// Copyright (C) 2002-2012 Thomas Alten
// This file is part of the "Irrlicht Engine".
// For conditions of distribution and use, see copyright notice in irrlicht.h

#ifndef IRR_C_WAD_READER_H_INCLUDED
#define IRR_C_WAD_READER_H_INCLUDED

#include "IrrCompileConfig.h"
#ifdef __IRR_COMPILE_WITH_WAD_ARCHIVE_LOADER_

#include "IReferenceCounted.h"
#include "IReadFile.h"
#include "irrArray.h"
#include "irrString.h"
#include "IFileSystem.h"
#include "CFileList.h"


namespace irr
{
namespace io
{

	enum eWADFileTypes
	{
		WAD_FORMAT_UNKNOWN	= 0,
		WAD_FORMAT_QUAKE2	= 1,
		WAD_FORMAT_HALFLIFE	= 2,

		WAD_CMP_NONE	= 0,
		WAD_CMP_LZSS	= 1,

		WAD_TYP_NONE	= 0,
		WAD_TYP_LABEL	= 1,

		WAD_TYP_LUMPY	= 64,				// 64 + grab command number
		WAD_TYP_PALETTE	= 64,
		WAD_TYP_QTEX	= 65,
		WAD_TYP_QPIC	= 66,
		WAD_TYP_SOUND	= 67,
		WAD_TYP_MIPTEX	= 68,
		WAD_TYP_MIPTEX_HALFLIFE = 67,
		WAD_TYP_FONT	= 70,
	};

// byte-align structures
#include "irrpack.h"

	struct SWADFileHeader
	{
		c8 tag[4]; // type of WAD format WAD2 = quake2, WAD3 = halflife
		u32 numlumps;
		u32 infotableofs;
	} PACK_STRUCT;

	struct SWADFileEntryOriginal
	{
		u32	filepos;
		u32 disksize;
		u32 size;					// uncompressed
		u8 type;
		u8 compression;
		u8 pad[2];
		u8 name[16];				// must be null terminated
	} PACK_STRUCT;

// Default alignment
#include "irrunpack.h"

	struct SWADFileEntry
	{
		io::path simpleFileName;
		bool operator < (const SWADFileEntry& other) const
		{
			return simpleFileName < other.simpleFileName;
		}

		io::path wadFileName;
		SWADFileEntryOriginal header;
	};

	//! Archiveloader capable of loading WAD Archives
	class CArchiveLoaderWAD : public IArchiveLoader
	{
	public:

		//! Constructor
		CArchiveLoaderWAD(io::IFileSystem* fs);

		//! returns true if the file maybe is able to be loaded by this class
		//! based on the file extension (e.g. ".zip")
		virtual bool isALoadableFileFormat(const io::path& filename) const IRR_OVERRIDE;

		//! Check if the file might be loaded by this class
		/** Check might look into the file.
		\param file File handle to check.
		\return True if file seems to be loadable. */
		virtual bool isALoadableFileFormat(io::IReadFile* file) const IRR_OVERRIDE;

		//! Check to see if the loader can create archives of this type.
		/** Check based on the archive type.
		\param fileType The archive type to check.
		\return True if the archile loader supports this type, false if not */
		virtual bool isALoadableFileFormat(E_FILE_ARCHIVE_TYPE fileType) const IRR_OVERRIDE;

		//! Creates an archive from the filename
		/** \param file File handle to check.
		\return Pointer to newly created archive, or 0 upon error. */
		virtual IFileArchive* createArchive(const io::path& filename, bool ignoreCase, bool ignorePaths) const IRR_OVERRIDE;

		//! creates/loads an archive from the file.
		//! \return Pointer to the created archive. Returns 0 if loading failed.
		virtual io::IFileArchive* createArchive(io::IReadFile* file, bool ignoreCase, bool ignorePaths) const IRR_OVERRIDE;

	private:
		io::IFileSystem* FileSystem;
	};


	//! reads from WAD
	class CWADReader : public IFileArchive, virtual CFileList
	{
	public:

		CWADReader(IReadFile* file, bool ignoreCase, bool ignorePaths);
		virtual ~CWADReader();

		// file archive methods

		//! return the id of the file Archive
		virtual const io::path& getArchiveName() const IRR_OVERRIDE;

		//! opens a file by file name
		virtual IReadFile* createAndOpenFile(const io::path& filename) IRR_OVERRIDE;

		//! opens a file by index
		virtual IReadFile* createAndOpenFile(u32 index) IRR_OVERRIDE;

		//! returns the list of files
		virtual const IFileList* getFileList() const IRR_OVERRIDE;

		//! get the class Type
		virtual E_FILE_ARCHIVE_TYPE getType() const IRR_OVERRIDE { return EFAT_WAD; }


	private:

		io::path Type;

		//! scans for a local header, returns false if there is no more local file header.
		bool scanLocalHeader();

		//! splits filename from zip file into useful filenames and paths
		void extractFilename(SWADFileEntry* entry);

		io::path Base;
		io::path MountPoint;

		IReadFile* File;

		eWADFileTypes WadType;
		SWADFileHeader Header;

		//core::array<SWADFileEntry> FileInfo;

		io::IFileSystem* FileSystem;
	};

} // end namespace io
} // end namespace irr

#endif	// #ifdef __IRR_COMPILE_WITH_WAD_ARCHIVE_LOADER_

#endif