void CCLEncoder::doEncodeFrame(CImage<uint8_t> * pFrame, CBitstream * pBstr, FRAME_TYPE frame_type)
{
(*static_cast<CImage<float>*>(m_imgF)) = (*pFrame);
m_imgF->CopyToDevice();
transform(m_imgF, m_img, m_predTab, frame_type);
entropy(m_img, m_predTab, pBstr, frame_type);
itransform(m_imgF, m_img, m_predTab, frame_type);
}
void CCLParallelEncoder::doEncodeFrame(CImage<uint8_t> * pFrame, CBitstream * pBstr, FRAME_TYPE frame_type)
{
(*(CImage<float>*)m_imgF) = (*pFrame);
m_imgF->CopyToDevice();
transform(m_imgF, m_img, m_predTab, frame_type);
m_dev.Finish();
m_img->CopyToHost();
m_predTab->CopyToHost();
itransform(m_imgF, m_img, m_predTab, frame_type);
entropy(m_img, m_predTab, pBstr, frame_type);
m_dev.Finish();
}
void putseq()
{
/* this routine assumes (N % M) == 0 */
int i, j, k, f, f0, n, np, nb, sxf, syf, sxb, syb;
int ipflag;
FILE *fd;
char name[256];
unsigned char *neworg[3], *newref[3];
static char ipb[5] = {' ','I','P','B','D'};
rc_init_seq(); /* initialize rate control */
/* sequence header, sequence extension and sequence display extension */
putseqhdr();
if (!mpeg1)
{
putseqext();
putseqdispext();
}
/* optionally output some text data (description, copyright or whatever) */
if (strlen(id_string) > 1)
putuserdata(id_string);
/* loop through all frames in encoding/decoding order */
for (i=0; i<nframes; i++)
{
if (!quiet)
{
fprintf(stderr,"Encoding frame %d ",i);
fflush(stderr);
}
/* f0: lowest frame number in current GOP
*
* first GOP contains N-(M-1) frames,
* all other GOPs contain N frames
*/
f0 = N*((i+(M-1))/N) - (M-1);
if (f0<0)
f0=0;
if (i==0 || (i-1)%M==0)
{
/* I or P frame */
for (j=0; j<3; j++)
{
/* shuffle reference frames */
neworg[j] = oldorgframe[j];
newref[j] = oldrefframe[j];
oldorgframe[j] = neworgframe[j];
oldrefframe[j] = newrefframe[j];
neworgframe[j] = neworg[j];
newrefframe[j] = newref[j];
}
/* f: frame number in display order */
f = (i==0) ? 0 : i+M-1;
if (f>=nframes)
f = nframes - 1;
if (i==f0) /* first displayed frame in GOP is I */
{
/* I frame */
pict_type = I_TYPE;
forw_hor_f_code = forw_vert_f_code = 15;
back_hor_f_code = back_vert_f_code = 15;
/* n: number of frames in current GOP
*
* first GOP contains (M-1) less (B) frames
*/
n = (i==0) ? N-(M-1) : N;
/* last GOP may contain less frames */
if (n > nframes-f0)
n = nframes-f0;
/* number of P frames */
if (i==0)
np = (n + 2*(M-1))/M - 1; /* first GOP */
else
np = (n + (M-1))/M - 1;
/* number of B frames */
nb = n - np - 1;
rc_init_GOP(np,nb);
putgophdr(f0,i==0); /* set closed_GOP in first GOP only */
}
else
{
/* P frame */
pict_type = P_TYPE;
forw_hor_f_code = motion_data[0].forw_hor_f_code;
forw_vert_f_code = motion_data[0].forw_vert_f_code;
back_hor_f_code = back_vert_f_code = 15;
sxf = motion_data[0].sxf;
syf = motion_data[0].syf;
}
}
else
{
/* B frame */
for (j=0; j<3; j++)
{
neworg[j] = auxorgframe[j];
newref[j] = auxframe[j];
}
/* f: frame number in display order */
f = i - 1;
pict_type = B_TYPE;
n = (i-2)%M + 1; /* first B: n=1, second B: n=2, ... */
forw_hor_f_code = motion_data[n].forw_hor_f_code;
forw_vert_f_code = motion_data[n].forw_vert_f_code;
back_hor_f_code = motion_data[n].back_hor_f_code;
back_vert_f_code = motion_data[n].back_vert_f_code;
sxf = motion_data[n].sxf;
syf = motion_data[n].syf;
sxb = motion_data[n].sxb;
syb = motion_data[n].syb;
}
temp_ref = f - f0;
frame_pred_dct = frame_pred_dct_tab[pict_type-1];
q_scale_type = qscale_tab[pict_type-1];
intravlc = intravlc_tab[pict_type-1];
altscan = altscan_tab[pict_type-1];
fprintf(statfile,"\nFrame %d (#%d in display order):\n",i,f);
fprintf(statfile," picture_type=%c\n",ipb[pict_type]);
fprintf(statfile," temporal_reference=%d\n",temp_ref);
fprintf(statfile," frame_pred_frame_dct=%d\n",frame_pred_dct);
fprintf(statfile," q_scale_type=%d\n",q_scale_type);
fprintf(statfile," intra_vlc_format=%d\n",intravlc);
fprintf(statfile," alternate_scan=%d\n",altscan);
if (pict_type!=I_TYPE)
{
fprintf(statfile," forward search window: %d...%d / %d...%d\n",
-sxf,sxf,-syf,syf);
fprintf(statfile," forward vector range: %d...%d.5 / %d...%d.5\n",
-(4<<forw_hor_f_code),(4<<forw_hor_f_code)-1,
-(4<<forw_vert_f_code),(4<<forw_vert_f_code)-1);
}
if (pict_type==B_TYPE)
{
fprintf(statfile," backward search window: %d...%d / %d...%d\n",
-sxb,sxb,-syb,syb);
fprintf(statfile," backward vector range: %d...%d.5 / %d...%d.5\n",
-(4<<back_hor_f_code),(4<<back_hor_f_code)-1,
-(4<<back_vert_f_code),(4<<back_vert_f_code)-1);
}
sprintf(name,tplorg,f+frame0);
readframe(name,neworg);
if (fieldpic)
{
if (!quiet)
{
fprintf(stderr,"\nfirst field (%s) ",topfirst ? "top" : "bot");
fflush(stderr);
}
pict_struct = topfirst ? TOP_FIELD : BOTTOM_FIELD;
#ifndef SKIP_PREDICTION
motion_estimation(oldorgframe[0],neworgframe[0],
oldrefframe[0],newrefframe[0],
neworg[0],newref[0],
sxf,syf,sxb,syb,mbinfo,0,0);
predict(oldrefframe,newrefframe,predframe,0,mbinfo);
#endif
#ifndef SKIP_BLOCK_ENCODE
dct_type_estimation(predframe[0],neworg[0],mbinfo);
transform(predframe,neworg,mbinfo,blocks);
#endif
putpict(neworg[0]);
#ifndef SKIP_BLOCK_DECODE
for (k=0; k<mb_height2*mb_width; k++)
{
if (mbinfo[k].mb_type & MB_INTRA)
for (j=0; j<block_count; j++)
iquant_intra(blocks[k*block_count+j],blocks[k*block_count+j],
dc_prec,intra_q,mbinfo[k].mquant);
else
for (j=0;j<block_count;j++)
iquant_non_intra(blocks[k*block_count+j],blocks[k*block_count+j],
inter_q,mbinfo[k].mquant);
}
itransform(predframe,newref,mbinfo,blocks);
#endif
calcSNR(neworg,newref);
stats();
if (!quiet)
{
fprintf(stderr,"second field (%s) ",topfirst ? "bot" : "top");
fflush(stderr);
}
pict_struct = topfirst ? BOTTOM_FIELD : TOP_FIELD;
ipflag = (pict_type==I_TYPE);
if (ipflag)
{
/* first field = I, second field = P */
pict_type = P_TYPE;
forw_hor_f_code = motion_data[0].forw_hor_f_code;
forw_vert_f_code = motion_data[0].forw_vert_f_code;
back_hor_f_code = back_vert_f_code = 15;
sxf = motion_data[0].sxf;
syf = motion_data[0].syf;
}
#ifndef SKIP_PREDICTION
motion_estimation(oldorgframe[0],neworgframe[0],
oldrefframe[0],newrefframe[0],
neworg[0],newref[0],
sxf,syf,sxb,syb,mbinfo,1,ipflag);
predict(oldrefframe,newrefframe,predframe,1,mbinfo);
#endif
#ifndef SKIP_BLOCK_ENCOED
dct_type_estimation(predframe[0],neworg[0],mbinfo);
transform(predframe,neworg,mbinfo,blocks);
#endif
putpict(neworg[0]);
#ifndef SKIP_BLOCK_DECODE
for (k=0; k<mb_height2*mb_width; k++)
{
if (mbinfo[k].mb_type & MB_INTRA)
for (j=0; j<block_count; j++)
iquant_intra(blocks[k*block_count+j],blocks[k*block_count+j],
dc_prec,intra_q,mbinfo[k].mquant);
else
for (j=0;j<block_count;j++)
iquant_non_intra(blocks[k*block_count+j],blocks[k*block_count+j],
inter_q,mbinfo[k].mquant);
}
itransform(predframe,newref,mbinfo,blocks);
#endif
calcSNR(neworg,newref);
stats();
}
else
{
pict_struct = FRAME_PICTURE;
/* do motion_estimation
*
* uses source frames (...orgframe) for full pel search
* and reconstructed frames (...refframe) for half pel search
*/
#ifndef SKIP_PREDICTION
motion_estimation(oldorgframe[0],neworgframe[0],
oldrefframe[0],newrefframe[0],
neworg[0],newref[0],
sxf,syf,sxb,syb,mbinfo,0,0);
predict(oldrefframe,newrefframe,predframe,0,mbinfo);
#endif
#ifndef SKIP_BLOCK_ENCODE
dct_type_estimation(predframe[0],neworg[0],mbinfo);
transform(predframe,neworg,mbinfo,blocks);
#endif
putpict(neworg[0]);
#ifndef SKIP_BLOCK_DECODE
for (k=0; k<mb_height*mb_width; k++)
{
if (mbinfo[k].mb_type & MB_INTRA)
for (j=0; j<block_count; j++)
iquant_intra(blocks[k*block_count+j],blocks[k*block_count+j],
dc_prec,intra_q,mbinfo[k].mquant);
else
for (j=0;j<block_count;j++)
iquant_non_intra(blocks[k*block_count+j],blocks[k*block_count+j],
inter_q,mbinfo[k].mquant);
}
itransform(predframe,newref,mbinfo,blocks);
#endif
calcSNR(neworg,newref);
stats();
}
sprintf(name,tplref,f+frame0);
writeframe(name,newref);
}
putseqend();
}