1 // Created on: 2000-05-30
2 // Created by: Sergey MOZOKHIN
3 // Copyright (c) 2000-2014 OPEN CASCADE SAS
5 // This file is part of Open CASCADE Technology software library.
7 // This library is free software; you can redistribute it and/or modify it under
8 // the terms of the GNU Lesser General Public License version 2.1 as published
9 // by the Free Software Foundation, with special exception defined in the file
10 // OCCT_LGPL_EXCEPTION.txt. Consult the file LICENSE_LGPL_21.txt included in OCCT
11 // distribution for complete text of the license and disclaimer of any warranty.
13 // Alternatively, this file may be used under the terms of Open CASCADE
14 // commercial license or contractual agreement.
17 #include <AIS_InteractiveContext.hxx>
18 #include <Aspect_TypeOfMarker.hxx>
19 #include <Bnd_Box.hxx>
22 #include <Draw_Interpretor.hxx>
23 #include <Draw_PluginMacro.hxx>
24 #include <Draw_ProgressIndicator.hxx>
25 #include <Graphic3d_MaterialAspect.hxx>
26 #include <MeshVS_DataMapOfIntegerAsciiString.hxx>
27 #include <MeshVS_DeformedDataSource.hxx>
28 #include <MeshVS_Drawer.hxx>
29 #include <MeshVS_DrawerAttribute.hxx>
30 #include <MeshVS_ElementalColorPrsBuilder.hxx>
31 #include <MeshVS_Mesh.hxx>
32 #include <MeshVS_MeshEntityOwner.hxx>
33 #include <MeshVS_MeshPrsBuilder.hxx>
34 #include <MeshVS_NodalColorPrsBuilder.hxx>
35 #include <MeshVS_PrsBuilder.hxx>
36 #include <MeshVS_TextPrsBuilder.hxx>
37 #include <MeshVS_VectorPrsBuilder.hxx>
38 #include <OSD_Path.hxx>
39 #include <Quantity_Color.hxx>
40 #include <Quantity_HArray1OfColor.hxx>
41 #include <Quantity_NameOfColor.hxx>
43 #include <SelectMgr_SelectionManager.hxx>
44 #include <Standard_ErrorHandler.hxx>
45 #include <StdSelect_ViewerSelector3d.hxx>
47 #include <StlAPI_Writer.hxx>
48 #include <StlMesh_Mesh.hxx>
49 #include <StlMesh_SequenceOfMeshTriangle.hxx>
50 #include <TColgp_SequenceOfXYZ.hxx>
51 #include <TCollection_AsciiString.hxx>
52 #include <TColStd_Array1OfReal.hxx>
53 #include <TColStd_HPackedMapOfInteger.hxx>
54 #include <TColStd_MapIteratorOfPackedMapOfInteger.hxx>
55 #include <TopoDS_Shape.hxx>
56 #include <V3d_View.hxx>
57 #include <ViewerTest.hxx>
58 #include <VrmlAPI.hxx>
59 #include <VrmlAPI_Writer.hxx>
60 #include <VrmlData_DataMapOfShapeAppearance.hxx>
61 #include <VrmlData_Scene.hxx>
62 #include <VrmlData_ShapeConvert.hxx>
64 #include <XSDRAWIGES.hxx>
65 #include <XSDRAWSTEP.hxx>
66 #include <XSDRAWSTLVRML.hxx>
67 #include <XSDRAWSTLVRML_DataSource.hxx>
68 #include <XSDRAWSTLVRML_DataSource3D.hxx>
69 #include <XSDRAWSTLVRML_DrawableMesh.hxx>
71 // avoid warnings on 'extern "C"' functions returning C++ classes
73 #pragma warning(4:4190)
80 extern Standard_Boolean VDisplayAISObject (const TCollection_AsciiString& theName,
81 const Handle(AIS_InteractiveObject)& theAISObj,
82 Standard_Boolean theReplaceIfExists = Standard_True);
84 static Standard_Integer writestl
85 (Draw_Interpretor& di, Standard_Integer argc, const char** argv)
87 if (argc < 3 || argc > 4) {
88 di << "Use: " << argv[0]
89 << " shape file [ascii/binary (0/1) : 1 by default]\n";
91 TopoDS_Shape aShape = DBRep::Get(argv[1]);
92 Standard_Boolean isASCIIMode = Standard_False;
94 isASCIIMode = (Draw::Atoi(argv[3]) == 0);
96 StlAPI_Writer aWriter;
97 aWriter.ASCIIMode() = isASCIIMode;
98 StlAPI_ErrorStatus aStatus = aWriter.Write (aShape, argv[2]);
102 case StlAPI_MeshIsEmpty: di << "** Error **: Mesh is empty. Please, compute triangulation before."; break;
103 case StlAPI_CannotOpenFile: di << "** Error **: Cannot create/open a file with the passed name."; break;
104 case StlAPI_StatusOK: default: break;
110 static Standard_Integer readstl
111 (Draw_Interpretor& di, Standard_Integer argc, const char** argv)
113 if (argc<3) di << "wrong number of parameters" << "\n";
115 TopoDS_Shape aShape ;
116 StlAPI::Read(aShape,argv[2]);
117 DBRep::Set(argv[1],aShape);
122 static Standard_Integer writevrml
123 (Draw_Interpretor& di, Standard_Integer argc, const char** argv)
125 if (argc < 3 || argc > 5)
127 di << "wrong number of parameters\n";
131 TopoDS_Shape aShape = DBRep::Get(argv[1]);
133 // Get the optional parameters
134 Standard_Integer aVersion = 2;
135 Standard_Integer aType = 1;
138 aVersion = Draw::Atoi(argv[3]);
140 aType = Draw::Atoi(argv[4]);
144 aVersion = Max(1, aVersion);
145 aVersion = Min(2, aVersion);
146 aType = Max(0, aType);
147 aType = Min(2, aType);
149 VrmlAPI_Writer writer;
153 case 0: writer.SetRepresentation(VrmlAPI_ShadedRepresentation); break;
154 case 1: writer.SetRepresentation(VrmlAPI_WireFrameRepresentation); break;
155 case 2: writer.SetRepresentation(VrmlAPI_BothRepresentation); break;
158 writer.Write(aShape, argv[2], aVersion);
163 //=======================================================================
164 //function : loadvrml
166 //=======================================================================
168 static Standard_Integer loadvrml
169 (Draw_Interpretor& di, Standard_Integer argc, const char** argv)
171 if (argc<3) di << "wrong number of parameters" << "\n";
173 TopoDS_Shape aShape ;
174 VrmlData_DataMapOfShapeAppearance aShapeAppMap;
176 //-----------------------------------------------------------
178 istream aStream (&aFic);
180 if (aFic.open(argv[2], ios::in)) {
182 // Get path of the VRML file.
183 OSD_Path aPath(argv[2]);
184 TCollection_AsciiString aVrmlDir(".");
185 TCollection_AsciiString aDisk = aPath.Disk();
186 TCollection_AsciiString aTrek = aPath.Trek();
187 if (!aTrek.IsEmpty())
189 if (!aDisk.IsEmpty())
193 aTrek.ChangeAll('|', '/');
197 VrmlData_Scene aScene;
199 aScene.SetVrmlDir (aVrmlDir);
201 const char * aStr = 0L;
202 switch (aScene.Status()) {
204 case VrmlData_StatusOK:
206 aShape = aScene.GetShape(aShapeAppMap);
209 case VrmlData_EmptyData: aStr = "EmptyData"; break;
210 case VrmlData_UnrecoverableError: aStr = "UnrecoverableError"; break;
211 case VrmlData_GeneralError: aStr = "GeneralError"; break;
212 case VrmlData_EndOfFile: aStr = "EndOfFile"; break;
213 case VrmlData_NotVrmlFile: aStr = "NotVrmlFile"; break;
214 case VrmlData_CannotOpenFile: aStr = "CannotOpenFile"; break;
215 case VrmlData_VrmlFormatError: aStr = "VrmlFormatError"; break;
216 case VrmlData_NumericInputError: aStr = "NumericInputError"; break;
217 case VrmlData_IrrelevantNumber: aStr = "IrrelevantNumber"; break;
218 case VrmlData_BooleanInputError: aStr = "BooleanInputError"; break;
219 case VrmlData_StringInputError: aStr = "StringInputError"; break;
220 case VrmlData_NodeNameUnknown: aStr = "NodeNameUnknown"; break;
221 case VrmlData_NonPositiveSize: aStr = "NonPositiveSize"; break;
222 case VrmlData_ReadUnknownNode: aStr = "ReadUnknownNode"; break;
223 case VrmlData_NonSupportedFeature: aStr = "NonSupportedFeature"; break;
224 case VrmlData_OutputStreamUndefined:aStr = "OutputStreamUndefined"; break;
225 case VrmlData_NotImplemented: aStr = "NotImplemented"; break;
230 di << " ++ VRML Error: " << aStr << " in line "
231 << aScene.GetLineError() << "\n";
234 DBRep::Set(argv[1],aShape);
238 di << "cannot open file\n";
242 //-----------------------------------------------------------
247 //-----------------------------------------------------------------------------
248 static Standard_Integer createmesh
249 (Draw_Interpretor& di, Standard_Integer argc, const char** argv )
253 di << "Wrong number of parameters\n";
254 di << "Use: " << argv[0] << " <mesh name> <stl file>\n";
258 Handle(AIS_InteractiveContext) aContext = ViewerTest::GetAISContext();
259 if (aContext.IsNull())
261 di << "No active view. Please call 'vinit' first\n";
265 // Progress indicator
266 OSD_Path aFile( argv[2] );
267 Handle(Draw_ProgressIndicator) aProgress = new Draw_ProgressIndicator (di, 1);
268 Handle(StlMesh_Mesh) aSTLMesh = RWStl::ReadFile (aFile, aProgress);
270 di << "Reading OK...\n";
271 Handle( XSDRAWSTLVRML_DataSource ) aDS = new XSDRAWSTLVRML_DataSource( aSTLMesh );
272 di << "Data source is created successful\n";
273 Handle( MeshVS_Mesh ) aMesh = new MeshVS_Mesh();
274 di << "MeshVS_Mesh is created successful\n";
276 aMesh->SetDataSource( aDS );
277 aMesh->AddBuilder( new MeshVS_MeshPrsBuilder( aMesh.operator->() ), Standard_True );
279 aMesh->GetDrawer()->SetColor( MeshVS_DA_EdgeColor, Quantity_NOC_YELLOW );
281 // Hide all nodes by default
282 Handle(TColStd_HPackedMapOfInteger) aNodes = new TColStd_HPackedMapOfInteger();
283 Standard_Integer aLen = aSTLMesh->Vertices().Length();
284 for ( Standard_Integer anIndex = 1; anIndex <= aLen; anIndex++ )
285 aNodes->ChangeMap().Add( anIndex );
286 aMesh->SetHiddenNodes( aNodes );
287 aMesh->SetSelectableNodes ( aNodes );
289 VDisplayAISObject(argv[1], aMesh);
290 aContext->Deactivate( aMesh );
292 Draw::Set( argv[1], new XSDRAWSTLVRML_DrawableMesh( aMesh ) );
293 Handle( V3d_View ) aView = ViewerTest::CurrentView();
294 if ( !aView.IsNull() )
299 //-----------------------------------------------------------------------------
301 static Standard_Integer create3d
302 (Draw_Interpretor& di, Standard_Integer argc, const char** argv )
306 di << "Wrong number of parameters\n";
307 di << "Use: " << argv[0] << " <mesh name>\n";
311 Handle(AIS_InteractiveContext) aContext = ViewerTest::GetAISContext();
312 if (aContext.IsNull())
314 di << "No active view. Please call 'vinit' first\n";
318 Handle( XSDRAWSTLVRML_DataSource3D ) aDS = new XSDRAWSTLVRML_DataSource3D();
319 di << "Data source is created successful\n";
320 Handle( MeshVS_Mesh ) aMesh = new MeshVS_Mesh();
321 di << "MeshVS_Mesh is created successful\n";
323 aMesh->SetDataSource( aDS );
324 aMesh->AddBuilder( new MeshVS_MeshPrsBuilder( aMesh.operator->() ), Standard_True );
326 aMesh->GetDrawer()->SetColor( MeshVS_DA_EdgeColor, Quantity_NOC_YELLOW );
328 // Hide all nodes by default
329 Handle(TColStd_HPackedMapOfInteger) aNodes = new TColStd_HPackedMapOfInteger();
330 Standard_Integer aLen = aDS->GetAllNodes().Extent();
331 for ( Standard_Integer anIndex = 1; anIndex <= aLen; anIndex++ )
332 aNodes->ChangeMap().Add( anIndex );
333 aMesh->SetHiddenNodes( aNodes );
334 aMesh->SetSelectableNodes ( aNodes );
336 VDisplayAISObject(argv[1], aMesh);
337 aContext->Deactivate( aMesh );
339 Draw::Set( argv[1], new XSDRAWSTLVRML_DrawableMesh( aMesh ) );
340 Handle( V3d_View ) aView = ViewerTest::CurrentView();
341 if ( !aView.IsNull() )
347 Handle( MeshVS_Mesh ) getMesh( const char* theName, Draw_Interpretor& di)
349 Handle( XSDRAWSTLVRML_DrawableMesh ) aDrawMesh =
350 Handle( XSDRAWSTLVRML_DrawableMesh )::DownCast( Draw::Get( theName ) );
352 if( aDrawMesh.IsNull() )
354 di << "There is no such object\n";
359 Handle( MeshVS_Mesh ) aMesh = aDrawMesh->GetMesh();
362 di << "There is invalid mesh\n";
370 //-----------------------------------------------------------------------------
371 static Standard_Integer setcolor
372 (Draw_Interpretor& di, Standard_Integer argc, const char** argv, Standard_Integer theParam )
375 di << "Wrong number of parameters\n";
378 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[1], di );
379 if( !aMesh.IsNull() )
381 Standard_Real aRed = Draw::Atof (argv[2]);
382 Standard_Real aGreen = Draw::Atof (argv[3]);
383 Standard_Real aBlue = Draw::Atof (argv[4]);
384 aMesh->GetDrawer()->SetColor( (MeshVS_DrawerAttribute)theParam,
385 Quantity_Color( aRed, aGreen, aBlue, Quantity_TOC_RGB ) );
387 Handle( AIS_InteractiveContext ) aContext = ViewerTest::GetAISContext();
389 if( aContext.IsNull() )
390 di << "The context is null\n";
392 aContext->Redisplay( aMesh );
397 //-----------------------------------------------------------------------------
398 static Standard_Integer meshcolor
399 (Draw_Interpretor& theInterp, Standard_Integer argc, const char** argv )
401 return setcolor( theInterp, argc, argv, MeshVS_DA_InteriorColor );
403 //-----------------------------------------------------------------------------
404 static Standard_Integer linecolor
405 (Draw_Interpretor& theInterp, Standard_Integer argc, const char** argv )
407 return setcolor( theInterp, argc, argv, MeshVS_DA_EdgeColor );
409 //-----------------------------------------------------------------------------
410 static Standard_Integer meshmat
411 (Draw_Interpretor& di, Standard_Integer argc, const char** argv )
414 di << "Wrong number of parameters\n";
417 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[1], di );
418 if( !aMesh.IsNull() )
420 Standard_Integer aMaterial = Draw::Atoi (argv[2]);
422 Graphic3d_MaterialAspect aMatAsp =
423 (Graphic3d_MaterialAspect)(Graphic3d_NameOfMaterial)aMaterial;
427 Standard_Real aTransparency = Draw::Atof(argv[3]);
428 aMatAsp.SetTransparency(aTransparency);
430 aMesh->GetDrawer()->SetMaterial( MeshVS_DA_FrontMaterial, aMatAsp );
431 aMesh->GetDrawer()->SetMaterial( MeshVS_DA_BackMaterial, aMatAsp );
433 Handle( AIS_InteractiveContext ) aContext = ViewerTest::GetAISContext();
435 if( aContext.IsNull() )
436 di << "The context is null\n";
438 aContext->Redisplay( aMesh );
443 //-----------------------------------------------------------------------------
444 static Standard_Integer shrink
445 (Draw_Interpretor& di, Standard_Integer argc, const char** argv )
448 di << "Wrong number of parameters\n";
451 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[1], di );
452 if( !aMesh.IsNull() )
454 Standard_Real aShrinkCoeff = Draw::Atof (argv[2]);
455 aMesh->GetDrawer()->SetDouble( MeshVS_DA_ShrinkCoeff, aShrinkCoeff );
457 Handle( AIS_InteractiveContext ) aContext = ViewerTest::GetAISContext();
459 if( aContext.IsNull() )
460 di << "The context is null\n";
462 aContext->Redisplay( aMesh );
468 //-----------------------------------------------------------------------------
469 static Standard_Integer closed (Draw_Interpretor& theDI, Standard_Integer theArgc, const char** theArgv)
473 theDI << "Wrong number of parameters.\n";
477 Handle(MeshVS_Mesh) aMesh = getMesh (theArgv[1], theDI);
480 Standard_Boolean aFlag = Draw::Atoi (theArgv[2]) != 0;
481 aMesh->GetDrawer()->SetBoolean (MeshVS_DA_SupressBackFaces, aFlag);
483 Handle( AIS_InteractiveContext ) aContext = ViewerTest::GetAISContext();
484 if (aContext.IsNull())
486 theDI << "The context is null\n";
490 aContext->Redisplay (aMesh);
497 //-----------------------------------------------------------------------------
499 static Standard_Integer mdisplay
500 (Draw_Interpretor& di, Standard_Integer argc, const char** argv )
503 di << "Wrong number of parameters\n";
506 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[1], di );
507 if( !aMesh.IsNull() )
509 Handle( AIS_InteractiveContext ) aContext = ViewerTest::GetAISContext();
511 if( aContext.IsNull() )
512 di << "The context is null\n";
515 Standard_DISABLE_DEPRECATION_WARNINGS
516 if (aContext->HasOpenedContext())
518 aContext->CloseLocalContext();
520 Standard_ENABLE_DEPRECATION_WARNINGS
522 aContext->Display( aMesh );
528 //-----------------------------------------------------------------------------
529 static Standard_Integer merase
530 (Draw_Interpretor& di, Standard_Integer argc, const char** argv )
533 di << "Wrong number of parameters\n";
536 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[1], di );
537 if( !aMesh.IsNull() )
539 Handle( AIS_InteractiveContext ) aContext = ViewerTest::GetAISContext();
541 if( aContext.IsNull() )
542 di << "The context is null\n";
545 Standard_DISABLE_DEPRECATION_WARNINGS
546 if (aContext->HasOpenedContext())
548 aContext->CloseLocalContext();
550 Standard_ENABLE_DEPRECATION_WARNINGS
552 aContext->Erase( aMesh );
556 di << "Mesh is null\n";
560 //-----------------------------------------------------------------------------
561 static Standard_Integer hidesel
562 (Draw_Interpretor& di, Standard_Integer argc, const char** argv )
566 di << "Wrong number of parameters\n";
567 di << "Use: " << argv[0] << " <mesh name>\n";
571 Handle( AIS_InteractiveContext ) aContext = ViewerTest::GetAISContext();
572 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[1], di );
575 di << "The mesh is invalid\n";
579 if( aContext.IsNull() )
580 di << "The context is null\n";
583 Handle(TColStd_HPackedMapOfInteger) aHiddenNodes = aMesh->GetHiddenNodes();
584 if (aHiddenNodes.IsNull())
586 aHiddenNodes = new TColStd_HPackedMapOfInteger();
588 Handle(TColStd_HPackedMapOfInteger) aHiddenElements = aMesh->GetHiddenElems();
589 if (aHiddenElements.IsNull())
591 aHiddenElements = new TColStd_HPackedMapOfInteger();
593 for( aContext->InitSelected(); aContext->MoreSelected(); aContext->NextSelected() )
595 Handle( MeshVS_MeshEntityOwner ) anOwner =
596 Handle( MeshVS_MeshEntityOwner )::DownCast( aContext->SelectedOwner() );
597 if( !anOwner.IsNull() )
599 if( anOwner->Type()==MeshVS_ET_Node )
601 aHiddenNodes->ChangeMap().Add( anOwner->ID() );
605 aHiddenElements->ChangeMap().Add( anOwner->ID() );
609 aContext->ClearSelected();
610 aMesh->SetHiddenNodes( aHiddenNodes );
611 aMesh->SetHiddenElems( aHiddenElements );
612 aContext->Redisplay( aMesh );
617 //-----------------------------------------------------------------------------
618 static Standard_Integer showonly
619 (Draw_Interpretor& di, Standard_Integer argc, const char** argv )
623 di << "Wrong number of parameters\n";
624 di << "Use: " << argv[0] << " <mesh name>\n";
629 Handle( AIS_InteractiveContext ) aContext = ViewerTest::GetAISContext();
630 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[1], di );
633 di << "The mesh is invalid\n";
637 if( aContext.IsNull() )
638 di << "The context is null\n";
641 Handle(TColStd_HPackedMapOfInteger) aHiddenNodes =
642 new TColStd_HPackedMapOfInteger(aMesh->GetDataSource()->GetAllNodes());
643 Handle(TColStd_HPackedMapOfInteger) aHiddenElements =
644 new TColStd_HPackedMapOfInteger(aMesh->GetDataSource()->GetAllElements());
645 for( aContext->InitSelected(); aContext->MoreSelected(); aContext->NextSelected() )
647 Handle( MeshVS_MeshEntityOwner ) anOwner =
648 Handle( MeshVS_MeshEntityOwner )::DownCast( aContext->SelectedOwner() );
649 if( !anOwner.IsNull() )
651 if( anOwner->Type() == MeshVS_ET_Node )
653 aHiddenNodes->ChangeMap().Remove( anOwner->ID() );
657 aHiddenElements->ChangeMap().Remove( anOwner->ID() );
661 aMesh->SetHiddenNodes( aHiddenNodes );
662 aMesh->SetHiddenElems( aHiddenElements );
663 aContext->Redisplay( aMesh );
668 //-----------------------------------------------------------------------------
669 static Standard_Integer showall
670 (Draw_Interpretor& di, Standard_Integer argc, const char** argv )
674 di << "Wrong number of parameters\n";
675 di << "Use: " << argv[0] << " <mesh name>\n";
679 Handle( AIS_InteractiveContext ) aContext = ViewerTest::GetAISContext();
680 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[1], di );
683 di << "The mesh is invalid\n";
687 if( aContext.IsNull() )
688 di << "The context is null\n";
691 aMesh->SetHiddenNodes( new TColStd_HPackedMapOfInteger() );
692 aMesh->SetHiddenElems( new TColStd_HPackedMapOfInteger() );
693 aContext->Redisplay( aMesh );
699 //-----------------------------------------------------------------------------
700 static Standard_Integer meshcolors( Draw_Interpretor& di,
701 Standard_Integer argc,
709 di << "Wrong number of parameters\n";
710 di << "Use : meshcolors <mesh name> <mode> <isreflect>\n";
711 di << "mode : {elem1|elem2|nodal|nodaltex|none}\n";
712 di << " elem1 - different color for each element\n";
713 di << " elem2 - one color for one side\n";
714 di << " nodal - different color for each node\n";
715 di << " nodaltex - different color for each node with texture interpolation\n";
716 di << " none - clear\n";
717 di << "isreflect : {0|1} \n";
722 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[ 1 ], di );
724 if ( aMesh.IsNull() )
726 di << "Mesh not found\n";
729 Handle(AIS_InteractiveContext) anIC = ViewerTest::GetAISContext();
732 di << "The context is null\n";
735 if( !aMesh.IsNull() )
737 TCollection_AsciiString aMode = TCollection_AsciiString (argv[2]);
738 Quantity_Color aColor1( (Quantity_NameOfColor)( Quantity_NOC_BLUE1 ) );
739 Quantity_Color aColor2( (Quantity_NameOfColor)( Quantity_NOC_RED1 ) );
740 if( aMode.IsEqual("elem1") || aMode.IsEqual("elem2") || aMode.IsEqual("nodal") || aMode.IsEqual("nodaltex") || aMode.IsEqual("none") )
742 Handle(MeshVS_PrsBuilder) aTempBuilder;
743 Standard_Integer aReflection = Draw::Atoi(argv[3]);
745 for (Standard_Integer aCount = 0 ; aCount < aMesh->GetBuildersCount(); aCount++ ){
746 aTempBuilder = aMesh->FindBuilder("MeshVS_ElementalColorPrsBuilder");
747 if( !aTempBuilder.IsNull())
748 aMesh->RemoveBuilderById(aTempBuilder->GetId());
750 aTempBuilder = aMesh->FindBuilder("MeshVS_NodalColorPrsBuilder");
751 if( !aTempBuilder.IsNull())
752 aMesh->RemoveBuilderById(aTempBuilder->GetId());
755 if( aMode.IsEqual("elem1") || aMode.IsEqual("elem2") )
757 Handle(MeshVS_ElementalColorPrsBuilder) aBuilder = new MeshVS_ElementalColorPrsBuilder(
758 aMesh, MeshVS_DMF_ElementalColorDataPrs | MeshVS_DMF_OCCMask );
760 const TColStd_PackedMapOfInteger& anAllElements = aMesh->GetDataSource()->GetAllElements();
761 TColStd_MapIteratorOfPackedMapOfInteger anIter( anAllElements );
763 if( aMode.IsEqual("elem1") )
764 for ( ; anIter.More(); anIter.Next() )
766 Quantity_Color aColor( (Quantity_NameOfColor)( anIter.Key() % Quantity_NOC_WHITE ) );
767 aBuilder->SetColor1( anIter.Key(), aColor );
770 for ( ; anIter.More(); anIter.Next() )
771 aBuilder->SetColor2( anIter.Key(), aColor1, aColor2 );
773 aMesh->AddBuilder( aBuilder, Standard_True );
777 if( aMode.IsEqual("nodal") )
779 Handle(MeshVS_NodalColorPrsBuilder) aBuilder = new MeshVS_NodalColorPrsBuilder(
780 aMesh, MeshVS_DMF_NodalColorDataPrs | MeshVS_DMF_OCCMask );
781 aMesh->AddBuilder( aBuilder, Standard_True );
784 const TColStd_PackedMapOfInteger& anAllNodes =
785 aMesh->GetDataSource()->GetAllNodes();
786 TColStd_MapIteratorOfPackedMapOfInteger anIter( anAllNodes );
787 for ( ; anIter.More(); anIter.Next() )
789 Quantity_Color aColor( (Quantity_NameOfColor)(
790 anIter.Key() % Quantity_NOC_WHITE ) );
791 aBuilder->SetColor( anIter.Key(), aColor );
793 aMesh->AddBuilder( aBuilder, Standard_True );
796 if(aMode.IsEqual("nodaltex"))
798 // assign nodal builder to the mesh
799 Handle(MeshVS_NodalColorPrsBuilder) aBuilder = new MeshVS_NodalColorPrsBuilder(
800 aMesh, MeshVS_DMF_NodalColorDataPrs | MeshVS_DMF_OCCMask);
801 aMesh->AddBuilder(aBuilder, Standard_True);
802 aBuilder->UseTexture(Standard_True);
804 // prepare color map for texture
805 Aspect_SequenceOfColor aColorMap;
806 aColorMap.Append((Quantity_NameOfColor) Quantity_NOC_RED);
807 aColorMap.Append((Quantity_NameOfColor) Quantity_NOC_YELLOW);
808 aColorMap.Append((Quantity_NameOfColor) Quantity_NOC_BLUE1);
810 // prepare scale map for mesh - it will be assigned to mesh as texture coordinates
811 // make mesh color interpolated from minimum X coord to maximum X coord
812 Handle(MeshVS_DataSource) aDataSource = aMesh->GetDataSource();
813 Standard_Real aMinX, aMinY, aMinZ, aMaxX, aMaxY, aMaxZ;
815 // get bounding box for calculations
816 aDataSource->GetBoundingBox().Get(aMinX, aMinY, aMinZ, aMaxX, aMaxY, aMaxZ);
817 Standard_Real aDelta = aMaxX - aMinX;
819 // assign color scale map values (0..1) to nodes
820 TColStd_DataMapOfIntegerReal aScaleMap;
821 TColStd_Array1OfReal aCoords(1, 3);
822 Standard_Integer aNbNodes;
823 MeshVS_EntityType aType;
826 const TColStd_PackedMapOfInteger& anAllNodes =
827 aMesh->GetDataSource()->GetAllNodes();
828 TColStd_MapIteratorOfPackedMapOfInteger anIter(anAllNodes);
829 for (; anIter.More(); anIter.Next())
831 //get node coordinates to aCoord variable
832 aDataSource->GetGeom(anIter.Key(), Standard_False, aCoords, aNbNodes, aType);
834 Standard_Real aScaleValue;
837 aScaleValue = (aCoords.Value(1) - (Standard_Real) aMinX) / aDelta;
838 } catch(Standard_Failure) {
842 aScaleMap.Bind(anIter.Key(), aScaleValue);
845 //set color map for builder and a color for invalid scale value
846 aBuilder->SetColorMap(aColorMap);
847 aBuilder->SetInvalidColor(Quantity_NOC_BLACK);
848 aBuilder->SetTextureCoords(aScaleMap);
849 aMesh->AddBuilder(aBuilder, Standard_True);
852 aMesh->GetDrawer()->SetBoolean (MeshVS_DA_ColorReflection, aReflection != 0);
854 anIC->Redisplay( aMesh );
858 di << "Wrong mode name\n";
863 catch ( Standard_Failure )
870 //-----------------------------------------------------------------------------
871 static Standard_Integer meshvectors( Draw_Interpretor& di,
872 Standard_Integer argc,
877 di << "Wrong number of parameters\n";
878 di << "Use : meshvectors <mesh name> < -mode {elem|nodal|none} > [-maxlen len] [-color name] [-arrowpart ratio] [-issimple {1|0}]\n";
879 di << "Supported mode values:\n";
880 di << " elem - vector per element\n";
881 di << " nodal - vector per node\n";
882 di << " none - clear\n";
887 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[ 1 ], di );
889 if ( aMesh.IsNull() )
891 di << "Mesh not found\n";
894 Handle(AIS_InteractiveContext) anIC = ViewerTest::GetAISContext();
897 di << "The context is null\n";
901 TCollection_AsciiString aParam;
902 TCollection_AsciiString aMode("none");
903 Standard_Real aMaxlen(1.0);
904 Quantity_Color aColor(Quantity_NOC_ORANGE);
905 Standard_Real anArrowPart(0.1);
906 Standard_Boolean isSimplePrs(Standard_False);
908 for (Standard_Integer anIdx = 2; anIdx < argc; anIdx++)
910 if (!aParam.IsEmpty())
912 if (aParam == "-mode")
916 else if (aParam == "-maxlen")
918 aMaxlen = Draw::Atof(argv[anIdx]);
920 else if (aParam == "-color")
922 aColor = ViewerTest::GetColorFromName(argv[anIdx]);
924 else if (aParam == "-arrowpart")
926 anArrowPart = Draw::Atof(argv[anIdx]);
928 else if (aParam == "-issimple")
930 isSimplePrs = Draw::Atoi(argv[anIdx]) != 0;
934 else if (argv[anIdx][0] == '-')
936 aParam = argv[anIdx];
940 if( !aMode.IsEqual("elem") && !aMode.IsEqual("nodal") && !aMode.IsEqual("none") )
942 di << "Wrong mode name\n";
946 Handle(MeshVS_PrsBuilder) aTempBuilder;
948 aTempBuilder = aMesh->FindBuilder("MeshVS_VectorPrsBuilder");
949 if( !aTempBuilder.IsNull())
950 aMesh->RemoveBuilderById(aTempBuilder->GetId());
952 if( !aMode.IsEqual("none") )
954 Handle(MeshVS_VectorPrsBuilder) aBuilder = new MeshVS_VectorPrsBuilder( aMesh.operator->(),
957 MeshVS_DMF_VectorDataPrs,
963 Standard_Boolean anIsElement = aMode.IsEqual("elem");
964 const TColStd_PackedMapOfInteger& anAllIDs = anIsElement ? aMesh->GetDataSource()->GetAllElements() :
965 aMesh->GetDataSource()->GetAllNodes();
967 Standard_Integer aNbNodes;
968 MeshVS_EntityType aEntType;
970 TColStd_Array1OfReal aCoords(1, 3);
972 TColStd_MapIteratorOfPackedMapOfInteger anIter( anAllIDs );
973 for ( ; anIter.More(); anIter.Next() )
975 Standard_Boolean IsValidData = Standard_False;
977 aMesh->GetDataSource()->GetGeomType(anIter.Key(), anIsElement, aEntType);
978 if (aEntType == MeshVS_ET_Face)
979 IsValidData = aMesh->GetDataSource()->GetNormal(anIter.Key(), 3, aCoords.ChangeValue(1), aCoords.ChangeValue(2), aCoords.ChangeValue(3));
981 IsValidData = aMesh->GetDataSource()->GetGeom(anIter.Key(), Standard_False, aCoords, aNbNodes, aEntType);
986 aNorm = gp_Vec(aCoords.Value(1), aCoords.Value(2), aCoords.Value(3));
987 if(aNorm.Magnitude() < gp::Resolution())
989 aNorm = gp_Vec(0,0,1); //method GetGeom(...) returns coordinates of nodes
994 aNorm = gp_Vec(0,0,1);
996 aBuilder->SetVector(anIsElement, anIter.Key(), aNorm.Normalized());
999 aMesh->AddBuilder( aBuilder, Standard_False );
1000 aMesh->GetDrawer()->SetDouble ( MeshVS_DA_VectorArrowPart, anArrowPart );
1003 anIC->Redisplay( aMesh );
1007 //-----------------------------------------------------------------------------
1009 static Standard_Integer meshtext( Draw_Interpretor& di,
1010 Standard_Integer argc,
1015 di << "Wrong number of parameters\n";
1016 di << "Use : meshtext <mesh name>\n";
1020 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[ 1 ], di );
1022 if ( aMesh.IsNull() )
1024 di << "Mesh not found\n";
1028 Handle(AIS_InteractiveContext) anIC = ViewerTest::GetAISContext();
1029 if ( anIC.IsNull() )
1031 di << "The context is null\n";
1035 // Prepare triangle labels
1036 MeshVS_DataMapOfIntegerAsciiString aLabels;
1037 Standard_Integer aLen = aMesh->GetDataSource()->GetAllElements().Extent();
1038 for ( Standard_Integer anIndex = 1; anIndex <= aLen; anIndex++ ){
1039 aLabels.Bind( anIndex, TCollection_AsciiString( anIndex ) );
1042 Handle(MeshVS_TextPrsBuilder) aTextBuilder = new MeshVS_TextPrsBuilder( aMesh.operator->(), 20., Quantity_NOC_YELLOW );
1043 aTextBuilder->SetTexts( Standard_True, aLabels );
1044 aMesh->AddBuilder( aTextBuilder );
1049 static Standard_Integer meshdeform( Draw_Interpretor& di,
1050 Standard_Integer argc,
1055 di << "Wrong number of parameters\n";
1056 di << "Use : meshdeform <mesh name> < -mode {on|off} > [-scale scalefactor]\n";
1060 Handle( MeshVS_Mesh ) aMesh = getMesh( argv[ 1 ], di );
1062 if ( aMesh.IsNull() )
1064 di << "Mesh not found\n";
1067 Handle(AIS_InteractiveContext) anIC = ViewerTest::GetAISContext();
1068 if ( anIC.IsNull() )
1070 di << "The context is null\n";
1074 TCollection_AsciiString aParam;
1075 TCollection_AsciiString aMode("off");
1076 Standard_Real aScale(1.0);
1078 for (Standard_Integer anIdx = 2; anIdx < argc; anIdx++)
1080 if (!aParam.IsEmpty())
1082 if (aParam == "-mode")
1084 aMode = argv[anIdx];
1086 else if (aParam == "-scale")
1088 aScale = Draw::Atof(argv[anIdx]);
1092 else if (argv[anIdx][0] == '-')
1094 aParam = argv[anIdx];
1098 if(!aMode.IsEqual("on") && !aMode.IsEqual("off"))
1100 di << "Wrong mode name\n";
1104 Handle ( MeshVS_DeformedDataSource ) aDefDS =
1105 new MeshVS_DeformedDataSource( aMesh->GetDataSource() , aScale );
1107 const TColStd_PackedMapOfInteger& anAllIDs = aMesh->GetDataSource()->GetAllNodes();
1109 Standard_Integer aNbNodes;
1110 MeshVS_EntityType aEntType;
1112 TColStd_MapIteratorOfPackedMapOfInteger anIter( anAllIDs );
1113 for ( ; anIter.More(); anIter.Next() )
1115 TColStd_Array1OfReal aCoords(1, 3);
1116 aMesh->GetDataSource()->GetGeom(anIter.Key(), Standard_False, aCoords, aNbNodes, aEntType);
1118 gp_Vec aNorm = gp_Vec(aCoords.Value(1), aCoords.Value(2), aCoords.Value(3));
1119 if( !aNorm.Magnitude() )
1120 aNorm = gp_Vec(0,0,1);
1121 aDefDS->SetVector(anIter.Key(), aNorm.Normalized());
1124 aMesh->SetDataSource(aDefDS);
1126 anIC->Redisplay( aMesh );
1128 Handle( V3d_View ) aView = ViewerTest::CurrentView();
1129 if ( !aView.IsNull() )
1135 static Standard_Integer mesh_edge_width( Draw_Interpretor& di,
1136 Standard_Integer argc,
1144 di << "Wrong number of parameters\n";
1145 di << "Use : mesh_edge_width <mesh name> <width>\n";
1149 Handle(MeshVS_Mesh) aMesh = getMesh( argv[ 1 ], di );
1150 if ( aMesh.IsNull() )
1152 di << "Mesh not found\n";
1156 const char* aWidthStr = argv[ 2 ];
1157 if ( aWidthStr == 0 || Draw::Atof( aWidthStr ) <= 0 )
1159 di << "Width must be real value more than zero\n";
1163 double aWidth = Draw::Atof( aWidthStr );
1165 Handle(AIS_InteractiveContext) anIC = ViewerTest::GetAISContext();
1166 if ( anIC.IsNull() )
1168 di << "The context is null\n";
1172 Handle(MeshVS_Drawer) aDrawer = aMesh->GetDrawer();
1173 if ( aDrawer.IsNull() )
1175 di << "The drawer is null\n";
1179 aDrawer->SetDouble( MeshVS_DA_EdgeWidth, aWidth );
1180 anIC->Redisplay( aMesh );
1182 catch ( Standard_Failure )
1190 //-----------------------------------------------------------------------------
1192 static Standard_Integer meshinfo(Draw_Interpretor& di,
1193 Standard_Integer argc,
1198 di << "Wrong number of parameters. Use : meshinfo mesh\n";
1202 Handle(MeshVS_Mesh) aMesh = getMesh(argv[ 1 ], di);
1203 if ( aMesh.IsNull() )
1205 di << "Mesh not found\n";
1209 Handle(XSDRAWSTLVRML_DataSource) stlMeshSource = Handle(XSDRAWSTLVRML_DataSource)::DownCast(aMesh->GetDataSource());
1210 if (!stlMeshSource.IsNull())
1212 const TColStd_PackedMapOfInteger& nodes = stlMeshSource->GetAllNodes();
1213 const TColStd_PackedMapOfInteger& tris = stlMeshSource->GetAllElements();
1215 di << "Nb nodes = " << nodes.Extent() << "\n";
1216 di << "Nb triangles = " << tris.Extent() << "\n";
1222 //-----------------------------------------------------------------------------
1224 void XSDRAWSTLVRML::InitCommands (Draw_Interpretor& theCommands)
1226 const char* g = "XSTEP-STL/VRML"; // Step transfer file commands
1227 //XSDRAW::LoadDraw(theCommands);
1229 theCommands.Add ("writevrml", "shape file [version VRML#1.0/VRML#2.0 (1/2): 2 by default] [representation shaded/wireframe/both (0/1/2): 1 by default]",__FILE__,writevrml,g);
1230 theCommands.Add ("writestl", "shape file [ascii/binary (0/1) : 1 by default] [InParallel (0/1) : 0 by default]",__FILE__,writestl,g);
1231 theCommands.Add ("readstl", "shape file",__FILE__,readstl,g);
1232 theCommands.Add ("loadvrml" , "shape file",__FILE__,loadvrml,g);
1234 theCommands.Add ("meshfromstl", "creates MeshVS_Mesh from STL file", __FILE__, createmesh, g );
1235 theCommands.Add ("mesh3delem", "creates 3d element mesh to test", __FILE__, create3d, g );
1236 theCommands.Add ("meshshadcolor", "change MeshVS_Mesh shading color", __FILE__, meshcolor, g );
1237 theCommands.Add ("meshlinkcolor", "change MeshVS_Mesh line color", __FILE__, linecolor, g );
1238 theCommands.Add ("meshmat", "change MeshVS_Mesh material and transparency", __FILE__, meshmat, g );
1239 theCommands.Add ("meshshrcoef", "change MeshVS_Mesh shrink coeff", __FILE__, shrink, g );
1240 theCommands.Add ("meshclosed", "meshclosed meshname (0/1) \nChange MeshVS_Mesh drawing mode. 0 - not closed object, 1 - closed object", __FILE__, closed, g);
1241 theCommands.Add ("meshshow", "display MeshVS_Mesh object", __FILE__, mdisplay, g );
1242 theCommands.Add ("meshhide", "erase MeshVS_Mesh object", __FILE__, merase, g );
1243 theCommands.Add ("meshhidesel", "hide selected entities", __FILE__, hidesel, g );
1244 theCommands.Add ("meshshowsel", "show only selected entities", __FILE__, showonly, g );
1245 theCommands.Add ("meshshowall", "show all entities", __FILE__, showall, g );
1246 theCommands.Add ("meshcolors", "display color presentation", __FILE__, meshcolors, g );
1247 theCommands.Add ("meshvectors", "display sample vectors", __FILE__, meshvectors, g );
1248 theCommands.Add ("meshtext", "display text labels", __FILE__, meshtext, g );
1249 theCommands.Add ("meshdeform", "display deformed mesh", __FILE__, meshdeform, g );
1250 theCommands.Add ("mesh_edge_width", "set width of edges", __FILE__, mesh_edge_width, g );
1251 theCommands.Add ("meshinfo", "displays the number of nodes and triangles", __FILE__, meshinfo, g );
1254 //==============================================================================
1255 // XSDRAWSTLVRML::Factory
1256 //==============================================================================
1257 void XSDRAWSTLVRML::Factory(Draw_Interpretor& theDI)
1259 XSDRAWIGES::InitSelect();
1260 XSDRAWIGES::InitToBRep(theDI);
1261 XSDRAWIGES::InitFromBRep(theDI);
1262 XSDRAWSTEP::InitCommands(theDI);
1263 XSDRAWSTLVRML::InitCommands(theDI);
1264 XSDRAW::LoadDraw(theDI);
1266 theDI << "Draw Plugin : All TKXSDRAW commands are loaded\n";
1270 // Declare entry point PLUGINFACTORY
1271 DPLUGIN(XSDRAWSTLVRML)