[TD]Py routines for CenterLines

This commit is contained in:
wandererfan
2019-12-13 17:01:04 -05:00
committed by WandererFan
parent 1ad2d35bce
commit a04d0f2574
21 changed files with 924 additions and 328 deletions

View File

@@ -108,18 +108,94 @@ PyObject* CenterLinePy::copy(PyObject *args)
return cpy;
}
PyObject* CenterLinePy::setFormat(PyObject* args)
//PyObject* CenterLinePy::setFormat(PyObject* args)
//{
//// Base::Console().Message("CLPI::setFormat()\n");
// PyObject* pTuple;
// int style = 1;
// double weight = 0.50;
// double red = 0.0, green = 0.0, blue = 0.0, alpha = 0.0;
// App::Color c(red, blue, green, alpha);
// bool visible = 1;
// if (!PyArg_ParseTuple(args, "O", &pTuple)) {
// return NULL;
// }
//
// TechDraw::CenterLine* cl = this->getCenterLinePtr();
// if (PyTuple_Check(pTuple)) {
// int tSize = (int) PyTuple_Size(pTuple);
// if (tSize > 3) {
// PyObject* pStyle = PyTuple_GetItem(pTuple,0);
// style = (int) PyLong_AsLong(pStyle);
// PyObject* pWeight = PyTuple_GetItem(pTuple,1);
// weight = PyFloat_AsDouble(pWeight);
// PyObject* pColor = PyTuple_GetItem(pTuple,2);
// c = DrawUtil::pyTupleToColor(pColor);
// PyObject* pVisible = PyTuple_GetItem(pTuple,3);
// visible = (bool) PyLong_AsLong(pVisible);
// cl->m_format.m_style = style;
// cl->m_format.m_weight = weight;
// cl->m_format.m_color = c;
// cl->m_format.m_visible = visible;
// }
// } else {
// Base::Console().Error("CLPI::setFormat - not a tuple!\n");
// }
//
// return Py_None;
//}
//PyObject* CenterLinePy::getFormat(PyObject *args)
//{
// (void) args;
//// Base::Console().Message("CLPI::getFormat()\n");
// TechDraw::CenterLine* cl = this->getCenterLinePtr();
// PyObject* pStyle = PyLong_FromLong((long) cl->m_format.m_style);
// PyObject* pWeight = PyFloat_FromDouble(cl->m_format.m_weight);
// PyObject* pColor = DrawUtil::colorToPyTuple(cl->m_format.m_color);
// PyObject* pVisible = PyBool_FromLong((long) cl->m_format.m_visible);
// PyObject* result = PyTuple_New(4);
// PyTuple_SET_ITEM(result, 0, pStyle);
// PyTuple_SET_ITEM(result, 1, pWeight);
// PyTuple_SET_ITEM(result, 2, pColor);
// PyTuple_SET_ITEM(result, 3, pVisible);
// return result;
//}
Py::Object CenterLinePy::getFormat(void) const
{
// Base::Console().Message("CLPI::setFormat()\n");
PyObject* pTuple;
// Base::Console().Message("CLP::getFormat()\n");
TechDraw::CenterLine* cl = this->getCenterLinePtr();
PyObject* pStyle = PyLong_FromLong((long) cl->m_format.m_style);
PyObject* pWeight = PyFloat_FromDouble(cl->m_format.m_weight);
PyObject* pColor = DrawUtil::colorToPyTuple(cl->m_format.m_color);
PyObject* pVisible = PyBool_FromLong((long) cl->m_format.m_visible);
PyObject* result = PyTuple_New(4);
PyTuple_SET_ITEM(result, 0, pStyle);
PyTuple_SET_ITEM(result, 1, pWeight);
PyTuple_SET_ITEM(result, 2, pColor);
PyTuple_SET_ITEM(result, 3, pVisible);
return Py::asObject(result);
}
void CenterLinePy::setFormat(Py::Object arg)
{
// Base::Console().Message("CLP::setFormat()\n");
PyObject* pTuple = arg.ptr();
int style = 1;
double weight = 0.50;
double red = 0.0, green = 0.0, blue = 0.0, alpha = 0.0;
App::Color c(red, blue, green, alpha);
bool visible = 1;
if (!PyArg_ParseTuple(args, "O", &pTuple)) {
return NULL;
}
TechDraw::CenterLine* cl = this->getCenterLinePtr();
if (PyTuple_Check(pTuple)) {
@@ -142,29 +218,6 @@ PyObject* CenterLinePy::setFormat(PyObject* args)
} else {
Base::Console().Error("CLPI::setFormat - not a tuple!\n");
}
return Py_None;
}
PyObject* CenterLinePy::getFormat(PyObject *args)
{
(void) args;
// Base::Console().Message("CLPI::getFormat()\n");
TechDraw::CenterLine* cl = this->getCenterLinePtr();
PyObject* pStyle = PyLong_FromLong((long) cl->m_format.m_style);
PyObject* pWeight = PyFloat_FromDouble(cl->m_format.m_weight);
PyObject* pColor = DrawUtil::colorToPyTuple(cl->m_format.m_color);
PyObject* pVisible = PyBool_FromLong((long) cl->m_format.m_visible);
PyObject* result = PyTuple_New(4);
PyTuple_SET_ITEM(result, 0, pStyle);
PyTuple_SET_ITEM(result, 1, pWeight);
PyTuple_SET_ITEM(result, 2, pColor);
PyTuple_SET_ITEM(result, 3, pVisible);
return result;
}
Py::String CenterLinePy::getTag(void) const
@@ -173,6 +226,252 @@ Py::String CenterLinePy::getTag(void) const
return Py::String(tmp);
}
Py::Long CenterLinePy::getType(void) const
{
int tmp = getCenterLinePtr()->m_type;
return Py::Long(tmp);
}
Py::Long CenterLinePy::getMode(void) const
{
int tmp = getCenterLinePtr()->m_mode;
return Py::Long(tmp);
}
void CenterLinePy::setMode(Py::Long arg)
{
PyObject* p = arg.ptr();
if (PyLong_Check(p)) {
long int temp = PyLong_AsLong(p);
getCenterLinePtr()->m_mode = (int) temp;
} else {
std::string error = std::string("type must be 'Integer', not ");
error += p->ob_type->tp_name;
throw Py::TypeError(error);
}
}
Py::Float CenterLinePy::getHorizShift(void) const
{
double shift = getCenterLinePtr()->getHShift();
return Py::asObject(PyFloat_FromDouble(shift));
}
void CenterLinePy::setHorizShift(Py::Float arg)
{
PyObject* p = arg.ptr();
if (PyFloat_Check(p)) {
double hshift = PyFloat_AsDouble(p);
double vshift = getCenterLinePtr()->getVShift();
getCenterLinePtr()->setShifts(hshift, vshift);
} else {
std::string error = std::string("type must be 'Float', not ");
error += p->ob_type->tp_name;
throw Py::TypeError(error);
}
}
Py::Float CenterLinePy::getVertShift(void) const
{
double shift = getCenterLinePtr()->getVShift();
return Py::asObject(PyFloat_FromDouble(shift));
}
void CenterLinePy::setVertShift(Py::Float arg)
{
PyObject* p = arg.ptr();
if (PyFloat_Check(p)) {
double vshift = PyFloat_AsDouble(p);
double hshift = getCenterLinePtr()->getHShift();
getCenterLinePtr()->setShifts(hshift, vshift);
} else {
std::string error = std::string("type must be 'Float', not ");
error += p->ob_type->tp_name;
throw Py::TypeError(error);
}
}
Py::Float CenterLinePy::getRotation(void) const
{
double rot = getCenterLinePtr()->getRotate();
return Py::asObject(PyFloat_FromDouble(rot));
}
void CenterLinePy::setRotation(Py::Float arg)
{
PyObject* p = arg.ptr();
if (PyFloat_Check(p)) {
double rot = PyFloat_AsDouble(p);
getCenterLinePtr()->setRotate(rot);
} else {
std::string error = std::string("type must be 'Float', not ");
error += p->ob_type->tp_name;
throw Py::TypeError(error);
}
}
Py::Float CenterLinePy::getExtension(void) const
{
double rot = getCenterLinePtr()->getExtend();
return Py::asObject(PyFloat_FromDouble(rot));
}
void CenterLinePy::setExtension(Py::Float arg)
{
PyObject* p = arg.ptr();
if (PyFloat_Check(p)) {
double ext = PyFloat_AsDouble(p);
getCenterLinePtr()->setExtend(ext);
} else {
std::string error = std::string("type must be 'Float', not ");
error += p->ob_type->tp_name;
throw Py::TypeError(error);
}
}
Py::Boolean CenterLinePy::getFlip(void) const
{
bool flip = getCenterLinePtr()->getFlip();
if (flip) {
Py_RETURN_TRUE;
} else {
Py_RETURN_FALSE;
}
}
void CenterLinePy::setFlip(Py::Boolean arg)
{
PyObject* p = arg.ptr();
if (PyBool_Check(p)) {
if (p == Py_True) {
getCenterLinePtr()->setFlip(true);
} else {
getCenterLinePtr()->setFlip(false);
}
} else {
std::string error = std::string("type must be 'Boolean', not ");
error += p->ob_type->tp_name;
throw Py::TypeError(error);
}
}
Py::Object CenterLinePy::getEdges(void) const
{
// Base::Console().Message("CLP::getEdges()\n");
TechDraw::CenterLine* cl = this->getCenterLinePtr();
std::vector<std::string> edges = cl->m_edges;
int size = edges.size();
PyObject* result = PyList_New(size);
for (auto& e: edges) {
PyList_Append(result, PyUnicode_FromString(e.c_str()));
}
return Py::asObject(result);
}
void CenterLinePy::setEdges(Py::Object arg)
{
// Base::Console().Message("CLP::setEdges()\n");
PyObject* pList = arg.ptr();
TechDraw::CenterLine* cl = this->getCenterLinePtr();
std::vector<std::string> temp;
if (PyList_Check(pList)) {
int tSize = (int) PyList_Size(pList);
int i = 0;
for ( ; i < tSize; i++) {
PyObject* item = PyList_GetItem(pList, (Py_ssize_t) i);
const char* utf8 = PyUnicode_AsUTF8(item);
std::string ssTemp(utf8);
temp.push_back(ssTemp);
}
cl->m_edges = temp;
} else {
Base::Console().Error("CLPI::setEdges - input not a list!\n");
}
}
Py::Object CenterLinePy::getFaces(void) const
{
// Base::Console().Message("CLP::getFaces()\n");
TechDraw::CenterLine* cl = this->getCenterLinePtr();
std::vector<std::string> faces = cl->m_faces;
int size = faces.size();
PyObject* result = PyList_New(size);
for (auto& f: faces) {
PyList_Append(result, PyUnicode_FromString(f.c_str()));
}
return Py::asObject(result);
}
void CenterLinePy::setFaces(Py::Object arg)
{
// Base::Console().Message("CLP::setFaces()\n");
PyObject* pList = arg.ptr();
TechDraw::CenterLine* cl = this->getCenterLinePtr();
std::vector<std::string> temp;
if (PyList_Check(pList)) {
int tSize = (int) PyList_Size(pList);
int i = 0;
for ( ; i < tSize; i++) {
PyObject* item = PyList_GetItem(pList, (Py_ssize_t) i);
const char* utf8 = PyUnicode_AsUTF8(item);
std::string ssTemp(utf8);
temp.push_back(ssTemp);
}
cl->m_faces = temp;
} else {
Base::Console().Error("CLPI::setFaces - input not a list!\n");
}
}
Py::Object CenterLinePy::getPoints(void) const
{
// Base::Console().Message("CLP::getPoints()\n");
TechDraw::CenterLine* cl = this->getCenterLinePtr();
std::vector<std::string> points = cl->m_verts;
int size = points.size();
PyObject* result = PyList_New(size);
for (auto& p: points) {
PyList_Append(result, PyUnicode_FromString(p.c_str()));
}
return Py::asObject(result);
}
void CenterLinePy::setPoints(Py::Object arg)
{
// Base::Console().Message("CLP::setPoints()\n");
PyObject* pList = arg.ptr();
TechDraw::CenterLine* cl = this->getCenterLinePtr();
std::vector<std::string> temp;
if (PyList_Check(pList)) {
int tSize = (int) PyList_Size(pList);
int i = 0;
for ( ; i < tSize; i++) {
PyObject* item = PyList_GetItem(pList, (Py_ssize_t) i);
const char* utf8 = PyUnicode_AsUTF8(item);
std::string ssTemp(utf8);
temp.push_back(ssTemp);
}
cl->m_verts = temp;
} else {
Base::Console().Error("CLPI::setPoints - input not a list!\n");
}
}
PyObject *CenterLinePy::getCustomAttributes(const char* /*attr*/) const
{
return 0;