encodetest.cpp

FFMpeg Libraries for Symbian OS

#include <stdio.h>
//#include <stdint.h>
#include <inttypes.h>
#include <string.h>
#include "avcodec.h"

class VideoEncoder {
public:
	VideoEncoder(CodecID codecId, int width, int height, int bitrate) {
		codec = avcodec_find_encoder(codecId);
		if (!codec) {
			fprintf(stderr, "Codec not found\n");
			return;
		}
		c = avcodec_alloc_context();
		picture = avcodec_alloc_frame();
		c->bit_rate = bitrate;
		c->width = width;
		c->height = height;
		c->time_base = (AVRational) {1, 25};
		c->gop_size = 10;
//		c->max_b_frames = 1;
		c->pix_fmt = PIX_FMT_YUV420P;

		picture->linesize[0] = c->width;
		picture->linesize[1] = c->width/2;
		picture->linesize[2] = c->width/2;

		if (avcodec_open(c, codec) < 0) {
			fprintf(stderr, "Could not open codec\n");
			return;
		}

		frames = 0;
	}

	void finish() {
		while (true) {
			int written = avcodec_encode_video(c, outbuf, sizeof(outbuf), NULL);
			if (written <= 0)
				break;
			storeData(written);
			frames++;
		}
	}

	virtual ~VideoEncoder() {
		avcodec_close(c);
		av_free(c);
		av_free(picture);
	}

	void encodeFrame(uint8_t *frame) {
		picture->data[0] = frame;
		picture->data[1] = frame + c->width*c->height;
		picture->data[2] = frame + c->width*c->height + (c->width/2) * (c->height/2);

		int written = avcodec_encode_video(c, outbuf, sizeof(outbuf), picture);
		storeData(written);
		frames++;
	}

	virtual void storeData(int size) = 0;

private:
	AVCodec* codec;
	AVCodecContext* c;
	AVFrame* picture;
protected:
	uint8_t outbuf[100000];
	int frames;
};

class MpegFileEncoder : public VideoEncoder {
public:
	MpegFileEncoder(int width, int height, int bitrate) : VideoEncoder(CODEC_ID_MPEG1VIDEO, width, height, bitrate) {
		out = fopen("e:\\video.mpg", "wb");
	}
	~MpegFileEncoder() {
		outbuf[0] = 0x00;
		outbuf[1] = 0x00;
		outbuf[2] = 0x01;
		outbuf[3] = 0xb7;
		fwrite(outbuf, 4, 1, out);
		fclose(out);
	}
	void storeData(int size) {
		fwrite(outbuf, size, 1, out);
		printf("writing frame %d, %d bytes\n", frames, size);
	}

private:
	FILE *out;
};

void doTest() {
	avcodec_init();
	avcodec_register_all();
	int width = 176;
	int height = 144;
	MpegFileEncoder enc(width, height, 100000);
	int size = width*height + 2*(width/2)*(height/2);
	uint8_t* buffer = new uint8_t[size];
	for (int i = 0; i < 15; i++) {
		for (int y = 0; y < height; y++) {
			for (int x = 0; x < width; x++) {
				int Y = (x + y + 20*i) & 0xff;
				int U = x << 1;
				int V = y << 1;
				buffer[y*width + x] = Y;
				buffer[width*height + (y/2)*(width/2) + (x/2)] = U;
				buffer[width*height + (width/2)*(height/2) + (y/2)*(width/2) + (x/2)] = V;
			}
		}
		enc.encodeFrame(buffer);
	}
	delete [] buffer;
	enc.finish();
}


extern "C" int entrypoint() {
	doTest();
	return 0;
}