test/source/blender/freestyle/intern/scene_graph/TriangleRep.h

/* SPDX-License-Identifier: GPL-2.0-or-later */

#pragma once

/** \file
 * \ingroup freestyle
 * \brief Class to define the representation of a triangle
 */

//! inherits from class Rep
#include "Rep.h"

namespace Freestyle {

/** Base class for all lines objects */
class TriangleRep : public Rep {
 public:
  /** Line description style */
  enum TRIANGLE_STYLE {
    FILL,
    LINES,
  };

 private:
  TRIANGLE_STYLE _Style;
  Vec3r _vertices[3];
  Vec3r _colors[3];

 public:
  inline TriangleRep() : Rep()
  {
    _Style = FILL;
  }

  /** Builds a triangle from 3 vertices
   *  v0
   *    first vertex
   *  v1
   *    second vertex
   *  v2
   *    third vertex
   */
  inline TriangleRep(const Vec3r &v0, const Vec3r &v1, const Vec3r &v2) : Rep()
  {
    _vertices[0] = v0;
    _vertices[1] = v1;
    _vertices[2] = v2;
    _Style = FILL;
  }

  inline TriangleRep(const Vec3r &v0,
                     const Vec3r &c0,
                     const Vec3r &v1,
                     const Vec3r &c1,
                     const Vec3r &v2,
                     const Vec3r &c2)
      : Rep()
  {
    _vertices[0] = v0;
    _vertices[1] = v1;
    _vertices[2] = v2;
    _colors[0] = c0;
    _colors[1] = c1;
    _colors[2] = c2;
    _Style = FILL;
  }

  virtual ~TriangleRep()
  {
  }

  /** accessors */
  inline const TRIANGLE_STYLE style() const
  {
    return _Style;
  }

  inline const Vec3r &vertex(int index) const
  {
    return _vertices[index];
  }

  inline const Vec3r &color(int index) const
  {
    return _colors[index];
  }

  /** modifiers */
  inline void setStyle(const TRIANGLE_STYLE iStyle)
  {
    _Style = iStyle;
  }

  inline void setVertex(int index, const Vec3r &iVertex)
  {
    _vertices[index] = iVertex;
  }

  inline void setColor(int index, const Vec3r &iColor)
  {
    _colors[index] = iColor;
  }

  inline void setVertices(const Vec3r &v0, const Vec3r &v1, const Vec3r &v2)
  {
    _vertices[0] = v0;
    _vertices[1] = v1;
    _vertices[2] = v2;
  }

  inline void setColors(const Vec3r &c0, const Vec3r &c1, const Vec3r &c2)
  {
    _colors[0] = c0;
    _colors[1] = c1;
    _colors[2] = c2;
  }

  /** Accept the corresponding visitor */
  virtual void accept(SceneVisitor &v)
  {
    Rep::accept(v);
    v.visitTriangleRep(*this);
  }

  /** Computes the triangle bounding box. */
  virtual void ComputeBBox();
};

} /* namespace Freestyle */