-SMDS_Iterator<const SMDS_MeshElement *> * SMDS_VolumeOfNodes::
- elementsIterator(SMDSAbs_ElementType type) const
-{
- class MyIterator:public SMDS_Iterator<const SMDS_MeshElement*>
- {
- const vector<const SMDS_MeshNode*>& mySet;
- int index;
- public:
- MyIterator(const vector<const SMDS_MeshNode*>& s):mySet(s),index(0)
- {}
-
- bool more()
- {
- return index<mySet.size();
- }
-
- const SMDS_MeshElement* next()
- {
- index++;
- return mySet[index-1];
- }
- };
- switch(type)
- {
- case SMDSAbs_Volume:return SMDS_MeshElement::elementsIterator(SMDSAbs_Volume);
- case SMDSAbs_Node:return new MyIterator(myNodes);
- default: MESSAGE("ERROR : Iterator not implemented");
- }
+/*!
+ * \brief Iterator on node of volume
+ */
+class SMDS_VolumeOfNodes_MyIterator:public SMDS_NodeArrayElemIterator
+{
+ public:
+ SMDS_VolumeOfNodes_MyIterator(const SMDS_MeshNode* const* s, int l):
+ SMDS_NodeArrayElemIterator( s, & s[ l ]) {}
+};
+
+/*!
+ * \brief Iterator on faces or edges of volume
+ */
+class _MySubIterator : public SMDS_ElemIterator
+{
+ vector< const SMDS_MeshElement* > myElems;
+ int myIndex;
+public:
+ _MySubIterator(const SMDS_VolumeOfNodes* vol, SMDSAbs_ElementType type):myIndex(0) {
+ SMDS_VolumeTool vTool(vol);
+ if (type == SMDSAbs_Face)
+ vTool.GetAllExistingFaces( myElems );
+ else
+ vTool.GetAllExistingFaces( myElems );
+ }
+ /// Return true if and only if there are other object in this iterator
+ virtual bool more() { return myIndex < myElems.size(); }
+
+ /// Return the current object and step to the next one
+ virtual const SMDS_MeshElement* next() { return myElems[ myIndex++ ]; }
+};
+
+SMDS_ElemIteratorPtr SMDS_VolumeOfNodes::elementsIterator(SMDSAbs_ElementType type) const
+{
+ switch(type)
+ {
+ case SMDSAbs_Volume:
+ return SMDS_MeshElement::elementsIterator(SMDSAbs_Volume);
+ case SMDSAbs_Node:
+ return SMDS_ElemIteratorPtr(new SMDS_VolumeOfNodes_MyIterator(myNodes,myNbNodes));
+ case SMDSAbs_Face:
+ return SMDS_ElemIteratorPtr(new _MySubIterator(this,SMDSAbs_Face));
+ case SMDSAbs_Edge:
+ return SMDS_ElemIteratorPtr(new _MySubIterator(this,SMDSAbs_Edge));
+ default:
+ MESSAGE("ERROR : Iterator not implemented");
+ return SMDS_ElemIteratorPtr((SMDS_ElemIterator*)NULL);
+ }