vrshoot

annotate libs/assimp/Q3BSPZipArchive.h @ 0:b2f14e535253

find changesets by author, revision, files, or words in the commit message
initial commit
author John Tsiombikas <nuclear@member.fsf.org>
date Sat, 01 Feb 2014 19:58:19 +0200
parents
children
rev   line source
nuclear@0 1 /*
nuclear@0 2 Open Asset Import Library (assimp)
nuclear@0 3 ----------------------------------------------------------------------
nuclear@0 4
nuclear@0 5 Copyright (c) 2006-2008, assimp team
nuclear@0 6 All rights reserved.
nuclear@0 7
nuclear@0 8 Redistribution and use of this software in source and binary forms,
nuclear@0 9 with or without modification, are permitted provided that the
nuclear@0 10 following conditions are met:
nuclear@0 11
nuclear@0 12 * Redistributions of source code must retain the above
nuclear@0 13 copyright notice, this list of conditions and the
nuclear@0 14 following disclaimer.
nuclear@0 15
nuclear@0 16 * Redistributions in binary form must reproduce the above
nuclear@0 17 copyright notice, this list of conditions and the
nuclear@0 18 following disclaimer in the documentation and/or other
nuclear@0 19 materials provided with the distribution.
nuclear@0 20
nuclear@0 21 * Neither the name of the assimp team, nor the names of its
nuclear@0 22 contributors may be used to endorse or promote products
nuclear@0 23 derived from this software without specific prior
nuclear@0 24 written permission of the assimp team.
nuclear@0 25
nuclear@0 26 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
nuclear@0 27 "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
nuclear@0 28 LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
nuclear@0 29 A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
nuclear@0 30 OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
nuclear@0 31 SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
nuclear@0 32 LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
nuclear@0 33 DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
nuclear@0 34 THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
nuclear@0 35 (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
nuclear@0 36 OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
nuclear@0 37
nuclear@0 38 ----------------------------------------------------------------------
nuclear@0 39 */
nuclear@0 40 #ifndef AI_Q3BSP_ZIPARCHIVE_H_INC
nuclear@0 41 #define AI_Q3BSP_ZIPARCHIVE_H_INC
nuclear@0 42
nuclear@0 43 #include "../contrib/unzip/unzip.h"
nuclear@0 44 #include "assimp/IOStream.hpp"
nuclear@0 45 #include "assimp/IOSystem.hpp"
nuclear@0 46 #include <string>
nuclear@0 47 #include <vector>
nuclear@0 48 #include <map>
nuclear@0 49 #include <cassert>
nuclear@0 50
nuclear@0 51 namespace Assimp
nuclear@0 52 {
nuclear@0 53 namespace Q3BSP
nuclear@0 54 {
nuclear@0 55
nuclear@0 56 // ------------------------------------------------------------------------------------------------
nuclear@0 57 /// \class ZipFile
nuclear@0 58 /// \ingroup Assimp::Q3BSP
nuclear@0 59 ///
nuclear@0 60 /// \brief
nuclear@0 61 // ------------------------------------------------------------------------------------------------
nuclear@0 62 class ZipFile : public IOStream
nuclear@0 63 {
nuclear@0 64 public:
nuclear@0 65 ZipFile( const std::string &rFileName, unzFile zipFile ) :
nuclear@0 66 m_Name( rFileName ),
nuclear@0 67 m_zipFile( zipFile )
nuclear@0 68 {
nuclear@0 69 ai_assert( NULL != m_zipFile );
nuclear@0 70 }
nuclear@0 71
nuclear@0 72 ~ZipFile()
nuclear@0 73 {
nuclear@0 74 m_zipFile = NULL;
nuclear@0 75 }
nuclear@0 76
nuclear@0 77 size_t Read(void* pvBuffer, size_t pSize, size_t pCount )
nuclear@0 78 {
nuclear@0 79 size_t bytes_read = 0;
nuclear@0 80 if ( NULL == m_zipFile )
nuclear@0 81 return bytes_read;
nuclear@0 82
nuclear@0 83 // search file and place file pointer there
nuclear@0 84 if ( unzLocateFile( m_zipFile, m_Name.c_str(), 0 ) == UNZ_OK )
nuclear@0 85 {
nuclear@0 86 // get file size, etc.
nuclear@0 87 unz_file_info fileInfo;
nuclear@0 88 unzGetCurrentFileInfo( m_zipFile, &fileInfo, 0, 0, 0, 0, 0, 0 );
nuclear@0 89 const size_t size = pSize * pCount;
nuclear@0 90 assert( size <= fileInfo.uncompressed_size );
nuclear@0 91
nuclear@0 92 // The file has EXACTLY the size of uncompressed_size. In C
nuclear@0 93 // you need to mark the last character with '\0', so add
nuclear@0 94 // another character
nuclear@0 95 unzOpenCurrentFile( m_zipFile );
nuclear@0 96 const int ret = unzReadCurrentFile( m_zipFile, pvBuffer, fileInfo.uncompressed_size);
nuclear@0 97 size_t filesize = fileInfo.uncompressed_size;
nuclear@0 98 if ( ret < 0 || size_t(ret) != filesize )
nuclear@0 99 {
nuclear@0 100 return 0;
nuclear@0 101 }
nuclear@0 102 bytes_read = ret;
nuclear@0 103 unzCloseCurrentFile( m_zipFile );
nuclear@0 104 }
nuclear@0 105 return bytes_read;
nuclear@0 106 }
nuclear@0 107
nuclear@0 108 size_t Write(const void* /*pvBuffer*/, size_t /*pSize*/, size_t /*pCount*/)
nuclear@0 109 {
nuclear@0 110 return 0;
nuclear@0 111 }
nuclear@0 112
nuclear@0 113 size_t FileSize() const
nuclear@0 114 {
nuclear@0 115 if ( NULL == m_zipFile )
nuclear@0 116 return 0;
nuclear@0 117 if ( unzLocateFile( m_zipFile, m_Name.c_str(), 0 ) == UNZ_OK )
nuclear@0 118 {
nuclear@0 119 unz_file_info fileInfo;
nuclear@0 120 unzGetCurrentFileInfo( m_zipFile, &fileInfo, 0, 0, 0, 0, 0, 0 );
nuclear@0 121 return fileInfo.uncompressed_size;
nuclear@0 122 }
nuclear@0 123 return 0;
nuclear@0 124 }
nuclear@0 125
nuclear@0 126 aiReturn Seek(size_t /*pOffset*/, aiOrigin /*pOrigin*/)
nuclear@0 127 {
nuclear@0 128 return aiReturn_FAILURE;
nuclear@0 129 }
nuclear@0 130
nuclear@0 131 size_t Tell() const
nuclear@0 132 {
nuclear@0 133 return 0;
nuclear@0 134 }
nuclear@0 135
nuclear@0 136 void Flush()
nuclear@0 137 {
nuclear@0 138 // empty
nuclear@0 139 }
nuclear@0 140
nuclear@0 141 private:
nuclear@0 142 std::string m_Name;
nuclear@0 143 unzFile m_zipFile;
nuclear@0 144 };
nuclear@0 145
nuclear@0 146 // ------------------------------------------------------------------------------------------------
nuclear@0 147 /// \class Q3BSPZipArchive
nuclear@0 148 /// \ingroup Assimp::Q3BSP
nuclear@0 149 ///
nuclear@0 150 /// \brief IMplements a zip archive like the WinZip archives. Will be also used to import data
nuclear@0 151 /// from a P3K archive ( Quake level format ).
nuclear@0 152 // ------------------------------------------------------------------------------------------------
nuclear@0 153 class Q3BSPZipArchive : public Assimp::IOSystem
nuclear@0 154 {
nuclear@0 155 public:
nuclear@0 156 static const unsigned int FileNameSize = 256;
nuclear@0 157
nuclear@0 158 public:
nuclear@0 159 Q3BSPZipArchive( const std::string & rFile );
nuclear@0 160 ~Q3BSPZipArchive();
nuclear@0 161 bool Exists( const char* pFile) const;
nuclear@0 162 char getOsSeparator() const;
nuclear@0 163 IOStream* Open(const char* pFile, const char* pMode = "rb");
nuclear@0 164 void Close( IOStream* pFile);
nuclear@0 165 bool isOpen() const;
nuclear@0 166 void getFileList( std::vector<std::string> &rFileList );
nuclear@0 167
nuclear@0 168 private:
nuclear@0 169 bool mapArchive();
nuclear@0 170
nuclear@0 171 private:
nuclear@0 172 unzFile m_ZipFileHandle;
nuclear@0 173 std::map<std::string, IOStream*> m_ArchiveMap;
nuclear@0 174 std::vector<std::string> m_FileList;
nuclear@0 175 bool m_bDirty;
nuclear@0 176 };
nuclear@0 177
nuclear@0 178 // ------------------------------------------------------------------------------------------------
nuclear@0 179
nuclear@0 180 } // Namespace Q3BSP
nuclear@0 181 } // Namespace Assimp
nuclear@0 182
nuclear@0 183 #endif // AI_Q3BSP_ZIPARCHIVE_H_INC