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.15 2002-11-02 16:12:27 chl Exp $ |
* $Id: encoder.c,v 1.76.2.21 2002-11-23 14:57:09 syskin Exp $ |
43 |
* |
* |
44 |
****************************************************************************/ |
****************************************************************************/ |
45 |
|
|
107 |
}; |
}; |
108 |
|
|
109 |
|
|
|
static void __inline |
|
|
image_null(IMAGE * image) |
|
|
{ |
|
|
image->y = image->u = image->v = NULL; |
|
|
} |
|
|
|
|
|
|
|
110 |
/***************************************************************************** |
/***************************************************************************** |
111 |
* Encoder creation |
* Encoder creation |
112 |
* |
* |
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 |
|
|
617 |
start_timer(); |
start_timer(); |
618 |
if (image_input |
if (image_input |
619 |
(&pEnc->queue[pEnc->queue_tail], pEnc->mbParam.width, pEnc->mbParam.height, |
(&pEnc->queue[pEnc->queue_tail], pEnc->mbParam.width, pEnc->mbParam.height, |
620 |
pEnc->mbParam.edged_width, pFrame->image, pFrame->colorspace)) |
pEnc->mbParam.edged_width, pFrame->image, pFrame->stride, pFrame->colorspace, pFrame->general & XVID_INTERLACING)) |
621 |
return; |
return; |
622 |
stop_conv_timer(); |
stop_conv_timer(); |
623 |
|
|
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); |
715 |
pFrame->intra = 0; |
pFrame->intra = 0; |
716 |
|
|
717 |
|
emms(); |
718 |
|
|
719 |
return XVID_ERR_OK; |
return XVID_ERR_OK; |
720 |
} |
} |
721 |
|
|
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); |
736 |
|
|
737 |
|
emms(); |
738 |
|
|
739 |
return XVID_ERR_OK; |
return XVID_ERR_OK; |
740 |
} |
} |
741 |
|
|
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); |
771 |
|
|
772 |
|
emms(); |
773 |
|
|
774 |
return XVID_ERR_OK; |
return XVID_ERR_OK; |
775 |
} |
} |
776 |
} |
} |
801 |
start_timer(); |
start_timer(); |
802 |
if (image_input |
if (image_input |
803 |
(&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height, |
(&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height, |
804 |
pEnc->mbParam.edged_width, pFrame->image, pFrame->colorspace)) |
pEnc->mbParam.edged_width, pFrame->image, pFrame->stride, pFrame->colorspace, pFrame->general & XVID_INTERLACING)) |
805 |
|
{ |
806 |
|
emms(); |
807 |
return XVID_ERR_FORMAT; |
return XVID_ERR_FORMAT; |
808 |
|
} |
809 |
stop_conv_timer(); |
stop_conv_timer(); |
810 |
|
|
811 |
// queue input frame, and dequue next image |
// queue input frame, and dequue next image |
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); |
847 |
|
emms(); |
848 |
return XVID_ERR_OK; |
return XVID_ERR_OK; |
849 |
} |
} |
850 |
|
|
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 |
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", |
949 |
pEnc->bframenum_head, pEnc->bframenum_tail, |
pEnc->bframenum_head, pEnc->bframenum_tail, |
965 |
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"); |
966 |
} |
} |
967 |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
968 |
|
bframes_count = 0; |
969 |
|
|
970 |
pFrame->intra = 0; |
pFrame->intra = 0; |
971 |
|
|
972 |
} else { |
} else { |
973 |
|
|
974 |
FrameCodeI(pEnc, &bs, &bits); |
FrameCodeI(pEnc, &bs, &bits); |
975 |
|
bframes_count = 0; |
976 |
pFrame->intra = 1; |
pFrame->intra = 1; |
977 |
|
|
978 |
pEnc->bframenum_dx50bvop = -1; |
pEnc->bframenum_dx50bvop = -1; |
991 |
* go screwy with divx 5.00 |
* go screwy with divx 5.00 |
992 |
*/ |
*/ |
993 |
} else if (pEnc->bframenum_tail >= pEnc->mbParam.max_bframes || mode != 0) { |
} else if (pEnc->bframenum_tail >= pEnc->mbParam.max_bframes || mode != 0) { |
994 |
|
// } else if (pFrame->intra == 0 || pEnc->bframenum_tail >= pEnc->mbParam.max_bframes || mode != 0) { |
995 |
/* |
/* |
996 |
* This will be coded as a Predicted Frame |
* This will be coded as a Predicted Frame |
997 |
*/ |
*/ |
1005 |
} |
} |
1006 |
|
|
1007 |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
FrameCodeP(pEnc, &bs, &bits, 1, 0); |
1008 |
|
bframes_count = 0; |
1009 |
pFrame->intra = 0; |
pFrame->intra = 0; |
1010 |
pEnc->flush_bframes = 1; |
pEnc->flush_bframes = 1; |
1011 |
|
|
1012 |
if ((pEnc->global & XVID_GLOBAL_PACKED)) { |
if ((pEnc->global & XVID_GLOBAL_PACKED) && (pEnc->bframenum_tail > 0)) { |
1013 |
BitstreamPadAlways(&bs); |
BitstreamPadAlways(&bs); |
1014 |
input_valid = 0; |
input_valid = 0; |
1015 |
goto ipvop_loop; |
goto ipvop_loop; |
1025 |
} |
} |
1026 |
|
|
1027 |
if (pFrame->bquant < 1) { |
if (pFrame->bquant < 1) { |
1028 |
pEnc->current->quant = |
pEnc->current->quant = ((((pEnc->reference->quant + pEnc->current->quant) * |
1029 |
((pEnc->reference->quant + |
pEnc->bquant_ratio) / 2) + pEnc->bquant_offset)/100; |
1030 |
pEnc->current->quant) * pEnc->bquant_ratio) / 200; |
|
1031 |
} else { |
} else { |
1032 |
pEnc->current->quant = pFrame->bquant; |
pEnc->current->quant = pFrame->bquant; |
1033 |
} |
} |
1034 |
|
|
1035 |
if (pEnc->current->quant < 1) |
if (pEnc->current->quant < 1) |
1036 |
pEnc->current->quant = 1; |
pEnc->current->quant = 1; |
1037 |
|
else if (pEnc->current->quant > 31) |
|
if (pEnc->current->quant > 31) |
|
1038 |
pEnc->current->quant = 31; |
pEnc->current->quant = 31; |
1039 |
|
|
|
|
|
1040 |
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", |
1041 |
pEnc->bframenum_head, pEnc->bframenum_tail, |
pEnc->bframenum_head, pEnc->bframenum_tail, |
1042 |
pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size,pEnc->current->quant); |
pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size,pEnc->current->quant); |
1043 |
|
|
|
|
|
|
|
|
1044 |
/* store frame into bframe buffer & swap ref back to current */ |
/* store frame into bframe buffer & swap ref back to current */ |
1045 |
SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]); |
SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]); |
1046 |
SWAP(pEnc->current, pEnc->reference); |
SWAP(pEnc->current, pEnc->reference); |
1047 |
|
|
1048 |
pEnc->bframenum_tail++; |
pEnc->bframenum_tail++; |
1049 |
|
|
1050 |
pFrame->intra = 0; |
// bframe report by koepi |
1051 |
|
pFrame->intra = 2; |
1052 |
pFrame->length = 0; |
pFrame->length = 0; |
1053 |
|
|
1054 |
input_valid = 0; |
input_valid = 0; |
1088 |
stop_global_timer(); |
stop_global_timer(); |
1089 |
write_timer(); |
write_timer(); |
1090 |
|
|
1091 |
|
emms(); |
1092 |
return XVID_ERR_OK; |
return XVID_ERR_OK; |
1093 |
} |
} |
1094 |
|
|
1143 |
start_timer(); |
start_timer(); |
1144 |
if (image_input |
if (image_input |
1145 |
(&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height, |
(&pEnc->current->image, pEnc->mbParam.width, pEnc->mbParam.height, |
1146 |
pEnc->mbParam.edged_width, pFrame->image, pFrame->colorspace) < 0) |
pEnc->mbParam.edged_width, pFrame->image, pFrame->stride, pFrame->colorspace, pFrame->general & XVID_INTERLACING) < 0) |
1147 |
return XVID_ERR_FORMAT; |
return XVID_ERR_FORMAT; |
1148 |
stop_conv_timer(); |
stop_conv_timer(); |
1149 |
|
|
1634 |
|
|
1635 |
if (pEnc->current->global_flags & XVID_GMC) { |
if (pEnc->current->global_flags & XVID_GMC) { |
1636 |
// printf("Global Motion = %d %d quarterpel=%d\n", pEnc->current->GMC_MV.x, pEnc->current->GMC_MV.y,pEnc->current->quarterpel); |
// printf("Global Motion = %d %d quarterpel=%d\n", pEnc->current->GMC_MV.x, pEnc->current->GMC_MV.y,pEnc->current->quarterpel); |
1637 |
|
DPRINTF(DPRINTF_HEADER, "Global Motion = %d %d quarterpel=%d\n", pEnc->current->GMC_MV.x, pEnc->current->GMC_MV.y,pEnc->current->quarterpel); |
1638 |
pEnc->current->coding_type = S_VOP; |
pEnc->current->coding_type = S_VOP; |
1639 |
} else |
} else |
1640 |
pEnc->current->coding_type = P_VOP; |
pEnc->current->coding_type = P_VOP; |
1916 |
((int32_t)(pEnc->current->stamp - pEnc->reference->stamp)), // time_pp |
((int32_t)(pEnc->current->stamp - pEnc->reference->stamp)), // time_pp |
1917 |
pEnc->reference->mbs, f_ref, |
pEnc->reference->mbs, f_ref, |
1918 |
&pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv, |
&pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv, |
1919 |
pEnc->current->mbs, b_ref, &pEnc->vInterH, |
pEnc->current, b_ref, &pEnc->vInterH, |
1920 |
&pEnc->vInterV, &pEnc->vInterHV); |
&pEnc->vInterV, &pEnc->vInterHV); |
1921 |
|
|
1922 |
|
|