qoccdocument.cpp

Go to the documentation of this file.

/************************************************************************************
**
** This file is part of the QtOPENCASCADE Toolkit.
**
** Copyright (C) 2006, 2007, 2008 QtOCC Team Members
**               Peter Dolbey, Marc Britten, Stephane Routelous
**               Stefan Boeykens, Pawel Dobrolowski, Walter Steffe
**               Álvaro Castro Castilla, Dirk Burski, Fotis Sioutis
**
** This library is free software; you can redistribute it and/or
** modify it under the terms of the GNU Lesser General Public
** License as published by the Free Software Foundation; either
** version 2.1 of the License, or (at your option) any later version.
** 
** This library is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
** Lesser General Public License for more details.
** 
** You should have received a copy of the GNU Lesser General Public
** License along with this library; if not, write to the Free Software
** Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
**
***********************************************************************************/
/*!
\class  QoccDocument
\brief  This class provides a base object for "proxy" OpenCascade objects,
        and a set of starter files for new classes
\author Peter C. Dolbey
*/

#include <qnamespace.h>
#include "qoccapplication.h"
#include "qoccdocument.h"
#include "qoccinternal.h"

#include <TNaming_NamedShape.hxx>
#include <TDF_ChildIterator.hxx>
#include <TDataStd_Integer.hxx>
#include <TPrsStd_AISViewer.hxx>
#include <GEOM_Object.hxx>

QoccDocument::QoccDocument( const Handle(TDocStd_Document)& aDoc, int anID )
{
        myDocument = aDoc;
        myID  = anID;
}

QoccDocument::~QoccDocument( )
{
}

void QoccDocument::undo()
{
        qGeomApp->Undo( this );
}

void QoccDocument::redo()
{
        qGeomApp->Redo( this );
}

Handle(TPrsStd_AISPresentation) QoccDocument::RegisterPresentation(Handle(GEOM_Object) TheGeomObject)
{
  TDF_Label aResultLabel = TheGeomObject->GetLastFunction()->GetEntry().FindChild(RESULT_LABEL);
  Handle(TPrsStd_AISPresentation) prs_main = TPrsStd_AISPresentation::Set(aResultLabel, TNaming_NamedShape::GetID());
  prs_main->Display(Standard_True);
  return prs_main;
}

bool QoccDocument::RemovePresentation(Handle(GEOM_Object) TheGeomObject)
{
  if (TheGeomObject != NULL)
  {
        if (!TheGeomObject.IsNull())
        {
          TDF_Label aResultLabel = TheGeomObject->GetLastFunction()->GetEntry().FindChild(RESULT_LABEL);
          Handle(TPrsStd_AISPresentation) prs_ref;
          aResultLabel.FindAttribute(TPrsStd_AISPresentation::GetID(), prs_ref);
          prs_ref->Erase(Standard_True);
        }
  }
  return true;
}

void QoccDocument::UpdatePresentations(TDF_Label aRoot)
{
  TDF_Label LabGEOM = aRoot;

  if (!LabGEOM.IsNull())
  {
        for (TDF_ChildIterator it(LabGEOM); it.More(); it.Next())
        {
          TDF_Label Label = it.Value();
          Handle_GEOM_Object this_obj = GEOM_Object::GetObject  (Label);
          if (!this_obj.IsNull())
          {
                int func_num = this_obj->GetNbFunctions();
                for (int i = 1 ; i <= func_num; i++)
                {
                  Handle(GEOM_Function) aFunction = this_obj->GetFunction(i);
                  if (!aFunction.IsNull())
                  {
                        TDF_Label aResultLabel = aFunction->GetEntry().FindChild(RESULT_LABEL);
                        Handle(TNaming_NamedShape) TNS;
                        if (!aResultLabel.FindAttribute(TNaming_NamedShape::GetID(), TNS))
                        {
                          continue;
                        }
                        Handle(TDataStd_Integer) TDI;
                        if (aResultLabel.FindAttribute(TDataStd_Integer::GetID(), TDI))
                        {
                          if(!TDI->Get())
                          {
                                continue;
                          }
                        }
                        Handle(TPrsStd_AISPresentation) prs;
                        if (!aResultLabel.FindAttribute(TPrsStd_AISPresentation::GetID(),prs))
                        {
                          prs = TPrsStd_AISPresentation::Set(aResultLabel,TNaming_NamedShape::GetID());
                        }

                        if (prs->IsDisplayed())
                        {
                          prs->Display(Standard_True);
                        }
                  }
                }
          }
          UpdatePresentations(Label);
        }
  }
}

void QoccDocument::SetDocumentAndID(Handle(TDocStd_Document) aDocument, int anID)
{
        myDocument = aDocument;
        myID  = anID;
}

---