simpleEvaluator.cpp

Go to the documentation of this file.

#include <fstream>
#include <stdio.h>
#include <unistd.h>
#include <sys/time.h>

#include <ISM/TableHelper.hpp>
#include <ISM/Recognizer.hpp>

#define ISM_DATA "/media/share/data"

using namespace ISM;
using namespace std;

void drawISM(string model, string patternName, map<string, map<string, unsigned> >graphVizData, unsigned testsPerformed, unsigned timeSteps)
{
  if (testsPerformed == 0)
    {
      testsPerformed = 1;
    }
  ofstream file;
  stringstream filePath;
  filePath<<std::string(ISM_DATA)+"/evaluation/gedeck_evaluation/complexity/"<<patternName<<"_"<<model<<"_ISMWithVotes.dot";
  string filename = filePath.str();
  filePath.str("");
  vector<pair<string, string> > alreadyTakenCombinations;
  ios_base::iostate exceptionMask = file.exceptions() | ios::failbit | ios::badbit;
  file.exceptions(exceptionMask);
  //map<string, set<string> > referencesPerObject;
  try
    {
      file.open(filename);
      file<<"digraph "<<model<<" {\n";
      for (auto type : graphVizData)
        {
          for (auto voteIt : type.second)
            {
              file<<voteIt.first<<"[shape=\"box\"];\n";
              //referencesPerObject[type.first].insert(voteIt.first);
            }
        }
      for (auto typeIt : graphVizData)
        {
          for (auto voteIt : typeIt.second)
            {
              if (find(alreadyTakenCombinations.begin(), alreadyTakenCombinations.end(), make_pair(voteIt.first, typeIt.first)) == alreadyTakenCombinations.end())
                {
                  //cout<<"from "<<typeIt.first<<" to "<<voteIt.first<<": "<<voteIt.second<<endl;exit(0);
                  file<<voteIt.first<<" -> "<<typeIt.first<<"[ label=\" &#8746;"
                      <<(unsigned)((((double)voteIt.second) / testsPerformed) / timeSteps)<<"&#124;&#8594;"
                    //<<(unsigned)((((double)typeIt.second.size() / testsPerformed) / timeSteps) / referencesPerObject.at(typeIt.first).size())<<"&#124;&#8594;"
                      <<(unsigned)(((double)voteIt.second) / testsPerformed)
                    //<<(unsigned)(((double)typeIt.second.size() / testsPerformed) / referencesPerObject.at(typeIt.first).size())
                      <<"\", decorate=\"true\", labelfoat=\"true\", labelfontcolor=\"red\", dir=\"back\"];\n";
                  alreadyTakenCombinations.push_back(make_pair(voteIt.first, typeIt.first));
                }
            }     
        }
      file<<"}\n\n";
      file.flush();
      file.close();
    }
  catch (ios_base::failure& e)
    {
      cerr<<e.what()<<"\n";
      //exit(1);
    }
}

void addVizData(std::map<std::string, std::map<std::string, unsigned> >& graphVizData, const RecognizerPtr& recognizer)
{
  std::map<ObjectPtr, std::vector<VotedPosePtr> > votingCache = recognizer->getVotingCache();
  for (auto it : votingCache)
    {
      for (auto voteIt : it.second)
        {
          ++graphVizData[it.first->getType()][voteIt->vote->patternName];
        }
    }    
}


int main (int argc, char**argv)
{
  std::vector<std::pair<std::string, std::string> > modeldbs = 
    {
      std::make_pair("best", std::string(ISM_DATA)+"/evaluation/gedeck_evaluation/sourcedbs/record.sqlite.optedTrainer_example_1_best"),
      std::make_pair("best_naive", std::string(ISM_DATA)+"/evaluation/gedeck_evaluation/sourcedbs/record.sqlite.optedTrainer_example_1_best_naive"),
      std::make_pair("star", std::string(ISM_DATA)+"/evaluation/gedeck_evaluation/sourcedbs/record.sqlite.optedTrainer_example_1_star"),
      std::make_pair("fully_meshed", std::string(ISM_DATA)+"/evaluation/gedeck_evaluation/sourcedbs/record.sqlite.optedTrainer_example_1_fullyMeshed"),
      std::make_pair("fully_meshed_naive",std::string(ISM_DATA)+"/evaluation/gedeck_evaluation/sourcedbs/record.sqlite.optedTrainer_example_1_fullyMeshed_naive")
    };
  ofstream output;
  output.open(std::string(ISM_DATA)+"/evaluation/gedeck_evaluation/output.csv");
  output<<"model, avgTime, falsePositives\n";      
   for (auto model : modeldbs)
    {
      std::map<std::string, std::map<std::string, unsigned> > vizData;
      cout<<"trying to open db file\n";
      TableHelperPtr validSetsTable(new TableHelper(std::string(ISM_DATA)+"/evaluation/gedeck_evaluation/sourcedbs/record.sqlite.optedTrainer_example_1_validTestSets"));
      cout<<"could open valid sets\n";
      auto validPatterns = validSetsTable->getRecordedPattern("optedTrainer_example_1");
      vector<ObjectSetPtr> validSets;
      if (validPatterns != 0)
        {
          validSets = validPatterns->objectSets;
        }
      struct timeval start, end;
      long time = 0;
      TableHelperPtr invalidSetsTable(new TableHelper(std::string(ISM_DATA)+"/evaluation/gedeck_evaluation/sourcedbs/record.sqlite.optedTrainer_example_1_invalidTestSets"));
      cout<<"could open invalid sets\n ";
      auto invalidPatterns = invalidSetsTable->getRecordedPattern("optedTrainer_example_1");
      vector<ObjectSetPtr> invalidSets;
      if (invalidPatterns != 0)
        {
          invalidSets = invalidPatterns->objectSets;
        }  
      cout<<invalidSets.size()<<endl;
      RecognizerPtr recog(new Recognizer(model.second, 0.03, 10, false));
      cout<<"could create recognizer\n";
      for (auto validSet : validSets)
        {
          if (model.first == "fully_meshed_naive")
            {
              break;
            }
          cout<<".";cout.flush();
          gettimeofday(&start, NULL);
          auto res = recog->recognizePattern(validSet, 0.0, 1);
          gettimeofday(&end, NULL);
          time += ((end.tv_sec - start.tv_sec) * 1000) + ((end.tv_usec - start.tv_usec) / 1000);
          addVizData(vizData, recog);
        }
      unsigned fps = 0;
      for (auto invalidSet : invalidSets)
        {
          cout<<".";cout.flush();
          gettimeofday(&start, NULL);
          auto res = recog->recognizePattern(invalidSet, 0.0, 1);
          gettimeofday(&end, NULL);
          time += ((end.tv_sec - start.tv_sec) * 1000) + ((end.tv_usec - start.tv_usec) / 1000);
          addVizData(vizData, recog);
          if (res.size() > 0 && res[0]->getConfidence() == 1 && res[0]->getPatternName() == "optedTrainer_example_1")
            {
              ++fps;
            }
          if (model.first == "fully_meshed_naive")
            {
              break;
            }
        }
      double avg;
      if (model.first == "fully_meshed_naive")
        {
          drawISM(model.first, "optedTrainer_example_1", vizData, 1, 112.0);
          avg = ((double)time) / 1.0;
        }
      else
        {
          drawISM(model.first, "optedTrainer_example_1", vizData, validSets.size() + invalidSets.size(), 112.0);
          avg = ((double)time) / (validSets.size() + invalidSets.size());
        }
      cout<<fps<<endl;
      output<<model.first<<","<<avg<<","<<fps<<endl;
    }
  output.close();
}