39 |
* MinChen <chenm001@163.com> |
* MinChen <chenm001@163.com> |
40 |
* 14.04.2002 added FrameCodeB() |
* 14.04.2002 added FrameCodeB() |
41 |
* |
* |
42 |
* $Id: encoder.c,v 1.76.2.17 2002-11-11 20:53:34 Isibaar Exp $ |
* $Id: encoder.c,v 1.76.2.22 2002-11-27 11:50:33 suxen_drol Exp $ |
43 |
* |
* |
44 |
****************************************************************************/ |
****************************************************************************/ |
45 |
|
|
327 |
pEnc->global = pParam->global; |
pEnc->global = pParam->global; |
328 |
pEnc->mbParam.max_bframes = pParam->max_bframes; |
pEnc->mbParam.max_bframes = pParam->max_bframes; |
329 |
pEnc->bquant_ratio = pParam->bquant_ratio; |
pEnc->bquant_ratio = pParam->bquant_ratio; |
330 |
|
pEnc->bquant_offset = pParam->bquant_offset; |
331 |
pEnc->frame_drop_ratio = pParam->frame_drop_ratio; |
pEnc->frame_drop_ratio = pParam->frame_drop_ratio; |
332 |
pEnc->bframes = NULL; |
pEnc->bframes = NULL; |
333 |
|
|
667 |
uint32_t bits, mode; |
uint32_t bits, mode; |
668 |
|
|
669 |
int input_valid = 1; |
int input_valid = 1; |
670 |
|
int bframes_count = 0; |
671 |
|
|
672 |
#ifdef _DEBUG_PSNR |
#ifdef _DEBUG_PSNR |
673 |
float psnr; |
float psnr; |
708 |
SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]); |
SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]); |
709 |
|
|
710 |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
711 |
|
bframes_count = 0; |
712 |
|
|
713 |
BitstreamPad(&bs); |
BitstreamPad(&bs); |
714 |
pFrame->length = BitstreamLength(&bs); |
pFrame->length = BitstreamLength(&bs); |
729 |
|
|
730 |
BitstreamPad(&bs); |
BitstreamPad(&bs); |
731 |
pFrame->length = BitstreamLength(&bs); |
pFrame->length = BitstreamLength(&bs); |
732 |
pFrame->intra = 0; |
pFrame->intra = 2; |
733 |
|
|
734 |
if (input_valid) |
if (input_valid) |
735 |
queue_image(pEnc, pFrame); |
queue_image(pEnc, pFrame); |
764 |
pEnc->current->seconds = tmp; |
pEnc->current->seconds = tmp; |
765 |
|
|
766 |
pFrame->length = BitstreamLength(&bs); |
pFrame->length = BitstreamLength(&bs); |
767 |
pFrame->intra = 0; |
pFrame->intra = 4; |
768 |
|
|
769 |
if (input_valid) |
if (input_valid) |
770 |
queue_image(pEnc, pFrame); |
queue_image(pEnc, pFrame); |
840 |
pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size); |
pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size); |
841 |
|
|
842 |
BitstreamPutBits(&bs, 0x7f, 8); |
BitstreamPutBits(&bs, 0x7f, 8); |
843 |
pFrame->intra = 0; |
pFrame->intra = 5; |
844 |
} |
} |
845 |
|
|
846 |
pFrame->length = BitstreamLength(&bs); |
pFrame->length = BitstreamLength(&bs); |
927 |
* ivop/pvop/bvop selection |
* ivop/pvop/bvop selection |
928 |
* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */ |
* %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% */ |
929 |
|
|
|
|
|
930 |
if (pEnc->iFrameNum == 0 || pFrame->intra == 1 || pEnc->bframenum_dx50bvop >= 0 || |
if (pEnc->iFrameNum == 0 || pFrame->intra == 1 || pEnc->bframenum_dx50bvop >= 0 || |
931 |
(pFrame->intra < 0 && pEnc->iMaxKeyInterval > 0 && |
(pFrame->intra < 0 && pEnc->iMaxKeyInterval > 0 && |
932 |
pEnc->iFrameNum >= pEnc->iMaxKeyInterval) |
pEnc->iFrameNum >= pEnc->iMaxKeyInterval) |
933 |
|| /*image_mad(&pEnc->reference->image, &pEnc->current->image, |
|| 2 == (mode = MEanalysis(&pEnc->reference->image, pEnc->current, |
934 |
pEnc->mbParam.edged_width, pEnc->mbParam.width, |
&pEnc->mbParam, pEnc->iMaxKeyInterval, |
935 |
pEnc->mbParam.height) > 30) {*/ |
(pFrame->intra < 0) ? pEnc->iFrameNum : 0, |
936 |
2 == (mode = MEanalysis(&pEnc->reference->image, &pEnc->current->image, |
bframes_count++))) { |
|
&pEnc->mbParam, pEnc->current->mbs, pEnc->current->fcode))) { |
|
937 |
|
|
938 |
/* |
/* |
939 |
* This will be coded as an Intra Frame |
* This will be coded as an Intra Frame |
940 |
*/ |
*/ |
941 |
|
if ((pEnc->current->global_flags & XVID_QUARTERPEL)) |
942 |
|
pEnc->mbParam.m_quarterpel = 1; |
943 |
|
else |
944 |
|
pEnc->mbParam.m_quarterpel = 0; |
945 |
|
|
946 |
|
if (pEnc->current->global_flags & XVID_MPEGQUANT) pEnc->mbParam.m_quant_type = MPEG4_QUANT; |
947 |
|
|
948 |
|
if ((pEnc->current->global_flags & XVID_CUSTOM_QMATRIX) > 0) { |
949 |
|
if (pFrame->quant_intra_matrix != NULL) |
950 |
|
set_intra_matrix(pFrame->quant_intra_matrix); |
951 |
|
if (pFrame->quant_inter_matrix != NULL) |
952 |
|
set_inter_matrix(pFrame->quant_inter_matrix); |
953 |
|
} |
954 |
|
|
955 |
|
|
956 |
DPRINTF(DPRINTF_DEBUG,"*** IFRAME bf: head=%i tail=%i queue: head=%i tail=%i size=%i", |
DPRINTF(DPRINTF_DEBUG,"*** IFRAME bf: head=%i tail=%i queue: head=%i tail=%i size=%i", |
957 |
pEnc->bframenum_head, pEnc->bframenum_tail, |
pEnc->bframenum_head, pEnc->bframenum_tail, |
973 |
image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 100, "DX50 BVOP->PVOP"); |
image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 100, "DX50 BVOP->PVOP"); |
974 |
} |
} |
975 |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
976 |
|
bframes_count = 0; |
977 |
|
|
978 |
pFrame->intra = 0; |
pFrame->intra = 0; |
979 |
|
|
980 |
} else { |
} else { |
981 |
|
|
982 |
FrameCodeI(pEnc, &bs, &bits); |
FrameCodeI(pEnc, &bs, &bits); |
983 |
|
bframes_count = 0; |
984 |
pFrame->intra = 1; |
pFrame->intra = 1; |
985 |
|
|
986 |
pEnc->bframenum_dx50bvop = -1; |
pEnc->bframenum_dx50bvop = -1; |
998 |
* NB : sequences like "IIBB" decode fine with msfdam but, |
* NB : sequences like "IIBB" decode fine with msfdam but, |
999 |
* go screwy with divx 5.00 |
* go screwy with divx 5.00 |
1000 |
*/ |
*/ |
1001 |
} else if ((pEnc->bframenum_tail >= pEnc->mbParam.max_bframes) || (mode != 0)) { |
} else if (pEnc->bframenum_tail >= pEnc->mbParam.max_bframes || mode != 0) { |
1002 |
|
// } else if (pFrame->intra == 0 || pEnc->bframenum_tail >= pEnc->mbParam.max_bframes || mode != 0) { |
1003 |
/* |
/* |
1004 |
* This will be coded as a Predicted Frame |
* This will be coded as a Predicted Frame |
1005 |
*/ |
*/ |
1013 |
} |
} |
1014 |
|
|
1015 |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
1016 |
|
bframes_count = 0; |
1017 |
pFrame->intra = 0; |
pFrame->intra = 0; |
1018 |
pEnc->flush_bframes = 1; |
pEnc->flush_bframes = 1; |
1019 |
|
|
1033 |
} |
} |
1034 |
|
|
1035 |
if (pFrame->bquant < 1) { |
if (pFrame->bquant < 1) { |
1036 |
pEnc->current->quant = |
pEnc->current->quant = ((((pEnc->reference->quant + pEnc->current->quant) * |
1037 |
((pEnc->reference->quant + |
pEnc->bquant_ratio) / 2) + pEnc->bquant_offset)/100; |
1038 |
pEnc->current->quant) * pEnc->bquant_ratio) / 200; |
|
1039 |
} else { |
} else { |
1040 |
pEnc->current->quant = pFrame->bquant; |
pEnc->current->quant = pFrame->bquant; |
1041 |
} |
} |
1042 |
|
|
1043 |
if (pEnc->current->quant < 1) |
if (pEnc->current->quant < 1) |
1044 |
pEnc->current->quant = 1; |
pEnc->current->quant = 1; |
1045 |
|
else if (pEnc->current->quant > 31) |
|
if (pEnc->current->quant > 31) |
|
1046 |
pEnc->current->quant = 31; |
pEnc->current->quant = 31; |
1047 |
|
|
|
|
|
1048 |
DPRINTF(DPRINTF_DEBUG,"*** BFRAME (store) bf: head=%i tail=%i queue: head=%i tail=%i size=%i quant=%i\n", |
DPRINTF(DPRINTF_DEBUG,"*** BFRAME (store) bf: head=%i tail=%i queue: head=%i tail=%i size=%i quant=%i\n", |
1049 |
pEnc->bframenum_head, pEnc->bframenum_tail, |
pEnc->bframenum_head, pEnc->bframenum_tail, |
1050 |
pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size,pEnc->current->quant); |
pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size,pEnc->current->quant); |
1051 |
|
|
|
|
|
|
|
|
1052 |
/* store frame into bframe buffer & swap ref back to current */ |
/* store frame into bframe buffer & swap ref back to current */ |
1053 |
SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]); |
SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]); |
1054 |
SWAP(pEnc->current, pEnc->reference); |
SWAP(pEnc->current, pEnc->reference); |
1055 |
|
|
1056 |
pEnc->bframenum_tail++; |
pEnc->bframenum_tail++; |
1057 |
|
|
1058 |
pFrame->intra = 0; |
// bframe report by koepi |
1059 |
|
pFrame->intra = 2; |
1060 |
pFrame->length = 0; |
pFrame->length = 0; |
1061 |
|
|
1062 |
input_valid = 0; |
input_valid = 0; |
1924 |
((int32_t)(pEnc->current->stamp - pEnc->reference->stamp)), // time_pp |
((int32_t)(pEnc->current->stamp - pEnc->reference->stamp)), // time_pp |
1925 |
pEnc->reference->mbs, f_ref, |
pEnc->reference->mbs, f_ref, |
1926 |
&pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv, |
&pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv, |
1927 |
pEnc->current->mbs, b_ref, &pEnc->vInterH, |
pEnc->current, b_ref, &pEnc->vInterH, |
1928 |
&pEnc->vInterV, &pEnc->vInterHV); |
&pEnc->vInterV, &pEnc->vInterHV); |
1929 |
|
|
1930 |
|
|