// 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.

//: i1 pdn 03.04.99  BUC60301

#include <Geom2d_Point.hxx>
#include <Interface_Check.hxx>
#include <Interface_HGraph.hxx>
#include <Interface_InterfaceModel.hxx>
#include <Interface_Macros.hxx>
#include <Message_Messenger.hxx>
#include <Message_ProgressScope.hxx>
#include <TColStd_HSequenceOfTransient.hxx>
#include <Transfer_ResultFromModel.hxx>
#include <Transfer_ResultFromTransient.hxx>
#include <Transfer_SimpleBinderOfTransient.hxx>
#include <Transfer_TransientProcess.hxx>
#include <XSControl_Controller.hxx>
#include <XSControl_TransferReader.hxx>
#include <XSControl_Vars.hxx>
#include <XSControl_WorkSession.hxx>

IMPLEMENT_STANDARD_RTTIEXT(XSControl_WorkSession, IFSelect_WorkSession)

namespace
{
static Standard_Mutex WS_GLOBAL_MUTEX; //!< Mutex to prevent data races during reading and writing.
}

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

XSControl_WorkSession::XSControl_WorkSession()
    : myTransferReader(new XSControl_TransferReader),
      myTransferWriter(new XSControl_TransferWriter),
      myVars(new XSControl_Vars)
{
}

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

void XSControl_WorkSession::ClearData(const Standard_Integer mode)
{
  // 1-2-3-4 : standard IFSelect
  if (mode >= 1 && mode <= 4)
    IFSelect_WorkSession::ClearData(mode);

  // 5 : Transferts seuls
  // 6 : Resultats forces seuls
  // 7 : Management, y compris tous transferts (forces/calcules), views

  if (mode == 5 || mode == 7)
  {
    myTransferReader->Clear(-1);
    myTransferWriter->Clear(-1);
  }
  if (mode == 6 && !myTransferReader.IsNull())
    myTransferReader->Clear(1);
  myTransferReader->SetGraph(HGraph());
}

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

Standard_Boolean XSControl_WorkSession::SelectNorm(const Standard_CString normname)
{
  const Standard_Mutex::Sentry aMutexLock(WS_GLOBAL_MUTEX);
  // Old norm and results
  myTransferReader->Clear(-1);
  //  ????  En toute rigueur, menage a faire dans XWS : virer les items
  //        ( a la limite, pourquoi pas, refaire XWS en entier)

  Handle(XSControl_Controller) newadapt = XSControl_Controller::Recorded(normname);
  if (newadapt.IsNull())
    return Standard_False;
  if (newadapt == myController)
    return Standard_True;
  SetController(newadapt);
  return Standard_True;
}

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

void XSControl_WorkSession::SetController(const Handle(XSControl_Controller)& ctl)
{
  myController = ctl;

  SetLibrary(myController->WorkLibrary());
  SetProtocol(myController->Protocol());

  ClearItems();
  ClearFinalModifiers();
  ClearShareOut(Standard_False);
  ClearFile();

  // Set worksession parameters from the controller
  Handle(XSControl_WorkSession) aWorkSession(this);
  myController->Customise(aWorkSession);

  myTransferReader->SetController(myController);
  myTransferWriter->SetController(myController);
}

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

Standard_CString XSControl_WorkSession::SelectedNorm(const Standard_Boolean rsc) const
{
  // JR/Hp :
  Standard_CString astr = (Standard_CString)(myController.IsNull() ? "" : myController->Name(rsc));
  return astr;
}

//              ##########################################
//              ############  Contexte de Transfert ######
//              ##########################################

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

void XSControl_WorkSession::SetAllContext(
  const NCollection_DataMap<TCollection_AsciiString, Handle(Standard_Transient)>& context)
{
  myContext                   = context;
  myTransferReader->Context() = context;
}

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

void XSControl_WorkSession::ClearContext()
{
  myContext.Clear();
  myTransferReader->Context().Clear();
}

//              ##########################################
//              ############    RESULTATS FORCES    ######
//              ##########################################

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

Standard_Boolean XSControl_WorkSession::PrintTransferStatus(const Standard_Integer num,
                                                            const Standard_Boolean wri,
                                                            Standard_OStream&      S) const
{
  const Handle(Transfer_FinderProcess)& FP = myTransferWriter->FinderProcess();
  Handle(Transfer_TransientProcess)     TP = myTransferReader->TransientProcess();

  Handle(Transfer_Binder)    binder;
  Handle(Transfer_Finder)    finder;
  Handle(Standard_Transient) ent;

  //   ***   WRITE  ***
  if (wri)
  {
    if (FP.IsNull())
      return Standard_False;
    if (num == 0)
      return Standard_False;

    Standard_Integer ne = 0, nr = 0, max = FP->NbMapped(), maxr = FP->NbRoots();
    if (num > 0)
    {
      if (num > max)
        return Standard_False;
      ne     = num;
      finder = FP->Mapped(ne);
      nr     = FP->RootIndex(finder);
    }
    else if (num < 0)
    {
      nr = -num;
      if (nr > maxr)
        return Standard_False;
      finder = FP->Root(nr);
      ne     = FP->MapIndex(finder);
    }

    S << "Transfer Write item n0." << ne << " of " << max;
    if (nr > 0)
    {
      S << "  ** Transfer Root n0." << ne;
    }
    S << std::endl;
    ent = FP->FindTransient(finder);
    S << " -> Type " << finder->DynamicType()->Name() << std::endl;
    FP->StartTrace(binder, finder, 0, 0); // pb sout/S
    if (!ent.IsNull())
    {
      S << " ** Resultat Transient, type " << ent->DynamicType()->Name();
      const Handle(Interface_InterfaceModel)& model = Model();
      if (!model.IsNull())
      {
        S << " In output Model, Entity ";
        model->Print(ent, S);
      }
      S << std::endl;
    }
  }

  //    ***   READ   ***
  else
  {
    if (TP.IsNull())
      return Standard_False;
    Handle(Interface_InterfaceModel) model = TP->Model();
    if (model.IsNull())
      std::cout << "No Model" << std::endl;
    else if (model != Model())
      std::cout << "Model different from the session" << std::endl;
    if (num == 0)
      return Standard_False;

    Standard_Integer ne = 0, nr = 0, max = TP->NbMapped(), maxr = TP->NbRoots();
    if (num > 0)
    {
      if (num > max)
        return Standard_False;
      ne  = num;
      ent = TP->Mapped(ne);
      nr  = TP->RootIndex(finder);
    }
    else if (num < 0)
    {
      nr = -num;
      if (nr > maxr)
        return Standard_False;
      ent = TP->Root(nr);
      ne  = TP->MapIndex(ent);
    }

    S << "Transfer Read item n0." << ne << " of " << max;
    if (nr > 0)
    {
      S << "  ** Transfer Root n0." << ne;
    }
    S << std::endl;
    if (!model.IsNull())
    {
      S << " In Model, Entity ";
      model->Print(ent, S);
    }
    binder = TP->MapItem(ne);
    S << std::endl;
    TP->StartTrace(binder, ent, 0, 0);
  }

  //   ***   CHECK (commun READ+WRITE)   ***
  if (!binder.IsNull())
  {
    const Handle(Interface_Check) ch = binder->Check();
    Standard_Integer              i, nbw = ch->NbWarnings(), nbf = ch->NbFails();
    if (nbw > 0)
    {
      S << " - Warnings : " << nbw << " :\n";
      for (i = 1; i <= nbw; i++)
        S << ch->CWarning(i) << std::endl;
    }
    if (nbf > 0)
    {
      S << " - Fails : " << nbf << " :\n";
      for (i = 1; i <= nbf; i++)
        S << ch->CFail(i) << std::endl;
    }
  }
  return Standard_True;
}

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

void XSControl_WorkSession::InitTransferReader(const Standard_Integer mode)
{
  if (mode == 0 || mode == 5)
    myTransferReader->Clear(-1); // full clear
  if (myTransferReader.IsNull())
    SetTransferReader(new XSControl_TransferReader);
  else
    SetTransferReader(myTransferReader);

  // mode = 0 fait par SetTransferReader suite a Nullify
  if (mode == 1)
  {
    if (!myTransferReader.IsNull())
      myTransferReader->Clear(-1);
    else
      SetTransferReader(new XSControl_TransferReader);
  }
  if (mode == 2)
  {
    Handle(Transfer_TransientProcess) TP = myTransferReader->TransientProcess();
    if (TP.IsNull())
    {
      TP = new Transfer_TransientProcess;
      myTransferReader->SetTransientProcess(TP);
      TP->SetGraph(HGraph());
    }
    Handle(TColStd_HSequenceOfTransient) lis = myTransferReader->RecordedList();
    Standard_Integer                     i, nb = lis->Length();
    for (i = 1; i <= nb; i++)
      TP->SetRoot(lis->Value(i));
  }
  if (mode == 3)
  {
    Handle(Transfer_TransientProcess) TP = myTransferReader->TransientProcess();
    if (TP.IsNull())
      return;
    Standard_Integer i, nb = TP->NbRoots();
    for (i = 1; i <= nb; i++)
      myTransferReader->RecordResult(TP->Root(i));
  }
  if (mode == 4 || mode == 5)
    myTransferReader->BeginTransfer();
}

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

void XSControl_WorkSession::SetTransferReader(const Handle(XSControl_TransferReader)& TR)
{
  if (myTransferReader != TR) // i1 pdn 03.04.99 BUC60301
    myTransferReader = TR;
  if (TR.IsNull())
    return;
  TR->SetController(myController);
  TR->SetGraph(HGraph());
  if (!TR->TransientProcess().IsNull())
    return;
  Handle(Transfer_TransientProcess) TP =
    new Transfer_TransientProcess(Model().IsNull() ? 100 : Model()->NbEntities() + 100);
  TP->SetGraph(HGraph());
  TP->SetErrorHandle(Standard_True);
  TR->SetTransientProcess(TP);
}

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

Handle(Transfer_TransientProcess) XSControl_WorkSession::MapReader() const
{
  return myTransferReader->TransientProcess();
}

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

Standard_Boolean XSControl_WorkSession::SetMapReader(const Handle(Transfer_TransientProcess)& TP)
{
  if (TP.IsNull())
    return Standard_False;
  if (TP->Model().IsNull())
    TP->SetModel(Model());
  TP->SetGraph(HGraph());
  if (TP->Model() != Model())
    return Standard_False;
  //  TR ne doit pas bouger, c est un "crochet" pour signatures, selections ...
  //  En revanche, mieux vaut le RAZ
  //  Handle(XSControl_TransferReader) TR = new XSControl_TransferReader;
  Handle(XSControl_TransferReader) TR = myTransferReader;
  TR->Clear(-1);

  SetTransferReader(TR);       // avec le meme mais le reinitialise
  TR->SetTransientProcess(TP); // et prend le nouveau TP
  return Standard_True;
}

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

Handle(Standard_Transient) XSControl_WorkSession::Result(const Handle(Standard_Transient)& ent,
                                                         const Standard_Integer mode) const
{
  Standard_Integer ouca = (mode % 10);
  Standard_Integer kica = (mode / 10);

  Handle(Transfer_Binder)          binder;
  Handle(Transfer_ResultFromModel) resu;

  if (ouca != 1)
    resu = myTransferReader->FinalResult(ent);
  if (mode == 20)
    return resu;

  if (!resu.IsNull())
    binder = resu->MainResult()->Binder();
  if (binder.IsNull() && ouca > 0)
    binder = myTransferReader->TransientProcess()->Find(ent);

  if (kica == 1)
    return binder;
  DeclareAndCast(Transfer_SimpleBinderOfTransient, trb, binder);
  if (!trb.IsNull())
    return trb->Result();
  return binder;
}

//              ##########################################
//              ############    TRANSFERT    #############
//              ##########################################

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

Standard_Integer XSControl_WorkSession::TransferReadOne(const Handle(Standard_Transient)& ent,
                                                        const Message_ProgressRange& theProgress)
{
  Handle(Interface_InterfaceModel) model = Model();
  if (ent == model)
    return TransferReadRoots(theProgress);

  Handle(TColStd_HSequenceOfTransient) list = GiveList(ent);
  if (list->Length() == 1)
    return myTransferReader->TransferOne(list->Value(1), Standard_True, theProgress);
  else
    return myTransferReader->TransferList(list, Standard_True, theProgress);
}

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

Standard_Integer XSControl_WorkSession::TransferReadRoots(const Message_ProgressRange& theProgress)
{
  return myTransferReader->TransferRoots(Graph(), theProgress);
}

//              ##########################################
//              ############    TRANSFERT  WRITE
//              ##########################################

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

Handle(Interface_InterfaceModel) XSControl_WorkSession::NewModel()
{
  const Standard_Mutex::Sentry     aMutexLock(WS_GLOBAL_MUTEX);
  Handle(Interface_InterfaceModel) newmod;
  if (myController.IsNull())
    return newmod;
  newmod = myController->NewModel();

  SetModel(newmod);
  if (!myTransferReader->TransientProcess().IsNull())
    myTransferReader->TransientProcess()->Clear();
  // clear all contains of WS
  myTransferReader->Clear(3);
  myTransferWriter->Clear(-1);

  return newmod;
}

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

IFSelect_ReturnStatus XSControl_WorkSession::TransferWriteShape(
  const TopoDS_Shape&          shape,
  const Standard_Boolean       compgraph,
  const Message_ProgressRange& theProgress)
{
  const Standard_Mutex::Sentry aMutexLock(WS_GLOBAL_MUTEX);
  IFSelect_ReturnStatus        status;
  if (myController.IsNull())
    return IFSelect_RetError;
  const Handle(Interface_InterfaceModel)& model = Model();
  if (model.IsNull() || shape.IsNull())
  {
    return IFSelect_RetVoid;
  }

  status = myTransferWriter->TransferWriteShape(model, shape, theProgress);
  if (theProgress.UserBreak())
    return IFSelect_RetStop;
  //  qui s occupe de tout, try/catch inclus

  // skl insert param compgraph for XDE writing 10.12.2003
  if (compgraph)
    ComputeGraph(Standard_True);

  return status;
}

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

Interface_CheckIterator XSControl_WorkSession::TransferWriteCheckList() const
{
  return myTransferWriter->ResultCheckList(Model());
}

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

void XSControl_WorkSession::ClearBinders()
{
  const Handle(Transfer_FinderProcess)& FP = myTransferWriter->FinderProcess();
  // Due to big number of chains of binders it is necessary to
  // collect head binders of each chain in the sequence
  TColStd_SequenceOfTransient aSeqBnd;
  TColStd_SequenceOfTransient aSeqShapes;
  Standard_Integer            i = 1;
  for (; i <= FP->NbMapped(); i++)
  {
    Handle(Transfer_Binder) bnd = FP->MapItem(i);
    if (!bnd.IsNull())
      aSeqBnd.Append(bnd);
    Handle(Standard_Transient) ash(FP->Mapped(i));
    aSeqShapes.Append(ash);
  }
  // removing finder process containing result of translation.
  FP->Clear();
  ClearData(1);
  ClearData(5);

  // removing each chain of binders
  while (aSeqBnd.Length() > 0)
  {
    Handle(Transfer_Binder)    aBnd = Handle(Transfer_Binder)::DownCast(aSeqBnd.Value(1));
    Handle(Standard_Transient) ash  = aSeqShapes.Value(1);
    aSeqBnd.Remove(1);
    aSeqShapes.Remove(1);
    ash.Nullify();
    while (!aBnd.IsNull())
    {
      Handle(Transfer_Binder) aBndNext = aBnd->NextResult();
      aBnd.Nullify();
      aBnd = aBndNext;
    }
  }
}