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

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

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 780 by schoenebeck, Sun Sep 25 13:40:37 2005 UTC revision 1713 by persson, Thu Mar 6 20:42:22 2008 UTC
# Line 1  Line 1 
1  /***************************************************************************  /***************************************************************************
2   *                                                                         *   *                                                                         *
3   *   libgig - C++ cross-platform Gigasampler format file loader library    *   *   libgig - C++ cross-platform Gigasampler format file access library    *
4   *                                                                         *   *                                                                         *
5   *   Copyright (C) 2003-2005 by Christian Schoenebeck                      *   *   Copyright (C) 2003-2007 by Christian Schoenebeck                      *
6   *                              <cuse@users.sourceforge.net>               *   *                              <cuse@users.sourceforge.net>               *
7   *                                                                         *   *                                                                         *
8   *   This library is free software; you can redistribute it and/or modify  *   *   This library is free software; you can redistribute it and/or modify  *
# Line 21  Line 21 
21   *   MA  02111-1307  USA                                                   *   *   MA  02111-1307  USA                                                   *
22   ***************************************************************************/   ***************************************************************************/
23    
24    #include <algorithm>
25  #include <string.h>  #include <string.h>
 #include <sstream>  
26    
27  #include "RIFF.h"  #include "RIFF.h"
28    
29    #include "helper.h"
30    
31  namespace RIFF {  namespace RIFF {
32    
33  // *************** Helper Functions **************  // *************** Internal functions **************
34  // *  // *
35    
36      template<class T> inline String ToString(T o) {      /// Returns a human readable path of the given chunk.
37          std::stringstream ss;      static String __resolveChunkPath(Chunk* pCk) {
38          ss << o;          String sPath;
39          return ss.str();          for (Chunk* pChunk = pCk; pChunk; pChunk = pChunk->GetParent()) {
40                if (pChunk->GetChunkID() == CHUNK_ID_LIST) {
41                    List* pList = (List*) pChunk;
42                    sPath = "->'" + pList->GetListTypeString() + "'" + sPath;
43                } else {
44                    sPath = "->'" + pChunk->GetChunkIDString() + "'" + sPath;
45                }
46            }
47            return sPath;
48      }      }
49    
50    
# Line 49  namespace RIFF { Line 59  namespace RIFF {
59          ulPos      = 0;          ulPos      = 0;
60          pParent    = NULL;          pParent    = NULL;
61          pChunkData = NULL;          pChunkData = NULL;
62            ulChunkDataSize = 0;
63            ChunkID    = CHUNK_ID_RIFF;
64          this->pFile = pFile;          this->pFile = pFile;
65      }      }
66    
# Line 61  namespace RIFF { Line 73  namespace RIFF {
73          pParent       = Parent;          pParent       = Parent;
74          ulPos         = 0;          ulPos         = 0;
75          pChunkData    = NULL;          pChunkData    = NULL;
76            ulChunkDataSize = 0;
77          ReadHeader(StartPos);          ReadHeader(StartPos);
78      }      }
79    
# Line 70  namespace RIFF { Line 83  namespace RIFF {
83          this->pParent    = pParent;          this->pParent    = pParent;
84          ulPos            = 0;          ulPos            = 0;
85          pChunkData       = NULL;          pChunkData       = NULL;
86            ulChunkDataSize  = 0;
87          ChunkID          = uiChunkID;          ChunkID          = uiChunkID;
88          CurrentChunkSize = 0;          CurrentChunkSize = 0;
89          NewChunkSize     = uiBodySize;          NewChunkSize     = uiBodySize;
90      }      }
91    
92      Chunk::~Chunk() {      Chunk::~Chunk() {
93            pFile->UnlogResized(this);
94          if (pChunkData) delete[] pChunkData;          if (pChunkData) delete[] pChunkData;
95      }      }
96    
# Line 87  namespace RIFF { Line 102  namespace RIFF {
102          if (lseek(pFile->hFileRead, fPos, SEEK_SET) != -1) {          if (lseek(pFile->hFileRead, fPos, SEEK_SET) != -1) {
103              read(pFile->hFileRead, &ChunkID, 4);              read(pFile->hFileRead, &ChunkID, 4);
104              read(pFile->hFileRead, &CurrentChunkSize, 4);              read(pFile->hFileRead, &CurrentChunkSize, 4);
105            #elif defined(WIN32)
106            if (SetFilePointer(pFile->hFileRead, fPos, NULL/*32 bit*/, FILE_BEGIN) != INVALID_SET_FILE_POINTER) {
107                DWORD dwBytesRead;
108                ReadFile(pFile->hFileRead, &ChunkID, 4, &dwBytesRead, NULL);
109                ReadFile(pFile->hFileRead, &CurrentChunkSize, 4, &dwBytesRead, NULL);
110          #else          #else
111          if (!fseek(pFile->hFileRead, fPos, SEEK_SET)) {          if (!fseek(pFile->hFileRead, fPos, SEEK_SET)) {
112              fread(&ChunkID, 4, 1, pFile->hFileRead);              fread(&ChunkID, 4, 1, pFile->hFileRead);
# Line 94  namespace RIFF { Line 114  namespace RIFF {
114          #endif // POSIX          #endif // POSIX
115              #if WORDS_BIGENDIAN              #if WORDS_BIGENDIAN
116              if (ChunkID == CHUNK_ID_RIFF) {              if (ChunkID == CHUNK_ID_RIFF) {
117                  bEndianNative = false;                  pFile->bEndianNative = false;
118              }              }
119              #else // little endian              #else // little endian
120              if (ChunkID == CHUNK_ID_RIFX) {              if (ChunkID == CHUNK_ID_RIFX) {
# Line 135  namespace RIFF { Line 155  namespace RIFF {
155              write(pFile->hFileWrite, &uiNewChunkID, 4);              write(pFile->hFileWrite, &uiNewChunkID, 4);
156              write(pFile->hFileWrite, &uiNewChunkSize, 4);              write(pFile->hFileWrite, &uiNewChunkSize, 4);
157          }          }
158            #elif defined(WIN32)
159            if (SetFilePointer(pFile->hFileWrite, fPos, NULL/*32 bit*/, FILE_BEGIN) != INVALID_SET_FILE_POINTER) {
160                DWORD dwBytesWritten;
161                WriteFile(pFile->hFileWrite, &uiNewChunkID, 4, &dwBytesWritten, NULL);
162                WriteFile(pFile->hFileWrite, &uiNewChunkSize, 4, &dwBytesWritten, NULL);
163            }
164          #else          #else
165          if (!fseek(pFile->hFileWrite, fPos, SEEK_SET)) {          if (!fseek(pFile->hFileWrite, fPos, SEEK_SET)) {
166              fwrite(&uiNewChunkID, 4, 1, pFile->hFileWrite);              fwrite(&uiNewChunkID, 4, 1, pFile->hFileWrite);
# Line 210  namespace RIFF { Line 236  namespace RIFF {
236       *  - RIFF::stream_closed :       *  - RIFF::stream_closed :
237       *    the data stream was closed somehow, no more reading possible       *    the data stream was closed somehow, no more reading possible
238       *  - RIFF::stream_end_reached :       *  - RIFF::stream_end_reached :
239       *    alreaady reached the end of the chunk data, no more reading       *    already reached the end of the chunk data, no more reading
240       *    possible without SetPos()       *    possible without SetPos()
241       */       */
242      stream_state_t Chunk::GetState() {      stream_state_t Chunk::GetState() {
# Line 218  namespace RIFF { Line 244  namespace RIFF {
244        std::cout << "Chunk::GetState()" << std::endl;        std::cout << "Chunk::GetState()" << std::endl;
245        #endif // DEBUG        #endif // DEBUG
246          #if POSIX          #if POSIX
247          if (pFile->hFileRead == 0)    return stream_closed;          if (pFile->hFileRead == 0) return stream_closed;
248            #elif defined (WIN32)
249            if (pFile->hFileRead == INVALID_HANDLE_VALUE)
250                return stream_closed;
251          #else          #else
252          if (pFile->hFileRead == NULL) return stream_closed;          if (pFile->hFileRead == NULL) return stream_closed;
253          #endif // POSIX          #endif // POSIX
# Line 252  namespace RIFF { Line 281  namespace RIFF {
281          unsigned long readWords = read(pFile->hFileRead, pData, WordCount * WordSize);          unsigned long readWords = read(pFile->hFileRead, pData, WordCount * WordSize);
282          if (readWords < 1) return 0;          if (readWords < 1) return 0;
283          readWords /= WordSize;          readWords /= WordSize;
284            #elif defined(WIN32)
285            if (SetFilePointer(pFile->hFileRead, ulStartPos + ulPos, NULL/*32 bit*/, FILE_BEGIN) == INVALID_SET_FILE_POINTER) return 0;
286            DWORD readWords;
287            ReadFile(pFile->hFileRead, pData, WordCount * WordSize, &readWords, NULL);
288            if (readWords < 1) return 0;
289            readWords /= WordSize;
290          #else // standard C functions          #else // standard C functions
291          if (fseek(pFile->hFileRead, ulStartPos + ulPos, SEEK_SET)) return 0;          if (fseek(pFile->hFileRead, ulStartPos + ulPos, SEEK_SET)) return 0;
292          unsigned long readWords = fread(pData, WordSize, WordCount, pFile->hFileRead);          unsigned long readWords = fread(pData, WordSize, WordCount, pFile->hFileRead);
# Line 293  namespace RIFF { Line 328  namespace RIFF {
328       *  @see Resize()       *  @see Resize()
329       */       */
330      unsigned long Chunk::Write(void* pData, unsigned long WordCount, unsigned long WordSize) {      unsigned long Chunk::Write(void* pData, unsigned long WordCount, unsigned long WordSize) {
331          if (pFile->Mode == stream_mode_read)          if (pFile->Mode != stream_mode_read_write)
332              throw Exception("Writing chunk data in read-only file access mode was attempted");              throw Exception("Cannot write data to chunk, file has to be opened in read+write mode first");
333          if (ulPos >= CurrentChunkSize || ulPos + WordCount * WordSize >= CurrentChunkSize)          if (ulPos >= CurrentChunkSize || ulPos + WordCount * WordSize > CurrentChunkSize)
334              throw Exception("End of chunk reached while trying to write data");              throw Exception("End of chunk reached while trying to write data");
335          if (!pFile->bEndianNative && WordSize != 1) {          if (!pFile->bEndianNative && WordSize != 1) {
336              switch (WordSize) {              switch (WordSize) {
# Line 321  namespace RIFF { Line 356  namespace RIFF {
356          unsigned long writtenWords = write(pFile->hFileWrite, pData, WordCount * WordSize);          unsigned long writtenWords = write(pFile->hFileWrite, pData, WordCount * WordSize);
357          if (writtenWords < 1) throw Exception("POSIX IO Error while trying to write chunk data");          if (writtenWords < 1) throw Exception("POSIX IO Error while trying to write chunk data");
358          writtenWords /= WordSize;          writtenWords /= WordSize;
359            #elif defined(WIN32)
360            if (SetFilePointer(pFile->hFileWrite, ulStartPos + ulPos, NULL/*32 bit*/, FILE_BEGIN) == INVALID_SET_FILE_POINTER) {
361                throw Exception("Could not seek to position " + ToString(ulPos) +
362                                " in chunk (" + ToString(ulStartPos + ulPos) + " in file)");
363            }
364            DWORD writtenWords;
365            WriteFile(pFile->hFileWrite, pData, WordCount * WordSize, &writtenWords, NULL);
366            if (writtenWords < 1) throw Exception("Windows IO Error while trying to write chunk data");
367            writtenWords /= WordSize;
368          #else // standard C functions          #else // standard C functions
369          if (fseek(pFile->hFileWrite, ulStartPos + ulPos, SEEK_SET)) {          if (fseek(pFile->hFileWrite, ulStartPos + ulPos, SEEK_SET)) {
370              throw Exception("Could not seek to position " + ToString(ulPos) +              throw Exception("Could not seek to position " + ToString(ulPos) +
# Line 667  namespace RIFF { Line 711  namespace RIFF {
711       *       *
712       * <b>Caution:</b> the buffer pointer will be invalidated once       * <b>Caution:</b> the buffer pointer will be invalidated once
713       * File::Save() was called. You have to call LoadChunkData() again to       * File::Save() was called. You have to call LoadChunkData() again to
714       * get a new pointer whenever File::Save() was called.       * get a new, valid pointer whenever File::Save() was called.
715         *
716         * You can call LoadChunkData() again if you previously scheduled to
717         * enlarge this chunk with a Resize() call. In that case the buffer will
718         * be enlarged to the new, scheduled chunk size and you can already
719         * place the new chunk data to the buffer and finally call File::Save()
720         * to enlarge the chunk physically and write the new data in one rush.
721         * This approach is definitely recommended if you have to enlarge and
722         * write new data to a lot of chunks.
723       *       *
724       * @returns a pointer to the data in RAM on success, NULL otherwise       * @returns a pointer to the data in RAM on success, NULL otherwise
725         * @throws Exception if data buffer could not be enlarged
726       * @see ReleaseChunkData()       * @see ReleaseChunkData()
727       */       */
728      void* Chunk::LoadChunkData() {      void* Chunk::LoadChunkData() {
729          if (!pChunkData) {          if (!pChunkData && pFile->Filename != "") {
730              #if POSIX              #if POSIX
731              if (lseek(pFile->hFileRead, ulStartPos, SEEK_SET) == -1) return NULL;              if (lseek(pFile->hFileRead, ulStartPos, SEEK_SET) == -1) return NULL;
732              pChunkData = new uint8_t[GetSize()];              #elif defined(WIN32)
733              if (!pChunkData) return NULL;              if (SetFilePointer(pFile->hFileRead, ulStartPos, NULL/*32 bit*/, FILE_BEGIN) == INVALID_SET_FILE_POINTER) return NULL;
             unsigned long readWords = read(pFile->hFileRead, pChunkData, GetSize());  
734              #else              #else
735              if (fseek(pFile->hFileRead, ulStartPos, SEEK_SET)) return NULL;              if (fseek(pFile->hFileRead, ulStartPos, SEEK_SET)) return NULL;
736              pChunkData = new uint8_t[GetSize()];              #endif // POSIX
737                unsigned long ulBufferSize = (CurrentChunkSize > NewChunkSize) ? CurrentChunkSize : NewChunkSize;
738                pChunkData = new uint8_t[ulBufferSize];
739              if (!pChunkData) return NULL;              if (!pChunkData) return NULL;
740                memset(pChunkData, 0, ulBufferSize);
741                #if POSIX
742                unsigned long readWords = read(pFile->hFileRead, pChunkData, GetSize());
743                #elif defined(WIN32)
744                DWORD readWords;
745                ReadFile(pFile->hFileRead, pChunkData, GetSize(), &readWords, NULL);
746                #else
747              unsigned long readWords = fread(pChunkData, 1, GetSize(), pFile->hFileRead);              unsigned long readWords = fread(pChunkData, 1, GetSize(), pFile->hFileRead);
748              #endif // POSIX              #endif // POSIX
749              if (readWords != GetSize()) {              if (readWords != GetSize()) {
750                  delete[] pChunkData;                  delete[] pChunkData;
751                  return (pChunkData = NULL);                  return (pChunkData = NULL);
752              }              }
753                ulChunkDataSize = ulBufferSize;
754            } else if (NewChunkSize > ulChunkDataSize) {
755                uint8_t* pNewBuffer = new uint8_t[NewChunkSize];
756                if (!pNewBuffer) throw Exception("Could not enlarge chunk data buffer to " + ToString(NewChunkSize) + " bytes");
757                memset(pNewBuffer, 0 , NewChunkSize);
758                memcpy(pNewBuffer, pChunkData, ulChunkDataSize);
759                delete[] pChunkData;
760                pChunkData      = pNewBuffer;
761                ulChunkDataSize = NewChunkSize;
762          }          }
763          return pChunkData;          return pChunkData;
764      }      }
# Line 718  namespace RIFF { Line 788  namespace RIFF {
788       *       *
789       * <b>Caution:</b> You cannot directly write to enlarged chunks before       * <b>Caution:</b> You cannot directly write to enlarged chunks before
790       * calling File::Save() as this might exceed the current chunk's body       * calling File::Save() as this might exceed the current chunk's body
791       * boundary.       * boundary!
792       *       *
793       * @param iNewSize - new chunk body size in bytes (must be greater than zero)       * @param iNewSize - new chunk body size in bytes (must be greater than zero)
794       * @throws RIFF::Exception  if \a iNewSize is less than 1       * @throws RIFF::Exception  if \a iNewSize is less than 1
795       * @see File::Save()       * @see File::Save()
796       */       */
797      void Chunk::Resize(int iNewSize) {      void Chunk::Resize(int iNewSize) {
798          if (iNewSize <= 0) throw Exception("Chunk size must be at least one byte");          if (iNewSize <= 0)
799                throw Exception("There is at least one empty chunk (zero size): " + __resolveChunkPath(this));
800            if (NewChunkSize == iNewSize) return;
801          NewChunkSize = iNewSize;          NewChunkSize = iNewSize;
802          pFile->LogAsResized(this);          pFile->LogAsResized(this);
803      }      }
# Line 743  namespace RIFF { Line 815  namespace RIFF {
815       *          (including its header size of course)       *          (including its header size of course)
816       */       */
817      unsigned long Chunk::WriteChunk(unsigned long ulWritePos, unsigned long ulCurrentDataOffset) {      unsigned long Chunk::WriteChunk(unsigned long ulWritePos, unsigned long ulCurrentDataOffset) {
818          unsigned long ulOriginalPos = ulWritePos;          const unsigned long ulOriginalPos = ulWritePos;
819          ulWritePos += CHUNK_HEADER_SIZE;          ulWritePos += CHUNK_HEADER_SIZE;
820    
821            if (pFile->Mode != stream_mode_read_write)
822                throw Exception("Cannot write list chunk, file has to be opened in read+write mode");
823    
824          // if the whole chunk body was loaded into RAM          // if the whole chunk body was loaded into RAM
825          if (pChunkData) {          if (pChunkData) {
826              // in case the chunk size was changed, reallocate the data in RAM with the chunk's new size              // make sure chunk data buffer in RAM is at least as large as the new chunk size
827              if (NewChunkSize != CurrentChunkSize) {              LoadChunkData();
                 uint8_t* pNewBuffer = new uint8_t[NewChunkSize];  
                 if (NewChunkSize > CurrentChunkSize) {  
                     memcpy(pNewBuffer, pChunkData, CurrentChunkSize);  
                     memset(pNewBuffer + CurrentChunkSize, 0, NewChunkSize - CurrentChunkSize);  
                 } else {  
                     memcpy(pNewBuffer, pChunkData, NewChunkSize);  
                 }  
                 delete[] pChunkData;  
                 pChunkData = pNewBuffer;  
             }  
   
828              // write chunk data from RAM persistently to the file              // write chunk data from RAM persistently to the file
829              #if POSIX              #if POSIX
830              lseek(pFile->hFileWrite, ulWritePos, SEEK_SET);              lseek(pFile->hFileWrite, ulWritePos, SEEK_SET);
831              if (write(pFile->hFileWrite, pChunkData, NewChunkSize) != NewChunkSize) {              if (write(pFile->hFileWrite, pChunkData, NewChunkSize) != NewChunkSize) {
832                  throw Exception("Writing Chunk data (from RAM) failed");                  throw Exception("Writing Chunk data (from RAM) failed");
833              }              }
834                #elif defined(WIN32)
835                SetFilePointer(pFile->hFileWrite, ulWritePos, NULL/*32 bit*/, FILE_BEGIN);
836                DWORD dwBytesWritten;
837                WriteFile(pFile->hFileWrite, pChunkData, NewChunkSize, &dwBytesWritten, NULL);
838                if (dwBytesWritten != NewChunkSize) {
839                    throw Exception("Writing Chunk data (from RAM) failed");
840                }
841              #else              #else
842              fseek(pFile->hFileWrite, ulWritePos, SEEK_SET);              fseek(pFile->hFileWrite, ulWritePos, SEEK_SET);
843              if (fwrite(pChunkData, 1, NewChunkSize, pFile->hFileWrite) != NewChunkSize) {              if (fwrite(pChunkData, 1, NewChunkSize, pFile->hFileWrite) != NewChunkSize) {
# Line 777  namespace RIFF { Line 848  namespace RIFF {
848              // move chunk data from the end of the file to the appropriate position              // move chunk data from the end of the file to the appropriate position
849              int8_t* pCopyBuffer = new int8_t[4096];              int8_t* pCopyBuffer = new int8_t[4096];
850              unsigned long ulToMove = (NewChunkSize < CurrentChunkSize) ? NewChunkSize : CurrentChunkSize;              unsigned long ulToMove = (NewChunkSize < CurrentChunkSize) ? NewChunkSize : CurrentChunkSize;
851                #if defined(WIN32)
852                DWORD iBytesMoved = 1; // we have to pass it via pointer to the Windows API, thus the correct size must be ensured
853                #else
854              int iBytesMoved = 1;              int iBytesMoved = 1;
855                #endif
856              for (unsigned long ulOffset = 0; iBytesMoved > 0; ulOffset += iBytesMoved, ulToMove -= iBytesMoved) {              for (unsigned long ulOffset = 0; iBytesMoved > 0; ulOffset += iBytesMoved, ulToMove -= iBytesMoved) {
857                  iBytesMoved = (ulToMove < 4096) ? ulToMove : 4096;                  iBytesMoved = (ulToMove < 4096) ? ulToMove : 4096;
858                  #if POSIX                  #if POSIX
# Line 785  namespace RIFF { Line 860  namespace RIFF {
860                  iBytesMoved = read(pFile->hFileRead, pCopyBuffer, iBytesMoved);                  iBytesMoved = read(pFile->hFileRead, pCopyBuffer, iBytesMoved);
861                  lseek(pFile->hFileWrite, ulWritePos + ulOffset, SEEK_SET);                  lseek(pFile->hFileWrite, ulWritePos + ulOffset, SEEK_SET);
862                  iBytesMoved = write(pFile->hFileWrite, pCopyBuffer, iBytesMoved);                  iBytesMoved = write(pFile->hFileWrite, pCopyBuffer, iBytesMoved);
863                    #elif defined(WIN32)
864                    SetFilePointer(pFile->hFileRead, ulStartPos + ulCurrentDataOffset + ulOffset, NULL/*32 bit*/, FILE_BEGIN);
865                    ReadFile(pFile->hFileRead, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
866                    SetFilePointer(pFile->hFileWrite, ulWritePos + ulOffset, NULL/*32 bit*/, FILE_BEGIN);
867                    WriteFile(pFile->hFileWrite, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
868                  #else                  #else
869                  fseek(pFile->hFileRead, ulStartPos + ulCurrentDataOffset + ulOffset, SEEK_SET);                  fseek(pFile->hFileRead, ulStartPos + ulCurrentDataOffset + ulOffset, SEEK_SET);
870                  iBytesMoved = fread(pCopyBuffer, 1, iBytesMoved, pFile->hFileRead);                  iBytesMoved = fread(pCopyBuffer, 1, iBytesMoved, pFile->hFileRead);
# Line 806  namespace RIFF { Line 886  namespace RIFF {
886    
887          // add pad byte if needed          // add pad byte if needed
888          if ((ulStartPos + NewChunkSize) % 2 != 0) {          if ((ulStartPos + NewChunkSize) % 2 != 0) {
889              char cPadByte = 0;              const char cPadByte = 0;
890              #if POSIX              #if POSIX
891                lseek(pFile->hFileWrite, ulStartPos + NewChunkSize, SEEK_SET);
892              write(pFile->hFileWrite, &cPadByte, 1);              write(pFile->hFileWrite, &cPadByte, 1);
893                #elif defined(WIN32)
894                SetFilePointer(pFile->hFileWrite, ulStartPos + NewChunkSize, NULL/*32 bit*/, FILE_BEGIN);
895                DWORD dwBytesWritten;
896                WriteFile(pFile->hFileWrite, &cPadByte, 1, &dwBytesWritten, NULL);
897              #else              #else
898                fseek(pFile->hFileWrite, ulStartPos + NewChunkSize, SEEK_SET);
899              fwrite(&cPadByte, 1, 1, pFile->hFileWrite);              fwrite(&cPadByte, 1, 1, pFile->hFileWrite);
900              #endif              #endif
901              return ulStartPos + NewChunkSize + 1;              return ulStartPos + NewChunkSize + 1;
# Line 1063  namespace RIFF { Line 1149  namespace RIFF {
1149      Chunk* List::AddSubChunk(uint32_t uiChunkID, uint uiBodySize) {      Chunk* List::AddSubChunk(uint32_t uiChunkID, uint uiBodySize) {
1150          if (uiBodySize == 0) throw Exception("Chunk body size must be at least 1 byte");          if (uiBodySize == 0) throw Exception("Chunk body size must be at least 1 byte");
1151          if (!pSubChunks) LoadSubChunks();          if (!pSubChunks) LoadSubChunks();
1152          Chunk* pNewChunk = new Chunk(pFile, this, uiChunkID, uiBodySize);          Chunk* pNewChunk = new Chunk(pFile, this, uiChunkID, 0);
1153          pSubChunks->push_back(pNewChunk);          pSubChunks->push_back(pNewChunk);
1154          (*pSubChunksMap)[uiChunkID] = pNewChunk;          (*pSubChunksMap)[uiChunkID] = pNewChunk;
1155          pFile->ResizedChunks.push_back(pNewChunk);          pNewChunk->Resize(uiBodySize);
1156            NewChunkSize += CHUNK_HEADER_SIZE;
1157            pFile->LogAsResized(this);
1158          return pNewChunk;          return pNewChunk;
1159      }      }
1160    
1161        /** @brief Moves a sub chunk.
1162         *
1163         * Moves a sub chunk from one position in a list to another
1164         * position in the same list. The pSrc chunk is placed before the
1165         * pDst chunk.
1166         *
1167         * @param pSrc - sub chunk to be moved
1168         * @param pDst - the position to move to. pSrc will be placed
1169         *               before pDst. If pDst is 0, pSrc will be placed
1170         *               last in list.
1171         */
1172        void List::MoveSubChunk(Chunk* pSrc, Chunk* pDst) {
1173            if (!pSubChunks) LoadSubChunks();
1174            pSubChunks->remove(pSrc);
1175            ChunkList::iterator iter = find(pSubChunks->begin(), pSubChunks->end(), pDst);
1176            pSubChunks->insert(iter, pSrc);
1177        }
1178    
1179      /** @brief Creates a new list sub chunk.      /** @brief Creates a new list sub chunk.
1180       *       *
1181       * Creates and adds a new list sub chunk to this list chunk. Note that       * Creates and adds a new list sub chunk to this list chunk. Note that
# Line 1084  namespace RIFF { Line 1190  namespace RIFF {
1190          List* pNewListChunk = new List(pFile, this, uiListType);          List* pNewListChunk = new List(pFile, this, uiListType);
1191          pSubChunks->push_back(pNewListChunk);          pSubChunks->push_back(pNewListChunk);
1192          (*pSubChunksMap)[CHUNK_ID_LIST] = pNewListChunk;          (*pSubChunksMap)[CHUNK_ID_LIST] = pNewListChunk;
1193            NewChunkSize += LIST_HEADER_SIZE;
1194            pFile->LogAsResized(this);
1195          return pNewListChunk;          return pNewListChunk;
1196      }      }
1197    
# Line 1091  namespace RIFF { Line 1199  namespace RIFF {
1199       *       *
1200       * Removes the sub chunk given by \a pSubChunk from this list and frees       * Removes the sub chunk given by \a pSubChunk from this list and frees
1201       * it completely from RAM. The given chunk can either be a normal sub       * it completely from RAM. The given chunk can either be a normal sub
1202       * chunk or a list sub chunk. You should call File::Save() to make this       * chunk or a list sub chunk. In case the given chunk is a list chunk,
1203       * change persistent at any time.       * all its subchunks (if any) will be removed recursively as well. You
1204         * should call File::Save() to make this change persistent at any time.
1205       *       *
1206       * @param pSubChunk - sub chunk or sub list chunk to be removed       * @param pSubChunk - sub chunk or sub list chunk to be removed
1207       */       */
# Line 1123  namespace RIFF { Line 1232  namespace RIFF {
1232          #if POSIX          #if POSIX
1233          lseek(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, SEEK_SET);          lseek(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1234          read(pFile->hFileRead, &ListType, 4);          read(pFile->hFileRead, &ListType, 4);
1235            #elif defined(WIN32)
1236            SetFilePointer(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, NULL/*32 bit*/, FILE_BEGIN);
1237            DWORD dwBytesRead;
1238            ReadFile(pFile->hFileRead, &ListType, 4, &dwBytesRead, NULL);
1239          #else          #else
1240          fseek(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, SEEK_SET);          fseek(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1241          fread(&ListType, 4, 1, pFile->hFileRead);          fread(&ListType, 4, 1, pFile->hFileRead);
# Line 1143  namespace RIFF { Line 1256  namespace RIFF {
1256          #if POSIX          #if POSIX
1257          lseek(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, SEEK_SET);          lseek(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1258          write(pFile->hFileWrite, &ListType, 4);          write(pFile->hFileWrite, &ListType, 4);
1259            #elif defined(WIN32)
1260            SetFilePointer(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, NULL/*32 bit*/, FILE_BEGIN);
1261            DWORD dwBytesWritten;
1262            WriteFile(pFile->hFileWrite, &ListType, 4, &dwBytesWritten, NULL);
1263          #else          #else
1264          fseek(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, SEEK_SET);          fseek(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1265          fwrite(&ListType, 4, 1, pFile->hFileWrite);          fwrite(&ListType, 4, 1, pFile->hFileWrite);
# Line 1156  namespace RIFF { Line 1273  namespace RIFF {
1273          if (!pSubChunks) {          if (!pSubChunks) {
1274              pSubChunks    = new ChunkList();              pSubChunks    = new ChunkList();
1275              pSubChunksMap = new ChunkMap();              pSubChunksMap = new ChunkMap();
1276                #if defined(WIN32)
1277                if (pFile->hFileRead == INVALID_HANDLE_VALUE) return;
1278                #else
1279              if (!pFile->hFileRead) return;              if (!pFile->hFileRead) return;
1280                #endif
1281              unsigned long uiOriginalPos = GetPos();              unsigned long uiOriginalPos = GetPos();
1282              SetPos(0); // jump to beginning of list chunk body              SetPos(0); // jump to beginning of list chunk body
1283              while (RemainingBytes() >= CHUNK_HEADER_SIZE) {              while (RemainingBytes() >= CHUNK_HEADER_SIZE) {
# Line 1182  namespace RIFF { Line 1303  namespace RIFF {
1303          }          }
1304      }      }
1305    
1306        void List::LoadSubChunksRecursively() {
1307            for (List* pList = GetFirstSubList(); pList; pList = GetNextSubList())
1308                pList->LoadSubChunksRecursively();
1309        }
1310    
1311      /** @brief Write list chunk persistently e.g. to disk.      /** @brief Write list chunk persistently e.g. to disk.
1312       *       *
1313       * Stores the list chunk persistently to its actual "physical" file. All       * Stores the list chunk persistently to its actual "physical" file. All
# Line 1197  namespace RIFF { Line 1323  namespace RIFF {
1323       *          (including its header size of course)       *          (including its header size of course)
1324       */       */
1325      unsigned long List::WriteChunk(unsigned long ulWritePos, unsigned long ulCurrentDataOffset) {      unsigned long List::WriteChunk(unsigned long ulWritePos, unsigned long ulCurrentDataOffset) {
1326          unsigned long ulOriginalPos = ulWritePos;          const unsigned long ulOriginalPos = ulWritePos;
1327          ulWritePos += LIST_HEADER_SIZE;          ulWritePos += LIST_HEADER_SIZE;
1328    
1329            if (pFile->Mode != stream_mode_read_write)
1330                throw Exception("Cannot write list chunk, file has to be opened in read+write mode");
1331    
1332          // write all subchunks (including sub list chunks) recursively          // write all subchunks (including sub list chunks) recursively
1333          if (pSubChunks) {          if (pSubChunks) {
1334              for (ChunkList::iterator iter = pSubChunks->begin(), end = pSubChunks->end(); iter != end; ++iter) {              for (ChunkList::iterator iter = pSubChunks->begin(), end = pSubChunks->end(); iter != end; ++iter) {
# Line 1211  namespace RIFF { Line 1340  namespace RIFF {
1340          CurrentChunkSize = NewChunkSize = ulWritePos - ulOriginalPos - LIST_HEADER_SIZE;          CurrentChunkSize = NewChunkSize = ulWritePos - ulOriginalPos - LIST_HEADER_SIZE;
1341          WriteHeader(ulOriginalPos);          WriteHeader(ulOriginalPos);
1342    
1343            // offset of this list chunk in new written file may have changed
1344            ulStartPos = ulOriginalPos + LIST_HEADER_SIZE;
1345    
1346          return ulWritePos;          return ulWritePos;
1347      }      }
1348    
# Line 1241  namespace RIFF { Line 1373  namespace RIFF {
1373       * "from scratch". Note: there must be no empty chunks or empty list       * "from scratch". Note: there must be no empty chunks or empty list
1374       * chunks when trying to make the new RIFF file persistent with Save()!       * chunks when trying to make the new RIFF file persistent with Save()!
1375       *       *
1376       * @see AddSubChunk(), AddSubList()       * Note: by default, the RIFF file will be saved in native endian
1377       */       * format; that is, as a RIFF file on little-endian machines and
1378      File::File() : List(this) {       * as a RIFX file on big-endian. To change this behaviour, call
1379         * SetByteOrder() before calling Save().
1380         *
1381         * @param FileType - four-byte identifier of the RIFF file type
1382         * @see AddSubChunk(), AddSubList(), SetByteOrder()
1383         */
1384        File::File(uint32_t FileType) : List(this) {
1385            #if defined(WIN32)
1386            hFileRead = hFileWrite = INVALID_HANDLE_VALUE;
1387            #else
1388          hFileRead = hFileWrite = 0;          hFileRead = hFileWrite = 0;
1389            #endif
1390            Mode = stream_mode_closed;
1391          bEndianNative = true;          bEndianNative = true;
1392          ulStartPos = RIFF_HEADER_SIZE;          ulStartPos = RIFF_HEADER_SIZE;
1393            ListType = FileType;
1394      }      }
1395    
1396      /** @brief Load existing RIFF file.      /** @brief Load existing RIFF file.
# Line 1268  namespace RIFF { Line 1412  namespace RIFF {
1412              hFileRead = hFileWrite = 0;              hFileRead = hFileWrite = 0;
1413              throw RIFF::Exception("Can't open \"" + path + "\"");              throw RIFF::Exception("Can't open \"" + path + "\"");
1414          }          }
1415            #elif defined(WIN32)
1416            hFileRead = hFileWrite = CreateFile(
1417                                         path.c_str(), GENERIC_READ,
1418                                         FILE_SHARE_READ | FILE_SHARE_WRITE,
1419                                         NULL, OPEN_EXISTING,
1420                                         FILE_ATTRIBUTE_NORMAL |
1421                                         FILE_FLAG_RANDOM_ACCESS, NULL
1422                                     );
1423            if (hFileRead == INVALID_HANDLE_VALUE) {
1424                hFileRead = hFileWrite = INVALID_HANDLE_VALUE;
1425                throw RIFF::Exception("Can't open \"" + path + "\"");
1426            }
1427          #else          #else
1428          hFileRead = hFileWrite = fopen(path.c_str(), "rb");          hFileRead = hFileWrite = fopen(path.c_str(), "rb");
1429          if (!hFile) throw RIFF::Exception("Can't open \"" + path + "\"");          if (!hFileRead) throw RIFF::Exception("Can't open \"" + path + "\"");
1430          #endif // POSIX          #endif // POSIX
1431            Mode = stream_mode_read;
1432          ulStartPos = RIFF_HEADER_SIZE;          ulStartPos = RIFF_HEADER_SIZE;
1433          ReadHeader(0);          ReadHeader(0);
1434          if (ChunkID != CHUNK_ID_RIFF) {          if (ChunkID != CHUNK_ID_RIFF) {
# Line 1308  namespace RIFF { Line 1465  namespace RIFF {
1465                          hFileRead = hFileWrite = 0;                          hFileRead = hFileWrite = 0;
1466                          throw Exception("Could not (re)open file \"" + Filename + "\" in read mode");                          throw Exception("Could not (re)open file \"" + Filename + "\" in read mode");
1467                      }                      }
1468                        #elif defined(WIN32)
1469                        if (hFileRead != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1470                        hFileRead = hFileWrite = CreateFile(
1471                                                     Filename.c_str(), GENERIC_READ,
1472                                                     FILE_SHARE_READ | FILE_SHARE_WRITE,
1473                                                     NULL, OPEN_EXISTING,
1474                                                     FILE_ATTRIBUTE_NORMAL |
1475                                                     FILE_FLAG_RANDOM_ACCESS,
1476                                                     NULL
1477                                                 );
1478                        if (hFileRead == INVALID_HANDLE_VALUE) {
1479                            hFileRead = hFileWrite = INVALID_HANDLE_VALUE;
1480                            throw Exception("Could not (re)open file \"" + Filename + "\" in read mode");
1481                        }
1482                      #else                      #else
1483                      if (hFileRead) fclose(hFileRead);                      if (hFileRead) fclose(hFileRead);
1484                      hFileRead = hFileWrite = fopen(path.c_str(), "rb");                      hFileRead = hFileWrite = fopen(Filename.c_str(), "rb");
1485                      if (!hFileRead) throw Exception("Could not (re)open file \"" + Filename + "\" in read mode");                      if (!hFileRead) throw Exception("Could not (re)open file \"" + Filename + "\" in read mode");
1486                      #endif                      #endif
1487                      __resetPos(); // reset read/write position of ALL 'Chunk' objects                      __resetPos(); // reset read/write position of ALL 'Chunk' objects
1488                      return true;                      break;
1489                  case stream_mode_read_write:                  case stream_mode_read_write:
1490                      #if POSIX                      #if POSIX
1491                      if (hFileRead) close(hFileRead);                      if (hFileRead) close(hFileRead);
# Line 1323  namespace RIFF { Line 1494  namespace RIFF {
1494                          hFileRead = hFileWrite = open(Filename.c_str(), O_RDONLY | O_NONBLOCK);                          hFileRead = hFileWrite = open(Filename.c_str(), O_RDONLY | O_NONBLOCK);
1495                          throw Exception("Could not open file \"" + Filename + "\" in read+write mode");                          throw Exception("Could not open file \"" + Filename + "\" in read+write mode");
1496                      }                      }
1497                        #elif defined(WIN32)
1498                        if (hFileRead != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1499                        hFileRead = hFileWrite = CreateFile(
1500                                                     Filename.c_str(),
1501                                                     GENERIC_READ | GENERIC_WRITE,
1502                                                     FILE_SHARE_READ,
1503                                                     NULL, OPEN_ALWAYS,
1504                                                     FILE_ATTRIBUTE_NORMAL |
1505                                                     FILE_FLAG_RANDOM_ACCESS,
1506                                                     NULL
1507                                                 );
1508                        if (hFileRead == INVALID_HANDLE_VALUE) {
1509                            hFileRead = hFileWrite = CreateFile(
1510                                                         Filename.c_str(), GENERIC_READ,
1511                                                         FILE_SHARE_READ | FILE_SHARE_WRITE,
1512                                                         NULL, OPEN_EXISTING,
1513                                                         FILE_ATTRIBUTE_NORMAL |
1514                                                         FILE_FLAG_RANDOM_ACCESS,
1515                                                         NULL
1516                                                     );
1517                            throw Exception("Could not (re)open file \"" + Filename + "\" in read+write mode");
1518                        }
1519                      #else                      #else
1520                      if (hFileRead) fclose(hFileRead);                      if (hFileRead) fclose(hFileRead);
1521                      hFileRead = hFileWrite = fopen(path.c_str(), "r+b");                      hFileRead = hFileWrite = fopen(Filename.c_str(), "r+b");
1522                      if (!hFileRead) {                      if (!hFileRead) {
1523                          hFileRead = hFileWrite = fopen(path.c_str(), "rb");                          hFileRead = hFileWrite = fopen(Filename.c_str(), "rb");
1524                          throw Exception("Could not open file \"" + Filename + "\" in read+write mode");                          throw Exception("Could not open file \"" + Filename + "\" in read+write mode");
1525                      }                      }
1526                      #endif                      #endif
1527                      __resetPos(); // reset read/write position of ALL 'Chunk' objects                      __resetPos(); // reset read/write position of ALL 'Chunk' objects
1528                      return true;                      break;
1529                    case stream_mode_closed:
1530                        #if POSIX
1531                        if (hFileRead)  close(hFileRead);
1532                        if (hFileWrite) close(hFileWrite);
1533                        #elif defined(WIN32)
1534                        if (hFileRead  != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1535                        if (hFileWrite != INVALID_HANDLE_VALUE) CloseHandle(hFileWrite);
1536                        #else
1537                        if (hFileRead)  fclose(hFileRead);
1538                        if (hFileWrite) fclose(hFileWrite);
1539                        #endif
1540                        hFileRead = hFileWrite = 0;
1541                        break;
1542                  default:                  default:
1543                      throw Exception("Unknown file access mode");                      throw Exception("Unknown file access mode");
1544              }              }
1545                Mode = NewMode;
1546                return true;
1547          }          }
1548          return false;          return false;
1549      }      }
1550    
1551        /** @brief Set the byte order to be used when saving.
1552         *
1553         * Set the byte order to be used in the file. A value of
1554         * endian_little will create a RIFF file, endian_big a RIFX file
1555         * and endian_native will create a RIFF file on little-endian
1556         * machines and RIFX on big-endian machines.
1557         *
1558         * @param Endian - endianess to use when file is saved.
1559         */
1560        void File::SetByteOrder(endian_t Endian) {
1561            #if WORDS_BIGENDIAN
1562            bEndianNative = Endian != endian_little;
1563            #else
1564            bEndianNative = Endian != endian_big;
1565            #endif
1566        }
1567    
1568      /** @brief Save changes to same file.      /** @brief Save changes to same file.
1569       *       *
1570       * Make all changes of all chunks persistent by writing them to the       * Make all changes of all chunks persistent by writing them to the
# Line 1351  namespace RIFF { Line 1576  namespace RIFF {
1576       *                         chunk or any kind of IO error occured       *                         chunk or any kind of IO error occured
1577       */       */
1578      void File::Save() {      void File::Save() {
1579            // make sure the RIFF tree is built (from the original file)
1580            LoadSubChunksRecursively();
1581    
1582          // reopen file in write mode          // reopen file in write mode
1583          bool bModeWasChanged = SetMode(stream_mode_read_write);          SetMode(stream_mode_read_write);
1584    
1585          // to be able to save the whole file without loading everything into          // to be able to save the whole file without loading everything into
1586          // RAM and without having to store the data in a temporary file, we          // RAM and without having to store the data in a temporary file, we
# Line 1363  namespace RIFF { Line 1591  namespace RIFF {
1591    
1592          // first we sum up all positive chunk size changes (and skip all negative ones)          // first we sum up all positive chunk size changes (and skip all negative ones)
1593          unsigned long ulPositiveSizeDiff = 0;          unsigned long ulPositiveSizeDiff = 0;
1594          for (ChunkList::iterator iter = ResizedChunks.begin(), end = ResizedChunks.end(); iter != end; ++iter) {          for (std::set<Chunk*>::const_iterator iter = ResizedChunks.begin(), end = ResizedChunks.end(); iter != end; ++iter) {
1595              if ((*iter)->GetNewSize() == 0) throw Exception("There is at least one empty chunk (zero size)");              if ((*iter)->GetNewSize() == 0) {
1596              unsigned long ulDiff = (*iter)->GetNewSize() - (*iter)->GetSize() + 1L; // +1 in case we have to add a pad byte                  throw Exception("There is at least one empty chunk (zero size): " + __resolveChunkPath(*iter));
1597                }
1598                unsigned long ulDiff = (*iter)->GetNewSize() + (*iter)->GetNewSize() % 2 - (*iter)->GetSize() - (*iter)->GetSize() % 2;
1599              if (ulDiff > 0) ulPositiveSizeDiff += ulDiff;              if (ulDiff > 0) ulPositiveSizeDiff += ulDiff;
1600          }          }
1601    
# Line 1379  namespace RIFF { Line 1609  namespace RIFF {
1609              // ... and move current data by the same amount towards end of file.              // ... and move current data by the same amount towards end of file.
1610              int8_t* pCopyBuffer = new int8_t[4096];              int8_t* pCopyBuffer = new int8_t[4096];
1611              const unsigned long ulFileSize = GetSize() + RIFF_HEADER_SIZE;              const unsigned long ulFileSize = GetSize() + RIFF_HEADER_SIZE;
1612                #if defined(WIN32)
1613                DWORD iBytesMoved = 1; // we have to pass it via pointer to the Windows API, thus the correct size must be ensured
1614                #else
1615              int iBytesMoved = 1;              int iBytesMoved = 1;
1616              for (unsigned long ulPos = 0; iBytesMoved > 0; ulPos += iBytesMoved) {              #endif
1617                  const unsigned long ulToMove = ulFileSize - ulPos;              for (unsigned long ulPos = ulFileSize; iBytesMoved > 0; ) {
1618                  iBytesMoved = (ulToMove < 4096) ? ulToMove : 4096;                  iBytesMoved = (ulPos < 4096) ? ulPos : 4096;
1619                    ulPos -= iBytesMoved;
1620                  #if POSIX                  #if POSIX
1621                  lseek(hFileRead, ulPos, SEEK_SET);                  lseek(hFileRead, ulPos, SEEK_SET);
1622                  iBytesMoved = read(hFileRead, pCopyBuffer, iBytesMoved);                  iBytesMoved = read(hFileRead, pCopyBuffer, iBytesMoved);
1623                  lseek(hFileWrite, ulPos + ulPositiveSizeDiff, SEEK_SET);                  lseek(hFileWrite, ulPos + ulPositiveSizeDiff, SEEK_SET);
1624                  iBytesMoved = write(hFileWrite, pCopyBuffer, iBytesMoved);                  iBytesMoved = write(hFileWrite, pCopyBuffer, iBytesMoved);
1625                    #elif defined(WIN32)
1626                    SetFilePointer(hFileRead, ulPos, NULL/*32 bit*/, FILE_BEGIN);
1627                    ReadFile(hFileRead, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
1628                    SetFilePointer(hFileWrite, ulPos + ulPositiveSizeDiff, NULL/*32 bit*/, FILE_BEGIN);
1629                    WriteFile(hFileWrite, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
1630                  #else                  #else
1631                  fseek(hFileRead, ulPos, SEEK_SET);                  fseek(hFileRead, ulPos, SEEK_SET);
1632                  iBytesMoved = fread(pCopyBuffer, 1, iBytesMoved, hFileRead);                  iBytesMoved = fread(pCopyBuffer, 1, iBytesMoved, hFileRead);
# Line 1400  namespace RIFF { Line 1639  namespace RIFF {
1639          }          }
1640    
1641          // rebuild / rewrite complete RIFF tree          // rebuild / rewrite complete RIFF tree
1642          unsigned long ulTotalSize = WriteChunk(0, ulPositiveSizeDiff);          unsigned long ulTotalSize  = WriteChunk(0, ulPositiveSizeDiff);
1643            unsigned long ulActualSize = __GetFileSize(hFileWrite);
1644    
1645          // resize file to the final size          // resize file to the final size
1646          if (ulTotalSize < ulWorkingFileSize) ResizeFile(ulTotalSize);          if (ulTotalSize < ulActualSize) ResizeFile(ulTotalSize);
1647    
1648          // forget all resized chunks          // forget all resized chunks
1649          ResizedChunks.clear();          ResizedChunks.clear();
   
         // reopen file in read mode  
         if (bModeWasChanged) SetMode(stream_mode_read);  
1650      }      }
1651    
1652      /** @brief Save changes to another file.      /** @brief Save changes to another file.
# Line 1417  namespace RIFF { Line 1654  namespace RIFF {
1654       * Make all changes of all chunks persistent by writing them to another       * Make all changes of all chunks persistent by writing them to another
1655       * file. <b>Caution:</b> this method is optimized for writing to       * file. <b>Caution:</b> this method is optimized for writing to
1656       * <b>another</b> file, do not use it to save the changes to the same       * <b>another</b> file, do not use it to save the changes to the same
1657       * file! Use File::Save() in that case instead!       * file! Use File::Save() in that case instead! Ignoring this might
1658         * result in a corrupted file, especially in case chunks were resized!
1659       *       *
1660       * After calling this method, this File object will be associated with       * After calling this method, this File object will be associated with
1661       * the new file (given by \a path) afterwards.       * the new file (given by \a path) afterwards.
# Line 1425  namespace RIFF { Line 1663  namespace RIFF {
1663       * @param path - path and file name where everything should be written to       * @param path - path and file name where everything should be written to
1664       */       */
1665      void File::Save(const String& path) {      void File::Save(const String& path) {
1666            //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
1667    
1668            // make sure the RIFF tree is built (from the original file)
1669            LoadSubChunksRecursively();
1670    
1671            if (Filename.length() > 0) SetMode(stream_mode_read);
1672          // open the other (new) file for writing and truncate it to zero size          // open the other (new) file for writing and truncate it to zero size
1673          #if POSIX          #if POSIX
1674          hFileWrite = open(path.c_str(), O_RDWR | O_CREAT | O_TRUNC);          hFileWrite = open(path.c_str(), O_RDWR | O_CREAT, S_IRUSR | S_IWUSR | S_IRGRP);
1675          if (hFileWrite < 0) {          if (hFileWrite < 0) {
1676              hFileWrite = hFileRead;              hFileWrite = hFileRead;
1677              throw Exception("Could not open file \"" + path + "\" for writing");              throw Exception("Could not open file \"" + path + "\" for writing");
1678          }          }
1679            #elif defined(WIN32)
1680            hFileWrite = CreateFile(
1681                             path.c_str(), GENERIC_WRITE, FILE_SHARE_READ,
1682                             NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL |
1683                             FILE_FLAG_RANDOM_ACCESS, NULL
1684                         );
1685            if (hFileWrite == INVALID_HANDLE_VALUE) {
1686                hFileWrite = hFileRead;
1687                throw Exception("Could not open file \"" + path + "\" for writing");
1688            }
1689          #else          #else
1690          hFileWrite = fopen(path.c_str(), "w+b");          hFileWrite = fopen(path.c_str(), "w+b");
1691          if (!hFileWrite) {          if (!hFileWrite) {
# Line 1439  namespace RIFF { Line 1693  namespace RIFF {
1693              throw Exception("Could not open file \"" + path + "\" for writing");              throw Exception("Could not open file \"" + path + "\" for writing");
1694          }          }
1695          #endif // POSIX          #endif // POSIX
1696            Mode = stream_mode_read_write;
1697    
1698          // write complete RIFF tree to the other (new) file          // write complete RIFF tree to the other (new) file
1699          WriteChunk(0, 0);          unsigned long ulTotalSize  = WriteChunk(0, 0);
1700            unsigned long ulActualSize = __GetFileSize(hFileWrite);
1701    
1702            // resize file to the final size (if the file was originally larger)
1703            if (ulTotalSize < ulActualSize) ResizeFile(ulTotalSize);
1704    
1705          // forget all resized chunks          // forget all resized chunks
1706          ResizedChunks.clear();          ResizedChunks.clear();
1707    
1708            #if POSIX
1709            if (hFileWrite) close(hFileWrite);
1710            #elif defined(WIN32)
1711            if (hFileWrite != INVALID_HANDLE_VALUE) CloseHandle(hFileWrite);
1712            #else
1713            if (hFileWrite) fclose(hFileWrite);
1714            #endif
1715            hFileWrite = hFileRead;
1716    
1717          // associate new file with this File object from now on          // associate new file with this File object from now on
1718          Filename = path;          Filename = path;
1719          stream_mode_t oldMode = Mode;          Mode = (stream_mode_t) -1;       // Just set it to an undefined mode ...
1720          Mode = (stream_mode_t) -1; // Just set it to an undefined mode ...          SetMode(stream_mode_read_write); // ... so SetMode() has to reopen the file handles.
         SetMode(oldMode);          // ... so SetMode() has to reopen the file handles.  
1721      }      }
1722    
1723      void File::ResizeFile(unsigned long ulNewSize) {      void File::ResizeFile(unsigned long ulNewSize) {
1724          #if POSIX          #if POSIX
1725          if (ftruncate(hFileWrite, ulNewSize) < 0)          if (ftruncate(hFileWrite, ulNewSize) < 0)
1726              throw Exception("Could not resize file \"" + Filename + "\"");              throw Exception("Could not resize file \"" + Filename + "\"");
1727            #elif defined(WIN32)
1728            if (
1729                SetFilePointer(hFileWrite, ulNewSize, NULL/*32 bit*/, FILE_BEGIN) == INVALID_SET_FILE_POINTER ||
1730                !SetEndOfFile(hFileWrite)
1731            ) throw Exception("Could not resize file \"" + Filename + "\"");
1732          #else          #else
1733          # error Sorry, this version of libgig only supports POSIX systems yet.          # error Sorry, this version of libgig only supports POSIX and Windows systems yet.
1734          # error Reason: portable implementation of RIFF::File::ResizeFile() is missing!          # error Reason: portable implementation of RIFF::File::ResizeFile() is missing (yet)!
1735          #endif          #endif
1736      }      }
1737    
# Line 1469  namespace RIFF { Line 1741  namespace RIFF {
1741         #endif // DEBUG         #endif // DEBUG
1742          #if POSIX          #if POSIX
1743          if (hFileRead) close(hFileRead);          if (hFileRead) close(hFileRead);
1744            #elif defined(WIN32)
1745            if (hFileRead != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1746          #else          #else
1747          if (hFileRead) fclose(hFileRead);          if (hFileRead) fclose(hFileRead);
1748          #endif // POSIX          #endif // POSIX
1749      }      }
1750    
1751      void File::LogAsResized(Chunk* pResizedChunk) {      void File::LogAsResized(Chunk* pResizedChunk) {
1752          ResizedChunks.push_back(pResizedChunk);          ResizedChunks.insert(pResizedChunk);
1753        }
1754    
1755        void File::UnlogResized(Chunk* pResizedChunk) {
1756            ResizedChunks.erase(pResizedChunk);
1757      }      }
1758    
1759      unsigned long File::GetFileSize() {      unsigned long File::GetFileSize() {
1760          #if POSIX          return __GetFileSize(hFileRead);
1761        }
1762    
1763        #if POSIX
1764        unsigned long File::__GetFileSize(int hFile) {
1765          struct stat filestat;          struct stat filestat;
1766          fstat(hFileRead, &filestat);          fstat(hFile, &filestat);
1767          long size = filestat.st_size;          long size = filestat.st_size;
         #else // standard C functions  
         long curpos = ftell(hFileRead);  
         fseek(hFileRead, 0, SEEK_END);  
         long size = ftell(hFileRead);  
         fseek(hFileRead, curpos, SEEK_SET);  
         #endif // POSIX  
1768          return size;          return size;
1769      }      }
1770        #elif defined(WIN32)
1771        unsigned long File::__GetFileSize(HANDLE hFile) {
1772            DWORD dwSize = ::GetFileSize(hFile, NULL /*32bit*/);
1773            if (dwSize == INVALID_FILE_SIZE)
1774                throw Exception("Windows FS error: could not determine file size");
1775            return dwSize;
1776        }
1777        #else // standard C functions
1778        unsigned long File::__GetFileSize(FILE* hFile) {
1779            long curpos = ftell(hFile);
1780            fseek(hFile, 0, SEEK_END);
1781            long size = ftell(hFile);
1782            fseek(hFile, curpos, SEEK_SET);
1783            return size;
1784        }
1785        #endif
1786    
1787    
1788  // *************** Exception ***************  // *************** Exception ***************
Legend:
Removed from v.780  
changed lines
  Added in v.1713
  ViewVC Help
Powered by ViewVC