smesh/src/StdMeshers/StdMeshers_ImportSource.cxx
2013-04-01 13:05:47 +00:00

491 lines
16 KiB
C++

// Copyright (C) 2007-2013 CEA/DEN, EDF R&D, 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.
//
// 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 StdMeshers_ImportSource1D : implementaion of SMESH idl descriptions
// File : StdMeshers_ImportSource1D.cxx
// Module : SMESH
//
#include "StdMeshers_ImportSource.hxx"
#include "SMESHDS_GroupOnGeom.hxx"
#include "SMESHDS_Mesh.hxx"
#include "SMESH_Algo.hxx"
#include "SMESH_Gen.hxx"
#include "SMESH_Group.hxx"
#include "SMESH_Mesh.hxx"
#include "SMESH_subMeshEventListener.hxx"
#include "utilities.h"
#include <Standard_ErrorHandler.hxx>
#include <boost/shared_ptr.hpp>
using namespace std;
//=============================================================================
/*!
* Creates StdMeshers_ImportSource1D
*/
//=============================================================================
StdMeshers_ImportSource1D::StdMeshers_ImportSource1D(int hypId,
int studyId,
SMESH_Gen * gen)
:SMESH_Hypothesis(hypId, studyId, gen),
_toCopyMesh(false),
_toCopyGroups(false)
{
_name = "ImportSource1D";
_param_algo_dim = 1; // is used by StdMeshers_Import_1D;
}
//=============================================================================
/*!
* Creates StdMeshers_ImportSource2D
*/
//=============================================================================
StdMeshers_ImportSource2D::StdMeshers_ImportSource2D(int hypId,
int studyId,
SMESH_Gen * gen)
:StdMeshers_ImportSource1D(hypId, studyId, gen)
{
_name = "ImportSource2D";
_param_algo_dim = 2; // is used by StdMeshers_Import_2D;
}
//=============================================================================
/*!
*
*/
//=============================================================================
StdMeshers_ImportSource1D::~StdMeshers_ImportSource1D()
{
}
//=============================================================================
/*!
* Sets groups to import elements from
*/
//=============================================================================
void StdMeshers_ImportSource1D::SetGroups(const std::vector<SMESH_Group*>& groups)
{
if (_groups != groups)
{
_groups = groups;
NotifySubMeshesHypothesisModification();
}
}
void StdMeshers_ImportSource1D::SetCopySourceMesh(bool toCopyMesh, bool toCopyGroups)
{
if ( !toCopyMesh ) toCopyGroups = false;
if ( _toCopyMesh != toCopyMesh || _toCopyGroups != toCopyGroups )
{
_toCopyMesh = toCopyMesh; _toCopyGroups = toCopyGroups;
NotifySubMeshesHypothesisModification();
}
}
void StdMeshers_ImportSource1D::GetCopySourceMesh(bool& toCopyMesh, bool& toCopyGroups) const
{
toCopyMesh = _toCopyMesh; toCopyGroups = _toCopyGroups;
}
namespace
{
//================================================================================
/*!
* \brief Return only alive groups
*/
//================================================================================
vector<SMESH_Group*> getValidGroups(const vector<SMESH_Group*>& groups,
StudyContextStruct* studyContext)
{
vector<SMESH_Group*> okGroups;
for ( int i = 0; i < groups.size(); ++i )
{
try
{
// we expect SIGSEGV on a dead group
OCC_CATCH_SIGNALS;
SMESH_Group* okGroup = 0;
map<int, SMESH_Mesh*>::iterator itm = itm = studyContext->mapMesh.begin();
for ( ; !okGroup && itm != studyContext->mapMesh.end(); itm++)
{
SMESH_Mesh::GroupIteratorPtr gIt = itm->second->GetGroups();
while ( gIt->more() && !okGroup )
if ( gIt->next() == groups[i] )
okGroup = groups[i];
}
if ( okGroup )
okGroups.push_back( okGroup );
}
catch(...)
{
}
}
return okGroups;
}
//================================================================================
/*!
* \brief Pack meshes into a pair of ints
*/
//================================================================================
pair<int, int> getResMapKey(const SMESHDS_Mesh& srcMesh, const SMESHDS_Mesh& tgtMesh)
{
return make_pair( srcMesh.GetPersistentId() , tgtMesh.GetPersistentId() );
}
//================================================================================
/*!
* \brief Return a target mesh by a pair of ints
*/
//================================================================================
SMESH_Mesh* getTgtMeshByKey( const pair<int, int> & resMapKey,
StudyContextStruct* studyContext)
{
int tgtID = resMapKey.second;
SMESH_Mesh* tgtMesh = 0;
map<int, SMESH_Mesh*>::iterator itm = itm = studyContext->mapMesh.begin();
for ( ; !tgtMesh && itm != studyContext->mapMesh.end(); itm++)
{
tgtMesh = (*itm).second;
if ( tgtMesh->GetMeshDS()->GetPersistentId() != tgtID )
tgtMesh = 0;
}
return tgtMesh;
}
//================================================================================
/*!
* \brief Return a target mesh by a pair of ints
*/
//================================================================================
int getSrcMeshID( const pair<int, int> & resMapKey )
{
return resMapKey.first;
}
}
//=============================================================================
/*!
* Returns groups to import elements from
*/
//=============================================================================
const std::vector<SMESH_Group*>& StdMeshers_ImportSource1D::GetGroups() const
{
// filter off deleted groups
vector<SMESH_Group*> okGroups = getValidGroups( _groups,
_gen->GetStudyContext(_studyId) );
if ( okGroups.size() != _groups.size() )
((StdMeshers_ImportSource1D*)this)->_groups = okGroups;
return _groups;
}
//================================================================================
/*!
* \brief Return source meshes
*/
//================================================================================
std::vector<SMESH_Mesh*> StdMeshers_ImportSource1D::GetSourceMeshes() const
{
// GetPersistentId()'s of meshes
set<int> meshIDs;
const vector<SMESH_Group*>& groups = GetGroups();
if ( !groups.empty() )
{
for ( unsigned i = 0; i < groups.size(); ++i )
{
const SMESHDS_GroupBase* gDS = groups[i]->GetGroupDS();
int id = gDS->GetMesh()->GetPersistentId();
meshIDs.insert( id );
}
}
else
{
if ( _resultGroups.empty() )
((StdMeshers_ImportSource1D*)this)->RestoreGroups(_groups);
TResGroupMap::const_iterator key_groups = _resultGroups.begin();
for ( ; key_groups != _resultGroups.end(); ++key_groups )
meshIDs.insert( getSrcMeshID( key_groups->first ));
}
// Find corresponding meshes
vector<SMESH_Mesh*> meshes;
if ( !meshIDs.empty() )
{
StudyContextStruct* studyContext = _gen->GetStudyContext(_studyId);
for ( set<int>::iterator id = meshIDs.begin(); id != meshIDs.end(); ++id )
{
map<int, SMESH_Mesh*>::iterator itm = itm = studyContext->mapMesh.begin();
for ( ; itm != studyContext->mapMesh.end(); itm++)
{
SMESH_Mesh* mesh = (*itm).second;
if ( mesh->GetMeshDS()->GetPersistentId() == *id )
{
meshes.push_back( mesh );
break;
}
}
}
}
return meshes;
}
//================================================================================
/*!
* \brief Return submeshes whose events affect the target mesh
*/
//================================================================================
std::vector<SMESH_subMesh*>
StdMeshers_ImportSource1D::GetSourceSubMeshes(const SMESH_Mesh* srcMesh) const
{
if ( !srcMesh->HasShapeToMesh() )
{
SMESH_Mesh* srcM = const_cast< SMESH_Mesh* >( srcMesh );
return vector<SMESH_subMesh*>(1, srcM->GetSubMesh( srcM->GetShapeToMesh()));
}
set<int> shapeIDs;
const vector<SMESH_Group*>& groups = GetGroups();
const SMESHDS_Mesh * srcMeshDS = srcMesh->GetMeshDS();
for ( size_t i = 0; i < groups.size(); ++i )
{
SMESHDS_GroupBase * grDS = groups[i]->GetGroupDS();
if ( grDS->GetMesh() != srcMeshDS )
continue;
if ( SMESHDS_GroupOnGeom* gog = dynamic_cast<SMESHDS_GroupOnGeom*>( grDS ))
{
shapeIDs.insert( srcMeshDS->ShapeToIndex( gog->GetShape() ));
}
else
{
SMDS_ElemIteratorPtr elIt = grDS->GetElements();
while ( elIt->more() )
shapeIDs.insert( elIt->next()->getshapeId() );
}
}
if ( !shapeIDs.empty() && *shapeIDs.begin() < 1 )
{
shapeIDs.erase( shapeIDs.begin() );
shapeIDs.insert( 1 );
}
vector<SMESH_subMesh*> smVec( shapeIDs.size());
set<int>::iterator sID = shapeIDs.begin();
for ( int i = 0; sID != shapeIDs.end(); ++sID, ++i )
smVec[i] = srcMesh->GetSubMeshContaining( *sID );
return smVec;
}
//=============================================================================
/*!
* Save _toCopyMesh and _toCopyGroups to a stream
*/
//=============================================================================
ostream & StdMeshers_ImportSource1D::SaveTo(ostream & save)
{
resultGroupsToIntVec();
save << " " << _toCopyMesh << " " << _toCopyGroups;
save << " " << _resultGroupsStorage.size();
for ( unsigned i = 0; i < _resultGroupsStorage.size(); ++i )
save << " " << _resultGroupsStorage[i];
return save;
}
//=============================================================================
/*!
* Load _toCopyMesh and _toCopyGroups from a stream
*/
//=============================================================================
istream & StdMeshers_ImportSource1D::LoadFrom(istream & load)
{
load >> _toCopyMesh >> _toCopyGroups;
_resultGroupsStorage.clear();
int val;
if ( load >> val )
{
_resultGroupsStorage.reserve(val);
while ( _resultGroupsStorage.size() < _resultGroupsStorage.capacity() && load >> val )
_resultGroupsStorage.push_back( val );
}
return load;
}
//================================================================================
/*!
* \brief Convert result groups into _resultGroupsStorage
*/
//================================================================================
void StdMeshers_ImportSource1D::resultGroupsToIntVec()
{
_resultGroupsStorage.clear();
// store result groups
TResGroupMap::iterator key2groups = _resultGroups.begin();
for ( ; key2groups != _resultGroups.end(); ++key2groups )
{
const pair<int, int>& key = key2groups->first;
const vector<SMESH_Group*>& groups = key2groups->second;
// mesh ids, nb groups
_resultGroupsStorage.push_back( key.first );
_resultGroupsStorage.push_back( key.second );
_resultGroupsStorage.push_back( groups.size() );
for ( unsigned i = 0; i < groups.size(); ++i )
{
// store group names as sequence of ints each standing for a char
// of a name; that is to avoid pb with names containing white spaces
string name = groups[i]->GetGroupDS()->GetStoreName();
_resultGroupsStorage.push_back( name.size() );
for ( unsigned j = 0; j < name.size(); ++j )
_resultGroupsStorage.push_back( name[j] );
}
}
}
//================================================================================
/*!
* \brief Restore source groups and result groups by _resultGroupsStorage
*/
//================================================================================
void StdMeshers_ImportSource1D::RestoreGroups(const std::vector<SMESH_Group*>& groups)
{
_groups = groups;
_resultGroups.clear();
int i = 0;
while ( i < _resultGroupsStorage.size() )
{
int key1 = _resultGroupsStorage[i++];
int key2 = _resultGroupsStorage[i++];
pair<int, int> resMapKey( key1, key2 );
SMESH_Mesh* mesh = getTgtMeshByKey( resMapKey, _gen->GetStudyContext(_studyId));
// restore mesh ids at least
_resultGroups.insert( make_pair (resMapKey,vector<SMESH_Group*>() ));
int nbGroups = _resultGroupsStorage[i++];
for ( int j = 0; j < nbGroups; ++j )
{
string::size_type nameSize = _resultGroupsStorage[i++];
string groupName(nameSize, '\0');
for ( unsigned k = 0; k < nameSize; ++k )
groupName[k] = (char) _resultGroupsStorage[i++];
// find a group by name
if ( mesh )
{
SMESH_Group* group = 0;
SMESH_Mesh::GroupIteratorPtr gIt = mesh->GetGroups();
while ( !group && gIt->more() )
{
group = gIt->next();
if ( !group->GetGroupDS() || groupName != group->GetGroupDS()->GetStoreName() )
group = 0;
}
if ( group )
_resultGroups[ resMapKey ].push_back( group );
}
}
}
}
//================================================================================
/*!
* \brief Remember groups imported from other mesh
* \param groups - result groups
* \param srcMesh - source mesh
* \param tgtMesh - destination mesh
*/
//================================================================================
void StdMeshers_ImportSource1D::StoreResultGroups(const std::vector<SMESH_Group*>& groups,
const SMESHDS_Mesh& srcMesh,
const SMESHDS_Mesh& tgtMesh)
{
_resultGroups[ getResMapKey(srcMesh,tgtMesh) ] = groups;
}
//================================================================================
/*!
* \brief Return groups imported from other mesh
* \param srcMesh - source mesh
* \param tgtMesh - destination mesh
* \retval const std::vector<SMESH_Group*>& - groups
*/
//================================================================================
std::vector<SMESH_Group*>*
StdMeshers_ImportSource1D::GetResultGroups(const SMESHDS_Mesh& srcMesh,
const SMESHDS_Mesh& tgtMesh)
{
TResGroupMap::iterator key2groups = _resultGroups.find( getResMapKey(srcMesh,tgtMesh ));
if ( key2groups == _resultGroups.end() )
return 0;
vector<SMESH_Group*> vec = getValidGroups((*key2groups).second,
_gen->GetStudyContext(_studyId) );
if ( vec.size() != key2groups->second.size())
key2groups->second = vec;
return & key2groups->second;
}
//================================================================================
/*!
* \brief Initialize ImportSource value by the mesh built on the geometry
* \param theMesh - the built mesh
* \param theShape - the geometry of interest
* \retval bool - true if parameter values have been successfully defined
*/
//================================================================================
bool StdMeshers_ImportSource1D::SetParametersByMesh(const SMESH_Mesh*, const TopoDS_Shape&)
{
return false;
}
//================================================================================
/*!
* \brief Initialize my parameter values by default parameters.
* \retval bool - true if parameter values have been successfully defined
*/
//================================================================================
bool StdMeshers_ImportSource1D::SetParametersByDefaults(const TDefaults&, const SMESH_Mesh* )
{
return false;
}