DriverMED_Family.h

Go to the documentation of this file.

// Copyright (C) 2007-2023  CEA, EDF, OPEN CASCADE
//
// Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 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
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// 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.com
//
 
//  SMESH DriverMED : tool to split groups on families
//  File   : DriverMED_Family.hxx
//  Author : Julia DOROVSKIKH
//  Module : SMESH
//
#ifndef _INCLUDE_DRIVERMED_FAMILY
#define _INCLUDE_DRIVERMED_FAMILY
 
#include "SMESH_DriverMED.hxx"
#include "DriverMED.hxx"
 
#include "SMDS_Mesh.hxx"
#include "SMESHDS_GroupBase.hxx"
#include "SMESHDS_SubMesh.hxx"
#include "MED_Common.hxx"
 
#include <boost/container/flat_set.hpp>
#include <boost/shared_ptr.hpp>
#include <set>
#include <limits>
 
#define REST_NODES_FAMILY 1
#define FIRST_NODE_FAMILY 2
 
#define REST_EDGES_FAMILY   -1
#define REST_FACES_FAMILY   -2
#define REST_VOLUMES_FAMILY -3
#define REST_0DELEM_FAMILY  -4
#define REST_BALL_FAMILY    -5
#define FIRST_ELEM_FAMILY   -6
 
// Not In Group families
#define NIG_EDGES_FAMILY   INT_MAX-1
#define NIG_FACES_FAMILY   INT_MAX-2
#define NIG_VOLS_FAMILY    INT_MAX-3
#define NIG_0DELEM_FAMILY  INT_MAX-4
#define NIG_BALL_FAMILY    INT_MAX-5
#define NIG_GROUP_PREFIX "NOT_IN_GRP"
 
typedef std::list<DriverMED_FamilyPtr                   > DriverMED_FamilyPtrList;
typedef std::map<int,SMESHDS_SubMesh*                   > SMESHDS_SubMeshPtrMap;
typedef std::list<SMESHDS_GroupBase*                    > SMESHDS_GroupBasePtrList;
typedef std::set<const SMDS_MeshElement*,TIDCompare     > ElementsSet;
typedef boost::container::flat_set< SMDSAbs_ElementType > ElemTypeSet;
 
class MESHDRIVERMED_EXPORT DriverMED_Family
{
 public:
 
  DriverMED_Family();
 
 
  static 
  DriverMED_FamilyPtrList
  MakeFamilies (SMESHDS_SubMeshIteratorPtr      theSubMeshes,
                const SMESHDS_GroupBasePtrList& theGroups,
                const bool doGroupOfNodes,
                const bool doGroupOfEdges,
                const bool doGroupOfFaces,
                const bool doGroupOfVolumes,
                const bool doGroupOf0DElems,
                const bool doGroupOfBalls,
                const bool doAllInGroups);
 
  template<class LowLevelWriter>
  MED::PFamilyInfo 
  GetFamilyInfo(const LowLevelWriter& theWrapper, 
                const MED::PMeshInfo& theMeshInfo) const;
 
  const ElementsSet& GetElements () const;
 
  int GetId () const;
 
  void SetId (const int theId);
 
 public:
 
  // Methods for groups reading from MED
 
  void AddElement(const SMDS_MeshElement* theElement);
 
  const MED::TStringSet& GetGroupNames() const;
  void AddGroupName(std::string theGroupName);
 
  void SetType(const SMDSAbs_ElementType theType);
  SMDSAbs_ElementType GetType();
  const ElemTypeSet&  GetTypes() const;
 
  bool MemberOf(std::string theGroupName) const;
 
  int  GetGroupAttributVal() const;
  void SetGroupAttributVal( int theValue);
 
  size_t NbElements( SMDSAbs_ElementType ) const;
 
 private:
  void Init (SMESHDS_GroupBase* group);
 
  static
  DriverMED_FamilyPtrList SplitByType(SMESHDS_SubMesh* theSubMesh,
                                      const int        theId);
 
 
  void Split (DriverMED_FamilyPtr by,
              DriverMED_FamilyPtr common);
 
  bool IsEmpty () const;
 
 
 private:
  int                 myId;
  SMDSAbs_ElementType myType;
  ElementsSet         myElements;
  MED::TStringSet     myGroupNames;
  int                 myGroupAttributVal;
  ElemTypeSet         myTypes; // Issue 0020576
};
 
#include "MED_Factory.hxx"
 
#include <set>
#include <string>
//=============================================================================
//=============================================================================
template<class LowLevelWriter>
MED::PFamilyInfo
DriverMED_Family::GetFamilyInfo(const LowLevelWriter& theWrapper, 
                                const MED::PMeshInfo& theMeshInfo) const
{
  std::ostringstream aStr;
  aStr << "FAM_" << myId;
  std::set<std::string>::const_iterator aGrIter = myGroupNames.begin();
  for(; aGrIter != myGroupNames.end(); aGrIter++){
    aStr << "_" << *aGrIter;
  }
  std::string aValue = aStr.str();
  // PAL19785,0019867 - med forbids whitespace to be the last char in the name
  int maxSize = MED::GetNOMLength();
  int lastCharPos = std::min( maxSize, (int) aValue.size() ) - 1;
  while ( isspace( aValue[ lastCharPos ] ))
    aValue.resize( lastCharPos-- );
 
  MED::PFamilyInfo anInfo;
  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: ";
//  set<string>::iterator aGrIter = myGroupNames.begin();
//  for (; aGrIter != myGroupNames.end(); aGrIter++)
//  {
//    cout << " " << *aGrIter;
//  }
//  cout << endl;
//
//  cout << "Elements: ";
//  set<const SMDS_MeshElement *>::iterator anIter = myElements.begin();
//  for (; anIter != myElements.end(); anIter++)
//  {
//    cout << " " << (*anIter)->GetID();
//  }
//  cout << endl;
 
  return anInfo;
}
 
#endif