summaryrefslogtreecommitdiff
path: root/src/ff_factory.h
blob: bc5865678f39600e15c672f7068207e027a8c780 (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
#ifndef _FF_FACTORY_H_
#define _FF_FACTORY_H_

#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;
  void DisplayList() const;
  void Register(const std::string& ffname, FFFactoryBase* factory);
 private:
  FFRegistry() {}
  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;
};

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));
  }
};

#endif