FreeCAD/src/Mod/Part/App/ImportIges.cpp
2016-05-14 16:53:27 +02:00

213 lines
8.6 KiB
C++

/***************************************************************************
* Copyright (c) 2011 Werner Mayer <wmayer[at]users.sourceforge.net> *
* *
* This file is part of the FreeCAD CAx development system. *
* *
* This library is free software; you can redistribute it and/or *
* modify it under the terms of the GNU Library General Public *
* License as published by the Free Software Foundation; either *
* version 2 of the License, or (at your option) any later version. *
* *
* 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 Library General Public License for more details. *
* *
* You should have received a copy of the GNU Library General Public *
* License along with this library; see the file COPYING.LIB. If not, *
* write to the Free Software Foundation, Inc., 59 Temple Place, *
* Suite 330, Boston, MA 02111-1307, USA *
* *
***************************************************************************/
#include "PreCompiled.h"
#ifndef _PreComp_
# include <fcntl.h>
# include <BRep_Builder.hxx>
# include <Interface_Static.hxx>
# include <TopTools_HSequenceOfShape.hxx>
# include <IGESControl_Controller.hxx>
# include <IGESControl_Reader.hxx>
# include <TopoDS.hxx>
# include <TopoDS_Shape.hxx>
# include <TopoDS_Shell.hxx>
# include <TopoDS_Solid.hxx>
# include <TopoDS_Compound.hxx>
# include <TopExp_Explorer.hxx>
# include <Message_MsgFile.hxx>
# include <TColStd_HSequenceOfTransient.hxx>
# include <TColStd_HSequenceOfTransient.hxx>
# include <IGESBasic_Group.hxx>
# include <IGESSolid_ManifoldSolid.hxx>
# include <IGESBasic_SingularSubfigure.hxx>
#endif
#include <XSControl_WorkSession.hxx>
#include <XSControl_TransferReader.hxx>
#include <XSControl_WorkSession.hxx>
#include <XSControl_TransferReader.hxx>
#include <Transfer_TransientProcess.hxx>
#include <Interface_EntityIterator.hxx>
#include <Quantity_Color.hxx>
#include <TCollection_ExtendedString.hxx>
#include <Base/Console.h>
#include <Base/Sequencer.h>
#include <App/Application.h>
#include <App/Document.h>
#include "ImportIges.h"
#include "PartFeature.h"
#include "ProgressIndicator.h"
using namespace Part;
int Part::ImportIgesParts(App::Document *pcDoc, const char* FileName)
{
try {
Base::FileInfo fi(FileName);
// read iges file
IGESControl_Controller::Init();
// load data exchange message files
Message_MsgFile::LoadFromEnv("CSF_XSMessage","IGES");
// load shape healing message files
Message_MsgFile::LoadFromEnv("CSF_SHMessageStd","SHAPEStd");
IGESControl_Reader aReader;
if (aReader.ReadFile((const Standard_CString)FileName) != IFSelect_RetDone)
throw Base::Exception("Error in reading IGES");
// Ignore construction elements
// http://www.opencascade.org/org/forum/thread_20603/?forum=3
aReader.SetReadVisible(Standard_True);
// check file conformity and output stats
aReader.PrintCheckLoad(Standard_True,IFSelect_GeneralInfo);
#if 1
std::string aName = fi.fileNamePure();
Handle_Message_ProgressIndicator pi = new ProgressIndicator(100);
pi->NewScope(100, "Reading IGES file...");
pi->Show();
aReader.WS()->MapReader()->SetProgress(pi);
// make model
aReader.ClearShapes();
//Standard_Integer nbRootsForTransfer = aReader.NbRootsForTransfer();
aReader.TransferRoots();
pi->EndScope();
// put all other free-flying shapes into a single compound
Standard_Boolean emptyComp = Standard_True;
BRep_Builder builder;
TopoDS_Compound comp;
builder.MakeCompound(comp);
Standard_Integer nbShapes = aReader.NbShapes();
for (Standard_Integer i=1; i<=nbShapes; i++) {
TopoDS_Shape aShape = aReader.Shape(i);
if (!aShape.IsNull()) {
if (aShape.ShapeType() == TopAbs_SOLID ||
aShape.ShapeType() == TopAbs_COMPOUND ||
aShape.ShapeType() == TopAbs_SHELL) {
App::DocumentObject* obj = pcDoc->addObject("Part::Feature", aName.c_str());
static_cast<Part::Feature*>(obj)->Shape.setValue(aShape);
}
else {
builder.Add(comp, aShape);
emptyComp = Standard_False;
}
}
}
if (!emptyComp) {
std::string name = fi.fileNamePure();
Part::Feature *pcFeature = static_cast<Part::Feature*>(pcDoc->addObject
("Part::Feature", name.c_str()));
pcFeature->Shape.setValue(comp);
}
#else
// put all other free-flying shapes into a single compound
Standard_Boolean emptyComp = Standard_True;
BRep_Builder builder;
TopoDS_Compound comp;
builder.MakeCompound(comp);
// get all entities
Handle_TColStd_HSequenceOfTransient aRootList=aReader.GiveList("xst-transferrable-roots");
Base::SequencerLauncher seq("Reading IGES file...", aRootList->Length());
Standard_Integer j;
for (j=1; j<=aRootList->Length(); j++) {
seq.next();
Handle(IGESData_IGESEntity) igesEntity = Handle(IGESData_IGESEntity)::DownCast(aRootList->Value(j));
if (igesEntity.IsNull()) continue;
// clear any old shape
aReader.ClearShapes();
#ifdef _DEBUG
std::string type = igesEntity->DynamicType()->Name();
(void)type;
#endif
// is it a group, singular sub-figure or solid?
if (igesEntity->IsKind(STANDARD_TYPE(IGESBasic_Group)) ||
igesEntity->IsKind(STANDARD_TYPE(IGESBasic_SingularSubfigure)) ||
igesEntity->IsKind(STANDARD_TYPE(IGESSolid_ManifoldSolid))) {
try {
if (aReader.TransferEntity(igesEntity)) {
if (aReader.NbShapes()>0) {
// get the shape
std::string aName = fi.fileNamePure();
if (igesEntity->HasShortLabel()) {
Handle(TCollection_HAsciiString) aLabel=igesEntity->ShortLabel();
aName = aLabel->ToCString();
}
TopoDS_Shape aShape=aReader.OneShape();
App::DocumentObject* obj = pcDoc->addObject("Part::Feature", aName.c_str());
obj->Label.setValue(aName);
static_cast<Part::Feature*>(obj)->Shape.setValue(aShape);
int iColor;
if (igesEntity->RankColor()>-1) {
iColor = igesEntity->RankColor();
}
}
}
}
catch (Standard_Failure) {
}
}
// normal shapes
else {
try {
if (aReader.TransferEntity(igesEntity)) {
if (aReader.NbShapes()>0) {
TopoDS_Shape aShape=aReader.OneShape();
builder.Add(comp, aShape);
emptyComp = Standard_False;
}
}
}
catch (Standard_Failure) {
}
}
}
if (!emptyComp) {
std::string name = fi.fileNamePure();
Part::Feature *pcFeature = static_cast<Part::Feature*>(pcDoc->addObject
("Part::Feature", name.c_str()));
pcFeature->Shape.setValue(comp);
}
#endif
}
catch (Standard_Failure) {
Handle(Standard_Failure) aFail = Standard_Failure::Caught();
throw Base::Exception(aFail->GetMessageString());
}
return 0;
}