Skip to content
Snippets Groups Projects
ReadingMachine.cpp 2.23 KiB
Newer Older
#include "ReadingMachine.hpp"
#include "util.hpp"

Franck Dary's avatar
Franck Dary committed
ReadingMachine::ReadingMachine(std::filesystem::path path) : path(path)
Franck Dary's avatar
Franck Dary committed
  dicts.emplace(std::make_pair("", Dict::State::Open));

Franck Dary's avatar
Franck Dary committed
  std::FILE * file = std::fopen(path.c_str(), "r");
Franck Dary's avatar
Franck Dary committed
  std::string fileContent;
  std::vector<std::string> lines;
  while (!std::feof(file))
  {
    if (buffer != std::fgets(buffer, 1024, file))
      break;
    // If line is blank or commented (# or //), ignore it
    if (util::doIfNameMatch(std::regex("((\\s|\\t)*)(((#|//).*)|)(\n|)"), buffer, [](auto){}))
      continue;
    if (buffer[std::strlen(buffer)-1] == '\n')
      buffer[std::strlen(buffer)-1] = '\0';

    lines.emplace_back(buffer);
Franck Dary's avatar
Franck Dary committed

  try
  {
    unsigned int curLine = 0;
    if (!util::doIfNameMatch(std::regex("Name : (.+)"), lines[curLine++], [this](auto sm){name = sm[1];}))
      util::myThrow("No name specified");

    while (util::doIfNameMatch(std::regex("Classifier : (.+) (.+) (.+)"), lines[curLine++], [this](auto sm){classifier.reset(new Classifier(sm[1], sm[2], sm[3]));}));
    if (!classifier.get())
      util::myThrow("No Classifier specified");

    --curLine;
    //std::vector<std::string_view> restOfFile;
    //while (curLine < lines.size() and !util::doIfNameMatch(std::regex("Strategy(.*)"),lines[curLine], [](auto){}))
    //  restOfFile.emplace_back(lines[curLine++]);

    //featureFunction.reset(new FeatureFunction(restOfFile));

    auto restOfFile = std::vector<std::string_view>(lines.begin()+curLine, lines.end());

Franck Dary's avatar
Franck Dary committed
    strategy.reset(new Strategy(restOfFile));

Franck Dary's avatar
Franck Dary committed
  } catch(std::exception & e) {util::myThrow(fmt::format("during reading of '{}' : {}", path.string(), e.what()));}
}

ReadingMachine::ReadingMachine(const std::string & filename, const std::vector<std::string> & models, const std::vector<std::string> & dicts)
{

TransitionSet & ReadingMachine::getTransitionSet()
{
  return classifier->getTransitionSet();
}

Strategy & ReadingMachine::getStrategy()
{
  return *strategy;
}

Franck Dary's avatar
Franck Dary committed
Dict & ReadingMachine::getDict(const std::string & state)
{
  auto found = dicts.find(state);

  if (found == dicts.end())
    return dicts.at("");

  return found->second;
}

Classifier * ReadingMachine::getClassifier()
{
  return classifier.get();
}