1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
#include "NvBlastExtDataConverter.h"
#include <string>
#include <iostream>
#include "tclap/CmdLine.h"
#include <stdint.h>
#include <iterator>
#include <fstream>
#include <functional>
#define IDE_DEBUG_MODE 0
int customMain(std::vector<std::string>& args)
{
try
{
// setup cmd line
TCLAP::CmdLine cmd("Blast SDK: Data Converter", ' ', "0.1");
TCLAP::ValueArg<int32_t> outversionArg("v", "outversion", "Output binary block version. Pass -1 or ignore this parameter to convert to latest version.", false, -1, "outversion");
cmd.add(outversionArg);
TCLAP::ValueArg<std::string> outfileArg("o", "outfile", "Output binary file.", true, "", "outfile");
cmd.add(outfileArg);
TCLAP::ValueArg<std::string> infileArg("i", "infile", "Input binary file.", true, "", "infile");
cmd.add(infileArg);
// parse cmd input
cmd.parse(args);
// get cmd parse results
std::string infile = infileArg.getValue();
std::string outfile = outfileArg.getValue();
int32_t outBlockVersion = outversionArg.getValue();
// read input file
std::ifstream infileStream(infile.c_str(), std::ios::binary);
if (!infileStream.is_open())
{
std::cerr << "FAIL: Can't open input file: " << infile << std::endl;
return 0;
}
std::vector<char> inBlock((std::istreambuf_iterator<char>(infileStream)), std::istreambuf_iterator<char>());
infileStream.close();
// convert
std::vector<char> outBlock;
if (Nv::Blast::convertDataBlock(outBlock, inBlock, outBlockVersion >= 0 ? (uint32_t*)&outBlockVersion : nullptr))
{
std::ofstream outfileStream(outfile, std::ios::binary | std::ios::out);
if (!outfileStream.is_open())
{
std::cerr << "FAIL: Can't open output file: " << outfile << std::endl;
return 0;
}
outfileStream.write((char*)&outBlock[0], outBlock.size());
outfileStream.close();
std::cout << "Conversion success, result written in file: " << outfile << std::endl;
}
else
{
std::cout << "Conversion failed." << std::endl;
}
}
catch (TCLAP::ArgException &e) // catch any exceptions
{
std::cout << "error: " << e.error() << " for arg " << e.argId() << std::endl;
}
return 0;
}
int main(int argc, const char* const* argv)
{
std::vector<std::string> args;
#if IDE_DEBUG_MODE
NV_UNUSED(argc);
NV_UNUSED(argv);
args.push_back("");
args.push_back("-i wall.blast");
args.push_back("-o wall_new.blast");
args.push_back("-v 5");
#else
for (int i = 0; i < argc; i++)
args.push_back(argv[i]);
#endif
return customMain(args);
}
|