You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
113 lines
3.0 KiB
C++
113 lines
3.0 KiB
C++
#include "FITKOCCModelExport.h"
|
|
|
|
// OCC
|
|
#include <TopoDS_Shape.hxx>
|
|
#include <TopoDS_Compound.hxx>
|
|
#include <BRep_Builder.hxx>
|
|
#include <BRepTools.hxx>
|
|
#include <STEPControl_Writer.hxx>
|
|
#include <IGESControl_Writer.hxx>
|
|
#include <StlAPI.hxx>
|
|
|
|
// App
|
|
#include "FITK_Kernel/FITKAppFramework/FITKAppFramework.h"
|
|
#include "FITK_Kernel/FITKAppFramework/FITKGlobalData.h"
|
|
|
|
// Abstract geometry
|
|
#include "FITK_Interface/FITKInterfaceGeometry/FITKGeoCommandList.h"
|
|
|
|
namespace OCC
|
|
{
|
|
bool FITKOCCModelExport::update()
|
|
{
|
|
// 检查文件路径。
|
|
if (_exportFileName.isEmpty())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
// 获取全局几何数据列表。
|
|
Interface::FITKGeoCommandList* geoData = FITKGLODATA->getGeometryData<Interface::FITKGeoCommandList>();
|
|
if (!geoData)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
// 初始化组合对象。
|
|
BRep_Builder builder;
|
|
TopoDS_Compound compound;
|
|
builder.MakeCompound(compound);
|
|
|
|
// 遍历所有模型，获取形状并组合成组合对象。
|
|
QList<Interface::FITKAbsGeoCommand*> rootCmds = geoData->getRootCommandList();
|
|
int nGeo = rootCmds.count();
|
|
for (int i = 0; i < nGeo; i++)
|
|
{
|
|
Interface::FITKAbsGeoCommand* cmd = rootCmds[i];
|
|
if (!cmd)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
// 获取OCC模型代理器。
|
|
FITKAbstractOCCModel* occModel = cmd->getTShapeAgent<FITKAbstractOCCModel>();
|
|
if (!occModel)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
// 获取形状并添加至组合对象。
|
|
TopoDS_Shape* shape = occModel->getShape();
|
|
builder.Add(compound, *shape);
|
|
}
|
|
|
|
// 文件写出。
|
|
bool flag = wirteFile(_exportFileName, compound);
|
|
|
|
return flag;
|
|
}
|
|
|
|
bool FITKOCCModelExport::wirteFile(QString fileName, TopoDS_Shape & shape)
|
|
{
|
|
// 检查输入信息。
|
|
if (fileName.isEmpty() || shape.IsNull())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
// 获取文件后缀。
|
|
QString lowerSuffix = fileName.split(".").last().trimmed().toLower();
|
|
bool flag = false;
|
|
QByteArray baName = fileName.toUtf8();
|
|
|
|
// 导出brep。
|
|
if (lowerSuffix == "brep")
|
|
{
|
|
flag = BRepTools::Write(shape, baName);
|
|
}
|
|
// 导出stp/step。
|
|
else if (lowerSuffix == "stp" || lowerSuffix == "step")
|
|
{
|
|
STEPControl_Writer writer;
|
|
writer.Transfer(shape, STEPControl_AsIs);
|
|
flag = writer.Write(baName);
|
|
}
|
|
// 写出igs/iges。
|
|
else if (lowerSuffix == "igs" || lowerSuffix == "iges")
|
|
{
|
|
IGESControl_Writer writer;
|
|
writer.AddShape(shape);
|
|
flag = writer.Write(baName);
|
|
}
|
|
// 写出stl。
|
|
else if (lowerSuffix == "stl")
|
|
{
|
|
flag = StlAPI::Write(shape, baName);
|
|
}
|
|
|
|
return flag;
|
|
}
|
|
}
|
|
|
|
|