feat(fortran): Added fortran bindings
Building of the C API GridFire can now be used from fotran using the gridfire_mod fortran module. This exposes the same, limited, set of funcitonality that the C API does.
This commit is contained in:
258
src/extern/fortran/gridfire_mod.f90
vendored
Normal file
258
src/extern/fortran/gridfire_mod.f90
vendored
Normal file
@@ -0,0 +1,258 @@
|
||||
module gridfire_mod
|
||||
use iso_c_binding
|
||||
implicit none
|
||||
|
||||
enum, bind (C)
|
||||
enumerator :: FDSSE_NON_4DSTAR_ERROR = -102
|
||||
enumerator :: FDSSE_UNKNOWN_ERROR = -101
|
||||
|
||||
enumerator :: FDSSE_SUCCESS = 1
|
||||
enumerator :: FDSSE_UNKNOWN_SYMBOL_ERROR = 100
|
||||
|
||||
enumerator :: FDSSE_SPECIES_ERROR = 101
|
||||
enumerator :: FDSSE_INVALID_COMPOSITION_ERROR = 102
|
||||
enumerator :: FDSSE_COMPOSITION_ERROR = 103
|
||||
|
||||
enumerator :: GF_NON_GRIDFIRE_ERROR = -2
|
||||
enumerator :: GF_UNKNOWN_ERROR = -1
|
||||
enumerator :: GF_SUCCESS = 0
|
||||
|
||||
enumerator :: GF_INVALID_QSE_SOLUTION_ERROR = 5
|
||||
enumerator :: GF_FAILED_TO_PARTITION_ERROR = 6
|
||||
enumerator :: GF_NETWORK_RESIZED_ERROR = 7
|
||||
enumerator :: GF_UNABLE_TO_SET_NETWORK_REACTIONS_ERROR = 8
|
||||
enumerator :: GF_BAD_COLLECTION_ERROR = 9
|
||||
enumerator :: GF_BAD_RHS_ENIGNE_ERROR = 10
|
||||
enumerator :: GF_STALE_JACOBIAN_ERROR = 11
|
||||
enumerator :: GF_UNINITIALIZED_JACOBIAN_ERROR = 12
|
||||
enumerator :: GF_UNKNONWN_JACOBIAN_ERROR = 13
|
||||
enumerator :: GF_JACOBIAN_ERROR = 14
|
||||
enumerator :: GF_ENGINE_ERROR = 15
|
||||
|
||||
enumerator :: GF_MISSING_BASE_REACTION_ERROR = 16
|
||||
enumerator :: GF_MISSING_SEED_SPECIES_ERROR = 17
|
||||
enumerator :: GF_MISSING_KEY_REACTION_ERROR = 18
|
||||
enumerator :: GF_POLICY_ERROR = 19
|
||||
|
||||
enumerator :: GF_REACTION_PARSING_ERROR = 20
|
||||
enumerator :: GF_REACTOION_ERROR = 21
|
||||
|
||||
enumerator :: GF_SINGULAR_JACOBIAN_ERROR = 22
|
||||
enumerator :: GF_ILL_CONDITIONED_JACOBIAN_ERROR = 23
|
||||
enumerator :: GF_CVODE_SOLVER_FAILURE_ERROR = 24
|
||||
enumerator :: GF_KINSOL_SOLVER_FAILURE_ERROR = 25
|
||||
enumerator :: GF_SUNDIALS_ERROR = 26
|
||||
enumerator :: GF_SOLVER_ERROR = 27
|
||||
|
||||
enumerator :: GF_HASHING_ERROR = 28
|
||||
enumerator :: GF_UTILITY_ERROR = 29
|
||||
|
||||
enumerator :: GF_DEBUG_ERRROR = 30
|
||||
|
||||
enumerator :: GF_GRIDFIRE_ERROR = 31
|
||||
end enum
|
||||
|
||||
interface
|
||||
! void* gf_init()
|
||||
function gf_init() bind(C, name="gf_init")
|
||||
import :: c_ptr
|
||||
type(c_ptr) :: gf_init
|
||||
end function gf_init
|
||||
|
||||
! void gf_free(void* gf)
|
||||
subroutine gf_free(gf) bind(C, name="gf_free")
|
||||
import :: c_ptr
|
||||
type(c_ptr), value :: gf
|
||||
end subroutine gf_free
|
||||
|
||||
! char* gf_get_last_error_message(void* ptr);
|
||||
function gf_get_last_error_message(ptr) result(c_msg) bind(C, name="gf_get_last_error_message")
|
||||
import
|
||||
type(c_ptr), value :: ptr
|
||||
type(c_ptr) :: c_msg
|
||||
end function
|
||||
|
||||
! int gf_register_species(void* ptr, const int num_species, const char** species_names);
|
||||
function gf_register_species(ptr, num_species, species_names) result(ierr) bind(C, name="gf_register_species")
|
||||
import
|
||||
type(c_ptr), value :: ptr
|
||||
integer(c_int), value :: num_species
|
||||
type(c_ptr), dimension(*), intent(in) :: species_names ! Array of C pointers
|
||||
integer(c_int) :: ierr
|
||||
end function
|
||||
|
||||
! int gf_construct_engine_from_policy(void* ptr, const char* policy_name, const double *abundances, size_t num_species);
|
||||
function gf_construct_engine_from_policy(ptr, policy_name, abundances, num_species) result(ierr) &
|
||||
bind(C, name="gf_construct_engine_from_policy")
|
||||
import
|
||||
type(c_ptr), value :: ptr
|
||||
character(kind=c_char), dimension(*), intent(in) :: policy_name
|
||||
real(c_double), dimension(*), intent(in) :: abundances
|
||||
integer(c_size_t), value :: num_species
|
||||
integer(c_int) :: ierr
|
||||
end function
|
||||
|
||||
! int gf_construct_solver_from_engine(void* ptr, const char* solver_name);
|
||||
function gf_construct_solver_from_engine(ptr, solver_name) result(ierr) &
|
||||
bind(C, name="gf_construct_solver_from_engine")
|
||||
import
|
||||
type(c_ptr), value :: ptr
|
||||
character(kind=c_char), dimension(*), intent(in) :: solver_name
|
||||
integer(c_int) :: ierr
|
||||
end function
|
||||
|
||||
! int gf_evolve(...)
|
||||
function gf_evolve(ptr, Y_in, num_species, T, rho, dt, Y_out, energy_out, dEps_dT, dEps_dRho, mass_lost) result(ierr) &
|
||||
bind(C, name="gf_evolve")
|
||||
import
|
||||
type(c_ptr), value :: ptr
|
||||
real(c_double), dimension(*), intent(in) :: Y_in
|
||||
integer(c_size_t), value :: num_species
|
||||
real(c_double), value :: T, rho, dt
|
||||
real(c_double), dimension(*), intent(out) :: Y_out
|
||||
real(c_double), intent(out) :: energy_out, dEps_dT, dEps_dRho, mass_lost
|
||||
integer(c_int) :: ierr
|
||||
end function
|
||||
end interface
|
||||
|
||||
type :: GridFire
|
||||
type(c_ptr) :: ctx = c_null_ptr
|
||||
integer(c_size_t) :: num_species = 0
|
||||
contains
|
||||
procedure :: gff_init
|
||||
procedure :: gff_free
|
||||
procedure :: register_species
|
||||
procedure :: setup_policy
|
||||
procedure :: setup_solver
|
||||
procedure :: evolve
|
||||
procedure :: get_last_error
|
||||
end type GridFire
|
||||
|
||||
contains
|
||||
subroutine gff_init(self)
|
||||
class(GridFire), intent(out) :: self
|
||||
|
||||
self%ctx = gf_init()
|
||||
end subroutine gff_init
|
||||
|
||||
subroutine gff_free(self)
|
||||
class(GridFire), intent(inout) :: self
|
||||
|
||||
if (c_associated(self%ctx)) then
|
||||
call gf_free(self%ctx)
|
||||
self%ctx = c_null_ptr
|
||||
end if
|
||||
end subroutine gff_free
|
||||
|
||||
function get_last_error(self) result(msg)
|
||||
class(GridFire), intent(in) :: self
|
||||
character(len=:), allocatable :: msg
|
||||
type(c_ptr) :: c_msg_ptr
|
||||
character(kind=c_char), pointer :: char_ptr(:)
|
||||
integer :: i, len_str
|
||||
|
||||
c_msg_ptr = gf_get_last_error_message(self%ctx)
|
||||
if (.not. c_associated(c_msg_ptr)) then
|
||||
msg = "GridFire: Unknown Error (Null Pointer returned)"
|
||||
return
|
||||
end if
|
||||
|
||||
call c_f_pointer(c_msg_ptr, char_ptr, [1024])
|
||||
len_str = 0
|
||||
do i = 1, 1024
|
||||
if (char_ptr(i) == c_null_char) exit
|
||||
len_str = len_str + 1
|
||||
end do
|
||||
|
||||
msg = repeat(' ', len_str+10)
|
||||
msg(1:10) = "GridFire: "
|
||||
do i = 1, len_str
|
||||
msg(i+10:i+10) = char_ptr(i)
|
||||
end do
|
||||
end function get_last_error
|
||||
|
||||
subroutine register_species(self, species_list)
|
||||
class(GridFire), intent(inout) :: self
|
||||
character(len=*), dimension(:), intent(in) :: species_list
|
||||
|
||||
type(c_ptr), allocatable, dimension(:) :: c_ptrs
|
||||
character(kind=c_char, len=:), allocatable, target :: temp_strs(:)
|
||||
integer :: i, n, ierr
|
||||
|
||||
print *, "Registering ", size(species_list), " species."
|
||||
n = size(species_list)
|
||||
self%num_species = int(n, c_size_t)
|
||||
|
||||
allocate(c_ptrs(n))
|
||||
allocate(character(len=len(species_list(1))+1) :: temp_strs(n)) ! +1 for null terminator
|
||||
|
||||
do i = 1, n
|
||||
temp_strs(i) = trim(species_list(i)) // c_null_char
|
||||
c_ptrs(i) = c_loc(temp_strs(i))
|
||||
end do
|
||||
|
||||
print *, "Calling gf_register_species..."
|
||||
ierr = gf_register_species(self%ctx, int(n, c_int), c_ptrs)
|
||||
print *, "gf_register_species returned with code: ", ierr
|
||||
|
||||
if (ierr /= GF_SUCCESS .AND. ierr /= FDSSE_SUCCESS) then
|
||||
print *, "GridFire: ", self%get_last_error()
|
||||
error stop
|
||||
end if
|
||||
end subroutine register_species
|
||||
|
||||
subroutine setup_policy(self, policy_name, abundances)
|
||||
class(GridFire), intent(in) :: self
|
||||
character(len=*), intent(in) :: policy_name
|
||||
real(c_double), dimension(:), intent(in) :: abundances
|
||||
integer(c_int) :: ierr
|
||||
|
||||
if (size(abundances) /= self%num_species) then
|
||||
print *, "GridFire Error: Abundance array size mismatch."
|
||||
error stop
|
||||
end if
|
||||
|
||||
ierr = gf_construct_engine_from_policy(self%ctx, &
|
||||
trim(policy_name) // c_null_char, &
|
||||
abundances, &
|
||||
self%num_species)
|
||||
|
||||
if (ierr /= GF_SUCCESS .AND. ierr /= FDSSE_SUCCESS) then
|
||||
print *, "GridFire Policy Error: ", self%get_last_error()
|
||||
error stop
|
||||
end if
|
||||
end subroutine setup_policy
|
||||
|
||||
subroutine setup_solver(self, solver_name)
|
||||
class(GridFire), intent(in) :: self
|
||||
character(len=*), intent(in) :: solver_name
|
||||
integer(c_int) :: ierr
|
||||
|
||||
ierr = gf_construct_solver_from_engine(self%ctx, trim(solver_name) // c_null_char)
|
||||
if (ierr /= GF_SUCCESS .AND. ierr /= FDSSE_SUCCESS) then
|
||||
print *, "GridFire Solver Error: ", self%get_last_error()
|
||||
error stop
|
||||
end if
|
||||
end subroutine setup_solver
|
||||
|
||||
subroutine evolve(self, Y_in, T, rho, dt, Y_out, energy, dedt, dedrho, mass_lost, ierr)
|
||||
class(GridFire), intent(in) :: self
|
||||
real(c_double), dimension(:), intent(in) :: Y_in
|
||||
real(c_double), value :: T, rho, dt
|
||||
real(c_double), dimension(:), intent(out) :: Y_out
|
||||
real(c_double), intent(out) :: energy, dedt, dedrho, mass_lost
|
||||
integer, intent(out) :: ierr
|
||||
integer(c_int) :: c_ierr
|
||||
|
||||
c_ierr = gf_evolve(self%ctx, &
|
||||
Y_in, self%num_species, &
|
||||
T, rho, dt, &
|
||||
Y_out, &
|
||||
energy, dedt, dedrho, mass_lost)
|
||||
|
||||
ierr = int(c_ierr)
|
||||
if (ierr /= GF_SUCCESS .AND. ierr /= FDSSE_SUCCESS) then
|
||||
print *, "GridFire Evolve Error: ", self%get_last_error()
|
||||
end if
|
||||
end subroutine evolve
|
||||
end module gridfire_mod
|
||||
11
src/extern/fortran/meson.build
vendored
Normal file
11
src/extern/fortran/meson.build
vendored
Normal file
@@ -0,0 +1,11 @@
|
||||
gridfire_fortran_lib = library('gridfire_fortran',
|
||||
'gridfire_mod.f90',
|
||||
link_with: libgridfire_extern,
|
||||
install: true,
|
||||
install_dir: get_option('libdir')
|
||||
)
|
||||
|
||||
gridfire_fortran_dep = declare_dependency(
|
||||
link_with: [gridfire_fortran_lib, libgridfire_extern],
|
||||
include_directories: include_directories('.')
|
||||
)
|
||||
@@ -62,7 +62,7 @@ extern "C" {
|
||||
|
||||
void* gf_init();
|
||||
|
||||
int gf_free(void* ctx);
|
||||
void gf_free(void* ctx);
|
||||
|
||||
int gf_register_species(void* ptr, const int num_species, const char** species_names);
|
||||
|
||||
|
||||
3
src/extern/lib/gridfire_extern.cpp
vendored
3
src/extern/lib/gridfire_extern.cpp
vendored
@@ -8,9 +8,8 @@ extern "C" {
|
||||
return new GridFireContext();
|
||||
}
|
||||
|
||||
int gf_free(void* ctx) {
|
||||
void gf_free(void* ctx) {
|
||||
delete static_cast<GridFireContext*>(ctx);
|
||||
return 0;
|
||||
}
|
||||
|
||||
int gf_register_species(void* ptr, const int num_species, const char** species_names) {
|
||||
|
||||
2
src/extern/meson.build
vendored
2
src/extern/meson.build
vendored
@@ -22,3 +22,5 @@ gridfire_extern_dep = declare_dependency(
|
||||
)
|
||||
|
||||
install_subdir('include/gridfire', install_dir: get_option('includedir'))
|
||||
|
||||
subdir('fortran')
|
||||
|
||||
Reference in New Issue
Block a user