[Part]add PropertyTopoShapeList
This commit is contained in:
193
src/Mod/Part/App/PropertyTopoShapeList.cpp
Normal file
193
src/Mod/Part/App/PropertyTopoShapeList.cpp
Normal file
@@ -0,0 +1,193 @@
|
||||
/***************************************************************************
|
||||
* Copyright (c) 2023 WandererFan <wandererfan@gmail.com> *
|
||||
* *
|
||||
* 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"
|
||||
|
||||
#include <Base/Console.h>
|
||||
#include <Base/Reader.h>
|
||||
#include <Base/Writer.h>
|
||||
|
||||
#include "TopoShapePy.h"
|
||||
#include "Part2DObject.h"
|
||||
|
||||
#include "PropertyTopoShapeList.h"
|
||||
|
||||
using namespace App;
|
||||
using namespace Base;
|
||||
using namespace std;
|
||||
using namespace Part;
|
||||
|
||||
|
||||
//**************************************************************************
|
||||
// PropertyTopoShapeList
|
||||
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||||
|
||||
TYPESYSTEM_SOURCE(Part::PropertyTopoShapeList, App::PropertyLists)
|
||||
|
||||
//**************************************************************************
|
||||
// Construction/Destruction
|
||||
|
||||
PropertyTopoShapeList::PropertyTopoShapeList()
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
PropertyTopoShapeList::~PropertyTopoShapeList()
|
||||
{
|
||||
for (std::vector<TopoShape*>::iterator it = _lValueList.begin(); it != _lValueList.end(); ++it)
|
||||
if (*it) delete *it;
|
||||
}
|
||||
|
||||
void PropertyTopoShapeList::setSize(int newSize)
|
||||
{
|
||||
for (unsigned int i = newSize; i < _lValueList.size(); i++)
|
||||
delete _lValueList[i];
|
||||
_lValueList.resize(newSize);
|
||||
}
|
||||
|
||||
int PropertyTopoShapeList::getSize() const
|
||||
{
|
||||
return static_cast<int>(_lValueList.size());
|
||||
}
|
||||
|
||||
void PropertyTopoShapeList::setValue(TopoShape* lValue)
|
||||
{
|
||||
if (lValue) {
|
||||
aboutToSetValue();
|
||||
_lValueList.resize(1);
|
||||
_lValueList[0] = lValue;
|
||||
hasSetValue();
|
||||
}
|
||||
}
|
||||
|
||||
void PropertyTopoShapeList::setValues(const std::vector<TopoShape*>& lValue)
|
||||
{
|
||||
aboutToSetValue();
|
||||
_lValueList.resize(lValue.size());
|
||||
for (unsigned int i = 0; i < lValue.size(); i++)
|
||||
_lValueList[i] = lValue[i];
|
||||
hasSetValue();
|
||||
}
|
||||
|
||||
PyObject *PropertyTopoShapeList::getPyObject()
|
||||
{
|
||||
PyObject* list = PyList_New(getSize());
|
||||
for (int i = 0; i < getSize(); i++)
|
||||
PyList_SetItem( list, i, _lValueList[i]->getPyObject());
|
||||
return list;
|
||||
}
|
||||
|
||||
|
||||
void PropertyTopoShapeList::setPyObject(PyObject *value)
|
||||
{
|
||||
// check container of this property to notify about changes
|
||||
Part2DObject* part2d = dynamic_cast<Part2DObject*>(this->getContainer());
|
||||
|
||||
if (PySequence_Check(value)) {
|
||||
Py_ssize_t nSize = PySequence_Size(value);
|
||||
std::vector<TopoShape*> values;
|
||||
values.resize(nSize);
|
||||
|
||||
for (Py_ssize_t i=0; i < nSize; ++i) {
|
||||
PyObject* item = PySequence_GetItem(value, i);
|
||||
if (!PyObject_TypeCheck(item, &(TopoShapePy::Type))) {
|
||||
std::string error = std::string("types in list must be 'Shape', not ");
|
||||
error += item->ob_type->tp_name;
|
||||
throw Base::TypeError(error);
|
||||
}
|
||||
|
||||
values[i] = static_cast<TopoShapePy*>(item)->getTopoShapePtr();
|
||||
}
|
||||
|
||||
setValues(values);
|
||||
if (part2d)
|
||||
part2d->acceptGeometry();
|
||||
}
|
||||
else if (PyObject_TypeCheck(value, &(TopoShapePy::Type))) {
|
||||
TopoShapePy *pcObject = static_cast<TopoShapePy*>(value);
|
||||
setValue(pcObject->getTopoShapePtr());
|
||||
if (part2d)
|
||||
part2d->acceptGeometry();
|
||||
}
|
||||
else {
|
||||
std::string error = std::string("type must be 'Shape' or list of 'Shape', not ");
|
||||
error += value->ob_type->tp_name;
|
||||
throw Base::TypeError(error);
|
||||
}
|
||||
}
|
||||
|
||||
void PropertyTopoShapeList::Save(Writer &writer) const
|
||||
{
|
||||
writer.Stream() << writer.ind() << "<ShapeList count=\"" << getSize() <<"\">" << endl;
|
||||
writer.incInd();
|
||||
for (int i = 0; i < getSize(); i++) {
|
||||
_lValueList[i]->Save(writer);
|
||||
}
|
||||
writer.decInd();
|
||||
writer.Stream() << writer.ind() << "</ShapeList>" << endl ;
|
||||
}
|
||||
|
||||
void PropertyTopoShapeList::Restore(Base::XMLReader &reader)
|
||||
{
|
||||
reader.readElement("ShapeList");
|
||||
int count = reader.getAttributeAsInteger("count");
|
||||
std::vector<TopoShape*> values;
|
||||
values.reserve(count);
|
||||
for (int i = 0; i < count; i++) {
|
||||
reader.readElement("TopoShape");
|
||||
std::string file (reader.getAttribute("file") );
|
||||
if (!file.empty()) {
|
||||
// add file to reader's list of files to be loaded by each newShape's RestoreDocFile()
|
||||
TopoShape* newShape = new TopoShape();
|
||||
values.push_back(newShape);
|
||||
reader.addFile(file.c_str(), newShape);
|
||||
}
|
||||
}
|
||||
reader.readEndElement("ShapeList");
|
||||
|
||||
// set the list to the restored values
|
||||
setValues(std::move(values));
|
||||
}
|
||||
|
||||
|
||||
App::Property *PropertyTopoShapeList::Copy() const
|
||||
{
|
||||
PropertyTopoShapeList *p = new PropertyTopoShapeList();
|
||||
p->setValues(_lValueList);
|
||||
return p;
|
||||
}
|
||||
|
||||
void PropertyTopoShapeList::Paste(const Property &from)
|
||||
{
|
||||
const PropertyTopoShapeList& FromList = dynamic_cast<const PropertyTopoShapeList&>(from);
|
||||
setValues(FromList._lValueList);
|
||||
}
|
||||
|
||||
unsigned int PropertyTopoShapeList::getMemSize() const
|
||||
{
|
||||
int size = sizeof(PropertyTopoShapeList);
|
||||
for (int i = 0; i < getSize(); i++)
|
||||
size += _lValueList[i]->getMemSize();
|
||||
return size;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user