DataGenerator.h

Go to the documentation of this file.

#ifndef DATAGENERATOR_H
#define DATAGENERATOR_H
 
// will be using pugiXML for parsing and generating
// git clone https://github.com/zeux/pugixml.git
// forked repo and uncomment in pugixml/src/pugiconfig.hpp
// line 48 #define PUGIXML_HEADER_ONLY
 
#include "../../extsrc/pugixml/src/pugixml.hpp"
#include "../app.h"
#include "AlternativesPerformance.h"
#include "Criterion.h"
#include "PerformanceTable.h"
#include <iostream>
#include <tuple>
 
class DataGenerator {
public:
  DataGenerator(Config &config);
 
  void datasetGenerator(int nb_criteria, int nb_alternative, int nb_categories,
                        std::string datasetName = "", bool overwrite = 1,
                        unsigned long int seed = 0);
 
  void modelGenerator(int nb_criteria, int nb_categories, std::string modelName,
                      bool overwrite = 1, unsigned long int seed = 0);
 
  void saveDataset(std::string fileName, AlternativesPerformance altPerf,
                   int nb_categories, bool overwrite = 1,
                   std::string datasetName = "");
 
  AlternativesPerformance loadDataset(std::string fileName);
 
  void saveModel(std::string fileName, float lambda, Criteria criteria,
                 PerformanceTable pt, bool overwrite = 1,
                 std::string modelName = "");
 
  std::tuple<float, Criteria, PerformanceTable> loadModel(std::string fileName);
 
  pugi::xml_document openXmlFile(std::string filename);
 
  std::string getXmlFileType(std::string fileName);
 
  int getNumberOfCriteria(std::string fileName);
 
  int getNumberOfCategories(std::string fileName);
 
  float getThresholdValue(std::string fileName);
 
  int getNumberOfAlternatives(std::string fileName);
 
  std::vector<Perf> getAlternativePerformance(std::string fileName,
                                              std::string alt_id);
  std::vector<std::string> getAlternativeIds(std::string fileName);
 
  std::vector<std::string> getCriteriaIds(std::string fileName);
 
  Criterion getCriterion(std::string fileName, std::string crit_id);
 
  int getAlternativeAssignment(std::string fileName, std::string alt_id);
 
  std::vector<float> getCriterionCategoryLimits(std::string fileName,
                                                std::string crit_id);
 
  bool checkDataCompatability(std::string fileName);
 
private:
  Config &conf;
};
 
#endif