ryzom/tools/make_anim_by_race/main.cpp

   1 // Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
   2 // Copyright (C) 2010  Winch Gate Property Limited
   3 //
   4 // This program is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU Affero General Public License as
   6 // published by the Free Software Foundation, either version 3 of the
   7 // License, or (at your option) any later version.
   8 //
   9 // This program is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU Affero General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Affero General Public License
  15 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17
  18 #include "nel/misc/path.h"
  19 #include "nel/misc/file.h"
  20 #include "nel/misc/common.h"
  21 #include "nel/misc/algo.h"
  22
  23
  24 using namespace std;
  25 using namespace NLMISC;
  26
  27
  28 // ***************************************************************************
  29 void    buildRaceAnimNames(std::vector<string> &raceAnimNames, const std::string &animName)
  30 {
  31         if(animName.compare(0, 3, "fy_")!=0)
  32         {
  33                 nlwarning("ERROR: all .anim must begin with fy_");
  34                 exit(-1);
  35         }
  36
  37         raceAnimNames.resize(4);
  38         raceAnimNames[0]= animName;
  39         raceAnimNames[1]= animName;
  40         raceAnimNames[2]= animName;
  41         raceAnimNames[3]= animName;
  42         raceAnimNames[0].replace(0, 3, "fy_");
  43         raceAnimNames[1].replace(0, 3, "ma_");
  44         raceAnimNames[2].replace(0, 3, "tr_");
  45         raceAnimNames[3].replace(0, 3, "zo_");
  46
  47         // Force ""
  48         for(uint i=0;i<raceAnimNames.size();i++)
  49         {
  50                 raceAnimNames[i]= string("\"") + raceAnimNames[i] + string("\"");
  51         }
  52 }
  53
  54 // ***************************************************************************
  55 string::size_type       findAnimName(const string &lineLwr, const std::vector<string> &raceAnimNames)
  56 {
  57         // line Must contains Name="filename", else CAN BE A LOAD CHAR ANIMATION!!!!
  58         if(lineLwr.find("name=\"filename\"")==string::npos)
  59                 return -1;
  60
  61         // in the animset, the original file can be a "tr_" ... Not necessarily a "fy_"
  62         for(uint i=0;i<raceAnimNames.size();i++)
  63         {
  64                 string::size_type       pos= lineLwr.find(raceAnimNames[i]);
  65                 if(pos!=string::npos)
  66                         return pos;
  67         }
  68         return string::npos;
  69 }
  70
  71 // ***************************************************************************
  72 void    appendRaceAnim(vector<string> &animSetText, uint startBlock, const vector<string> &copyText, uint nameLineInBlock, uint nameIndexInLine, const string &raceAnimName)
  73 {
  74         // add empty space
  75         animSetText.insert(animSetText.begin()+startBlock, copyText.size(), string());
  76
  77         // Fill line by line
  78         for(uint i=0;i<copyText.size();i++)
  79         {
  80                 string  line= copyText[i];
  81
  82                 // If this is the line we have to change the anim name
  83                 if(i==nameLineInBlock)
  84                         line.replace(nameIndexInLine, raceAnimName.size(), raceAnimName);
  85
  86                 // If this is the line we have to specify the race node
  87                 if(i==copyText.size()-2)
  88                 {
  89                         string  peopleEnum;
  90                         if(raceAnimName.compare(1,3,"fy_")==0)
  91                                 peopleEnum= "Fyros";
  92                         else if(raceAnimName.compare(1,3,"ma_")==0)
  93                                 peopleEnum= "Matis";
  94                         else if(raceAnimName.compare(1,3,"tr_")==0)
  95                                 peopleEnum= "Tryker";
  96                         else if(raceAnimName.compare(1,3,"zo_")==0)
  97                                 peopleEnum= "Zorai";
  98                         else
  99                         {
 100                                 nlwarning("ERROR: generated anim can be only fy_, ma_, zo_, tr_. Ask a coder");
 101                                 exit(-1);
 102                         }
 103                         line=toString("          <ATOM Name=\"Race Restriction\" Value=\"%s\"/>", peopleEnum.c_str());
 104                 }
 105
 106                 // copy
 107                 animSetText[startBlock+i]= line;
 108         }
 109 }
 110
 111 // ***************************************************************************
 112 void    makeAnimByRace(const std::string &animSetFile, const std::vector<string> &animList)
 113 {
 114         // *** Read the animset file.
 115         CIFile  iFile;
 116         iFile.open(animSetFile, true);
 117         // Read all text
 118         static vector<string>   animSetText;
 119         animSetText.clear();
 120         while(!iFile.eof())
 121         {
 122                 char    tmp[50000];
 123                 iFile.getline(tmp, 50000);
 124                 animSetText.push_back(tmp);
 125         }
 126         iFile.close();
 127
 128
 129         bool    someChangeDone= false;
 130
 131         // *** For each possible anim
 132         for(uint i=0;i<animList.size();i++)
 133         {
 134                 // get the possible anim file name (lowered)
 135                 static vector<string>   raceAnimNames;
 136                 raceAnimNames.clear();
 137                 buildRaceAnimNames(raceAnimNames, toLower(CFile::getFilename(animList[i])));
 138
 139                 // For each line of the animSet
 140                 uint    lastStructLine= 0;
 141                 bool    raceRestrictionFound= false;
 142                 for(uint j=0;j<animSetText.size();)
 143                 {
 144                         string  line= animSetText[j];
 145                         string  lineLwr= toLower(line);
 146
 147                         // Find <LOG> TAg? => stop
 148                         if(line.find("<LOG>")!=string::npos)
 149                                 break;
 150
 151                         // Find a STRUCT start?
 152                         if(line.find("<STRUCT>")!=string::npos)
 153                         {
 154                                 lastStructLine= j;
 155                                 raceRestrictionFound= false;
 156                         }
 157
 158                         // Find a RaceRestriction?
 159                         if( line.find("Name=\"Race Restriction\"")!=string::npos )
 160                                 raceRestrictionFound= true;
 161
 162                         // Find the anim name?
 163                         string::size_type       nameIndexInLine= findAnimName(lineLwr, raceAnimNames);
 164                         if(nameIndexInLine!=string::npos)
 165                         {
 166                                 // Find the enclosing struct
 167                                 nlassert(lastStructLine!=0);
 168                                 uint    startBlock= lastStructLine;
 169                                 uint    nameLineInBlock= j-startBlock;
 170                                 uint    endBlock= 0;
 171                                 for(uint k=j+1;k<animSetText.size();k++)
 172                                 {
 173                                         string  line= animSetText[k];
 174
 175                                         // Find a RaceRestriction?
 176                                         if( line.find("Name=\"Race Restriction\"")!=string::npos )
 177                                                 raceRestrictionFound= true;
 178
 179                                         // end of block?
 180                                         if(line.find("</STRUCT>")!=string::npos)
 181                                         {
 182                                                 // endBlock is exclusive
 183                                                 endBlock= k+1;
 184                                                 break;
 185                                         }
 186                                 }
 187
 188                                 // if not found, abort
 189                                 if(endBlock==0)
 190                                         break;
 191
 192                                 // if a raceRestriction has been found, no op (already done)
 193                                 if(raceRestrictionFound)
 194                                 {
 195                                         j= endBlock;
 196                                 }
 197                                 else
 198                                 {
 199                                         // LOG
 200                                         InfoLog->displayRawNL("%s: Specifying %s by race",
 201                                                 CFile::getFilename(animSetFile).c_str(),
 202                                                 CFile::getFilename(animList[i]).c_str());
 203
 204                                         // *** Start a copy paste ^^
 205                                         // Copy
 206                                         static vector<string>   copyText;
 207                                         copyText.clear();
 208                                         for(uint k=startBlock;k<endBlock;k++)
 209                                         {
 210                                                 // add an empty line before </STRUCT>, for race selection node (filled later)
 211                                                 if(k==endBlock-1)
 212                                                         copyText.push_back(string());
 213                                                 copyText.push_back(animSetText[k]);
 214                                         }
 215
 216                                         // erase this part
 217                                         animSetText.erase(animSetText.begin()+startBlock, animSetText.begin()+endBlock);
 218                                         uint    nextBlock= startBlock;
 219
 220                                         // Append for each race
 221                                         for(uint k=0;k<raceAnimNames.size();k++)
 222                                         {
 223                                                 appendRaceAnim(animSetText, nextBlock, copyText, nameLineInBlock, (uint)nameIndexInLine, raceAnimNames[k]);
 224                                                 // nextBlock is then shifted
 225                                                 nextBlock+= (uint)copyText.size();
 226                                         }
 227
 228                                         someChangeDone= true;
 229
 230                                         // *** then let j point to next block
 231                                         j= nextBlock;
 232                                 }
 233                         }
 234                         else
 235                         {
 236                                 j++;
 237                         }
 238                 }
 239         }
 240
 241         // *** Write the animset file.
 242         if(someChangeDone)
 243         {
 244                 COFile  oFile;
 245                 oFile.open(animSetFile, false, true);
 246                 // Write all text
 247                 for(uint i=0;i<animSetText.size();i++)
 248                 {
 249                         string  str= animSetText[i];
 250                         str+= "\n";
 251                         oFile.serialBuffer((uint8*)str.c_str(), (uint)str.size());
 252                 }
 253         }
 254 }
 255
 256
 257
 258 // ***************************************************************************
 259 int usage()
 260 {
 261         printf("Usage: make_anim_by_race  new_anim_dir  animset_dir");
 262         return -1;
 263 }
 264
 265
 266 // ***************************************************************************
 267 int main(int argc, char *argv[])
 268 {
 269         NLMISC::createDebug();
 270
 271         // make_anim_by_race new_anim_dir animset_dir
 272         if(argc!=3)
 273                 return usage();
 274         string animDir= argv[1];
 275         string animSetDir= argv[2];
 276
 277         // Get the list of .anim to make by race
 278         vector<string>  files;
 279         CPath::getPathContent(animDir, false, false, true, files);
 280         // Filter .anim
 281         vector<string>  animList;
 282         InfoLog->displayRawNL("");
 283         InfoLog->displayRawNL("********************");
 284         InfoLog->displayRawNL("**** .anim list ****");
 285         InfoLog->displayRawNL("********************");
 286         for(uint i=0;i<files.size();i++)
 287         {
 288                 if(testWildCard(files[i], "*.anim"))
 289                 {
 290                         animList.push_back(files[i]);
 291                         InfoLog->displayRawNL(animList.back().c_str());
 292                 }
 293
 294         }
 295
 296         // Get the list of .animset to make by race
 297         files.clear();
 298         CPath::getPathContent(animSetDir, true, false, true, files);
 299         vector<string>  animSetList;
 300         InfoLog->displayRawNL("");
 301         InfoLog->displayRawNL("*****************************");
 302         InfoLog->displayRawNL("**** .animation_set list ****");
 303         InfoLog->displayRawNL("*****************************");
 304         for(uint i=0;i<files.size();i++)
 305         {
 306                 if(testWildCard(files[i], "*.animation_set"))
 307                 {
 308                         animSetList.push_back(files[i]);
 309                         InfoLog->displayRawNL(animSetList.back().c_str());
 310                 }
 311         }
 312
 313         InfoLog->displayRawNL("");
 314         InfoLog->displayRawNL("**************************");
 315         InfoLog->displayRawNL("**** Starting Process ****");
 316         InfoLog->displayRawNL("**************************");
 317         // For each animset, test if can replace some anim
 318         for(uint i=0;i<animSetList.size();i++)
 319         {
 320                 makeAnimByRace(animSetList[i], animList);
 321         }
 322
 323         return 0;
 324 }
 325