cMT.h

   1 // Machine Translation driver
   2 // Copyright © 2009 The University of Chicago
   3 #ifndef CMT_H
   4 #define CMT_H
   5
   6 #include <QString>
   7
   8 class LinguisticaMainWindow;
   9 class mTVolca;
  10 class cMTModel1;
  11 class cMTModel2Norm;
  12 class mTForSortingItem;
  13
  14 template<class V> class Q3SortedList;
  15 template<class K, class V> class QMap;
  16
  17 typedef Q3SortedList<mTForSortingItem> mTSortedList;
  18 typedef QMap<int, double> IntToDouble;
  19 typedef QMap<int, IntToDouble*> IntToIntToDouble;
  20
  21 class cMT {
  22 public:
  23         LinguisticaMainWindow* m_parent;
  24         QString m_projectDirectory;
  25
  26         mTVolca* m_Volca;
  27
  28         /// Model 1
  29         cMTModel1* m_Model1;
  30         /// Model 2
  31         cMTModel2Norm* m_Model2Norm;
  32
  33         /// Log parameter
  34         QString m_MTLog;
  35 public:
  36         // construction/destruction.
  37
  38         cMT(LinguisticaMainWindow*, QString);
  39         virtual ~cMT();
  40
  41         // disable default-construction, copy
  42 private:
  43         cMT();
  44         cMT(const cMT& x);
  45         cMT& operator=(const cMT& x);
  46 public:
  47         // training.
  48
  49         void readTrainingCorpus();
  50         /// Model 1
  51         void trainModel1(int);
  52         /// Model 2
  53         void trainModel2Norm(int, bool);
  54
  55         // output to logs.
  56
  57         /// Log Model1
  58         void logTAfterModel1();
  59         /// Log Model2
  60         void logTandAAfterModel2Norm();
  61 };
  62
  63 class mTForSortingItem {
  64 public:
  65         QString m_name;
  66         int m_type;
  67         double m_doubleValue;
  68 public:
  69         // construction/destruction.
  70
  71         mTForSortingItem() : m_name(), m_type(0), m_doubleValue(0.0) { }
  72         mTForSortingItem(QString name, double value, int type)
  73                 : m_name(name), m_type(type),
  74                 m_doubleValue(m_type == 1 ? value : 0.0) { }
  75         // destructor implicitly defined
  76
  77         // copy construction/assignment implicitly defined
  78
  79         // compare.
  80
  81         bool operator<(const mTForSortingItem& x)
  82         {
  83                 return m_type == 1 ?
  84                         m_doubleValue > x.m_doubleValue :
  85                         true;
  86         }
  87         bool operator==(const mTForSortingItem& x)
  88         {
  89                 return m_type == 1 ?
  90                         m_doubleValue == x.m_doubleValue :
  91                         true;
  92         }
  93 };
  94
  95 #endif // CMT_H