/* *  Copyright (c) 2010 The WebM project authors. All Rights Reserved. * *  Use of this source code is governed by a BSD-style license *  that can be found in the LICENSE file in the root of the source *  tree. An additional intellectual property rights grant can be found *  in the file PATENTS.  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/vpx_encoder.h"#include "vpx/vp8cx.h"#define interface (vpx_codec_vp8_cx())#define fourcc    0x30385056@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 */     if(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);     if(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;    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, VPX_IMG_FMT_I420, 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;}