Salome HOME
Added a mechanism to store AIS_AngleDimension objects and display them all along...
[modules/geom.git] / src / IGESImport / IGESImport.cxx
1 // Copyright (C) 2007-2012  CEA/DEN, EDF R&D, OPEN CASCADE
2 //
3 // Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
4 // CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
5 //
6 // This library is free software; you can redistribute it and/or
7 // modify it under the terms of the GNU Lesser General Public
8 // License as published by the Free Software Foundation; either
9 // version 2.1 of the License.
10 //
11 // This library is distributed in the hope that it will be useful,
12 // but WITHOUT ANY WARRANTY; without even the implied warranty of
13 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 // Lesser General Public License for more details.
15 //
16 // You should have received a copy of the GNU Lesser General Public
17 // License along with this library; if not, write to the Free Software
18 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
19 //
20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
21 //
22 //  File:        IGESImport.cxx
23 //  Created:     Wed May 19 14:36:35 2004
24 //  Author:      Pavel TELKOV
25
26 #include "utilities.h"
27
28 #include <Basics_Utils.hxx>
29
30 #include <IFSelect_ReturnStatus.hxx>
31 #include <IGESControl_Reader.hxx>
32 #include <IGESData_IGESModel.hxx>
33 #include <IGESData_IGESEntity.hxx>
34
35 #include <Interface_Static.hxx>
36 #include <Interface_InterfaceModel.hxx>
37 #include <XSControl_TransferReader.hxx>
38 #include <XSControl_WorkSession.hxx>
39 #include <Transfer_TransientProcess.hxx>
40 #include <Transfer_Binder.hxx>
41 #include <TransferBRep.hxx>
42
43 #include <TNaming_Builder.hxx>
44 #include <TDF_TagSource.hxx>
45 #include <TDataStd_Name.hxx>
46 #include <TDF_Label.hxx>
47
48 #include <TCollection_HAsciiString.hxx>
49 #include <TopoDS_Shape.hxx>
50 #include <TopoDS_Vertex.hxx>
51 #include <BRep_Builder.hxx>
52 #include <gp_Pnt.hxx>
53
54 #ifdef WNT
55  #if defined IGESIMPORT_EXPORTS || defined IGESImport_EXPORTS
56   #if defined WIN32
57    #define IGESIMPORT_EXPORT __declspec( dllexport )
58   #else
59    #define IGESIMPORT_EXPORT
60   #endif
61  #else
62   #if defined WIN32
63    #define IGESIMPORT_EXPORT __declspec( dllimport )
64   #else
65    #define IGESIMPORT_EXPORT
66   #endif
67  #endif
68 #else
69  #define IGESIMPORT_EXPORT
70 #endif
71
72 //=============================================================================
73 /*!
74  *
75  */
76 //=============================================================================
77
78 extern "C"
79 {
80 IGESIMPORT_EXPORT
81   TopoDS_Shape Import (const TCollection_AsciiString& theFileName,
82                        const TCollection_AsciiString& theFormatName,
83                        TCollection_AsciiString&       theError,
84                        const TDF_Label&               theShapeLabel)
85   {
86     // Set "C" numeric locale to save numbers correctly
87     Kernel_Utils::Localizer loc;
88
89     IGESControl_Reader aReader;
90     TopoDS_Shape aResShape;
91     Interface_Static::SetCVal("xstep.cascade.unit","M");
92     try {
93       IFSelect_ReturnStatus status = aReader.ReadFile(theFileName.ToCString());
94
95       if (status == IFSelect_RetDone) {
96
97         if (theFormatName == "IGES_UNIT") {
98           Handle(IGESData_IGESModel) aModel =
99             Handle(IGESData_IGESModel)::DownCast(aReader.Model());
100           gp_Pnt P(1.0,0.0,0.0);
101           if (!aModel.IsNull()) {
102             Handle(TCollection_HAsciiString) aUnitName =
103               aModel->GlobalSection().UnitName();
104             //cout<<"aUnitName = "<<aUnitName->ToCString()<<endl;
105             //cout<<"aUnitFlag = "<<aModel->GlobalSection().UnitFlag()<<endl;
106             if (aUnitName->String()=="MM") {
107               P = gp_Pnt(0.001,0.0,0.0);
108             }
109             else if (aUnitName->String()=="CM") {
110               P = gp_Pnt(0.01,0.0,0.0);
111             }
112           }
113           BRep_Builder B;
114           TopoDS_Vertex V;
115           B.MakeVertex(V,P,1.e-7);
116           aResShape = V;
117           return aResShape;
118         }
119         if (theFormatName == "IGES_SCALE") {
120           //cout<<"need re-scale a model"<<endl;
121           // set UnitFlag to 'meter'
122           Handle(IGESData_IGESModel) aModel =
123             Handle(IGESData_IGESModel)::DownCast(aReader.Model());
124           if (!aModel.IsNull()) {
125             IGESData_GlobalSection aGS = aModel->GlobalSection();
126             aGS.SetUnitFlag(6);
127             aModel->SetGlobalSection(aGS);
128           }
129         }
130
131         MESSAGE("ImportIGES : all Geometry Transfer");
132         //OCC 5.1.2 porting
133         //     aReader.Clear();
134         //     aReader.TransferRoots(false);
135         aReader.ClearShapes();
136         aReader.TransferRoots();
137
138         MESSAGE("ImportIGES : count of shapes produced = " << aReader.NbShapes());
139         aResShape = aReader.OneShape();
140
141         // BEGIN: Store names of sub-shapes from file
142         Handle(Interface_InterfaceModel) Model = aReader.WS()->Model();
143         Handle(XSControl_TransferReader) TR = aReader.WS()->TransferReader();
144         if (!TR.IsNull()) {
145           Handle(Transfer_TransientProcess) TP = /*TransientProcess();*/TR->TransientProcess();
146           Standard_Integer nb = Model->NbEntities();
147           for (Standard_Integer i = 1; i <= nb; i++) {
148             Handle(IGESData_IGESEntity) ent = Handle(IGESData_IGESEntity)::DownCast(Model->Value(i));
149             if (ent.IsNull() || ! ent->HasName()) continue;
150
151             // find target shape
152             Handle(Transfer_Binder) binder = TP->Find(ent);
153             if (binder.IsNull()) continue;
154             TopoDS_Shape S = TransferBRep::ShapeResult(binder);
155             if (S.IsNull()) continue;
156
157             // create label and set shape
158             TDF_Label L;
159             TDF_TagSource aTag;
160             L = aTag.NewChild(theShapeLabel);
161             TNaming_Builder tnBuild (L);
162             tnBuild.Generated(S);
163
164             // set a name
165             TCollection_AsciiString string = ent->NameValue()->String();
166             string.LeftAdjust();
167             string.RightAdjust();
168             TCollection_ExtendedString str (string);
169             TDataStd_Name::Set(L, str);
170           }
171         }
172         // END: Store names
173       } else {
174 //        switch (status) {
175 //        case IFSelect_RetVoid:
176 //          theError = "Nothing created or No data to process";
177 //          break;
178 //        case IFSelect_RetError:
179 //          theError = "Error in command or input data";
180 //          break;
181 //        case IFSelect_RetFail:
182 //          theError = "Execution was run, but has failed";
183 //          break;
184 //        case IFSelect_RetStop:
185 //          theError = "Execution has been stopped. Quite possible, an exception was raised";
186 //          break;
187 //        default:
188 //          break;
189 //        }
190         theError = "Wrong format of the imported file. Can't import file.";
191         aResShape.Nullify();
192       }
193     }
194     catch(Standard_Failure) {
195       Handle(Standard_Failure) aFail = Standard_Failure::Caught();
196       theError = aFail->GetMessageString();
197       aResShape.Nullify();
198     }
199     return aResShape;
200   }
201 }