#ifndef TDL_INCLUDE_IR_CODEGEN_STORAGE_ALLOC_H
#define TDL_INCLUDE_IR_CODEGEN_STORAGE_ALLOC_H

#include <map>
#include <set>

namespace tdl{

namespace ir{
  class value;
  class function;
}

namespace codegen{

class layout;
class target_tuner;
class liveness;
class buffer_info_pass;

class allocation {
public:
  allocation(liveness *live, buffer_info_pass *buffer_info)
    : liveness_(live), buffer_info_(buffer_info){ }

  // accessors
  unsigned get_offset(ir::value *x)    const { return offsets_.at(x); }
  unsigned get_allocated_size()        const { return allocated_size_; }

  // run
  void run();

private:
  std::map<ir::value*, unsigned> offsets_;
  std::map<ir::value*, unsigned> num_bytes_;
  size_t allocated_size_;
  // dependences
  liveness *liveness_;
  buffer_info_pass *buffer_info_;
};

}
}

#endif