/[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 798 by schoenebeck, Thu Nov 3 23:49:11 2005 UTC revision 1301 by persson, Sat Aug 25 09:59:53 2007 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 22  Line 22 
22   ***************************************************************************/   ***************************************************************************/
23    
24  #include <string.h>  #include <string.h>
 #include <sstream>  
25    
26  #include "RIFF.h"  #include "RIFF.h"
27    
28    #include "helper.h"
29    
30  namespace RIFF {  namespace RIFF {
31    
32  // *************** Helper Functions **************  // *************** Internal functions **************
33  // *  // *
34    
35      template<class T> inline String ToString(T o) {      /// Returns a human readable path of the given chunk.
36          std::stringstream ss;      String __resolveChunkPath(Chunk* pCk) {
37          ss << o;          String sPath;
38          return ss.str();          for (Chunk* pChunk = pCk; pChunk; pChunk = pChunk->GetParent()) {
39                if (pChunk->GetChunkID() == CHUNK_ID_LIST) {
40                    List* pList = (List*) pChunk;
41                    sPath = "->'" + pList->GetListTypeString() + "'" + sPath;
42                } else {
43                    sPath = "->'" + pChunk->GetChunkIDString() + "'" + sPath;
44                }
45            }
46            return sPath;
47      }      }
48    
49    
# Line 49  namespace RIFF { Line 58  namespace RIFF {
58          ulPos      = 0;          ulPos      = 0;
59          pParent    = NULL;          pParent    = NULL;
60          pChunkData = NULL;          pChunkData = NULL;
61            ulChunkDataSize = 0;
62          ChunkID    = CHUNK_ID_RIFF;          ChunkID    = CHUNK_ID_RIFF;
63          this->pFile = pFile;          this->pFile = pFile;
64      }      }
# Line 62  namespace RIFF { Line 72  namespace RIFF {
72          pParent       = Parent;          pParent       = Parent;
73          ulPos         = 0;          ulPos         = 0;
74          pChunkData    = NULL;          pChunkData    = NULL;
75            ulChunkDataSize = 0;
76          ReadHeader(StartPos);          ReadHeader(StartPos);
77      }      }
78    
# Line 71  namespace RIFF { Line 82  namespace RIFF {
82          this->pParent    = pParent;          this->pParent    = pParent;
83          ulPos            = 0;          ulPos            = 0;
84          pChunkData       = NULL;          pChunkData       = NULL;
85            ulChunkDataSize  = 0;
86          ChunkID          = uiChunkID;          ChunkID          = uiChunkID;
87          CurrentChunkSize = 0;          CurrentChunkSize = 0;
88          NewChunkSize     = uiBodySize;          NewChunkSize     = uiBodySize;
89      }      }
90    
91      Chunk::~Chunk() {      Chunk::~Chunk() {
92            if (CurrentChunkSize != NewChunkSize) pFile->UnlogResized(this);
93          if (pChunkData) delete[] pChunkData;          if (pChunkData) delete[] pChunkData;
94      }      }
95    
# Line 88  namespace RIFF { Line 101  namespace RIFF {
101          if (lseek(pFile->hFileRead, fPos, SEEK_SET) != -1) {          if (lseek(pFile->hFileRead, fPos, SEEK_SET) != -1) {
102              read(pFile->hFileRead, &ChunkID, 4);              read(pFile->hFileRead, &ChunkID, 4);
103              read(pFile->hFileRead, &CurrentChunkSize, 4);              read(pFile->hFileRead, &CurrentChunkSize, 4);
104            #elif defined(WIN32)
105            if (SetFilePointer(pFile->hFileRead, fPos, NULL/*32 bit*/, FILE_BEGIN) != INVALID_SET_FILE_POINTER) {
106                DWORD dwBytesRead;
107                ReadFile(pFile->hFileRead, &ChunkID, 4, &dwBytesRead, NULL);
108                ReadFile(pFile->hFileRead, &CurrentChunkSize, 4, &dwBytesRead, NULL);
109          #else          #else
110          if (!fseek(pFile->hFileRead, fPos, SEEK_SET)) {          if (!fseek(pFile->hFileRead, fPos, SEEK_SET)) {
111              fread(&ChunkID, 4, 1, pFile->hFileRead);              fread(&ChunkID, 4, 1, pFile->hFileRead);
# Line 95  namespace RIFF { Line 113  namespace RIFF {
113          #endif // POSIX          #endif // POSIX
114              #if WORDS_BIGENDIAN              #if WORDS_BIGENDIAN
115              if (ChunkID == CHUNK_ID_RIFF) {              if (ChunkID == CHUNK_ID_RIFF) {
116                  bEndianNative = false;                  pFile->bEndianNative = false;
117              }              }
118              #else // little endian              #else // little endian
119              if (ChunkID == CHUNK_ID_RIFX) {              if (ChunkID == CHUNK_ID_RIFX) {
# Line 136  namespace RIFF { Line 154  namespace RIFF {
154              write(pFile->hFileWrite, &uiNewChunkID, 4);              write(pFile->hFileWrite, &uiNewChunkID, 4);
155              write(pFile->hFileWrite, &uiNewChunkSize, 4);              write(pFile->hFileWrite, &uiNewChunkSize, 4);
156          }          }
157            #elif defined(WIN32)
158            if (SetFilePointer(pFile->hFileWrite, fPos, NULL/*32 bit*/, FILE_BEGIN) != INVALID_SET_FILE_POINTER) {
159                DWORD dwBytesWritten;
160                WriteFile(pFile->hFileWrite, &uiNewChunkID, 4, &dwBytesWritten, NULL);
161                WriteFile(pFile->hFileWrite, &uiNewChunkSize, 4, &dwBytesWritten, NULL);
162            }
163          #else          #else
164          if (!fseek(pFile->hFileWrite, fPos, SEEK_SET)) {          if (!fseek(pFile->hFileWrite, fPos, SEEK_SET)) {
165              fwrite(&uiNewChunkID, 4, 1, pFile->hFileWrite);              fwrite(&uiNewChunkID, 4, 1, pFile->hFileWrite);
# Line 219  namespace RIFF { Line 243  namespace RIFF {
243        std::cout << "Chunk::GetState()" << std::endl;        std::cout << "Chunk::GetState()" << std::endl;
244        #endif // DEBUG        #endif // DEBUG
245          #if POSIX          #if POSIX
246          if (pFile->hFileRead == 0)    return stream_closed;          if (pFile->hFileRead == 0) return stream_closed;
247            #elif defined (WIN32)
248            if (pFile->hFileRead == INVALID_HANDLE_VALUE)
249                return stream_closed;
250          #else          #else
251          if (pFile->hFileRead == NULL) return stream_closed;          if (pFile->hFileRead == NULL) return stream_closed;
252          #endif // POSIX          #endif // POSIX
# Line 253  namespace RIFF { Line 280  namespace RIFF {
280          unsigned long readWords = read(pFile->hFileRead, pData, WordCount * WordSize);          unsigned long readWords = read(pFile->hFileRead, pData, WordCount * WordSize);
281          if (readWords < 1) return 0;          if (readWords < 1) return 0;
282          readWords /= WordSize;          readWords /= WordSize;
283            #elif defined(WIN32)
284            if (SetFilePointer(pFile->hFileRead, ulStartPos + ulPos, NULL/*32 bit*/, FILE_BEGIN) == INVALID_SET_FILE_POINTER) return 0;
285            DWORD readWords;
286            ReadFile(pFile->hFileRead, pData, WordCount * WordSize, &readWords, NULL);
287            if (readWords < 1) return 0;
288            readWords /= WordSize;
289          #else // standard C functions          #else // standard C functions
290          if (fseek(pFile->hFileRead, ulStartPos + ulPos, SEEK_SET)) return 0;          if (fseek(pFile->hFileRead, ulStartPos + ulPos, SEEK_SET)) return 0;
291          unsigned long readWords = fread(pData, WordSize, WordCount, pFile->hFileRead);          unsigned long readWords = fread(pData, WordSize, WordCount, pFile->hFileRead);
# Line 322  namespace RIFF { Line 355  namespace RIFF {
355          unsigned long writtenWords = write(pFile->hFileWrite, pData, WordCount * WordSize);          unsigned long writtenWords = write(pFile->hFileWrite, pData, WordCount * WordSize);
356          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");
357          writtenWords /= WordSize;          writtenWords /= WordSize;
358            #elif defined(WIN32)
359            if (SetFilePointer(pFile->hFileWrite, ulStartPos + ulPos, NULL/*32 bit*/, FILE_BEGIN) == INVALID_SET_FILE_POINTER) {
360                throw Exception("Could not seek to position " + ToString(ulPos) +
361                                " in chunk (" + ToString(ulStartPos + ulPos) + " in file)");            
362            }
363            DWORD writtenWords;
364            WriteFile(pFile->hFileWrite, pData, WordCount * WordSize, &writtenWords, NULL);
365            if (writtenWords < 1) throw Exception("Windows IO Error while trying to write chunk data");
366            writtenWords /= WordSize;
367          #else // standard C functions          #else // standard C functions
368          if (fseek(pFile->hFileWrite, ulStartPos + ulPos, SEEK_SET)) {          if (fseek(pFile->hFileWrite, ulStartPos + ulPos, SEEK_SET)) {
369              throw Exception("Could not seek to position " + ToString(ulPos) +              throw Exception("Could not seek to position " + ToString(ulPos) +
# Line 668  namespace RIFF { Line 710  namespace RIFF {
710       *       *
711       * <b>Caution:</b> the buffer pointer will be invalidated once       * <b>Caution:</b> the buffer pointer will be invalidated once
712       * File::Save() was called. You have to call LoadChunkData() again to       * File::Save() was called. You have to call LoadChunkData() again to
713       * get a new pointer whenever File::Save() was called.       * get a new, valid pointer whenever File::Save() was called.
714         *
715         * You can call LoadChunkData() again if you previously scheduled to
716         * enlarge this chunk with a Resize() call. In that case the buffer will
717         * be enlarged to the new, scheduled chunk size and you can already
718         * place the new chunk data to the buffer and finally call File::Save()
719         * to enlarge the chunk physically and write the new data in one rush.
720         * This approach is definitely recommended if you have to enlarge and
721         * write new data to a lot of chunks.
722       *       *
723       * @returns a pointer to the data in RAM on success, NULL otherwise       * @returns a pointer to the data in RAM on success, NULL otherwise
724         * @throws Exception if data buffer could not be enlarged
725       * @see ReleaseChunkData()       * @see ReleaseChunkData()
726       */       */
727      void* Chunk::LoadChunkData() {      void* Chunk::LoadChunkData() {
728          if (!pChunkData) {          if (!pChunkData && pFile->Filename != "") {
729              #if POSIX              #if POSIX
730              if (lseek(pFile->hFileRead, ulStartPos, SEEK_SET) == -1) return NULL;              if (lseek(pFile->hFileRead, ulStartPos, SEEK_SET) == -1) return NULL;
731              pChunkData = new uint8_t[GetSize()];              #elif defined(WIN32)
732              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());  
733              #else              #else
734              if (fseek(pFile->hFileRead, ulStartPos, SEEK_SET)) return NULL;              if (fseek(pFile->hFileRead, ulStartPos, SEEK_SET)) return NULL;
735              pChunkData = new uint8_t[GetSize()];              #endif // POSIX
736                unsigned long ulBufferSize = (CurrentChunkSize > NewChunkSize) ? CurrentChunkSize : NewChunkSize;
737                pChunkData = new uint8_t[ulBufferSize];
738              if (!pChunkData) return NULL;              if (!pChunkData) return NULL;
739                memset(pChunkData, 0, ulBufferSize);
740                #if POSIX
741                unsigned long readWords = read(pFile->hFileRead, pChunkData, GetSize());
742                #elif defined(WIN32)
743                DWORD readWords;
744                ReadFile(pFile->hFileRead, pChunkData, GetSize(), &readWords, NULL);
745                #else
746              unsigned long readWords = fread(pChunkData, 1, GetSize(), pFile->hFileRead);              unsigned long readWords = fread(pChunkData, 1, GetSize(), pFile->hFileRead);
747              #endif // POSIX              #endif // POSIX
748              if (readWords != GetSize()) {              if (readWords != GetSize()) {
749                  delete[] pChunkData;                  delete[] pChunkData;
750                  return (pChunkData = NULL);                  return (pChunkData = NULL);
751              }              }
752                ulChunkDataSize = ulBufferSize;
753            } else if (NewChunkSize > ulChunkDataSize) {
754                uint8_t* pNewBuffer = new uint8_t[NewChunkSize];
755                if (!pNewBuffer) throw Exception("Could not enlarge chunk data buffer to " + ToString(NewChunkSize) + " bytes");
756                memset(pNewBuffer, 0 , NewChunkSize);
757                memcpy(pNewBuffer, pChunkData, ulChunkDataSize);
758                delete[] pChunkData;
759                pChunkData      = pNewBuffer;
760                ulChunkDataSize = NewChunkSize;
761          }          }
762          return pChunkData;          return pChunkData;
763      }      }
# Line 719  namespace RIFF { Line 787  namespace RIFF {
787       *       *
788       * <b>Caution:</b> You cannot directly write to enlarged chunks before       * <b>Caution:</b> You cannot directly write to enlarged chunks before
789       * calling File::Save() as this might exceed the current chunk's body       * calling File::Save() as this might exceed the current chunk's body
790       * boundary.       * boundary!
791       *       *
792       * @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)
793       * @throws RIFF::Exception  if \a iNewSize is less than 1       * @throws RIFF::Exception  if \a iNewSize is less than 1
794       * @see File::Save()       * @see File::Save()
795       */       */
796      void Chunk::Resize(int iNewSize) {      void Chunk::Resize(int iNewSize) {
797          if (iNewSize <= 0) throw Exception("Chunk size must be at least one byte");          if (iNewSize <= 0)
798                throw Exception("There is at least one empty chunk (zero size): " + __resolveChunkPath(this));
799            if (NewChunkSize == iNewSize) return;
800          NewChunkSize = iNewSize;          NewChunkSize = iNewSize;
801          pFile->LogAsResized(this);          pFile->LogAsResized(this);
802      }      }
# Line 752  namespace RIFF { Line 822  namespace RIFF {
822    
823          // if the whole chunk body was loaded into RAM          // if the whole chunk body was loaded into RAM
824          if (pChunkData) {          if (pChunkData) {
825              // 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
826              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;  
             }  
   
827              // write chunk data from RAM persistently to the file              // write chunk data from RAM persistently to the file
828              #if POSIX              #if POSIX
829              lseek(pFile->hFileWrite, ulWritePos, SEEK_SET);              lseek(pFile->hFileWrite, ulWritePos, SEEK_SET);
830              if (write(pFile->hFileWrite, pChunkData, NewChunkSize) != NewChunkSize) {              if (write(pFile->hFileWrite, pChunkData, NewChunkSize) != NewChunkSize) {
831                  throw Exception("Writing Chunk data (from RAM) failed");                  throw Exception("Writing Chunk data (from RAM) failed");
832              }              }
833                #elif defined(WIN32)
834                SetFilePointer(pFile->hFileWrite, ulWritePos, NULL/*32 bit*/, FILE_BEGIN);
835                DWORD dwBytesWritten;
836                WriteFile(pFile->hFileWrite, pChunkData, NewChunkSize, &dwBytesWritten, NULL);
837                if (dwBytesWritten != NewChunkSize) {
838                    throw Exception("Writing Chunk data (from RAM) failed");
839                }
840              #else              #else
841              fseek(pFile->hFileWrite, ulWritePos, SEEK_SET);              fseek(pFile->hFileWrite, ulWritePos, SEEK_SET);
842              if (fwrite(pChunkData, 1, NewChunkSize, pFile->hFileWrite) != NewChunkSize) {              if (fwrite(pChunkData, 1, NewChunkSize, pFile->hFileWrite) != NewChunkSize) {
# Line 781  namespace RIFF { Line 847  namespace RIFF {
847              // 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
848              int8_t* pCopyBuffer = new int8_t[4096];              int8_t* pCopyBuffer = new int8_t[4096];
849              unsigned long ulToMove = (NewChunkSize < CurrentChunkSize) ? NewChunkSize : CurrentChunkSize;              unsigned long ulToMove = (NewChunkSize < CurrentChunkSize) ? NewChunkSize : CurrentChunkSize;
850                #if defined(WIN32)
851                DWORD iBytesMoved = 1; // we have to pass it via pointer to the Windows API, thus the correct size must be ensured
852                #else
853              int iBytesMoved = 1;              int iBytesMoved = 1;
854                #endif
855              for (unsigned long ulOffset = 0; iBytesMoved > 0; ulOffset += iBytesMoved, ulToMove -= iBytesMoved) {              for (unsigned long ulOffset = 0; iBytesMoved > 0; ulOffset += iBytesMoved, ulToMove -= iBytesMoved) {
856                  iBytesMoved = (ulToMove < 4096) ? ulToMove : 4096;                  iBytesMoved = (ulToMove < 4096) ? ulToMove : 4096;
857                  #if POSIX                  #if POSIX
# Line 789  namespace RIFF { Line 859  namespace RIFF {
859                  iBytesMoved = read(pFile->hFileRead, pCopyBuffer, iBytesMoved);                  iBytesMoved = read(pFile->hFileRead, pCopyBuffer, iBytesMoved);
860                  lseek(pFile->hFileWrite, ulWritePos + ulOffset, SEEK_SET);                  lseek(pFile->hFileWrite, ulWritePos + ulOffset, SEEK_SET);
861                  iBytesMoved = write(pFile->hFileWrite, pCopyBuffer, iBytesMoved);                  iBytesMoved = write(pFile->hFileWrite, pCopyBuffer, iBytesMoved);
862                    #elif defined(WIN32)
863                    SetFilePointer(pFile->hFileRead, ulStartPos + ulCurrentDataOffset + ulOffset, NULL/*32 bit*/, FILE_BEGIN);
864                    ReadFile(pFile->hFileRead, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
865                    SetFilePointer(pFile->hFileWrite, ulWritePos + ulOffset, NULL/*32 bit*/, FILE_BEGIN);
866                    WriteFile(pFile->hFileWrite, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
867                  #else                  #else
868                  fseek(pFile->hFileRead, ulStartPos + ulCurrentDataOffset + ulOffset, SEEK_SET);                  fseek(pFile->hFileRead, ulStartPos + ulCurrentDataOffset + ulOffset, SEEK_SET);
869                  iBytesMoved = fread(pCopyBuffer, 1, iBytesMoved, pFile->hFileRead);                  iBytesMoved = fread(pCopyBuffer, 1, iBytesMoved, pFile->hFileRead);
# Line 814  namespace RIFF { Line 889  namespace RIFF {
889              #if POSIX              #if POSIX
890              lseek(pFile->hFileWrite, ulStartPos + NewChunkSize, SEEK_SET);              lseek(pFile->hFileWrite, ulStartPos + NewChunkSize, SEEK_SET);
891              write(pFile->hFileWrite, &cPadByte, 1);              write(pFile->hFileWrite, &cPadByte, 1);
892                #elif defined(WIN32)
893                SetFilePointer(pFile->hFileWrite, ulStartPos + NewChunkSize, NULL/*32 bit*/, FILE_BEGIN);
894                DWORD dwBytesWritten;
895                WriteFile(pFile->hFileWrite, &cPadByte, 1, &dwBytesWritten, NULL);
896              #else              #else
897              fseek(pFile->hFileWrite, ulStartPos + NewChunkSize, SEEK_SET);              fseek(pFile->hFileWrite, ulStartPos + NewChunkSize, SEEK_SET);
898              fwrite(&cPadByte, 1, 1, pFile->hFileWrite);              fwrite(&cPadByte, 1, 1, pFile->hFileWrite);
# Line 1076  namespace RIFF { Line 1155  namespace RIFF {
1155          return pNewChunk;          return pNewChunk;
1156      }      }
1157    
1158        /** @brief Moves a sub chunk.
1159         *
1160         * Moves a sub chunk from one position in a list to another
1161         * position in the same list. The pSrc chunk is placed before the
1162         * pDst chunk.
1163         *
1164         * @param pSrc - sub chunk to be moved
1165         * @param pDst - the position to move to. pSrc will be placed
1166         *               before pDst. If pDst is 0, pSrc will be placed
1167         *               last in list.
1168         */
1169        void List::MoveSubChunk(Chunk* pSrc, Chunk* pDst) {
1170            if (!pSubChunks) LoadSubChunks();
1171            pSubChunks->remove(pSrc);
1172            ChunkList::iterator iter = find(pSubChunks->begin(), pSubChunks->end(), pDst);
1173            pSubChunks->insert(iter, pSrc);
1174        }
1175    
1176      /** @brief Creates a new list sub chunk.      /** @brief Creates a new list sub chunk.
1177       *       *
1178       * 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 1097  namespace RIFF { Line 1194  namespace RIFF {
1194       *       *
1195       * 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
1196       * 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
1197       * 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,
1198       * change persistent at any time.       * all its subchunks (if any) will be removed recursively as well. You
1199         * should call File::Save() to make this change persistent at any time.
1200       *       *
1201       * @param pSubChunk - sub chunk or sub list chunk to be removed       * @param pSubChunk - sub chunk or sub list chunk to be removed
1202       */       */
# Line 1129  namespace RIFF { Line 1227  namespace RIFF {
1227          #if POSIX          #if POSIX
1228          lseek(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, SEEK_SET);          lseek(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1229          read(pFile->hFileRead, &ListType, 4);          read(pFile->hFileRead, &ListType, 4);
1230            #elif defined(WIN32)
1231            SetFilePointer(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, NULL/*32 bit*/, FILE_BEGIN);
1232            DWORD dwBytesRead;
1233            ReadFile(pFile->hFileRead, &ListType, 4, &dwBytesRead, NULL);
1234          #else          #else
1235          fseek(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, SEEK_SET);          fseek(pFile->hFileRead, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1236          fread(&ListType, 4, 1, pFile->hFileRead);          fread(&ListType, 4, 1, pFile->hFileRead);
# Line 1149  namespace RIFF { Line 1251  namespace RIFF {
1251          #if POSIX          #if POSIX
1252          lseek(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, SEEK_SET);          lseek(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1253          write(pFile->hFileWrite, &ListType, 4);          write(pFile->hFileWrite, &ListType, 4);
1254            #elif defined(WIN32)
1255            SetFilePointer(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, NULL/*32 bit*/, FILE_BEGIN);
1256            DWORD dwBytesWritten;
1257            WriteFile(pFile->hFileWrite, &ListType, 4, &dwBytesWritten, NULL);
1258          #else          #else
1259          fseek(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, SEEK_SET);          fseek(pFile->hFileWrite, fPos + CHUNK_HEADER_SIZE, SEEK_SET);
1260          fwrite(&ListType, 4, 1, pFile->hFileWrite);          fwrite(&ListType, 4, 1, pFile->hFileWrite);
# Line 1162  namespace RIFF { Line 1268  namespace RIFF {
1268          if (!pSubChunks) {          if (!pSubChunks) {
1269              pSubChunks    = new ChunkList();              pSubChunks    = new ChunkList();
1270              pSubChunksMap = new ChunkMap();              pSubChunksMap = new ChunkMap();
1271                #if defined(WIN32)
1272                if (pFile->hFileRead == INVALID_HANDLE_VALUE) return;
1273                #else
1274              if (!pFile->hFileRead) return;              if (!pFile->hFileRead) return;
1275                #endif
1276              unsigned long uiOriginalPos = GetPos();              unsigned long uiOriginalPos = GetPos();
1277              SetPos(0); // jump to beginning of list chunk body              SetPos(0); // jump to beginning of list chunk body
1278              while (RemainingBytes() >= CHUNK_HEADER_SIZE) {              while (RemainingBytes() >= CHUNK_HEADER_SIZE) {
# Line 1188  namespace RIFF { Line 1298  namespace RIFF {
1298          }          }
1299      }      }
1300    
1301        void List::LoadSubChunksRecursively() {
1302            for (List* pList = GetFirstSubList(); pList; pList = GetNextSubList())
1303                pList->LoadSubChunksRecursively();
1304        }
1305    
1306      /** @brief Write list chunk persistently e.g. to disk.      /** @brief Write list chunk persistently e.g. to disk.
1307       *       *
1308       * Stores the list chunk persistently to its actual "physical" file. All       * Stores the list chunk persistently to its actual "physical" file. All
# Line 1253  namespace RIFF { Line 1368  namespace RIFF {
1368       * "from scratch". Note: there must be no empty chunks or empty list       * "from scratch". Note: there must be no empty chunks or empty list
1369       * chunks when trying to make the new RIFF file persistent with Save()!       * chunks when trying to make the new RIFF file persistent with Save()!
1370       *       *
1371         * Note: by default, the RIFF file will be saved in native endian
1372         * format; that is, as a RIFF file on little-endian machines and
1373         * as a RIFX file on big-endian. To change this behaviour, call
1374         * SetByteOrder() before calling Save().
1375         *
1376       * @param FileType - four-byte identifier of the RIFF file type       * @param FileType - four-byte identifier of the RIFF file type
1377       * @see AddSubChunk(), AddSubList()       * @see AddSubChunk(), AddSubList(), SetByteOrder()
1378       */       */
1379      File::File(uint32_t FileType) : List(this) {      File::File(uint32_t FileType) : List(this) {
1380            #if defined(WIN32)
1381            hFileRead = hFileWrite = INVALID_HANDLE_VALUE;
1382            #else
1383          hFileRead = hFileWrite = 0;          hFileRead = hFileWrite = 0;
1384            #endif
1385          Mode = stream_mode_closed;          Mode = stream_mode_closed;
1386          bEndianNative = true;          bEndianNative = true;
1387          ulStartPos = RIFF_HEADER_SIZE;          ulStartPos = RIFF_HEADER_SIZE;
# Line 1283  namespace RIFF { Line 1407  namespace RIFF {
1407              hFileRead = hFileWrite = 0;              hFileRead = hFileWrite = 0;
1408              throw RIFF::Exception("Can't open \"" + path + "\"");              throw RIFF::Exception("Can't open \"" + path + "\"");
1409          }          }
1410            #elif defined(WIN32)
1411            hFileRead = hFileWrite = CreateFile(
1412                                         path.c_str(), GENERIC_READ,
1413                                         FILE_SHARE_READ | FILE_SHARE_WRITE,
1414                                         NULL, OPEN_EXISTING,
1415                                         FILE_ATTRIBUTE_NORMAL, NULL
1416                                     );
1417            if (hFileRead == INVALID_HANDLE_VALUE) {
1418                hFileRead = hFileWrite = INVALID_HANDLE_VALUE;
1419                throw RIFF::Exception("Can't open \"" + path + "\"");
1420            }
1421          #else          #else
1422          hFileRead = hFileWrite = fopen(path.c_str(), "rb");          hFileRead = hFileWrite = fopen(path.c_str(), "rb");
1423          if (!hFile) throw RIFF::Exception("Can't open \"" + path + "\"");          if (!hFileRead) throw RIFF::Exception("Can't open \"" + path + "\"");
1424          #endif // POSIX          #endif // POSIX
1425          Mode = stream_mode_read;          Mode = stream_mode_read;
1426          ulStartPos = RIFF_HEADER_SIZE;          ulStartPos = RIFF_HEADER_SIZE;
# Line 1324  namespace RIFF { Line 1459  namespace RIFF {
1459                          hFileRead = hFileWrite = 0;                          hFileRead = hFileWrite = 0;
1460                          throw Exception("Could not (re)open file \"" + Filename + "\" in read mode");                          throw Exception("Could not (re)open file \"" + Filename + "\" in read mode");
1461                      }                      }
1462                        #elif defined(WIN32)
1463                        if (hFileRead != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1464                        hFileRead = hFileWrite = CreateFile(
1465                                                     Filename.c_str(), GENERIC_READ,
1466                                                     FILE_SHARE_READ | FILE_SHARE_WRITE,
1467                                                     NULL, OPEN_EXISTING,
1468                                                     FILE_ATTRIBUTE_NORMAL, NULL
1469                                                 );
1470                        if (hFileRead == INVALID_HANDLE_VALUE) {
1471                            hFileRead = hFileWrite = INVALID_HANDLE_VALUE;
1472                            throw Exception("Could not (re)open file \"" + Filename + "\" in read mode");
1473                        }
1474                      #else                      #else
1475                      if (hFileRead) fclose(hFileRead);                      if (hFileRead) fclose(hFileRead);
1476                      hFileRead = hFileWrite = fopen(path.c_str(), "rb");                      hFileRead = hFileWrite = fopen(Filename.c_str(), "rb");
1477                      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");
1478                      #endif                      #endif
1479                      __resetPos(); // reset read/write position of ALL 'Chunk' objects                      __resetPos(); // reset read/write position of ALL 'Chunk' objects
# Line 1339  namespace RIFF { Line 1486  namespace RIFF {
1486                          hFileRead = hFileWrite = open(Filename.c_str(), O_RDONLY | O_NONBLOCK);                          hFileRead = hFileWrite = open(Filename.c_str(), O_RDONLY | O_NONBLOCK);
1487                          throw Exception("Could not open file \"" + Filename + "\" in read+write mode");                          throw Exception("Could not open file \"" + Filename + "\" in read+write mode");
1488                      }                      }
1489                        #elif defined(WIN32)
1490                        if (hFileRead != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1491                        hFileRead = hFileWrite = CreateFile(
1492                                                     Filename.c_str(),
1493                                                     GENERIC_READ | GENERIC_WRITE,
1494                                                     FILE_SHARE_READ,
1495                                                     NULL, OPEN_ALWAYS,
1496                                                     FILE_ATTRIBUTE_NORMAL, NULL
1497                                                 );
1498                        if (hFileRead == INVALID_HANDLE_VALUE) {
1499                            hFileRead = hFileWrite = CreateFile(
1500                                                         Filename.c_str(), GENERIC_READ,
1501                                                         FILE_SHARE_READ | FILE_SHARE_WRITE,
1502                                                         NULL, OPEN_EXISTING,
1503                                                         FILE_ATTRIBUTE_NORMAL, NULL
1504                                                     );
1505                            throw Exception("Could not (re)open file \"" + Filename + "\" in read+write mode");
1506                        }
1507                      #else                      #else
1508                      if (hFileRead) fclose(hFileRead);                      if (hFileRead) fclose(hFileRead);
1509                      hFileRead = hFileWrite = fopen(path.c_str(), "r+b");                      hFileRead = hFileWrite = fopen(Filename.c_str(), "r+b");
1510                      if (!hFileRead) {                      if (!hFileRead) {
1511                          hFileRead = hFileWrite = fopen(path.c_str(), "rb");                          hFileRead = hFileWrite = fopen(Filename.c_str(), "rb");
1512                          throw Exception("Could not open file \"" + Filename + "\" in read+write mode");                          throw Exception("Could not open file \"" + Filename + "\" in read+write mode");
1513                      }                      }
1514                      #endif                      #endif
# Line 1353  namespace RIFF { Line 1518  namespace RIFF {
1518                      #if POSIX                      #if POSIX
1519                      if (hFileRead)  close(hFileRead);                      if (hFileRead)  close(hFileRead);
1520                      if (hFileWrite) close(hFileWrite);                      if (hFileWrite) close(hFileWrite);
1521                        #elif defined(WIN32)
1522                        if (hFileRead  != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1523                        if (hFileWrite != INVALID_HANDLE_VALUE) CloseHandle(hFileWrite);
1524                      #else                      #else
1525                      if (hFileRead)  fclose(hFileRead);                      if (hFileRead)  fclose(hFileRead);
1526                      if (hFileWrite) fclose(hFileWrite);                      if (hFileWrite) fclose(hFileWrite);
# Line 1368  namespace RIFF { Line 1536  namespace RIFF {
1536          return false;          return false;
1537      }      }
1538    
1539        /** @brief Set the byte order to be used when saving.
1540         *
1541         * Set the byte order to be used in the file. A value of
1542         * endian_little will create a RIFF file, endian_big a RIFX file
1543         * and endian_native will create a RIFF file on little-endian
1544         * machines and RIFX on big-endian machines.
1545         *
1546         * @param Endian - endianess to use when file is saved.
1547         */
1548        void File::SetByteOrder(endian_t Endian) {
1549            #if WORDS_BIGENDIAN
1550            bEndianNative = Endian != endian_little;
1551            #else
1552            bEndianNative = Endian != endian_big;
1553            #endif
1554        }
1555    
1556      /** @brief Save changes to same file.      /** @brief Save changes to same file.
1557       *       *
1558       * 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 1379  namespace RIFF { Line 1564  namespace RIFF {
1564       *                         chunk or any kind of IO error occured       *                         chunk or any kind of IO error occured
1565       */       */
1566      void File::Save() {      void File::Save() {
1567            // make sure the RIFF tree is built (from the original file)
1568            LoadSubChunksRecursively();
1569    
1570          // reopen file in write mode          // reopen file in write mode
1571          SetMode(stream_mode_read_write);          SetMode(stream_mode_read_write);
1572    
# Line 1392  namespace RIFF { Line 1580  namespace RIFF {
1580          // 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)
1581          unsigned long ulPositiveSizeDiff = 0;          unsigned long ulPositiveSizeDiff = 0;
1582          for (ChunkList::iterator iter = ResizedChunks.begin(), end = ResizedChunks.end(); iter != end; ++iter) {          for (ChunkList::iterator iter = ResizedChunks.begin(), end = ResizedChunks.end(); iter != end; ++iter) {
1583              if ((*iter)->GetNewSize() == 0) throw Exception("There is at least one empty chunk (zero size)");              if ((*iter)->GetNewSize() == 0) {
1584                    throw Exception("There is at least one empty chunk (zero size): " + __resolveChunkPath(*iter));
1585                }
1586              if ((*iter)->GetNewSize() + 1L > (*iter)->GetSize()) {              if ((*iter)->GetNewSize() + 1L > (*iter)->GetSize()) {
1587                  unsigned long ulDiff = (*iter)->GetNewSize() - (*iter)->GetSize() + 1L; // +1 in case we have to add a pad byte                  unsigned long ulDiff = (*iter)->GetNewSize() - (*iter)->GetSize() + 1L; // +1 in case we have to add a pad byte
1588                  ulPositiveSizeDiff += ulDiff;                  ulPositiveSizeDiff += ulDiff;
# Line 1409  namespace RIFF { Line 1599  namespace RIFF {
1599              // ... and move current data by the same amount towards end of file.              // ... and move current data by the same amount towards end of file.
1600              int8_t* pCopyBuffer = new int8_t[4096];              int8_t* pCopyBuffer = new int8_t[4096];
1601              const unsigned long ulFileSize = GetSize() + RIFF_HEADER_SIZE;              const unsigned long ulFileSize = GetSize() + RIFF_HEADER_SIZE;
1602              int iBytesMoved = 1;              #if defined(WIN32)
1603                DWORD iBytesMoved = 1; // we have to pass it via pointer to the Windows API, thus the correct size must be ensured
1604                #else
1605                int iBytesMoved = 1;
1606                #endif
1607              for (unsigned long ulPos = 0; iBytesMoved > 0; ulPos += iBytesMoved) {              for (unsigned long ulPos = 0; iBytesMoved > 0; ulPos += iBytesMoved) {
1608                  const unsigned long ulToMove = ulFileSize - ulPos;                  const unsigned long ulToMove = ulFileSize - ulPos;
1609                  iBytesMoved = (ulToMove < 4096) ? ulToMove : 4096;                  iBytesMoved = (ulToMove < 4096) ? ulToMove : 4096;
# Line 1418  namespace RIFF { Line 1612  namespace RIFF {
1612                  iBytesMoved = read(hFileRead, pCopyBuffer, iBytesMoved);                  iBytesMoved = read(hFileRead, pCopyBuffer, iBytesMoved);
1613                  lseek(hFileWrite, ulPos + ulPositiveSizeDiff, SEEK_SET);                  lseek(hFileWrite, ulPos + ulPositiveSizeDiff, SEEK_SET);
1614                  iBytesMoved = write(hFileWrite, pCopyBuffer, iBytesMoved);                  iBytesMoved = write(hFileWrite, pCopyBuffer, iBytesMoved);
1615                    #elif defined(WIN32)
1616                    SetFilePointer(hFileRead, ulPos, NULL/*32 bit*/, FILE_BEGIN);
1617                    ReadFile(hFileRead, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
1618                    SetFilePointer(hFileWrite, ulPos + ulPositiveSizeDiff, NULL/*32 bit*/, FILE_BEGIN);
1619                    WriteFile(hFileWrite, pCopyBuffer, iBytesMoved, &iBytesMoved, NULL);
1620                  #else                  #else
1621                  fseek(hFileRead, ulPos, SEEK_SET);                  fseek(hFileRead, ulPos, SEEK_SET);
1622                  iBytesMoved = fread(pCopyBuffer, 1, iBytesMoved, hFileRead);                  iBytesMoved = fread(pCopyBuffer, 1, iBytesMoved, hFileRead);
# Line 1456  namespace RIFF { Line 1655  namespace RIFF {
1655      void File::Save(const String& path) {      void File::Save(const String& path) {
1656          //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          //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
1657    
1658            // make sure the RIFF tree is built (from the original file)
1659            LoadSubChunksRecursively();
1660    
1661          if (Filename.length() > 0) SetMode(stream_mode_read);          if (Filename.length() > 0) SetMode(stream_mode_read);
1662          // 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
1663          #if POSIX          #if POSIX
# Line 1464  namespace RIFF { Line 1666  namespace RIFF {
1666              hFileWrite = hFileRead;              hFileWrite = hFileRead;
1667              throw Exception("Could not open file \"" + path + "\" for writing");              throw Exception("Could not open file \"" + path + "\" for writing");
1668          }          }
1669            #elif defined(WIN32)
1670            hFileWrite = CreateFile(
1671                             path.c_str(), GENERIC_WRITE, FILE_SHARE_READ,
1672                             NULL, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL
1673                         );
1674            if (hFileWrite == INVALID_HANDLE_VALUE) {
1675                hFileWrite = hFileRead;
1676                throw Exception("Could not open file \"" + path + "\" for writing");
1677            }
1678          #else          #else
1679          hFileWrite = fopen(path.c_str(), "w+b");          hFileWrite = fopen(path.c_str(), "w+b");
1680          if (!hFileWrite) {          if (!hFileWrite) {
# Line 1483  namespace RIFF { Line 1694  namespace RIFF {
1694          // forget all resized chunks          // forget all resized chunks
1695          ResizedChunks.clear();          ResizedChunks.clear();
1696    
1697          if (Filename.length() > 0) {          #if POSIX
1698              #if POSIX          if (hFileWrite) close(hFileWrite);
1699              close(hFileWrite);          #elif defined(WIN32)
1700              #else          if (hFileWrite != INVALID_HANDLE_VALUE) CloseHandle(hFileWrite);
1701              fclose(hFileWrite);          #else
1702              #endif          if (hFileWrite) fclose(hFileWrite);
1703              hFileWrite = hFileRead;          #endif
1704          }          hFileWrite = hFileRead;
1705    
1706          // associate new file with this File object from now on          // associate new file with this File object from now on
1707          Filename = path;          Filename = path;
# Line 1502  namespace RIFF { Line 1713  namespace RIFF {
1713          #if POSIX          #if POSIX
1714          if (ftruncate(hFileWrite, ulNewSize) < 0)          if (ftruncate(hFileWrite, ulNewSize) < 0)
1715              throw Exception("Could not resize file \"" + Filename + "\"");              throw Exception("Could not resize file \"" + Filename + "\"");
1716            #elif defined(WIN32)
1717            if (
1718                SetFilePointer(hFileWrite, ulNewSize, NULL/*32 bit*/, FILE_BEGIN) == INVALID_SET_FILE_POINTER ||
1719                !SetEndOfFile(hFileWrite)
1720            ) throw Exception("Could not resize file \"" + Filename + "\"");
1721          #else          #else
1722          # error Sorry, this version of libgig only supports POSIX systems yet.          # error Sorry, this version of libgig only supports POSIX and Windows systems yet.
1723          # error Reason: portable implementation of RIFF::File::ResizeFile() is missing (yet)!          # error Reason: portable implementation of RIFF::File::ResizeFile() is missing (yet)!
1724          #endif          #endif
1725      }      }
# Line 1514  namespace RIFF { Line 1730  namespace RIFF {
1730         #endif // DEBUG         #endif // DEBUG
1731          #if POSIX          #if POSIX
1732          if (hFileRead) close(hFileRead);          if (hFileRead) close(hFileRead);
1733            #elif defined(WIN32)
1734            if (hFileRead != INVALID_HANDLE_VALUE) CloseHandle(hFileRead);
1735          #else          #else
1736          if (hFileRead) fclose(hFileRead);          if (hFileRead) fclose(hFileRead);
1737          #endif // POSIX          #endif // POSIX
# Line 1523  namespace RIFF { Line 1741  namespace RIFF {
1741          ResizedChunks.push_back(pResizedChunk);          ResizedChunks.push_back(pResizedChunk);
1742      }      }
1743    
1744        void File::UnlogResized(Chunk* pResizedChunk) {
1745            ResizedChunks.remove(pResizedChunk);
1746        }
1747    
1748      unsigned long File::GetFileSize() {      unsigned long File::GetFileSize() {
1749          return __GetFileSize(hFileRead);          return __GetFileSize(hFileRead);
1750      }      }
# Line 1534  namespace RIFF { Line 1756  namespace RIFF {
1756          long size = filestat.st_size;          long size = filestat.st_size;
1757          return size;          return size;
1758      }      }
1759        #elif defined(WIN32)
1760        unsigned long File::__GetFileSize(HANDLE hFile) {
1761            DWORD dwSize = ::GetFileSize(hFile, NULL /*32bit*/);
1762            if (dwSize == INVALID_FILE_SIZE)
1763                throw Exception("Windows FS error: could not determine file size");
1764            return dwSize;
1765        }
1766      #else // standard C functions      #else // standard C functions
1767      unsigned long File::__GetFileSize(FILE* hFile) {      unsigned long File::__GetFileSize(FILE* hFile) {
1768          long curpos = ftell(hFile);          long curpos = ftell(hFile);
Legend:
Removed from v.798  
changed lines
  Added in v.1301
  ViewVC Help
Powered by ViewVC