/[svn]/libgig/trunk/src/RIFF.cpp
ViewVC logotype

Annotation of /libgig/trunk/src/RIFF.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 2450 - (hide annotations) (download)
Wed May 8 17:53:07 2013 UTC (10 years, 11 months ago) by persson
File size: 77333 byte(s)
* added write support for CtrlTrigger midi rule
* added read and write support for Legato and Alternator midi rules

1 schoenebeck 2 /***************************************************************************
2     * *
3 schoenebeck 933 * libgig - C++ cross-platform Gigasampler format file access library *
4 schoenebeck 2 * *
5 persson 2450 * Copyright (C) 2003-2013 by Christian Schoenebeck *
6 schoenebeck 384 * <cuse@users.sourceforge.net> *
7 schoenebeck 2 * *
8     * This library is free software; you can redistribute it and/or modify *
9     * it under the terms of the GNU General Public License as published by *
10     * the Free Software Foundation; either version 2 of the License, or *
11     * (at your option) any later version. *
12     * *
13     * This library is distributed in the hope that it will be useful, *
14     * but WITHOUT ANY WARRANTY; without even the implied warranty of *
15     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
16     * GNU General Public License for more details. *
17     * *
18     * You should have received a copy of the GNU General Public License *
19     * along with this library; if not, write to the Free Software *
20     * Foundation, Inc., 59 Temple Place, Suite 330, Boston, *
21     * MA 02111-1307 USA *
22     ***************************************************************************/
23 schoenebeck 780
24 persson 1713 #include <algorithm>
25 schoenebeck 1953 #include <set>
26 schoenebeck 780 #include <string.h>
27    
28 schoenebeck 2 #include "RIFF.h"
29    
30 schoenebeck 800 #include "helper.h"
31    
32 schoenebeck 2 namespace RIFF {
33    
34 schoenebeck 1105 // *************** Internal functions **************
35     // *
36    
37     /// Returns a human readable path of the given chunk.
38 schoenebeck 1383 static String __resolveChunkPath(Chunk* pCk) {
39 schoenebeck 1105 String sPath;
40     for (Chunk* pChunk = pCk; pChunk; pChunk = pChunk->GetParent()) {
41     if (pChunk->GetChunkID() == CHUNK_ID_LIST) {
42     List* pList = (List*) pChunk;
43     sPath = "->'" + pList->GetListTypeString() + "'" + sPath;
44     } else {
45     sPath = "->'" + pChunk->GetChunkIDString() + "'" + sPath;
46     }
47     }
48     return sPath;
49     }
50    
51    
52    
53 schoenebeck 2 // *************** Chunk **************
54     // *
55    
56 schoenebeck 780 Chunk::Chunk(File* pFile) {
57 schoenebeck 2 #if DEBUG
58 schoenebeck 780 std::cout << "Chunk::Chunk(File* pFile)" << std::endl;
59 schoenebeck 2 #endif // DEBUG
60     ulPos = 0;
61     pParent = NULL;
62     pChunkData = NULL;
63 schoenebeck 1863 CurrentChunkSize = 0;
64     NewChunkSize = 0;
65 schoenebeck 800 ulChunkDataSize = 0;
66 schoenebeck 798 ChunkID = CHUNK_ID_RIFF;
67 schoenebeck 780 this->pFile = pFile;
68 schoenebeck 2 }
69    
70 schoenebeck 780 Chunk::Chunk(File* pFile, unsigned long StartPos, List* Parent) {
71 schoenebeck 2 #if DEBUG
72 schoenebeck 780 std::cout << "Chunk::Chunk(File*,ulong,bool,List*),StartPos=" << StartPos << std::endl;
73 schoenebeck 2 #endif // DEBUG
74 schoenebeck 780 this->pFile = pFile;
75 schoenebeck 2 ulStartPos = StartPos + CHUNK_HEADER_SIZE;
76     pParent = Parent;
77     ulPos = 0;
78     pChunkData = NULL;
79 schoenebeck 1863 CurrentChunkSize = 0;
80     NewChunkSize = 0;
81 schoenebeck 800 ulChunkDataSize = 0;
82 schoenebeck 2 ReadHeader(StartPos);
83     }
84    
85 schoenebeck 780 Chunk::Chunk(File* pFile, List* pParent, uint32_t uiChunkID, uint uiBodySize) {
86     this->pFile = pFile;
87     ulStartPos = 0; // arbitrary usually, since it will be updated when we write the chunk
88     this->pParent = pParent;
89     ulPos = 0;
90     pChunkData = NULL;
91 schoenebeck 1863 ChunkID = uiChunkID;
92 schoenebeck 800 ulChunkDataSize = 0;
93 schoenebeck 780 CurrentChunkSize = 0;
94     NewChunkSize = uiBodySize;
95     }
96    
97 schoenebeck 2 Chunk::~Chunk() {
98 persson 1869 if (pFile) pFile->UnlogResized(this);
99 schoenebeck 2 if (pChunkData) delete[] pChunkData;
100     }
101    
102     void Chunk::ReadHeader(unsigned long fPos) {
103     #if DEBUG
104     std::cout << "Chunk::Readheader(" << fPos << ") ";
105     #endif // DEBUG
106 schoenebeck 1863 ChunkID = 0;
107     NewChunkSize = CurrentChunkSize = 0;
108 schoenebeck 2 #if POSIX
109 schoenebeck 780 if (lseek(pFile->hFileRead, fPos, SEEK_SET) != -1) {
110     read(pFile->hFileRead, &ChunkID, 4);
111     read(pFile->hFileRead, &CurrentChunkSize, 4);
112 schoenebeck 1050 #elif defined(WIN32)
113     if (SetFilePointer(pFile->hFileRead, fPos, NULL/*32 bit*/, FILE_BEGIN) != INVALID_SET_FILE_POINTER) {
114     DWORD dwBytesRead;
115     ReadFile(pFile->hFileRead, &ChunkID, 4, &dwBytesRead, NULL);
116     ReadFile(pFile->hFileRead, &CurrentChunkSize, 4, &dwBytesRead, NULL);
117 schoenebeck 2 #else
118 schoenebeck 780 if (!fseek(pFile->hFileRead, fPos, SEEK_SET)) {
119     fread(&ChunkID, 4, 1, pFile->hFileRead);
120     fread(&CurrentChunkSize, 4, 1, pFile->hFileRead);
121 schoenebeck 2 #endif // POSIX
122     #if WORDS_BIGENDIAN
123     if (ChunkID == CHUNK_ID_RIFF) {
124 schoenebeck 808 pFile->bEndianNative = false;
125 schoenebeck 2 }
126     #else // little endian
127     if (ChunkID == CHUNK_ID_RIFX) {
128 schoenebeck 780 pFile->bEndianNative = false;
129 schoenebeck 2 ChunkID = CHUNK_ID_RIFF;
130     }
131     #endif // WORDS_BIGENDIAN
132 schoenebeck 780 if (!pFile->bEndianNative) {
133 schoenebeck 11 //swapBytes_32(&ChunkID);
134 schoenebeck 780 swapBytes_32(&CurrentChunkSize);
135 schoenebeck 2 }
136 schoenebeck 11 #if DEBUG
137     std::cout << "ckID=" << convertToString(ChunkID) << " ";
138 schoenebeck 1863 std::cout << "ckSize=" << CurrentChunkSize << " ";
139     std::cout << "bEndianNative=" << pFile->bEndianNative << std::endl;
140 schoenebeck 11 #endif // DEBUG
141 schoenebeck 780 NewChunkSize = CurrentChunkSize;
142 schoenebeck 2 }
143     }
144    
145 schoenebeck 780 void Chunk::WriteHeader(unsigned long fPos) {
146     uint32_t uiNewChunkID = ChunkID;
147     if (ChunkID == CHUNK_ID_RIFF) {
148     #if WORDS_BIGENDIAN
149     if (pFile->bEndianNative) uiNewChunkID = CHUNK_ID_RIFX;
150     #else // little endian
151     if (!pFile->bEndianNative) uiNewChunkID = CHUNK_ID_RIFX;
152     #endif // WORDS_BIGENDIAN
153     }
154    
155     uint32_t uiNewChunkSize = NewChunkSize;
156     if (!pFile->bEndianNative) {
157     swapBytes_32(&uiNewChunkSize);
158     }
159    
160     #if POSIX
161     if (lseek(pFile->hFileWrite, fPos, SEEK_SET) != -1) {
162     write(pFile->hFileWrite, &uiNewChunkID, 4);
163     write(pFile->hFileWrite, &uiNewChunkSize, 4);
164     }
165 schoenebeck 1050 #elif defined(WIN32)
166     if (SetFilePointer(pFile->hFileWrite, fPos, NULL/*32 bit*/, FILE_BEGIN) != INVALID_SET_FILE_POINTER) {
167     DWORD dwBytesWritten;
168     WriteFile(pFile->hFileWrite, &uiNewChunkID, 4, &dwBytesWritten, NULL);
169     WriteFile(pFile->hFileWrite, &uiNewChunkSize, 4, &dwBytesWritten, NULL);
170     }
171 schoenebeck 780 #else
172     if (!fseek(pFile->hFileWrite, fPos, SEEK_SET)) {
173     fwrite(&uiNewChunkID, 4, 1, pFile->hFileWrite);
174     fwrite(&uiNewChunkSize, 4, 1, pFile->hFileWrite);
175     }
176     #endif // POSIX
177     }
178    
179 schoenebeck 11 /**
180     * Returns the String representation of the chunk's ID (e.g. "RIFF",
181     * "LIST").
182     */
183 schoenebeck 2 String Chunk::GetChunkIDString() {
184     return convertToString(ChunkID);
185     }
186    
187 schoenebeck 11 /**
188     * Sets the position within the chunk body, thus within the data portion
189     * of the chunk (in bytes).
190     *
191 schoenebeck 780 * <b>Caution:</b> the position will be reset to zero whenever
192     * File::Save() was called.
193     *
194 schoenebeck 11 * @param Where - position offset (in bytes)
195     * @param Whence - optional: defines to what <i>\a Where</i> relates to,
196     * if omitted \a Where relates to beginning of the chunk
197     * data
198     */
199 schoenebeck 2 unsigned long Chunk::SetPos(unsigned long Where, stream_whence_t Whence) {
200     #if DEBUG
201     std::cout << "Chunk::SetPos(ulong)" << std::endl;
202     #endif // DEBUG
203     switch (Whence) {
204     case stream_curpos:
205     ulPos += Where;
206     break;
207     case stream_end:
208 schoenebeck 780 ulPos = CurrentChunkSize - 1 - Where;
209 schoenebeck 2 break;
210     case stream_backward:
211     ulPos -= Where;
212     break;
213     case stream_start: default:
214     ulPos = Where;
215     break;
216     }
217 schoenebeck 780 if (ulPos > CurrentChunkSize) ulPos = CurrentChunkSize;
218 schoenebeck 2 return ulPos;
219     }
220    
221 schoenebeck 11 /**
222     * Returns the number of bytes left to read in the chunk body.
223     * When reading data from the chunk using the Read*() Methods, the
224     * position within the chunk data (that is the chunk body) will be
225     * incremented by the number of read bytes and RemainingBytes() returns
226     * how much data is left to read from the current position to the end
227     * of the chunk data.
228     *
229     * @returns number of bytes left to read
230     */
231 schoenebeck 2 unsigned long Chunk::RemainingBytes() {
232     #if DEBUG
233 schoenebeck 780 std::cout << "Chunk::Remainingbytes()=" << CurrentChunkSize - ulPos << std::endl;
234 schoenebeck 2 #endif // DEBUG
235 schoenebeck 1863 return (CurrentChunkSize > ulPos) ? CurrentChunkSize - ulPos : 0;
236 schoenebeck 2 }
237    
238 schoenebeck 11 /**
239     * Returns the current state of the chunk object.
240     * Following values are possible:
241     * - RIFF::stream_ready :
242     * chunk data can be read (this is the usual case)
243     * - RIFF::stream_closed :
244     * the data stream was closed somehow, no more reading possible
245     * - RIFF::stream_end_reached :
246 schoenebeck 1383 * already reached the end of the chunk data, no more reading
247 schoenebeck 11 * possible without SetPos()
248     */
249 schoenebeck 2 stream_state_t Chunk::GetState() {
250     #if DEBUG
251     std::cout << "Chunk::GetState()" << std::endl;
252     #endif // DEBUG
253     #if POSIX
254 schoenebeck 1050 if (pFile->hFileRead == 0) return stream_closed;
255     #elif defined (WIN32)
256     if (pFile->hFileRead == INVALID_HANDLE_VALUE)
257     return stream_closed;
258 schoenebeck 2 #else
259 schoenebeck 780 if (pFile->hFileRead == NULL) return stream_closed;
260 schoenebeck 2 #endif // POSIX
261 schoenebeck 780 if (ulPos < CurrentChunkSize) return stream_ready;
262     else return stream_end_reached;
263 schoenebeck 2 }
264    
265     /**
266     * Reads \a WordCount number of data words with given \a WordSize and
267     * copies it into a buffer pointed by \a pData. The buffer has to be
268     * allocated and be sure to provide the correct \a WordSize, as this
269     * will be important and taken into account for eventual endian
270     * correction (swapping of bytes due to different native byte order of
271     * a system). The position within the chunk will automatically be
272     * incremented.
273     *
274     * @param pData destination buffer
275     * @param WordCount number of data words to read
276     * @param WordSize size of each data word to read
277     * @returns number of successfully read data words or 0 if end
278     * of file reached or error occured
279     */
280     unsigned long Chunk::Read(void* pData, unsigned long WordCount, unsigned long WordSize) {
281     #if DEBUG
282     std::cout << "Chunk::Read(void*,ulong,ulong)" << std::endl;
283     #endif // DEBUG
284 schoenebeck 1851 if (ulStartPos == 0) return 0; // is only 0 if this is a new chunk, so nothing to read (yet)
285 schoenebeck 780 if (ulPos >= CurrentChunkSize) return 0;
286     if (ulPos + WordCount * WordSize >= CurrentChunkSize) WordCount = (CurrentChunkSize - ulPos) / WordSize;
287 schoenebeck 2 #if POSIX
288 schoenebeck 780 if (lseek(pFile->hFileRead, ulStartPos + ulPos, SEEK_SET) < 0) return 0;
289     unsigned long readWords = read(pFile->hFileRead, pData, WordCount * WordSize);
290 schoenebeck 2 if (readWords < 1) return 0;
291     readWords /= WordSize;
292 schoenebeck 1050 #elif defined(WIN32)
293     if (SetFilePointer(pFile->hFileRead, ulStartPos + ulPos, NULL/*32 bit*/, FILE_BEGIN) == INVALID_SET_FILE_POINTER) return 0;
294     DWORD readWords;
295     ReadFile(pFile->hFileRead, pData, WordCount * WordSize, &readWords, NULL);
296     if (readWords < 1) return 0;
297     readWords /= WordSize;
298 schoenebeck 2 #else // standard C functions
299 schoenebeck 780 if (fseek(pFile->hFileRead, ulStartPos + ulPos, SEEK_SET)) return 0;
300     unsigned long readWords = fread(pData, WordSize, WordCount, pFile->hFileRead);
301 schoenebeck 2 #endif // POSIX
302 schoenebeck 780 if (!pFile->bEndianNative && WordSize != 1) {
303 schoenebeck 2 switch (WordSize) {
304     case 2:
305     for (unsigned long iWord = 0; iWord < readWords; iWord++)
306     swapBytes_16((uint16_t*) pData + iWord);
307     break;
308     case 4:
309     for (unsigned long iWord = 0; iWord < readWords; iWord++)
310     swapBytes_32((uint32_t*) pData + iWord);
311     break;
312     default:
313     for (unsigned long iWord = 0; iWord < readWords; iWord++)
314     swapBytes((uint8_t*) pData + iWord * WordSize, WordSize);
315     break;
316     }
317     }
318     SetPos(readWords * WordSize, stream_curpos);
319     return readWords;
320     }
321    
322 schoenebeck 780 /**
323     * Writes \a WordCount number of data words with given \a WordSize from
324     * the buffer pointed by \a pData. Be sure to provide the correct
325     * \a WordSize, as this will be important and taken into account for
326     * eventual endian correction (swapping of bytes due to different
327     * native byte order of a system). The position within the chunk will
328     * automatically be incremented.
329     *
330     * @param pData source buffer (containing the data)
331     * @param WordCount number of data words to write
332     * @param WordSize size of each data word to write
333     * @returns number of successfully written data words
334     * @throws RIFF::Exception if write operation would exceed current
335     * chunk size or any IO error occured
336     * @see Resize()
337     */
338     unsigned long Chunk::Write(void* pData, unsigned long WordCount, unsigned long WordSize) {
339 schoenebeck 798 if (pFile->Mode != stream_mode_read_write)
340     throw Exception("Cannot write data to chunk, file has to be opened in read+write mode first");
341     if (ulPos >= CurrentChunkSize || ulPos + WordCount * WordSize > CurrentChunkSize)
342 schoenebeck 780 throw Exception("End of chunk reached while trying to write data");
343     if (!pFile->bEndianNative && WordSize != 1) {
344     switch (WordSize) {
345     case 2:
346     for (unsigned long iWord = 0; iWord < WordCount; iWord++)
347     swapBytes_16((uint16_t*) pData + iWord);
348     break;
349     case 4:
350     for (unsigned long iWord = 0; iWord < WordCount; iWord++)
351     swapBytes_32((uint32_t*) pData + iWord);
352     break;
353     default:
354     for (unsigned long iWord = 0; iWord < WordCount; iWord++)
355     swapBytes((uint8_t*) pData + iWord * WordSize, WordSize);
356     break;
357     }
358     }
359     #if POSIX
360     if (lseek(pFile->hFileWrite, ulStartPos + ulPos, SEEK_SET) < 0) {
361     throw Exception("Could not seek to position " + ToString(ulPos) +
362     " in chunk (" + ToString(ulStartPos + ulPos) + " in file)");
363     }
364     unsigned long writtenWords = write(pFile->hFileWrite, pData, WordCount * WordSize);
365     if (writtenWords < 1) throw Exception("POSIX IO Error while trying to write chunk data");
366     writtenWords /= WordSize;
367 schoenebeck 1050 #elif defined(WIN32)
368     if (SetFilePointer(pFile->hFileWrite, ulStartPos + ulPos, NULL/*32 bit*/, FILE_BEGIN) == INVALID_SET_FILE_POINTER) {
369     throw Exception("Could not seek to position " + ToString(ulPos) +
370 schoenebeck 1459 " in chunk (" + ToString(ulStartPos + ulPos) + " in file)");
371 schoenebeck 1050 }
372     DWORD writtenWords;
373     WriteFile(pFile->hFileWrite, pData, WordCount * WordSize, &writtenWords, NULL);
374     if (writtenWords < 1) throw Exception("Windows IO Error while trying to write chunk data");
375     writtenWords /= WordSize;
376 schoenebeck 780 #else // standard C functions
377     if (fseek(pFile->hFileWrite, ulStartPos + ulPos, SEEK_SET)) {
378     throw Exception("Could not seek to position " + ToString(ulPos) +
379     " in chunk (" + ToString(ulStartPos + ulPos) + " in file)");
380     }
381     unsigned long writtenWords = fwrite(pData, WordSize, WordCount, pFile->hFileWrite);
382     #endif // POSIX
383     SetPos(writtenWords * WordSize, stream_curpos);
384     return writtenWords;
385     }
386    
387 schoenebeck 2 /** Just an internal wrapper for the main <i>Read()</i> method with additional Exception throwing on errors. */
388     unsigned long Chunk::ReadSceptical(void* pData, unsigned long WordCount, unsigned long WordSize) {
389     unsigned long readWords = Read(pData, WordCount, WordSize);
390     if (readWords != WordCount) throw RIFF::Exception("End of chunk data reached.");
391     return readWords;
392     }
393    
394     /**
395     * Reads \a WordCount number of 8 Bit signed integer words and copies it
396     * into the buffer pointed by \a pData. The buffer has to be allocated.
397     * The position within the chunk will automatically be incremented.
398     *
399     * @param pData destination buffer
400     * @param WordCount number of 8 Bit signed integers to read
401     * @returns number of read integers
402     * @throws RIFF::Exception if an error occured or less than
403     * \a WordCount integers could be read!
404     */
405     unsigned long Chunk::ReadInt8(int8_t* pData, unsigned long WordCount) {
406     #if DEBUG
407     std::cout << "Chunk::ReadInt8(int8_t*,ulong)" << std::endl;
408     #endif // DEBUG
409     return ReadSceptical(pData, WordCount, 1);
410     }
411    
412     /**
413 schoenebeck 780 * Writes \a WordCount number of 8 Bit signed integer words from the
414     * buffer pointed by \a pData to the chunk's body, directly to the
415     * actual "physical" file. The position within the chunk will
416     * automatically be incremented. Note: you cannot write beyond the
417     * boundaries of the chunk, to append data to the chunk call Resize()
418     * before.
419     *
420     * @param pData source buffer (containing the data)
421     * @param WordCount number of 8 Bit signed integers to write
422     * @returns number of written integers
423     * @throws RIFF::Exception if an IO error occured
424     * @see Resize()
425     */
426     unsigned long Chunk::WriteInt8(int8_t* pData, unsigned long WordCount) {
427     return Write(pData, WordCount, 1);
428     }
429    
430     /**
431 schoenebeck 2 * Reads \a WordCount number of 8 Bit unsigned integer words and copies
432     * it into the buffer pointed by \a pData. The buffer has to be
433     * allocated. The position within the chunk will automatically be
434     * incremented.
435     *
436     * @param pData destination buffer
437     * @param WordCount number of 8 Bit unsigned integers to read
438     * @returns number of read integers
439     * @throws RIFF::Exception if an error occured or less than
440     * \a WordCount integers could be read!
441     */
442     unsigned long Chunk::ReadUint8(uint8_t* pData, unsigned long WordCount) {
443     #if DEBUG
444     std::cout << "Chunk::ReadUint8(uint8_t*,ulong)" << std::endl;
445     #endif // DEBUG
446     return ReadSceptical(pData, WordCount, 1);
447     }
448    
449     /**
450 schoenebeck 780 * Writes \a WordCount number of 8 Bit unsigned integer words from the
451     * buffer pointed by \a pData to the chunk's body, directly to the
452     * actual "physical" file. The position within the chunk will
453     * automatically be incremented. Note: you cannot write beyond the
454     * boundaries of the chunk, to append data to the chunk call Resize()
455     * before.
456     *
457     * @param pData source buffer (containing the data)
458     * @param WordCount number of 8 Bit unsigned integers to write
459     * @returns number of written integers
460     * @throws RIFF::Exception if an IO error occured
461     * @see Resize()
462     */
463     unsigned long Chunk::WriteUint8(uint8_t* pData, unsigned long WordCount) {
464     return Write(pData, WordCount, 1);
465     }
466    
467     /**
468 schoenebeck 2 * Reads \a WordCount number of 16 Bit signed integer words and copies
469     * it into the buffer pointed by \a pData. The buffer has to be
470     * allocated. Endian correction will automatically be done if needed.
471     * The position within the chunk will automatically be incremented.
472     *
473     * @param pData destination buffer
474     * @param WordCount number of 16 Bit signed integers to read
475     * @returns number of read integers
476     * @throws RIFF::Exception if an error occured or less than
477     * \a WordCount integers could be read!
478     */
479     unsigned long Chunk::ReadInt16(int16_t* pData, unsigned long WordCount) {
480     #if DEBUG
481     std::cout << "Chunk::ReadInt16(int16_t*,ulong)" << std::endl;
482     #endif // DEBUG
483     return ReadSceptical(pData, WordCount, 2);
484     }
485    
486     /**
487 schoenebeck 780 * Writes \a WordCount number of 16 Bit signed integer words from the
488     * buffer pointed by \a pData to the chunk's body, directly to the
489     * actual "physical" file. The position within the chunk will
490     * automatically be incremented. Note: you cannot write beyond the
491     * boundaries of the chunk, to append data to the chunk call Resize()
492     * before.
493     *
494     * @param pData source buffer (containing the data)
495     * @param WordCount number of 16 Bit signed integers to write
496     * @returns number of written integers
497     * @throws RIFF::Exception if an IO error occured
498     * @see Resize()
499     */
500     unsigned long Chunk::WriteInt16(int16_t* pData, unsigned long WordCount) {
501     return Write(pData, WordCount, 2);
502     }
503    
504     /**
505 schoenebeck 2 * Reads \a WordCount number of 16 Bit unsigned integer words and copies
506     * it into the buffer pointed by \a pData. The buffer has to be
507     * allocated. Endian correction will automatically be done if needed.
508     * The position within the chunk will automatically be incremented.
509     *
510     * @param pData destination buffer
511     * @param WordCount number of 8 Bit unsigned integers to read
512     * @returns number of read integers
513     * @throws RIFF::Exception if an error occured or less than
514     * \a WordCount integers could be read!
515     */
516     unsigned long Chunk::ReadUint16(uint16_t* pData, unsigned long WordCount) {
517     #if DEBUG
518     std::cout << "Chunk::ReadUint16(uint16_t*,ulong)" << std::endl;
519     #endif // DEBUG
520     return ReadSceptical(pData, WordCount, 2);
521     }
522    
523     /**
524 schoenebeck 780 * Writes \a WordCount number of 16 Bit unsigned integer words from the
525     * buffer pointed by \a pData to the chunk's body, directly to the
526     * actual "physical" file. The position within the chunk will
527     * automatically be incremented. Note: you cannot write beyond the
528     * boundaries of the chunk, to append data to the chunk call Resize()
529     * before.
530     *
531     * @param pData source buffer (containing the data)
532     * @param WordCount number of 16 Bit unsigned integers to write
533     * @returns number of written integers
534     * @throws RIFF::Exception if an IO error occured
535     * @see Resize()
536     */
537     unsigned long Chunk::WriteUint16(uint16_t* pData, unsigned long WordCount) {
538     return Write(pData, WordCount, 2);
539     }
540    
541     /**
542 schoenebeck 2 * Reads \a WordCount number of 32 Bit signed integer words and copies
543     * it into the buffer pointed by \a pData. The buffer has to be
544     * allocated. Endian correction will automatically be done if needed.
545     * The position within the chunk will automatically be incremented.
546     *
547     * @param pData destination buffer
548     * @param WordCount number of 32 Bit signed integers to read
549     * @returns number of read integers
550     * @throws RIFF::Exception if an error occured or less than
551     * \a WordCount integers could be read!
552     */
553     unsigned long Chunk::ReadInt32(int32_t* pData, unsigned long WordCount) {
554     #if DEBUG
555     std::cout << "Chunk::ReadInt32(int32_t*,ulong)" << std::endl;
556     #endif // DEBUG
557     return ReadSceptical(pData, WordCount, 4);
558     }
559    
560     /**
561 schoenebeck 780 * Writes \a WordCount number of 32 Bit signed integer words from the
562     * buffer pointed by \a pData to the chunk's body, directly to the
563     * actual "physical" file. The position within the chunk will
564     * automatically be incremented. Note: you cannot write beyond the
565     * boundaries of the chunk, to append data to the chunk call Resize()
566     * before.
567     *
568     * @param pData source buffer (containing the data)
569     * @param WordCount number of 32 Bit signed integers to write
570     * @returns number of written integers
571     * @throws RIFF::Exception if an IO error occured
572     * @see Resize()
573     */
574     unsigned long Chunk::WriteInt32(int32_t* pData, unsigned long WordCount) {
575     return Write(pData, WordCount, 4);
576     }
577    
578     /**
579 schoenebeck 2 * Reads \a WordCount number of 32 Bit unsigned integer words and copies
580     * it into the buffer pointed by \a pData. The buffer has to be
581     * allocated. Endian correction will automatically be done if needed.
582     * The position within the chunk will automatically be incremented.
583     *
584     * @param pData destination buffer
585     * @param WordCount number of 32 Bit unsigned integers to read
586     * @returns number of read integers
587     * @throws RIFF::Exception if an error occured or less than
588     * \a WordCount integers could be read!
589     */
590     unsigned long Chunk::ReadUint32(uint32_t* pData, unsigned long WordCount) {
591     #if DEBUG
592     std::cout << "Chunk::ReadUint32(uint32_t*,ulong)" << std::endl;
593     #endif // DEBUG
594     return ReadSceptical(pData, WordCount, 4);
595     }
596    
597     /**
598 persson 2450 * Reads a null-padded string of size characters and copies it
599     * into the string \a s. The position within the chunk will
600     * automatically be incremented.
601     *
602     * @param s destination string
603     * @param size number of characters to read
604     * @throws RIFF::Exception if an error occured or less than
605     * \a size characters could be read!
606     */
607     void Chunk::ReadString(String& s, int size) {
608     char* buf = new char[size];
609     ReadSceptical(buf, 1, size);
610     s.assign(buf, std::find(buf, buf + size, '\0'));
611     delete[] buf;
612     }
613    
614     /**
615 schoenebeck 780 * Writes \a WordCount number of 32 Bit unsigned integer words from the
616     * buffer pointed by \a pData to the chunk's body, directly to the
617     * actual "physical" file. The position within the chunk will
618     * automatically be incremented. Note: you cannot write beyond the
619     * boundaries of the chunk, to append data to the chunk call Resize()
620     * before.
621     *
622     * @param pData source buffer (containing the data)
623     * @param WordCount number of 32 Bit unsigned integers to write
624     * @returns number of written integers
625     * @throws RIFF::Exception if an IO error occured
626     * @see Resize()
627     */
628     unsigned long Chunk::WriteUint32(uint32_t* pData, unsigned long WordCount) {
629     return Write(pData, WordCount, 4);
630     }
631    
632     /**
633 schoenebeck 2 * Reads one 8 Bit signed integer word and increments the position within
634     * the chunk.
635     *
636     * @returns read integer word
637     * @throws RIFF::Exception if an error occured
638     */
639     int8_t Chunk::ReadInt8() {
640     #if DEBUG
641     std::cout << "Chunk::ReadInt8()" << std::endl;
642     #endif // DEBUG
643     int8_t word;
644     ReadSceptical(&word,1,1);
645     return word;
646     }
647    
648     /**
649     * Reads one 8 Bit unsigned integer word and increments the position
650     * within the chunk.
651     *
652     * @returns read integer word
653     * @throws RIFF::Exception if an error occured
654     */
655     uint8_t Chunk::ReadUint8() {
656     #if DEBUG
657     std::cout << "Chunk::ReadUint8()" << std::endl;
658     #endif // DEBUG
659     uint8_t word;
660     ReadSceptical(&word,1,1);
661     return word;
662     }
663    
664     /**
665     * Reads one 16 Bit signed integer word and increments the position
666     * within the chunk. Endian correction will automatically be done if
667     * needed.
668     *
669     * @returns read integer word
670     * @throws RIFF::Exception if an error occured
671     */
672     int16_t Chunk::ReadInt16() {
673     #if DEBUG
674     std::cout << "Chunk::ReadInt16()" << std::endl;
675     #endif // DEBUG
676     int16_t word;
677     ReadSceptical(&word,1,2);
678     return word;
679     }
680    
681     /**
682     * Reads one 16 Bit unsigned integer word and increments the position
683     * within the chunk. Endian correction will automatically be done if
684     * needed.
685     *
686     * @returns read integer word
687     * @throws RIFF::Exception if an error occured
688     */
689     uint16_t Chunk::ReadUint16() {
690     #if DEBUG
691     std::cout << "Chunk::ReadUint16()" << std::endl;
692     #endif // DEBUG
693     uint16_t word;
694     ReadSceptical(&word,1,2);
695     return word;
696     }
697    
698     /**
699     * Reads one 32 Bit signed integer word and increments the position
700     * within the chunk. Endian correction will automatically be done if
701     * needed.
702     *
703     * @returns read integer word
704     * @throws RIFF::Exception if an error occured
705     */
706     int32_t Chunk::ReadInt32() {
707     #if DEBUG
708     std::cout << "Chunk::ReadInt32()" << std::endl;
709     #endif // DEBUG
710     int32_t word;
711     ReadSceptical(&word,1,4);
712     return word;
713     }
714    
715     /**
716     * Reads one 32 Bit unsigned integer word and increments the position
717     * within the chunk. Endian correction will automatically be done if
718     * needed.
719     *
720     * @returns read integer word
721     * @throws RIFF::Exception if an error occured
722     */
723     uint32_t Chunk::ReadUint32() {
724     #if DEBUG
725     std::cout << "Chunk::ReadUint32()" << std::endl;
726     #endif // DEBUG
727     uint32_t word;
728     ReadSceptical(&word,1,4);
729     return word;
730     }
731    
732 schoenebeck 780 /** @brief Load chunk body into RAM.
733     *
734     * Loads the whole chunk body into memory. You can modify the data in
735     * RAM and save the data by calling File::Save() afterwards.
736     *
737     * <b>Caution:</b> the buffer pointer will be invalidated once
738     * File::Save() was called. You have to call LoadChunkData() again to
739 schoenebeck 800 * get a new, valid pointer whenever File::Save() was called.
740 schoenebeck 780 *
741 schoenebeck 800 * You can call LoadChunkData() again if you previously scheduled to
742     * enlarge this chunk with a Resize() call. In that case the buffer will
743     * be enlarged to the new, scheduled chunk size and you can already
744     * place the new chunk data to the buffer and finally call File::Save()
745     * to enlarge the chunk physically and write the new data in one rush.
746     * This approach is definitely recommended if you have to enlarge and
747     * write new data to a lot of chunks.
748     *
749 schoenebeck 780 * @returns a pointer to the data in RAM on success, NULL otherwise
750 schoenebeck 800 * @throws Exception if data buffer could not be enlarged
751 schoenebeck 780 * @see ReleaseChunkData()
752     */
753 schoenebeck 2 void* Chunk::LoadChunkData() {
754 schoenebeck 1851 if (!pChunkData && pFile->Filename != "" && ulStartPos != 0) {
755 schoenebeck 2 #if POSIX
756 schoenebeck 780 if (lseek(pFile->hFileRead, ulStartPos, SEEK_SET) == -1) return NULL;
757 schoenebeck 1050 #elif defined(WIN32)
758     if (SetFilePointer(pFile->hFileRead, ulStartPos, NULL/*32 bit*/, FILE_BEGIN) == INVALID_SET_FILE_POINTER) return NULL;
759 schoenebeck 800 #else
760     if (fseek(pFile->hFileRead, ulStartPos, SEEK_SET)) return NULL;
761     #endif // POSIX
762     unsigned long ulBufferSize = (CurrentChunkSize > NewChunkSize) ? CurrentChunkSize : NewChunkSize;
763     pChunkData = new uint8_t[ulBufferSize];
764 schoenebeck 2 if (!pChunkData) return NULL;
765 schoenebeck 800 memset(pChunkData, 0, ulBufferSize);
766     #if POSIX
767 schoenebeck 780 unsigned long readWords = read(pFile->hFileRead, pChunkData, GetSize());
768 schoenebeck 1050 #elif defined(WIN32)
769     DWORD readWords;
770     ReadFile(pFile->hFileRead, pChunkData, GetSize(), &readWords, NULL);
771 schoenebeck 2 #else
772 schoenebeck 780 unsigned long readWords = fread(pChunkData, 1, GetSize(), pFile->hFileRead);
773 schoenebeck 2 #endif // POSIX
774     if (readWords != GetSize()) {
775     delete[] pChunkData;
776     return (pChunkData = NULL);
777     }
778 schoenebeck 800 ulChunkDataSize = ulBufferSize;
779     } else if (NewChunkSize > ulChunkDataSize) {
780     uint8_t* pNewBuffer = new uint8_t[NewChunkSize];
781     if (!pNewBuffer) throw Exception("Could not enlarge chunk data buffer to " + ToString(NewChunkSize) + " bytes");
782     memset(pNewBuffer, 0 , NewChunkSize);
783     memcpy(pNewBuffer, pChunkData, ulChunkDataSize);
784     delete[] pChunkData;
785     pChunkData = pNewBuffer;
786     ulChunkDataSize = NewChunkSize;
787 schoenebeck 2 }
788     return pChunkData;
789     }
790    
791 schoenebeck 780 /** @brief Free loaded chunk body from RAM.
792     *
793     * Frees loaded chunk body data from memory (RAM). You should call
794     * File::Save() before calling this method if you modified the data to
795     * make the changes persistent.
796     */
797 schoenebeck 2 void Chunk::ReleaseChunkData() {
798     if (pChunkData) {
799     delete[] pChunkData;
800     pChunkData = NULL;
801     }
802     }
803    
804 schoenebeck 780 /** @brief Resize chunk.
805     *
806     * Resizes this chunk's body, that is the actual size of data possible
807     * to be written to this chunk. This call will return immediately and
808     * just schedule the resize operation. You should call File::Save() to
809     * actually perform the resize operation(s) "physically" to the file.
810     * As this can take a while on large files, it is recommended to call
811     * Resize() first on all chunks which have to be resized and finally to
812     * call File::Save() to perform all those resize operations in one rush.
813     *
814     * <b>Caution:</b> You cannot directly write to enlarged chunks before
815     * calling File::Save() as this might exceed the current chunk's body
816 schoenebeck 800 * boundary!
817 schoenebeck 780 *
818     * @param iNewSize - new chunk body size in bytes (must be greater than zero)
819     * @throws RIFF::Exception if \a iNewSize is less than 1
820     * @see File::Save()
821     */
822     void Chunk::Resize(int iNewSize) {
823 schoenebeck 1105 if (iNewSize <= 0)
824     throw Exception("There is at least one empty chunk (zero size): " + __resolveChunkPath(this));
825 schoenebeck 800 if (NewChunkSize == iNewSize) return;
826 schoenebeck 780 NewChunkSize = iNewSize;
827     pFile->LogAsResized(this);
828     }
829 schoenebeck 2
830 schoenebeck 780 /** @brief Write chunk persistently e.g. to disk.
831     *
832     * Stores the chunk persistently to its actual "physical" file.
833     *
834     * @param ulWritePos - position within the "physical" file where this
835     * chunk should be written to
836     * @param ulCurrentDataOffset - offset of current (old) data within
837     * the file
838     * @returns new write position in the "physical" file, that is
839     * \a ulWritePos incremented by this chunk's new size
840     * (including its header size of course)
841     */
842     unsigned long Chunk::WriteChunk(unsigned long ulWritePos, unsigned long ulCurrentDataOffset) {
843 schoenebeck 798 const unsigned long ulOriginalPos = ulWritePos;
844 schoenebeck 780 ulWritePos += CHUNK_HEADER_SIZE;
845 schoenebeck 2
846 schoenebeck 798 if (pFile->Mode != stream_mode_read_write)
847     throw Exception("Cannot write list chunk, file has to be opened in read+write mode");
848    
849 schoenebeck 780 // if the whole chunk body was loaded into RAM
850     if (pChunkData) {
851 schoenebeck 800 // make sure chunk data buffer in RAM is at least as large as the new chunk size
852     LoadChunkData();
853 schoenebeck 780 // write chunk data from RAM persistently to the file
854     #if POSIX
855     lseek(pFile->hFileWrite, ulWritePos, SEEK_SET);
856     if (write(pFile->hFileWrite, pChunkData, NewChunkSize) != NewChunkSize) {
857     throw Exception("Writing Chunk data (from RAM) failed");
858     }
859 schoenebeck 1050 #elif defined(WIN32)
860     SetFilePointer(pFile->hFileWrite, ulWritePos, NULL/*32 bit*/, FILE_BEGIN);
861     DWORD dwBytesWritten;
862     WriteFile(pFile->hFileWrite, pChunkData, NewChunkSize, &dwBytesWritten, NULL);
863     if (dwBytesWritten != NewChunkSize) {
864     throw Exception("Writing Chunk data (from RAM) failed");
865     }
866 schoenebeck 780 #else
867     fseek(pFile->hFileWrite, ulWritePos, SEEK_SET);
868     if (fwrite(pChunkData, 1, NewChunkSize, pFile->hFileWrite) != NewChunkSize) {
869     throw Exception("Writing Chunk data (from RAM) failed");
870     }
871     #endif // POSIX
872     } else {
873     // move chunk data from the end of the file to the appropriate position
874     int8_t* pCopyBuffer = new int8_t[4096];
875     unsigned long ulToMove = (NewChunkSize < CurrentChunkSize) ? NewChunkSize : CurrentChunkSize;
876 schoenebeck 1050 #if defined(WIN32)
877     DWORD iBytesMoved = 1; // we have to pass it via pointer to the Windows API, thus the correct size must be ensured
878     #else
879 schoenebeck 780 int iBytesMoved = 1;
880 schoenebeck 1050 #endif
881 persson 2120 for (unsigned long ulOffset = 0; ulToMove > 0 && iBytesMoved > 0; ulOffset += iBytesMoved, ulToMove -= iBytesMoved) {
882 schoenebeck 780 iBytesMoved = (ulToMove < 4096) ? ulToMove : 4096;
883     #if POSIX
884     lseek(pFile->hFileRead, ulStartPos + ulCurrentDataOffset + ulOffset, SEEK_SET);
885     iBytesMoved = read(pFile->hFileRead, pCopyBuffer, iBytesMoved);
886     lseek(pFile->hFileWrite, ulWritePos + ulOffset, SEEK_SET);
887     iBytesMoved = write(pFile->hFileWrite, pCopyBuffer, iBytesMoved);
888 schoenebeck 1050 #elif defined(WIN32)
889     SetFilePointer(pFile->hFileRead, ulStartPos + ulCurrentDataOffset + ulOffset, NULL/*32 bit*/, FILE_BEGIN);
890     ReadFile(pFile->hFileRead, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
891     SetFilePointer(pFile->hFileWrite, ulWritePos + ulOffset, NULL/*32 bit*/, FILE_BEGIN);
892     WriteFile(pFile->hFileWrite, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
893 schoenebeck 780 #else
894     fseek(pFile->hFileRead, ulStartPos + ulCurrentDataOffset + ulOffset, SEEK_SET);
895     iBytesMoved = fread(pCopyBuffer, 1, iBytesMoved, pFile->hFileRead);
896     fseek(pFile->hFileWrite, ulWritePos + ulOffset, SEEK_SET);
897     iBytesMoved = fwrite(pCopyBuffer, 1, iBytesMoved, pFile->hFileWrite);
898     #endif
899     }
900     delete[] pCopyBuffer;
901     if (iBytesMoved < 0) throw Exception("Writing Chunk data (from file) failed");
902     }
903    
904     // update this chunk's header
905     CurrentChunkSize = NewChunkSize;
906     WriteHeader(ulOriginalPos);
907    
908     // update chunk's position pointers
909     ulStartPos = ulOriginalPos + CHUNK_HEADER_SIZE;
910     ulPos = 0;
911    
912     // add pad byte if needed
913     if ((ulStartPos + NewChunkSize) % 2 != 0) {
914 schoenebeck 798 const char cPadByte = 0;
915 schoenebeck 780 #if POSIX
916 schoenebeck 798 lseek(pFile->hFileWrite, ulStartPos + NewChunkSize, SEEK_SET);
917 schoenebeck 780 write(pFile->hFileWrite, &cPadByte, 1);
918 schoenebeck 1050 #elif defined(WIN32)
919     SetFilePointer(pFile->hFileWrite, ulStartPos + NewChunkSize, NULL/*32 bit*/, FILE_BEGIN);
920     DWORD dwBytesWritten;
921     WriteFile(pFile->hFileWrite, &cPadByte, 1, &dwBytesWritten, NULL);
922 schoenebeck 780 #else
923 schoenebeck 798 fseek(pFile->hFileWrite, ulStartPos + NewChunkSize, SEEK_SET);
924 schoenebeck 780 fwrite(&cPadByte, 1, 1, pFile->hFileWrite);
925     #endif
926     return ulStartPos + NewChunkSize + 1;
927     }
928    
929     return ulStartPos + NewChunkSize;
930     }
931    
932     void Chunk::__resetPos() {
933     ulPos = 0;
934     }
935    
936    
937    
938 schoenebeck 2 // *************** List ***************
939     // *
940    
941 schoenebeck 780 List::List(File* pFile) : Chunk(pFile) {
942 schoenebeck 2 #if DEBUG
943 schoenebeck 780 std::cout << "List::List(File* pFile)" << std::endl;
944 schoenebeck 2 #endif // DEBUG
945     pSubChunks = NULL;
946     pSubChunksMap = NULL;
947     }
948    
949 schoenebeck 780 List::List(File* pFile, unsigned long StartPos, List* Parent)
950     : Chunk(pFile, StartPos, Parent) {
951 schoenebeck 2 #if DEBUG
952 schoenebeck 780 std::cout << "List::List(File*,ulong,bool,List*)" << std::endl;
953 schoenebeck 2 #endif // DEBUG
954     pSubChunks = NULL;
955     pSubChunksMap = NULL;
956     ReadHeader(StartPos);
957     ulStartPos = StartPos + LIST_HEADER_SIZE;
958     }
959    
960 schoenebeck 780 List::List(File* pFile, List* pParent, uint32_t uiListID)
961     : Chunk(pFile, pParent, CHUNK_ID_LIST, 0) {
962     pSubChunks = NULL;
963     pSubChunksMap = NULL;
964     ListType = uiListID;
965     }
966    
967 schoenebeck 2 List::~List() {
968     #if DEBUG
969     std::cout << "List::~List()" << std::endl;
970     #endif // DEBUG
971 persson 1869 DeleteChunkList();
972     }
973    
974     void List::DeleteChunkList() {
975 schoenebeck 2 if (pSubChunks) {
976     ChunkList::iterator iter = pSubChunks->begin();
977     ChunkList::iterator end = pSubChunks->end();
978     while (iter != end) {
979     delete *iter;
980     iter++;
981     }
982     delete pSubChunks;
983 persson 1869 pSubChunks = NULL;
984 schoenebeck 2 }
985 persson 1869 if (pSubChunksMap) {
986     delete pSubChunksMap;
987     pSubChunksMap = NULL;
988     }
989 schoenebeck 2 }
990    
991 schoenebeck 11 /**
992     * Returns subchunk with chunk ID <i>\a ChunkID</i> within this chunk
993     * list. Use this method if you expect only one subchunk of that type in
994     * the list. It there are more than one, it's undetermined which one of
995     * them will be returned! If there are no subchunks with that desired
996     * chunk ID, NULL will be returned.
997     *
998     * @param ChunkID - chunk ID of the sought subchunk
999     * @returns pointer to the subchunk or NULL if there is none of
1000     * that ID
1001     */
1002 schoenebeck 2 Chunk* List::GetSubChunk(uint32_t ChunkID) {
1003     #if DEBUG
1004     std::cout << "List::GetSubChunk(uint32_t)" << std::endl;
1005     #endif // DEBUG
1006     if (!pSubChunksMap) LoadSubChunks();
1007     return (*pSubChunksMap)[ChunkID];
1008     }
1009    
1010 schoenebeck 11 /**
1011     * Returns sublist chunk with list type <i>\a ListType</i> within this
1012     * chunk list. Use this method if you expect only one sublist chunk of
1013     * that type in the list. It there are more than one, it's undetermined
1014     * which one of them will be returned! If there are no sublists with
1015     * that desired list type, NULL will be returned.
1016     *
1017     * @param ListType - list type of the sought sublist
1018     * @returns pointer to the sublist or NULL if there is none of
1019     * that type
1020     */
1021 schoenebeck 2 List* List::GetSubList(uint32_t ListType) {
1022     #if DEBUG
1023     std::cout << "List::GetSubList(uint32_t)" << std::endl;
1024     #endif // DEBUG
1025     if (!pSubChunks) LoadSubChunks();
1026     ChunkList::iterator iter = pSubChunks->begin();
1027     ChunkList::iterator end = pSubChunks->end();
1028     while (iter != end) {
1029     if ((*iter)->GetChunkID() == CHUNK_ID_LIST) {
1030     List* l = (List*) *iter;
1031     if (l->GetListType() == ListType) return l;
1032     }
1033     iter++;
1034     }
1035     return NULL;
1036     }
1037    
1038 schoenebeck 11 /**
1039     * Returns the first subchunk within the list. You have to call this
1040     * method before you can call GetNextSubChunk(). Recall it when you want
1041     * to start from the beginning of the list again.
1042     *
1043     * @returns pointer to the first subchunk within the list, NULL
1044     * otherwise
1045     */
1046 schoenebeck 2 Chunk* List::GetFirstSubChunk() {
1047     #if DEBUG
1048     std::cout << "List::GetFirstSubChunk()" << std::endl;
1049     #endif // DEBUG
1050     if (!pSubChunks) LoadSubChunks();
1051     ChunksIterator = pSubChunks->begin();
1052     return (ChunksIterator != pSubChunks->end()) ? *ChunksIterator : NULL;
1053     }
1054    
1055 schoenebeck 11 /**
1056     * Returns the next subchunk within the list. You have to call
1057     * GetFirstSubChunk() before you can use this method!
1058     *
1059     * @returns pointer to the next subchunk within the list or NULL if
1060     * end of list is reached
1061     */
1062 schoenebeck 2 Chunk* List::GetNextSubChunk() {
1063     #if DEBUG
1064     std::cout << "List::GetNextSubChunk()" << std::endl;
1065     #endif // DEBUG
1066     if (!pSubChunks) return NULL;
1067     ChunksIterator++;
1068     return (ChunksIterator != pSubChunks->end()) ? *ChunksIterator : NULL;
1069     }
1070    
1071 schoenebeck 11 /**
1072     * Returns the first sublist within the list (that is a subchunk with
1073     * chunk ID "LIST"). You have to call this method before you can call
1074     * GetNextSubList(). Recall it when you want to start from the beginning
1075     * of the list again.
1076     *
1077     * @returns pointer to the first sublist within the list, NULL
1078     * otherwise
1079     */
1080 schoenebeck 2 List* List::GetFirstSubList() {
1081     #if DEBUG
1082     std::cout << "List::GetFirstSubList()" << std::endl;
1083     #endif // DEBUG
1084     if (!pSubChunks) LoadSubChunks();
1085     ListIterator = pSubChunks->begin();
1086     ChunkList::iterator end = pSubChunks->end();
1087     while (ListIterator != end) {
1088     if ((*ListIterator)->GetChunkID() == CHUNK_ID_LIST) return (List*) *ListIterator;
1089     ListIterator++;
1090     }
1091     return NULL;
1092     }
1093    
1094 schoenebeck 11 /**
1095     * Returns the next sublist (that is a subchunk with chunk ID "LIST")
1096     * within the list. You have to call GetFirstSubList() before you can
1097     * use this method!
1098     *
1099     * @returns pointer to the next sublist within the list, NULL if
1100     * end of list is reached
1101     */
1102 schoenebeck 2 List* List::GetNextSubList() {
1103     #if DEBUG
1104     std::cout << "List::GetNextSubList()" << std::endl;
1105     #endif // DEBUG
1106     if (!pSubChunks) return NULL;
1107     if (ListIterator == pSubChunks->end()) return NULL;
1108     ListIterator++;
1109     ChunkList::iterator end = pSubChunks->end();
1110     while (ListIterator != end) {
1111     if ((*ListIterator)->GetChunkID() == CHUNK_ID_LIST) return (List*) *ListIterator;
1112     ListIterator++;
1113     }
1114     return NULL;
1115     }
1116    
1117 schoenebeck 11 /**
1118 schoenebeck 515 * Returns number of subchunks within the list.
1119 schoenebeck 11 */
1120 schoenebeck 2 unsigned int List::CountSubChunks() {
1121     if (!pSubChunks) LoadSubChunks();
1122     return pSubChunks->size();
1123     }
1124    
1125 schoenebeck 11 /**
1126     * Returns number of subchunks within the list with chunk ID
1127     * <i>\a ChunkId</i>.
1128     */
1129 schoenebeck 2 unsigned int List::CountSubChunks(uint32_t ChunkID) {
1130     unsigned int result = 0;
1131     if (!pSubChunks) LoadSubChunks();
1132     ChunkList::iterator iter = pSubChunks->begin();
1133     ChunkList::iterator end = pSubChunks->end();
1134     while (iter != end) {
1135     if ((*iter)->GetChunkID() == ChunkID) {
1136     result++;
1137     }
1138     iter++;
1139     }
1140     return result;
1141     }
1142    
1143 schoenebeck 11 /**
1144     * Returns number of sublists within the list.
1145     */
1146 schoenebeck 2 unsigned int List::CountSubLists() {
1147     return CountSubChunks(CHUNK_ID_LIST);
1148     }
1149    
1150 schoenebeck 11 /**
1151     * Returns number of sublists within the list with list type
1152     * <i>\a ListType</i>
1153     */
1154 schoenebeck 2 unsigned int List::CountSubLists(uint32_t ListType) {
1155     unsigned int result = 0;
1156     if (!pSubChunks) LoadSubChunks();
1157     ChunkList::iterator iter = pSubChunks->begin();
1158     ChunkList::iterator end = pSubChunks->end();
1159     while (iter != end) {
1160     if ((*iter)->GetChunkID() == CHUNK_ID_LIST) {
1161     List* l = (List*) *iter;
1162     if (l->GetListType() == ListType) result++;
1163     }
1164     iter++;
1165     }
1166     return result;
1167     }
1168    
1169 schoenebeck 780 /** @brief Creates a new sub chunk.
1170     *
1171     * Creates and adds a new sub chunk to this list chunk. Note that the
1172     * chunk's body size given by \a uiBodySize must be greater than zero.
1173     * You have to call File::Save() to make this change persistent to the
1174     * actual file and <b>before</b> performing any data write operations
1175     * on the new chunk!
1176     *
1177     * @param uiChunkID - chunk ID of the new chunk
1178     * @param uiBodySize - size of the new chunk's body, that is its actual
1179     * data size (without header)
1180     * @throws RIFF::Exception if \a uiBodySize equals zero
1181     */
1182     Chunk* List::AddSubChunk(uint32_t uiChunkID, uint uiBodySize) {
1183     if (uiBodySize == 0) throw Exception("Chunk body size must be at least 1 byte");
1184     if (!pSubChunks) LoadSubChunks();
1185 schoenebeck 798 Chunk* pNewChunk = new Chunk(pFile, this, uiChunkID, 0);
1186 schoenebeck 780 pSubChunks->push_back(pNewChunk);
1187     (*pSubChunksMap)[uiChunkID] = pNewChunk;
1188 schoenebeck 798 pNewChunk->Resize(uiBodySize);
1189 persson 1678 NewChunkSize += CHUNK_HEADER_SIZE;
1190     pFile->LogAsResized(this);
1191 schoenebeck 780 return pNewChunk;
1192     }
1193    
1194 persson 1102 /** @brief Moves a sub chunk.
1195     *
1196     * Moves a sub chunk from one position in a list to another
1197     * position in the same list. The pSrc chunk is placed before the
1198     * pDst chunk.
1199     *
1200     * @param pSrc - sub chunk to be moved
1201     * @param pDst - the position to move to. pSrc will be placed
1202     * before pDst. If pDst is 0, pSrc will be placed
1203     * last in list.
1204     */
1205     void List::MoveSubChunk(Chunk* pSrc, Chunk* pDst) {
1206     if (!pSubChunks) LoadSubChunks();
1207     pSubChunks->remove(pSrc);
1208     ChunkList::iterator iter = find(pSubChunks->begin(), pSubChunks->end(), pDst);
1209     pSubChunks->insert(iter, pSrc);
1210     }
1211    
1212 schoenebeck 780 /** @brief Creates a new list sub chunk.
1213     *
1214     * Creates and adds a new list sub chunk to this list chunk. Note that
1215     * you have to add sub chunks / sub list chunks to the new created chunk
1216     * <b>before</b> trying to make this change persisten to the actual
1217     * file with File::Save()!
1218     *
1219     * @param uiListType - list ID of the new list chunk
1220     */
1221     List* List::AddSubList(uint32_t uiListType) {
1222     if (!pSubChunks) LoadSubChunks();
1223     List* pNewListChunk = new List(pFile, this, uiListType);
1224     pSubChunks->push_back(pNewListChunk);
1225     (*pSubChunksMap)[CHUNK_ID_LIST] = pNewListChunk;
1226 persson 1678 NewChunkSize += LIST_HEADER_SIZE;
1227     pFile->LogAsResized(this);
1228 schoenebeck 780 return pNewListChunk;
1229     }
1230    
1231     /** @brief Removes a sub chunk.
1232     *
1233     * Removes the sub chunk given by \a pSubChunk from this list and frees
1234     * it completely from RAM. The given chunk can either be a normal sub
1235 schoenebeck 1093 * chunk or a list sub chunk. In case the given chunk is a list chunk,
1236     * all its subchunks (if any) will be removed recursively as well. You
1237     * should call File::Save() to make this change persistent at any time.
1238 schoenebeck 780 *
1239     * @param pSubChunk - sub chunk or sub list chunk to be removed
1240     */
1241     void List::DeleteSubChunk(Chunk* pSubChunk) {
1242     if (!pSubChunks) LoadSubChunks();
1243     pSubChunks->remove(pSubChunk);
1244     if ((*pSubChunksMap)[pSubChunk->GetChunkID()] == pSubChunk) {
1245     pSubChunksMap->erase(pSubChunk->GetChunkID());
1246     // try to find another chunk of the same chunk ID
1247     ChunkList::iterator iter = pSubChunks->begin();
1248     ChunkList::iterator end = pSubChunks->end();
1249     for (; iter != end; ++iter) {
1250     if ((*iter)->GetChunkID() == pSubChunk->GetChunkID()) {
1251     (*pSubChunksMap)[pSubChunk->GetChunkID()] = *iter;
1252     break; // we're done, stop search
1253     }
1254     }
1255     }
1256     delete pSubChunk;
1257     }
1258    
1259 schoenebeck 2 void List::ReadHeader(unsigned long fPos) {
1260     #if DEBUG
1261     std::cout << "List::Readheader(ulong) ";
1262     #endif // DEBUG
1263     Chunk::ReadHeader(fPos);
1264 persson 1885 if (CurrentChunkSize < 4) return;
1265 schoenebeck 780 NewChunkSize = CurrentChunkSize -= 4;
1266 schoenebeck 2 #if POSIX
1267 schoenebeck 780 lseek(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1268     read(pFile->hFileRead, &ListType, 4);
1269 schoenebeck 1050 #elif defined(WIN32)
1270     SetFilePointer(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, NULL/*32 bit*/, FILE_BEGIN);
1271     DWORD dwBytesRead;
1272     ReadFile(pFile->hFileRead, &ListType, 4, &dwBytesRead, NULL);
1273 schoenebeck 2 #else
1274 schoenebeck 780 fseek(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1275     fread(&ListType, 4, 1, pFile->hFileRead);
1276 schoenebeck 2 #endif // POSIX
1277     #if DEBUG
1278     std::cout << "listType=" << convertToString(ListType) << std::endl;
1279     #endif // DEBUG
1280 schoenebeck 780 if (!pFile->bEndianNative) {
1281 schoenebeck 11 //swapBytes_32(&ListType);
1282 schoenebeck 2 }
1283     }
1284    
1285 schoenebeck 780 void List::WriteHeader(unsigned long fPos) {
1286     // the four list type bytes officially belong the chunk's body in the RIFF format
1287     NewChunkSize += 4;
1288     Chunk::WriteHeader(fPos);
1289     NewChunkSize -= 4; // just revert the +4 incrementation
1290     #if POSIX
1291     lseek(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1292     write(pFile->hFileWrite, &ListType, 4);
1293 schoenebeck 1050 #elif defined(WIN32)
1294     SetFilePointer(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, NULL/*32 bit*/, FILE_BEGIN);
1295     DWORD dwBytesWritten;
1296     WriteFile(pFile->hFileWrite, &ListType, 4, &dwBytesWritten, NULL);
1297 schoenebeck 780 #else
1298     fseek(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1299     fwrite(&ListType, 4, 1, pFile->hFileWrite);
1300     #endif // POSIX
1301     }
1302    
1303 schoenebeck 2 void List::LoadSubChunks() {
1304     #if DEBUG
1305     std::cout << "List::LoadSubChunks()";
1306     #endif // DEBUG
1307     if (!pSubChunks) {
1308     pSubChunks = new ChunkList();
1309     pSubChunksMap = new ChunkMap();
1310 persson 1301 #if defined(WIN32)
1311     if (pFile->hFileRead == INVALID_HANDLE_VALUE) return;
1312     #else
1313 schoenebeck 780 if (!pFile->hFileRead) return;
1314 persson 1301 #endif
1315 schoenebeck 515 unsigned long uiOriginalPos = GetPos();
1316     SetPos(0); // jump to beginning of list chunk body
1317 schoenebeck 2 while (RemainingBytes() >= CHUNK_HEADER_SIZE) {
1318     Chunk* ck;
1319 schoenebeck 11 uint32_t ckid;
1320     Read(&ckid, 4, 1);
1321 schoenebeck 2 #if DEBUG
1322     std::cout << " ckid=" << convertToString(ckid) << std::endl;
1323     #endif // DEBUG
1324     if (ckid == CHUNK_ID_LIST) {
1325 schoenebeck 780 ck = new RIFF::List(pFile, ulStartPos + ulPos - 4, this);
1326 schoenebeck 2 SetPos(ck->GetSize() + LIST_HEADER_SIZE - 4, RIFF::stream_curpos);
1327     }
1328     else { // simple chunk
1329 schoenebeck 780 ck = new RIFF::Chunk(pFile, ulStartPos + ulPos - 4, this);
1330 schoenebeck 2 SetPos(ck->GetSize() + CHUNK_HEADER_SIZE - 4, RIFF::stream_curpos);
1331     }
1332     pSubChunks->push_back(ck);
1333     (*pSubChunksMap)[ckid] = ck;
1334     if (GetPos() % 2 != 0) SetPos(1, RIFF::stream_curpos); // jump over pad byte
1335     }
1336 schoenebeck 515 SetPos(uiOriginalPos); // restore position before this call
1337 schoenebeck 2 }
1338     }
1339    
1340 schoenebeck 833 void List::LoadSubChunksRecursively() {
1341     for (List* pList = GetFirstSubList(); pList; pList = GetNextSubList())
1342     pList->LoadSubChunksRecursively();
1343     }
1344    
1345 schoenebeck 780 /** @brief Write list chunk persistently e.g. to disk.
1346     *
1347     * Stores the list chunk persistently to its actual "physical" file. All
1348     * subchunks (including sub list chunks) will be stored recursively as
1349     * well.
1350     *
1351     * @param ulWritePos - position within the "physical" file where this
1352     * list chunk should be written to
1353     * @param ulCurrentDataOffset - offset of current (old) data within
1354     * the file
1355     * @returns new write position in the "physical" file, that is
1356     * \a ulWritePos incremented by this list chunk's new size
1357     * (including its header size of course)
1358     */
1359     unsigned long List::WriteChunk(unsigned long ulWritePos, unsigned long ulCurrentDataOffset) {
1360 schoenebeck 798 const unsigned long ulOriginalPos = ulWritePos;
1361 schoenebeck 780 ulWritePos += LIST_HEADER_SIZE;
1362    
1363 schoenebeck 798 if (pFile->Mode != stream_mode_read_write)
1364     throw Exception("Cannot write list chunk, file has to be opened in read+write mode");
1365    
1366 schoenebeck 780 // write all subchunks (including sub list chunks) recursively
1367     if (pSubChunks) {
1368     for (ChunkList::iterator iter = pSubChunks->begin(), end = pSubChunks->end(); iter != end; ++iter) {
1369     ulWritePos = (*iter)->WriteChunk(ulWritePos, ulCurrentDataOffset);
1370     }
1371     }
1372    
1373     // update this list chunk's header
1374     CurrentChunkSize = NewChunkSize = ulWritePos - ulOriginalPos - LIST_HEADER_SIZE;
1375     WriteHeader(ulOriginalPos);
1376    
1377 schoenebeck 798 // offset of this list chunk in new written file may have changed
1378     ulStartPos = ulOriginalPos + LIST_HEADER_SIZE;
1379    
1380 schoenebeck 780 return ulWritePos;
1381     }
1382    
1383     void List::__resetPos() {
1384     Chunk::__resetPos();
1385     if (pSubChunks) {
1386     for (ChunkList::iterator iter = pSubChunks->begin(), end = pSubChunks->end(); iter != end; ++iter) {
1387     (*iter)->__resetPos();
1388     }
1389     }
1390     }
1391    
1392 schoenebeck 11 /**
1393     * Returns string representation of the lists's id
1394     */
1395 schoenebeck 2 String List::GetListTypeString() {
1396     return convertToString(ListType);
1397     }
1398    
1399    
1400    
1401     // *************** File ***************
1402     // *
1403    
1404 schoenebeck 1953 //HACK: to avoid breaking DLL compatibility to older versions of libgig we roll the new std::set<Chunk*> into the old std::list<Chunk*> container, should be replaced on member variable level soon though
1405     #define _GET_RESIZED_CHUNKS() \
1406     (reinterpret_cast<std::set<Chunk*>*>(ResizedChunks.front()))
1407    
1408 schoenebeck 780 /** @brief Create new RIFF file.
1409     *
1410     * Use this constructor if you want to create a new RIFF file completely
1411     * "from scratch". Note: there must be no empty chunks or empty list
1412     * chunks when trying to make the new RIFF file persistent with Save()!
1413     *
1414 persson 1207 * Note: by default, the RIFF file will be saved in native endian
1415     * format; that is, as a RIFF file on little-endian machines and
1416     * as a RIFX file on big-endian. To change this behaviour, call
1417     * SetByteOrder() before calling Save().
1418     *
1419 schoenebeck 798 * @param FileType - four-byte identifier of the RIFF file type
1420 persson 1207 * @see AddSubChunk(), AddSubList(), SetByteOrder()
1421 schoenebeck 780 */
1422 schoenebeck 798 File::File(uint32_t FileType) : List(this) {
1423 schoenebeck 1953 //HACK: see _GET_RESIZED_CHUNKS() comment
1424     ResizedChunks.push_back(reinterpret_cast<Chunk*>(new std::set<Chunk*>));
1425 schoenebeck 1050 #if defined(WIN32)
1426     hFileRead = hFileWrite = INVALID_HANDLE_VALUE;
1427     #else
1428 schoenebeck 780 hFileRead = hFileWrite = 0;
1429 schoenebeck 1050 #endif
1430 schoenebeck 798 Mode = stream_mode_closed;
1431 schoenebeck 780 bEndianNative = true;
1432     ulStartPos = RIFF_HEADER_SIZE;
1433 schoenebeck 798 ListType = FileType;
1434 schoenebeck 780 }
1435    
1436     /** @brief Load existing RIFF file.
1437     *
1438     * Loads an existing RIFF file with all its chunks.
1439     *
1440     * @param path - path and file name of the RIFF file to open
1441     * @throws RIFF::Exception if error occured while trying to load the
1442     * given RIFF file
1443     */
1444     File::File(const String& path) : List(this), Filename(path) {
1445 persson 2155 #if DEBUG
1446     std::cout << "File::File("<<path<<")" << std::endl;
1447     #endif // DEBUG
1448     try {
1449     bEndianNative = true;
1450     //HACK: see _GET_RESIZED_CHUNKS() comment
1451     ResizedChunks.push_back(reinterpret_cast<Chunk*>(new std::set<Chunk*>));
1452     #if POSIX
1453     hFileRead = hFileWrite = open(path.c_str(), O_RDONLY | O_NONBLOCK);
1454     if (hFileRead <= 0) {
1455     hFileRead = hFileWrite = 0;
1456     throw RIFF::Exception("Can't open \"" + path + "\"");
1457     }
1458     #elif defined(WIN32)
1459     hFileRead = hFileWrite = CreateFile(
1460     path.c_str(), GENERIC_READ,
1461     FILE_SHARE_READ | FILE_SHARE_WRITE,
1462     NULL, OPEN_EXISTING,
1463     FILE_ATTRIBUTE_NORMAL |
1464     FILE_FLAG_RANDOM_ACCESS, NULL
1465     );
1466     if (hFileRead == INVALID_HANDLE_VALUE) {
1467     hFileRead = hFileWrite = INVALID_HANDLE_VALUE;
1468     throw RIFF::Exception("Can't open \"" + path + "\"");
1469     }
1470     #else
1471     hFileRead = hFileWrite = fopen(path.c_str(), "rb");
1472     if (!hFileRead) throw RIFF::Exception("Can't open \"" + path + "\"");
1473     #endif // POSIX
1474     Mode = stream_mode_read;
1475     ulStartPos = RIFF_HEADER_SIZE;
1476     ReadHeader(0);
1477     if (ChunkID != CHUNK_ID_RIFF && ChunkID != CHUNK_ID_RIFX) {
1478     throw RIFF::Exception("Not a RIFF file");
1479     }
1480 schoenebeck 2 }
1481 persson 2155 catch (...) {
1482     Cleanup();
1483     throw;
1484 schoenebeck 1050 }
1485 schoenebeck 2 }
1486    
1487 schoenebeck 780 String File::GetFileName() {
1488     return Filename;
1489     }
1490    
1491     stream_mode_t File::GetMode() {
1492     return Mode;
1493     }
1494    
1495     /** @brief Change file access mode.
1496     *
1497     * Changes files access mode either to read-only mode or to read/write
1498     * mode.
1499     *
1500     * @param NewMode - new file access mode
1501     * @returns true if mode was changed, false if current mode already
1502     * equals new mode
1503     * @throws RIFF::Exception if new file access mode is unknown
1504     */
1505     bool File::SetMode(stream_mode_t NewMode) {
1506     if (NewMode != Mode) {
1507     switch (NewMode) {
1508     case stream_mode_read:
1509     #if POSIX
1510     if (hFileRead) close(hFileRead);
1511     hFileRead = hFileWrite = open(Filename.c_str(), O_RDONLY | O_NONBLOCK);
1512     if (hFileRead < 0) {
1513     hFileRead = hFileWrite = 0;
1514     throw Exception("Could not (re)open file \"" + Filename + "\" in read mode");
1515     }
1516 schoenebeck 1050 #elif defined(WIN32)
1517     if (hFileRead != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1518     hFileRead = hFileWrite = CreateFile(
1519     Filename.c_str(), GENERIC_READ,
1520     FILE_SHARE_READ | FILE_SHARE_WRITE,
1521     NULL, OPEN_EXISTING,
1522 schoenebeck 1459 FILE_ATTRIBUTE_NORMAL |
1523     FILE_FLAG_RANDOM_ACCESS,
1524     NULL
1525 schoenebeck 1050 );
1526     if (hFileRead == INVALID_HANDLE_VALUE) {
1527     hFileRead = hFileWrite = INVALID_HANDLE_VALUE;
1528     throw Exception("Could not (re)open file \"" + Filename + "\" in read mode");
1529     }
1530 schoenebeck 780 #else
1531     if (hFileRead) fclose(hFileRead);
1532 schoenebeck 1050 hFileRead = hFileWrite = fopen(Filename.c_str(), "rb");
1533 schoenebeck 780 if (!hFileRead) throw Exception("Could not (re)open file \"" + Filename + "\" in read mode");
1534     #endif
1535     __resetPos(); // reset read/write position of ALL 'Chunk' objects
1536 schoenebeck 798 break;
1537 schoenebeck 780 case stream_mode_read_write:
1538     #if POSIX
1539     if (hFileRead) close(hFileRead);
1540     hFileRead = hFileWrite = open(Filename.c_str(), O_RDWR | O_NONBLOCK);
1541     if (hFileRead < 0) {
1542     hFileRead = hFileWrite = open(Filename.c_str(), O_RDONLY | O_NONBLOCK);
1543     throw Exception("Could not open file \"" + Filename + "\" in read+write mode");
1544     }
1545 schoenebeck 1050 #elif defined(WIN32)
1546     if (hFileRead != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1547     hFileRead = hFileWrite = CreateFile(
1548     Filename.c_str(),
1549     GENERIC_READ | GENERIC_WRITE,
1550     FILE_SHARE_READ,
1551     NULL, OPEN_ALWAYS,
1552 schoenebeck 1459 FILE_ATTRIBUTE_NORMAL |
1553     FILE_FLAG_RANDOM_ACCESS,
1554     NULL
1555 schoenebeck 1050 );
1556     if (hFileRead == INVALID_HANDLE_VALUE) {
1557     hFileRead = hFileWrite = CreateFile(
1558     Filename.c_str(), GENERIC_READ,
1559     FILE_SHARE_READ | FILE_SHARE_WRITE,
1560     NULL, OPEN_EXISTING,
1561 schoenebeck 1459 FILE_ATTRIBUTE_NORMAL |
1562     FILE_FLAG_RANDOM_ACCESS,
1563     NULL
1564 schoenebeck 1050 );
1565 persson 1301 throw Exception("Could not (re)open file \"" + Filename + "\" in read+write mode");
1566 schoenebeck 1050 }
1567 schoenebeck 780 #else
1568     if (hFileRead) fclose(hFileRead);
1569 schoenebeck 1050 hFileRead = hFileWrite = fopen(Filename.c_str(), "r+b");
1570 schoenebeck 780 if (!hFileRead) {
1571 schoenebeck 1050 hFileRead = hFileWrite = fopen(Filename.c_str(), "rb");
1572 schoenebeck 780 throw Exception("Could not open file \"" + Filename + "\" in read+write mode");
1573     }
1574     #endif
1575     __resetPos(); // reset read/write position of ALL 'Chunk' objects
1576 schoenebeck 798 break;
1577     case stream_mode_closed:
1578     #if POSIX
1579     if (hFileRead) close(hFileRead);
1580     if (hFileWrite) close(hFileWrite);
1581 schoenebeck 1050 #elif defined(WIN32)
1582     if (hFileRead != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1583     if (hFileWrite != INVALID_HANDLE_VALUE) CloseHandle(hFileWrite);
1584 schoenebeck 798 #else
1585     if (hFileRead) fclose(hFileRead);
1586     if (hFileWrite) fclose(hFileWrite);
1587     #endif
1588     hFileRead = hFileWrite = 0;
1589     break;
1590 schoenebeck 780 default:
1591     throw Exception("Unknown file access mode");
1592     }
1593 schoenebeck 798 Mode = NewMode;
1594     return true;
1595 schoenebeck 780 }
1596     return false;
1597     }
1598    
1599 persson 1184 /** @brief Set the byte order to be used when saving.
1600     *
1601     * Set the byte order to be used in the file. A value of
1602     * endian_little will create a RIFF file, endian_big a RIFX file
1603     * and endian_native will create a RIFF file on little-endian
1604     * machines and RIFX on big-endian machines.
1605     *
1606     * @param Endian - endianess to use when file is saved.
1607     */
1608     void File::SetByteOrder(endian_t Endian) {
1609     #if WORDS_BIGENDIAN
1610     bEndianNative = Endian != endian_little;
1611     #else
1612     bEndianNative = Endian != endian_big;
1613     #endif
1614     }
1615    
1616 schoenebeck 780 /** @brief Save changes to same file.
1617     *
1618     * Make all changes of all chunks persistent by writing them to the
1619     * actual (same) file. The file might temporarily grow to a higher size
1620     * than it will have at the end of the saving process, in case chunks
1621     * were grown.
1622     *
1623     * @throws RIFF::Exception if there is an empty chunk or empty list
1624     * chunk or any kind of IO error occured
1625     */
1626     void File::Save() {
1627 schoenebeck 833 // make sure the RIFF tree is built (from the original file)
1628     LoadSubChunksRecursively();
1629    
1630 schoenebeck 780 // reopen file in write mode
1631 schoenebeck 798 SetMode(stream_mode_read_write);
1632 schoenebeck 780
1633     // to be able to save the whole file without loading everything into
1634     // RAM and without having to store the data in a temporary file, we
1635     // enlarge the file with the sum of all _positive_ chunk size
1636     // changes, move current data towards the end of the file with the
1637     // calculated sum and finally update / rewrite the file by copying
1638     // the old data back to the right position at the beginning of the file
1639    
1640     // first we sum up all positive chunk size changes (and skip all negative ones)
1641     unsigned long ulPositiveSizeDiff = 0;
1642 schoenebeck 1953 std::set<Chunk*>* resizedChunks = _GET_RESIZED_CHUNKS();
1643     for (std::set<Chunk*>::const_iterator iter = resizedChunks->begin(), end = resizedChunks->end(); iter != end; ++iter) {
1644 schoenebeck 1095 if ((*iter)->GetNewSize() == 0) {
1645 schoenebeck 1105 throw Exception("There is at least one empty chunk (zero size): " + __resolveChunkPath(*iter));
1646 schoenebeck 1095 }
1647 persson 1859 unsigned long newSizePadded = (*iter)->GetNewSize() + (*iter)->GetNewSize() % 2;
1648     unsigned long oldSizePadded = (*iter)->GetSize() + (*iter)->GetSize() % 2;
1649     if (newSizePadded > oldSizePadded) ulPositiveSizeDiff += newSizePadded - oldSizePadded;
1650 schoenebeck 780 }
1651    
1652     unsigned long ulWorkingFileSize = GetFileSize();
1653    
1654     // if there are positive size changes...
1655     if (ulPositiveSizeDiff > 0) {
1656     // ... we enlarge this file first ...
1657     ulWorkingFileSize += ulPositiveSizeDiff;
1658     ResizeFile(ulWorkingFileSize);
1659     // ... and move current data by the same amount towards end of file.
1660     int8_t* pCopyBuffer = new int8_t[4096];
1661     const unsigned long ulFileSize = GetSize() + RIFF_HEADER_SIZE;
1662 schoenebeck 1050 #if defined(WIN32)
1663     DWORD iBytesMoved = 1; // we have to pass it via pointer to the Windows API, thus the correct size must be ensured
1664     #else
1665 schoenebeck 1459 int iBytesMoved = 1;
1666 schoenebeck 1050 #endif
1667 persson 1678 for (unsigned long ulPos = ulFileSize; iBytesMoved > 0; ) {
1668     iBytesMoved = (ulPos < 4096) ? ulPos : 4096;
1669     ulPos -= iBytesMoved;
1670 schoenebeck 780 #if POSIX
1671     lseek(hFileRead, ulPos, SEEK_SET);
1672     iBytesMoved = read(hFileRead, pCopyBuffer, iBytesMoved);
1673     lseek(hFileWrite, ulPos + ulPositiveSizeDiff, SEEK_SET);
1674     iBytesMoved = write(hFileWrite, pCopyBuffer, iBytesMoved);
1675 schoenebeck 1050 #elif defined(WIN32)
1676     SetFilePointer(hFileRead, ulPos, NULL/*32 bit*/, FILE_BEGIN);
1677     ReadFile(hFileRead, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
1678     SetFilePointer(hFileWrite, ulPos + ulPositiveSizeDiff, NULL/*32 bit*/, FILE_BEGIN);
1679     WriteFile(hFileWrite, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
1680 schoenebeck 780 #else
1681     fseek(hFileRead, ulPos, SEEK_SET);
1682     iBytesMoved = fread(pCopyBuffer, 1, iBytesMoved, hFileRead);
1683     fseek(hFileWrite, ulPos + ulPositiveSizeDiff, SEEK_SET);
1684     iBytesMoved = fwrite(pCopyBuffer, 1, iBytesMoved, hFileWrite);
1685     #endif
1686     }
1687     delete[] pCopyBuffer;
1688     if (iBytesMoved < 0) throw Exception("Could not modify file while trying to enlarge it");
1689     }
1690    
1691     // rebuild / rewrite complete RIFF tree
1692 schoenebeck 798 unsigned long ulTotalSize = WriteChunk(0, ulPositiveSizeDiff);
1693     unsigned long ulActualSize = __GetFileSize(hFileWrite);
1694 schoenebeck 780
1695     // resize file to the final size
1696 schoenebeck 798 if (ulTotalSize < ulActualSize) ResizeFile(ulTotalSize);
1697 schoenebeck 780
1698     // forget all resized chunks
1699 schoenebeck 1953 resizedChunks->clear();
1700 schoenebeck 780 }
1701    
1702     /** @brief Save changes to another file.
1703     *
1704     * Make all changes of all chunks persistent by writing them to another
1705     * file. <b>Caution:</b> this method is optimized for writing to
1706     * <b>another</b> file, do not use it to save the changes to the same
1707 schoenebeck 798 * file! Use File::Save() in that case instead! Ignoring this might
1708     * result in a corrupted file, especially in case chunks were resized!
1709 schoenebeck 780 *
1710     * After calling this method, this File object will be associated with
1711     * the new file (given by \a path) afterwards.
1712     *
1713     * @param path - path and file name where everything should be written to
1714     */
1715     void File::Save(const String& path) {
1716 schoenebeck 798 //TODO: we should make a check here if somebody tries to write to the same file and automatically call the other Save() method in that case
1717    
1718 schoenebeck 833 // make sure the RIFF tree is built (from the original file)
1719     LoadSubChunksRecursively();
1720    
1721 schoenebeck 798 if (Filename.length() > 0) SetMode(stream_mode_read);
1722 schoenebeck 780 // open the other (new) file for writing and truncate it to zero size
1723     #if POSIX
1724 schoenebeck 798 hFileWrite = open(path.c_str(), O_RDWR | O_CREAT, S_IRUSR | S_IWUSR | S_IRGRP);
1725 schoenebeck 780 if (hFileWrite < 0) {
1726     hFileWrite = hFileRead;
1727     throw Exception("Could not open file \"" + path + "\" for writing");
1728     }
1729 schoenebeck 1050 #elif defined(WIN32)
1730     hFileWrite = CreateFile(
1731     path.c_str(), GENERIC_WRITE, FILE_SHARE_READ,
1732 schoenebeck 1459 NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL |
1733     FILE_FLAG_RANDOM_ACCESS, NULL
1734 schoenebeck 1050 );
1735     if (hFileWrite == INVALID_HANDLE_VALUE) {
1736     hFileWrite = hFileRead;
1737     throw Exception("Could not open file \"" + path + "\" for writing");
1738     }
1739 schoenebeck 780 #else
1740     hFileWrite = fopen(path.c_str(), "w+b");
1741     if (!hFileWrite) {
1742     hFileWrite = hFileRead;
1743     throw Exception("Could not open file \"" + path + "\" for writing");
1744     }
1745     #endif // POSIX
1746 schoenebeck 798 Mode = stream_mode_read_write;
1747 schoenebeck 780
1748     // write complete RIFF tree to the other (new) file
1749 schoenebeck 798 unsigned long ulTotalSize = WriteChunk(0, 0);
1750     unsigned long ulActualSize = __GetFileSize(hFileWrite);
1751 schoenebeck 780
1752 schoenebeck 798 // resize file to the final size (if the file was originally larger)
1753     if (ulTotalSize < ulActualSize) ResizeFile(ulTotalSize);
1754    
1755 schoenebeck 780 // forget all resized chunks
1756 schoenebeck 1953 _GET_RESIZED_CHUNKS()->clear();
1757 schoenebeck 780
1758 persson 1301 #if POSIX
1759     if (hFileWrite) close(hFileWrite);
1760     #elif defined(WIN32)
1761     if (hFileWrite != INVALID_HANDLE_VALUE) CloseHandle(hFileWrite);
1762     #else
1763     if (hFileWrite) fclose(hFileWrite);
1764     #endif
1765     hFileWrite = hFileRead;
1766 schoenebeck 798
1767 schoenebeck 780 // associate new file with this File object from now on
1768     Filename = path;
1769 schoenebeck 798 Mode = (stream_mode_t) -1; // Just set it to an undefined mode ...
1770     SetMode(stream_mode_read_write); // ... so SetMode() has to reopen the file handles.
1771 schoenebeck 780 }
1772    
1773     void File::ResizeFile(unsigned long ulNewSize) {
1774     #if POSIX
1775     if (ftruncate(hFileWrite, ulNewSize) < 0)
1776     throw Exception("Could not resize file \"" + Filename + "\"");
1777 schoenebeck 1050 #elif defined(WIN32)
1778     if (
1779     SetFilePointer(hFileWrite, ulNewSize, NULL/*32 bit*/, FILE_BEGIN) == INVALID_SET_FILE_POINTER ||
1780     !SetEndOfFile(hFileWrite)
1781     ) throw Exception("Could not resize file \"" + Filename + "\"");
1782 schoenebeck 780 #else
1783 schoenebeck 1050 # error Sorry, this version of libgig only supports POSIX and Windows systems yet.
1784 schoenebeck 798 # error Reason: portable implementation of RIFF::File::ResizeFile() is missing (yet)!
1785 schoenebeck 780 #endif
1786     }
1787    
1788 schoenebeck 2 File::~File() {
1789     #if DEBUG
1790     std::cout << "File::~File()" << std::endl;
1791     #endif // DEBUG
1792 persson 2155 Cleanup();
1793     }
1794    
1795     void File::Cleanup() {
1796 schoenebeck 2 #if POSIX
1797 schoenebeck 780 if (hFileRead) close(hFileRead);
1798 schoenebeck 1050 #elif defined(WIN32)
1799     if (hFileRead != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1800 schoenebeck 2 #else
1801 schoenebeck 780 if (hFileRead) fclose(hFileRead);
1802 schoenebeck 2 #endif // POSIX
1803 persson 1869 DeleteChunkList();
1804     pFile = NULL;
1805 schoenebeck 1953 //HACK: see _GET_RESIZED_CHUNKS() comment
1806     delete _GET_RESIZED_CHUNKS();
1807 schoenebeck 2 }
1808    
1809 schoenebeck 780 void File::LogAsResized(Chunk* pResizedChunk) {
1810 schoenebeck 1953 _GET_RESIZED_CHUNKS()->insert(pResizedChunk);
1811 schoenebeck 780 }
1812    
1813 schoenebeck 1095 void File::UnlogResized(Chunk* pResizedChunk) {
1814 schoenebeck 1953 _GET_RESIZED_CHUNKS()->erase(pResizedChunk);
1815 schoenebeck 1095 }
1816    
1817 schoenebeck 2 unsigned long File::GetFileSize() {
1818 schoenebeck 798 return __GetFileSize(hFileRead);
1819     }
1820    
1821     #if POSIX
1822     unsigned long File::__GetFileSize(int hFile) {
1823 schoenebeck 2 struct stat filestat;
1824 schoenebeck 798 fstat(hFile, &filestat);
1825 schoenebeck 2 long size = filestat.st_size;
1826     return size;
1827     }
1828 schoenebeck 1050 #elif defined(WIN32)
1829     unsigned long File::__GetFileSize(HANDLE hFile) {
1830 schoenebeck 1063 DWORD dwSize = ::GetFileSize(hFile, NULL /*32bit*/);
1831     if (dwSize == INVALID_FILE_SIZE)
1832     throw Exception("Windows FS error: could not determine file size");
1833     return dwSize;
1834 schoenebeck 1050 }
1835 schoenebeck 798 #else // standard C functions
1836     unsigned long File::__GetFileSize(FILE* hFile) {
1837     long curpos = ftell(hFile);
1838     fseek(hFile, 0, SEEK_END);
1839     long size = ftell(hFile);
1840     fseek(hFile, curpos, SEEK_SET);
1841     return size;
1842     }
1843     #endif
1844 schoenebeck 2
1845    
1846     // *************** Exception ***************
1847     // *
1848    
1849     void Exception::PrintMessage() {
1850     std::cout << "RIFF::Exception: " << Message << std::endl;
1851     }
1852    
1853 schoenebeck 518
1854     // *************** functions ***************
1855     // *
1856    
1857     /**
1858     * Returns the name of this C++ library. This is usually "libgig" of
1859     * course. This call is equivalent to DLS::libraryName() and
1860     * gig::libraryName().
1861     */
1862     String libraryName() {
1863     return PACKAGE;
1864     }
1865    
1866     /**
1867     * Returns version of this C++ library. This call is equivalent to
1868     * DLS::libraryVersion() and gig::libraryVersion().
1869     */
1870     String libraryVersion() {
1871     return VERSION;
1872     }
1873    
1874 schoenebeck 2 } // namespace RIFF

  ViewVC Help
Powered by ViewVC