diff options
Diffstat (limited to 'examples/encoder_tmpl.c')
| -rw-r--r-- | examples/encoder_tmpl.c | 189 |
1 files changed, 189 insertions, 0 deletions
diff --git a/examples/encoder_tmpl.c b/examples/encoder_tmpl.c new file mode 100644 index 000000000..2df893af8 --- /dev/null +++ b/examples/encoder_tmpl.c @@ -0,0 +1,189 @@ +/* + * Copyright (c) 2010 The VP8 project authors. All Rights Reserved. + * + * Use of this source code is governed by a BSD-style license and patent + * grant that can be found in the LICENSE file in the root of the source + * tree. All contributing project authors may be found in the AUTHORS + * file in the root of the source tree. + */ + + +/* +@*INTRODUCTION + */ +#include <stdio.h> +#include <stdlib.h> +#include <stdarg.h> +#include <string.h> +#define VPX_CODEC_DISABLE_COMPAT 1 +#include "vpx_encoder.h" +#if CONFIG_VP8_ENCODER && !defined(interface) +#include "vp8cx.h" +#define interface (&vpx_codec_vp8_cx_algo) +#define fourcc 0x30385056 +#endif +@EXTRA_INCLUDES + +#define IVF_FILE_HDR_SZ (32) +#define IVF_FRAME_HDR_SZ (12) + +static void mem_put_le16(char *mem, unsigned int val) { + mem[0] = val; + mem[1] = val>>8; +} + +static void mem_put_le32(char *mem, unsigned int val) { + mem[0] = val; + mem[1] = val>>8; + mem[2] = val>>16; + mem[3] = val>>24; +} + +static void die(const char *fmt, ...) { + va_list ap; + + va_start(ap, fmt); + vprintf(fmt, ap); + if(fmt[strlen(fmt)-1] != '\n') + printf("\n"); + exit(EXIT_FAILURE); +} + +@DIE_CODEC + +static int read_frame(FILE *f, vpx_image_t *img) { + size_t nbytes, to_read; + int res = 1; + + to_read = img->w*img->h*3/2; + nbytes = fread(img->planes[0], 1, to_read, f); + if(nbytes != to_read) { + res = 0; + if(nbytes > 0) + printf("Warning: Read partial frame. Check your width & height!\n"); + } + return res; +} + +static void write_ivf_file_header(FILE *outfile, + const vpx_codec_enc_cfg_t *cfg, + int frame_cnt) { + char header[32]; + + if(cfg->g_pass != VPX_RC_ONE_PASS && cfg->g_pass != VPX_RC_LAST_PASS) + return; + header[0] = 'D'; + header[1] = 'K'; + header[2] = 'I'; + header[3] = 'F'; + mem_put_le16(header+4, 0); /* version */ + mem_put_le16(header+6, 32); /* headersize */ + mem_put_le32(header+8, fourcc); /* headersize */ + mem_put_le16(header+12, cfg->g_w); /* width */ + mem_put_le16(header+14, cfg->g_h); /* height */ + mem_put_le32(header+16, cfg->g_timebase.den); /* rate */ + mem_put_le32(header+20, cfg->g_timebase.num); /* scale */ + mem_put_le32(header+24, frame_cnt); /* length */ + mem_put_le32(header+28, 0); /* unused */ + + fwrite(header, 1, 32, outfile); +} + + +static void write_ivf_frame_header(FILE *outfile, + const vpx_codec_cx_pkt_t *pkt) +{ + char header[12]; + vpx_codec_pts_t pts; + + if(pkt->kind != VPX_CODEC_CX_FRAME_PKT) + return; + + pts = pkt->data.frame.pts; + mem_put_le32(header, pkt->data.frame.sz); + mem_put_le32(header+4, pts&0xFFFFFFFF); + mem_put_le32(header+8, pts >> 32); + + fwrite(header, 1, 12, outfile); +} + +int main(int argc, char **argv) { + FILE *infile, *outfile; + vpx_codec_ctx_t codec; + vpx_codec_enc_cfg_t cfg; + int frame_cnt = 0; + unsigned char file_hdr[IVF_FILE_HDR_SZ]; + unsigned char frame_hdr[IVF_FRAME_HDR_SZ]; + vpx_image_t raw; + vpx_codec_err_t res; + long width; + long height; + int frame_avail; + int got_data; + int flags = 0; +@@@@TWOPASS_VARS + + /* Open files */ +@@@@USAGE + width = strtol(argv[1], NULL, 0); + height = strtol(argv[2], NULL, 0); + if(width < 16 || width%2 || height <16 || height%2) + die("Invalid resolution: %ldx%ld", width, height); + if(!vpx_img_alloc(&raw, IMG_FMT_YV12, width, height, 1)) + die("Faile to allocate image", width, height); + if(!(outfile = fopen(argv[4], "wb"))) + die("Failed to open %s for writing", argv[4]); + + printf("Using %s\n",vpx_codec_iface_name(interface)); + +@@@@ENC_DEF_CFG + +@@@@ENC_SET_CFG +@@@@ENC_SET_CFG2 + + write_ivf_file_header(outfile, &cfg, 0); + +@@@@TWOPASS_LOOP_BEGIN + + /* Open input file for this encoding pass */ + if(!(infile = fopen(argv[3], "rb"))) + die("Failed to open %s for reading", argv[3]); + +@@@@@@@@ENC_INIT + + frame_avail = 1; + got_data = 0; + while(frame_avail || got_data) { + vpx_codec_iter_t iter = NULL; + const vpx_codec_cx_pkt_t *pkt; + +@@@@@@@@@@@@PER_FRAME_CFG +@@@@@@@@@@@@ENCODE_FRAME + got_data = 0; + while( (pkt = vpx_codec_get_cx_data(&codec, &iter)) ) { + got_data = 1; + switch(pkt->kind) { +@@@@@@@@@@@@@@@@PROCESS_FRAME +@@@@@@@@@@@@@@@@PROCESS_STATS + default: + break; + } + printf(pkt->kind == VPX_CODEC_CX_FRAME_PKT + && (pkt->data.frame.flags & VPX_FRAME_IS_KEY)? "K":"."); + fflush(stdout); + } + frame_cnt++; + } + printf("\n"); + fclose(infile); +@@@@TWOPASS_LOOP_END + + printf("Processed %d frames.\n",frame_cnt-1); +@@@@DESTROY + + /* Try to rewrite the file header with the actual frame count */ + if(!fseek(outfile, 0, SEEK_SET)) + write_ivf_file_header(outfile, &cfg, frame_cnt-1); + fclose(outfile); + return EXIT_SUCCESS; +} |