common-helpers.hpp

Go to the documentation of this file.

#pragma once

#include <registrypo.hpp>
#include <kenlmdetect.hpp>

namespace ucam {
namespace fsttools {

template <
  template <template <class> class
            , class> class TaskOneT
  , template <template <class> class
              , class> class TaskTwoT
  , template <class> class DataT
  , class ArcT
  >
struct RunTask2 {
  explicit RunTask2(util::RegistryPO const &rg){
  using util::Runner2;
  ( Runner2<
    TaskOneT< DataT, ArcT>
    , TaskTwoT< DataT, ArcT>
    >
    ( rg ) ) ();
  }
};

template <
  template <template <class> class
            , class> class TaskOneT
  , template <template <class> class
              , class> class TaskTwoT
  , template <template <class> class
              , class> class TaskThreeT
  , template <class> class DataT
  , class ArcT
  >
struct RunTask3 {
  explicit RunTask3(util::RegistryPO const &rg){
  using util::Runner3;
  ( Runner3<
    TaskOneT< DataT, ArcT>
    , TaskTwoT< DataT, ArcT>
    , TaskThreeT< DataT, ArcT>
    >
    ( rg ) ) ();
  }
};

#ifdef KENLM_MAX_ORDER

template< template < template <class, class> class
                     , class
                     , class> class RunTaskT
          , template <class, class> class DataT
          , class ArcT
          >
inline void runTaskWithKenLMTemplate(ucam::util::RegistryPO const &rg) {
  using namespace lm::ngram;
  using namespace ucam::util;
  // Detect here kenlm binary type
  // it's a bit ugly this way of initializing the correct kenlm handler
  ModelType kenmt = ucam::util::detectkenlm
      (rg.getVectorString (HifstConstants::kLmLoad, 0) );

  switch (kenmt) {
    case PROBING:
      (RunTaskT<DataT, ProbingModel, ArcT>(rg));
      break;
    case REST_PROBING:
      (RunTaskT<DataT, RestProbingModel, ArcT>(rg));
      break;
    case TRIE:
      (RunTaskT<DataT, TrieModel, ArcT>(rg));
    break;
    case QUANT_TRIE:
      (RunTaskT<DataT, QuantTrieModel, ArcT>(rg));
      break;
    case ARRAY_TRIE:
      (RunTaskT<DataT, ArrayTrieModel, ArcT>(rg));
      break;
    case QUANT_ARRAY_TRIE:
      (RunTaskT<DataT, QuantArrayTrieModel, ArcT>(rg));
      break;
    case KENLM_NPLM:
#ifdef WITH_NPLM
      (RunTaskT<DataT, lm::np::Model, ArcT>(rg));
    break;
#endif
    std::cerr << "Unsuported format: KENLM_NPLM. Did you compile NPLM library?" << std::endl;
    exit(EXIT_FAILURE);
  }
};
#endif

}}  // end namespaces