#ifndef TDL_INCLUDE_IR_CODEGEN_TUNE_H
#define TDL_INCLUDE_IR_CODEGEN_TUNE_H

#include <map>
#include <set>
#include <vector>

namespace tdl{

namespace ir{
  class value;
  class module;
  class instruction;
  class function;
}

namespace codegen{

class tune {
  typedef std::pair<ir::value*, unsigned> node_t;
  typedef std::map <node_t, std::set<node_t>> graph_t;

private:
  void add_constraint(node_t x, node_t y);
  void init_c_phi(ir::instruction *i);
  void init_c_graph(ir::instruction *v);
  void connected_components(node_t x, const std::vector<unsigned*> vals, std::set<node_t> &nodes, graph_t &graph);
  void create_grids(std::vector<ir::instruction*> &grids, std::map<unsigned*, ir::instruction*> &references, ir::function *fn);


public:
  std::vector<unsigned *> get_params(ir::module& mod);
  std::map<std::string, unsigned *> get_params(ir::instruction* i);
  unsigned *get_param(ir::value *value, const std::string &key) { return params_[value][key]; }
  bool check_constraints(ir::module &fn, std::map<ir::value *, std::vector<std::string>> &errors);
  void run(ir::module &mod);

private:
  std::map<ir::value*, std::map<std::string, unsigned*>> params_;
  std::vector<unsigned*> pool_;
  graph_t dependencies_;
  std::set<node_t> nodes_;
  std::map<node_t, unsigned> static_params_;
};


}
}

#endif