/* ScummVM - Graphic Adventure Engine
 *
 * ScummVM is the legal property of its developers, whose names
 * are too numerous to list here. Please refer to the COPYRIGHT
 * file distributed with this source distribution.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.

 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.

 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 * $URL$
 * $Id$
 *
 */

//
// Heavily based on ffmpeg code.
//
// Copyright (c) 2001 Fabrice Bellard.
// First version by Francois Revol revol@free.fr
// Seek function by Gael Chardon gael.dev@4now.net
//

#ifndef MOHAWK_QT_PLAYER_H
#define MOHAWK_QT_PLAYER_H

#include "common/scummsys.h"
#include "common/queue.h"
#include "graphics/video/codecs/codec.h"
#include "sound/audiostream.h"
#include "sound/mixer.h"

namespace Common {
	class File;
}

namespace Mohawk {

class QueuedAudioStream : public Audio::AudioStream {
public:
	QueuedAudioStream(int rate, int channels, bool autofree = true);
	~QueuedAudioStream();

	int readBuffer(int16 *buffer, const int numSamples);
	bool isStereo() const { return _channels == 2; }
	int getRate() const { return _rate; }
	bool endOfData() const { return _queue.empty(); }
	bool endOfStream() const { return _finished; }

	void queueAudioStream(Audio::AudioStream *audStream);
	void finish() { _finished = true; }

	uint32 streamsInQueue() { return _queue.size(); }

private:
	bool _autofree;
	bool _finished;
	int _rate;
	int _channels;

	Common::Queue<Audio::AudioStream*> _queue;
};

enum ScaleMode {
	kScaleNormal = 1,
	kScaleHalf = 2,
	kScaleQuarter = 4
};

class QTPlayer {
public:
	QTPlayer();
	virtual ~QTPlayer();

	/**
	 * Returns the width of the video
	 * @return the width of the video
	 */
	uint16 getWidth();

	/**
	 * Returns the height of the video
	 * @return the height of the video
	 */
	uint16 getHeight();

	/**
	 * Returns the amount of frames in the video
	 * @return the amount of frames in the video
	 */
	uint32 getFrameCount();
	
	/**
	 * Returns the bits per pixel of the video
	 * @return the bits per pixel of the video
	 */
	byte getBitsPerPixel();
	
	/**
	 * Returns the codec tag of the video
	 * @return the codec tag of the video
	 */
	uint32 getCodecTag();

	/**
	 * Load a QuickTime video file from a SeekableReadStream
	 * @param stream	the stream to load
	 */
	bool loadFile(Common::SeekableReadStream* stream);

	/**
	 * Close a QuickTime encoded video file
	 */
	void closeFile();
	
	/**
	 * Returns the downscale mode of the video
	 * @return the downscale mode of the video
	 */
	ScaleMode getScaleMode();
	
	/**
	 * Returns the palette of the video
	 * @return the palette of the video
	 */
	byte *getPalette() { return _palette; }
	
	/**
	 * Set the beginning offset of the video so we can modify the offsets in the stco
	 * atom of videos inside the Mohawk archives
	 * @param the beginning offset of the video
	 */
	void setChunkBeginOffset(uint32 offset) { _beginOffset = offset; }
	
	int32 getCurFrame() { return _curFrame; }
	void addPauseTime(uint32 p) { _lastFrameStart += p; _nextFrameStart += p; }
	Graphics::Surface *getNextFrame();
	void updateAudioBuffer();
	void startAudio();
	void stopAudio();
	void pauseAudio();
	void resumeAudio();
	bool needsUpdate();
	bool endOfVideo();
	void stop();
	void reset();

protected:
	// This is the file handle from which data is read from. It can be the actual file handle or a decompressed stream.
	Common::SeekableReadStream *_fd;

	struct MOVatom {
		uint32 type;
		uint32 offset;
		uint32 size;
	};

	struct ParseTable {
		uint32 type;
		int (QTPlayer::*func)(MOVatom atom);
	};

	struct MOVstts {
		int count;
		int duration;
	};

	struct MOVstsc {
		uint32 first;
		uint32 count;
		uint32 id;
	};

	enum CodecType {
		CODEC_TYPE_MOV_OTHER,
		CODEC_TYPE_VIDEO,
		CODEC_TYPE_AUDIO
	};

	struct MOVStreamContext {
		MOVStreamContext() { memset(this, 0, sizeof(MOVStreamContext)); }
		~MOVStreamContext() {
			delete[] chunk_offsets;
			delete[] stts_data;
			delete[] ctts_data;
			delete[] sample_to_chunk;
			delete[] sample_sizes;
			delete[] keyframes;
			delete extradata;
		}

		int ffindex; /* the ffmpeg stream id */
		int is_ff_stream; /* Is this stream presented to ffmpeg ? i.e. is this an audio or video stream ? */
		uint32 next_chunk;
		uint32 chunk_count;
		uint32 *chunk_offsets;
		int stts_count;
		MOVstts *stts_data;
		int ctts_count;
		MOVstts *ctts_data;
		int edit_count; /* number of 'edit' (elst atom) */
		uint32 sample_to_chunk_sz;
		MOVstsc *sample_to_chunk;
		int32 sample_to_chunk_index;
		int sample_to_time_index;
		uint32 sample_to_time_sample;
		uint32 sample_to_time_time;
		int sample_to_ctime_index;
		int sample_to_ctime_sample;
		uint32 sample_size;
		uint32 sample_count;
		uint32 *sample_sizes;
		uint32 keyframe_count;
		uint32 *keyframes;
		int32 time_scale;
		int time_rate;
		uint32 current_sample;
		uint32 left_in_chunk; /* how many samples before next chunk */

		uint16 width;
		uint16 height;
		int codec_type;
		uint32 codec_tag;
		char codec_name[32];
		uint16 bits_per_sample;
		uint16 color_table_id;
		bool palettized;
		Common::SeekableReadStream *extradata;

		uint16 stsd_version;
		uint16 channels;
		uint16 sample_rate;
		uint32 samples_per_frame;
		uint32 bytes_per_frame;

		uint32 nb_frames;
		uint32 duration;
		uint32 start_time;
		ScaleMode scaleMode;
	};

	const ParseTable *_parseTable;
	bool _foundMOOV;
	bool _foundMDAT;
	uint32 _timeScale;
	uint32 _duration;
	uint32 _mdatOffset;
	uint32 _mdatSize;
	uint32 _next_chunk_offset;
	MOVStreamContext *_partial;
	uint32 _numStreams;
	int _ni;
	ScaleMode _scaleMode;
	MOVStreamContext *_streams[20];
	byte _palette[256 * 4];

	void initParseTable();
	Audio::AudioStream *createAudioStream(Common::SeekableReadStream *stream);
	bool checkAudioCodecSupport(uint32 tag);
	Common::SeekableReadStream *getNextFramePacket();
	void resetInternal();
	uint32 getFrameDuration();

	QueuedAudioStream *_audStream;
	int8 _videoStreamIndex;
	int8 _audioStreamIndex;
	uint _curAudioChunk;
	uint32 _beginOffset;
	
	Graphics::Codec *createCodec(uint32 codecTag, byte bitsPerPixel);
	Graphics::Codec *_videoCodec;
	bool _noCodecFound;
	int32 _curFrame;
	uint32 _lastFrameStart, _nextFrameStart; // In 1/100 ms
	Audio::SoundHandle _audHandle;

	int readDefault(MOVatom atom);
	int readLeaf(MOVatom atom);
	int readELST(MOVatom atom);
	int readHDLR(MOVatom atom);
	int readMDAT(MOVatom atom);
	int readMDHD(MOVatom atom);
	int readMOOV(MOVatom atom);
	int readMVHD(MOVatom atom);
	int readTKHD(MOVatom atom);
	int readTRAK(MOVatom atom);
	int readSTCO(MOVatom atom);
	int readSTSC(MOVatom atom);
	int readSTSD(MOVatom atom);
	int readSTSS(MOVatom atom);
	int readSTSZ(MOVatom atom);
	int readSTTS(MOVatom atom);
	int readCMOV(MOVatom atom);
	int readWAVE(MOVatom atom);
};

} // End of namespace Mohawk

#endif