32 |
* |
* |
33 |
* History |
* History |
34 |
* |
* |
35 |
|
* 10.07.2002 added BFRAMES_DEC_DEBUG support |
36 |
|
* MinChen <chenm001@163.com> |
37 |
* 20.06.2002 bframe patch |
* 20.06.2002 bframe patch |
38 |
* 08.05.2002 fix some problem in DEBUG mode; |
* 08.05.2002 fix some problem in DEBUG mode; |
39 |
* MinChen <chenm001@163.com> |
* MinChen <chenm001@163.com> |
43 |
* |
* |
44 |
****************************************************************************/ |
****************************************************************************/ |
45 |
|
|
46 |
|
|
47 |
#include <stdlib.h> |
#include <stdlib.h> |
48 |
#include <stdio.h> |
#include <stdio.h> |
49 |
#include <math.h> |
#include <math.h> |
210 |
|
|
211 |
/* 1 keyframe each 10 seconds */ |
/* 1 keyframe each 10 seconds */ |
212 |
|
|
213 |
if (pParam->max_key_interval == 0) |
if (pParam->max_key_interval <= 0) |
214 |
pParam->max_key_interval = 10 * pParam->fincr / pParam->fbase; |
pParam->max_key_interval = 10 * pParam->fincr / pParam->fbase; |
215 |
|
|
|
|
|
216 |
pEnc = (Encoder *) xvid_malloc(sizeof(Encoder), CACHE_LINE); |
pEnc = (Encoder *) xvid_malloc(sizeof(Encoder), CACHE_LINE); |
217 |
if (pEnc == NULL) |
if (pEnc == NULL) |
218 |
return XVID_ERR_MEMORY; |
return XVID_ERR_MEMORY; |
237 |
|
|
238 |
pEnc->mbParam.m_quant_type = H263_QUANT; |
pEnc->mbParam.m_quant_type = H263_QUANT; |
239 |
|
|
240 |
|
#ifdef _SMP |
241 |
|
pEnc->mbParam.num_threads = MIN(pParam->num_threads, MAXNUMTHREADS); |
242 |
|
#endif |
243 |
|
|
244 |
pEnc->sStat.fMvPrevSigma = -1; |
pEnc->sStat.fMvPrevSigma = -1; |
245 |
|
|
246 |
/* Fill rate control parameters */ |
/* Fill rate control parameters */ |
345 |
pEnc->global = pParam->global; |
pEnc->global = pParam->global; |
346 |
pEnc->mbParam.max_bframes = pParam->max_bframes; |
pEnc->mbParam.max_bframes = pParam->max_bframes; |
347 |
pEnc->bquant_ratio = pParam->bquant_ratio; |
pEnc->bquant_ratio = pParam->bquant_ratio; |
348 |
|
pEnc->frame_drop_ratio = pParam->frame_drop_ratio; |
349 |
pEnc->bframes = NULL; |
pEnc->bframes = NULL; |
350 |
|
|
351 |
if (pEnc->mbParam.max_bframes > 0) { |
if (pEnc->mbParam.max_bframes > 0) { |
423 |
pEnc->mbParam.m_seconds = 0; |
pEnc->mbParam.m_seconds = 0; |
424 |
pEnc->mbParam.m_ticks = 0; |
pEnc->mbParam.m_ticks = 0; |
425 |
pEnc->m_framenum = 0; |
pEnc->m_framenum = 0; |
426 |
|
pEnc->last_pframe = 1; |
427 |
#endif |
#endif |
428 |
|
|
429 |
pParam->handle = (void *) pEnc; |
pParam->handle = (void *) pEnc; |
854 |
else |
else |
855 |
pEnc->current->quant = pFrame->quant; |
pEnc->current->quant = pFrame->quant; |
856 |
|
|
857 |
if (pEnc->current->quant < 1) |
/* if (pEnc->current->quant < 1) |
858 |
pEnc->current->quant = 1; |
pEnc->current->quant = 1; |
859 |
|
|
860 |
if (pEnc->current->quant > 31) |
if (pEnc->current->quant > 31) |
861 |
pEnc->current->quant = 31; |
pEnc->current->quant = 31; |
862 |
|
*/ |
863 |
pEnc->current->global_flags = pFrame->general; |
pEnc->current->global_flags = pFrame->general; |
864 |
pEnc->current->motion_flags = pFrame->motion; |
pEnc->current->motion_flags = pFrame->motion; |
865 |
|
|
1006 |
* This will be coded as a Bidirectional Frame |
* This will be coded as a Bidirectional Frame |
1007 |
*/ |
*/ |
1008 |
|
|
|
DPRINTF(DPRINTF_DEBUG,"*** BFRAME (store) bf: head=%i tail=%i queue: head=%i tail=%i size=%i", |
|
|
pEnc->bframenum_head, pEnc->bframenum_tail, |
|
|
pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size); |
|
|
|
|
1009 |
if ((pEnc->global & XVID_GLOBAL_DEBUG)) { |
if ((pEnc->global & XVID_GLOBAL_DEBUG)) { |
1010 |
image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 200, "BVOP"); |
image_printf(&pEnc->current->image, pEnc->mbParam.edged_width, pEnc->mbParam.height, 5, 200, "BVOP"); |
1011 |
} |
} |
1017 |
} else { |
} else { |
1018 |
pEnc->current->quant = pFrame->bquant; |
pEnc->current->quant = pFrame->bquant; |
1019 |
} |
} |
1020 |
|
if (pEnc->current->quant < 1) |
1021 |
|
pEnc->current->quant = 1; |
1022 |
|
|
1023 |
|
if (pEnc->current->quant > 31) |
1024 |
|
pEnc->current->quant = 31; |
1025 |
|
|
1026 |
|
|
1027 |
|
DPRINTF(DPRINTF_DEBUG,"*** BFRAME (store) bf: head=%i tail=%i queue: head=%i tail=%i size=%i quant=%i\n", |
1028 |
|
pEnc->bframenum_head, pEnc->bframenum_tail, |
1029 |
|
pEnc->queue_head, pEnc->queue_tail, pEnc->queue_size,pEnc->current->quant); |
1030 |
|
|
1031 |
|
|
1032 |
|
|
1033 |
/* store frame into bframe buffer & swap ref back to current */ |
/* store frame into bframe buffer & swap ref back to current */ |
1034 |
SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]); |
SWAP(pEnc->current, pEnc->bframes[pEnc->bframenum_tail]); |
1609 |
} else { |
} else { |
1610 |
|
|
1611 |
#ifdef _SMP |
#ifdef _SMP |
1612 |
if (NUMTHREADS > 1) |
if (pEnc->mbParam.num_threads > 1) |
1613 |
bIntra = |
bIntra = |
1614 |
SMP_MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference, |
SMP_MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference, |
1615 |
&pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV, |
&pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV, |
1616 |
iLimit); |
iLimit); |
1617 |
else |
else |
1618 |
#endif |
#endif |
|
|
|
1619 |
bIntra = |
bIntra = |
1620 |
MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference, |
MotionEstimation(&pEnc->mbParam, pEnc->current, pEnc->reference, |
1621 |
&pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV, |
&pEnc->vInterH, &pEnc->vInterV, &pEnc->vInterHV, |
1622 |
iLimit); |
iLimit); |
1623 |
|
|
|
|
|
1624 |
} |
} |
1625 |
stop_motion_timer(); |
stop_motion_timer(); |
1626 |
|
|
1732 |
|
|
1733 |
pEnc->sStat.fMvPrevSigma = fSigma; |
pEnc->sStat.fMvPrevSigma = fSigma; |
1734 |
|
|
1735 |
|
#ifdef BFRAMES |
1736 |
|
/* frame drop code */ |
1737 |
|
// DPRINTF(DPRINTF_DEBUG, "kmu %i %i %i", pEnc->sStat.kblks, pEnc->sStat.mblks, pEnc->sStat.ublks); |
1738 |
|
if (pEnc->sStat.kblks + pEnc->sStat.mblks <= |
1739 |
|
(pEnc->frame_drop_ratio * pEnc->mbParam.mb_width * pEnc->mbParam.mb_height) / 100) |
1740 |
|
{ |
1741 |
|
pEnc->sStat.kblks = pEnc->sStat.mblks = 0; |
1742 |
|
pEnc->sStat.ublks = pEnc->mbParam.mb_width * pEnc->mbParam.mb_height; |
1743 |
|
|
1744 |
|
BitstreamReset(bs); |
1745 |
|
BitstreamWriteVopHeader(bs, &pEnc->mbParam, pEnc->current, 0); |
1746 |
|
|
1747 |
|
// copy reference frame details into the current frame |
1748 |
|
pEnc->current->quant = pEnc->reference->quant; |
1749 |
|
pEnc->current->motion_flags = pEnc->reference->motion_flags; |
1750 |
|
pEnc->current->rounding_type = pEnc->reference->rounding_type; |
1751 |
|
pEnc->current->fcode = pEnc->reference->fcode; |
1752 |
|
pEnc->current->bcode = pEnc->reference->bcode; |
1753 |
|
image_copy(&pEnc->current->image, &pEnc->reference->image, pEnc->mbParam.edged_width, pEnc->mbParam.height); |
1754 |
|
memcpy(pEnc->current->mbs, pEnc->reference->mbs, sizeof(MACROBLOCK) * pEnc->mbParam.mb_width * pEnc->mbParam.mb_height); |
1755 |
|
|
1756 |
|
} |
1757 |
|
#endif |
1758 |
|
|
1759 |
*pBits = BitstreamPos(bs) - *pBits; |
*pBits = BitstreamPos(bs) - *pBits; |
1760 |
|
|
1761 |
|
#ifdef BFRAMES |
1762 |
|
pEnc->time_pp = ((int32_t)pEnc->mbParam.fbase - (int32_t)pEnc->last_pframe + (int32_t)pEnc->mbParam.m_ticks) % (int32_t)pEnc->mbParam.fbase; |
1763 |
|
pEnc->last_pframe = pEnc->mbParam.m_ticks; |
1764 |
|
#endif |
1765 |
|
|
1766 |
return 0; // inter |
return 0; // inter |
1767 |
} |
} |
1768 |
|
|
1783 |
IMAGE *f_ref = &pEnc->reference->image; |
IMAGE *f_ref = &pEnc->reference->image; |
1784 |
IMAGE *b_ref = &pEnc->current->image; |
IMAGE *b_ref = &pEnc->current->image; |
1785 |
|
|
1786 |
|
#ifdef BFRAMES_DEC_DEBUG |
1787 |
|
FILE *fp; |
1788 |
|
static char first=0; |
1789 |
|
#define BFRAME_DEBUG if (!first && fp){ \ |
1790 |
|
fprintf(fp,"Y=%3d X=%3d MB=%2d CBP=%02X\n",y,x,mb->mode,mb->cbp); \ |
1791 |
|
} |
1792 |
|
|
1793 |
|
if (!first){ |
1794 |
|
fp=fopen("C:\\XVIDDBGE.TXT","w"); |
1795 |
|
} |
1796 |
|
#endif |
1797 |
|
|
1798 |
// forward |
// forward |
1799 |
image_setedges(f_ref, pEnc->mbParam.edged_width, |
image_setedges(f_ref, pEnc->mbParam.edged_width, |
1800 |
pEnc->mbParam.edged_height, pEnc->mbParam.width, |
pEnc->mbParam.edged_height, pEnc->mbParam.width, |
1818 |
stop_inter_timer(); |
stop_inter_timer(); |
1819 |
|
|
1820 |
start_timer(); |
start_timer(); |
1821 |
MotionEstimationBVOP(&pEnc->mbParam, frame, pEnc->reference->mbs, f_ref, |
MotionEstimationBVOP(&pEnc->mbParam, frame, |
1822 |
|
((int32_t)pEnc->mbParam.fbase + (int32_t)pEnc->mbParam.m_ticks + 1 - (int32_t)pEnc->last_pframe) % pEnc->mbParam.fbase, |
1823 |
|
pEnc->time_pp, |
1824 |
|
pEnc->reference->mbs, f_ref, |
1825 |
&pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv, |
&pEnc->f_refh, &pEnc->f_refv, &pEnc->f_refhv, |
1826 |
pEnc->current->mbs, b_ref, &pEnc->vInterH, |
pEnc->current->mbs, b_ref, &pEnc->vInterH, |
1827 |
&pEnc->vInterV, &pEnc->vInterHV); |
&pEnc->vInterV, &pEnc->vInterHV); |
1864 |
if (mb->mode == MODE_NOT_CODED) { |
if (mb->mode == MODE_NOT_CODED) { |
1865 |
mb->mvs[0].x = 0; |
mb->mvs[0].x = 0; |
1866 |
mb->mvs[0].y = 0; |
mb->mvs[0].y = 0; |
1867 |
|
|
1868 |
|
mb->cbp = 0; |
1869 |
|
#ifdef BFRAMES_DEC_DEBUG |
1870 |
|
BFRAME_DEBUG |
1871 |
|
#endif |
1872 |
continue; |
continue; |
1873 |
} |
} |
1874 |
|
|
1884 |
qcoeff); |
qcoeff); |
1885 |
//mb->cbp = MBTransQuantBVOP(&pEnc->mbParam, x, y, dct_codes, qcoeff, &frame->image, frame->quant); |
//mb->cbp = MBTransQuantBVOP(&pEnc->mbParam, x, y, dct_codes, qcoeff, &frame->image, frame->quant); |
1886 |
|
|
1887 |
|
if ( (mb->mode == MODE_DIRECT) && (mb->cbp == 0) |
1888 |
if ((mb->mode == MODE_INTERPOLATE || mb->mode == MODE_DIRECT) |
&& (mb->deltamv.x == 0) && (mb->deltamv.y == 0) ) { |
1889 |
&& mb->cbp == 0 && mb->mvs[0].x == 0 && mb->mvs[0].y == 0) { |
mb->mode = MODE_DIRECT_NONE_MV; // skipped |
|
mb->mode = 5; // skipped |
|
1890 |
} |
} |
1891 |
|
|
1892 |
|
/* update predictors for forward and backward vectors */ |
1893 |
if (mb->mode == MODE_INTERPOLATE || mb->mode == MODE_FORWARD) { |
if (mb->mode == MODE_INTERPOLATE || mb->mode == MODE_FORWARD) { |
1894 |
mb->pmvs[0].x = mb->mvs[0].x - forward.x; |
mb->pmvs[0].x = mb->mvs[0].x - forward.x; |
1895 |
mb->pmvs[0].y = mb->mvs[0].y - forward.y; |
mb->pmvs[0].y = mb->mvs[0].y - forward.y; |
1903 |
backward.x = mb->b_mvs[0].x; |
backward.x = mb->b_mvs[0].x; |
1904 |
backward.y = mb->b_mvs[0].y; |
backward.y = mb->b_mvs[0].y; |
1905 |
} |
} |
1906 |
|
|
1907 |
// DPRINTF("%05i : [%i %i] M=%i CBP=%i MVS=%i,%i forward=%i,%i", pEnc->m_framenum, x, y, mb->mode, mb->cbp, mb->mvs[0].x, mb->mvs[0].y, forward.x, forward.y); |
// DPRINTF("%05i : [%i %i] M=%i CBP=%i MVS=%i,%i forward=%i,%i", pEnc->m_framenum, x, y, mb->mode, mb->cbp, mb->mvs[0].x, mb->mvs[0].y, forward.x, forward.y); |
1908 |
|
|
1909 |
|
#ifdef BFRAMES_DEC_DEBUG |
1910 |
|
BFRAME_DEBUG |
1911 |
|
#endif |
1912 |
start_timer(); |
start_timer(); |
1913 |
MBCodingBVOP(mb, qcoeff, frame->fcode, frame->bcode, bs, |
MBCodingBVOP(mb, qcoeff, frame->fcode, frame->bcode, bs, |
1914 |
&pEnc->sStat); |
&pEnc->sStat); |
1921 |
// TODO: dynamic fcode/bcode ??? |
// TODO: dynamic fcode/bcode ??? |
1922 |
|
|
1923 |
*pBits = BitstreamPos(bs) - *pBits; |
*pBits = BitstreamPos(bs) - *pBits; |
1924 |
|
|
1925 |
|
#ifdef BFRAMES_DEC_DEBUG |
1926 |
|
if (!first){ |
1927 |
|
first=1; |
1928 |
|
if (fp) |
1929 |
|
fclose(fp); |
1930 |
|
} |
1931 |
|
#endif |
1932 |
} |
} |
1933 |
#endif |
#endif |
1934 |
|
|
1935 |
|
|
1936 |
|
/* in case internal output is needed somewhere... */ |
1937 |
|
/* { |
1938 |
|
FILE *filehandle; |
1939 |
|
filehandle=fopen("last-b.pgm","wb"); |
1940 |
|
if (filehandle) |
1941 |
|
{ |
1942 |
|
fprintf(filehandle,"P5\n\n"); // |
1943 |
|
fprintf(filehandle,"%d %d 255\n",pEnc->mbParam.edged_width,pEnc->mbParam.edged_height); |
1944 |
|
fwrite(frame->image.y,pEnc->mbParam.edged_width,pEnc->mbParam.edged_height,filehandle); |
1945 |
|
fclose(filehandle); |
1946 |
|
} |
1947 |
|
} |
1948 |
|
*/ |