]> SALOME platform Git repositories - plugins/canrecplugin.git/blob - src/CANRECPLUGINEngine/CANRECPluginImpl_IOperations.cxx
Salome HOME
81b20d8ffa097c894ed4af3d6d6d934bbccea039
[plugins/canrecplugin.git] / src / CANRECPLUGINEngine / CANRECPluginImpl_IOperations.cxx
1 // Copyright (C) 2014-2015  OPEN CASCADE
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19
20 #include "CANRECPluginImpl_IOperations.hxx"
21 #include "CANRECPluginImpl_Driver.hxx"
22 #include "CANRECPluginImpl_ICanRec.hxx"
23 #include "CANRECPluginImpl_Types.hxx"
24
25 #include <GEOM_PythonDump.hxx>
26
27 #include <utilities.h>
28
29 #include <Standard_ErrorHandler.hxx>
30
31 CANRECPluginImpl_IOperations::CANRECPluginImpl_IOperations( GEOM_Engine* theEngine, int theDocID)
32   : GEOM_IOperations( theEngine, theDocID )
33 {
34   MESSAGE( "CANRECPluginImpl_IOperations::CANRECPluginImpl_IOperations()" );
35 }
36
37 CANRECPluginImpl_IOperations::~CANRECPluginImpl_IOperations()
38 {
39   MESSAGE( "CANRECPluginImpl_IOperations::~CANRECPluginImpl_IOperations()" );
40 }
41
42 //=============================================================================
43 /*!
44  *  Create a canonical recognition for the object
45  *  \param theMergeSurf set merging surfaces
46  *  \param theMergeCurves set merging curves
47  *  \param theTolerance set tolerance
48  *  \param theObject is initial object
49  *  \param theStat is the statistics. A set of integers. For details please
50  *         see CANRECPlugin.idl
51  *  \return New GEOM_Object, containing the created shape.
52  */
53 //=============================================================================
54 Handle(GEOM_Object)
55 CANRECPluginImpl_IOperations::MakeCanonicalRecognition
56                   (Handle(GEOM_Object)                theObject,
57                    bool                               theMergeSurf,
58                    bool                               theMergeCurves,
59                    double                             theTolerance,
60                    Handle_TColStd_HSequenceOfInteger &theStat)
61 {
62   SetErrorCode( KO );
63
64   if (theStat.IsNull()) {
65     theStat = new TColStd_HSequenceOfInteger;
66   } else {
67     theStat->Clear();
68   }
69
70   if ( theObject.IsNull() ) return NULL;
71
72   // Add a new object
73   Handle(GEOM_Object) aShape = GetEngine()->AddObject( GetDocID(), GEOM_CANONICALRECOGNITION );
74
75   // Add a new shape function with parameters
76   Handle(GEOM_Function) aFunction = aShape->AddFunction( CANRECPluginImpl_Driver::GetID(), CANONICALRECOGNITION_OBJ );
77   if ( aFunction.IsNull() ) return NULL;
78
79   // Check if the function is set correctly
80   if ( aFunction->GetDriverGUID() != CANRECPluginImpl_Driver::GetID() ) return NULL;
81
82   Handle(GEOM_Function) aRefFunction = theObject->GetLastFunction();
83
84   if ( aRefFunction.IsNull() ) return NULL;
85
86   CANRECPluginImpl_ICanRec aData( aFunction );
87   aData.SetNeedMergeSurf( theMergeSurf );
88   aData.SetNeedMergeCurves( theMergeCurves );
89   aData.SetTolerance( theTolerance );
90   aData.SetObject( aRefFunction );
91
92   // Compute the resulting value
93   try {
94     OCC_CATCH_SIGNALS;
95     if ( !GetSolver()->ComputeFunction( aFunction ) ) {
96       SetErrorCode( "Canonical Recognition driver failed" );
97       return NULL;
98     }
99   }
100   catch ( Standard_Failure ) {
101     Handle(Standard_Failure) aFail = Standard_Failure::Caught();
102     SetErrorCode( aFail->GetMessageString() );
103     return NULL;
104   }
105
106   Handle(TColStd_HArray1OfInteger) anArray = aData.GetStatistics();
107
108   if (anArray.IsNull() == Standard_False) {
109     Standard_Integer       i   = anArray->Lower();
110     const Standard_Integer iUp = anArray->Upper();
111
112     for (; i <= iUp; ++i) {
113       theStat->Append(anArray->Value(i));
114     }
115   }
116   
117   // Make a Python command
118   GEOM::TPythonDump(aFunction) << aShape << " = " << "geompy.MakeCanonicalRecognition(" << 
119     theObject << ", " << theMergeSurf << ", " << theMergeCurves << ", " << theTolerance <<  ")";
120
121   SetErrorCode( OK );
122
123   return aShape;
124 }
125
126 //=============================================================================
127 /*!
128  *  \brief Get the number of canonical faces in the shape. 
129  *
130  *  \param theObject the input object (solid, compound, compsolid).
131  *
132  *  \return the number of canonical faces in the object.
133  */
134 //=============================================================================
135 Standard_Integer CANRECPluginImpl_IOperations::GetNbCanonicalFaces
136                                     (const Handle(GEOM_Object) &theObject)
137 {
138   SetErrorCode(KO);
139
140   TopoDS_Shape aShape = theObject->GetValue();
141
142   if (aShape.IsNull()) {
143     return -1;
144   }
145
146   const Standard_Integer aResult =
147     CANRECPluginImpl_Driver::GetNbCanonicalFaces(aShape);
148
149   if (aResult >= 0) {
150     SetErrorCode(OK);
151
152     return aResult;
153   }
154
155   return -1;
156 }