+ implement FemMesh::getVolumesByFace, replace inner list by tuple

This commit is contained in:
wmayer 2015-05-08 21:45:13 +02:00
parent cd6088332f
commit 9e89f2ff9a
4 changed files with 87 additions and 10 deletions

View File

@ -404,14 +404,50 @@ std::set<long> FemMesh::getSurfaceNodes(long ElemId, short FaceId, float Angle)
return result;
}
/* That function returns map containing volume ID and face number
/*! That function returns map containing volume ID and face ID.
*/
std::map<int, int> FemMesh::getVolumesByFace(const TopoDS_Face &face) const
{
std::map<int, int> result;
std::set<int> nodes_on_face = getNodesByFace(face);
SMDS_VolumeIteratorPtr vol_iter = myMesh->GetMeshDS()->volumesIterator();
while (vol_iter->more()) {
const SMDS_MeshVolume* vol = vol_iter->next();
int numFaces = vol->NbFaces();
SMDS_ElemIteratorPtr face_iter = vol->facesIterator();
while (face_iter->more()) {
const SMDS_MeshFace* face = static_cast<const SMDS_MeshFace*>(face_iter->next());
int numNodes = face->NbNodes();
std::set<int> face_nodes;
for (int i=0; i<numNodes; i++) {
face_nodes.insert(face->GetNode(i)->GetID());
}
std::vector<int> element_face_nodes;
std::set_intersection(nodes_on_face.begin(), nodes_on_face.end(), face_nodes.begin(), face_nodes.end(),
std::back_insert_iterator<std::vector<int> >(element_face_nodes));
if (element_face_nodes.size() == numNodes) {
result[vol->GetID()] = face->GetID();
break;
}
}
}
return result;
}
/*! That function returns map containing volume ID and face number
* as per CalculiX definition for tetrahedral elements. See CalculiX
* documentation for the details.
*/
std::map<int, int> FemMesh::getccxVolumesByFace(const TopoDS_Face &face) const
{
std::map<int, int> result;
std::set<int> nodes_on_face = FemMesh::getNodesByFace(face);
std::set<int> nodes_on_face = getNodesByFace(face);
static std::map<int, std::vector<int> > elem_order;
if (elem_order.empty()) {

View File

@ -93,6 +93,8 @@ public:
std::set<int> getNodesByEdge(const TopoDS_Edge &edge) const;
/// retrieving by vertex
std::set<int> getNodesByVertex(const TopoDS_Vertex &vertex) const;
/// retrieving volume IDs and face IDs number by face
std::map<int, int> getVolumesByFace(const TopoDS_Face &face) const;
/// retrieving volume IDs and CalculiX face number by face
std::map<int, int> getccxVolumesByFace(const TopoDS_Face &face) const;
//@}

View File

@ -84,9 +84,14 @@
<UserDocu>Make a copy of this FEM mesh.</UserDocu>
</Documentation>
</Methode>
<Methode Name="getVolumesByFace" Const="true">
<Documentation>
<UserDocu>Return a dict of volume IDs and face IDs which belong to a TopoFace</UserDocu>
</Documentation>
</Methode>
<Methode Name="getccxVolumesByFace" Const="true">
<Documentation>
<UserDocu>Return a list of volume IDs which belong to a TopoFace</UserDocu>
<UserDocu>Return a dict of volume IDs and ccx face numbers which belong to a TopoFace</UserDocu>
</Documentation>
</Methode>
<Methode Name="getNodeById" Const="true">

View File

@ -514,6 +514,39 @@ PyObject* FemMeshPy::setTransform(PyObject *args)
Py_Return;
}
PyObject* FemMeshPy::getVolumesByFace(PyObject *args)
{
PyObject *pW;
if (!PyArg_ParseTuple(args, "O!", &(Part::TopoShapeFacePy::Type), &pW))
return 0;
try {
const TopoDS_Shape& sh = static_cast<Part::TopoShapeFacePy*>(pW)->getTopoShapePtr()->_Shape;
if (sh.IsNull()) {
PyErr_SetString(Base::BaseExceptionFreeCADError, "Face is empty");
return 0;
}
const TopoDS_Face& fc = TopoDS::Face(sh);
Py::List ret;
std::map<int, int> resultSet = getFemMeshPtr()->getVolumesByFace(fc);
for (std::map<int, int>::const_iterator it = resultSet.begin();it!=resultSet.end();++it) {
Py::Tuple vol_face(2);
vol_face.setItem(0, Py::Int(it->first));
vol_face.setItem(1, Py::Int(it->second));
ret.append(vol_face);
}
return Py::new_reference_to(ret);
}
catch (Standard_Failure) {
Handle_Standard_Failure e = Standard_Failure::Caught();
PyErr_SetString(Base::BaseExceptionFreeCADError, e->GetMessageString());
return 0;
}
}
PyObject* FemMeshPy::getccxVolumesByFace(PyObject *args)
{
PyObject *pW;
@ -522,22 +555,23 @@ PyObject* FemMeshPy::getccxVolumesByFace(PyObject *args)
try {
const TopoDS_Shape& sh = static_cast<Part::TopoShapeFacePy*>(pW)->getTopoShapePtr()->_Shape;
const TopoDS_Face& fc = TopoDS::Face(sh);
if (sh.IsNull()) {
PyErr_SetString(Base::BaseExceptionFreeCADError, "Face is empty");
return 0;
}
const TopoDS_Face& fc = TopoDS::Face(sh);
Py::List ret;
std::map<int, int> resultSet = getFemMeshPtr()->getccxVolumesByFace(fc);
for (std::map<int, int>::const_iterator it = resultSet.begin();it!=resultSet.end();++it) {
Py::List vol_face;
vol_face.append(Py::Int(it->first));
vol_face.append(Py::Int(it->second));
ret.append(vol_face);
}
Py::Tuple vol_face(2);
vol_face.setItem(0, Py::Int(it->first));
vol_face.setItem(1, Py::Int(it->second));
ret.append(vol_face);
}
return Py::new_reference_to(ret);
}
catch (Standard_Failure) {
Handle_Standard_Failure e = Standard_Failure::Caught();