#ifndef _TERCPPTERCALC_H___ #define _TERCPPTERCALC_H___ #include #include #include #include #include "hashMap.h" #include "hashMapInfos.h" #include "hashMapStringInfos.h" #include "terAlignment.h" #include "tools.h" #include "terShift.h" #include "alignmentStruct.h" #include "bestShiftStruct.h" using namespace std; using namespace Tools; using namespace HashMapSpace; namespace TERCpp { // typedef size_t WERelement[2]; // Vecteur d'alignement contenant le hash du mot et son evaluation (0=ok, 1=sub, 2=ins, 3=del) typedef vector vecTerShift; /** @author */ class terCalc { private : // Vecteur d'alignement contenant le hash du mot et son evaluation (0=ok, 1=sub, 2=ins, 3=del) WERalignment l_WERalignment; // HashMap contenant les caleurs de hash de chaque mot hashMap bagOfWords; int MAX_SHIFT_SIZE; /* Variables for some internal counting. */ int NUM_SEGMENTS_SCORED; int NUM_SHIFTS_CONSIDERED; int NUM_BEAM_SEARCH_CALLS; int MAX_SHIFT_DIST; bool PRINT_DEBUG; /* These are resized by the MIN_EDIT_DIST code if they aren't big enough */ double S[1000][1000]; char P[1000][1000]; vector refSpans; vector hypSpans; int BEAM_WIDTH; public: int shift_cost; int insert_cost; int delete_cost; int substitute_cost; int match_cost; double INF; terCalc(); // ~terCalc(); // size_t* hashVec ( vector s ); void setDebugMode ( bool b ); int WERCalculation ( size_t * ref, size_t * hyp ); int WERCalculation ( vector ref, vector hyp ); int WERCalculation ( vector ref, vector hyp ); // string vectorToString(vector vec); // vector subVector(vector vec, int start, int end); hashMapInfos BuildWordMatches ( vector hyp, vector ref ); terAlignment MinEditDist ( vector hyp, vector ref, vector curHypSpans ); bool spanIntersection ( vecInt refSpan, vecInt hypSpan ); terAlignment TER ( vector hyp, vector ref , float avRefLength ); terAlignment TER ( vector hyp, vector ref ); terAlignment TER ( vector hyp, vector ref ); bestShiftStruct CalcBestShift ( vector cur, vector hyp, vector ref, hashMapInfos rloc, terAlignment cur_align ); void FindAlignErr ( terAlignment align, bool* herr, bool* rerr, int* ralign ); vector GatherAllPossShifts ( vector hyp, vector ref, hashMapInfos rloc, terAlignment align, bool* herr, bool* rerr, int* ralign ); alignmentStruct PerformShift ( vector words, terShift s ); alignmentStruct PerformShift ( vector words, int start, int end, int newloc ); }; } #endif // _TERCPPTERCALC_H___