[cvs] / xvidcore / src / plugins / plugin_2pass1.c Repository:
ViewVC logotype

Annotation of /xvidcore/src/plugins/plugin_2pass1.c

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.1.2.7 - (view) (download)

1 : suxen_drol 1.1.2.1 /******************************************************************************
2 :     *
3 : edgomez 1.1.2.5 * XviD Bit Rate Controller Library
4 :     * - VBR 2 pass bitrate controler implementation -
5 : suxen_drol 1.1.2.1 *
6 : edgomez 1.1.2.5 * Copyright (C) 2002-2003 Edouard Gomez <ed.gomez@free.fr>
7 : suxen_drol 1.1.2.1 *
8 : edgomez 1.1.2.5 * The curve treatment algorithm is the one implemented by Foxer <email?> and
9 :     * Dirk Knop <dknop@gwdg.de> for the XviD vfw dynamic library.
10 : suxen_drol 1.1.2.1 *
11 : edgomez 1.1.2.5 * This program is free software; you can redistribute it and/or modify
12 :     * it under the terms of the GNU General Public License as published by
13 :     * the Free Software Foundation; either version 2 of the License, or
14 :     * (at your option) any later version.
15 : suxen_drol 1.1.2.1 *
16 : edgomez 1.1.2.5 * This program is distributed in the hope that it will be useful,
17 :     * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 :     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 :     * GNU General Public License for more details.
20 : suxen_drol 1.1.2.1 *
21 : edgomez 1.1.2.5 * You should have received a copy of the GNU General Public License
22 :     * along with this program; if not, write to the Free Software
23 :     * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
24 : suxen_drol 1.1.2.1 *
25 : edgomez 1.1.2.5 * $Id$
26 : suxen_drol 1.1.2.1 *
27 :     *****************************************************************************/
28 :    
29 :     #include <stdio.h>
30 :    
31 :     #include "../xvid.h"
32 :     #include "../image/image.h"
33 :    
34 :    
35 :     /* context struct */
36 :     typedef struct
37 :     {
38 :     FILE * stat_file;
39 : suxen_drol 1.1.2.3
40 :     double fq_error;
41 : suxen_drol 1.1.2.1 } rc_2pass1_t;
42 :    
43 :    
44 :    
45 :     static int rc_2pass1_create(xvid_plg_create_t * create, rc_2pass1_t ** handle)
46 :     {
47 :     xvid_plugin_2pass1_t * param = (xvid_plugin_2pass1_t *)create->param;
48 :     rc_2pass1_t * rc;
49 :    
50 :     /* check filename */
51 : edgomez 1.1.2.6 if ((param->filename == NULL) ||
52 :     (param->filename != NULL && param->filename[0] == '\0'))
53 : suxen_drol 1.1.2.1 return XVID_ERR_FAIL;
54 :    
55 :     /* allocate context struct */
56 :     if((rc = malloc(sizeof(rc_2pass1_t))) == NULL)
57 :     return(XVID_ERR_MEMORY);
58 :    
59 : edgomez 1.1.2.7 /* Initialize safe defaults for 2pass 1 */
60 : suxen_drol 1.1.2.1 rc->stat_file = NULL;
61 :    
62 :     /* Open the 1st pass file */
63 :     if((rc->stat_file = fopen(param->filename, "w+")) == NULL)
64 :     return(XVID_ERR_FAIL);
65 : suxen_drol 1.1.2.3
66 : suxen_drol 1.1.2.1 /*
67 :     * The File Header
68 :     */
69 : edgomez 1.1.2.5 #if 0
70 :     fprintf(rc->stat_file, "# XviD 2pass stat file\n");
71 : suxen_drol 1.1.2.1 fprintf(rc->stat_file, "version %i.%i.%i\n",XVID_MAJOR(XVID_VERSION), XVID_MINOR(XVID_VERSION), XVID_PATCH(XVID_VERSION));
72 :     fprintf(rc->stat_file, "start\n");
73 : edgomez 1.1.2.5 fprintf(rc->stat_file, "type quantizer length kblocks mblocks ublocks\n");
74 :     #endif
75 : edgomez 1.1.2.7
76 : suxen_drol 1.1.2.3 rc->fq_error = 0;
77 : suxen_drol 1.1.2.1
78 :     *handle = rc;
79 :     return(0);
80 :     }
81 :    
82 :    
83 :     static int rc_2pass1_destroy(rc_2pass1_t * rc, xvid_plg_destroy_t * destroy)
84 :     {
85 :     fclose(rc->stat_file);
86 :    
87 :     free(rc);
88 :     return(0);
89 :     }
90 :    
91 :    
92 :     static int rc_2pass1_before(rc_2pass1_t * rc, xvid_plg_data_t * data)
93 :     {
94 : suxen_drol 1.1.2.3 if (data->quant <= 0) {
95 :     if (data->zone && data->zone->mode == XVID_ZONE_QUANT) {
96 :     rc->fq_error += (double)data->zone->increment / (double)data->zone->base;
97 :     data->quant = (int)rc->fq_error;
98 :     rc->fq_error -= data->quant;
99 : edgomez 1.1.2.7
100 : suxen_drol 1.1.2.3 }else {
101 :     data->quant = 2;
102 :     }
103 :     }
104 : suxen_drol 1.1.2.1 return 0;
105 :     }
106 :    
107 :    
108 :     static int rc_2pass1_after(rc_2pass1_t * rc, xvid_plg_data_t * data)
109 :     {
110 :     char type;
111 :    
112 :     /* Frame type in ascii I/P/B */
113 :     switch(data->type) {
114 :     case XVID_TYPE_IVOP:
115 :     type = 'i';
116 :     break;
117 :     case XVID_TYPE_PVOP:
118 :     type = 'p';
119 :     break;
120 :     case XVID_TYPE_BVOP:
121 :     type = 'b';
122 :     break;
123 :     case XVID_TYPE_SVOP:
124 :     type = 's';
125 :     break;
126 :     default: /* Should not go here */
127 :     return(XVID_ERR_FAIL);
128 :     }
129 :    
130 :     /* write the resulting statistics */
131 :    
132 :     fprintf(rc->stat_file, "%c %d %d %d %d %d\n",
133 :     type,
134 :     data->quant,
135 :     data->kblks,
136 :     data->mblks,
137 : suxen_drol 1.1.2.2 data->ublks,
138 :     data->length);
139 : suxen_drol 1.1.2.1
140 :     return(0);
141 :     }
142 :    
143 :    
144 :    
145 :     int xvid_plugin_2pass1(void * handle, int opt, void * param1, void * param2)
146 :     {
147 :     switch(opt)
148 :     {
149 :     case XVID_PLG_INFO :
150 :     return 0;
151 :    
152 :     case XVID_PLG_CREATE :
153 :     return rc_2pass1_create((xvid_plg_create_t*)param1, param2);
154 :    
155 :     case XVID_PLG_DESTROY :
156 :     return rc_2pass1_destroy((rc_2pass1_t*)handle, (xvid_plg_destroy_t*)param1);
157 :    
158 :     case XVID_PLG_BEFORE :
159 :     return rc_2pass1_before((rc_2pass1_t*)handle, (xvid_plg_data_t*)param1);
160 :    
161 :     case XVID_PLG_AFTER :
162 :     return rc_2pass1_after((rc_2pass1_t*)handle, (xvid_plg_data_t*)param1);
163 :     }
164 :    
165 :     return XVID_ERR_FAIL;
166 :     }

No admin address has been configured
ViewVC Help
Powered by ViewVC 1.0.4