dc/d0f/tst_nuclide_set_8f90-example.html

program tstnuclideset


#include "ScaleSTL/FortranTestMacros.h"


use origen_nuclideset_m

use scalestl_vec_int_m

use origen_definitions_m


implicit none


type(origen_nuclideset) :: nucset,nucset2

type(scalestl_vec_int) :: vec_ids,inds

type(scalestl_vec_str) :: vec_str

integer,allocatable :: native_ids(:)

integer,parameter :: l=c_size_t

integer(L) :: n

type(scalestl_vec_int) :: remember

logical(C_BOOL) :: found


!initialize a nuclide set and make sure it is empty

call nucset % initialize()

expect_true(nucset % has_ids() ) !internal ids are non-null

expect_eq(0,nucset % ids_size() )


!create a Vec_Int and manage it in the nuclide set

!(once you use manageptr, it is owned by the NuclideSet--do not destroy)

call vec_ids % initialize()

call vec_ids % push_back(10008016)

call vec_ids % push_back(10008017)

call vec_ids % push_back(10054135)

call vec_ids % push_back(20092235)

call vec_ids % push_back(20092238)

call vec_ids % push_back(30054135)

expect_eq(6,vec_ids%size())


!dangerous!

!only do this when you know exactly what's going on, e.g. in a unit test

remember % instance_ptr = vec_ids % instance_ptr

call nucset % manageptr_ids(vec_ids)

!has been automatically nullified to you can't kill it

assert_not_c_assoc(vec_ids)


!get the pointer and make sure memory is same

call nucset % getptr_ids_vec(vec_ids)

assert_c_assoc2(remember,vec_ids)


!get a copy of the ids in a native fortran vector

call nucset % get_ids(native_ids)


expect_eq(6,size(native_ids,1))

expect_eq(10008016,native_ids(1))

expect_eq(10008017,native_ids(2))

expect_eq(10054135,native_ids(3))

expect_eq(20092235,native_ids(4))

expect_eq(20092238,native_ids(5))

expect_eq(30054135,native_ids(6))


!check the lookup_index

expect_eq(1_l,nucset%lookup_index(10008016))

expect_eq(2_l,nucset%lookup_index(10008017))

expect_eq(3_l,nucset%lookup_index(10054135))

expect_eq(4_l,nucset%lookup_index(20092235))

expect_eq(5_l,nucset%lookup_index(20092238))

expect_eq(6_l,nucset%lookup_index(30054135))


!not found as index (must be exact match)

expect_eq(nucset%NOT_FOUND(),nucset%lookup_index(54135))


!found fission product version with last_izzzaaa lookup

expect_eq(6_l,nucset%lookup_index_guess(54135))


!now get both 30054135 and 10054135

call inds%initialize()

found = nucset%lookup_index_all(inds,54135)

expect_true(found)

expect_eq(2_l,inds%size())

expect_eq(3_l,inds%at(1_l))

expect_eq(6_l,inds%at(2_l))


!save the original with a smart/shallow copy

call nucset2 % initialize_copy(nucset)


!change one value and set the new ids

!the data will be reallocated completely

native_ids(4) = 21092235

call nucset % set_ids(native_ids)


!test the pointer on the new data

call nucset % getptr_ids_vec(vec_ids)

expect_eq(21092235,vec_ids%at(4_l))


!test the pointer on the old data

call nucset2 % getptr_ids_vec(vec_ids)

expect_eq(20092235,vec_ids%at(4_l))


!destroy and free all the smart pointers

call nucset % destroy()

call nucset2 % destroy()


!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

! Test symbol_to_izzzaaa

!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

call vec_ids % initialize()

call vec_str % initialize()

call vec_str % push_back("u235")

call vec_str % push_back("u235m")

call vec_str % push_back("pu239")

call vec_str % push_back("h1")

call vec_str % push_back("zr")

call nucset % convert_symbol_to_izzzaaa(vec_str,vec_ids)

expect_eq(92235,vec_ids%at(1_l))

expect_eq(1092235,vec_ids%at(2_l))

expect_eq(94239,vec_ids%at(3_l))

expect_eq(1001,vec_ids%at(4_l))

expect_eq(40000,vec_ids%at(5_l))


call nucset % convert_sizzzaaa_to_zzzaaai(vec_ids)

expect_eq(922350,vec_ids%at(1_l))

expect_eq(922351,vec_ids%at(2_l))

expect_eq(942390,vec_ids%at(3_l))

expect_eq( 10010,vec_ids%at(4_l))

expect_eq(400000,vec_ids%at(5_l))


call vec_ids % destroy()

call vec_str % destroy()


end program