Setting.h

#ifndef SETTING_H
#define SETTING_H
#define INT long
#define REAL float
#include <cstdio>
#include <cstring>
#include <string>
 
std::string inPath = "../data/FB15K/";
std::string outPath = "../data/FB15K/";
std::string ent_file = "";
std::string rel_file = "";
std::string train_file = "";
std::string valid_file = "";
std::string test_file = "";
 
extern "C" void setInPath(char *path) {
  INT len = strlen(path);
  inPath = "";
  for (INT i = 0; i < len; i++)
    inPath = inPath + path[i];
  printf("Input Files Path : %s\n", inPath.c_str());
}
 
extern "C" void setOutPath(char *path) {
  INT len = strlen(path);
  outPath = "";
  for (INT i = 0; i < len; i++)
    outPath = outPath + path[i];
  printf("Output Files Path : %s\n", outPath.c_str());
}
 
extern "C" void setTrainPath(char *path) {
  INT len = strlen(path);
  train_file = "";
  for (INT i = 0; i < len; i++)
    train_file = train_file + path[i];
  printf("Training Files Path : %s\n", train_file.c_str());
}
 
extern "C" void setValidPath(char *path) {
  INT len = strlen(path);
  valid_file = "";
  for (INT i = 0; i < len; i++)
    valid_file = valid_file + path[i];
  printf("Valid Files Path : %s\n", valid_file.c_str());
}
 
extern "C" void setTestPath(char *path) {
  INT len = strlen(path);
  test_file = "";
  for (INT i = 0; i < len; i++)
    test_file = test_file + path[i];
  printf("Test Files Path : %s\n", test_file.c_str());
}
 
extern "C" void setEntPath(char *path) {
  INT len = strlen(path);
  ent_file = "";
  for (INT i = 0; i < len; i++)
    ent_file = ent_file + path[i];
  printf("Entity Files Path : %s\n", ent_file.c_str());
}
 
extern "C" void setRelPath(char *path) {
  INT len = strlen(path);
  rel_file = "";
  for (INT i = 0; i < len; i++)
    rel_file = rel_file + path[i];
  printf("Relation Files Path : %s\n", rel_file.c_str());
}
 
/*
============================================================
*/
 
INT workThreads = 1;
 
extern "C" void setWorkThreads(INT threads) { workThreads = threads; }
 
extern "C" INT getWorkThreads() { return workThreads; }
 
/*
============================================================
*/
 
INT relationTotal = 0;
INT entityTotal = 0;
INT tripleTotal = 0;
INT testTotal = 0;
INT trainTotal = 0;
INT validTotal = 0;
 
extern "C" INT getEntityTotal() { return entityTotal; }
 
extern "C" INT getRelationTotal() { return relationTotal; }
 
extern "C" INT getTripleTotal() { return tripleTotal; }
 
extern "C" INT getTrainTotal() { return trainTotal; }
 
extern "C" INT getTestTotal() { return testTotal; }
 
extern "C" INT getValidTotal() { return validTotal; }
/*
============================================================
*/
 
INT bernFlag = 0;
 
extern "C" void setBern(INT con) { bernFlag = con; }
 
#endif