167 lines
6.9 KiB
C++
167 lines
6.9 KiB
C++
// SPDX-License-Identifier: LGPL-2.1-or-later
|
|
/****************************************************************************
|
|
* *
|
|
* Copyright (c) 2023 Ondsel <development@ondsel.com> *
|
|
* *
|
|
* This file is part of FreeCAD. *
|
|
* *
|
|
* FreeCAD is free software: you can redistribute it and/or modify it *
|
|
* under the terms of the GNU Lesser General Public License as *
|
|
* published by the Free Software Foundation, either version 2.1 of the *
|
|
* License, or (at your option) any later version. *
|
|
* *
|
|
* FreeCAD 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 *
|
|
* Lesser General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Lesser General Public *
|
|
* License along with FreeCAD. If not, see *
|
|
* <https://www.gnu.org/licenses/>. *
|
|
* *
|
|
***************************************************************************/
|
|
|
|
|
|
#ifndef ASSEMBLY_AssemblyObject_H
|
|
#define ASSEMBLY_AssemblyObject_H
|
|
|
|
|
|
#include <GeomAbs_CurveType.hxx>
|
|
#include <GeomAbs_SurfaceType.hxx>
|
|
|
|
#include <Mod/Assembly/AssemblyGlobal.h>
|
|
|
|
#include <App/FeaturePython.h>
|
|
#include <App/Part.h>
|
|
#include <App/PropertyLinks.h>
|
|
|
|
namespace MbD
|
|
{
|
|
class ASMTPart;
|
|
class ASMTAssembly;
|
|
class ASMTJoint;
|
|
class ASMTMarker;
|
|
class ASMTPart;
|
|
} // namespace MbD
|
|
|
|
namespace Base
|
|
{
|
|
class Placement;
|
|
class Rotation;
|
|
} // namespace Base
|
|
|
|
namespace Assembly
|
|
{
|
|
|
|
class JointGroup;
|
|
|
|
// This enum has to be the same as the one in JointObject.py
|
|
enum class JointType
|
|
{
|
|
Fixed,
|
|
Revolute,
|
|
Cylindrical,
|
|
Slider,
|
|
Ball,
|
|
Distance
|
|
};
|
|
|
|
class AssemblyExport AssemblyObject: public App::Part
|
|
{
|
|
PROPERTY_HEADER_WITH_OVERRIDE(Assembly::AssemblyObject);
|
|
|
|
public:
|
|
AssemblyObject();
|
|
~AssemblyObject() override;
|
|
|
|
PyObject* getPyObject() override;
|
|
|
|
/// returns the type name of the ViewProvider
|
|
const char* getViewProviderName() const override
|
|
{
|
|
return "AssemblyGui::ViewProviderAssembly";
|
|
}
|
|
|
|
int solve(bool enableRedo = false);
|
|
void savePlacementsForUndo();
|
|
void undoSolve();
|
|
void clearUndo();
|
|
void exportAsASMT(std::string fileName);
|
|
std::shared_ptr<MbD::ASMTAssembly> makeMbdAssembly();
|
|
std::shared_ptr<MbD::ASMTPart>
|
|
makeMbdPart(std::string& name, Base::Placement plc = Base::Placement(), double mass = 1.0);
|
|
std::shared_ptr<MbD::ASMTPart> getMbDPart(App::DocumentObject* obj);
|
|
std::shared_ptr<MbD::ASMTMarker> makeMbdMarker(std::string& name, Base::Placement& plc);
|
|
std::vector<std::shared_ptr<MbD::ASMTJoint>> makeMbdJoint(App::DocumentObject* joint);
|
|
std::shared_ptr<MbD::ASMTJoint> makeMbdJointOfType(App::DocumentObject* joint,
|
|
JointType jointType);
|
|
std::shared_ptr<MbD::ASMTJoint> makeMbdJointDistance(App::DocumentObject* joint);
|
|
std::shared_ptr<MbD::ASMTJoint> makeMbdJointDistanceFaceVertex(App::DocumentObject* joint);
|
|
std::shared_ptr<MbD::ASMTJoint> makeMbdJointDistanceEdgeVertex(App::DocumentObject* joint);
|
|
std::shared_ptr<MbD::ASMTJoint> makeMbdJointDistanceFaceEdge(App::DocumentObject* joint);
|
|
std::shared_ptr<MbD::ASMTJoint> makeMbdJointDistanceEdgeEdge(App::DocumentObject* joint);
|
|
std::shared_ptr<MbD::ASMTJoint> makeMbdJointDistanceFaceFace(App::DocumentObject* joint);
|
|
|
|
std::string handleOneSideOfJoint(App::DocumentObject* joint,
|
|
const char* propObjLinkName,
|
|
const char* propPlcName);
|
|
void jointParts(std::vector<App::DocumentObject*> joints);
|
|
std::vector<App::DocumentObject*> getJoints();
|
|
std::vector<App::DocumentObject*> getGroundedJoints();
|
|
void fixGroundedPart(App::DocumentObject* obj, Base::Placement& plc, std::string& jointName);
|
|
std::vector<App::DocumentObject*> fixGroundedParts();
|
|
|
|
void removeUnconnectedJoints(std::vector<App::DocumentObject*>& joints,
|
|
std::vector<App::DocumentObject*> groundedObjs);
|
|
void traverseAndMarkConnectedParts(App::DocumentObject* currentPart,
|
|
std::set<App::DocumentObject*>& connectedParts,
|
|
const std::vector<App::DocumentObject*>& joints);
|
|
std::vector<App::DocumentObject*>
|
|
getConnectedParts(App::DocumentObject* part, const std::vector<App::DocumentObject*>& joints);
|
|
|
|
JointGroup* getJointGroup();
|
|
|
|
void swapJCS(App::DocumentObject* joint);
|
|
|
|
void setNewPlacements();
|
|
void recomputeJointPlacements(std::vector<App::DocumentObject*> joints);
|
|
|
|
bool isPartConnected(App::DocumentObject* obj);
|
|
|
|
double getObjMass(App::DocumentObject* obj);
|
|
void setObjMasses(std::vector<std::pair<App::DocumentObject*, double>> objectMasses);
|
|
|
|
bool isEdgeType(App::DocumentObject* obj, const char* elName, GeomAbs_CurveType type);
|
|
bool isFaceType(App::DocumentObject* obj, const char* elName, GeomAbs_SurfaceType type);
|
|
double getFaceRadius(App::DocumentObject* obj, const char* elName);
|
|
double getEdgeRadius(App::DocumentObject* obj, const char* elName);
|
|
|
|
// getters to get from properties
|
|
double getJointDistance(App::DocumentObject* joint);
|
|
JointType getJointType(App::DocumentObject* joint);
|
|
const char* getElementFromProp(App::DocumentObject* obj, const char* propName);
|
|
std::string getElementTypeFromProp(App::DocumentObject* obj, const char* propName);
|
|
Base::Placement getPlacementFromProp(App::DocumentObject* obj, const char* propName);
|
|
App::DocumentObject* getLinkObjFromProp(App::DocumentObject* joint, const char* propName);
|
|
App::DocumentObject*
|
|
getLinkedObjFromNameProp(App::DocumentObject* joint, const char* pObjName, const char* pPart);
|
|
|
|
private:
|
|
std::shared_ptr<MbD::ASMTAssembly> mbdAssembly;
|
|
|
|
std::unordered_map<App::DocumentObject*, std::shared_ptr<MbD::ASMTPart>> objectPartMap;
|
|
std::vector<std::pair<App::DocumentObject*, double>> objMasses;
|
|
|
|
std::vector<std::pair<App::DocumentObject*, Base::Placement>> previousPositions;
|
|
|
|
// void handleChangedPropertyType(Base::XMLReader &reader, const char *TypeName, App::Property
|
|
// *prop) override;
|
|
};
|
|
|
|
// using AssemblyObjectPython = App::FeaturePythonT<AssemblyObject>;
|
|
|
|
} // namespace Assembly
|
|
|
|
|
|
#endif // ASSEMBLY_AssemblyObject_H
|