inode.hpp

#ifndef INODE_H
#define INODE_H

#include <vector>
#include <algorithm>
#include "ivisitor.hpp"

namespace mcts {
using std::vector;

// ----------------------------------------------------------------------
// ----------------------------------------------------------------------
template <typename Context, typename Config> class INode {
  typedef typename IVisitor<Context, Config>::visitor_t visitor_t;

public:
  typedef INode<Context, Config> node_t;

  virtual ~INode() {}

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual void expand() = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual node_t *parent() const = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual node_t *select_child() = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual node_t *select_recusively() = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual Config *config() const = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual int nb_samples() const = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual const vector<node_t *> children() const = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual void add_child(node_t *child) = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual const Context context() const = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual double ev() const = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual double std_dev() const = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual double variance() const = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual double simulate() const = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual void accept(visitor_t *visitor) = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual void backpropagate(const double &value) = 0;

  // ----------------------------------------------------------------------
  // ----------------------------------------------------------------------
  virtual unsigned count_recursive() const {
    unsigned sum = 0;
    vector<node_t *> children = this->children();
    for (unsigned i = 0; i < children.size(); ++i) {
      sum += children[i]->count_recursive();
    }
    return 1 + sum;
  }
};
}

#endif