#include CORBA_SERVER_HEADER(SMESH_Mesh)
#include CORBA_SERVER_HEADER(SMESH_Hypothesis)
#include CORBA_CLIENT_HEADER(GEOM_Gen)
-#include CORBA_CLIENT_HEADER(GEOM_Shape)
#include CORBA_CLIENT_HEADER(SALOMEDS)
#include CORBA_CLIENT_HEADER(SALOMEDS_Attributes)
-
-class SMESH_Mesh_i;
-
-#include "SMESH_HypothesisFactory_i.hxx"
#include "SMESH_Mesh_i.hxx"
+#include "SMESH_Hypothesis_i.hxx"
#include "SALOME_Component_i.hxx"
#include "SALOME_NamingService.hxx"
#include "SMESH_Gen.hxx"
-#include "SMESH_topo.hxx"
#include "GEOM_Client.hxx"
-#include <HDFOI.hxx>
+#include <TCollection_AsciiString.hxx>
+#include <Resource_DataMapOfAsciiStringAsciiString.hxx>
+#include <TColStd_HSequenceOfAsciiString.hxx>
#include <map>
+#include <sstream>
+
+class SMESH_Mesh_i;
+class SALOME_LifeCycleCORBA;
-typedef struct studyContext_iStruct
+// ===========================================================
+// Study context - stores study-connected objects references
+// ==========================================================
+class StudyContext
{
- map<int,SMESH_Mesh_i*> mapMesh_i;
-} StudyContext_iStruct;
+public:
+ // constructor
+ StudyContext() {}
+ // destructor
+ ~StudyContext()
+ {
+ mapIdToIOR.clear();
+ mapIdToId.clear();
+ }
+ // register object in the internal map and return its id
+ int addObject( string theIOR )
+ {
+ int nextId = getNextId();
+ mapIdToIOR[ nextId ] = theIOR;
+ return nextId;
+ }
+ // find the object id in the internal map by the IOR
+ int findId( string theIOR )
+ {
+ map<int, string>::iterator imap;
+ for ( imap = mapIdToIOR.begin(); imap != mapIdToIOR.end(); ++imap ) {
+ if ( imap->second == theIOR )
+ return imap->first;
+ }
+ return 0;
+ }
+ // get object's IOR by id
+ string getIORbyId( const int theId )
+ {
+ if ( mapIdToIOR.find( theId ) != mapIdToIOR.end() )
+ return mapIdToIOR[ theId ];
+ return string( "" );
+ }
+ // get object's IOR by old id
+ string getIORbyOldId( const int theOldId )
+ {
+ if ( mapIdToId.find( theOldId ) != mapIdToId.end() )
+ return getIORbyId( mapIdToId[ theOldId ] );
+ return string( "" );
+ }
+ // maps old object id to the new one (used when restoring data)
+ void mapOldToNew( const int oldId, const int newId ) {
+ mapIdToId[ oldId ] = newId;
+ }
+
+private:
+ // get next free object identifier
+ int getNextId()
+ {
+ int id = 1;
+ while( mapIdToIOR.find( id ) != mapIdToIOR.end() )
+ id++;
+ return id;
+ }
+
+ map<int, string> mapIdToIOR; // persistent-to-transient map
+ map<int, int> mapIdToId; // used to translate object from persistent to transient form
+};
+// ===========================================================
+// SMESH module's engine
+// ==========================================================
class SMESH_Gen_i:
- public POA_SMESH::SMESH_Gen,
- public Engines_Component_i
+ public virtual POA_SMESH::SMESH_Gen,
+ public virtual Engines_Component_i
{
public:
-
+ // Get last created instance of the class
+ static SMESH_Gen_i* GetSMESHGen() { return mySMESHGen;}
+ // Get ORB object
+ static CORBA::ORB_var GetORB() { return myOrb;}
+ // Get SMESH module's POA object
+ static PortableServer::POA_var GetPOA() { return myPoa;}
+ // Get Naming Service object
+ static SALOME_NamingService* GetNS();
+ // Get SALOME_LifeCycleCORBA object
+ static SALOME_LifeCycleCORBA* GetLCC();
+ // Retrieve and get GEOM engine reference
+ static GEOM::GEOM_Gen_ptr GetGeomEngine();
+ // Get object of the CORBA reference
+ static PortableServer::ServantBase_var GetServant( CORBA::Object_ptr theObject );
+ // Get CORBA object corresponding to the SALOMEDS::SObject
+ static CORBA::Object_var SObjectToObject( SALOMEDS::SObject_ptr theSObject );
+ // Get the SALOMEDS::SObject corresponding to a CORBA object
+ static SALOMEDS::SObject_ptr ObjectToSObject(SALOMEDS::Study_ptr theStudy,
+ CORBA::Object_ptr theObject);
+ // Get GEOM Object correspoding to TopoDS_Shape
+ GEOM::GEOM_Object_ptr ShapeToGeomObject (const TopoDS_Shape& theShape );
+ // Get TopoDS_Shape correspoding to GEOM_Object
+ TopoDS_Shape GeomObjectToShape(GEOM::GEOM_Object_ptr theGeomObject);
+
+ // Default constructor
SMESH_Gen_i();
- SMESH_Gen_i(CORBA::ORB_ptr orb,
- PortableServer::POA_ptr poa,
- PortableServer::ObjectId * contId,
- const char *instanceName,
- const char *interfaceName);
+ // Standard constructor
+ SMESH_Gen_i( CORBA::ORB_ptr orb,
+ PortableServer::POA_ptr poa,
+ PortableServer::ObjectId* contId,
+ const char* instanceName,
+ const char* interfaceName );
+ // Destructor
virtual ~SMESH_Gen_i();
- SMESH::SMESH_Hypothesis_ptr CreateHypothesis(const char* anHyp,
- CORBA::Long studyId)
- throw (SALOME::SALOME_Exception);
+ // *****************************************
+ // Interface methods
+ // *****************************************
+
+ // Set current study
+ void SetCurrentStudy( SALOMEDS::Study_ptr theStudy );
+ // Get current study
+ SALOMEDS::Study_ptr GetCurrentStudy();
+
+ // Create hypothesis/algorothm of given type
+ SMESH::SMESH_Hypothesis_ptr CreateHypothesis (const char* theHypType,
+ const char* theLibName)
+ throw ( SALOME::SALOME_Exception );
+
+ // Create empty mesh on a shape
+ SMESH::SMESH_Mesh_ptr CreateMesh( GEOM::GEOM_Object_ptr theShapeObject )
+ throw ( SALOME::SALOME_Exception );
+
+ // Create mesh(es) and import data from UNV file
+ SMESH::SMESH_Mesh_ptr CreateMeshesFromUNV( const char* theFileName )
+ throw ( SALOME::SALOME_Exception );
+
+ // Create mesh(es) and import data from MED file
+ SMESH::mesh_array* CreateMeshesFromMED( const char* theFileName,
+ SMESH::DriverMED_ReadStatus& theStatus )
+ throw ( SALOME::SALOME_Exception );
+
+ // Create mesh(es) and import data from STL file
+ SMESH::SMESH_Mesh_ptr CreateMeshesFromSTL( const char* theFileName )
+ throw ( SALOME::SALOME_Exception );
+
+ // Compute mesh on a shape
+ CORBA::Boolean Compute( SMESH::SMESH_Mesh_ptr theMesh,
+ GEOM::GEOM_Object_ptr theShapeObject )
+ throw ( SALOME::SALOME_Exception );
+
+ // Returns true if mesh contains enough data to be computed
+ CORBA::Boolean IsReadyToCompute( SMESH::SMESH_Mesh_ptr theMesh,
+ GEOM::GEOM_Object_ptr theShapeObject )
+ throw ( SALOME::SALOME_Exception );
+
+ // Returns errors of hypotheses definintion
+ SMESH::algo_error_array* GetAlgoState( SMESH::SMESH_Mesh_ptr theMesh,
+ GEOM::GEOM_Object_ptr theSubObject )
+ throw ( SALOME::SALOME_Exception );
+
+ // Get sub-shapes unique ID's list
+ SMESH::long_array* GetSubShapesId( GEOM::GEOM_Object_ptr theMainShapeObject,
+ const SMESH::object_array& theListOfSubShape )
+ throw ( SALOME::SALOME_Exception );
+
+
+ // ****************************************************
+ // Interface inherited methods (from SALOMEDS::Driver)
+ // ****************************************************
+
+ // Save SMESH data
+ SALOMEDS::TMPFile* Save( SALOMEDS::SComponent_ptr theComponent,
+ const char* theURL,
+ bool isMultiFile );
+ // Load SMESH data
+ bool Load( SALOMEDS::SComponent_ptr theComponent,
+ const SALOMEDS::TMPFile& theStream,
+ const char* theURL,
+ bool isMultiFile );
+ // Save SMESH data in ASCII format
+ SALOMEDS::TMPFile* SaveASCII( SALOMEDS::SComponent_ptr theComponent,
+ const char* theURL,
+ bool isMultiFile );
+ // Load SMESH data in ASCII format
+ bool LoadASCII( SALOMEDS::SComponent_ptr theComponent,
+ const SALOMEDS::TMPFile& theStream,
+ const char* theURL,
+ bool isMultiFile );
+
+ // Create filter manager
+ SMESH::FilterManager_ptr CreateFilterManager();
+
+ // Return a pattern mesher
+ SMESH::SMESH_Pattern_ptr GetPattern();
+
+ // Clears study-connected data when it is closed
+ void Close( SALOMEDS::SComponent_ptr theComponent );
- SMESH::SMESH_Mesh_ptr Init(GEOM::GEOM_Gen_ptr geomEngine,
- CORBA::Long studyId,
- GEOM::GEOM_Shape_ptr aShape)
- throw (SALOME::SALOME_Exception);
-
- CORBA::Boolean Compute(SMESH::SMESH_Mesh_ptr aMesh,
- GEOM::GEOM_Shape_ptr aShape)
- throw (SALOME::SALOME_Exception);
-
- CORBA::Boolean IsReadyToCompute(SMESH::SMESH_Mesh_ptr aMesh,
- GEOM::GEOM_Shape_ptr aShape)
- throw (SALOME::SALOME_Exception);
-
- SMESH::long_array* GetSubShapesId(GEOM::GEOM_Gen_ptr geomEngine,
- CORBA::Long studyId,
- GEOM::GEOM_Shape_ptr mainShape,
- const SMESH::shape_array& listOfSubShape)
- throw (SALOME::SALOME_Exception);
-
- SMESH::SMESH_Mesh_ptr Import(CORBA::Long studyId, const char* fileName,
- const char* fileType);
-
- // inherited methods from SALOMEDS::Driver
-
- SALOMEDS::TMPFile* Save(SALOMEDS::SComponent_ptr theComponent,
- const char* theURL,
- bool isMultiFile);
- bool Load(SALOMEDS::SComponent_ptr theComponent,
- const SALOMEDS::TMPFile& theStream,
- const char* theURL,
- bool isMultiFile);
-
- SALOMEDS::TMPFile* SaveASCII(SALOMEDS::SComponent_ptr theComponent,
- const char* theURL,
- bool isMultiFile);
- bool LoadASCII(SALOMEDS::SComponent_ptr theComponent,
- const SALOMEDS::TMPFile& theStream,
- const char* theURL,
- bool isMultiFile);
-
- void Close(SALOMEDS::SComponent_ptr theComponent);
+ // Get component data type
char* ComponentDataType();
- char* IORToLocalPersistentID(SALOMEDS::SObject_ptr theSObject,
- const char* IORString,
- CORBA::Boolean isMultiFile,
- CORBA::Boolean isASCII);
- char* LocalPersistentIDToIOR(SALOMEDS::SObject_ptr theSObject,
- const char* aLocalPersistentID,
- CORBA::Boolean isMultiFile,
- CORBA::Boolean isASCII);
-
- bool CanPublishInStudy(CORBA::Object_ptr theIOR) { return false; }
- SALOMEDS::SObject_ptr PublishInStudy(SALOMEDS::Study_ptr theStudy,
- SALOMEDS::SObject_ptr theSObject,
- CORBA::Object_ptr theObject,
- const char* theName) throw (SALOME::SALOME_Exception) {
- SALOMEDS::SObject_var aResultSO;
- return aResultSO._retn();
- }
+ // Transform data from transient form to persistent
+ char* IORToLocalPersistentID( SALOMEDS::SObject_ptr theSObject,
+ const char* IORString,
+ CORBA::Boolean isMultiFile,
+ CORBA::Boolean isASCII );
+ // Transform data from persistent form to transient
+ char* LocalPersistentIDToIOR( SALOMEDS::SObject_ptr theSObject,
+ const char* aLocalPersistentID,
+ CORBA::Boolean isMultiFile,
+ CORBA::Boolean isASCII );
- CORBA::Boolean CanCopy(SALOMEDS::SObject_ptr theObject) {return false;}
- SALOMEDS::TMPFile* CopyFrom(SALOMEDS::SObject_ptr theObject, CORBA::Long& theObjectID) {return false;}
- CORBA::Boolean CanPaste(const char* theComponentName, CORBA::Long theObjectID) {return false;}
- SALOMEDS::SObject_ptr PasteInto(const SALOMEDS::TMPFile& theStream,
- CORBA::Long theObjectID,
- SALOMEDS::SObject_ptr theObject) {
+ // Returns true if object can be published in the study
+ bool CanPublishInStudy( CORBA::Object_ptr theIOR );
+ // Publish object in the study
+ SALOMEDS::SObject_ptr PublishInStudy( SALOMEDS::Study_ptr theStudy,
+ SALOMEDS::SObject_ptr theSObject,
+ CORBA::Object_ptr theObject,
+ const char* theName )
+ throw ( SALOME::SALOME_Exception );
+
+ // Copy-paste methods - returns true if object can be copied to the clipboard
+ CORBA::Boolean CanCopy( SALOMEDS::SObject_ptr theObject ) { return false; }
+ // Copy-paste methods - copy object to the clipboard
+ SALOMEDS::TMPFile* CopyFrom( SALOMEDS::SObject_ptr theObject, CORBA::Long& theObjectID ) { return false; }
+ // Copy-paste methods - returns true if object can be pasted from the clipboard
+ CORBA::Boolean CanPaste( const char* theComponentName, CORBA::Long theObjectID ) { return false; }
+ // Copy-paste methods - paste object from the clipboard
+ SALOMEDS::SObject_ptr PasteInto( const SALOMEDS::TMPFile& theStream,
+ CORBA::Long theObjectID,
+ SALOMEDS::SObject_ptr theObject ) {
SALOMEDS::SObject_var aResultSO;
return aResultSO._retn();
}
- GEOM_Client* _ShapeReader;
-protected:
- SMESH_topo* ExploreMainShape(GEOM::GEOM_Gen_ptr geomEngine,
- CORBA::Long studyId,
- GEOM::GEOM_Shape_ptr aShape);
+ // ============
+ // Dump python
+ // ============
+
+ virtual Engines::TMPFile* DumpPython(CORBA::Object_ptr theStudy,
+ CORBA::Boolean isPublished,
+ CORBA::Boolean& isValidScript);
+
+ void AddToPythonScript (int theStudyID, const TCollection_AsciiString& theString);
+
+ void RemoveLastFromPythonScript (int theStudyID);
+
+ static void AddToCurrentPyScript (const TCollection_AsciiString& theString);
+
+ void SavePython (SALOMEDS::Study_ptr theStudy);
+
+ TCollection_AsciiString DumpPython_impl (int theStudyID,
+ Resource_DataMapOfAsciiStringAsciiString& theObjectNames,
+ Resource_DataMapOfAsciiStringAsciiString& theNames,
+ bool isPublished,
+ bool& aValidScript,
+ const TCollection_AsciiString& theSavedTrace);
+
+ TCollection_AsciiString GetNewPythonLines (int theStudyID);
+
+ void CleanPythonTrace (int theStudyID);
+
+ // Dump python comfort methods
+
+ static TCollection_AsciiString& AddObject(TCollection_AsciiString& theStr,
+ CORBA::Object_ptr theObject);
+ // add object to script string
+
+ template <class _array>
+ static TCollection_AsciiString& AddArray(TCollection_AsciiString& theStr,
+ const _array & array)
+ // put array contents into theStr like this: "[ 1, 2, 5 ]"
+ {
+ ostringstream sout; // can convert long int, and TCollection_AsciiString cant
+ sout << "[ ";
+ for (int i = 1; i <= array.length(); i++) {
+ sout << array[i-1];
+ if ( i < array.length() )
+ sout << ", ";
+ }
+ sout << " ]";
+ theStr += (char*) sout.str().c_str();
+ return theStr;
+ }
+
+ // *****************************************
+ // Internal methods
+ // *****************************************
+public:
+ // Get shape reader
+ GEOM_Client* GetShapeReader();
+
+ // Tags definition
+ static long GetHypothesisRootTag();
+ static long GetAlgorithmsRootTag();
+ static long GetRefOnShapeTag();
+ static long GetRefOnAppliedHypothesisTag();
+ static long GetRefOnAppliedAlgorithmsTag();
+ static long GetSubMeshOnVertexTag();
+ static long GetSubMeshOnEdgeTag();
+ static long GetSubMeshOnFaceTag();
+ static long GetSubMeshOnSolidTag();
+ static long GetSubMeshOnCompoundTag();
+ static long GetSubMeshOnWireTag();
+ static long GetSubMeshOnShellTag();
+ static long GetNodeGroupsTag();
+ static long GetEdgeGroupsTag();
+ static long GetFaceGroupsTag();
+ static long GetVolumeGroupsTag();
+
+ // publishing methods
+ SALOMEDS::SComponent_ptr PublishComponent(SALOMEDS::Study_ptr theStudy);
+ SALOMEDS::SObject_ptr PublishMesh (SALOMEDS::Study_ptr theStudy,
+ SMESH::SMESH_Mesh_ptr theMesh,
+ const char* theName = 0);
+ SALOMEDS::SObject_ptr PublishHypothesis (SALOMEDS::Study_ptr theStudy,
+ SMESH::SMESH_Hypothesis_ptr theHyp,
+ const char* theName = 0);
+ SALOMEDS::SObject_ptr PublishSubMesh (SALOMEDS::Study_ptr theStudy,
+ SMESH::SMESH_Mesh_ptr theMesh,
+ SMESH::SMESH_subMesh_ptr theSubMesh,
+ GEOM::GEOM_Object_ptr theShapeObject,
+ const char* theName = 0);
+ SALOMEDS::SObject_ptr PublishGroup (SALOMEDS::Study_ptr theStudy,
+ SMESH::SMESH_Mesh_ptr theMesh,
+ SMESH::SMESH_GroupBase_ptr theGroup,
+ GEOM::GEOM_Object_ptr theShapeObject,
+ const char* theName = 0);
+ bool AddHypothesisToShape(SALOMEDS::Study_ptr theStudy,
+ SMESH::SMESH_Mesh_ptr theMesh,
+ GEOM::GEOM_Object_ptr theShapeObject,
+ SMESH::SMESH_Hypothesis_ptr theHyp);
+ bool RemoveHypothesisFromShape(SALOMEDS::Study_ptr theStudy,
+ SMESH::SMESH_Mesh_ptr theMesh,
+ GEOM::GEOM_Object_ptr theShapeObject,
+ SMESH::SMESH_Hypothesis_ptr theHyp);
+ SALOMEDS::SObject_ptr GetMeshOrSubmeshByShape (SALOMEDS::Study_ptr theStudy,
+ SMESH::SMESH_Mesh_ptr theMesh,
+ GEOM::GEOM_Object_ptr theShape);
+ static void SetName(SALOMEDS::SObject_ptr theSObject,
+ const char* theName,
+ const char* theDefaultName = 0);
+
+ // Get study context
+ StudyContext* GetCurrentStudyContext();
+
+ // Register an object in a StudyContext; return object id
+ int RegisterObject(CORBA::Object_ptr theObject);
+
+ // Get current study ID
+ int GetCurrentStudyID()
+ { return myCurrentStudy->_is_nil() ? -1 : myCurrentStudy->StudyId(); }
+
+private:
+ // Create hypothesis of given type
+ SMESH::SMESH_Hypothesis_ptr createHypothesis( const char* theHypName,
+ const char* theLibName)
+ throw ( SALOME::SALOME_Exception );
+
+ // Create empty mesh on shape
+ SMESH::SMESH_Mesh_ptr createMesh()
+ throw ( SALOME::SALOME_Exception );
+
+ static void loadGeomData( SALOMEDS::SComponent_ptr theCompRoot );
private:
- void loadMesh(char * name, HDFfile * hdf_file, char * meshFile,
- SALOMEDS::Study_var study);
- void loadHypothesis(char * name, HDFfile * hdf_file, char * hypothesisFile,
- int studyId);
- void loadAlgorithms(char * name, HDFfile * hdf_file, char * algorithmsFile,
- int studyId);
- void loadAppliedHypothesis(HDFgroup * hdfGroupMeshId, bool _found,
- SALOMEDS::Study_var Study, SMESH::SMESH_Mesh_var myNewMesh,
- GEOM::GEOM_Shape_var aShape);
- GEOM::GEOM_Gen_var getGeomEngine();
- GEOM::GEOM_Shape_var getShape(SALOMEDS::Study_var Study, char * refFromFile);
-
-void loadAppliedAlgorithms(HDFgroup * hdfGroupMeshId,
- bool _found, SALOMEDS::Study_var Study, SMESH::SMESH_Mesh_var myNewMesh,
- GEOM::GEOM_Shape_var aShape);
-void loadSubMeshes(HDFgroup * hdfGroupMeshId, char * msgname,
- SALOMEDS::Study_var Study, SMESH::SMESH_Mesh_var myNewMesh);
-
- SMESH_HypothesisFactory_i _hypothesisFactory_i;
- ::SMESH_Gen _impl; // no namespace here
- int _localId; // unique Id of created objects, within SMESH_Gen_i entity
-
- map<int, StudyContext_iStruct*> _mapStudyContext_i;
- map <string, string> _SMESHCorbaObj;
+
+ static CORBA::ORB_var myOrb; // ORB reference
+ static PortableServer::POA_var myPoa; // POA reference
+ static SALOME_NamingService* myNS; // Naming Service
+ static SALOME_LifeCycleCORBA* myLCC; // Life Cycle CORBA
+ static SMESH_Gen_i* mySMESHGen; // Point to last created instance of the class
+ ::SMESH_Gen myGen; // SMESH_Gen local implementation
+
+ // hypotheses managing
+ map<string, GenericHypothesisCreator_i*> myHypCreatorMap;
+
+ map<int, StudyContext*> myStudyContextMap; // Map of study context objects
+
+ GEOM_Client* myShapeReader; // Shape reader
+ SALOMEDS::Study_var myCurrentStudy; // Current study
+
+ // Dump Python: trace of API methods calls
+ std::map < int, Handle(TColStd_HSequenceOfAsciiString) > myPythonScripts;
};
+
+namespace SMESH
+{
+ template<class T>
+ T
+ DownCast(CORBA::Object_ptr theArg)
+ {
+ return dynamic_cast<T>(SMESH_Gen_i::GetServant(theArg).in());
+ }
+}
+
+
#endif