// Copyright (c) 2009, Karl Blomster
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
//   * Redistributions of source code must retain the above copyright notice,
//     this list of conditions and the following disclaimer.
//   * Redistributions in binary form must reproduce the above copyright notice,
//     this list of conditions and the following disclaimer in the documentation
//     and/or other materials provided with the distribution.
//   * Neither the name of the Aegisub Group nor the names of its contributors
//     may be used to endorse or promote products derived from this software
//     without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
//
// -----------------------------------------------------------------------------
//
// AEGISUB
//
// Website: http://aegisub.cellosoft.com
// Contact: mailto:zeratul@cellosoft.com
//


#pragma once

#include "include/aegisub/video_provider.h"
#include <wx/wxprec.h>
#include <wx/filename.h>
#include <wx/log.h>
#include <stdio.h>
#include <vector>

// ffmpeg uses 80, so I'm p sure this isn't too small
#define YUV4MPEG_HEADER_MAXLEN 128


class YUV4MPEGVideoProvider : public VideoProvider {
private:
	enum Y4M_PixelFormat {
		Y4M_PIXFMT_NONE		= -1,
		Y4M_PIXFMT_420JPEG,			// afaict the only difference between
		Y4M_PIXFMT_420MPEG2,		// these three is the chroma sample location,
		Y4M_PIXFMT_420PALDV,		// and nobody cares about that.
		Y4M_PIXFMT_411,
		Y4M_PIXFMT_422,
		Y4M_PIXFMT_444,
		Y4M_PIXFMT_444ALPHA,
		Y4M_PIXFMT_MONO,
	};

	enum Y4M_InterlacingMode {
		Y4M_ILACE_NOTSET = -1, // not to be confused with Y4M_ILACE_UNKNOWN
		Y4M_ILACE_PROGRESSIVE,
		Y4M_ILACE_TFF,
		Y4M_ILACE_BFF,
		Y4M_ILACE_MIXED,
		Y4M_ILACE_UNKNOWN,
	};

	// this is currently unused :(
	enum Y4M_FrameFlags {
		Y4M_FFLAG_NOTSET	= -1,
		Y4M_FFLAG_NONE		= 0x0000,
		// repeat field/frame flags
		Y4M_FFLAG_R_TFF		= 0x0001, // TFF
		Y4M_FFLAG_R_TFF_R	= 0x0002, // TFF and repeat
		Y4M_FFLAG_R_BFF		= 0x0004, // BFF
		Y4M_FFLAG_R_BFF_R	= 0x0008, // BFF and repeat
		Y4M_FFLAG_R_P		= 0x0010, // progressive
		Y4M_FFLAG_R_P_R		= 0x0020, // progressive and repeat once
		Y4M_FFLAG_R_P_RR	= 0x0040, // progressive and repeat twice
		// temporal sampling flags
		Y4M_FFLAG_T_P		= 0x0080, // progressive (fields sampled at the same time)
		Y4M_FFLAG_T_I		= 0x0100, // interlaced (fields sampled at different times)
		// spatial sampling flags
		Y4M_FFLAG_C_P		= 0x0200, // progressive (whole frame subsampled)
		Y4M_FFLAG_C_I		= 0x0400, // interlaced (fields subsampled independently)
		Y4M_FFLAG_C_UNKNOWN = 0x0800, // unknown (only allowed for non-4:2:0 sampling)
	};

	FILE *sf;		// source file
	bool inited;
	int w, h;		// width/height
	int num_frames; // length of file in frames
	int frame_sz;	// size of each frame in bytes
	Y4M_PixelFormat pixfmt; // colorspace/pixel format
	Y4M_InterlacingMode imode; // interlacing mode
	struct {
		int num;
		int den;
	} fps_rat;		// framerate

	std::vector<int64_t> seek_table;	// the position in the file of each frame, in bytes
	int cur_fn;							// current frame number

	wxString errmsg;

	void LoadVideo(const Aegisub::String filename);
	void Close();
	
	void CheckFileFormat();
	void ParseFileHeader(const std::vector<wxString>& tags);
	Y4M_FrameFlags ParseFrameHeader(const std::vector<wxString>& tags);
	std::vector<wxString> ReadHeader(int64_t startpos, bool reset_pos=false);
	int IndexFile();

public:
	YUV4MPEGVideoProvider(Aegisub::String filename);
	~YUV4MPEGVideoProvider();

	const AegiVideoFrame GetFrame(int n);
	int GetPosition();
	int GetFrameCount();

	int GetWidth();
	int GetHeight();
	double GetFPS();
	bool AreKeyFramesLoaded() { return false; }
	wxArrayInt GetKeyFrames() { return wxArrayInt(); }
	bool IsVFR() { return false; };
	FrameRate GetTrueFrameRate() { return FrameRate(); }
	Aegisub::String GetDecoderName() { return L"YUV4MPEG"; }
	int GetDesiredCacheSize() { return 8; }
};