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
|
nuclear@0
|
41 #include "AssimpPCH.h"
|
nuclear@0
|
42
|
nuclear@0
|
43 #ifndef ASSIMP_BUILD_NO_Q3BSP_IMPORTER
|
nuclear@0
|
44
|
nuclear@0
|
45 #include "Q3BSPZipArchive.h"
|
nuclear@0
|
46 #include <algorithm>
|
nuclear@0
|
47 #include <cassert>
|
nuclear@0
|
48
|
nuclear@0
|
49 namespace Assimp
|
nuclear@0
|
50 {
|
nuclear@0
|
51 namespace Q3BSP
|
nuclear@0
|
52 {
|
nuclear@0
|
53
|
nuclear@0
|
54 // ------------------------------------------------------------------------------------------------
|
nuclear@0
|
55 // Constructor.
|
nuclear@0
|
56 Q3BSPZipArchive::Q3BSPZipArchive( const std::string& rFile ) :
|
nuclear@0
|
57 m_ZipFileHandle( NULL ),
|
nuclear@0
|
58 m_FileList(),
|
nuclear@0
|
59 m_bDirty( true )
|
nuclear@0
|
60 {
|
nuclear@0
|
61 if ( !rFile.empty() )
|
nuclear@0
|
62 {
|
nuclear@0
|
63 m_ZipFileHandle = unzOpen( rFile.c_str() );
|
nuclear@0
|
64 if ( NULL != m_ZipFileHandle )
|
nuclear@0
|
65 {
|
nuclear@0
|
66 mapArchive();
|
nuclear@0
|
67 }
|
nuclear@0
|
68 }
|
nuclear@0
|
69 }
|
nuclear@0
|
70
|
nuclear@0
|
71 // ------------------------------------------------------------------------------------------------
|
nuclear@0
|
72 // Destructor.
|
nuclear@0
|
73 Q3BSPZipArchive::~Q3BSPZipArchive()
|
nuclear@0
|
74 {
|
nuclear@0
|
75 if ( NULL != m_ZipFileHandle )
|
nuclear@0
|
76 {
|
nuclear@0
|
77 unzClose( m_ZipFileHandle );
|
nuclear@0
|
78 }
|
nuclear@0
|
79 m_ZipFileHandle = NULL;
|
nuclear@0
|
80 m_FileList.clear();
|
nuclear@0
|
81 }
|
nuclear@0
|
82
|
nuclear@0
|
83 // ------------------------------------------------------------------------------------------------
|
nuclear@0
|
84 // Returns true, if the archive is already open.
|
nuclear@0
|
85 bool Q3BSPZipArchive::isOpen() const
|
nuclear@0
|
86 {
|
nuclear@0
|
87 return ( NULL != m_ZipFileHandle );
|
nuclear@0
|
88 }
|
nuclear@0
|
89
|
nuclear@0
|
90 // ------------------------------------------------------------------------------------------------
|
nuclear@0
|
91 // Returns true, if the filename is part of the archive.
|
nuclear@0
|
92 bool Q3BSPZipArchive::Exists( const char* pFile ) const
|
nuclear@0
|
93 {
|
nuclear@0
|
94 ai_assert( NULL != pFile );
|
nuclear@0
|
95 if ( NULL == pFile )
|
nuclear@0
|
96 {
|
nuclear@0
|
97 return false;
|
nuclear@0
|
98 }
|
nuclear@0
|
99
|
nuclear@0
|
100 std::string rFile( pFile );
|
nuclear@0
|
101 std::vector<std::string>::const_iterator it = std::find( m_FileList.begin(), m_FileList.end(), rFile );
|
nuclear@0
|
102 if ( m_FileList.end() == it )
|
nuclear@0
|
103 {
|
nuclear@0
|
104 return false;
|
nuclear@0
|
105 }
|
nuclear@0
|
106
|
nuclear@0
|
107 return true;
|
nuclear@0
|
108 }
|
nuclear@0
|
109
|
nuclear@0
|
110 // ------------------------------------------------------------------------------------------------
|
nuclear@0
|
111 // Returns the separator delimiter.
|
nuclear@0
|
112 char Q3BSPZipArchive::getOsSeparator() const
|
nuclear@0
|
113 {
|
nuclear@0
|
114 return '/';
|
nuclear@0
|
115 }
|
nuclear@0
|
116
|
nuclear@0
|
117 // ------------------------------------------------------------------------------------------------
|
nuclear@0
|
118 // Opens a file, which is part of the archive.
|
nuclear@0
|
119 IOStream *Q3BSPZipArchive::Open( const char* pFile, const char* /*pMode*/ )
|
nuclear@0
|
120 {
|
nuclear@0
|
121 ai_assert( NULL != pFile );
|
nuclear@0
|
122
|
nuclear@0
|
123 std::string rItem( pFile );
|
nuclear@0
|
124 std::vector<std::string>::iterator it = std::find( m_FileList.begin(), m_FileList.end(), rItem );
|
nuclear@0
|
125 if ( m_FileList.end() == it )
|
nuclear@0
|
126 return NULL;
|
nuclear@0
|
127
|
nuclear@0
|
128 ZipFile *pZipFile = new ZipFile( *it, m_ZipFileHandle );
|
nuclear@0
|
129 m_ArchiveMap[ rItem ] = pZipFile;
|
nuclear@0
|
130
|
nuclear@0
|
131 return pZipFile;
|
nuclear@0
|
132 }
|
nuclear@0
|
133
|
nuclear@0
|
134 // ------------------------------------------------------------------------------------------------
|
nuclear@0
|
135 // Close a filestream.
|
nuclear@0
|
136 void Q3BSPZipArchive::Close( IOStream *pFile )
|
nuclear@0
|
137 {
|
nuclear@0
|
138 ai_assert( NULL != pFile );
|
nuclear@0
|
139
|
nuclear@0
|
140 std::map<std::string, IOStream*>::iterator it;
|
nuclear@0
|
141 for ( it = m_ArchiveMap.begin(); it != m_ArchiveMap.end(); ++it )
|
nuclear@0
|
142 {
|
nuclear@0
|
143 if ( (*it).second == pFile )
|
nuclear@0
|
144 {
|
nuclear@0
|
145 ZipFile *pZipFile = reinterpret_cast<ZipFile*>( (*it).second );
|
nuclear@0
|
146 delete pZipFile;
|
nuclear@0
|
147 m_ArchiveMap.erase( it );
|
nuclear@0
|
148 break;
|
nuclear@0
|
149 }
|
nuclear@0
|
150 }
|
nuclear@0
|
151 }
|
nuclear@0
|
152 // ------------------------------------------------------------------------------------------------
|
nuclear@0
|
153 // Returns the file-list of the archive.
|
nuclear@0
|
154 void Q3BSPZipArchive::getFileList( std::vector<std::string> &rFileList )
|
nuclear@0
|
155 {
|
nuclear@0
|
156 rFileList = m_FileList;
|
nuclear@0
|
157 }
|
nuclear@0
|
158
|
nuclear@0
|
159 // ------------------------------------------------------------------------------------------------
|
nuclear@0
|
160 // Maps the archive content.
|
nuclear@0
|
161 bool Q3BSPZipArchive::mapArchive()
|
nuclear@0
|
162 {
|
nuclear@0
|
163 if ( NULL == m_ZipFileHandle )
|
nuclear@0
|
164 return false;
|
nuclear@0
|
165
|
nuclear@0
|
166 if ( !m_bDirty )
|
nuclear@0
|
167 return true;
|
nuclear@0
|
168
|
nuclear@0
|
169 if ( !m_FileList.empty() )
|
nuclear@0
|
170 m_FileList.resize( 0 );
|
nuclear@0
|
171
|
nuclear@0
|
172 // At first ensure file is already open
|
nuclear@0
|
173 if ( UNZ_OK == unzGoToFirstFile( m_ZipFileHandle ) )
|
nuclear@0
|
174 {
|
nuclear@0
|
175 char filename[ FileNameSize ];
|
nuclear@0
|
176 unzGetCurrentFileInfo( m_ZipFileHandle, NULL, filename, FileNameSize, NULL, 0, NULL, 0 );
|
nuclear@0
|
177 m_FileList.push_back( filename );
|
nuclear@0
|
178 unzCloseCurrentFile( m_ZipFileHandle );
|
nuclear@0
|
179
|
nuclear@0
|
180 // Loop over all files
|
nuclear@0
|
181 while ( unzGoToNextFile( m_ZipFileHandle ) != UNZ_END_OF_LIST_OF_FILE )
|
nuclear@0
|
182 {
|
nuclear@0
|
183 char filename[ FileNameSize ];
|
nuclear@0
|
184 unzGetCurrentFileInfo( m_ZipFileHandle, NULL, filename, FileNameSize, NULL, 0, NULL, 0 );
|
nuclear@0
|
185 m_FileList.push_back( filename );
|
nuclear@0
|
186 unzCloseCurrentFile( m_ZipFileHandle );
|
nuclear@0
|
187 }
|
nuclear@0
|
188 }
|
nuclear@0
|
189
|
nuclear@0
|
190 std::sort( m_FileList.begin(), m_FileList.end() );
|
nuclear@0
|
191 m_bDirty = false;
|
nuclear@0
|
192
|
nuclear@0
|
193 return true;
|
nuclear@0
|
194 }
|
nuclear@0
|
195
|
nuclear@0
|
196 // ------------------------------------------------------------------------------------------------
|
nuclear@0
|
197
|
nuclear@0
|
198 } // Namespace Q3BSP
|
nuclear@0
|
199 } // Namespace Assimp
|
nuclear@0
|
200
|
nuclear@0
|
201 #endif // ASSIMP_BUILD_NO_Q3BSP_IMPORTER
|