contrib/vmap_extractor_v2/vmapextract/adtfile.cpp

   1 #include "adtfile.h"
   2
   3 #include <algorithm>
   4
   5 char * GetPlainName(char * FileName)
   6 {
   7     char * szTemp;
   8
   9     if((szTemp = strrchr(FileName, '\\')) != NULL)
  10         FileName = szTemp + 1;
  11     return FileName;
  12 }
  13
  14 void fixnamen(char *name, size_t len)
  15 {
  16     for (size_t i=0; i<len-3; i++)
  17     {
  18         if (i>0 && name[i]>='A' && name[i]<='Z' && isalpha(name[i-1]))
  19         {
  20             name[i] |= 0x20;
  21         } else if ((i==0 || !isalpha(name[i-1])) && name[i]>='a' && name[i]<='z')
  22         {
  23             name[i] &= ~0x20;
  24         }
  25     }
  26 }
  27 void fixname2(char *name, size_t len)
  28 {
  29     for (size_t i=0; i<len-3; i++)
  30     {
  31         if(name[i] == ' ')
  32         name[i] = '_';
  33     }
  34 }
  35
  36 ADTFile::ADTFile(char* filename): ADT(filename)
  37 {
  38     Adtfilename.append(filename);
  39 }
  40
  41 bool ADTFile::init(char *map_id)
  42 {
  43     if(ADT.isEof ())
  44         return false;
  45
  46     size_t size;
  47
  48     string xMap;
  49     string yMap;
  50
  51     Adtfilename.erase(Adtfilename.find(".adt"),4);
  52     string TempMapNumber;
  53     TempMapNumber = Adtfilename.substr(Adtfilename.length()-6,6);
  54     xMap = TempMapNumber.substr(TempMapNumber.find("_")+1,(TempMapNumber.find_last_of("_")-1) - (TempMapNumber.find("_")));
  55     yMap = TempMapNumber.substr(TempMapNumber.find_last_of("_")+1,(TempMapNumber.length()) - (TempMapNumber.find_last_of("_")));
  56     Adtfilename.erase((Adtfilename.length()-xMap.length()-yMap.length()-2), (xMap.length()+yMap.length()+2));
  57     string AdtMapNumber = xMap + ' ' + yMap + ' ' + GetPlainName((char*)Adtfilename.c_str());
  58     printf("Processing map %s...\n", AdtMapNumber.c_str());
  59     //printf("MapNumber = %s\n", TempMapNumber.c_str());
  60     //printf("xMap = %s\n", xMap.c_str());
  61     //printf("yMap = %s\n", yMap.c_str());
  62
  63     const char dirname[] = "buildings\\dir";
  64     FILE *dirfile;
  65     dirfile = fopen(dirname, "ab");
  66     if(!dirfile)
  67     {
  68         printf("Can't open dirfile!'%s'\n");
  69         return false;
  70     }
  71
  72
  73     while (!ADT.isEof())
  74     {
  75         char fourcc[5];
  76         ADT.read(&fourcc,4);
  77         ADT.read(&size, 4);
  78         flipcc(fourcc);
  79         fourcc[4] = 0;
  80
  81         size_t nextpos = ADT.getPos() + size;
  82
  83         if (!strcmp(fourcc,"MCIN"))
  84         {
  85
  86         }
  87         else if (!strcmp(fourcc,"MTEX"))
  88         {
  89         }
  90         else if (!strcmp(fourcc,"MMDX"))
  91         {
  92             if (size)
  93             {
  94                 char *buf = new char[size];
  95                 ADT.read(buf, size);
  96                 char *p=buf;
  97                 int t=0;
  98                 ModelInstansName = new string[size];
  99                 while (p<buf+size)
 100                 {
 101                     fixnamen(p,strlen(p));
 102                     string path(p);
 103                     char* s=GetPlainName(p);
 104                     fixname2(s,strlen(s));
 105                     p=p+strlen(p)+1;
 106                     ModelInstansName[t++] = s;
 107
 108                     // < 3.1.0 ADT MMDX section store filename.mdx filenames for corresponded .m2 file
 109                     std::string ext3 = path.size() >= 4 ? path.substr(path.size()-4,4) : "";
 110                     std::transform( ext3.begin(), ext3.end(), ext3.begin(), ::tolower );
 111                     if(ext3 == ".mdx")
 112                     {
 113                         // replace .mdx -> .m2
 114                         path.erase(path.length()-2,2);
 115                         path.append("2");
 116                     }
 117                     // >= 3.1.0 ADT MMDX section store filename.m2 filenames for corresponded .m2 file
 118                     // nothing do
 119
 120                     char szLocalFile[MAX_PATH];
 121                     sprintf(szLocalFile, ".\\buildings\\%s", s);
 122                     FILE * output = fopen(szLocalFile,"rb");
 123                     if(!output)
 124                     {
 125                         Model * m2 = new Model(path);
 126                         if(m2->open())
 127                             m2->ConvertToVMAPModel(szLocalFile);
 128                         delete m2;
 129                     }
 130                     else
 131                         fclose(output);
 132                 }
 133
 134                 delete[] buf;
 135             }
 136
 137         }
 138         else if (!strcmp(fourcc,"MWMO"))
 139         {
 140             if (size)
 141             {
 142                 char *buf = new char[size];
 143                 ADT.read(buf, size);
 144                 char *p=buf;
 145                 int q = 0;
 146                 WmoInstansName = new string[size];
 147                 while (p<buf+size)
 148                 {
 149                     string path(p);
 150                     char* s=GetPlainName(p);
 151                     fixnamen(s,strlen(s));
 152                     fixname2(s,strlen(s));
 153                     p=p+strlen(p)+1;
 154                     WmoInstansName[q++] = s;
 155                 }
 156                 delete[] buf;
 157             }
 158         }
 159         //======================
 160         else if (!strcmp(fourcc,"MDDF"))
 161         {
 162             if (size)
 163             {
 164                 nMDX = (int)size / 36;
 165                 for (int i=0; i<nMDX; ++i)
 166                 {
 167                     int id;
 168                     ADT.read(&id, 4);
 169                     ModelInstance inst(ADT,ModelInstansName[id].c_str(),map_id, dirfile);//!!!!!!!!!!!
 170                 }
 171                 delete[] ModelInstansName;
 172             }
 173         }
 174         else if (!strcmp(fourcc,"MODF"))
 175         {
 176             if (size)
 177             {
 178                 nWMO = (int)size / 64;
 179                 for (int i=0; i<nWMO; ++i)
 180                 {
 181                     int id;
 182                     ADT.read(&id, 4);
 183                     WMOInstance inst(ADT,WmoInstansName[id].c_str(),map_id, dirfile);//!!!!!!!!!!!!!
 184                 }
 185                 delete[] WmoInstansName;
 186             }
 187         }
 188         //======================
 189 #if 0
 190         else if (!strcmp(fourcc,"MDDF"))
 191         {
 192             if (size)
 193             {
 194                 nMDX = (int)size / 36;
 195                 for (int i=0; i<nMDX; ++i)
 196                 {
 197                     int id;
 198                     ADT.read(&id, 4);
 199                     ModelInstance inst(ADT,ModelInstansName[id].c_str(),AdtMapNumber.c_str(), dirfile);
 200                 }
 201                 delete[] ModelInstansName;
 202             }
 203         }
 204         else if (!strcmp(fourcc,"MODF"))
 205         {
 206             if (size)
 207             {
 208                 nWMO = (int)size / 64;
 209                 for (int i=0; i<nWMO; ++i)
 210                 {
 211                     int id;
 212                     ADT.read(&id, 4);
 213                     WMOInstance inst(ADT,WmoInstansName[id].c_str(),AdtMapNumber.c_str(), dirfile);
 214                 }
 215                 delete[] WmoInstansName;
 216             }
 217         }
 218 #endif
 219         ADT.seek(nextpos);
 220     }
 221     ADT.close();
 222     fclose(dirfile);
 223     return true;
 224 }
 225
 226 ADTFile::~ADTFile()
 227 {
 228     ADT.close();
 229 }