194 lines
7.9 KiB
C++
194 lines
7.9 KiB
C++
/******************************************************************************
|
|
* Copyright (c)2013 Jan Rheinlaender <jrheinlaender@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 <BRepAlgoAPI_Fuse.hxx>
|
|
# include <BRepAlgoAPI_Cut.hxx>
|
|
# include <BRepAlgoAPI_Common.hxx>
|
|
# include <BRepAlgoAPI_Section.hxx>
|
|
# include <gp_Trsf.hxx>
|
|
# include <gp_Pnt.hxx>
|
|
# include <gp_Dir.hxx>
|
|
# include <gp_Vec.hxx>
|
|
# include <gp_Ax1.hxx>
|
|
#include <BRepBuilderAPI_GTransform.hxx>
|
|
#endif
|
|
|
|
#include "Body.h"
|
|
#include "FeatureBoolean.h"
|
|
|
|
#include <Base/Console.h>
|
|
#include <Base/Exception.h>
|
|
#include <App/Document.h>
|
|
#include <App/Part.h>
|
|
|
|
using namespace PartDesign;
|
|
|
|
namespace PartDesign {
|
|
|
|
PROPERTY_SOURCE(PartDesign::Boolean, PartDesign::Feature)
|
|
|
|
const char* Boolean::TypeEnums[]= {"Fuse","Cut","Common","Section",NULL};
|
|
|
|
Boolean::Boolean()
|
|
{
|
|
ADD_PROPERTY(Type,((long)0));
|
|
Type.setEnums(TypeEnums);
|
|
ADD_PROPERTY(Bodies,(0));
|
|
Bodies.setSize(0);
|
|
}
|
|
|
|
short Boolean::mustExecute() const
|
|
{
|
|
if (Bodies.isTouched())
|
|
return 1;
|
|
return PartDesign::Feature::mustExecute();
|
|
}
|
|
|
|
App::DocumentObjectExecReturn *Boolean::execute(void)
|
|
{
|
|
// Check the parameters
|
|
const Part::Feature* baseFeature;
|
|
try {
|
|
baseFeature = getBaseObject();
|
|
} catch (const Base::Exception&) {
|
|
return new App::DocumentObjectExecReturn("Cannot do boolean operation with invalid BaseFeature");
|
|
}
|
|
|
|
std::vector<App::DocumentObject*> bodies = Bodies.getValues();
|
|
if (bodies.empty())
|
|
return App::DocumentObject::StdReturn;
|
|
|
|
// Get the base shape to operate on
|
|
Part::TopoShape baseTopShape = baseFeature->Shape.getShape();
|
|
if (baseTopShape._Shape.IsNull())
|
|
return new App::DocumentObjectExecReturn("Cannot do boolean operation with invalid base shape");
|
|
|
|
//get the body this boolean feature belongs to
|
|
PartDesign::Body* baseBody = NULL;
|
|
for(PartDesign::Body* b : this->getDocument()->getObjectsOfType<PartDesign::Body>()) {
|
|
if(b->isFeature(this)) {
|
|
baseBody = b;
|
|
break;
|
|
}
|
|
}
|
|
if(!baseBody)
|
|
return new App::DocumentObjectExecReturn("Cannot do boolean on feature which is not in a body");
|
|
|
|
//get the part every body should belong to
|
|
App::Part* part = NULL;
|
|
for(App::Part* p : this->getDocument()->getObjectsOfType<App::Part>()) {
|
|
if(p->hasObject(baseBody)) {
|
|
part = p;
|
|
break;
|
|
}
|
|
}
|
|
if(!part)
|
|
return new App::DocumentObjectExecReturn("Cannot do boolean on body which is not in a part");
|
|
|
|
|
|
// TODO: Why is Feature::getLocation() protected?
|
|
Base::Placement place = baseBody->Placement.getValue();
|
|
Base::Rotation rot(place.getRotation());
|
|
Base::Vector3d axis;
|
|
double angle;
|
|
rot.getValue(axis, angle);
|
|
gp_Trsf trf;
|
|
trf.SetRotation(gp_Ax1(gp_Pnt(), gp_Dir(axis.x, axis.y, axis.z)), angle);
|
|
trf.SetTranslationPart(gp_Vec(place.getPosition().x,place.getPosition().y,place.getPosition().z));
|
|
TopLoc_Location objLoc(trf);
|
|
|
|
TopoDS_Shape result = baseTopShape._Shape;
|
|
result.Move(objLoc);
|
|
|
|
// Get the operation type
|
|
std::string type = Type.getValueAsString();
|
|
|
|
for (std::vector<App::DocumentObject*>::const_iterator b = bodies.begin(); b != bodies.end(); b++)
|
|
{
|
|
// Extract the body shape. Its important to get the actual feature that provides the last solid in the body
|
|
// so that the placement will be right
|
|
PartDesign::Body* body = static_cast<PartDesign::Body*>(*b);
|
|
if(!part->hasObject(body))
|
|
return new App::DocumentObjectExecReturn("Cannot do boolean on bodies of different parts");
|
|
|
|
TopoDS_Shape shape = body->getTipShape()._Shape;
|
|
|
|
// Move the shape to the location of the base shape in the other body
|
|
Base::Placement pl = body->Placement.getValue();
|
|
// TODO: Why is Feature::getLocation() protected?
|
|
Base::Rotation rot(pl.getRotation());
|
|
Base::Vector3d axis;
|
|
double angle;
|
|
rot.getValue(axis, angle);
|
|
gp_Trsf trf;
|
|
trf.SetRotation(gp_Ax1(gp_Pnt(), gp_Dir(axis.x, axis.y, axis.z)), angle);
|
|
trf.SetTranslationPart(gp_Vec(pl.getPosition().x,pl.getPosition().y,pl.getPosition().z));
|
|
TopLoc_Location bLoc(trf);
|
|
shape.Move(bLoc);
|
|
|
|
TopoDS_Shape boolOp;
|
|
|
|
if (type == "Fuse") {
|
|
BRepAlgoAPI_Fuse mkFuse(result, shape);
|
|
if (!mkFuse.IsDone())
|
|
return new App::DocumentObjectExecReturn("Fusion of bodies failed", *b);
|
|
// we have to get the solids (fuse sometimes creates compounds)
|
|
boolOp = this->getSolid(mkFuse.Shape());
|
|
// lets check if the result is a solid
|
|
if (boolOp.IsNull())
|
|
return new App::DocumentObjectExecReturn("Resulting shape is not a solid", *b);
|
|
} else if (type == "Cut") {
|
|
BRepAlgoAPI_Cut mkCut(result, shape);
|
|
if (!mkCut.IsDone())
|
|
return new App::DocumentObjectExecReturn("Cut out of first body failed", *b);
|
|
boolOp = mkCut.Shape();
|
|
} else if (type == "Common") {
|
|
BRepAlgoAPI_Common mkCommon(result, shape);
|
|
if (!mkCommon.IsDone())
|
|
return new App::DocumentObjectExecReturn("Common operation with first body failed", *b);
|
|
boolOp = mkCommon.Shape();
|
|
} else if (type == "Section") {
|
|
BRepAlgoAPI_Section mkSection(result, shape);
|
|
if (!mkSection.IsDone())
|
|
return new App::DocumentObjectExecReturn("Section out of first body failed", *b);
|
|
// we have to get the solids
|
|
boolOp = this->getSolid(mkSection.Shape());
|
|
// lets check if the result is a solid
|
|
if (boolOp.IsNull())
|
|
return new App::DocumentObjectExecReturn("Resulting shape is not a solid", *b);
|
|
}
|
|
|
|
result = boolOp; // Use result of this operation for fuse/cut of next body
|
|
//bring the result geometry into the correct coordinance of the body the boolean belongs to
|
|
BRepBuilderAPI_GTransform mkTrf(result, objLoc.Inverted().Transformation());
|
|
result = mkTrf.Shape();
|
|
}
|
|
|
|
this->Shape.setValue(result);
|
|
return App::DocumentObject::StdReturn;
|
|
}
|
|
|
|
}
|