Method write

src/codegen/cppCodeGen.cpp:32–67

#include <unistd.h>

bool dnnc::cppCodeGen::write() {

  inferDataType typeInference(_graph);
  typeInference.main();

  std::string code = "";

  for (dnnParameters param : _graph.parameters()) {
    code += write(param);
  }
  size_t argv_index = 1;
  for (ioNode *term : _graph.inputs()) {
    code += write(*term, argv_index);
  }

  for (node *n : _graph) {
    if (n->ntype() == node::OPERATOR)
      code += write(*dynamic_cast<opNode *>(n));
  }

  // OUTPUTs are written with operators.

  std::ofstream out(_bundleDir.size()
                        ? (_bundleDir + FS_PATH_SEPARATOR + _outFile)
                        : _outFile);
  if (!out.is_open() || out.fail()) {
    std::cerr << "ERROR (CODEGEN): could not open file " + _outFile +
                     " to write.\n";
    return false;
  }
  out << writeIncludes() << "\n";
  out << writeUsageFunction() << "\n";
  out << writeMainFunction(code) << "\n";
  out.close();
  return code.length();
}

// \brief replace all characters that can't appear
// in C++ variable name.