implement beginSimulation and getResultMesh

This commit is contained in:
Shai Seger
2017-10-22 00:46:34 +03:00
committed by Yorik van Havre
parent e366a3cfab
commit fbbd0ce35b
7 changed files with 123 additions and 60 deletions

View File

@@ -67,6 +67,7 @@ PyMOD_INIT_FUNC(PathSimulator)
try {
Base::Interpreter().runString("import Part");
Base::Interpreter().runString("import Path");
Base::Interpreter().runString("import Mesh");
}
catch (const Base::Exception& e) {
PyErr_SetString(PyExc_ImportError, e.what());

View File

@@ -14,6 +14,7 @@ include_directories(
set(PathSimulator_LIBS
Path
Part
Mesh
FreeCADApp
)

View File

@@ -25,8 +25,18 @@ Start a simulation process on a box shape stock with given resolution\n</UserDoc
</Methode>
<Methode Name="SetCurrentTool">
<Documentation>
<UserDocu>SetCurrentTool(tool):\n
Set the current Path Tool for the subsequent simulator operations.\n</UserDocu>
<UserDocu>
SetCurrentTool(tool):\n
Set the current Path Tool for the subsequent simulator operations.\n
</UserDocu>
</Documentation>
</Methode>
<Methode Name="GetResultMesh">
<Documentation>
<UserDocu>
GetResultMesh():\n
Return the current mesh result of the simulation.\n
</UserDocu>
</Documentation>
</Methode>
<Attribute Name="Tool" ReadOnly="true">

View File

@@ -23,6 +23,8 @@
#include "PreCompiled.h"
#include "Mod/Path/PathSimulator/App/PathSim.h"
#include <Mod/Part/App/TopoShapePy.h>
#include <Mod/Mesh/App/MeshPy.h>
// inclusion of the generated files (generated out of PathSimPy.xml)
#include "PathSimPy.h"
@@ -49,19 +51,47 @@ int PathSimPy::PyInit(PyObject* /*args*/, PyObject* /*kwd*/)
}
PyObject* PathSimPy::BeginSimulation(PyObject * /*args*/, PyObject * /*kwds*/)
PyObject* PathSimPy::BeginSimulation(PyObject * args, PyObject * kwds)
{
PyErr_SetString(PyExc_NotImplementedError, "Not yet implemented");
return 0;
static char *kwlist[] = { "stock", "resolution", NULL };
PyObject *pObjStock;
float resolution;
if (!PyArg_ParseTupleAndKeywords(args, kwds, "O!f", kwlist, &(Part::TopoShapePy::Type), &pObjStock, &resolution))
return 0;
PathSim *sim = getPathSimPtr();
Part::TopoShape *stock = static_cast<Part::TopoShapePy*>(pObjStock)->getTopoShapePtr();
sim->BeginSimulation(stock, resolution);
Py_IncRef(Py_None);
return Py_None;
}
PyObject* PathSimPy::SetCurrentTool(PyObject * /*args*/)
{
PyErr_SetString(PyExc_NotImplementedError, "Not yet implemented");
return 0;
PyErr_SetString(PyExc_NotImplementedError, "Not yet implemented");
return 0;
}
PyObject* PathSimPy::GetResultMesh(PyObject * args)
{
if (!PyArg_ParseTuple(args, ""))
return 0;
cStock *stock = getPathSimPtr()->m_stock;
Mesh::MeshObject *mesh = new Mesh::MeshObject();
Mesh::MeshPy *meshpy = new Mesh::MeshPy(mesh);
stock->Tesselate(*mesh);
return meshpy;
}
/* test script
import PathSimulator
sim = PathSimulator.PathSim()
stock = Part.makeBox(20,20,5)
sim.BeginSimulation(stock,0.1)
msh = sim.GetResultMesh()
Mesh.show(msh)
*/

View File

@@ -31,6 +31,7 @@
# define PathSimulatorExport __declspec(dllexport)
# define PathExport __declspec(dllexport)
# define PartExport __declspec(dllimport)
# define MeshExport __declspec(dllimport)
#else // for Linux
# define PathSimulatorAppExport
#endif

View File

@@ -20,6 +20,7 @@
* *
***************************************************************************/
#include "PreCompiled.h"
#include <algorithm>
#include "VolSim.h"
@@ -151,7 +152,7 @@ float cStock::FindRectTop(int & xp, int & yp, int & x_size, int & y_size, bool s
return z;
}
int cStock::TesselTop(Model3D *model, int xp, int yp)
int cStock::TesselTop(Mesh::MeshObject & mesh, int xp, int yp)
{
int x_size, y_size;
float z = FindRectTop(xp, yp, x_size, y_size, true);
@@ -182,7 +183,7 @@ int cStock::TesselTop(Model3D *model, int xp, int yp)
Point3D pbr(xp + x_size, yp, z);
Point3D ptl(xp, yp + y_size, z);
Point3D ptr(xp + x_size, yp + y_size, z);
model->AddQuad(pbl, pbr, ptr, ptl);
AddQuad(mesh, pbl, pbr, ptr, ptl);
}
if (farRect)
@@ -294,7 +295,7 @@ void cStock::FindRectBot(int & xp, int & yp, int & x_size, int & y_size, bool sc
}
int cStock::TesselBot(Model3D *model, int xp, int yp)
int cStock::TesselBot(Mesh::MeshObject & mesh, int xp, int yp)
{
int x_size, y_size;
FindRectBot(xp, yp, x_size, y_size, true);
@@ -323,7 +324,7 @@ int cStock::TesselBot(Model3D *model, int xp, int yp)
Point3D pbr(xp + x_size, yp, m_pz);
Point3D ptl(xp, yp + y_size, m_pz);
Point3D ptr(xp + x_size, yp + y_size, m_pz);
model->AddQuad(pbl, ptl, ptr, pbr);
AddQuad(mesh, pbl, ptl, ptr, pbr);
if (farRect)
return -1;
@@ -332,7 +333,7 @@ int cStock::TesselBot(Model3D *model, int xp, int yp)
}
int cStock::TesselSidesX(Model3D *model, int yp)
int cStock::TesselSidesX(Mesh::MeshObject & mesh, int yp)
{
float lastz1 = m_pz;
if (yp < m_y)
@@ -356,14 +357,11 @@ int cStock::TesselSidesX(Model3D *model, int yp)
{
if (abs(newz1 - lastz1) < m_res && abs(newz2 - lastz2) < m_res)
continue;
Point3D pbl(lastpoint, yp, lastz1);
Point3D pbr(x, yp, lastz1);
Point3D ptl(lastpoint, yp, lastz2);
Point3D ptr(x, yp, lastz2);
if (lastz2 > lastz1)
model->AddQuad(pbl, pbr, ptr, ptl);
else
model->AddQuad(pbl, ptl, ptr, pbr);
Point3D pbl(lastpoint, yp, lastz1);
Point3D pbr(x, yp, lastz1);
Point3D ptl(lastpoint, yp, lastz2);
Point3D ptr(x, yp, lastz2);
AddQuad(mesh, pbl, ptl, ptr, pbr);
}
lastz1 = newz1;
lastz2 = newz2;
@@ -372,7 +370,7 @@ int cStock::TesselSidesX(Model3D *model, int yp)
return 0;
}
int cStock::TesselSidesY(Model3D *model, int xp)
int cStock::TesselSidesY(Mesh::MeshObject & mesh, int xp)
{
float lastz1 = m_pz;
if (xp < m_x)
@@ -400,10 +398,7 @@ int cStock::TesselSidesY(Model3D *model, int xp)
Point3D pbl(xp, y, lastz1);
Point3D ptr(xp, lastpoint, lastz2);
Point3D ptl(xp, y, lastz2);
if (lastz2 > lastz1)
model->AddQuad(pbl, pbr, ptr, ptl);
else
model->AddQuad(pbl, ptl, ptr, pbr);
AddQuad(mesh, pbl, ptl, ptr, pbr);
}
lastz1 = newz1;
lastz2 = newz2;
@@ -412,30 +407,38 @@ int cStock::TesselSidesY(Model3D *model, int xp)
return 0;
}
void cStock::AdjustCoordinates(Model3D *model)
void cStock::SetFacetPoints(MeshCore::MeshGeomFacet & facet, Point3D & p1, Point3D & p2, Point3D & p3)
{
for (int i = 0; i < model->triangles.size(); i++)
{
for (int j = 0; j < 3; j++)
{
Point3D & t = model->triangles[i].points[j];
t.x = (float)t.x * m_res + m_px;
t.y = (float)t.y * m_res + m_py;
}
}
facet._aclPoints[0][0] = p1.x * m_res + m_px;
facet._aclPoints[0][1] = p1.y * m_res + m_py;
facet._aclPoints[0][2] = p1.z;
facet._aclPoints[1][0] = p2.x * m_res + m_px;
facet._aclPoints[1][1] = p2.y * m_res + m_py;
facet._aclPoints[1][2] = p2.z;
facet._aclPoints[2][0] = p3.x * m_res + m_px;
facet._aclPoints[2][1] = p3.y * m_res + m_py;
facet._aclPoints[2][2] = p3.z;
facet.CalcNormal();
}
Model3D *cStock::Tesselate()
void cStock::AddQuad(Mesh::MeshObject & mesh, Point3D & p1, Point3D & p2, Point3D & p3, Point3D & p4)
{
MeshCore::MeshGeomFacet facet;
SetFacetPoints(facet, p1, p2, p3);
mesh.addFacet(facet);
SetFacetPoints(facet, p1, p3, p4);
mesh.addFacet(facet);
}
void cStock::Tesselate(Mesh::MeshObject & mesh)
{
Model3D *model = new Model3D();
for (int y = 0; y < m_y; y++)
{
for (int x = 0; x < m_x; x++)
{
int attr = m_attr[x][y];
if ((attr & SIM_TESSEL_TOP) == 0)
x += TesselTop(model, x, y);
x += TesselTop(mesh, x, y);
}
}
for (int y = 0; y < m_y; y++)
@@ -445,15 +448,13 @@ Model3D *cStock::Tesselate()
if ((m_stock[x][y] - m_pz) < m_res)
m_attr[x][y] |= SIM_TESSEL_BOT;
if ((m_attr[x][y] & SIM_TESSEL_BOT) == 0)
x += TesselBot(model, x, y);
x += TesselBot(mesh, x, y);
}
}
for (int y = 0; y <= m_y; y++)
TesselSidesX(model, y);
TesselSidesX(mesh, y);
for (int x = 0; x <= m_x; x++)
TesselSidesY(model, x);
AdjustCoordinates(model);
return model;
TesselSidesY(mesh, x);
}

View File

@@ -27,6 +27,7 @@
#define PATHSIMULATOR_VolSim_H
#include <vector>
#include <Mod/Mesh/App/Mesh.h>
#define SIM_EPSILON 0.00001
#define SIM_TESSEL_TOP 1
@@ -82,15 +83,32 @@ struct cLineSegment
struct Model3D
{
Model3D() {}
std::vector<Triangle3D> triangles;
Model3D(float px, float py, float res) : pos_x(px), pos_y(py), resolution(res) {}
void SetFacetPoints(MeshCore::MeshGeomFacet & facet, Point3D & p1, Point3D & p2, Point3D & p3)
{
facet._aclPoints[0][0] = p1.x * resolution + pos_x;
facet._aclPoints[0][1] = p1.y * resolution + pos_y;
facet._aclPoints[0][2] = p1.z;
facet._aclPoints[1][0] = p2.x * resolution + pos_x;
facet._aclPoints[1][1] = p2.y * resolution + pos_y;
facet._aclPoints[1][2] = p2.z;
facet._aclPoints[2][0] = p3.x * resolution + pos_x;
facet._aclPoints[2][1] = p3.y * resolution + pos_y;
facet._aclPoints[2][2] = p3.z;
facet.CalcNormal();
}
inline void AddQuad(Point3D & p1, Point3D & p2, Point3D & p3, Point3D & p4)
{
Triangle3D t1(p1, p2, p3);
Triangle3D t2(p1, p3, p4);
triangles.push_back(t1);
triangles.push_back(t2);
MeshCore::MeshGeomFacet facet;
SetFacetPoints(facet, p1, p2, p3);
mesh.addFacet(facet);
SetFacetPoints(facet, p1, p3, p4);
mesh.addFacet(facet);
}
float pos_x, pos_y, resolution;
Mesh::MeshObject mesh;
};
class cSimTool
@@ -145,22 +163,23 @@ class cStock
public:
cStock(float px, float py, float pz, float lx, float ly, float lz, float res);
~cStock();
Model3D *Tesselate();
void CreatePocket(float x, float y, float rad, float height);
void ApplyLinearTool(Point3D & p1, Point3D & p2, cSimTool &tool);
void ApplyCircularTool(Point3D & p1, Point3D & p2, Point3D & cent, cSimTool &tool, bool isCCW);
inline Point3D & ToInner(Point3D & p) {
void Tesselate(Mesh::MeshObject & mesh);
void CreatePocket(float x, float y, float rad, float height);
void ApplyLinearTool(Point3D & p1, Point3D & p2, cSimTool &tool);
void ApplyCircularTool(Point3D & p1, Point3D & p2, Point3D & cent, cSimTool &tool, bool isCCW);
inline Point3D & ToInner(Point3D & p) {
return Point3D((p.x - m_px) / m_res, (p.y - m_py) / m_res, p.z);
}
private:
float FindRectTop(int & xp, int & yp, int & x_size, int & y_size, bool scanHoriz);
void FindRectBot(int & xp, int & yp, int & x_size, int & y_size, bool scanHoriz);
int TesselTop(Model3D *model, int x, int y);
int TesselBot(Model3D *model, int x, int y);
int TesselSidesX(Model3D *model, int yp);
int TesselSidesY(Model3D *model, int xp);
void AdjustCoordinates(Model3D *model);
void SetFacetPoints(MeshCore::MeshGeomFacet & facet, Point3D & p1, Point3D & p2, Point3D & p3);
void AddQuad(Mesh::MeshObject & mesh, Point3D & p1, Point3D & p2, Point3D & p3, Point3D & p4);
int TesselTop(Mesh::MeshObject & mesh, int x, int y);
int TesselBot(Mesh::MeshObject & mesh, int x, int y);
int TesselSidesX(Mesh::MeshObject & mesh, int yp);
int TesselSidesY(Mesh::MeshObject & mesh, int xp);
Array2D<float> m_stock;
Array2D<char> m_attr;
float m_px, m_py, m_pz; // stock zero position