mirror of
https://git.dev.opencascade.org/repos/occt.git
synced 2025-04-10 18:51:21 +03:00
Improved vrml reading
This commit is contained in:
dbv
ema
parent
8b2102869f
commit
e1879ccf43
@ -63,6 +63,9 @@
|
||||
#include <BRepMesh_IncrementalMesh.hxx>
|
||||
#include <Geom_CylindricalSurface.hxx>
|
||||
|
||||
#include <VrmlData_ShapeNode.hxx>
|
||||
#include <VrmlData_Group.hxx>
|
||||
|
||||
#include <StlAPI.hxx>
|
||||
|
||||
#include <stdio.h>
|
||||
@ -829,7 +832,7 @@ TDF_Label ReadVrmlRec(const OSD_Path& thePath, const Handle(TDocStd_Document)& t
|
||||
ReadVrmlRec(aSubDirPath, theDoc, aNewLabel);
|
||||
}
|
||||
|
||||
aShapeTool->UpdateAssembly(aNewLabel);
|
||||
aShapeTool->UpdateAssemblies();
|
||||
}
|
||||
// At the end of operation update assemblies
|
||||
}
|
||||
@ -841,6 +844,249 @@ TDF_Label ReadVrmlRec(const OSD_Path& thePath, const Handle(TDocStd_Document)& t
|
||||
return aNewLabel;
|
||||
}
|
||||
|
||||
//=======================================================================
|
||||
//function : HaveShapeNode
|
||||
//purpose : Check that VRML node have Shape
|
||||
//=======================================================================
|
||||
Standard_Boolean HaveShapeNode (const Handle (VrmlData_Node)& theNode)
|
||||
{
|
||||
// Try a Shape type of node.
|
||||
const Handle (VrmlData_ShapeNode) aShapeNode = Handle (VrmlData_ShapeNode)::DownCast (theNode);
|
||||
if (!aShapeNode.IsNull ())
|
||||
{
|
||||
const Handle (VrmlData_Geometry) aGeometry = aShapeNode->Geometry ();
|
||||
if (!aGeometry.IsNull ())
|
||||
{
|
||||
const Handle (TopoDS_TShape) aTShape = aGeometry->TShape ();
|
||||
if (!aTShape.IsNull ())
|
||||
{
|
||||
return Standard_True;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Try a Group type of node.
|
||||
const Handle (VrmlData_Group) aGroupNode = Handle (VrmlData_Group)::DownCast (theNode);
|
||||
if (!aGroupNode.IsNull ())
|
||||
{
|
||||
for (VrmlData_Group::Iterator anIt = aGroupNode->NodeIterator (); anIt.More (); anIt.Next ())
|
||||
{
|
||||
if (HaveShapeNode (anIt.Value ()))
|
||||
{
|
||||
return Standard_True;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return Standard_False;
|
||||
}
|
||||
|
||||
//=======================================================================
|
||||
//function : ReadVrmlNode
|
||||
//purpose : Read VRML node to DECAF document
|
||||
//=======================================================================
|
||||
TDF_Label ReadVrmlNode (const Handle(VrmlData_Node)& theNode,
|
||||
const Handle(TDocStd_Document)& theDoc,
|
||||
const TDF_Label& theLabel = TDF_Label())
|
||||
{
|
||||
TDF_Label aNewLabel;
|
||||
Handle (XCAFDoc_ShapeTool) aShapeTool = XCAFDoc_DocumentTool::ShapeTool (theDoc->Main ());
|
||||
Handle (XCAFDoc_ColorTool) aColorTool = XCAFDoc_DocumentTool::ColorTool (theDoc->Main ());
|
||||
|
||||
// Try a Shape type of node.
|
||||
const Handle (VrmlData_ShapeNode) aShapeNode = Handle (VrmlData_ShapeNode)::DownCast (theNode);
|
||||
if (!aShapeNode.IsNull ())
|
||||
{
|
||||
const Handle (VrmlData_Geometry) aGeometry = aShapeNode->Geometry ();
|
||||
if (!aGeometry.IsNull ())
|
||||
{
|
||||
const Handle (TopoDS_TShape) aTShape = aGeometry->TShape ();
|
||||
if (!aTShape.IsNull ())
|
||||
{
|
||||
TopoDS_Shape aShape;
|
||||
aShape.Orientation (TopAbs_FORWARD);
|
||||
aShape.TShape (aTShape);
|
||||
|
||||
if (theLabel.IsNull () || !aShapeTool->IsAssembly (theLabel))
|
||||
{
|
||||
// Add new shape.
|
||||
aNewLabel = aShapeTool->AddShape (aShape, Standard_False);
|
||||
}
|
||||
else if (!theLabel.IsNull() && aShapeTool->IsAssembly (theLabel))
|
||||
{
|
||||
// Add shape as component.
|
||||
aNewLabel = aShapeTool->AddComponent (theLabel, aShape);
|
||||
|
||||
if (aShapeTool->IsReference (aNewLabel))
|
||||
{
|
||||
TDF_Label aRefLabel;
|
||||
if (aShapeTool->GetReferredShape (aNewLabel, aRefLabel))
|
||||
{
|
||||
aNewLabel = aRefLabel;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!aNewLabel.IsNull ())
|
||||
{
|
||||
TDataStd_Name::Set (aNewLabel, aShapeNode->Name ());
|
||||
Quantity_Color aFaceColor;
|
||||
Quantity_Color anEdgeColor (Quantity_NOC_BLACK);
|
||||
const Handle (VrmlData_Appearance) anAppearance = aShapeNode->Appearance ();
|
||||
if (!anAppearance.IsNull ())
|
||||
{
|
||||
aFaceColor = anAppearance->Material ()->DiffuseColor ();
|
||||
}
|
||||
aColorTool->SetColor (aNewLabel, aFaceColor, XCAFDoc_ColorSurf);
|
||||
aColorTool->SetColor (aNewLabel, anEdgeColor, XCAFDoc_ColorCurv);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Try a Group type of node.
|
||||
const Handle (VrmlData_Group) aGroupNode = Handle (VrmlData_Group)::DownCast (theNode);
|
||||
if (!aGroupNode.IsNull ()/* && HaveShapeNode (aGroupNode)*/)
|
||||
{
|
||||
// Create assembly
|
||||
TopoDS_Compound aCompound;
|
||||
BRep_Builder aBuilder;
|
||||
aBuilder.MakeCompound (aCompound);
|
||||
aCompound.Location (aGroupNode->GetTransform ());
|
||||
|
||||
if (theLabel.IsNull () || !aShapeTool->IsAssembly (theLabel))
|
||||
{
|
||||
// Add new shape.
|
||||
aNewLabel = aShapeTool->AddShape (aCompound);
|
||||
}
|
||||
else if (!theLabel.IsNull () && aShapeTool->IsAssembly (theLabel))
|
||||
{
|
||||
// Add shape as component.
|
||||
aNewLabel = aShapeTool->AddComponent (theLabel, aCompound, Standard_True);
|
||||
|
||||
if (aShapeTool->IsReference (aNewLabel))
|
||||
{
|
||||
TDF_Label aRefLabel;
|
||||
if (aShapeTool->GetReferredShape (aNewLabel, aRefLabel))
|
||||
{
|
||||
aNewLabel = aRefLabel;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (!aNewLabel.IsNull ())
|
||||
{
|
||||
TDataStd_Name::Set (aNewLabel, aGroupNode->Name ());
|
||||
}
|
||||
|
||||
// Add components.
|
||||
for (VrmlData_Group::Iterator anIt = aGroupNode->NodeIterator (); anIt.More (); anIt.Next ())
|
||||
{
|
||||
ReadVrmlNode (anIt.Value (), theDoc, aNewLabel);
|
||||
}
|
||||
|
||||
aShapeTool->UpdateAssemblies();
|
||||
}
|
||||
|
||||
return aNewLabel;
|
||||
}
|
||||
|
||||
//=======================================================================
|
||||
//function : ReadSingleVrml
|
||||
//purpose : Read VRML file to DECAF document
|
||||
//=======================================================================
|
||||
bool ReadSingleVrml (const OSD_Path& thePath, const Handle (TDocStd_Document)& theDoc)
|
||||
{
|
||||
// Get path of the VRML file.
|
||||
TCollection_AsciiString aPath (".");
|
||||
TCollection_AsciiString aDisk = thePath.Disk ();
|
||||
TCollection_AsciiString aTrek = thePath.Trek ();
|
||||
TCollection_AsciiString aName = thePath.Name ();
|
||||
TCollection_AsciiString anExt = thePath.Extension ();
|
||||
|
||||
if (!aTrek.IsEmpty())
|
||||
{
|
||||
if (!aDisk.IsEmpty())
|
||||
{
|
||||
aPath = aDisk;
|
||||
}
|
||||
else
|
||||
{
|
||||
aPath.Clear ();
|
||||
}
|
||||
aTrek.ChangeAll ('|', '/');
|
||||
aTrek.ChangeAll ('\\', '/');
|
||||
aPath += aTrek;
|
||||
|
||||
if (!aName.IsEmpty())
|
||||
{
|
||||
aPath += aName;
|
||||
}
|
||||
|
||||
if (!anExt.IsEmpty())
|
||||
{
|
||||
aPath += anExt;
|
||||
}
|
||||
}
|
||||
|
||||
// Analize the passed path.
|
||||
if (anExt != ".wrl")
|
||||
{
|
||||
std::cerr << "Reading failed. Unknown file format: " << anExt << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
// Open file.
|
||||
filebuf aFileBuf;
|
||||
aFileBuf.open (aPath.ToCString (), ios::in);
|
||||
if (aFileBuf.is_open () == Standard_False)
|
||||
{
|
||||
std::cerr << "Reading failed. Can not open file: " << aPath.ToCString () << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
// Read vrml file.
|
||||
Standard_IStream aStream (&aFileBuf);
|
||||
VrmlData_Scene aScene;
|
||||
aScene.SetVrmlDir (aPath);
|
||||
aScene << aStream;
|
||||
aFileBuf.close();
|
||||
if (aScene.Status () != VrmlData_StatusOK)
|
||||
{
|
||||
std::cerr << "Reading failed. Error: " << aScene.Status () << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
// Read shape.
|
||||
//VrmlData_DataMapOfShapeAppearance aShapeAppearMap;
|
||||
//TopoDS_Shape aShape = aScene.GetShape (aShapeAppearMap);
|
||||
//if (aShape.IsNull ())
|
||||
//{
|
||||
// std::cerr << "Reading failed. Getting TopoDS_Shape failed." << std::endl;
|
||||
// return false;
|
||||
//}
|
||||
|
||||
// Create assembly.
|
||||
TopoDS_Compound aCompound;
|
||||
BRep_Builder aBuilder;
|
||||
aBuilder.MakeCompound (aCompound);
|
||||
Handle (XCAFDoc_ShapeTool) aShapeTool = XCAFDoc_DocumentTool::ShapeTool (theDoc->Main ());
|
||||
TDF_Label aNewLabel = aShapeTool->AddShape (aCompound);
|
||||
if (!aNewLabel.IsNull ())
|
||||
{
|
||||
TDataStd_Name::Set (aNewLabel, aName);
|
||||
}
|
||||
|
||||
// Read nodes one by one and add to document.
|
||||
for (VrmlData_Scene::Iterator anIt = aScene.GetIterator (); anIt.More (); anIt.Next ())
|
||||
{
|
||||
ReadVrmlNode (anIt.Value (), theDoc, aNewLabel);
|
||||
}
|
||||
aShapeTool->UpdateAssemblies();
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
static Standard_Integer ReadVrml (Draw_Interpretor& di, Standard_Integer argc, const char** argv)
|
||||
{
|
||||
if ( argc != 3 ) {
|
||||
@ -858,7 +1104,16 @@ static Standard_Integer ReadVrml (Draw_Interpretor& di, Standard_Integer argc, c
|
||||
Draw::Set(argv[1], DD);
|
||||
}
|
||||
|
||||
ReadVrmlRec(OSD_Path(argv[2]), doc, TDF_Label());
|
||||
OSD_Path aPath (argv[2]);
|
||||
|
||||
if (aPath.Extension () == ".wrl")
|
||||
{
|
||||
ReadSingleVrml (aPath, doc);
|
||||
}
|
||||
else if (aPath.Extension ().IsEmpty ())
|
||||
{
|
||||
ReadVrmlRec (aPath, doc, TDF_Label ());
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user