/*
 * Copyright (c) 2018 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 __NEURUN_DUMPER_DOT_DOT_OPERAND_INFO_H__
#define __NEURUN_DUMPER_DOT_DOT_OPERAND_INFO_H__

#include <vector>

#include "IDotInfo.h"
#include "model/operand/Object.h"
#include "model/operand/Index.h"

namespace neurun
{
namespace dumper
{
namespace dot
{

class DotOperandInfo : public IDotInfo
{
public:
  enum class Type
  {
    UNDEFINED,
    MODEL_INPUT,
    MODEL_OUTPUT,
    INTERNAL
  };

public:
  static const std::string INPUT_SHAPE;
  static const std::string OUTPUT_SHAPE;
  static const std::string OPERAND_SHAPE;
  static const std::string BG_COLOR_SCHEME;
  static const std::string BG_COLORS[8];

public:
  DotOperandInfo(const neurun::model::operand::Index &index,
                 const neurun::model::operand::Object &object, Type type);

public:
  virtual std::string index_str() const override;
  virtual std::string label() const override;
  virtual std::string dot_shape() const override;
  virtual std::string bg_color_scheme() const override;
  virtual std::string bg_color() const override;

private:
  void addBackendLabel();

private:
  const neurun::model::operand::Index &_index;
  const neurun::model::operand::Object &_object;
  Type _type;

  std::vector<std::string> _labels;
};

} // namespace dot
} // namespace dumper
} // namespace neurun

#endif // __NEURUN_DUMPER_DOT_DOT_OPERAND_INFO_H__