Module.h

Go to the documentation of this file.

/*
 * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#ifndef __LUCI_MODULE_H__
#define __LUCI_MODULE_H__
 
#include <loco/IR/Graph.h>
 
#include <map>
#include <memory>
#include <vector>
 
namespace luci
{
 
class Module final
{
public:
  Module() = default;
 
  // Copy/Move is not allowed for Module
  Module(const Module &) = delete;
  Module(Module &&) = delete;
 
  ~Module() = default;
 
public:
  size_t size(void) const { return _graphs.size(); }
 
public:
  void add(std::unique_ptr<loco::Graph> &&g);
 
  loco::Graph *graph(void) const;
 
  loco::Graph *graph(size_t idx) const;
 
  // TODO provide graph accessor with a name
 
public:
  void ext_buffer(bool set) { _ext_buffer = set; }
  bool ext_buffer(void) const { return _ext_buffer; }
 
private:
  bool _ext_buffer = false;
 
public:
  void source_table(const std::map<uint32_t, std::string> &table) { _source_table = table; }
 
  const std::map<uint32_t, std::string> &source_table(void) const { return _source_table; }
 
private:
  std::vector<std::unique_ptr<loco::Graph>> _graphs;
 
private:
  std::map<uint32_t, std::string> _source_table;
};
 
std::unique_ptr<Module> make_module(void);
 
} // namespace luci
 
#endif // __LUCI_MODULE_H__