static void process_image(char *p)
{
int bpp = 16;
if (discard_count) {
discard_count -= 1;
printf("will discard %d more frames\n", discard_count);
return;
}
if (save_jpeg) {
unsigned char *src = (unsigned char *)p;
// convert from YUV422 to RGB888
YUV422toRGB888(g_pix_width, g_pix_height, src, g_img_buf);
// write jpeg
jpegWrite(g_img_buf);
} else {
save_yuv(bpp, p); /* 422 */
}
file_loop++;
}
/* return the encoded data in *pbuf and the size. Return < 0 if error */
int jctvc_encode_picture(uint8_t **pbuf, Image *img,
const HEVCEncodeParams *params)
{
TAppEncTop cTAppEncTop;
int argc;
char *argv[ARGV_MAX + 1];
char buf[1024], infilename[1024], outfilename[1024];
const char *str;
FILE *f;
uint8_t *out_buf;
int out_buf_len, i;
#ifdef WIN32
if (GetTempPath(sizeof(buf), buf) > sizeof(buf) - 1) {
fprintf(stderr, "Temporary path too long\n");
return -1;
}
#else
strcpy(buf, "/tmp/");
#endif
snprintf(infilename, sizeof(infilename), "%sout%d.yuv", buf, getpid());
snprintf(outfilename, sizeof(outfilename), "%sout%d.bin", buf, getpid());
save_yuv(img, infilename);
m_gcAnalyzeAll.clear();
m_gcAnalyzeI.clear();
m_gcAnalyzeP.clear();
m_gcAnalyzeB.clear();
m_gcAnalyzeAll_in.clear();
cTAppEncTop.create();
argc = 0;
add_opt(&argc, argv, "jctvc"); /* dummy executable name */
snprintf(buf, sizeof(buf),"--InputFile=%s", infilename);
add_opt(&argc, argv, buf);
snprintf(buf, sizeof(buf),"--BitstreamFile=%s", outfilename);
add_opt(&argc, argv, buf);
snprintf(buf, sizeof(buf),"--SourceWidth=%d", img->w);
add_opt(&argc, argv, buf);
snprintf(buf, sizeof(buf),"--SourceWidth=%d", img->w);
add_opt(&argc, argv, buf);
snprintf(buf, sizeof(buf),"--SourceHeight=%d", img->h);
add_opt(&argc, argv, buf);
snprintf(buf, sizeof(buf),"--InputBitDepth=%d", img->bit_depth);
add_opt(&argc, argv, buf);
switch(img->format) {
case BPG_FORMAT_GRAY:
str = "400";
break;
case BPG_FORMAT_420:
str = "420";
break;
case BPG_FORMAT_422:
str = "422";
break;
case BPG_FORMAT_444:
str = "444";
break;
default:
abort();
}
snprintf(buf, sizeof(buf),"--InputChromaFormat=%s", str);
add_opt(&argc, argv, buf);
snprintf(buf, sizeof(buf),"--QP=%d", params->qp);
add_opt(&argc, argv, buf);
snprintf(buf, sizeof(buf),"--SEIDecodedPictureHash=%d",
params->sei_decoded_picture_hash);
add_opt(&argc, argv, buf);
if (!params->verbose)
add_opt(&argc, argv, "--Verbose=0");
/* single frame */
add_opt(&argc, argv, "--FramesToBeEncoded=1");
/* no padding necessary (it is done before) */
add_opt(&argc, argv, "--ConformanceWindowMode=0");
/* dummy frame rate */
add_opt(&argc, argv, "--FrameRate=25");
/* general config */
add_opt(&argc, argv, "--Profile=main_444_16_intra");
add_opt(&argc, argv, "--QuadtreeTULog2MaxSize=5");
add_opt(&argc, argv, "--QuadtreeTUMaxDepthIntra=3");
add_opt(&argc, argv, "--IntraPeriod=1");
add_opt(&argc, argv, "--GOPSize=1");
add_opt(&argc, argv, "--TransformSkip=1");
add_opt(&argc, argv, "--TransformSkipFast=1");
/* Note: Format Range extension */
if (img->format == BPG_FORMAT_444) {
add_opt(&argc, argv, "--CrossComponentPrediction=1");
}
if (params->lossless) {
add_opt(&argc, argv, "--CostMode=lossless");
add_opt(&argc, argv, "--SAO=0");
add_opt(&argc, argv, "--LoopFilterDisable");
add_opt(&argc, argv, "--TransquantBypassEnableFlag");
add_opt(&argc, argv, "--CUTransquantBypassFlagForce");
add_opt(&argc, argv, "--ImplicitResidualDPCM");
add_opt(&argc, argv, "--GolombRiceParameterAdaptation");
add_opt(&argc, argv, "--HadamardME=0");
}
/* trailing NULL */
argv[argc] = NULL;
if (params->verbose >= 2) {
int i;
printf("Encode options:");
for(i = 0; i < argc; i++) {
printf(" %s", argv[i]);
}
printf("\n");
}
if(!cTAppEncTop.parseCfg( argc, argv )) {
fprintf(stderr, "Error while parsing options\n");
cTAppEncTop.destroy();
return -1;
}
cTAppEncTop.encode();
cTAppEncTop.destroy();
for(i = 0; i < argc; i++)
free(argv[i]);
unlink(infilename);
/* read output bitstream */
f = fopen(outfilename, "rb");
if (!f) {
fprintf(stderr, "Could not open '%s'\n", outfilename);
return -1;
}
fseek(f, 0, SEEK_END);
out_buf_len = ftell(f);
fseek(f, 0, SEEK_SET);
out_buf = (uint8_t *)malloc(out_buf_len);
if (fread(out_buf, 1, out_buf_len, f) != out_buf_len) {
fprintf(stderr, "read error\n");
fclose(f);
free(out_buf);
return -1;
}
fclose(f);
unlink(outfilename);
*pbuf = out_buf;
return out_buf_len;
}
