2017-05-31 22:38:17 +05:00
# ifdef NG_PYTHON
# ifdef OCCGEOMETRY
# include <../general/ngpython.hpp>
2019-07-11 19:23:22 +05:00
# include <core/python_ngcore.hpp>
2019-07-26 02:25:58 +05:00
# include "../meshing/python_mesh.hpp"
2017-05-31 22:38:17 +05:00
# include <meshing.hpp>
# include <occgeom.hpp>
2019-10-28 19:11:42 +05:00
# include <Standard_Version.hxx>
2017-05-31 22:38:17 +05:00
using namespace netgen ;
namespace netgen
{
extern std : : shared_ptr < NetgenGeometry > ng_geometry ;
}
2019-08-27 13:10:17 +05:00
static string occparameter_description = R " delimiter(
OCC Specific Meshing Parameters
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
2019-09-16 15:48:27 +05:00
closeedgefac : Optional [ float ] = 2.
2019-08-27 13:10:17 +05:00
Factor for meshing close edges , if None it is disabled .
minedgelen : Optional [ float ] = 0.001
Minimum edge length to be used for dividing edges to mesh points . If
None this is disabled .
) delimiter " ;
void CreateOCCParametersFromKwargs ( OCCParameters & occparam , py : : dict kwargs )
{
if ( kwargs . contains ( " minedgelen " ) )
{
auto val = kwargs . attr ( " pop " ) ( " minedgelen " ) ;
if ( val . is_none ( ) )
occparam . resthminedgelenenable = false ;
else
{
occparam . resthminedgelen = py : : cast < double > ( val ) ;
occparam . resthminedgelenenable = true ;
}
}
}
2017-05-31 22:38:17 +05:00
DLL_HEADER void ExportNgOCC ( py : : module & m )
{
2019-10-28 19:11:42 +05:00
m . attr ( " occ_version " ) = OCC_VERSION_COMPLETE ;
2018-04-17 22:37:29 +05:00
py : : class_ < OCCGeometry , shared_ptr < OCCGeometry > , NetgenGeometry > ( m , " OCCGeometry " , R " raw_string(Use LoadOCCGeometry to load the geometry from a *.step file.)raw_string " )
2017-05-31 22:38:17 +05:00
. def ( py : : init < > ( ) )
2019-07-26 02:25:58 +05:00
. def ( py : : init ( [ ] ( const string & filename )
{
shared_ptr < OCCGeometry > geo ;
if ( EndsWith ( filename , " .step " ) | | EndsWith ( filename , " .stp " ) )
geo . reset ( LoadOCC_STEP ( filename . c_str ( ) ) ) ;
else if ( EndsWith ( filename , " .brep " ) )
geo . reset ( LoadOCC_BREP ( filename . c_str ( ) ) ) ;
else if ( EndsWith ( filename , " .iges " ) )
geo . reset ( LoadOCC_IGES ( filename . c_str ( ) ) ) ;
else
throw Exception ( " Cannot load file " + filename + " \n Valid formats are: step, stp, brep, iges " ) ;
ng_geometry = geo ;
return geo ;
} ) , py : : arg ( " filename " ) ,
" Load OCC geometry from step, brep or iges file " )
2018-12-20 21:01:27 +05:00
. def ( NGSPickle < OCCGeometry > ( ) )
2020-06-01 00:58:21 +05:00
. def ( " Glue " , & OCCGeometry : : GlueGeometry )
2017-11-03 21:44:01 +05:00
. def ( " Heal " , [ ] ( OCCGeometry & self , double tolerance , bool fixsmalledges , bool fixspotstripfaces , bool sewfaces , bool makesolids , bool splitpartitions )
{
self . tolerance = tolerance ;
self . fixsmalledges = fixsmalledges ;
self . fixspotstripfaces = fixspotstripfaces ;
self . sewfaces = sewfaces ;
self . makesolids = makesolids ;
self . splitpartitions = splitpartitions ;
self . HealGeometry ( ) ;
self . BuildFMap ( ) ;
2018-03-13 02:38:21 +05:00
} , py : : arg ( " tolerance " ) = 1e-3 , py : : arg ( " fixsmalledges " ) = true , py : : arg ( " fixspotstripfaces " ) = true , py : : arg ( " sewfaces " ) = true , py : : arg ( " makesolids " ) = true , py : : arg ( " splitpartitions " ) = false , R " raw_string(Heal the OCCGeometry.)raw_string " , py : : call_guard < py : : gil_scoped_release > ( ) )
2019-08-13 21:45:27 +05:00
. def ( " SetFaceMeshsize " , [ ] ( OCCGeometry & self , size_t fnr , double meshsize )
{
self . SetFaceMaxH ( fnr , meshsize ) ;
} , " Set maximum meshsize for face fnr. Face numbers are 0 based. " )
2018-07-12 19:35:52 +05:00
. def ( " _visualizationData " , [ ] ( shared_ptr < OCCGeometry > occ_geo )
{
std : : vector < float > vertices ;
std : : vector < int > trigs ;
std : : vector < float > normals ;
std : : vector < float > min = { std : : numeric_limits < float > : : max ( ) ,
std : : numeric_limits < float > : : max ( ) ,
std : : numeric_limits < float > : : max ( ) } ;
std : : vector < float > max = { std : : numeric_limits < float > : : lowest ( ) ,
std : : numeric_limits < float > : : lowest ( ) ,
std : : numeric_limits < float > : : lowest ( ) } ;
std : : vector < string > surfnames ;
auto box = occ_geo - > GetBoundingBox ( ) ;
for ( int i = 0 ; i < 3 ; i + + )
{
min [ i ] = box . PMin ( ) [ i ] ;
max [ i ] = box . PMax ( ) [ i ] ;
}
occ_geo - > BuildVisualizationMesh ( 0.01 ) ;
gp_Pnt2d uv ;
gp_Pnt pnt ;
gp_Vec n ;
gp_Pnt p [ 3 ] ;
int count = 0 ;
for ( int i = 1 ; i < = occ_geo - > fmap . Extent ( ) ; i + + )
{
surfnames . push_back ( " occ_surface " + to_string ( i ) ) ;
auto face = TopoDS : : Face ( occ_geo - > fmap ( i ) ) ;
auto surf = BRep_Tool : : Surface ( face ) ;
TopLoc_Location loc ;
BRepAdaptor_Surface sf ( face , Standard_False ) ;
BRepLProp_SLProps prop ( sf , 1 , 1e-5 ) ;
Handle ( Poly_Triangulation ) triangulation = BRep_Tool : : Triangulation ( face , loc ) ;
if ( triangulation . IsNull ( ) )
cout < < " cannot visualize face " < < i < < endl ;
trigs . reserve ( trigs . size ( ) + triangulation - > NbTriangles ( ) * 4 ) ;
vertices . reserve ( vertices . size ( ) + triangulation - > NbTriangles ( ) * 3 * 3 ) ;
normals . reserve ( normals . size ( ) + triangulation - > NbTriangles ( ) * 3 * 3 ) ;
for ( int j = 1 ; j < triangulation - > NbTriangles ( ) + 1 ; j + + )
{
auto triangle = ( triangulation - > Triangles ( ) ) ( j ) ;
for ( int k = 1 ; k < 4 ; k + + )
p [ k - 1 ] = ( triangulation - > Nodes ( ) ) ( triangle ( k ) ) . Transformed ( loc ) ;
for ( int k = 1 ; k < 4 ; k + + )
{
vertices . insert ( vertices . end ( ) , { float ( p [ k - 1 ] . X ( ) ) , float ( p [ k - 1 ] . Y ( ) ) , float ( p [ k - 1 ] . Z ( ) ) } ) ;
trigs . insert ( trigs . end ( ) , { count , count + 1 , count + 2 , i } ) ;
count + = 3 ;
uv = ( triangulation - > UVNodes ( ) ) ( triangle ( k ) ) ;
prop . SetParameters ( uv . X ( ) , uv . Y ( ) ) ;
if ( prop . IsNormalDefined ( ) )
n = prop . Normal ( ) ;
else
{
gp_Vec a ( p [ 0 ] , p [ 1 ] ) ;
gp_Vec b ( p [ 0 ] , p [ 2 ] ) ;
n = b ^ a ;
}
if ( face . Orientation ( ) = = TopAbs_REVERSED ) n * = - 1 ;
normals . insert ( normals . end ( ) , { float ( n . X ( ) ) , float ( n . Y ( ) ) , float ( n . Z ( ) ) } ) ;
}
}
}
py : : gil_scoped_acquire ac ;
py : : dict res ;
py : : list snames ;
for ( auto name : surfnames )
snames . append ( py : : cast ( name ) ) ;
res [ " vertices " ] = MoveToNumpy ( vertices ) ;
res [ " triangles " ] = MoveToNumpy ( trigs ) ;
res [ " normals " ] = MoveToNumpy ( normals ) ;
res [ " surfnames " ] = snames ;
res [ " min " ] = MoveToNumpy ( min ) ;
res [ " max " ] = MoveToNumpy ( max ) ;
return res ;
} , py : : call_guard < py : : gil_scoped_release > ( ) )
2019-07-29 10:11:57 +05:00
. def ( " GenerateMesh " , [ ] ( shared_ptr < OCCGeometry > geo ,
MeshingParameters * pars , py : : kwargs kwargs )
2019-07-26 02:25:58 +05:00
{
MeshingParameters mp ;
2019-08-27 13:10:17 +05:00
OCCParameters occparam ;
2019-07-26 02:25:58 +05:00
{
py : : gil_scoped_acquire aq ;
2019-08-27 13:10:17 +05:00
if ( pars )
{
auto mp_kwargs = CreateDictFromFlags ( pars - > geometrySpecificParameters ) ;
CreateOCCParametersFromKwargs ( occparam , mp_kwargs ) ;
mp = * pars ;
}
CreateOCCParametersFromKwargs ( occparam , kwargs ) ;
2019-07-29 10:11:57 +05:00
CreateMPfromKwargs ( mp , kwargs ) ;
2019-07-26 02:25:58 +05:00
}
2019-10-02 20:20:13 +05:00
geo - > SetOCCParameters ( occparam ) ;
2019-07-26 02:25:58 +05:00
auto mesh = make_shared < Mesh > ( ) ;
mesh - > SetGeometry ( geo ) ;
2019-11-26 21:08:21 +05:00
auto result = geo - > GenerateMesh ( mesh , mp ) ;
if ( result ! = 0 )
throw Exception ( " Meshing failed! " ) ;
2019-10-02 20:20:13 +05:00
SetGlobalMesh ( mesh ) ;
2019-07-26 02:25:58 +05:00
ng_geometry = geo ;
return mesh ;
2019-07-29 10:11:57 +05:00
} , py : : arg ( " mp " ) = nullptr ,
2019-07-26 02:25:58 +05:00
py : : call_guard < py : : gil_scoped_release > ( ) ,
2019-08-27 13:10:17 +05:00
( meshingparameter_description + occparameter_description ) . c_str ( ) )
2019-07-26 02:25:58 +05:00
;
m . def ( " LoadOCCGeometry " , [ ] ( const string & filename )
2017-05-31 22:38:17 +05:00
{
2019-07-26 02:25:58 +05:00
cout < < " WARNING: LoadOCCGeometry is deprecated! Just use the OCCGeometry(filename) constructor. It is able to read brep and iges files as well! " < < endl ;
2017-05-31 22:38:17 +05:00
ifstream ist ( filename ) ;
OCCGeometry * instance = new OCCGeometry ( ) ;
instance = LoadOCC_STEP ( filename . c_str ( ) ) ;
2018-08-01 18:53:01 +05:00
ng_geometry = shared_ptr < OCCGeometry > ( instance , NOOP_Deleter ) ;
return ng_geometry ;
2019-07-26 02:25:58 +05:00
} , py : : call_guard < py : : gil_scoped_release > ( ) ) ;
2017-05-31 22:38:17 +05:00
}
2017-09-01 12:33:57 +05:00
PYBIND11_MODULE ( libNgOCC , m ) {
2017-05-31 22:38:17 +05:00
ExportNgOCC ( m ) ;
}
# endif // OCCGEOMETRY
# endif // NG_PYTHON