summaryrefslogtreecommitdiff
path: root/decoder/ff_factory.h
blob: 93681c5e0d97a06be9a4dab83825108db05aed5c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#ifndef _FF_FACTORY_H_
#define _FF_FACTORY_H_

/*TODO: register state identity separately from feature function identity?  as
 * in: string registry for name of state somewhere, assert that same result is
 * computed by all users?  or, we can just require that ff sharing same state
 * all be mashed into a single ffunc, which can just emit all the fid scores at
 * once.  that's fine.
 */

//TODO: use http://www.boost.org/doc/libs/1_43_0/libs/functional/factory/doc/html/index.html ?

#include <iostream>
#include <string>
#include <map>

#include <boost/shared_ptr.hpp>

class FeatureFunction;
class FFRegistry;
class FFFactoryBase;
extern boost::shared_ptr<FFRegistry> global_ff_registry;

class FFRegistry {
  friend int main(int argc, char** argv);
  friend class FFFactoryBase;
 public:
  boost::shared_ptr<FeatureFunction> Create(const std::string& ffname, const std::string& param) const;
  std::string usage(std::string const& ffname,bool params=true,bool verbose=true) const;
  void DisplayList() const;
  void Register(const std::string& ffname, FFFactoryBase* factory);
  void Register(FFFactoryBase* factory);
  FFRegistry() {}
 private:
  std::map<std::string, boost::shared_ptr<FFFactoryBase> > reg_;
};

struct FFFactoryBase {
  virtual ~FFFactoryBase();
  virtual boost::shared_ptr<FeatureFunction> Create(const std::string& param) const = 0;
  virtual std::string usage(bool params,bool verbose) const = 0;
};

template<class FF>
class FFFactory : public FFFactoryBase {
  boost::shared_ptr<FeatureFunction> Create(const std::string& param) const {
    return boost::shared_ptr<FeatureFunction>(new FF(param));
  }
  // called with false,false just gives feature name
  virtual std::string usage(bool params,bool verbose) const {
    return FF::usage(params,verbose);
  }

};

#endif