contrib/vmap_extractor_v2/vmapextract/mpq.cpp

   1 #include "mpq.h"
   2 //#include <vector>
   3 #include "Stormlib.h"
   4 #define __STORMLIB_SELF__
   5
   6 MPQArchiveSet gOpenArchives;
   7
   8 //xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
   9 MPQArchive::MPQArchive(const char* filename)
  10 {
  11     BOOL succ = SFileOpenArchive(filename, 0, 0,&hMPQ);
  12     if (succ)
  13         printf("Opening %s\n", filename);
  14     else
  15         printf("Error!!!Not open archive %s\n", filename);
  16 }
  17
  18 void MPQArchive::close()
  19 {
  20     SFileCloseArchive(hMPQ);
  21 }
  22
  23 bool MPQArchiveSet::Open( std::vector<std::string> const& archiveNames )
  24 {
  25     for (size_t i=0; i < archiveNames.size(); ++i)
  26     {
  27         MPQArchive mpqarch(archiveNames[i].c_str());
  28         if(mpqarch.isOpen())
  29             archives.push_back(mpqarch);
  30     }
  31
  32     return !archives.empty();
  33 }
  34
  35 MPQArchiveSet::~MPQArchiveSet()
  36 {
  37     // close archives
  38     for (ArchiveSet::iterator ar_itr = archives.begin(); ar_itr != archives.end(); ++ar_itr)
  39         ar_itr->close();
  40 }
  41
  42 MPQFile::MPQFile(const char* filename):
  43     eof(false),
  44     buffer(0),
  45     pointer(0),
  46     size(0)
  47 {
  48     for(ArchiveSet::const_iterator i=gOpenArchives.archives.begin(); i!=gOpenArchives.archives.end();++i)
  49     {
  50         HANDLE hFile = "";
  51         hMPQ = i->hMPQ;
  52         BOOL succ = SFileOpenFileEx(hMPQ,filename,0, &hFile);
  53         if (succ)
  54         {
  55             DWORD s = SFileGetFileSize(hFile, 0);
  56             if (!s)
  57             {
  58                 eof = true;
  59                 buffer = 0;
  60                 return;
  61             }
  62             size = (size_t)s;
  63             buffer = new char[s];
  64             SFileReadFile(hFile, buffer, s, 0, 0);
  65             SFileCloseFile(hFile);
  66
  67             eof = false;
  68             return;
  69         }
  70     }
  71
  72     eof = true;
  73     buffer = 0;
  74 }
  75
  76 MPQFile::~MPQFile()
  77 {
  78     close();
  79 }
  80
  81 size_t MPQFile::read(void* dest, size_t bytes)
  82 {
  83     if (eof)
  84         return 0;
  85
  86     size_t rpos = pointer + bytes;
  87     if (rpos > size)
  88     {
  89         bytes = size - pointer;
  90         eof = true;
  91     }
  92
  93     memcpy(dest, &(buffer[pointer]), bytes);
  94
  95     pointer = rpos;
  96
  97     return bytes;
  98 }
  99
 100 bool MPQFile::isEof()
 101 {
 102     return eof;
 103 }
 104
 105 void MPQFile::seek(int offset)
 106 {
 107     pointer = offset;
 108     eof = (pointer >= size);
 109 }
 110
 111 void MPQFile::seekRelative(int offset)
 112 {
 113     pointer += offset;
 114     eof = (pointer >= size);
 115 }
 116
 117 void MPQFile::close()
 118 {
 119     if (buffer)
 120         delete[] buffer;
 121     buffer = 0;
 122     eof = true;
 123 }
 124
 125 size_t MPQFile::getSize()
 126 {
 127     return size;
 128 }
 129
 130 size_t MPQFile::getPos()
 131 {
 132     return pointer;
 133 }
 134
 135 char* MPQFile::getBuffer()
 136 {
 137     return buffer;
 138 }
 139
 140 char* MPQFile::getPointer()
 141 {
 142     return buffer + pointer;
 143 }