// Created on: 2015-11-23
// Created by: Anastasia BORISOVA
// Copyright (c) 2015 OPEN CASCADE SAS
//
// This file is part of Open CASCADE Technology software library.
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License version 2.1 as published
// by the Free Software Foundation, with special exception defined in the file
// OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
// distribution for complete text of the license and disclaimer of any warranty.
//
// Alternatively, this file may be used under the terms of Open CASCADE
// commercial license or contractual agreement.

#include <AIS_RubberBand.hxx>
#include <BRepMesh_DataStructureOfDelaun.hxx>
#include <BRepMesh_Delaun.hxx>
#include <Graphic3d_ArrayOfPolygons.hxx>
#include <Graphic3d_ArrayOfPolylines.hxx>
#include <Graphic3d_AspectFillArea3d.hxx>
#include <Graphic3d_GraphicDriver.hxx>
#include <Graphic3d_ArrayOfTriangles.hxx>
#include <Graphic3d_TransModeFlags.hxx>
#include <Graphic3d_ZLayerId.hxx>
#include <Prs3d_LineAspect.hxx>
#include <Prs3d_Root.hxx>
#include <Prs3d_ShadingAspect.hxx>
#include <SelectMgr_EntityOwner.hxx>
#include <V3d_Viewer.hxx>
#include <V3d_View.hxx>


#define MEMORY_BLOCK_SIZE 512 * 7

IMPLEMENT_STANDARD_RTTIEXT(AIS_RubberBand, AIS_InteractiveObject)
//=======================================================================
//function : Constructor
//purpose  :
//=======================================================================
AIS_RubberBand::AIS_RubberBand()
{
  myDrawer->SetLineAspect (new Prs3d_LineAspect (Quantity_NOC_WHITE, Aspect_TOL_SOLID, 1.0));
  myDrawer->SetShadingAspect (new Prs3d_ShadingAspect());
  myDrawer->ShadingAspect()->SetMaterial (Graphic3d_NOM_PLASTIC);
  myDrawer->ShadingAspect()->Aspect()->SetInteriorStyle (Aspect_IS_EMPTY);
  myDrawer->ShadingAspect()->SetTransparency (1.0);
  myDrawer->ShadingAspect()->SetColor (Quantity_NOC_WHITE);

  SetTransformPersistence (new Graphic3d_TransformPers (Graphic3d_TMF_2d, Aspect_TOTP_LEFT_LOWER));
  SetZLayer (Graphic3d_ZLayerId_TopOSD);
}

//=======================================================================
//function : Constructor
//purpose  :
//=======================================================================
AIS_RubberBand::AIS_RubberBand (const Quantity_Color& theLineColor,
                                const Aspect_TypeOfLine theLineType,
                                const Standard_Real theWidth)
{
  myDrawer->SetLineAspect (new Prs3d_LineAspect (theLineColor, theLineType, theWidth));
  myDrawer->SetShadingAspect (new Prs3d_ShadingAspect());
  myDrawer->ShadingAspect()->SetMaterial (Graphic3d_NOM_PLASTIC);
  myDrawer->ShadingAspect()->Aspect()->SetInteriorStyle (Aspect_IS_EMPTY);
  myDrawer->ShadingAspect()->SetTransparency (1.0);
  myDrawer->ShadingAspect()->SetColor (Quantity_NOC_WHITE);

  SetTransformPersistence (new Graphic3d_TransformPers (Graphic3d_TMF_2d, Aspect_TOTP_LEFT_LOWER));
  SetZLayer (Graphic3d_ZLayerId_TopOSD);
}

//=======================================================================
//function : Constructor
//purpose  :
//=======================================================================
AIS_RubberBand::AIS_RubberBand (const Quantity_Color& theLineColor,
                                const Aspect_TypeOfLine theLineType,
                                const Quantity_Color theFillColor,
                                const Standard_Real theTransparency,
                                const Standard_Real theLineWidth)
{
  myDrawer->SetLineAspect (new Prs3d_LineAspect (theLineColor, theLineType, theLineWidth));
  myDrawer->SetShadingAspect (new Prs3d_ShadingAspect());
  myDrawer->ShadingAspect()->SetMaterial (Graphic3d_NOM_PLASTIC);
  myDrawer->ShadingAspect()->SetColor (theFillColor);
  myDrawer->ShadingAspect()->Aspect()->SetInteriorStyle (Aspect_IS_SOLID);
  myDrawer->ShadingAspect()->SetTransparency (theTransparency);

  SetTransformPersistence (new Graphic3d_TransformPers (Graphic3d_TMF_2d, Aspect_TOTP_LEFT_LOWER));
  SetZLayer (Graphic3d_ZLayerId_TopOSD);
}

//=======================================================================
//function : Destructor
//purpose  :
//=======================================================================
AIS_RubberBand::~AIS_RubberBand()
{
  myPoints.Clear();
  myTriangles.Nullify();
  myBorders.Nullify();
}

//=======================================================================
//function : SetRectangle
//purpose  :
//=======================================================================
void AIS_RubberBand::SetRectangle (const Standard_Integer theMinX, const Standard_Integer theMinY,
                                   const Standard_Integer theMaxX, const Standard_Integer theMaxY)
{
  myPoints.Clear();
  myPoints.Append (Graphic3d_Vec2i (theMinX, theMinY));
  myPoints.Append (Graphic3d_Vec2i (theMinX, theMaxY));
  myPoints.Append (Graphic3d_Vec2i (theMaxX, theMaxY));
  myPoints.Append (Graphic3d_Vec2i (theMaxX, theMinY));
}

//=======================================================================
//function : AddPoint
//purpose  :
//=======================================================================
void AIS_RubberBand::AddPoint (const Graphic3d_Vec2i& thePoint)
{
  myPoints.Append (thePoint);
}

//=======================================================================
//function : AddPoint
//purpose  :
//=======================================================================
void AIS_RubberBand::RemoveLastPoint()
{
  myPoints.Remove (myPoints.Length());
}

//=======================================================================
//function : GetPoints
//purpose  :
//=======================================================================
const NCollection_Sequence<Graphic3d_Vec2i>& AIS_RubberBand::Points() const
{
  return myPoints;
}

//=======================================================================
//function : LineColor
//purpose  :
//=======================================================================
Quantity_Color AIS_RubberBand::LineColor() const
{
  return myDrawer->LineAspect()->Aspect()->Color();
}

//=======================================================================
//function : SetLineColor
//purpose  :
//=======================================================================
void AIS_RubberBand::SetLineColor (const Quantity_Color& theColor)
{
  myDrawer->LineAspect()->SetColor (theColor);
}

//=======================================================================
//function : FillColor
//purpose  :
//=======================================================================
Quantity_Color AIS_RubberBand::FillColor() const
{
  return myDrawer->ShadingAspect()->Color();
}

//=======================================================================
//function : SetFillColor
//purpose  :
//=======================================================================
void AIS_RubberBand::SetFillColor (const Quantity_Color& theColor)
{
  myDrawer->ShadingAspect()->SetColor (theColor);
}

//=======================================================================
//function : SetLineWidth
//purpose  :
//=======================================================================
void AIS_RubberBand::SetLineWidth (const Standard_Real theWidth) const
{
  myDrawer->LineAspect()->SetWidth (theWidth);
}

//=======================================================================
//function : SetLineWidth
//purpose  :
//=======================================================================
Standard_Real AIS_RubberBand::LineWidth() const
{
  return myDrawer->LineAspect()->Aspect()->Width();
}

//=======================================================================
//function : SetLineType
//purpose  :
//=======================================================================
void AIS_RubberBand::SetLineType (const Aspect_TypeOfLine theType)
{
  myDrawer->LineAspect()->SetTypeOfLine (theType);
}

//=======================================================================
//function : LineType
//purpose  :
//=======================================================================
Aspect_TypeOfLine AIS_RubberBand::LineType() const
{
  return myDrawer->LineAspect()->Aspect()->Type();
}

//=======================================================================
//function : SetFillTransparency
//purpose  :
//=======================================================================
void AIS_RubberBand::SetFillTransparency (const Standard_Real theValue) const
{
  myDrawer->ShadingAspect()->SetTransparency (theValue);
}

//=======================================================================
//function : SetFillTransparency
//purpose  :
//=======================================================================
Standard_Real AIS_RubberBand::FillTransparency() const
{
  return myDrawer->ShadingAspect()->Transparency();
}

//=======================================================================
//function : SetFilling
//purpose  :
//=======================================================================
void AIS_RubberBand::SetFilling (const Standard_Boolean theIsFilling)
{
  myDrawer->ShadingAspect()->Aspect()->SetInteriorStyle (theIsFilling ? Aspect_IS_SOLID : Aspect_IS_EMPTY);
}

//=======================================================================
//function : SetFilling
//purpose  :
//=======================================================================
void AIS_RubberBand::SetFilling (const Quantity_Color theColor, const Standard_Real theTransparency)
{
  SetFilling (Standard_True);
  SetFillTransparency (theTransparency);
  SetFillColor (theColor);
}

//=======================================================================
//function : IsFilling
//purpose  :
//=======================================================================
Standard_Boolean AIS_RubberBand::IsFilling() const
{
  Aspect_InteriorStyle aStyle = myDrawer->ShadingAspect()->Aspect()->InteriorStyle();
  return aStyle != Aspect_IS_EMPTY;
}

//=======================================================================
//function : fillTriangles
//purpose  :
//=======================================================================
Standard_Boolean AIS_RubberBand::fillTriangles()
{
  Handle(NCollection_IncAllocator) anAllocator = new NCollection_IncAllocator (MEMORY_BLOCK_SIZE);
  Handle(BRepMesh_DataStructureOfDelaun) aMeshStructure = new BRepMesh_DataStructureOfDelaun(anAllocator);
  Standard_Integer aPtsLower = myPoints.Lower();
  Standard_Integer aPtsUpper = myPoints.Upper();
  BRepMesh::Array1OfInteger anIndexes (0, myPoints.Length() - 1);
  for (Standard_Integer aPtIdx = aPtsLower; aPtIdx <= aPtsUpper; ++aPtIdx)
  {
    gp_XY aP ((Standard_Real)myPoints.Value (aPtIdx).x(),
              (Standard_Real)myPoints.Value (aPtIdx).y());
    BRepMesh_Vertex aVertex (aP, aPtIdx, BRepMesh_Frontier);
    anIndexes.ChangeValue (aPtIdx - aPtsLower) = aMeshStructure->AddNode (aVertex);
  }

  Standard_Real aPtSum = 0;
  for (Standard_Integer aIdx = aPtsLower; aIdx <= aPtsUpper; ++aIdx)
  {
    Standard_Integer aNextIdx = (aIdx % myPoints.Length()) + 1;
    aPtSum += (Standard_Real)(myPoints.Value (aNextIdx).x() - myPoints.Value (aIdx).x())
             * (Standard_Real)(myPoints.Value (aNextIdx).y() + myPoints.Value (aIdx).y());
  }
  Standard_Boolean isClockwiseOrdered = aPtSum < 0;

  for (Standard_Integer aIdx = 0; aIdx < anIndexes.Length(); ++aIdx)
  {
    Standard_Integer aPtIdx = isClockwiseOrdered ? aIdx : (aIdx + 1) % anIndexes.Length();
    Standard_Integer aNextPtIdx = isClockwiseOrdered ? (aIdx + 1) % anIndexes.Length() : aIdx;
    BRepMesh_Edge anEdge (anIndexes.Value (aPtIdx),
                          anIndexes.Value (aNextPtIdx),
                          BRepMesh_Frontier);
    aMeshStructure->AddLink (anEdge);
  }

  BRepMesh_Delaun aTriangulation (aMeshStructure, anIndexes);
  const BRepMesh::MapOfInteger& aTriangles = aMeshStructure->ElementsOfDomain();
  if (aTriangles.Extent() < 1)
    return Standard_False;


  Standard_Boolean toFill = Standard_False;
  if (myTriangles.IsNull() || myTriangles->VertexNumber() != aTriangles.Extent() * 3)
  {
    toFill = Standard_True;
    myTriangles = new Graphic3d_ArrayOfTriangles (aTriangles.Extent() * 3, 0, Standard_True);
  }

  Standard_Integer aVertexIndex = 1;
  BRepMesh::MapOfInteger::Iterator aTriangleIt (aTriangles);
  for (; aTriangleIt.More(); aTriangleIt.Next())
  {
    const Standard_Integer aTriangleId = aTriangleIt.Key();
    const BRepMesh_Triangle& aCurrentTriangle = aMeshStructure->GetElement (aTriangleId);

    if (aCurrentTriangle.Movability() == BRepMesh_Deleted)
      continue;

    Standard_Integer aTriangleVerts[3];
    aMeshStructure->ElementNodes (aCurrentTriangle, aTriangleVerts);

    gp_Pnt2d aPts[3];
    for (Standard_Integer aVertIdx = 0; aVertIdx < 3; ++aVertIdx)
    {
      const BRepMesh_Vertex& aVertex = aMeshStructure->GetNode (aTriangleVerts[aVertIdx]);
      aPts[aVertIdx] = aVertex.Coord();
    }

    if (toFill)
    {
      gp_Dir aNorm = gp::DZ();
      for (Standard_Integer anIt = 0; anIt < 3; ++anIt)
      {
        myTriangles->AddVertex (aPts[anIt].X(), aPts[anIt].Y(), 0.0,
                                aNorm.X(), aNorm.Y(), aNorm.Z());
      }
    }
    else
    {
      for (Standard_Integer anIt = 0; anIt < 3; ++anIt)
      {
        myTriangles->SetVertice (aVertexIndex++, (Standard_ShortReal)aPts[anIt].X(), (Standard_ShortReal)aPts[anIt].Y(), 0.0f);
      }
    }
  }

  aMeshStructure.Nullify();
  anAllocator.Nullify();

  return Standard_True;
}

//=======================================================================
//function : Compute
//purpose  :
//=======================================================================
void AIS_RubberBand::Compute (const Handle(PrsMgr_PresentationManager3d)& /*thePresentationManager*/,
                              const Handle(Prs3d_Presentation)& thePresentation,
                              const Standard_Integer /*theMode*/)
{
  Handle (Graphic3d_Group) aGroup = Prs3d_Root::CurrentGroup (thePresentation);

  // Draw filling
  if (IsFilling() && fillTriangles())
  {
    aGroup->SetGroupPrimitivesAspect (myDrawer->ShadingAspect()->Aspect());
    aGroup->AddPrimitiveArray (myTriangles);
  }

  // Draw frame
  if (myBorders.IsNull() || myBorders->VertexNumber() != myPoints.Length() + 1)
  {
     myBorders = new Graphic3d_ArrayOfPolylines (myPoints.Length() + 1);
     for (Standard_Integer anIt = 1; anIt <= myPoints.Length(); anIt++)
     {
       myBorders->AddVertex ((Standard_Real)myPoints.Value (anIt).x(),
                             (Standard_Real)myPoints.Value (anIt).y(), 0.0);
     }

     myBorders->AddVertex ((Standard_Real)myPoints.Value(1).x(),
                           (Standard_Real)myPoints.Value(1).y(), 0.0);

  }
  else
  {
    for (Standard_Integer anIt = 1; anIt <= myPoints.Length(); anIt++)
    {
          myBorders->SetVertice (anIt, (Standard_ShortReal)myPoints.Value (anIt).x(),
                                 (Standard_ShortReal)myPoints.Value (anIt).y(), 0.0f);
    }

    myBorders->SetVertice (myPoints.Length() + 1, (Standard_ShortReal)myPoints.Value(1).x(),
                           (Standard_ShortReal)myPoints.Value(1).y(), 0.0f);
  }

  aGroup->SetGroupPrimitivesAspect (myDrawer->LineAspect()->Aspect());
  aGroup->AddPrimitiveArray (myBorders);
}