OCCT

// Created by: Peter KURNEV
// Copyright (c) 1999-2014 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 <BOPAlgo_BOP.hxx>
#include <BOPAlgo_Builder.hxx>
#include <BOPAlgo_Operation.hxx>
#include <BOPAlgo_PaveFiller.hxx>
#include <BOPAlgo_Section.hxx>
#include <BOPAlgo_Splitter.hxx>
#include <BOPDS_DS.hxx>
#include <BOPTest.hxx>
#include <BOPTest_DrawableShape.hxx>
#include <BOPTest_Objects.hxx>
#include <BRepTest_Objects.hxx>
#include <DBRep.hxx>
#include <OSD_Timer.hxx>
#include <TopoDS_Shape.hxx>
#include <Draw_ProgressIndicator.hxx>

#include <cstdio>
#include <cstring>
//
//
static int bfillds(Draw_Interpretor&, int, const char**);
static int bbuild(Draw_Interpretor&, int, const char**);
static int bbop(Draw_Interpretor&, int, const char**);
static int bsplit(Draw_Interpretor&, int, const char**);
static int buildbop(Draw_Interpretor&, int, const char**);

//=================================================================================================

void BOPTest::PartitionCommands(Draw_Interpretor& theCommands)
{
  static bool done = false;
  if (done)
  {
    return;
  }
  done = true;
  // Chapter's name
  const char* g = "BOPTest commands";
  // Commands
  theCommands.Add("bfillds",
                  "Performs intersection of the arguments added for the operation by baddobjects "
                  "and baddtools commands.\n"
                  "\t\tUsage: bfillds [-t]\n"
                  "\t\tWhere: -t is the optional parameter for enabling timer and showing elapsed "
                  "time of the operation",
                  __FILE__,
                  bfillds,
                  g);

  theCommands.Add(
    "bbuild",
    "Builds the result of General Fuse operation. Intersection (bfillds) has to be already "
    "performed by this moment.\n"
    "\t\tUsage: bbuild result [-t]\n"
    "\t\tWhere:\n"
    "\t\tresult - name of the result shape\n"
    "\t\t-t is the optional parameter for enabling timer and showing elapsed time of the operation",
    __FILE__,
    bbuild,
    g);

  theCommands.Add(
    "bbop",
    "Builds the result of Boolean operation. Intersection (bfillds) has to be already performed by "
    "this moment.\n"
    "\t\tUsage: bbop result op [-t]\n"
    "\t\tWhere:\n"
    "\t\tresult - name of the result shape\n"
    "\t\top - type of Boolean operation. Possible values:\n"
    "\t\t     - 0/common - for Common operation\n"
    "\t\t     - 1/fuse - for Fuse operation\n"
    "\t\t     - 2/cut - for Cut operation\n"
    "\t\t     - 3/tuc/cut21 - for Cut21 operation\n"
    "\t\t     - 4/section - for Section operation\n"
    "\t\t-t - optional parameter for enabling timer and showing elapsed time of the operation",
    __FILE__,
    bbop,
    g);

  theCommands.Add(
    "bsplit",
    "Builds the result of Split operation. Intersection (bfillds) has to be already performed by "
    "this moment.\n"
    "\t\tUsage: bsplit result [-t]\n"
    "\t\tWhere:\n"
    "\t\tresult - name of the result shape\n"
    "\t\t-t is the optional parameter for enabling timer and showing elapsed time of the operation",
    __FILE__,
    bsplit,
    g);

  theCommands.Add(
    "buildbop",
    "Builds the result of BOP basing on the GF, thus bbuild command has to be already performed\n"
    "\t\tThe command uses classification approach for building the result of BOP\n"
    "\t\t(thus it operates on solids only and can be used on open solids):\n"
    "\t\t - FUSE is built from the faces OUT of all arguments\n"
    "\t\t - COMMON is built from the faces IN any of the object/tools\n"
    "\t\t - CUT is built from the objects faces OUT of the tools and tools faces IN the objects.\n"
    "\t\tPlease note that history for solids will not be available.\n\n"
    "\t\tUsage: buildbop result -o s1 [s2 ...] -t s3 [s4 ...] -op operation (common/fuse/cut/tuc)\n"
    "\t\tWhere:\n"
    "\t\tresult      - result shape of the operation\n"
    "\t\ts1 s2 s3 s4 - arguments (solids) of the GF operation\n"
    "\t\toperation   - type of boolean operation",
    __FILE__,
    buildbop,
    g);
}

//=================================================================================================

int bfillds(Draw_Interpretor& di, int n, const char** a)
{
  if (n > 2)
  {
    di.PrintHelp(a[0]);
    return 1;
  }
  //
  char                                     buf[32];
  bool                                     bRunParallel, bNonDestructive, bShowTime;
  int                                      aNbS;
  double                                   aTol;
  NCollection_List<TopoDS_Shape>::Iterator aIt;
  NCollection_List<TopoDS_Shape>           aLC;
  NCollection_List<TopoDS_Shape>&          aLS = BOPTest_Objects::Shapes();
  aNbS                                         = aLS.Extent();
  if (!aNbS)
  {
    di << "No objects to process\n";
    return 0;
  }
  //
  bShowTime = false;
  //
  bRunParallel           = BOPTest_Objects::RunParallel();
  bNonDestructive        = BOPTest_Objects::NonDestructive();
  aTol                   = BOPTest_Objects::FuzzyValue();
  BOPAlgo_GlueEnum aGlue = BOPTest_Objects::Glue();
  //
  if (n == 2)
  {
    if (!strcmp(a[1], "-t"))
    {
      bShowTime = true;
    }
    else
    {
      di << "Warning: invalid key\n";
    }
  }
  //
  NCollection_List<TopoDS_Shape>& aLT = BOPTest_Objects::Tools();
  //
  aIt.Initialize(aLS);
  for (; aIt.More(); aIt.Next())
  {
    const TopoDS_Shape& aS = aIt.Value();
    aLC.Append(aS);
  }
  //
  aIt.Initialize(aLT);
  for (; aIt.More(); aIt.Next())
  {
    const TopoDS_Shape& aS = aIt.Value();
    aLC.Append(aS);
  }
  //
  BOPAlgo_PaveFiller& aPF = BOPTest_Objects::PaveFiller();
  //
  aPF.SetArguments(aLC);
  aPF.SetRunParallel(bRunParallel);
  aPF.SetNonDestructive(bNonDestructive);
  aPF.SetFuzzyValue(aTol);
  aPF.SetGlue(aGlue);
  aPF.SetUseOBB(BOPTest_Objects::UseOBB());
  //
  OSD_Timer aTimer;
  aTimer.Start();
  //
  occ::handle<Draw_ProgressIndicator> aProgress = new Draw_ProgressIndicator(di, 1);
  aPF.Perform(aProgress->Start());
  BOPTest::ReportAlerts(aPF.GetReport());
  if (aPF.HasErrors())
  {
    return 0;
  }
  //
  aTimer.Stop();
  //
  if (bShowTime)
  {
    Sprintf(buf, "  Tps: %7.2lf\n", aTimer.ElapsedTime());
    di << buf;
  }
  //
  return 0;
}

//=================================================================================================

int bbuild(Draw_Interpretor& di, int n, const char** a)
{
  if (n < 2 || n > 3)
  {
    di.PrintHelp(a[0]);
    return 1;
  }
  //
  BOPDS_DS* pDS = BOPTest_Objects::PDS();
  if (!pDS)
  {
    di << "Prepare PaveFiller first\n";
    return 0;
  }
  //
  char buf[128];
  bool bRunParallel, bShowTime;

  NCollection_List<TopoDS_Shape>::Iterator aIt;
  //
  BOPAlgo_PaveFiller& aPF = BOPTest_Objects::PaveFiller();
  //
  BOPTest_Objects::SetBuilderDefault();
  BOPAlgo_Builder& aBuilder = BOPTest_Objects::Builder();
  aBuilder.Clear();
  //
  NCollection_List<TopoDS_Shape>& aLSObj = BOPTest_Objects::Shapes();
  aIt.Initialize(aLSObj);
  for (; aIt.More(); aIt.Next())
  {
    const TopoDS_Shape& aS = aIt.Value();
    aBuilder.AddArgument(aS);
  }
  //
  NCollection_List<TopoDS_Shape>& aLSTool = BOPTest_Objects::Tools();
  aIt.Initialize(aLSTool);
  for (; aIt.More(); aIt.Next())
  {
    const TopoDS_Shape& aS = aIt.Value();
    aBuilder.AddArgument(aS);
  }
  //
  bShowTime    = false;
  bRunParallel = BOPTest_Objects::RunParallel();
  if (n == 3)
  {
    if (!strcmp(a[2], "-t"))
    {
      bShowTime = true;
    }
    else
    {
      di << "Warning: invalid key\n";
    }
  }
  aBuilder.SetRunParallel(bRunParallel);
  aBuilder.SetCheckInverted(BOPTest_Objects::CheckInverted());
  aBuilder.SetToFillHistory(BRepTest_Objects::IsHistoryNeeded());
  //
  occ::handle<Draw_ProgressIndicator> aProgress = new Draw_ProgressIndicator(di, 1);
  //
  OSD_Timer aTimer;
  aTimer.Start();
  //
  aBuilder.PerformWithFiller(aPF, aProgress->Start());
  BOPTest::ReportAlerts(aBuilder.GetReport());

  // Set history of GF operation into the session
  if (BRepTest_Objects::IsHistoryNeeded())
  {
    BRepTest_Objects::SetHistory(aPF.Arguments(), aBuilder);
  }

  if (aBuilder.HasErrors())
  {
    return 0;
  }
  //
  aTimer.Stop();
  //
  if (bShowTime)
  {
    Sprintf(buf, "  Tps: %7.2lf\n", aTimer.ElapsedTime());
    di << buf;
  }
  //
  const TopoDS_Shape& aR = aBuilder.Shape();
  if (aR.IsNull())
  {
    di << "Result is a null shape\n";
    return 0;
  }
  //
  DBRep::Set(a[1], aR);
  return 0;
}

//=================================================================================================

int bbop(Draw_Interpretor& di, int n, const char** a)
{
  if (n < 3 || n > 4)
  {
    di.PrintHelp(a[0]);
    return 1;
  }
  //
  BOPDS_DS* pDS = BOPTest_Objects::PDS();
  if (!pDS)
  {
    di << "Prepare PaveFiller first\n";
    return 0;
  }
  //
  BOPAlgo_Operation anOp = BOPTest::GetOperationType(a[2]);
  if (anOp == BOPAlgo_UNKNOWN)
  {
    di << "Invalid operation type\n";
    return 0;
  }

  bool bShowTime    = false;
  bool bRunParallel = BOPTest_Objects::RunParallel();
  if (n == 4)
  {
    if (!strcmp(a[3], "-t"))
    {
      bShowTime = true;
    }
    else
    {
      di << "Warning: invalid key\n";
    }
  }
  //
  BOPAlgo_PaveFiller& aPF = BOPTest_Objects::PaveFiller();
  //
  BOPAlgo_Builder* pBuilder = nullptr;

  if (anOp != BOPAlgo_SECTION)
  {
    pBuilder = &BOPTest_Objects::BOP();
  }
  else
  {
    pBuilder = &BOPTest_Objects::Section();
  }
  //
  pBuilder->Clear();
  //
  NCollection_List<TopoDS_Shape>&          aLSObj = BOPTest_Objects::Shapes();
  NCollection_List<TopoDS_Shape>::Iterator aIt(aLSObj);
  for (; aIt.More(); aIt.Next())
  {
    const TopoDS_Shape& aS = aIt.Value();
    pBuilder->AddArgument(aS);
  }
  //
  if (anOp != BOPAlgo_SECTION)
  {
    BOPAlgo_BOP* pBOP = (BOPAlgo_BOP*)pBuilder;
    //
    NCollection_List<TopoDS_Shape>& aLSTools = BOPTest_Objects::Tools();
    aIt.Initialize(aLSTools);
    for (; aIt.More(); aIt.Next())
    {
      const TopoDS_Shape& aS = aIt.Value();
      pBOP->AddTool(aS);
    }
    //
    pBOP->SetOperation(anOp);
  }
  else
  {
    NCollection_List<TopoDS_Shape>& aLSTools = BOPTest_Objects::Tools();
    aIt.Initialize(aLSTools);
    for (; aIt.More(); aIt.Next())
    {
      const TopoDS_Shape& aS = aIt.Value();
      pBuilder->AddArgument(aS);
    }
  }
  //
  pBuilder->SetRunParallel(bRunParallel);
  pBuilder->SetCheckInverted(BOPTest_Objects::CheckInverted());
  pBuilder->SetToFillHistory(BRepTest_Objects::IsHistoryNeeded());
  //
  occ::handle<Draw_ProgressIndicator> aProgress = new Draw_ProgressIndicator(di, 1);
  //
  OSD_Timer aTimer;
  aTimer.Start();
  //
  pBuilder->PerformWithFiller(aPF, aProgress->Start());
  BOPTest::ReportAlerts(pBuilder->GetReport());

  // Set history of Boolean operation into the session
  if (BRepTest_Objects::IsHistoryNeeded())
  {
    BRepTest_Objects::SetHistory(aPF.Arguments(), *pBuilder);
  }

  if (pBuilder->HasErrors())
  {
    return 0;
  }
  //
  aTimer.Stop();
  //
  if (bShowTime)
  {
    char buf[32];
    Sprintf(buf, "  Tps: %7.2lf\n", aTimer.ElapsedTime());
    di << buf;
  }
  //
  const TopoDS_Shape& aR = pBuilder->Shape();
  if (aR.IsNull())
  {
    di << "Result is a null shape\n";
    return 0;
  }
  //
  BOPTest_Objects::SetBuilder(pBuilder);
  //
  DBRep::Set(a[1], aR);
  return 0;
}

//=================================================================================================

int bsplit(Draw_Interpretor& di, int n, const char** a)
{
  if (n < 2 || n > 3)
  {
    di.PrintHelp(a[0]);
    return 1;
  }
  //
  BOPDS_DS* pDS = BOPTest_Objects::PDS();
  if (!pDS)
  {
    di << "Prepare PaveFiller first\n";
    return 0;
  }
  //
  BOPAlgo_PaveFiller& aPF = BOPTest_Objects::PaveFiller();
  //
  BOPAlgo_Splitter* pSplitter = &BOPTest_Objects::Splitter();
  pSplitter->Clear();
  //
  // set objects
  const NCollection_List<TopoDS_Shape>& aLSObjects = BOPTest_Objects::Shapes();
  pSplitter->SetArguments(aLSObjects);
  //
  // set tools
  NCollection_List<TopoDS_Shape>& aLSTools = BOPTest_Objects::Tools();
  pSplitter->SetTools(aLSTools);
  //
  // set options
  pSplitter->SetRunParallel(BOPTest_Objects::RunParallel());
  pSplitter->SetNonDestructive(BOPTest_Objects::NonDestructive());
  pSplitter->SetFuzzyValue(BOPTest_Objects::FuzzyValue());
  pSplitter->SetCheckInverted(BOPTest_Objects::CheckInverted());
  pSplitter->SetToFillHistory(BRepTest_Objects::IsHistoryNeeded());
  //
  // measure the time of the operation
  OSD_Timer aTimer;
  aTimer.Start();
  //
  // perform the operation
  occ::handle<Draw_ProgressIndicator> aProgress = new Draw_ProgressIndicator(di, 1);
  pSplitter->PerformWithFiller(aPF, aProgress->Start());
  //
  aTimer.Stop();
  BOPTest::ReportAlerts(pSplitter->GetReport());

  // Set history of Split operation into the session
  if (BRepTest_Objects::IsHistoryNeeded())
  {
    BRepTest_Objects::SetHistory(aPF.Arguments(), *pSplitter);
  }

  if (pSplitter->HasErrors())
  {
    return 0;
  }
  //
  // show time if necessary
  if (n == 3)
  {
    if (!strcmp(a[2], "-t"))
    {
      char buf[20];
      Sprintf(buf, "  Tps: %7.2lf\n", aTimer.ElapsedTime());
      di << buf;
    }
    else
    {
      di << "Warning: invalid key\n";
    }
  }
  //
  // Debug commands support
  BOPTest_Objects::SetBuilder(pSplitter);
  //
  const TopoDS_Shape& aR = pSplitter->Shape();
  if (aR.IsNull())
  {
    di << " null shape\n";
    return 0;
  }
  //
  DBRep::Set(a[1], aR);
  return 0;
}

//=================================================================================================

int buildbop(Draw_Interpretor& di, int n, const char** a)
{
  if (n < 3)
  {
    di.PrintHelp(a[0]);
    return 1;
  }

  BOPDS_DS* pDS = BOPTest_Objects::PDS();
  if (!pDS)
  {
    di << "Error: perform intersection of arguments first";
    return 1;
  }

  BOPAlgo_Builder* pBuilder = &BOPTest_Objects::Builder();
  if (pBuilder->HasErrors())
  {
    di << "Error: there were problems during GF";
    return 0;
  }

  if (pBuilder->Arguments().IsEmpty() || pBuilder->Shape().IsNull())
  {
    di << "Error: it seems the GF has not been yet performed";
    return 1;
  }

  // Get arguments and operation
  NCollection_List<TopoDS_Shape> aLObjects, aLTools;
  BOPAlgo_Operation              anOp = BOPAlgo_UNKNOWN;

  for (int i = 2; i < n; ++i)
  {
    if (!strcmp(a[i], "-o") || !strcmp(a[i], "-t"))
    {
      if (i == (n - 1))
      {
        di << "Error: shapes are expected after the key " << a[i];
        return 1;
      }

      NCollection_List<TopoDS_Shape>& aList = !strcmp(a[i], "-o") ? aLObjects : aLTools;
      int                             j     = i + 1;
      for (; j < n; ++j)
      {
        if (a[j][0] == '-')
        {
          // reached the following key
          i = j - 1;
          break;
        }
        else
        {
          // Get the shape
          TopoDS_Shape aS = DBRep::Get(a[j]);
          if (aS.IsNull())
          {
            di << "Error: " << a[j] << " is a null shape";
            return 1;
          }
          if (aS.ShapeType() != TopAbs_SOLID)
          {
            di << "Error: " << a[j] << " is not a solid";
            return 1;
          }
          if (pDS->Index(aS) < 0)
          {
            di << "Error: " << a[j] << " is not an argument of GF";
            return 1;
          }
          aList.Append(aS);
        }
      }
      // End of arguments is reached
      if (j == n)
      {
        break;
      }
    }
    else if (!strcmp(a[i], "-op"))
    {
      if (i == (n - 1))
      {
        di << "Error: operation type is expected after the key " << a[i];
        return 1;
      }

      ++i;
      if (!strcasecmp(a[i], "common"))
      {
        anOp = BOPAlgo_COMMON;
      }
      else if (!strcasecmp(a[i], "fuse"))
      {
        anOp = BOPAlgo_FUSE;
      }
      else if (!strcasecmp(a[i], "cut"))
      {
        anOp = BOPAlgo_CUT;
      }
      else if (!strcasecmp(a[i], "tuc"))
      {
        anOp = BOPAlgo_CUT21;
      }
      else
      {
        di << "Error: unknown operation type";
        return 1;
      }
    }
    else
    {
      di << "Error: " << a[i] << " invalid key";
      return 1;
    }
  }

  if (anOp == BOPAlgo_UNKNOWN)
  {
    di << "Error: operation has not been specified";
    return 1;
  }

  bool hasObjects = !aLObjects.IsEmpty();
  bool hasTools   = !aLTools.IsEmpty();
  if (!hasObjects && !hasTools)
  {
    di << "Error: no shapes are given";
    return 1;
  }

  // Create new report for the operation
  occ::handle<Message_Report>         aReport   = new Message_Report;
  occ::handle<Draw_ProgressIndicator> aProgress = new Draw_ProgressIndicator(di, 1);
  // Build specific operation
  pBuilder->BuildBOP(aLObjects, aLTools, anOp, aProgress->Start(), aReport);

  // Report alerts of the operation
  BOPTest::ReportAlerts(aReport);

  if (!aReport->GetAlerts(Message_Fail).IsEmpty())
  {
    return 0;
  }

  // Set history of Split operation into the session
  if (BRepTest_Objects::IsHistoryNeeded())
  {
    BRepTest_Objects::SetHistory(pDS->Arguments(), *pBuilder);
  }

  // Result shape
  const TopoDS_Shape& aR = pBuilder->Shape();
  // Draw result shape
  DBRep::Set(a[1], aR);

  return 0;
}