libvspell/penalty.h

   1 #ifndef __PENALTY_H__ // -*- tab-width: 2 mode: c++ -*-
   2 #define __PENALTY_H__
   3
   4 #ifndef __DAG_H__
   5 #include "dag.h"
   6 #endif
   7
   8 class PenaltyDAG : public DAG {
   9 protected:
  10         /// requires that dag->get_info() return a WordEntry*
  11         DAG *dag;
  12         float weight;
  13
  14 public:
  15         PenaltyDAG(DAG *dag_,float weight_):dag(dag_),weight(weight_) {}
  16
  17         void set_weight(float weight_) {
  18                 weight = weight_;
  19         }
  20
  21         float get_weight() const {
  22                 return weight;
  23         }
  24
  25         virtual uint node_begin() const {
  26                 return dag->node_begin();
  27         }
  28         virtual uint node_end() const {
  29                 return dag->node_end();
  30         }
  31         virtual uint node_count() const {
  32                 return dag->node_count();
  33         }
  34         virtual const void* node_info(uint node_id) const {
  35                 return dag->node_info(node_id);
  36         }
  37         virtual VocabIndex node_id(uint id) const {
  38                 return dag->node_id(id);
  39         }
  40         virtual void get_next(uint node_id,std::vector<uint> &next_id) const {
  41                 dag->get_next(node_id,next_id);
  42         }
  43         virtual float edge_value(uint node_from,uint node_to) const;
  44 };
  45
  46 #endif