/*
 * Copyright 2009, Stephan Aßmus <superstippi@gmx.de>
 * All rights reserved. Distributed under the terms of the GNU L-GPL license.
 */
#ifndef AV_FORMAT_WRITER_H
#define AV_FORMAT_WRITER_H


#include <Locker.h>

#include "WriterPlugin.h"

extern "C" {
	#include "avformat.h"
}


class AVFormatWriter : public Writer {
public:
								AVFormatWriter();
								~AVFormatWriter();

	virtual	status_t			Init(const media_file_format* fileFormat);

	virtual	status_t			SetCopyright(const char* copyright);
	virtual	status_t			CommitHeader();
	virtual	status_t			Flush();
	virtual	status_t			Close();

	virtual	status_t			AllocateCookie(void** cookie,
									const media_format* format,
									const media_codec_info* codecInfo);
	virtual	status_t			FreeCookie(void* cookie);

	virtual	status_t			SetCopyright(void* cookie,
									const char* copyright);

	virtual	status_t			AddTrackInfo(void* cookie, uint32 code,
									const void* data, size_t size,
									uint32 flags = 0);

	virtual	status_t			WriteChunk(void* cookie,
									const void* chunkBuffer, size_t chunkSize,
									media_encode_info* encodeInfo);

private:
	static	int					_Write(void* cookie, uint8* buffer,
									int bufferSize);

	static	off_t				_Seek(void* cookie, off_t offset, int whence);

private:
	class StreamCookie;

			AVFormatContext*	fContext;
			bool				fHeaderWritten;

			ByteIOContext		fIOContext;
			uint8*				fIOBuffer;

			StreamCookie**		fStreams;
			BLocker				fStreamLock;
};


#endif // AV_FORMAT_WRITER_H