// // main.cpp // tools/shaderScribe/src // // Created by Sam Gateau on 12/15/2014. // Copyright 2013 High Fidelity, Inc. // // Distributed under the Apache License, Version 2.0. // See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html #include "TextTemplate.h" #include #include #include #include using namespace std; int main (int argc, char** argv) { // process the command line arguments std::vector< std::string > inputs; std::string srcFilename; std::string destFilename; std::string targetName; TextTemplate::Vars vars; std::string lastVarName; bool listVars = false; bool showParseTree = false; bool makeCPlusPlus = false; auto config = std::make_shared(); enum Mode { READY = 0, GRAB_OUTPUT, GRAB_VAR_NAME, GRAB_VAR_VALUE, GRAB_INCLUDE_PATH, GRAB_TARGET_NAME, EXIT, } mode = READY; for (int ii = 1; (mode != EXIT) && (ii < argc); ii++) { inputs.push_back(argv[ii]); switch (mode) { case READY: { if (inputs.back() == "-o") { mode = GRAB_OUTPUT; } else if (inputs.back() == "-t") { mode = GRAB_TARGET_NAME; } else if (inputs.back() == "-D") { mode = GRAB_VAR_NAME; } else if (inputs.back() == "-I") { mode = GRAB_INCLUDE_PATH; } else if (inputs.back() == "-listVars") { listVars = true; mode = READY; } else if (inputs.back() == "-showParseTree") { showParseTree = true; mode = READY; } else if (inputs.back() == "-c++") { makeCPlusPlus = true; mode = READY; } else { // just grabbed the source filename, stop parameter parsing srcFilename = inputs.back(); mode = EXIT; } } break; case GRAB_OUTPUT: { destFilename = inputs.back(); mode = READY; } break; case GRAB_TARGET_NAME: { targetName = inputs.back(); mode = READY; } break; case GRAB_VAR_NAME: { // grab first the name of the var lastVarName = inputs.back(); mode = GRAB_VAR_VALUE; } break; case GRAB_VAR_VALUE: { // and then the value vars.insert(TextTemplate::Vars::value_type(lastVarName, inputs.back())); mode = READY; } break; case GRAB_INCLUDE_PATH: { config->addIncludePath(inputs.back().c_str()); mode = READY; } break; case EXIT: { // THis shouldn't happen } break; } } if (srcFilename.empty()) { cerr << "Usage: shaderScribe [OPTION]... inputFilename" << endl; cerr << "Where options include:" << endl; cerr << " -o filename: Send output to filename rather than standard output." << endl; cerr << " -t targetName: Set the targetName used, if not defined use the output filename 'name' and if not defined use the inputFilename 'name'" << endl; cerr << " -I include_directory: Declare a directory to be added to the includes search pool." << endl; cerr << " -D varname varvalue: Declare a var used to generate the output file." << endl; cerr << " varname and varvalue must be made of alpha numerical characters with no spaces." << endl; cerr << " -listVars : Will list the vars name and value in the standard output." << endl; cerr << " -showParseTree : Draw the tree obtained while parsing the source" << endl; cerr << " -c++ : Generate a c++ header file containing the output file stream stored as a char[] variable" << endl; return 0; } // Define targetName: if none, get destFilenmae, if none get srcFilename if (targetName.empty()) { if (destFilename.empty()) { targetName = srcFilename; } else { targetName = destFilename; } } // no clean it to have just a descent c var name if (!targetName.empty()) { // trim anything before '/' or '\' targetName = targetName.substr(targetName.find_last_of('/') + 1); targetName = targetName.substr(targetName.find_last_of('\\') + 1); // trim anything after '.' targetName = targetName.substr(0, targetName.find_first_of('.')); } // Add built in vars time_t endTime = chrono::system_clock::to_time_t(chrono::system_clock::now()); std::string endTimStr(ctime(&endTime)); vars["_SCRIBE_DATE"] = endTimStr.substr(0, endTimStr.length() - 1); // List vars? if (listVars) { cerr << "Vars:" << endl; for (auto v : vars) { cerr << " " << v.first << " = " << v.second << endl; } } // Open up source std::fstream srcStream; srcStream.open(srcFilename, std::fstream::in); if (!srcStream.is_open()) { cerr << "Failed to open source file <" << srcFilename << ">" << endl; return 0; } auto scribe = std::make_shared(srcFilename, config); // ready to parse and generate std::stringstream destStringStream; int numErrors = scribe->scribe(destStringStream, srcStream, vars); if (numErrors) { cerr << "Scribe " << srcFilename << "> failed: " << numErrors << " errors." << endl; return 0; }; if (showParseTree) { int level = 1; cerr << "Config trees:" << std::endl; config->displayTree(cerr, level); cerr << srcFilename << " tree:" << std::endl; scribe->displayTree(cerr, level); } std::ostringstream targetStringStream; if (makeCPlusPlus) { // Because there is a maximum size for literal strings declared in source we need to partition the // full source string stream into pages that seems to be around that value... const int MAX_STRING_LITERAL = 10000; std::string lineToken; auto pageSize = lineToken.length(); std::vector> pages(1, std::make_shared()); while (!destStringStream.eof()) { std::getline(destStringStream, lineToken); auto lineSize = lineToken.length() + 1; if (pageSize + lineSize > MAX_STRING_LITERAL) { pages.push_back(std::make_shared()); // reset pageStringStream pageSize = 0; } (*pages.back()) << lineToken << std::endl; pageSize += lineSize; } targetStringStream << "// File generated by Scribe " << vars["_SCRIBE_DATE"] << std::endl; targetStringStream << "#ifndef scribe_" << targetName << "_h" << std::endl; targetStringStream << "#define scribe_" << targetName << "_h" << std::endl << std::endl; targetStringStream << "const char " << targetName << "[] = \n"; // Write the pages content for (auto page : pages) { targetStringStream << "R\"SCRIBE(\n" << page->str() << "\n)SCRIBE\"\n"; } targetStringStream << ";\n" << std::endl << std::endl; targetStringStream << "#endif" << std::endl; } else { targetStringStream << destStringStream.str(); } // Destination stream if (!destFilename.empty()) { std::fstream destFileStream; destFileStream.open(destFilename, std::fstream::out); if (!destFileStream.is_open()) { cerr << "Scribe output file " << destFilename << "> failed to open." << endl; return 0; } destFileStream << targetStringStream.str(); } else { cerr << targetStringStream.str(); } return 0; }