Salome HOME
CCAR:remove trailing whitespace
[modules/geom.git] / src / STEPImport / STEPImport.cxx
index 3fb9cc511a2bec645797b29d5470f0c1cd9beced..05503980521931e8cd4d7266bea56528ba052f4e 100644 (file)
@@ -1,45 +1,81 @@
-// Copyright (C) 2005  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
-// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
-// 
-// 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.
-// 
-// 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.
+//  Copyright (C) 2007-2010  CEA/DEN, EDF R&D, OPEN CASCADE
 //
-// 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+//  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
 //
-// See http://www.salome-platform.org/
+//  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.
 //
-// File:       STEPImport.cxx
-// Created:    Wed May 19 14:41:10 2004
-// Author:     Pavel TELKOV
-//             <ptv@mutex.nnov.opencascade.com>
+//  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., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+//  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
+//  File:        STEPImport.cxx
+//  Created:     Wed May 19 14:41:10 2004
+//  Author:      Pavel TELKOV
 
 #include "utilities.h"
 
+#include <Basics_Utils.hxx>
+
 #include <BRep_Builder.hxx>
 
 #include <IFSelect_ReturnStatus.hxx>
 
 #include <STEPControl_Reader.hxx>
+#include <StepBasic_ProductDefinition.hxx>
+#include <StepBasic_ProductDefinitionFormation.hxx>
+#include <StepBasic_Product.hxx>
+#include <Interface_InterfaceModel.hxx>
+#include <XSControl_TransferReader.hxx>
+#include <XSControl_WorkSession.hxx>
+
+#include <Transfer_Binder.hxx>
+#include <TNaming_Builder.hxx>
+#include <TDataStd_Name.hxx>
+#include <Transfer_TransientProcess.hxx>
+#include <TransferBRep.hxx>
 
 #include <TCollection_AsciiString.hxx>
 #include <TopoDS_Compound.hxx>
 #include <TopoDS_Shape.hxx>
+#include <TDF_Label.hxx>
+#include <TDF_Tool.hxx>
+#include <Interface_Static.hxx>
 
+#include <TopTools_IndexedMapOfShape.hxx>
+#include <TopExp.hxx>
+#include <TopoDS_Iterator.hxx>
+#include <BRepTools.hxx>
+
+#include <Standard_Failure.hxx>
 #include <Standard_ErrorHandler.hxx> // CAREFUL ! position of this file is critic : see Lucien PIGNOLONI / OCC
 
 #ifdef WNT
-#include <SALOME_WNT.hxx>
+ #if defined STEPIMPORT_EXPORTS || defined STEPImport_EXPORTS
+  #if defined WIN32
+   #define STEPIMPORT_EXPORT __declspec( dllexport )
+  #else
+   #define STEPIMPORT_EXPORT
+  #endif
+ #else
+  #if defined WIN32
+   #define STEPIMPORT_EXPORT __declspec( dllimport )
+  #else
+   #define STEPIMPORT_EXPORT
+  #endif
+ #endif
 #else
-#define SALOME_WNT_EXPORT
+ #define STEPIMPORT_EXPORT
 #endif
 
 //=============================================================================
 
 extern "C"
 {
-SALOME_WNT_EXPORT
+  STEPIMPORT_EXPORT
   TopoDS_Shape Import (const TCollection_AsciiString& theFileName,
-                       TCollection_AsciiString&       theError)
+                       const TCollection_AsciiString& /*theFormatName*/,
+                       TCollection_AsciiString&       theError,
+                       const TDF_Label&               theShapeLabel)
   {
     MESSAGE("Import STEP model from file " << theFileName.ToCString());
+    // Set "C" numeric locale to save numbers correctly
+    Kernel_Utils::Localizer loc;
     TopoDS_Shape aResShape;
     //VRV: OCC 4.0 migration
     STEPControl_Reader aReader;
+    //VSR: 16/09/09: Convert to METERS
+    Interface_Static::SetCVal("xstep.cascade.unit","M");
+    Interface_Static::SetIVal("read.step.ideas", 1);
+    Interface_Static::SetIVal("read.step.nonmanifold", 1);
     //VRV: OCC 4.0 migration
     TopoDS_Compound compound;
     BRep_Builder B;
-    B.MakeCompound( compound );
+    B.MakeCompound(compound);
     try {
+#if (OCC_VERSION_MAJOR << 16 | OCC_VERSION_MINOR << 8 | OCC_VERSION_MAINTENANCE) > 0x060100
+      OCC_CATCH_SIGNALS;
+#endif
       IFSelect_ReturnStatus status = aReader.ReadFile(theFileName.ToCString());
 
       if (status == IFSelect_RetDone) {
-       Standard_Boolean failsonly = Standard_False ;
-       aReader.PrintCheckLoad (failsonly, IFSelect_ItemsByEntity);
-       /* Root transfers */
-       Standard_Integer nbr = aReader.NbRootsForTransfer();
-       aReader.PrintCheckTransfer (failsonly, IFSelect_ItemsByEntity);
-
-       for ( Standard_Integer n=1; n <= nbr; n++) {
-         Standard_Boolean ok = aReader.TransferRoot(n);
-         /* Collecting resulting entities */
-         Standard_Integer nbs = aReader.NbShapes();
-         if (!ok || nbs == 0)
-         {
-           // THROW_SALOME_CORBA_EXCEPTION("Exception catched in GEOM_Gen_i::ImportStep", SALOME::BAD_PARAM);
-           continue; // skip empty root
-         }
-         /* For a single entity */
-         else if (nbr == 1 && nbs == 1) {
-           aResShape = aReader.Shape(1);
-           break;
-         }
-
-         for ( Standard_Integer i=1; i<=nbs; i++ ) {
-           TopoDS_Shape aShape = aReader.Shape(i);
-           if ( aShape.IsNull() ) {
-             // THROW_SALOME_CORBA_EXCEPTION("Null shape in GEOM_Gen_i::ImportStep", SALOME::BAD_PARAM) ;
-             //return aResShape;
-             continue;
-           }
-           else {
-             B.Add( compound, aShape ) ;
-           }
-         }
-       }
-       if ( aResShape.IsNull() )
-         aResShape = compound;
-
-      } else {
+        Standard_Boolean failsonly = Standard_False;
+        aReader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity);
+        /* Root transfers */
+        Standard_Integer nbr = aReader.NbRootsForTransfer();
+        aReader.PrintCheckTransfer(failsonly, IFSelect_ItemsByEntity);
+
+        for (Standard_Integer n = 1; n <= nbr; n++) {
+          Standard_Boolean ok = aReader.TransferRoot(n);
+          /* Collecting resulting entities */
+          Standard_Integer nbs = aReader.NbShapes();
+          if (!ok || nbs == 0)
+          {
+            // THROW_SALOME_CORBA_EXCEPTION("Exception catched in GEOM_Gen_i::ImportStep", SALOME::BAD_PARAM);
+            continue; // skip empty root
+          }
+          /* For a single entity */
+          else if (nbr == 1 && nbs == 1) {
+            aResShape = aReader.Shape(1);
+            break;
+          }
+
+          for (Standard_Integer i = 1; i <= nbs; i++) {
+            TopoDS_Shape aShape = aReader.Shape(i);
+            if (aShape.IsNull()) {
+              // THROW_SALOME_CORBA_EXCEPTION("Null shape in GEOM_Gen_i::ImportStep", SALOME::BAD_PARAM) ;
+              //return aResShape;
+              continue;
+            }
+            else {
+              B.Add(compound, aShape);
+            }
+          }
+        }
+        if (aResShape.IsNull())
+          aResShape = compound;
+
+        // BEGIN: Store names of sub-shapes from file
+        TopTools_IndexedMapOfShape anIndices;
+        TopExp::MapShapes(aResShape, anIndices);
+
+        Handle(Interface_InterfaceModel) Model = aReader.WS()->Model();
+        Handle(XSControl_TransferReader) TR = aReader.WS()->TransferReader();
+        if (!TR.IsNull()) {
+          Handle(Transfer_TransientProcess) TP = TR->TransientProcess();
+          Handle(Standard_Type) tPD  = STANDARD_TYPE(StepBasic_ProductDefinition);
+
+          Standard_Integer nb = Model->NbEntities();
+          for (Standard_Integer ie = 1; ie <= nb; ie++) {
+            Handle(Standard_Transient) enti = Model->Value(ie);
+            if (enti->DynamicType() != tPD) continue;
+
+            Handle(StepBasic_ProductDefinition) PD =
+              Handle(StepBasic_ProductDefinition)::DownCast(enti);
+            if (PD.IsNull()) continue;
+
+            Handle(StepBasic_Product) Prod = PD->Formation()->OfProduct();
+            if (Prod->Name()->UsefullLength() <= 0) continue;
+
+            Handle(TCollection_HAsciiString) aName = Prod->Name();
+            TCollection_ExtendedString aNameExt (aName->ToCString());
+
+            // special check to pass names like "Open CASCADE STEP translator 6.3 1"
+            TCollection_AsciiString aSkipName ("Open CASCADE STEP translator");
+            if (aName->Length() >= aSkipName.Length()) {
+              if (aName->String().SubString(1, aSkipName.Length()).IsEqual(aSkipName))
+                continue;
+            }
+
+            // find target shape
+            Handle(Transfer_Binder) binder = TP->Find(enti);
+            if (binder.IsNull()) continue;
+            TopoDS_Shape S = TransferBRep::ShapeResult(binder);
+            if (S.IsNull()) continue;
+
+            // as PRODUCT can be included in the main shape
+            // several times, we look here for all iclusions.
+            Standard_Integer isub, nbSubs = anIndices.Extent();
+            for (isub = 1; isub <= nbSubs; isub++)
+            {
+              TopoDS_Shape aSub = anIndices.FindKey(isub);
+              if (aSub.IsPartner(S)) {
+                // create label and set shape
+                TDF_Label L;
+                TDF_TagSource aTag;
+                L = aTag.NewChild(theShapeLabel);
+                TNaming_Builder tnBuild (L);
+                //tnBuild.Generated(S);
+                tnBuild.Generated(aSub);
+
+                // set a name
+                TDataStd_Name::Set(L, aNameExt);
+              }
+            }
+          }
+        }
+        // END: Store names
+      }
+      else {
 //        switch (status) {
 //        case IFSelect_RetVoid:
 //          theError = "Nothing created or No data to process";
@@ -120,7 +228,7 @@ SALOME_WNT_EXPORT
 //          break;
 //        }
         theError = "Wrong format of the imported file. Can't import file.";
-       aResShape.Nullify();
+        aResShape.Nullify();
       }
     }
     catch (Standard_Failure) {
@@ -128,6 +236,7 @@ SALOME_WNT_EXPORT
       theError = aFail->GetMessageString();
       aResShape.Nullify();
     }
+    // Return previous locale
     return aResShape;
   }
 }