khanat-code-old/code/ryzom/tools/pdr_util/pdr_util.cpp
2010-05-27 07:31:08 -05:00

281 lines
6.5 KiB
C++

#include "nel/misc/types_nl.h"
#include "nel/misc/path.h"
#include "nel/misc/sheet_id.h"
#include <stdio.h>
#include <vector>
#include "game_share/persistent_data.h"
using namespace std;
using namespace NLMISC;
int main(int argc, char *argv[])
{
NLMISC::CApplicationContext context;
if (argc == 1)
{
printf("Usage : %s [-s<sheet_id_path>] [-x|-b|-t] [-o<outputFileName>] inputFileName\n", argv[0]);
printf(" -s : must contains the path where the sheet_id.bin can be found\n");
printf(" You could ignore this param is the sheet_id.bin can be found in the local directory\n");
printf(" -x : convert a binary pdr to XML format (exclude -b -t)\n");
printf(" -t : convert a binary pdr to text line format (exclude -b -x)\n");
printf(" -b : convert a XML or text pdr to binary format (exclude -x -t)\n");
printf(" -o : output filename\n");
return -1;
}
enum TConvMode
{
cm_undefined,
cm_to_xml,
cm_to_binary,
cm_to_txt,
};
enum TSourceFormat
{
sf_undefined,
sf_xml,
sf_txt
};
TConvMode mode = cm_undefined;
TSourceFormat sourceFormat = sf_undefined;
string sheetIdPath;
string fileName;
vector<string> filenames;
string outputFileName;
vector<string> args(argv, argv+argc);
for (uint i=1; i<args.size(); ++i)
{
if (args[i].size() >= 2 && args[i][0] == '-')
{
string paramValue = args[i].substr(2);
switch (args[i][1])
{
case 's':
sheetIdPath = paramValue;
break;
case 'x':
mode = cm_to_xml;
break;
case 't':
mode = cm_to_txt;
break;
case 'b':
mode = cm_to_binary;
break;
case 'o':
outputFileName = paramValue;
default:
fprintf(stderr, "Unknown parameter '%s'", args[i].c_str());
return -1;
}
}
else if(!args[i].empty())
{
filenames.push_back(args[i]);
}
/*
if (i == args.size()-1)
{
// last param, must be the filename
if (args[i].empty() || args[i][0] == '-')
{
fprintf(stderr, "Invalid or missing filename '%s'\n", args[i].c_str());
return -1;
}
fileName = args[i];
}
else
{
if (args[i].empty() || args[i][0] != '-' || args[i].size() < 2)
{
fprintf(stderr, "Invalid param '%s'\n", args[i].c_str());
return -1;
}
string paramValue = args[i].substr(2);
switch (args[i][1])
{
case 's':
sheetIdPath = paramValue;
break;
case 'x':
mode = cm_to_xml;
break;
case 't':
mode = cm_to_txt;
break;
case 'b':
mode = cm_to_binary;
break;
case 'o':
outputFileName = paramValue;
default:
fprintf(stderr, "Unknown parameter '%s'", args[i].c_str());
return -1;
}
}
*/
}
// init the sheet id
if (!sheetIdPath.empty())
{
CPath::addSearchPath(sheetIdPath, false, false);
}
CSheetId::init(false);
for(int f = 0; f < filenames.size(); f++)
{
fileName = filenames[f];
if (!CFile::isExists(fileName))
{
fprintf(stderr, "Couldn't find file '%s'", fileName.c_str());
// return -1;
continue;
}
if (mode == cm_undefined)
{
// try to automatically determine conversion mode
if (fileName.find(".xml") == (fileName.size() - 4))
{
printf("Choosing XML->BINARY conversion mode");
mode = cm_to_binary;
}
else if (fileName.find(".txt") == (fileName.size() -4))
{
printf("Choosing TXT->BINARY conversion mode");
mode = cm_to_binary;
}
else if (fileName.find(".bin") == (fileName.size() -4))
{
printf("Choosing BINARY->XML conversion mode");
mode = cm_to_xml;
}
else
{
fprintf(stderr, "Missing conversion mode flag (-x|-b|-t) and can't deduce mode from filename extension");
// return -1;
continue;
}
}
// determine source format when concerting to binary
if (mode == cm_to_binary)
{
if (fileName.find(".xml") == (fileName.size() - 4))
{
printf("Source file is in XML format");
sourceFormat = sf_xml;
}
/* else if (fileName.find(".txt") == (fileName.size() -4))
{
printf("Source file is in TXT format");
sourceFormat = sf_txt;
}
*/ else
{
fprintf(stderr, "Invalid source format, only support '.xml' files");
// return -1;
continue;
}
}
if (outputFileName.empty() || filenames.size() > 1)
{
// build a output name
outputFileName = fileName;
string inExt;
string outExt;
if (mode == cm_to_binary)
{
if (sourceFormat == sf_txt)
inExt = ".txt";
else
inExt = ".xml";
outExt = ".bin";
}
else if (mode == cm_to_xml)
{
outExt = ".xml";
inExt = ".bin";
}
else if (mode == cm_to_txt)
{
outExt = ".txt";
inExt = ".bin";
}
if (outputFileName.find(inExt) == (outputFileName.size()-inExt.size()))
{
// remove input ext from output filename
outputFileName = outputFileName.substr(0, outputFileName.size()-inExt.size());
}
// append output extension
outputFileName += outExt;
}
static CPersistentDataRecord pdr;
pdr.clear();
switch(mode)
{
case cm_to_binary:
if (sourceFormat == sf_txt)
{
printf("Converting from txt to bin is currently unpossible ! use xml format");
// return -1;
continue;
// printf("Converting '%s' (TXT) to '%s' (BINARY)\n", fileName.c_str(), outputFileName.c_str() );
}
else
printf("Converting '%s' (XML) to '%s' (BINARY)\n", fileName.c_str(), outputFileName.c_str() );
if (!pdr.readFromTxtFile(fileName.c_str()))
goto failureRead;
if (!pdr.writeToBinFile(outputFileName.c_str()))
goto failureWrite;
break;
case cm_to_xml:
printf("Converting '%s' (BINARY) to '%s' (XML)\n", fileName.c_str(), outputFileName.c_str() );
if (!pdr.readFromBinFile(fileName.c_str()))
goto failureRead;
if (!pdr.writeToTxtFile(outputFileName.c_str(), CPersistentDataRecord::XML_STRING))
goto failureWrite;
break;
case cm_to_txt:
printf("Converting '%s' (BINARY) to '%s' (TXT)\n", fileName.c_str(), outputFileName.c_str() );
if (!pdr.readFromBinFile(fileName.c_str()))
goto failureRead;
if (!pdr.writeToTxtFile(outputFileName.c_str(), CPersistentDataRecord::LINES_STRING))
goto failureWrite;
break;
}
// return 0;
continue;
failureRead:
fprintf(stderr, "Error while reading '%s', conversion aborted", fileName.c_str());
// return -1;
continue;
failureWrite:
fprintf(stderr, "Error while writing '%s', conversion aborted", outputFileName.c_str());
// return -1;
continue;
}
}