]> OCCT Git - occt.git/blame_incremental - samples/mfc/standard/Common/ImportExport/ImportExport.cpp
projects / occt.git / blame_incremental
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (non-incremental) | history | HEAD
Testing - Fix master validation workflow (#611)
[occt.git] / samples / mfc / standard / Common / ImportExport / ImportExport.cpp
... / ...
CommitLineData
1// ImportExport.cpp: implementation of the CImportExport class.
2//
3//////////////////////////////////////////////////////////////////////
4
5#include "stdafx.h"
6#include "ImportExport.h"
7#include <OCC_App.h>
8
9#include "SaveSTEPDlg.h"
10
11#include "TColStd_SequenceOfAsciiString.hxx"
12#include "TColStd_SequenceOfExtendedString.hxx"
13#include "OSD_Timer.hxx"
14
15#include "IGESControl_Reader.hxx"
16#include "STEPControl_Controller.hxx"
17
18#include <BRepAlgo.hxx>
19#include <IGESControl_Controller.hxx>
20#include <IGESControl_Writer.hxx>
21#include <Interface_Static.hxx>
22#include <STEPControl_Reader.hxx>
23#include <Geom_Curve.hxx>
24#include <STEPControl_Writer.hxx>
25#include <TopoDS_Compound.hxx>
26#include <Geom_Line.hxx>
27#include <StlAPI_Writer.hxx>
28#include <VrmlAPI_Writer.hxx>
29#include <VrmlData_Scene.hxx>
30#include <VrmlData_ShapeConvert.hxx>
31#include <VrmlData_Appearance.hxx>
32#include <VrmlData_Material.hxx>
33#include <VrmlData_Group.hxx>
34#include <VrmlData_ListOfNode.hxx>
35#include <VrmlData_ShapeNode.hxx>
36
37#include <XSControl_WorkSession.hxx>
38#include <XSControl_TransferReader.hxx>
39#include <STEPConstruct_Styles.hxx>
40#include <TColStd_HSequenceOfTransient.hxx>
41#include <STEPConstruct.hxx>
42#include <StepVisual_StyledItem.hxx>
43
44#ifdef _DEBUG
45#undef THIS_FILE
46static char THIS_FILE[]=__FILE__;
47//#define new DEBUG_NEW
48#endif
49
50//////////////////////////////////////////////////////////////////////
51// Construction/Destruction
52//////////////////////////////////////////////////////////////////////
53
54Handle(TopTools_HSequenceOfShape) CImportExport::BuildSequenceFromContext(const Handle(AIS_InteractiveContext)& anInteractiveContext,
55 Handle(Quantity_HArray1OfColor)& anArrayOfColors,
56 Handle(TColStd_HArray1OfReal)& anArrayOfTransparencies)
57{
58 Handle(TopTools_HSequenceOfShape) aSequence;
59 Standard_Integer nb = anInteractiveContext->NbSelected(), i = 1;
60 if (!nb)
61 return aSequence;
62
63 aSequence = new TopTools_HSequenceOfShape();
64 anArrayOfColors = new Quantity_HArray1OfColor(1, nb);
65 anArrayOfTransparencies = new TColStd_HArray1OfReal (1, nb);
66
67 Handle(AIS_InteractiveObject) picked;
68 for (anInteractiveContext->InitSelected(); anInteractiveContext->MoreSelected(); anInteractiveContext->NextSelected())
69 {
70 picked = anInteractiveContext->SelectedInteractive();
71 if (picked->IsKind (STANDARD_TYPE (AIS_Shape)))
72 {
73 Handle(AIS_Shape) aisShape = Handle(AIS_Shape)::DownCast(picked);
74 TopoDS_Shape aShape = aisShape->Shape();
75 aSequence->Append(aShape);
76
77 Quantity_Color color;
78 aisShape->Color(color);
79 anArrayOfColors->SetValue(i, color);
80
81 Standard_Real transparency = aisShape->Transparency();
82 anArrayOfTransparencies->SetValue(i, transparency);
83
84 i++;
85 }
86 }
87 return aSequence;
88}
89
90//======================================================================
91//= =
92//= BREP =
93//= =
94//======================================================================
95
96int CImportExport::ReadBREP (const Handle(AIS_InteractiveContext)& anInteractiveContext)
97{
98 Handle(TopTools_HSequenceOfShape) aSequence = CImportExport::ReadBREP();
99 if(aSequence->IsEmpty())
100 return 1;
101 Handle(AIS_Shape) aShape;
102 for(int i=1;i<= aSequence->Length();i++){
103 aShape = new AIS_Shape(aSequence->Value(i));
104 anInteractiveContext->SetDisplayMode(aShape, 1, Standard_False);
105 anInteractiveContext->Display(aShape, Standard_False);
106 const Handle(AIS_InteractiveObject)& aPrs = aShape; // A small trick to avoid compiler error (C2668).
107 anInteractiveContext->SetSelected (aPrs, Standard_False);
108 }
109 return 0;
110}
111
112Handle(TopTools_HSequenceOfShape) CImportExport::ReadBREP()
113{
114 CFileDialog dlg(TRUE,
115 NULL,
116 NULL,
117 OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
118 L"BREP Files (*.brep , *.rle)|*.brep; *.BREP; *.rle; *.RLE; |All Files (*.*)|*.*||",
119 NULL );
120
121 CString SHAREPATHValue;
122 SHAREPATHValue.GetEnvironmentVariable (L"CSF_OCCTDataPath");
123 CString initdir = (SHAREPATHValue + "\\occ");
124
125 dlg.m_ofn.lpstrInitialDir = initdir;
126
127 Handle(TopTools_HSequenceOfShape) aSequence= new TopTools_HSequenceOfShape();
128
129 if (dlg.DoModal() == IDOK)
130 {
131 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_WAIT));
132 CString filename = dlg.GetPathName();
133 TopoDS_Shape aShape;
134 Standard_Boolean result = ReadBREP (filename, aShape);
135 if (result)
136 {
137 if (!BRepAlgo::IsValid(aShape))
138 MessageBoxW (AfxGetMainWnd()->m_hWnd, L"Warning: The shape is not valid!", L"Cascade Warning", MB_ICONWARNING);
139
140 aSequence->Append(aShape);
141 }
142 else
143 MessageBoxW (AfxGetMainWnd()->m_hWnd, L"Error: The file was not read", L"Cascade Error", MB_ICONERROR);
144
145 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_ARROW));
146 }
147
148 return aSequence;
149}
150//----------------------------------------------------------------------
151
152Standard_Boolean CImportExport::ReadBREP(CString aFileName,
153 TopoDS_Shape& aShape)
154{
155 aShape.Nullify();
156
157 std::filebuf aFileBuf;
158 std::istream aStream (&aFileBuf);
159 if (!aFileBuf.open (aFileName, std::ios::in))
160 {
161 return Standard_False;
162 }
163
164 BRep_Builder aBuilder;
165 BRepTools::Read (aShape, aStream, aBuilder);
166 return !aShape.IsNull();
167}
168
169void CImportExport::SaveBREP(const Handle(AIS_InteractiveContext)& anInteractiveContext)
170{
171 anInteractiveContext->InitSelected();
172 if (anInteractiveContext->NbSelected() == 0){
173 AfxMessageBox (L"No shape selected for export!");
174 return;
175 }
176 Handle(TopTools_HSequenceOfShape) aHSequenceOfShape;
177 Handle(Quantity_HArray1OfColor) anArrayOfColors;
178 Handle(TColStd_HArray1OfReal) anArrayOfTransparencies;
179 aHSequenceOfShape = BuildSequenceFromContext(anInteractiveContext, anArrayOfColors, anArrayOfTransparencies);
180 int aLength = aHSequenceOfShape->Length();
181 if (aLength == 1){
182 TopoDS_Shape RES = aHSequenceOfShape->Value(1);
183 CImportExport::SaveBREP(RES);
184 } else {
185 TopoDS_Compound RES;
186 BRep_Builder MKCP;
187 MKCP.MakeCompound(RES);
188 for (Standard_Integer i=1;i<=aLength;i++)
189 {
190 TopoDS_Shape aShape= aHSequenceOfShape->Value(i);
191 if ( aShape.IsNull() )
192 {
193 continue;
194 }
195 MKCP.Add(RES, aShape);
196 }
197 CImportExport::SaveBREP(RES);
198 }
199}
200
201Standard_Boolean CImportExport::SaveBREP(const TopoDS_Shape& aShape)
202{
203 CFileDialog dlg (FALSE, L"*.brep",NULL,OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
204 L"BREP Files (*.brep)|*.brep;|BREP Files (*.BREP)|*.BREP;||", NULL);
205
206CString SHAREPATHValue;
207SHAREPATHValue.GetEnvironmentVariable (L"CSF_OCCTDataPath");
208CString initdir = (SHAREPATHValue + "\\occ");
209
210dlg.m_ofn.lpstrInitialDir = initdir;
211
212 Standard_Boolean result = Standard_False;
213 if (dlg.DoModal() == IDOK)
214 {
215 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_WAIT));
216 CString filename = dlg.GetPathName();
217 result = SaveBREP (filename, aShape);
218 if (!result)
219 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd,
220 L"Error : The shape or shapes were not saved.",
221 L"CasCade Error", MB_ICONERROR);
222 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_ARROW));
223 }
224 return result;
225}
226
227//----------------------------------------------------------------------------------------
228Standard_Boolean CImportExport::SaveBREP (CString aFileName,
229 const TopoDS_Shape& aShape)
230{
231 std::filebuf aFileBuf;
232 std::ostream aStream (&aFileBuf);
233 if (!aFileBuf.open (aFileName, std::ios::out))
234 {
235 return Standard_False;
236 }
237
238 BRepTools::Write (aShape, aStream);
239 return Standard_True;
240}
241
242//======================================================================
243//= =
244//= IGES =
245//= =
246//======================================================================
247
248
249
250void CImportExport::ReadIGES(const Handle(AIS_InteractiveContext)& anInteractiveContext)
251{
252 Handle(TopTools_HSequenceOfShape) aSequence = CImportExport::ReadIGES();
253 for(int i=1;i<= aSequence->Length();i++)
254 anInteractiveContext->Display (new AIS_Shape (aSequence->Value (i)), Standard_False);
255 anInteractiveContext->UpdateCurrentViewer();
256}
257
258Handle(TopTools_HSequenceOfShape) CImportExport::ReadIGES()// not by reference --> the sequence is created here !!
259{
260 CFileDialog dlg(TRUE,
261 NULL,
262 NULL,
263 OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
264 L"IGES Files (*.iges , *.igs)|*.iges; *.igs|All Files (*.*)|*.*||",
265 NULL );
266
267CString SHAREPATHValue;
268SHAREPATHValue.GetEnvironmentVariable (L"CSF_OCCTDataPath");
269CString initdir = (SHAREPATHValue + "\\iges");
270
271dlg.m_ofn.lpstrInitialDir = initdir;
272
273 Handle(TopTools_HSequenceOfShape) aSequence = new TopTools_HSequenceOfShape();
274 if (dlg.DoModal() == IDOK)
275 {
276 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_WAIT));
277 TCollection_AsciiString aFileName ((const wchar_t* )dlg.GetPathName());
278 Standard_Integer status = ReadIGES (aFileName.ToCString(), aSequence);
279 if (status != IFSelect_RetDone)
280 {
281 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, L"Error : The file is not read", L"CasCade Error", MB_ICONERROR);
282 }
283
284 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_ARROW));
285 }
286 return aSequence;
287}
288//----------------------------------------------------------------------
289
290Standard_Integer CImportExport::ReadIGES(const Standard_CString& aFileName,
291 Handle(TopTools_HSequenceOfShape)& aHSequenceOfShape)
292{
293
294 IGESControl_Reader Reader;
295
296 Standard_Integer status = Reader.ReadFile(aFileName);
297
298 if (status != IFSelect_RetDone) return status;
299 Reader.TransferRoots();
300 TopoDS_Shape aShape = Reader.OneShape();
301 aHSequenceOfShape->Append(aShape);
302
303 return status;
304}
305//----------------------------------------------------------------------
306
307void CImportExport::SaveIGES(const Handle(AIS_InteractiveContext)& anInteractiveContext)
308{
309 anInteractiveContext->InitSelected();
310 if (anInteractiveContext->NbSelected() == 0){
311 AfxMessageBox (L"No shape selected for export!");
312 return;
313 }
314 Handle(Quantity_HArray1OfColor) anArrayOfColors;
315 Handle(TColStd_HArray1OfReal) anArrayOfTransparencies;
316 CImportExport::SaveIGES(BuildSequenceFromContext(anInteractiveContext, anArrayOfColors, anArrayOfTransparencies));
317}
318
319Standard_Boolean CImportExport::SaveIGES(const Handle(TopTools_HSequenceOfShape)& aHSequenceOfShape)
320{
321 if (aHSequenceOfShape->Length() == 0)
322 {
323 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, L"No Shape in the HSequence!!", L"CasCade Warning", MB_ICONWARNING);
324 return Standard_False;
325 }
326
327 CFileDialog dlg(FALSE, L"*.iges",NULL,OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
328 L"IGES Files (*.iges )|*.iges;|IGES Files (*.igs )| *.igs;||", NULL);
329
330CString SHAREPATHValue;
331SHAREPATHValue.GetEnvironmentVariable (L"CSF_OCCTDataPath");
332CString initdir = (SHAREPATHValue + "\\iges");
333
334dlg.m_ofn.lpstrInitialDir = initdir;
335
336 Standard_Boolean result=Standard_False;
337 if (dlg.DoModal() == IDOK)
338 {
339 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_WAIT));
340
341 TCollection_AsciiString aFileName ((const wchar_t* )dlg.GetPathName());
342
343 result = SaveIGES (aFileName.ToCString(), aHSequenceOfShape);
344 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_ARROW));
345 }
346 return result;
347}
348//----------------------------------------------------------------------
349
350Standard_Boolean CImportExport::SaveIGES(const Standard_CString& aFileName,
351 const Handle(TopTools_HSequenceOfShape)& aHSequenceOfShape)
352{
353
354 IGESControl_Controller::Init();
355 IGESControl_Writer ICW (Interface_Static::CVal("XSTEP.iges.unit"),
356 Interface_Static::IVal("XSTEP.iges.writebrep.mode"));
357
358 for (Standard_Integer i=1;i<=aHSequenceOfShape->Length();i++)
359 ICW.AddShape (aHSequenceOfShape->Value(i));
360
361 ICW.ComputeModel();
362 Standard_Boolean result = ICW.Write(aFileName );
363 return result;
364}
365
366//======================================================================
367
368//======================================================================
369//= =
370//= STEP =
371//= =
372//======================================================================
373
374void CImportExport::ReadSTEP(const Handle(AIS_InteractiveContext)& anInteractiveContext)
375{
376 Handle(TopTools_HSequenceOfShape) aSequence = CImportExport::ReadSTEP();
377 if (!aSequence.IsNull()) {
378 for(int i=1;i<= aSequence->Length();i++)
379 anInteractiveContext->Display(new AIS_Shape(aSequence->Value(i)), Standard_False);
380 }
381}
382
383Handle(TopTools_HSequenceOfShape) CImportExport::ReadSTEP()// not by reference --> the sequence is created here !!
384{
385 CFileDialog dlg(TRUE,
386 NULL,
387 NULL,
388 OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
389 L"STEP Files (*.stp;*.step)|*.stp; *.step|All Files (*.*)|*.*||",
390 NULL );
391
392CString SHAREPATHValue;
393SHAREPATHValue.GetEnvironmentVariable (L"CSF_OCCTDataPath");
394CString initdir = (SHAREPATHValue + "\\step");
395
396dlg.m_ofn.lpstrInitialDir = initdir;
397
398 Handle(TopTools_HSequenceOfShape) aSequence= new TopTools_HSequenceOfShape();
399 if (dlg.DoModal() == IDOK)
400 {
401 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_WAIT));
402 TCollection_AsciiString aFileName ((const wchar_t* )dlg.GetPathName());
403 IFSelect_ReturnStatus ReturnStatus = ReadSTEP (aFileName.ToCString(), aSequence);
404 switch (ReturnStatus)
405 {
406 case IFSelect_RetError :
407 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, L"Not a valid Step file", L"ERROR", MB_ICONWARNING);
408 break;
409 case IFSelect_RetFail :
410 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, L"Reading has failed", L"ERROR", MB_ICONWARNING);
411 break;
412 case IFSelect_RetVoid :
413 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, L"Nothing to transfer", L"ERROR", MB_ICONWARNING);
414 break;
415 }
416 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_ARROW));
417 }
418 return aSequence;
419}
420
421IFSelect_ReturnStatus CImportExport::ReadSTEP(const Standard_CString& aFileName,
422 Handle(TopTools_HSequenceOfShape)& aHSequenceOfShape)
423{
424 aHSequenceOfShape->Clear();
425
426 // create additional log file
427 STEPControl_Reader aReader;
428 IFSelect_ReturnStatus status = aReader.ReadFile(aFileName);
429 if (status != IFSelect_RetDone)
430 return status;
431
432 aReader.WS()->TransferReader()->TransientProcess()->SetTraceLevel(2); // increase default trace level
433
434 Standard_Boolean failsonly = Standard_False;
435 aReader.PrintCheckLoad(failsonly, IFSelect_ItemsByEntity);
436
437 // Root transfers
438 Standard_Integer nbr = aReader.NbRootsForTransfer();
439 aReader.PrintCheckTransfer (failsonly, IFSelect_ItemsByEntity);
440 for ( Standard_Integer n = 1; n<=nbr; n++) {
441 /*Standard_Boolean ok =*/ aReader.TransferRoot(n);
442 }
443
444 // Collecting resulting entities
445 Standard_Integer nbs = aReader.NbShapes();
446 if (nbs == 0) {
447 return IFSelect_RetVoid;
448 }
449 for (Standard_Integer i=1; i<=nbs; i++) {
450 aHSequenceOfShape->Append(aReader.Shape(i));
451 }
452
453 return status;
454}
455
456
457//----------------------------------------------------------------------
458void CImportExport::SaveSTEP(const Handle(AIS_InteractiveContext)& anInteractiveContext)
459{
460 anInteractiveContext->InitSelected();
461 if (anInteractiveContext->NbSelected() == 0){
462 AfxMessageBox (L"No shape selected for export!");
463 return;
464 }
465 Handle(Quantity_HArray1OfColor) anArrayOfColors;
466 Handle(TColStd_HArray1OfReal) anArrayOfTransparencies;
467 CImportExport::SaveSTEP(BuildSequenceFromContext(anInteractiveContext, anArrayOfColors, anArrayOfTransparencies));
468}
469
470// Return True if no error
471Standard_Boolean TestFacetedBrep(const Handle(TopTools_HSequenceOfShape)& aHSequenceOfShape)
472{
473 Standard_Boolean OneErrorFound = Standard_False;
474 for (Standard_Integer i=1;i<=aHSequenceOfShape->Length();i++)
475 {
476 TopoDS_Shape aShape= aHSequenceOfShape->Value(i);
477
478 TopExp_Explorer Ex(aShape,TopAbs_FACE);
479 while (Ex.More() && !OneErrorFound)
480 {
481 // Get the Geom_Surface outside the TopoDS_Face
482 Handle(Geom_Surface) aSurface = BRep_Tool::Surface(TopoDS::Face(Ex.Current()));
483 // check if it is a plane.
484 if (!aSurface->IsKind(STANDARD_TYPE(Geom_Plane)))
485 OneErrorFound=Standard_True;
486 Ex.Next();
487 }
488 TopExp_Explorer Ex2(aShape,TopAbs_EDGE);
489 while (Ex2.More() && !OneErrorFound)
490 {
491 // Get the Geom_Curve outside the TopoDS_Face
492 Standard_Real FirstDummy,LastDummy;
493 Handle(Geom_Curve) aCurve = BRep_Tool::Curve(TopoDS::Edge(Ex2.Current()),FirstDummy,LastDummy);
494 // check if it is a line.
495 if (!aCurve->IsKind(STANDARD_TYPE(Geom_Line)))
496 OneErrorFound=Standard_True;
497 Ex2.Next();
498 }
499 }
500 return !OneErrorFound;
501}
502
503IFSelect_ReturnStatus CImportExport::SaveSTEP(const Handle(TopTools_HSequenceOfShape)& aHSequenceOfShape)
504{
505 if (aHSequenceOfShape->Length() == 0)
506 {
507 MessageBox (AfxGetApp()->m_pMainWnd->m_hWnd, L"No Shape in the HSequence!!", L"CasCade Warning", MB_ICONWARNING);
508 return IFSelect_RetError;
509 }
510
511 IFSelect_ReturnStatus status = IFSelect_RetVoid;
512
513 CFileSaveSTEPDialog aDlg(NULL);
514
515 aDlg.m_Cc1ModelType = STEPControl_AsIs;
516
517 if (aDlg.DoModal() == IDOK) {
518 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_WAIT));
519 TCollection_AsciiString aFileName ((const wchar_t* )aDlg.GetPathName());
520
521 STEPControl_StepModelType selection = aDlg.m_Cc1ModelType;
522
523 if(selection == STEPControl_FacetedBrep)
524
525 if (!TestFacetedBrep(aHSequenceOfShape))
526 {
527 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, L"At least one shape doesn't contain facetes", L"CasCade Warning", MB_ICONWARNING);
528 return IFSelect_RetError;
529 }
530
531
532 status = SaveSTEP (aFileName.ToCString(), aHSequenceOfShape, selection);
533 switch (status)
534 {
535 case IFSelect_RetError:
536 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, L"Incorrect Data", L"ERROR", MB_ICONWARNING);
537 break;
538 case IFSelect_RetFail:
539 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, L"Writing has failed", L"ERROR", MB_ICONWARNING);
540 break;
541 case IFSelect_RetVoid:
542 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, L"Nothing to transfer", L"ERROR", MB_ICONWARNING);
543 break;
544 }
545 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_ARROW));
546 }
547 return status;
548}
549//----------------------------------------------------------------------------------------
550IFSelect_ReturnStatus CImportExport::SaveSTEP(const Standard_CString& aFileName,
551 const Handle(TopTools_HSequenceOfShape)& aHSequenceOfShape,
552
553const STEPControl_StepModelType aValue /* =TopoDSToCc1Act_ManifoldSolidBrep */ )
554
555{
556 // CREATE THE WRITER
557
558 STEPControl_Writer aWriter;
559
560 IFSelect_ReturnStatus status;
561 for (Standard_Integer i=1;i<=aHSequenceOfShape->Length();i++)
562 {
563 status = aWriter.Transfer(aHSequenceOfShape->Value(i), aValue);
564 if ( status != IFSelect_RetDone ) return status;
565 }
566 status = aWriter.Write(aFileName);
567 return status;
568}
569
570
571
572//======================================================================
573//= =
574//= STL =
575//= =
576//======================================================================
577
578void CImportExport::SaveSTL(const Handle(AIS_InteractiveContext)& anInteractiveContext)
579{
580 anInteractiveContext->InitSelected();
581 if (anInteractiveContext->NbSelected() == 0){
582 AfxMessageBox (L"No shape selected for export!");
583 return;
584 }
585 Handle(Quantity_HArray1OfColor) anArrayOfColors;
586 Handle(TColStd_HArray1OfReal) anArrayOfTransparencies;
587 CImportExport::SaveSTL(BuildSequenceFromContext(anInteractiveContext, anArrayOfColors, anArrayOfTransparencies));
588}
589
590Standard_Boolean CImportExport::SaveSTL(const Handle(TopTools_HSequenceOfShape)& aHSequenceOfShape)
591{
592 CFileDialog dlg(FALSE, L"*.stl", NULL, OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
593 L"stl Files (*.stl)|*.stl;|STL Files (*.STL)|*.STL;||", NULL);
594
595CString SHAREPATHValue;
596SHAREPATHValue.GetEnvironmentVariable (L"CSF_OCCTDataPath");
597CString initdir = (SHAREPATHValue + "\\stl");
598
599dlg.m_ofn.lpstrInitialDir = initdir;
600
601 Standard_Boolean result = Standard_False;
602
603 if (dlg.DoModal() == IDOK) {
604 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_WAIT));
605 TCollection_AsciiString aFileName ((const wchar_t* )dlg.GetPathName());
606 TCollection_AsciiString Message;
607 result = SaveSTL (aFileName.ToCString(), aHSequenceOfShape, Message);
608 CString aMsg (TCollection_ExtendedString (Message).ToWideString());
609 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, aMsg, result ? L"CasCade" : L"CasCade Error", result ? MB_OK : MB_ICONERROR);
610 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_ARROW));
611 }
612 return result;
613}
614
615Standard_Boolean CImportExport::SaveSTL(const Standard_CString& aFileName,
616 const Handle(TopTools_HSequenceOfShape)& aHSequenceOfShape,
617 TCollection_AsciiString& ReturnMessage)
618{
619 Standard_Boolean ReturnValue = Standard_True;
620 if (aHSequenceOfShape->Length() == 0)
621 {
622 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, L"No Shape in the HSequence!!", L"CasCade Warning", MB_ICONWARNING);
623 return Standard_False;
624 }
625
626 ReturnMessage += "The Object have be saved in the file ";
627 ReturnMessage += aFileName;
628 ReturnMessage += "\n with the names : ";
629
630 TopoDS_Compound RES;
631 BRep_Builder MKCP;
632 MKCP.MakeCompound(RES);
633
634 for (Standard_Integer i=1;i<=aHSequenceOfShape->Length();i++)
635 {
636 TopoDS_Shape aShape= aHSequenceOfShape->Value(i);
637 TCollection_AsciiString anObjectName("anObjectName_");
638 anObjectName += i;
639 ReturnMessage += anObjectName;
640 ReturnMessage += " \n";
641
642 if ( aShape.IsNull() )
643 {
644 ReturnMessage += " Error : Invalid shape \n";
645 ReturnValue = Standard_False;
646 continue;
647 }
648
649 MKCP.Add(RES, aShape);
650 }
651
652 StlAPI_Writer myStlWriter;
653 myStlWriter.Write(RES, aFileName);
654
655 return ReturnValue;
656}
657
658
659//======================================================================
660//= =
661//= VRML =
662//= =
663//======================================================================
664
665void CImportExport::SaveVRML(const Handle(AIS_InteractiveContext)& anInteractiveContext)
666{
667 anInteractiveContext->InitSelected();
668 if (anInteractiveContext->NbSelected() == 0){
669 AfxMessageBox (L"No shape selected for export!");
670 return;
671 }
672 Handle(Quantity_HArray1OfColor) anArrayOfColors;
673 Handle(TColStd_HArray1OfReal) anArrayOfTransparencies;
674 CImportExport::SaveVRML(BuildSequenceFromContext(anInteractiveContext, anArrayOfColors, anArrayOfTransparencies),
675 anArrayOfColors, anArrayOfTransparencies);
676}
677
678Standard_Boolean CImportExport::SaveVRML(const Handle(TopTools_HSequenceOfShape)& aHSequenceOfShape,
679 const Handle(Quantity_HArray1OfColor)& anArrayOfColors,
680 const Handle(TColStd_HArray1OfReal)& anArrayOfTransparencies)
681{
682 CFileDialog dlg(FALSE, L"*.vrml", NULL, OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
683 L"vrml Files (*.vrml)|*.vrml;|vrm Files (*.vrm)|*.vrm;||", NULL);
684
685CString SHAREPATHValue;
686SHAREPATHValue.GetEnvironmentVariable (L"CSF_OCCTDataPath");
687CString initdir = (SHAREPATHValue + "\\vrml");
688
689dlg.m_ofn.lpstrInitialDir = initdir;
690
691 Standard_Boolean result = Standard_False;
692
693 if (dlg.DoModal() == IDOK) {
694 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_WAIT));
695 TCollection_AsciiString aFileName ((const wchar_t* )dlg.GetPathName());
696 TCollection_AsciiString Message;
697 result = SaveVRML (aFileName.ToCString(), aHSequenceOfShape, anArrayOfColors, anArrayOfTransparencies, Message);
698 CString aMsg (TCollection_ExtendedString(Message).ToWideString());
699 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, aMsg, result ? L"CasCade" : L"CasCade Error", result ? MB_OK : MB_ICONERROR);
700 SetCursor(AfxGetApp()->LoadStandardCursor(IDC_ARROW));
701 }
702 return result;
703}
704
705Standard_Boolean CImportExport::SaveVRML(const Standard_CString& aFileName,
706 const Handle(TopTools_HSequenceOfShape)& aHSequenceOfShape,
707 const Handle(Quantity_HArray1OfColor)& anArrayOfColors,
708 const Handle(TColStd_HArray1OfReal)& anArrayOfTransparencies,
709 TCollection_AsciiString& ReturnMessage)
710{
711 Standard_Boolean ReturnValue = Standard_True;
712 if (aHSequenceOfShape->Length() == 0)
713 {
714 MessageBoxW (AfxGetApp()->m_pMainWnd->m_hWnd, L"No Shape in the HSequence!!", L"CasCade Warning", MB_ICONWARNING);
715 return Standard_False;
716 }
717
718 ReturnMessage += "The Object has been saved in the file ";
719 ReturnMessage += aFileName;
720 ReturnMessage += "\n with the names : ";
721
722 // VRML scene.
723 VrmlData_Scene scene;
724 VrmlData_ShapeConvert converter(scene/*, 0.001*/); // from mm to meters
725 Standard_Integer iShape = 1; // Counter of shapes
726
727 for (int i = 1; i <= aHSequenceOfShape->Length(); i++)
728 {
729 // Shape
730 TopoDS_Shape shape = aHSequenceOfShape->Value(i);
731 if (shape.IsNull())
732 {
733 ReturnMessage += " Error : Invalid shape \n";
734 ReturnValue = Standard_False;
735 continue;
736 }
737
738 // Color
739 Quantity_Color color; // yellow
740 if (!anArrayOfColors.IsNull())
741 color = anArrayOfColors->Value(i);
742
743 // Transparency
744 Standard_Real transparency = 0.0;
745 if (!anArrayOfTransparencies.IsNull())
746 transparency = anArrayOfTransparencies->Value(i);
747
748 // Give a name to the shape.
749 TCollection_AsciiString name("Shape");
750 name += TCollection_AsciiString(iShape++);
751 converter.AddShape(shape, name.ToCString());
752 ReturnMessage += name;
753 ReturnMessage += '\n';
754
755 // Check presence of faces in the shape.
756 TopExp_Explorer expl(shape, TopAbs_FACE);
757 if (expl.More())
758 converter.Convert(true, false, 0.01); // faces only
759 else
760 converter.Convert(false, true, 0.01); // edges only
761
762 // Name of the color & transparency.
763 // It will be uniquely saved in VRML file.
764 TCollection_AsciiString cname = Quantity_Color::StringName(color.Name());
765 cname += transparency;
766
767 // Make the appearance (VRML attribute)
768 Handle(VrmlData_Appearance) appearance = Handle(VrmlData_Appearance)::DownCast(scene.FindNode(cname.ToCString()));
769 if (appearance.IsNull())
770 {
771 // Not found ... create a new one.
772 Handle(VrmlData_Material) material = new VrmlData_Material(scene, cname.ToCString(), 0.2, 0.2, transparency);
773 material->SetDiffuseColor(color);
774 material->SetEmissiveColor(color);
775 material->SetSpecularColor(color);
776 scene.AddNode(material, false);
777 appearance = new VrmlData_Appearance(scene, cname.ToCString());
778 appearance->SetMaterial(material);
779 scene.AddNode(appearance, false);
780 }
781
782 // Apply the material to the shape of entity.
783 Handle(VrmlData_Group) group = Handle(VrmlData_Group)::DownCast(scene.FindNode(name.ToCString()));
784 if (!group.IsNull())
785 {
786 VrmlData_ListOfNode::Iterator itr = group->NodeIterator();
787 for (; itr.More(); itr.Next())
788 {
789 Handle(VrmlData_Node) node = itr.Value();
790 if (node->DynamicType() == STANDARD_TYPE(VrmlData_ShapeNode))
791 {
792 Handle(VrmlData_ShapeNode) aShape = Handle(VrmlData_ShapeNode)::DownCast(node);
793 aShape->SetAppearance(appearance);
794 }
795 else if (itr.Value()->DynamicType() == STANDARD_TYPE(VrmlData_Group))
796 {
797 Handle(VrmlData_Group) groupc = Handle(VrmlData_Group)::DownCast(itr.Value());
798 VrmlData_ListOfNode::Iterator itrc = groupc->NodeIterator();
799 for (; itrc.More(); itrc.Next())
800 {
801 Handle(VrmlData_Node) nodec = itrc.Value();
802 if (nodec->DynamicType() == STANDARD_TYPE(VrmlData_ShapeNode))
803 {
804 Handle(VrmlData_ShapeNode) shapec = Handle(VrmlData_ShapeNode)::DownCast(nodec);
805 shapec->SetAppearance(appearance);
806 }
807 } // for of group nodes...
808 } // if (it is a shape node...
809 } // for of group nodes...
810 } // if (!group.IsNull...
811 } // iterator of shapes
812
813 // Call VRML writer
814 std::ofstream writer(aFileName);
815 writer<<scene;
816 writer.close();
817
818 /* Old approach to store shapes in VRML (without color & transparency).
819 TopoDS_Compound RES;
820 BRep_Builder MKCP;
821 MKCP.MakeCompound(RES);
822
823 for (Standard_Integer i=1;i<=aHSequenceOfShape->Length();i++)
824 {
825 TopoDS_Shape aShape= aHSequenceOfShape->Value(i);
826 TCollection_AsciiString anObjectName("anObjectName_");
827 anObjectName += i;
828 ReturnMessage += anObjectName;
829 ReturnMessage += " \n";
830
831 if ( aShape.IsNull() )
832 {
833 ReturnMessage += " Error : Invalid shape \n";
834 ReturnValue = Standard_False;
835 continue;
836 }
837
838 MKCP.Add(RES, aShape);
839 }
840
841 VrmlAPI_Writer myVrmlWriter;
842 myVrmlWriter.Write(RES, aFileName);
843 */
844
845 return ReturnValue;
846}
847
848
849
850
Open CASCADE Technology repository