229 |
|
|
230 |
pEnc->bitrate = pParam->rc_bitrate; |
pEnc->bitrate = pParam->rc_bitrate; |
231 |
|
|
232 |
pEnc->iFrameNum = 0; |
pEnc->iFrameNum = -1; |
233 |
pEnc->mbParam.iMaxKeyInterval = pParam->max_key_interval; |
pEnc->mbParam.iMaxKeyInterval = pParam->max_key_interval; |
234 |
|
|
235 |
/* try to allocate frame memory */ |
/* try to allocate frame memory */ |
254 |
|
|
255 |
/* try to allocate image memory */ |
/* try to allocate image memory */ |
256 |
|
|
257 |
#ifdef _DEBUG_PSNR |
if (pParam->global & XVID_GLOBAL_EXTRASTATS) |
258 |
image_null(&pEnc->sOriginal); |
image_null(&pEnc->sOriginal); |
|
#endif |
|
259 |
|
|
260 |
image_null(&pEnc->f_refh); |
image_null(&pEnc->f_refh); |
261 |
image_null(&pEnc->f_refv); |
image_null(&pEnc->f_refv); |
269 |
image_null(&pEnc->vInterHV); |
image_null(&pEnc->vInterHV); |
270 |
image_null(&pEnc->vInterHVf); |
image_null(&pEnc->vInterHVf); |
271 |
|
|
272 |
#ifdef _DEBUG_PSNR |
if (pParam->global & XVID_GLOBAL_EXTRASTATS) |
273 |
if (image_create |
{ if (image_create |
274 |
(&pEnc->sOriginal, pEnc->mbParam.edged_width, |
(&pEnc->sOriginal, pEnc->mbParam.edged_width, |
275 |
pEnc->mbParam.edged_height) < 0) |
pEnc->mbParam.edged_height) < 0) |
276 |
goto xvid_err_memory3; |
goto xvid_err_memory3; |
277 |
#endif |
} |
278 |
|
|
279 |
if (image_create |
if (image_create |
280 |
(&pEnc->f_refh, pEnc->mbParam.edged_width, |
(&pEnc->f_refh, pEnc->mbParam.edged_width, |
325 |
goto xvid_err_memory3; |
goto xvid_err_memory3; |
326 |
|
|
327 |
|
|
|
/* B Frames specific init */ |
|
328 |
|
|
329 |
pEnc->mbParam.global = pParam->global; |
pEnc->mbParam.global = pParam->global; |
330 |
|
|
331 |
|
/* B Frames specific init */ |
332 |
pEnc->mbParam.max_bframes = pParam->max_bframes; |
pEnc->mbParam.max_bframes = pParam->max_bframes; |
333 |
pEnc->mbParam.bquant_ratio = pParam->bquant_ratio; |
pEnc->mbParam.bquant_ratio = pParam->bquant_ratio; |
334 |
pEnc->mbParam.bquant_offset = pParam->bquant_offset; |
pEnc->mbParam.bquant_offset = pParam->bquant_offset; |
464 |
} |
} |
465 |
|
|
466 |
xvid_err_memory3: |
xvid_err_memory3: |
467 |
#ifdef _DEBUG_PSNR |
|
468 |
image_destroy(&pEnc->sOriginal, pEnc->mbParam.edged_width, |
if (pEnc->mbParam.global & XVID_GLOBAL_EXTRASTATS) |
469 |
|
{ image_destroy(&pEnc->sOriginal, pEnc->mbParam.edged_width, |
470 |
pEnc->mbParam.edged_height); |
pEnc->mbParam.edged_height); |
471 |
#endif |
} |
472 |
|
|
473 |
image_destroy(&pEnc->f_refh, pEnc->mbParam.edged_width, |
image_destroy(&pEnc->f_refh, pEnc->mbParam.edged_width, |
474 |
pEnc->mbParam.edged_height); |
pEnc->mbParam.edged_height); |
584 |
image_destroy(&pEnc->f_refhv, pEnc->mbParam.edged_width, |
image_destroy(&pEnc->f_refhv, pEnc->mbParam.edged_width, |
585 |
pEnc->mbParam.edged_height); |
pEnc->mbParam.edged_height); |
586 |
|
|
587 |
#ifdef _DEBUG_PSNR |
if (pEnc->mbParam.global & XVID_GLOBAL_EXTRASTATS) |
588 |
image_destroy(&pEnc->sOriginal, pEnc->mbParam.edged_width, |
{ image_destroy(&pEnc->sOriginal, pEnc->mbParam.edged_width, |
589 |
pEnc->mbParam.edged_height); |
pEnc->mbParam.edged_height); |
590 |
#endif |
} |
591 |
|
|
592 |
/* Encoder structure */ |
/* Encoder structure */ |
593 |
|
|
662 |
if (intra < 0) return -1; |
if (intra < 0) return -1; |
663 |
if (intra == 1) return I_VOP; |
if (intra == 1) return I_VOP; |
664 |
if (intra == 2) return B_VOP; |
if (intra == 2) return B_VOP; |
|
if (intra == 3) return S_VOP; |
|
665 |
|
|
666 |
return P_VOP; |
return P_VOP; |
667 |
} |
} |
690 |
int input_valid = 1; |
int input_valid = 1; |
691 |
int bframes_count = 0; |
int bframes_count = 0; |
692 |
|
|
|
#ifdef _DEBUG_PSNR |
|
693 |
float psnr; |
float psnr; |
694 |
char temp[128]; |
char temp[128]; |
|
#endif |
|
695 |
|
|
696 |
ENC_CHECK(pEnc); |
ENC_CHECK(pEnc); |
697 |
ENC_CHECK(pFrame); |
ENC_CHECK(pFrame); |
731 |
|
|
732 |
BitstreamPadAlways(&bs); |
BitstreamPadAlways(&bs); |
733 |
pFrame->length = BitstreamLength(&bs); |
pFrame->length = BitstreamLength(&bs); |
|
if(pEnc->current->coding_type == P_VOP) |
|
734 |
pFrame->intra = 0; |
pFrame->intra = 0; |
735 |
else |
|
|
pFrame->intra = 3; |
|
736 |
|
|
737 |
emms(); |
emms(); |
738 |
|
|
954 |
|
|
955 |
inc_frame_num(pEnc); |
inc_frame_num(pEnc); |
956 |
|
|
957 |
#ifdef _DEBUG_PSNR |
if (pFrame->general & XVID_EXTRASTATS) |
958 |
image_copy(&pEnc->sOriginal, &pEnc->current->image, |
{ image_copy(&pEnc->sOriginal, &pEnc->current->image, |
959 |
pEnc->mbParam.edged_width, pEnc->mbParam.height); |
pEnc->mbParam.edged_width, pEnc->mbParam.height); |
960 |
#endif |
} |
961 |
|
|
962 |
emms(); |
emms(); |
963 |
|
|
1059 |
} |
} |
1060 |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
1061 |
bframes_count = 0; |
bframes_count = 0; |
|
|
|
|
if(pEnc->current->coding_type == P_VOP) |
|
1062 |
pFrame->intra = 0; |
pFrame->intra = 0; |
|
else |
|
|
pFrame->intra = 3; |
|
|
|
|
1063 |
|
|
1064 |
} else { |
} else { |
1065 |
|
|
1097 |
|
|
1098 |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
1099 |
bframes_count = 0; |
bframes_count = 0; |
|
if(pEnc->current->coding_type == P_VOP) |
|
1100 |
pFrame->intra = 0; |
pFrame->intra = 0; |
|
else |
|
|
pFrame->intra = 3; |
|
1101 |
pEnc->flush_bframes = 1; |
pEnc->flush_bframes = 1; |
1102 |
|
|
1103 |
if ((pEnc->mbParam.global & XVID_GLOBAL_PACKED) && (pEnc->bframenum_tail > 0)) { |
if ((pEnc->mbParam.global & XVID_GLOBAL_PACKED) && (pEnc->bframenum_tail > 0)) { |
1155 |
pResult->kblks = pEnc->current->sStat.kblks; |
pResult->kblks = pEnc->current->sStat.kblks; |
1156 |
pResult->mblks = pEnc->current->sStat.mblks; |
pResult->mblks = pEnc->current->sStat.mblks; |
1157 |
pResult->ublks = pEnc->current->sStat.ublks; |
pResult->ublks = pEnc->current->sStat.ublks; |
|
} |
|
1158 |
|
|
1159 |
emms(); |
if (pFrame->general & XVID_EXTRASTATS) |
1160 |
|
{ pResult->sse_y = |
1161 |
#ifdef _DEBUG_PSNR |
plane_sse( pEnc->sOriginal.y, pEnc->current->image.y, |
|
psnr = |
|
|
image_psnr(&pEnc->sOriginal, &pEnc->current->image, |
|
1162 |
pEnc->mbParam.edged_width, pEnc->mbParam.width, |
pEnc->mbParam.edged_width, pEnc->mbParam.width, |
1163 |
pEnc->mbParam.height); |
pEnc->mbParam.height); |
1164 |
|
|
1165 |
printf("PSNR: %f\n", psnr); |
pResult->sse_u = |
1166 |
// DEBUG(temp); |
plane_sse( pEnc->sOriginal.u, pEnc->current->image.u, |
1167 |
#endif |
pEnc->mbParam.edged_width/2, pEnc->mbParam.width/2, |
1168 |
|
pEnc->mbParam.height/2); |
1169 |
|
|
1170 |
|
pResult->sse_v = |
1171 |
|
plane_sse( pEnc->sOriginal.v, pEnc->current->image.v, |
1172 |
|
pEnc->mbParam.edged_width/2, pEnc->mbParam.width/2, |
1173 |
|
pEnc->mbParam.height/2); |
1174 |
|
} |
1175 |
|
} |
1176 |
|
|
1177 |
|
emms(); |
1178 |
|
|
1179 |
if (pFrame->quant == 0) { |
if (pFrame->quant == 0) { |
1180 |
RateControlUpdate(&pEnc->rate_control, pEnc->current->quant, |
RateControlUpdate(&pEnc->rate_control, pEnc->current->quant, |
1209 |
uint32_t bits; |
uint32_t bits; |
1210 |
uint16_t write_vol_header = 0; |
uint16_t write_vol_header = 0; |
1211 |
|
|
|
#ifdef _DEBUG_PSNR |
|
1212 |
float psnr; |
float psnr; |
1213 |
uint8_t temp[128]; |
uint8_t temp[128]; |
|
#endif |
|
1214 |
|
|
1215 |
start_global_timer(); |
start_global_timer(); |
1216 |
|
|
1241 |
return XVID_ERR_FORMAT; |
return XVID_ERR_FORMAT; |
1242 |
stop_conv_timer(); |
stop_conv_timer(); |
1243 |
|
|
1244 |
#ifdef _DEBUG_PSNR |
if (pFrame->general & XVID_EXTRASTATS) |
1245 |
image_copy(&pEnc->sOriginal, &pEnc->current->image, |
{ image_copy(&pEnc->sOriginal, &pEnc->current->image, |
1246 |
pEnc->mbParam.edged_width, pEnc->mbParam.height); |
pEnc->mbParam.edged_width, pEnc->mbParam.height); |
1247 |
#endif |
} |
1248 |
|
|
1249 |
emms(); |
emms(); |
1250 |
|
|
1356 |
RateControlUpdate(&pEnc->rate_control, pEnc->current->quant, |
RateControlUpdate(&pEnc->rate_control, pEnc->current->quant, |
1357 |
pFrame->length, pFrame->intra); |
pFrame->length, pFrame->intra); |
1358 |
} |
} |
1359 |
#ifdef _DEBUG_PSNR |
if (pFrame->general & XVID_EXTRASTATS) |
1360 |
|
{ |
1361 |
psnr = |
psnr = |
1362 |
image_psnr(&pEnc->sOriginal, &pEnc->current->image, |
image_psnr(&pEnc->sOriginal, &pEnc->current->image, |
1363 |
pEnc->mbParam.edged_width, pEnc->mbParam.width, |
pEnc->mbParam.edged_width, pEnc->mbParam.width, |
1364 |
pEnc->mbParam.height); |
pEnc->mbParam.height); |
1365 |
|
|
1366 |
snprintf(temp, 127, "PSNR: %f\n", psnr); |
snprintf(temp, 127, "PSNR: %f\n", psnr); |
1367 |
// DEBUG(temp); |
} |
|
#endif |
|
1368 |
|
|
1369 |
pEnc->iFrameNum++; |
pEnc->iFrameNum++; |
1370 |
|
|
1828 |
pEnc->vGMC.y + 16*y*pEnc->mbParam.edged_width + 16*x, |
pEnc->vGMC.y + 16*y*pEnc->mbParam.edged_width + 16*x, |
1829 |
pEnc->mbParam.edged_width, 65536); |
pEnc->mbParam.edged_width, 65536); |
1830 |
|
|
1831 |
|
if (pEnc->current->motion_flags & PMV_CHROMA16) { |
1832 |
|
iSAD += sad8(pEnc->current->image.u + 8*y*(pEnc->mbParam.edged_width/2) + 8*x, |
1833 |
|
pEnc->vGMC.u + 8*y*(pEnc->mbParam.edged_width/2) + 8*x, pEnc->mbParam.edged_width/2); |
1834 |
|
|
1835 |
|
iSAD += sad8(pEnc->current->image.v + 8*y*(pEnc->mbParam.edged_width/2) + 8*x, |
1836 |
|
pEnc->vGMC.v + 8*y*(pEnc->mbParam.edged_width/2) + 8*x, pEnc->mbParam.edged_width/2); |
1837 |
|
} |
1838 |
|
|
1839 |
if (iSAD <= pMB->sad16) { /* mode decision GMC */ |
if (iSAD <= pMB->sad16) { /* mode decision GMC */ |
1840 |
|
|
1841 |
if (pEnc->mbParam.m_quarterpel) |
if (pEnc->mbParam.m_quarterpel) |