#include "SMESH_Gen_i.hxx"
#include "SMESH_Mesh.hxx"
#include "SMESH_Mesh_i.hxx"
+#include "SMDS_MeshFace.hxx"
+#include "SMDS_MeshVolume.hxx"
#include <TopExp_Explorer.hxx>
#include <TopoDS.hxx>
#include <TopoDS_Face.hxx>
#include <sstream>
+#include <set>
+
+//=======================================================================
+//function : dumpErrorCode
+//purpose :
+//=======================================================================
+
+static void addErrorCode(const char* thePyCommand)
+{
+ SMESH_Gen_i::AddToCurrentPyScript("if (isDone != 1):");
+ TCollection_AsciiString str ("\tprint \"");
+ str += (char*) thePyCommand;
+ str += ":\", pattern.GetErrorCode()";
+ SMESH_Gen_i::AddToCurrentPyScript( str );
+}
//=============================================================================
/*!
SMESH::SMESH_Pattern_ptr SMESH_Gen_i::GetPattern()
{
+ // Update Python script
+ SMESH_Gen_i::AddToCurrentPyScript( "pattern = smesh.GetPattern()" );
+
SMESH_Pattern_i* i = new SMESH_Pattern_i( this );
SMESH::SMESH_Pattern_var anObj = i->_this();
return anObj._retn();
{
}
-//=======================================================================
-//function : getShape
-//purpose :
-//=======================================================================
-
-TopoDS_Shape SMESH_Pattern_i::getShape( GEOM::GEOM_Object_ptr & theGeomObject )
-{
- return myGen->GetShapeReader()->GetShape( SMESH_Gen_i::GetGeomEngine(), theGeomObject );
-}
-
//=======================================================================
//function : getMesh
//purpose :
CORBA::Boolean SMESH_Pattern_i::LoadFromFile(const char* theFileContents)
{
+ // Update Python script
+ TCollection_AsciiString str( "isDone = pattern.LoadFromFile(" );
+ str += TCollection_AsciiString( (char*) theFileContents ) + ")";
+ SMESH_Gen_i::AddToCurrentPyScript( str );
+ addErrorCode( "LoadFromFile" );
+
return myPattern.Load( theFileContents );
}
if ( !aMesh )
return false;
- TopoDS_Face aFace = TopoDS::Face( getShape( theFace ));
+ TopoDS_Face aFace = TopoDS::Face( myGen->GeomObjectToShape( theFace ));
if ( aFace.IsNull() )
return false;
+ // Update Python script
+ TCollection_AsciiString str( "isDone = pattern.LoadFromFace( " );
+ SMESH_Gen_i::AddObject( str, theMesh ) += ", ";
+ SMESH_Gen_i::AddObject( str, theFace ) += ", ";
+ str += TCollection_AsciiString( theProject ) + " )";
+ SMESH_Gen_i::AddToCurrentPyScript( str );
+ addErrorCode( "LoadFromFace" );
+
return myPattern.Load( aMesh, aFace, theProject );
}
if ( !aMesh )
return false;
- TopoDS_Shape aShape = getShape( theBlock );
+ TopoDS_Shape aShape = myGen->GeomObjectToShape( theBlock );
if ( aShape.IsNull())
return false;
if ( !exp.More() )
return false;
+ // Update Python script
+ TCollection_AsciiString str( "isDone = pattern.LoadFrom3DBlock( " );
+ SMESH_Gen_i::AddObject( str, theMesh ) += ", ";
+ SMESH_Gen_i::AddObject( str, theBlock ) += " )";
+ SMESH_Gen_i::AddToCurrentPyScript( str );
+ addErrorCode( "LoadFrom3DBlock" );
+
return myPattern.Load( aMesh, TopoDS::Shell( exp.Current() ));
}
SMESH::point_array_var points = new SMESH::point_array;
list<const gp_XYZ *> xyzList;
- TopoDS_Shape F = getShape( theFace );
- TopoDS_Shape V = getShape( theVertexOnKeyPoint1 );
+ TopoDS_Shape F = myGen->GeomObjectToShape( theFace );
+ TopoDS_Shape V = myGen->GeomObjectToShape( theVertexOnKeyPoint1 );
if (!F.IsNull() && F.ShapeType() == TopAbs_FACE &&
!V.IsNull() && V.ShapeType() == TopAbs_VERTEX
}
}
+ // Update Python script
+ TCollection_AsciiString str( "pattern.ApplyToFace( " );
+ SMESH_Gen_i::AddObject( str, theFace ) += ", ";
+ SMESH_Gen_i::AddObject( str, theVertexOnKeyPoint1 ) += ", ";
+ str += TCollection_AsciiString( theReverse ) + " )";
+ SMESH_Gen_i::AddToCurrentPyScript( str );
+
return points._retn();
}
SMESH::point_array_var points = new SMESH::point_array;
list<const gp_XYZ *> xyzList;
- TopExp_Explorer exp( getShape( theBlock ), TopAbs_SHELL );
- TopoDS_Shape V000 = getShape( theVertex000 );
- TopoDS_Shape V001 = getShape( theVertex001 );
+ TopExp_Explorer exp( myGen->GeomObjectToShape( theBlock ), TopAbs_SHELL );
+ TopoDS_Shape V000 = myGen->GeomObjectToShape( theVertex000 );
+ TopoDS_Shape V001 = myGen->GeomObjectToShape( theVertex001 );
if (exp.More() &&
!V000.IsNull() && V000.ShapeType() == TopAbs_VERTEX &&
}
}
+ // Update Python script
+ TCollection_AsciiString str( "pattern.ApplyTo3DBlock( " );
+ SMESH_Gen_i::AddObject( str, theBlock ) += ", ";
+ SMESH_Gen_i::AddObject( str, theVertex000 ) += ", ";
+ SMESH_Gen_i::AddObject( str, theVertex001 ) += " )";
+ SMESH_Gen_i::AddToCurrentPyScript( str );
+
+ return points._retn();
+}
+
+//=======================================================================
+//function : ApplyToMeshFaces
+//purpose :
+//=======================================================================
+
+SMESH::point_array*
+ SMESH_Pattern_i::ApplyToMeshFaces(SMESH::SMESH_Mesh_ptr theMesh,
+ const SMESH::long_array& theFacesIDs,
+ CORBA::Long theNodeIndexOnKeyPoint1,
+ CORBA::Boolean theReverse)
+{
+ SMESH::point_array_var points = new SMESH::point_array;
+
+ ::SMESH_Mesh* aMesh = getMesh( theMesh );
+ if ( !aMesh )
+ return points._retn();
+
+ list<const gp_XYZ *> xyzList;
+ set<const SMDS_MeshFace*> fset;
+ for (int i = 0; i < theFacesIDs.length(); i++)
+ {
+ CORBA::Long index = theFacesIDs[i];
+ const SMDS_MeshElement * elem = aMesh->GetMeshDS()->FindElement(index);
+ if ( elem && elem->GetType() == SMDSAbs_Face )
+ fset.insert( static_cast<const SMDS_MeshFace *>( elem ));
+ }
+ if (myPattern.Apply( fset, theNodeIndexOnKeyPoint1, theReverse ) &&
+ myPattern.GetMappedPoints( xyzList ))
+ {
+ points->length( xyzList.size() );
+ list<const gp_XYZ *>::iterator xyzIt = xyzList.begin();
+ for ( int i = 0; xyzIt != xyzList.end(); xyzIt++ ) {
+ SMESH::PointStruct & p = points[ i++ ];
+ (*xyzIt)->Coord( p.x, p.y, p.z );
+ }
+ }
+
+ // Update Python script
+ TCollection_AsciiString str( "pattern.ApplyToMeshFaces( " );
+ SMESH_Gen_i::AddObject( str, theMesh ) += ", ";
+ SMESH_Gen_i::AddArray( str, theFacesIDs ) += ", ";
+ str += TCollection_AsciiString( (int)theNodeIndexOnKeyPoint1 ) + ", ";
+ str += TCollection_AsciiString( theReverse ) + " )";
+ SMESH_Gen_i::AddToCurrentPyScript( str );
+
+ return points._retn();
+}
+
+//=======================================================================
+//function : ApplyToHexahedrons
+//purpose :
+//=======================================================================
+
+SMESH::point_array*
+ SMESH_Pattern_i::ApplyToHexahedrons(SMESH::SMESH_Mesh_ptr theMesh,
+ const SMESH::long_array& theVolumesIDs,
+ CORBA::Long theNode000Index,
+ CORBA::Long theNode001Index)
+{
+ SMESH::point_array_var points = new SMESH::point_array;
+
+ ::SMESH_Mesh* aMesh = getMesh( theMesh );
+ if ( !aMesh )
+ return points._retn();
+
+ list<const gp_XYZ *> xyzList;
+ set<const SMDS_MeshVolume*> vset;
+ for (int i = 0; i < theVolumesIDs.length(); i++)
+ {
+ CORBA::Long index = theVolumesIDs[i];
+ const SMDS_MeshElement * elem = aMesh->GetMeshDS()->FindElement(index);
+ if ( elem && elem->GetType() == SMDSAbs_Volume && elem->NbNodes() == 8 )
+ vset.insert( static_cast<const SMDS_MeshVolume *>( elem ));
+ }
+ if (myPattern.Apply( vset, theNode000Index, theNode001Index ) &&
+ myPattern.GetMappedPoints( xyzList ))
+ {
+ points->length( xyzList.size() );
+ list<const gp_XYZ *>::iterator xyzIt = xyzList.begin();
+ for ( int i = 0; xyzIt != xyzList.end(); xyzIt++ ) {
+ SMESH::PointStruct & p = points[ i++ ];
+ (*xyzIt)->Coord( p.x, p.y, p.z );
+ }
+ }
+
+ // Update Python script
+ TCollection_AsciiString str( "pattern.ApplyToHexahedrons( " );
+ SMESH_Gen_i::AddObject( str, theMesh ) += ", ";
+ SMESH_Gen_i::AddArray( str, theVolumesIDs ) += ", ";
+ str += TCollection_AsciiString( (int)theNode000Index ) + ", ";
+ str += TCollection_AsciiString( (int)theNode001Index ) + " )";
+ SMESH_Gen_i::AddToCurrentPyScript( str );
+
return points._retn();
}
//purpose :
//=======================================================================
-CORBA::Boolean SMESH_Pattern_i::MakeMesh(SMESH::SMESH_Mesh_ptr theMesh)
+CORBA::Boolean SMESH_Pattern_i::MakeMesh (SMESH::SMESH_Mesh_ptr theMesh,
+ const CORBA::Boolean CreatePolygons,
+ const CORBA::Boolean CreatePolyedrs)
{
::SMESH_Mesh* aMesh = getMesh( theMesh );
if ( !aMesh )
return false;
- return myPattern.MakeMesh( aMesh );
+ // Update Python script
+ TCollection_AsciiString str( "isDone = pattern.MakeMesh( " );
+ SMESH_Gen_i::AddObject( str, theMesh ) += ", ";
+ str += TCollection_AsciiString( CreatePolygons ) + ", ";
+ str += TCollection_AsciiString( CreatePolyedrs ) + " )";
+ SMESH_Gen_i::AddToCurrentPyScript( str );
+ addErrorCode( "MakeMesh" );
+
+ return myPattern.MakeMesh( aMesh, CreatePolygons, CreatePolyedrs );
}
//=======================================================================
//purpose :
//=======================================================================
-SMESH::array_of_long_array* SMESH_Pattern_i::GetElementPoints()
+SMESH::array_of_long_array* SMESH_Pattern_i::GetElementPoints(CORBA::Boolean applied)
{
SMESH::array_of_long_array_var arrayOfArray = new SMESH::array_of_long_array;
- const list< list< int > >& listOfIdList = myPattern.GetElementPointIDs();
+ const list< list< int > >& listOfIdList = myPattern.GetElementPointIDs(applied);
arrayOfArray->length( listOfIdList.size() );
list< list< int > >::const_iterator llIt = listOfIdList.begin();
for ( int i = 0 ; llIt != listOfIdList.end(); llIt++, i++ )