OCCT

// 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 <BRepToIGESBRep_Entity.hxx>
#include <Geom_Surface.hxx>
#include <GeomToIGES_GeomCurve.hxx>
#include <GeomToIGES_GeomSurface.hxx>
#include <IGESControl_ActorWrite.hxx>
#include <IGESData_IGESEntity.hxx>
#include <IGESData_IGESModel.hxx>
#include <MoniTool_Macros.hxx>
#include <Interface_Static.hxx>
#include <Standard_Type.hxx>
#include <Transfer_Binder.hxx>
#include <Transfer_Finder.hxx>
#include <Transfer_FinderProcess.hxx>
#include <Transfer_SimpleBinderOfTransient.hxx>
#include <Transfer_TransientMapper.hxx>
#include <TransferBRep_ShapeMapper.hxx>
#include <XSAlgo.hxx>
#include <XSAlgo_ShapeProcessor.hxx>

IMPLEMENT_STANDARD_RTTIEXT(IGESControl_ActorWrite, Transfer_ActorOfFinderProcess)

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

IGESControl_ActorWrite::IGESControl_ActorWrite()
{
  ModeTrans() = 0;
}

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

bool IGESControl_ActorWrite::Recognize(const occ::handle<Transfer_Finder>& start)
{
  DeclareAndCast(TransferBRep_ShapeMapper, shmap, start);
  if (!shmap.IsNull())
  {
    return true;
  }
  DeclareAndCast(Transfer_TransientMapper, gemap, start);
  if (!gemap.IsNull())
  {
    occ::handle<Standard_Transient> geom = gemap->Value();
    DeclareAndCast(Geom_Curve, Curve, geom);
    DeclareAndCast(Geom_Surface, Surf, geom);
    if (!Curve.IsNull() || !Surf.IsNull())
    {
      return true;
    }
  }
  return false;
}

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

occ::handle<Transfer_Binder> IGESControl_ActorWrite::Transfer(
  const occ::handle<Transfer_Finder>&        start,
  const occ::handle<Transfer_FinderProcess>& FP,
  const Message_ProgressRange&               theProgress)
{
  XSAlgo_ShapeProcessor::PrepareForTransfer();

  DeclareAndCast(IGESData_IGESModel, modl, FP->Model());
  if (modl.IsNull())
  {
    return NullResult();
  }
  if (themodetrans < 0 || themodetrans > 1)
  {
    return NullResult();
  }
  occ::handle<Standard_Transient> ent;

  DeclareAndCast(TransferBRep_ShapeMapper, shmap, start);
  if (!shmap.IsNull())
  {
    TopoDS_Shape shape = shmap->Value();
    if (shape.IsNull())
    {
      return NullResult();
    }

    XSAlgo_ShapeProcessor aShapeProcessor(GetShapeFixParameters());
    shape = aShapeProcessor.ProcessShape(shape, GetShapeProcessFlags().first, theProgress);

    BRepToIGES_BREntity BR0;
    BR0.SetModel(modl);
    BR0.SetTransferProcess(FP);
    BRepToIGESBRep_Entity BR1;
    BR1.SetModel(modl);
    BR1.SetTransferProcess(FP);

    if (themodetrans == 0)
    {
      ent = BR0.TransferShape(shape, theProgress);
    }
    if (themodetrans == 1)
    {
      ent = BR1.TransferShape(shape, theProgress);
    }
    aShapeProcessor.MergeTransferInfo(FP);
    if (!ent.IsNull())
    {
      return TransientResult(ent);
    }
  }
  DeclareAndCast(Transfer_TransientMapper, gemap, start);
  if (!gemap.IsNull())
  {
    occ::handle<Standard_Transient> geom = gemap->Value();
    DeclareAndCast(Geom_Curve, Curve, geom);
    DeclareAndCast(Geom_Surface, Surf, geom);

    //  We recognize: Curve and Surface from Geom
    //   what about Point; Geom2d ?

    //  GeomToIGES_GeomPoint GP;
    GeomToIGES_GeomCurve GC;
    GC.SetModel(modl);
    GeomToIGES_GeomSurface GS;
    GS.SetModel(modl);
    if (!Curve.IsNull())
    {
      ent = GC.TransferCurve(Curve, Curve->FirstParameter(), Curve->LastParameter());
    }
    else if (!Surf.IsNull())
    {
      double U1, U2, V1, V2;
      Surf->Bounds(U1, U2, V1, V2);
      ent = GS.TransferSurface(Surf, U1, U2, V1, V2);
    }
    if (!ent.IsNull())
    {
      return TransientResult(ent);
    }
  }

  return NullResult();
}