namespace generator {
namespace cpp {

IncludeRenderer::IncludeRenderer(RendererContext context) : Renderer(context) {}

void IncludeRenderer::SelfHeader() {
  Include(SelfHeaderPath());
  BlankLine();
}

string IncludeRenderer::SelfHeaderPath() const {
  return io::JoinPath(context_.path_config.tf_root_dir,
                      context_.path_config.tf_output_dir,
                      context_.cpp_config.unit + "_ops.h");
}

namespace tensorflow {
namespace generator {
namespace cpp {

class IncludeRenderer : public Renderer {
 public:
  explicit IncludeRenderer(RendererContext context);

  string SelfHeaderPath() const;
  void SelfHeader();
  void Headers();

 private:
  void Include(const string &tf_file_path);
};

}  // namespace cpp
}  // namespace generator
}  // namespace tensorflow