// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
//
-// See http://www.salome-platform.org or email : webmaster.salome@opencascade.org
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
//
//
//
// $Header$
#include "DriverMED_Family.h"
+#include "MED_Factory.hxx"
+
#include <sstream>
+using namespace std;
+
+//=============================================================================
+/*!
+ * Default constructor
+ */
+//=============================================================================
+DriverMED_Family
+::DriverMED_Family():
+ myGroupAttributVal(0)
+{}
+
+
+//=============================================================================
+const ElementsSet&
+DriverMED_Family
+::GetElements () const
+{
+ return myElements;
+}
+
+int
+DriverMED_Family
+::GetId () const
+{
+ return myId;
+}
+
+void
+DriverMED_Family
+::SetId (const int theId)
+{
+ myId = theId;
+}
+
+void
+DriverMED_Family
+::AddElement(const SMDS_MeshElement* theElement)
+{
+ myElements.insert(theElement);
+}
+
+void
+DriverMED_Family
+::AddGroupName(std::string theGroupName)
+{
+ myGroupNames.insert(theGroupName);
+}
+
+void
+DriverMED_Family
+::SetType(const SMDSAbs_ElementType theType)
+{
+ myType = theType;
+}
+
+SMDSAbs_ElementType
+DriverMED_Family
+::GetType()
+{
+ return myType;
+}
+
+bool
+DriverMED_Family
+::MemberOf(std::string theGroupName) const
+{
+ return myGroupNames.find(theGroupName) != myGroupNames.end();
+}
+
+const MED::TStringSet&
+DriverMED_Family
+::GetGroupNames () const
+{
+ return myGroupNames;
+}
+
+
+int
+DriverMED_Family
+::GetGroupAttributVal() const
+{
+ return myGroupAttributVal;
+}
+
+void
+DriverMED_Family
+::SetGroupAttributVal( int theValue)
+{
+ myGroupAttributVal = theValue;
+}
+
+bool
+DriverMED_Family
+::IsEmpty () const
+{
+ return myElements.empty();
+}
+
+bool CompareColors( const SALOMEDS::Color& theColor, const SALOMEDS::Color& theRefColor )
+{
+ if( fabs( theColor.R - theRefColor.R ) < 0.01 &&
+ fabs( theColor.G - theRefColor.G ) < 0.01 &&
+ fabs( theColor.B - theRefColor.B ) < 0.01 )
+ return true;
+
+ return false;
+}
+
//=============================================================================
/*!
* Split each group from list <aGroups> on some parts (families)
* Resulting families have no common elements.
*/
//=============================================================================
-list<DriverMED_FamilyPtr> DriverMED_Family::MakeFamilies
- (const map <int, SMESHDS_SubMesh*>& theSubMeshes,
- const list<SMESHDS_Group*>& theGroups,
- const bool doGroupOfNodes,
- const bool doGroupOfEdges,
- const bool doGroupOfFaces,
- const bool doGroupOfVolumes)
+DriverMED_FamilyPtrList
+DriverMED_Family
+::MakeFamilies(const SMESHDS_SubMeshPtrMap& theSubMeshes,
+ const SMESHDS_GroupBasePtrList& theGroups,
+ const bool doGroupOfNodes,
+ const bool doGroupOfEdges,
+ const bool doGroupOfFaces,
+ const bool doGroupOfVolumes)
{
- list<DriverMED_FamilyPtr> aFamilies;
+ DriverMED_FamilyPtrList aFamilies;
string anAllNodesGroupName = "Group_Of_All_Nodes";
string anAllEdgesGroupName = "Group_Of_All_Edges";
int aElemFamId = FIRST_ELEM_FAMILY;
// Process sub-meshes
- map<int, SMESHDS_SubMesh*>::const_iterator aSMIter = theSubMeshes.begin();
+ SMESHDS_SubMeshPtrMap::const_iterator aSMIter = theSubMeshes.begin();
for (; aSMIter != theSubMeshes.end(); aSMIter++)
{
- list<DriverMED_FamilyPtr> aSMFams = SplitByType((*aSMIter).second, (*aSMIter).first);
- list<DriverMED_FamilyPtr>::iterator aSMFamsIter = aSMFams.begin();
+ const int anId = aSMIter->first;
+ SMESHDS_SubMesh* aSubMesh = aSMIter->second;
+ if ( aSubMesh->IsComplexSubmesh() )
+ continue; // submesh containing other submeshs
+ DriverMED_FamilyPtrList aSMFams = SplitByType(aSubMesh,anId);
+ DriverMED_FamilyPtrList::iterator aSMFamsIter = aSMFams.begin();
for (; aSMFamsIter != aSMFams.end(); aSMFamsIter++)
{
DriverMED_FamilyPtr aFam2 = (*aSMFamsIter);
-
- list<DriverMED_FamilyPtr>::iterator aFamsIter = aFamilies.begin();
+ DriverMED_FamilyPtrList::iterator aFamsIter = aFamilies.begin();
while (aFamsIter != aFamilies.end())
{
DriverMED_FamilyPtr aFam1 = *aFamsIter;
- list<DriverMED_FamilyPtr>::iterator aCurrIter = aFamsIter++;
+ DriverMED_FamilyPtrList::iterator aCurrIter = aFamsIter++;
if (aFam1->myType == aFam2->myType)
{
DriverMED_FamilyPtr aCommon (new DriverMED_Family);
{
aFamilies.erase(aCurrIter);
}
- if (aFam2->IsEmpty()) break;
+ if (aFam2->IsEmpty())
+ break;
}
}
// The rest elements of family
}
// Process groups
- list<SMESHDS_Group*>::const_iterator aGroupsIter = theGroups.begin();
- for (; aGroupsIter != theGroups.end(); aGroupsIter++)
+ SMESHDS_GroupBasePtrList::const_iterator aGroupsIter;
+
+ int id = 0;
+ ColorMap aColorMap;
+ for (aGroupsIter = theGroups.begin(); aGroupsIter != theGroups.end(); aGroupsIter++)
+ {
+ Quantity_Color aQColor = (*aGroupsIter)->GetColor();
+ SALOMEDS::Color aColor;
+ aColor.R = aQColor.Red();
+ aColor.G = aQColor.Green();
+ aColor.B = aQColor.Blue();
+
+ bool isFound = false;
+ for (ColorMap::iterator aColorIter = aColorMap.begin(); aColorIter != aColorMap.end(); aColorIter++)
+ {
+ SALOMEDS::Color aRefColor = aColorIter->second;
+ if( CompareColors( aColor, aRefColor ) )
+ {
+ isFound = true;
+ break;
+ }
+ }
+
+ if( !isFound )
+ aColorMap[ id++ ] = aColor;
+ }
+
+ for (aGroupsIter = theGroups.begin(); aGroupsIter != theGroups.end(); aGroupsIter++)
{
DriverMED_FamilyPtr aFam2 (new DriverMED_Family);
- aFam2->Init(*aGroupsIter);
+ aFam2->Init(*aGroupsIter, aColorMap);
- list<DriverMED_FamilyPtr>::iterator aFamsIter = aFamilies.begin();
+ DriverMED_FamilyPtrList::iterator aFamsIter = aFamilies.begin();
while (aFamsIter != aFamilies.end())
{
DriverMED_FamilyPtr aFam1 = *aFamsIter;
- list<DriverMED_FamilyPtr>::iterator aCurrIter = aFamsIter++;
+ DriverMED_FamilyPtrList::iterator aCurrIter = aFamsIter++;
if (aFam1->myType == aFam2->myType)
{
DriverMED_FamilyPtr aCommon (new DriverMED_Family);
aFam1->Split(aFam2, aCommon);
if (!aCommon->IsEmpty())
{
+ aCommon->SetGroupAttributVal(0);
aFamilies.push_back(aCommon);
}
if (aFam1->IsEmpty())
{
aFamilies.erase(aCurrIter);
}
- if (aFam2->IsEmpty()) break;
+ if (aFam2->IsEmpty())
+ break;
}
}
// The rest elements of group
}
}
- list<DriverMED_FamilyPtr>::iterator aFamsIter = aFamilies.begin();
+ DriverMED_FamilyPtrList::iterator aFamsIter = aFamilies.begin();
for (; aFamsIter != aFamilies.end(); aFamsIter++)
{
DriverMED_FamilyPtr aFam = *aFamsIter;
* Create TFamilyInfo for this family
*/
//=============================================================================
-MEDA::PFamilyInfo DriverMED_Family::GetFamilyInfo
- (const MEDA::PMeshInfo& theMeshInfo) const
+MED::PFamilyInfo
+DriverMED_Family::GetFamilyInfo(const MED::PWrapper& theWrapper,
+ const MED::PMeshInfo& theMeshInfo) const
{
- string aValue;
ostringstream aStr;
- aStr << myId;
- aValue = aStr.str();
- MED::TStringVector anAttrDescs (1, ""); // 1 attribute with empty description,
- MED::TIntVector anAttrIds (1, myId); // Id=0,
- MED::TIntVector anAttrVals (1, myId); // Value=0
-
- MEDA::PFamilyInfo anInfo = MEDA::TWrapper::CrFamilyInfo(theMeshInfo,
- aValue,
- myId,
- myGroupNames,
- anAttrDescs,anAttrIds,anAttrVals);
+ aStr << "FAM_" << myId;
+ set<string>::const_iterator aGrIter = myGroupNames.begin();
+ for(; aGrIter != myGroupNames.end(); aGrIter++){
+ aStr << "_" << *aGrIter;
+ }
+
+ MED::PFamilyInfo anInfo;
+ string aValue = aStr.str();
+ if(myId == 0 || myGroupAttributVal == 0){
+ anInfo = theWrapper->CrFamilyInfo(theMeshInfo,
+ aValue,
+ myId,
+ myGroupNames);
+ }else{
+ MED::TStringVector anAttrDescs (1, ""); // 1 attribute with empty description,
+ MED::TIntVector anAttrIds (1, myId); // Id=0,
+ MED::TIntVector anAttrVals (1, myGroupAttributVal);
+ anInfo = theWrapper->CrFamilyInfo(theMeshInfo,
+ aValue,
+ myId,
+ myGroupNames,
+ anAttrDescs,
+ anAttrIds,
+ anAttrVals);
+ }
// cout << endl;
// cout << "Groups: ";
//=============================================================================
/*!
- * Initialize the tool by SMESHDS_Group
+ * Initialize the tool by SMESHDS_GroupBase
*/
//=============================================================================
-void DriverMED_Family::Init (SMESHDS_Group* group)
+void DriverMED_Family::Init (SMESHDS_GroupBase* theGroup, const ColorMap& theColorMap)
{
// Elements
myElements.clear();
- group->InitIterator();
- while (group->More())
+ SMDS_ElemIteratorPtr elemIt = theGroup->GetElements();
+ while (elemIt->more())
{
- myElements.insert(group->Next());
+ myElements.insert(elemIt->next());
}
// Type
- myType = group->GetType();
+ myType = theGroup->GetType();
// Groups list
myGroupNames.clear();
- myGroupNames.insert(string(group->GetStoreName()));
+ myGroupNames.insert(string(theGroup->GetStoreName()));
+
+ myGroupAttributVal = 0;
+
+ ColorMap::const_iterator aColorIter = theColorMap.begin();
+ for (; aColorIter != theColorMap.end(); aColorIter++)
+ {
+ Quantity_Color aGroupQColor = theGroup->GetColor();
+ SALOMEDS::Color aGroupColor;
+ aGroupColor.R = aGroupQColor.Red();
+ aGroupColor.G = aGroupQColor.Green();
+ aGroupColor.B = aGroupQColor.Blue();
+
+ SALOMEDS::Color aColor = aColorIter->second;
+ if( CompareColors( aGroupColor, aColor ) )
+ {
+ myGroupAttributVal = aColorIter->first;
+ break;
+ }
+ }
}
//=============================================================================
* on the basis of the elements type.
*/
//=============================================================================
-list<DriverMED_FamilyPtr> DriverMED_Family::SplitByType (SMESHDS_SubMesh* theSubMesh,
- const int theId)
+DriverMED_FamilyPtrList
+DriverMED_Family
+::SplitByType (SMESHDS_SubMesh* theSubMesh,
+ const int theId)
{
- list<DriverMED_FamilyPtr> aFamilies;
+ DriverMED_FamilyPtrList aFamilies;
DriverMED_FamilyPtr aNodesFamily (new DriverMED_Family);
DriverMED_FamilyPtr anEdgesFamily (new DriverMED_Family);
DriverMED_FamilyPtr aFacesFamily (new DriverMED_Family);
DriverMED_FamilyPtr common)
{
// Elements
- set<const SMDS_MeshElement *>::iterator anIter = by->myElements.begin();
- for (; anIter != by->myElements.end(); anIter++)
+ ElementsSet::iterator anIter = by->myElements.begin();
+ while ( anIter != by->myElements.end())
{
if (myElements.find(*anIter) != myElements.end())
{
common->myElements.insert(*anIter);
myElements.erase(*anIter);
- by->myElements.erase(*anIter);
+ by->myElements.erase(anIter++);
}
+ else
+ anIter++;
}
if (!common->IsEmpty())
{
// Groups list
common->myGroupNames = myGroupNames;
- set<string>::iterator aGrNamesIter = by->myGroupNames.begin();
+ MED::TStringSet::iterator aGrNamesIter = by->myGroupNames.begin();
for (; aGrNamesIter != by->myGroupNames.end(); aGrNamesIter++)
{
common->myGroupNames.insert(*aGrNamesIter);
