nuclear@0: /*
nuclear@0: Open Asset Import Library (assimp)
nuclear@0: ----------------------------------------------------------------------
nuclear@0: 
nuclear@0: Copyright (c) 2006-2008, assimp team
nuclear@0: All rights reserved.
nuclear@0: 
nuclear@0: Redistribution and use of this software in source and binary forms, 
nuclear@0: with or without modification, are permitted provided that the 
nuclear@0: following conditions are met:
nuclear@0: 
nuclear@0: * Redistributions of source code must retain the above
nuclear@0:   copyright notice, this list of conditions and the
nuclear@0:   following disclaimer.
nuclear@0: 
nuclear@0: * Redistributions in binary form must reproduce the above
nuclear@0:   copyright notice, this list of conditions and the
nuclear@0:   following disclaimer in the documentation and/or other
nuclear@0:   materials provided with the distribution.
nuclear@0: 
nuclear@0: * Neither the name of the assimp team, nor the names of its
nuclear@0:   contributors may be used to endorse or promote products
nuclear@0:   derived from this software without specific prior
nuclear@0:   written permission of the assimp team.
nuclear@0: 
nuclear@0: THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 
nuclear@0: "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 
nuclear@0: LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
nuclear@0: A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 
nuclear@0: OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
nuclear@0: SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 
nuclear@0: LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
nuclear@0: DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 
nuclear@0: THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 
nuclear@0: (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 
nuclear@0: OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
nuclear@0: 
nuclear@0: ----------------------------------------------------------------------
nuclear@0: */
nuclear@0: #ifndef AI_Q3BSP_ZIPARCHIVE_H_INC
nuclear@0: #define AI_Q3BSP_ZIPARCHIVE_H_INC
nuclear@0: 
nuclear@0: #include "../contrib/unzip/unzip.h"
nuclear@0: #include "assimp/IOStream.hpp"
nuclear@0: #include "assimp/IOSystem.hpp"
nuclear@0: #include <string>
nuclear@0: #include <vector>
nuclear@0: #include <map>
nuclear@0: #include <cassert>
nuclear@0: 
nuclear@0: namespace Assimp
nuclear@0: {
nuclear@0: namespace Q3BSP
nuclear@0: {
nuclear@0: 
nuclear@0: // ------------------------------------------------------------------------------------------------
nuclear@0: ///	\class		ZipFile
nuclear@0: ///	\ingroup	Assimp::Q3BSP
nuclear@0: ///
nuclear@0: ///	\brief
nuclear@0: // ------------------------------------------------------------------------------------------------
nuclear@0: class ZipFile : public IOStream
nuclear@0: {
nuclear@0: public:
nuclear@0: 	ZipFile( const std::string &rFileName, unzFile zipFile ) :
nuclear@0: 		m_Name( rFileName ),
nuclear@0: 		m_zipFile( zipFile )
nuclear@0: 	{
nuclear@0: 		ai_assert( NULL != m_zipFile );
nuclear@0: 	}
nuclear@0: 	
nuclear@0: 	~ZipFile()
nuclear@0: 	{
nuclear@0: 		m_zipFile = NULL;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	size_t Read(void* pvBuffer, size_t pSize, size_t pCount )
nuclear@0: 	{
nuclear@0: 		size_t bytes_read = 0;
nuclear@0: 		if ( NULL == m_zipFile )
nuclear@0: 			return bytes_read;
nuclear@0: 		
nuclear@0: 		// search file and place file pointer there
nuclear@0: 		if ( unzLocateFile( m_zipFile, m_Name.c_str(), 0 ) == UNZ_OK )
nuclear@0: 		{
nuclear@0: 			// get file size, etc.
nuclear@0: 			unz_file_info fileInfo;
nuclear@0: 			unzGetCurrentFileInfo( m_zipFile, &fileInfo, 0, 0, 0, 0, 0, 0 );
nuclear@0: 			const size_t size = pSize * pCount;
nuclear@0: 			assert( size <= fileInfo.uncompressed_size );
nuclear@0: 			
nuclear@0: 			// The file has EXACTLY the size of uncompressed_size. In C
nuclear@0: 			// you need to mark the last character with '\0', so add 
nuclear@0: 			// another character
nuclear@0: 			unzOpenCurrentFile( m_zipFile );
nuclear@0: 			const int ret = unzReadCurrentFile( m_zipFile, pvBuffer, fileInfo.uncompressed_size);
nuclear@0: 			size_t filesize = fileInfo.uncompressed_size;
nuclear@0: 			if ( ret < 0 || size_t(ret) != filesize )
nuclear@0: 			{
nuclear@0: 				return 0;
nuclear@0: 			}
nuclear@0: 			bytes_read = ret;
nuclear@0: 			unzCloseCurrentFile( m_zipFile );
nuclear@0: 		}
nuclear@0: 		return bytes_read;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	size_t Write(const void* /*pvBuffer*/, size_t /*pSize*/, size_t /*pCount*/)
nuclear@0: 	{
nuclear@0: 		return 0;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	size_t FileSize() const
nuclear@0: 	{
nuclear@0: 		if ( NULL == m_zipFile )
nuclear@0: 			return 0;
nuclear@0: 		if ( unzLocateFile( m_zipFile, m_Name.c_str(), 0 ) == UNZ_OK ) 
nuclear@0: 		{
nuclear@0: 			unz_file_info fileInfo;
nuclear@0: 			unzGetCurrentFileInfo( m_zipFile, &fileInfo, 0, 0, 0, 0, 0, 0 );
nuclear@0: 			return fileInfo.uncompressed_size;
nuclear@0: 		}
nuclear@0: 		return 0;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	aiReturn Seek(size_t /*pOffset*/, aiOrigin /*pOrigin*/)
nuclear@0: 	{
nuclear@0: 		return aiReturn_FAILURE;
nuclear@0: 	}
nuclear@0: 
nuclear@0:     size_t Tell() const
nuclear@0: 	{
nuclear@0: 		return 0;
nuclear@0: 	}
nuclear@0: 
nuclear@0: 	void Flush()
nuclear@0: 	{
nuclear@0: 		// empty
nuclear@0: 	}
nuclear@0: 
nuclear@0: private:
nuclear@0: 	std::string m_Name;
nuclear@0: 	unzFile m_zipFile;
nuclear@0: };
nuclear@0: 
nuclear@0: // ------------------------------------------------------------------------------------------------
nuclear@0: ///	\class		Q3BSPZipArchive
nuclear@0: ///	\ingroup	Assimp::Q3BSP
nuclear@0: ///	
nuclear@0: ///	\brief	IMplements a zip archive like the WinZip archives. Will be also used to import data 
nuclear@0: ///	from a P3K archive ( Quake level format ).
nuclear@0: // ------------------------------------------------------------------------------------------------
nuclear@0: class Q3BSPZipArchive : public Assimp::IOSystem
nuclear@0: {
nuclear@0: public:
nuclear@0: 	static const unsigned int FileNameSize = 256;
nuclear@0: 
nuclear@0: public:
nuclear@0: 	Q3BSPZipArchive( const std::string & rFile );
nuclear@0: 	~Q3BSPZipArchive();
nuclear@0: 	bool Exists( const char* pFile) const;
nuclear@0: 	char getOsSeparator() const;
nuclear@0: 	IOStream* Open(const char* pFile, const char* pMode = "rb");
nuclear@0: 	void Close( IOStream* pFile);
nuclear@0: 	bool isOpen() const;
nuclear@0: 	void getFileList( std::vector<std::string> &rFileList );
nuclear@0: 
nuclear@0: private:
nuclear@0: 	bool mapArchive();
nuclear@0: 
nuclear@0: private:
nuclear@0: 	unzFile m_ZipFileHandle;
nuclear@0: 	std::map<std::string, IOStream*> m_ArchiveMap;
nuclear@0: 	std::vector<std::string> m_FileList;
nuclear@0: 	bool m_bDirty;
nuclear@0: };
nuclear@0: 
nuclear@0: // ------------------------------------------------------------------------------------------------
nuclear@0: 
nuclear@0: } // Namespace Q3BSP
nuclear@0: } // Namespace Assimp
nuclear@0: 
nuclear@0: #endif // AI_Q3BSP_ZIPARCHIVE_H_INC