Commit 37a31b40 authored by Jakub Klinkovský's avatar Jakub Klinkovský
Browse files

pytnl: export bindings for VTUReader

parent 999fd145

src/Python/pytnl/tnl/CMakeLists.txt

+15 −0
Original line number Diff line number Diff line
@@ -41,4 +41,19 @@ if( ${WITH_CI_FLAGS} ) 
endif()





# enable zlib and tinyxml2 (used by VTUReader)

find_package( ZLIB )

if( ZLIB_FOUND )

   target_compile_definitions(pytnl PUBLIC "-DHAVE_ZLIB")

   target_include_directories(pytnl PUBLIC ${ZLIB_INCLUDE_DIRS})

   target_link_libraries(pytnl PUBLIC ${ZLIB_LIBRARIES})

endif()



find_package( tinyxml2 QUIET )

if( tinyxml2_FOUND )

   target_compile_definitions(pytnl PUBLIC "-DHAVE_TINYXML2")

   target_link_libraries(pytnl PUBLIC tinyxml2::tinyxml2)

endif()





install( TARGETS pytnl DESTINATION ${PYTHON_SITE_PACKAGES_DIR} )

src/Python/pytnl/tnl/Mesh.cpp

+14 −8
Original line number Diff line number Diff line
@@ -3,16 +3,12 @@ 


#include "Mesh.h"

#include <TNL/Meshes/Readers/VTKReader.h>

#include <TNL/Meshes/Readers/VTUReader.h>



void export_Meshes( py::module & m )

template< typename Reader >

void export_reader( py::module & m, const char* name )

{

    export_Mesh< MeshOfEdges >( m, "MeshOfEdges" );

    export_Mesh< MeshOfTriangles >( m, "MeshOfTriangles" );

    export_Mesh< MeshOfTetrahedrons >( m, "MeshOfTetrahedrons" );



    using Reader = TNL::Meshes::Readers::VTKReader;



    py::class_< Reader >( m, "VTKReader" )

    py::class_< Reader >( m, name )

        .def(py::init<std::string>())

        .def("loadMesh", &Reader::template loadMesh< MeshOfEdges >)

        .def("loadMesh", &Reader::template loadMesh< MeshOfTriangles >)
@@ -28,3 +24,13 @@ void export_Meshes( py::module & m ) 
//            } )

    ;

}



void export_Meshes( py::module & m )

{

    export_Mesh< MeshOfEdges >( m, "MeshOfEdges" );

    export_Mesh< MeshOfTriangles >( m, "MeshOfTriangles" );

    export_Mesh< MeshOfTetrahedrons >( m, "MeshOfTetrahedrons" );



    export_reader< TNL::Meshes::Readers::VTKReader >( m, "VTKReader" );

    export_reader< TNL::Meshes::Readers::VTUReader >( m, "VTUReader" );

}