/*Copyright 2014 Francisco Alvaro

 This file is part of SESHAT.

    SESHAT is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    SESHAT is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with SESHAT.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef _SYMREC_
#define _SYMREC_

struct SegmentHyp;
class Sample;
class SymFeatures;

#include <cstdio>
#include <string>
#include <cstring>
#include <map>
#include <vector>
#include "rnnlib4seshat/DataSequence.hpp"
#include "rnnlib4seshat/NetcdfDataset.hpp"
#include "rnnlib4seshat/Mdrnn.hpp"
#include "rnnlib4seshat/MultilayerNet.hpp"
#include "rnnlib4seshat/Rprop.hpp"
#include "rnnlib4seshat/SteepestDescent.hpp"
#include "rnnlib4seshat/Trainer.hpp"
#include "rnnlib4seshat/WeightContainer.hpp"
#include "sample.h"
#include "symfeatures.h"

using namespace std;


class SymRec{
  SymFeatures *FEAS;
  DataHeader header_on, header_off;
  DataExportHandler deh_on, deh_off;
  WeightContainer *wc_on, *wc_off;
  Mdrnn *blstm_on, *blstm_off;
  float RNNalpha;
  
  //Symbol classes and types information
  int *type;
  map<string,int> cl2key;
  string *key2cl;
  
  int C; //Number of classes
  
  int  classify(Sample *M, SegmentHyp *SegHyp, const int NB, int *vclase, float *vpr, int *as, int *ds);
  void BLSTMclassification( Mdrnn *net, DataSequence *seq, pair<float,int> *claspr, const int NB );
  
 public:
  SymRec(char *path);
  ~SymRec();
  
  char *strClase(int c);
  int   keyClase(char *str);
  bool  checkClase(char *str);
  int   getNClases();
  int   symType(int k);
  
  int clasificar(Sample *M, int ncomp,     const int NB, int *vclase, float *vpr, int *as, int *ds);
  int clasificar(Sample *M, list<int> *LT, const int NB, int *vclase, float *vpr, int *as, int *ds);
};


#endif