mirror of
https://bitbucket.org/King_DuckZ/jumping-in-d.git
synced 2024-10-30 02:49:01 +00:00
257 lines
6.7 KiB
C++
257 lines
6.7 KiB
C++
#ifdef WIN32
|
|
#include <windows.h>
|
|
#endif
|
|
#include <unistd.h>
|
|
#include <iostream>
|
|
#include "TAppEncTop.h"
|
|
#include "TLibCommon/Debug.h"
|
|
#include "TLibEncoder/TEncAnalyze.h"
|
|
|
|
#include "bpgenc.h"
|
|
|
|
struct HEVCEncoderContext {
|
|
HEVCEncodeParams params;
|
|
char infilename[1024];
|
|
char outfilename[1024];
|
|
FILE *yuv_file;
|
|
int frame_count;
|
|
};
|
|
|
|
#define ARGV_MAX 256
|
|
|
|
static void add_opt(int *pargc, char **argv,
|
|
const char *str)
|
|
{
|
|
int argc;
|
|
argc = *pargc;
|
|
if (argc >= ARGV_MAX)
|
|
abort();
|
|
argv[argc++] = strdup(str);
|
|
*pargc = argc;
|
|
}
|
|
|
|
static HEVCEncoderContext *jctvc_open(const HEVCEncodeParams *params)
|
|
{
|
|
HEVCEncoderContext *s;
|
|
char buf[1024];
|
|
static int tmp_idx = 1;
|
|
|
|
s = (HEVCEncoderContext *)malloc(sizeof(HEVCEncoderContext));
|
|
memset(s, 0, sizeof(*s));
|
|
|
|
s->params = *params;
|
|
#ifdef WIN32
|
|
if (GetTempPath(sizeof(buf), buf) > sizeof(buf) - 1) {
|
|
fprintf(stderr, "Temporary path too long\n");
|
|
free(s);
|
|
return NULL;
|
|
}
|
|
#else
|
|
strcpy(buf, "/tmp/");
|
|
#endif
|
|
snprintf(s->infilename, sizeof(s->infilename), "%sout%d-%d.yuv", buf, getpid(), tmp_idx);
|
|
snprintf(s->outfilename, sizeof(s->outfilename), "%sout%d-%d.bin", buf, getpid(), tmp_idx);
|
|
tmp_idx++;
|
|
|
|
s->yuv_file = fopen(s->infilename, "wb");
|
|
if (!s->yuv_file) {
|
|
fprintf(stderr, "Could not open '%s'\n", s->infilename);
|
|
free(s);
|
|
return NULL;
|
|
}
|
|
return s;
|
|
}
|
|
|
|
static int jctvc_encode(HEVCEncoderContext *s, Image *img)
|
|
{
|
|
save_yuv1(img, s->yuv_file);
|
|
s->frame_count++;
|
|
return 0;
|
|
}
|
|
|
|
/* return the encoded data in *pbuf and the size. Return < 0 if error */
|
|
static int jctvc_close(HEVCEncoderContext *s, uint8_t **pbuf)
|
|
{
|
|
TAppEncTop cTAppEncTop;
|
|
int argc;
|
|
char *argv[ARGV_MAX + 1];
|
|
char buf[1024];
|
|
const char *str;
|
|
FILE *f;
|
|
uint8_t *out_buf;
|
|
int out_buf_len, i;
|
|
|
|
fclose(s->yuv_file);
|
|
s->yuv_file = NULL;
|
|
|
|
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", s->infilename);
|
|
add_opt(&argc, argv, buf);
|
|
snprintf(buf, sizeof(buf),"--BitstreamFile=%s", s->outfilename);
|
|
add_opt(&argc, argv, buf);
|
|
|
|
snprintf(buf, sizeof(buf),"--SourceWidth=%d", s->params.width);
|
|
add_opt(&argc, argv, buf);
|
|
snprintf(buf, sizeof(buf),"--SourceHeight=%d", s->params.height);
|
|
add_opt(&argc, argv, buf);
|
|
snprintf(buf, sizeof(buf),"--InputBitDepth=%d", s->params.bit_depth);
|
|
add_opt(&argc, argv, buf);
|
|
|
|
switch(s->params.chroma_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", s->params.qp);
|
|
add_opt(&argc, argv, buf);
|
|
|
|
snprintf(buf, sizeof(buf),"--SEIDecodedPictureHash=%d",
|
|
s->params.sei_decoded_picture_hash);
|
|
add_opt(&argc, argv, buf);
|
|
|
|
if (!s->params.verbose)
|
|
add_opt(&argc, argv, "--Verbose=0");
|
|
|
|
/* single frame */
|
|
snprintf(buf, sizeof(buf),"--FramesToBeEncoded=%d", s->frame_count);
|
|
add_opt(&argc, argv, buf);
|
|
|
|
/* 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, "--QuadtreeTULog2MaxSize=5");
|
|
if (s->params.compress_level == 9) {
|
|
add_opt(&argc, argv, "--QuadtreeTUMaxDepthIntra=4");
|
|
add_opt(&argc, argv, "--QuadtreeTUMaxDepthInter=4");
|
|
} else {
|
|
add_opt(&argc, argv, "--QuadtreeTUMaxDepthIntra=3");
|
|
add_opt(&argc, argv, "--QuadtreeTUMaxDepthInter=3");
|
|
}
|
|
|
|
if (s->params.intra_only) {
|
|
add_opt(&argc, argv, "--Profile=main_444_16_intra");
|
|
|
|
add_opt(&argc, argv, "--IntraPeriod=1");
|
|
add_opt(&argc, argv, "--GOPSize=1");
|
|
} else {
|
|
int gop_size;
|
|
|
|
add_opt(&argc, argv, "--Profile=main_444_16");
|
|
add_opt(&argc, argv, "--IntraPeriod=250");
|
|
gop_size = 1;
|
|
snprintf(buf, sizeof(buf), "--GOPSize=%d", gop_size);
|
|
add_opt(&argc, argv, buf);
|
|
|
|
for(i = 0; i < gop_size; i++) {
|
|
snprintf(buf, sizeof(buf), "--Frame%d=P 1 3 0.4624 0 0 0 1 1 -1 0", i + 1);
|
|
add_opt(&argc, argv, buf);
|
|
}
|
|
}
|
|
add_opt(&argc, argv, "--TransformSkip=1");
|
|
add_opt(&argc, argv, "--TransformSkipFast=1");
|
|
|
|
/* Note: Format Range extension */
|
|
if (s->params.chroma_format == BPG_FORMAT_444) {
|
|
add_opt(&argc, argv, "--CrossComponentPrediction=1");
|
|
}
|
|
|
|
if (s->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");
|
|
}
|
|
|
|
#if 0
|
|
/* TEST with several slices */
|
|
add_opt(&argc, argv, "--SliceMode=2");
|
|
add_opt(&argc, argv, "--SliceArgument=5");
|
|
#endif
|
|
|
|
/* trailing NULL */
|
|
argv[argc] = NULL;
|
|
|
|
if (s->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(s->infilename);
|
|
|
|
/* read output bitstream */
|
|
f = fopen(s->outfilename, "rb");
|
|
if (!f) {
|
|
fprintf(stderr, "Could not open '%s'\n", s->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(s->outfilename);
|
|
*pbuf = out_buf;
|
|
free(s);
|
|
return out_buf_len;
|
|
}
|
|
|
|
HEVCEncoder jctvc_encoder = {
|
|
.open = jctvc_open,
|
|
.encode = jctvc_encode,
|
|
.close = jctvc_close,
|
|
};
|