contrib/extractor/System.cpp

   1 #define _CRT_SECURE_NO_DEPRECATE
   2
   3 #include <stdio.h>
   4 #include <deque>
   5 #include <set>
   6
   7 #ifdef WIN32
   8 #include "direct.h"
   9 #else
  10 #include <sys/stat.h>
  11 #endif
  12
  13 #include "dbcfile.h"
  14 #include "mpq_libmpq.h"
  15
  16 extern unsigned int iRes;
  17 extern ArchiveSet gOpenArchives;
  18
  19 bool ConvertADT(char*,char*);
  20
  21 typedef struct{
  22     char name[64];
  23     unsigned int id;
  24 }map_id;
  25
  26 typedef unsigned char uint8;
  27 typedef unsigned short uint16;
  28 typedef unsigned int uint32;
  29
  30 map_id * map_ids;
  31 uint16 * areas;
  32 char output_path[128]=".";
  33 char input_path[128]=".";
  34 uint32 maxAreaId = 0;
  35
  36 enum Extract
  37 {
  38     EXTRACT_MAP = 1,
  39     EXTRACT_DBC = 2
  40 };
  41 int extract = EXTRACT_MAP | EXTRACT_DBC;
  42
  43 static char* const langs[]={"enGB", "enUS", "deDE", "esES", "frFR", "koKR", "zhCN", "zhTW", "enCN", "enTW", "esMX", "ruRU" };
  44 #define LANG_COUNT 12
  45
  46 #define ADT_RES 64
  47
  48 void CreateDir( const std::string& Path )
  49 {
  50     #ifdef WIN32
  51     _mkdir( Path.c_str());
  52     #else
  53     mkdir( Path.c_str(), 0777 );
  54     #endif
  55 }
  56
  57 bool FileExists( const char* FileName )
  58 {
  59     if( FILE* fp = fopen( FileName, "rb" ) )
  60     {
  61         fclose( fp );
  62         return true;
  63     }
  64
  65     return false;
  66 }
  67
  68 void Usage(char* prg)
  69 {
  70     printf("Usage:\n%s -[var] [value]\n-i set input path\n-o set output path\n-r set resolution\n-e extract only MAP(1)/DBC(2) - standard: both(3)\nExample: %s -r 256 -i \"c:\\games\\game\"",
  71     prg,prg);
  72     exit(1);
  73 }
  74
  75 void HandleArgs(int argc, char * arg[])
  76 {
  77     for(int c=1;c<argc;c++)
  78     {
  79         //i - input path
  80         //o - output path
  81         //r - resolution, array of (r * r) heights will be created
  82         //e - extract only MAP(1)/DBC(2) - standard both(3)
  83         if(arg[c][0] != '-')
  84             Usage(arg[0]);
  85
  86         switch(arg[c][1])
  87         {
  88             case 'i':
  89                 if(c+1<argc)//all ok
  90                     strcpy(input_path,arg[(c++) +1]);
  91                 else
  92                     Usage(arg[0]);
  93                 break;
  94             case 'o':
  95                 if(c+1<argc)//all ok
  96                     strcpy(output_path,arg[(c++) +1]);
  97                 else
  98                     Usage(arg[0]);
  99                 break;
 100             case 'r':
 101                 if(c+1<argc)//all ok
 102                     iRes=atoi(arg[(c++) +1]);
 103                 else
 104                     Usage(arg[0]);
 105                 break;
 106             case 'e':
 107                 if(c+1<argc)//all ok
 108                 {
 109                     extract=atoi(arg[(c++) +1]);
 110                     if(!(extract > 0 && extract < 4))
 111                         Usage(arg[0]);
 112                 }
 113                 else
 114                     Usage(arg[0]);
 115                 break;
 116         }
 117     }
 118 }
 119
 120 uint32 ReadMapDBC()
 121 {
 122     printf("Read Map.dbc file... ");
 123     DBCFile dbc("DBFilesClient\\Map.dbc");
 124     dbc.open();
 125
 126     uint32 map_count=dbc.getRecordCount();
 127     map_ids=new map_id[map_count];
 128     for(unsigned int x=0;x<map_count;x++)
 129     {
 130         map_ids[x].id=dbc.getRecord(x).getUInt(0);
 131         strcpy(map_ids[x].name,dbc.getRecord(x).getString(1));
 132     }
 133     printf("Done! (%u maps loaded)\n", map_count);
 134     return map_count;
 135 }
 136
 137 void ReadAreaTableDBC()
 138 {
 139     printf("Read AreaTable.dbc file... ");
 140     DBCFile dbc("DBFilesClient\\AreaTable.dbc");
 141     dbc.open();
 142
 143     unsigned int area_count=dbc.getRecordCount();
 144     uint32 maxid = dbc.getMaxId();
 145     areas=new uint16[maxid + 1];
 146     memset(areas, 0xff, sizeof(areas));
 147     for(unsigned int x=0; x<area_count;++x)
 148         areas[dbc.getRecord(x).getUInt(0)] = dbc.getRecord(x).getUInt(3);
 149
 150     maxAreaId = dbc.getMaxId();
 151
 152     printf("Done! (%u areas loaded)\n", area_count);
 153 }
 154
 155 void ExtractMapsFromMpq()
 156 {
 157     char mpq_filename[1024];
 158     char output_filename[1024];
 159
 160     printf("Extracting maps...\n");
 161
 162     uint32 map_count = ReadMapDBC();
 163
 164     ReadAreaTableDBC();
 165
 166     unsigned int total=map_count*ADT_RES*ADT_RES;
 167     unsigned int done=0;
 168
 169     std::string path = output_path;
 170     path += "/maps/";
 171     CreateDir(path);
 172
 173     for(unsigned int x = 0; x < ADT_RES; ++x)
 174     {
 175         for(unsigned int y = 0; y < ADT_RES; ++y)
 176         {
 177             for(unsigned int z = 0; z < map_count; ++z)
 178             {
 179                 sprintf(mpq_filename,"World\\Maps\\%s\\%s_%u_%u.adt",map_ids[z].name,map_ids[z].name,x,y);
 180                 sprintf(output_filename,"%s/maps/%03u%02u%02u.map",output_path,map_ids[z].id,y,x);
 181                 ConvertADT(mpq_filename,output_filename);
 182                 done++;
 183             }
 184             //draw progess bar
 185             printf("Processing........................%d%%\r",(100*done)/total);
 186         }
 187     }
 188
 189     delete [] areas;
 190     delete [] map_ids;
 191 }
 192
 193 //bool WMO(char* filename);
 194
 195 void ExtractDBCFiles(int locale, bool basicLocale)
 196 {
 197     printf("Extracting dbc files...\n");
 198
 199     set<string> dbcfiles;
 200
 201     // get DBC file list
 202     for(ArchiveSet::iterator i = gOpenArchives.begin(); i != gOpenArchives.end();++i)
 203     {
 204         vector<string> files;
 205         (*i)->GetFileListTo(files);
 206         for (vector<string>::iterator iter = files.begin(); iter != files.end(); ++iter)
 207             if (iter->rfind(".dbc") == iter->length() - strlen(".dbc"))
 208                     dbcfiles.insert(*iter);
 209     }
 210
 211     string path = output_path;
 212     path += "/dbc/";
 213     CreateDir(path);
 214     if(!basicLocale)
 215     {
 216         path += langs[locale];
 217         path += "/";
 218         CreateDir(path);
 219     }
 220
 221     // extract DBCs
 222     int count = 0;
 223     for (set<string>::iterator iter = dbcfiles.begin(); iter != dbcfiles.end(); ++iter)
 224     {
 225         string filename = path;
 226         filename += (iter->c_str() + strlen("DBFilesClient\\"));
 227
 228         FILE *output=fopen(filename.c_str(),"wb");
 229         if(!output)
 230         {
 231             printf("Can't create the output file '%s'\n",filename.c_str());
 232             continue;
 233         }
 234         MPQFile m(iter->c_str());
 235         if(!m.isEof())
 236             fwrite(m.getPointer(),1,m.getSize(),output);
 237
 238         fclose(output);
 239         ++count;
 240     }
 241     printf("Extracted %u DBC files\n\n", count);
 242 }
 243
 244 void LoadLocaleMPQFiles(int const locale)
 245 {
 246     char filename[512];
 247
 248     sprintf(filename,"%s/Data/%s/locale-%s.MPQ",input_path,langs[locale],langs[locale]);
 249     new MPQArchive(filename);
 250
 251     for(int i = 1; i < 5; ++i)
 252     {
 253         char ext[3] = "";
 254         if(i > 1)
 255             sprintf(ext, "-%i", i);
 256
 257         sprintf(filename,"%s/Data/%s/patch-%s%s.MPQ",input_path,langs[locale],langs[locale],ext);
 258         if(FileExists(filename))
 259             new MPQArchive(filename);
 260     }
 261 }
 262
 263 void LoadCommonMPQFiles()
 264 {
 265     char filename[512];
 266
 267     sprintf(filename,"%s/Data/common-2.MPQ",input_path);
 268     new MPQArchive(filename);
 269     sprintf(filename,"%s/Data/lichking.MPQ",input_path);
 270     new MPQArchive(filename);
 271     sprintf(filename,"%s/Data/expansion.MPQ",input_path);
 272     new MPQArchive(filename);
 273
 274     for(int i = 1; i < 5; ++i)
 275     {
 276         char ext[3] = "";
 277         if(i > 1)
 278             sprintf(ext, "-%i", i);
 279
 280         sprintf(filename,"%s/Data/patch%s.MPQ",input_path,ext);
 281         if(FileExists(filename))
 282             new MPQArchive(filename);
 283     }
 284 }
 285
 286 inline void CloseMPQFiles()
 287 {
 288     for(ArchiveSet::iterator j = gOpenArchives.begin(); j != gOpenArchives.end();++j) (*j)->close();
 289         gOpenArchives.clear();
 290 }
 291
 292 int main(int argc, char * arg[])
 293 {
 294     printf("Map & DBC Extractor\n");
 295     printf("===================\n\n");
 296
 297     HandleArgs(argc, arg);
 298
 299     int FirstLocale = -1;
 300
 301     for (int i = 0; i < LANG_COUNT; i++)
 302     {
 303         char tmp1[512];
 304         sprintf(tmp1, "%s/Data/%s/locale-%s.MPQ", input_path, langs[i], langs[i]);
 305         if (FileExists(tmp1))
 306         {
 307             printf("Detected locale: %s\n", langs[i]);
 308
 309             //Open MPQs
 310             LoadLocaleMPQFiles(i);
 311
 312             if((extract & EXTRACT_DBC) == 0)
 313             {
 314                 FirstLocale=i;
 315                 break;
 316             }
 317
 318             //Extract DBC files
 319             if(FirstLocale<0)
 320             {
 321                 ExtractDBCFiles(i, true);
 322                 FirstLocale = i;
 323             }
 324             else
 325                 ExtractDBCFiles(i, false);
 326
 327             //Close MPQs
 328             CloseMPQFiles();
 329         }
 330     }
 331
 332     if(FirstLocale<0)
 333     {
 334         printf("No locales detected\n");
 335         return 0;
 336     }
 337
 338     if (extract & EXTRACT_MAP)
 339     {
 340         printf("Using locale: %s\n", langs[FirstLocale]);
 341
 342         // Open MPQs
 343         LoadLocaleMPQFiles(FirstLocale);
 344         LoadCommonMPQFiles();
 345
 346         // Extract maps
 347         ExtractMapsFromMpq();
 348
 349         // Close MPQs
 350         CloseMPQFiles();
 351     }
 352
 353     return 0;
 354 }