gen_parameter.py

#!/usr/bin/env python

# Copyright 2011-2013 Pierre de Buyl
#
# This file is part of f90h5md
#
# f90h5md is free software and is licensed under the modified BSD license (see
# LICENSE file).

types = dict()
types['i'] = 'integer'
types['d'] = 'double precision'
types['l'] = 'logical'
types['c'] = 'character'
H5T = dict()
H5T['i'] = 'H5T_NATIVE_INTEGER'
H5T['d'] = 'H5T_NATIVE_DOUBLE'
H5T['c'] = 'a_type'
H5T['l'] = 'H5T_NATIVE_INTEGER'
dims = dict()
dims['s'] = ''
dims['1'] = '(:)'
dims['2'] = '(:,:)'

for t_k,t_v in types.iteritems():
    if (t_v=='character'):
        dims_var = dict()
        dims_var['s'] = ''
        t_v = 'character(len=*)'
    else:
        dims_var = dims
    for d_k,d_v in dims_var.iteritems():
        if (d_k == 's'):
            rank = 1
        else:
            rank = int(d_k)
        s=''
        s+="""  !> Writes a parameter to the parameter group.
  !! @param file_id hdf5 file ID.
  !! @param name name of the parameter.
  !! @param data value of the parameter.
  !! @private"""
        s+="""
  subroutine h5md_write_par_%s%s(file_id, name, data)
    integer(HID_T), intent(in) :: file_id
    character(len=*), intent(in) :: name
    %s, intent(in) :: data%s

    integer(HID_T) :: par_d, par_s
    integer(HSIZE_T) :: dims(%i)"""  % (t_k,d_k,t_v,d_v,rank)
        if (t_k=='c'):
            s+="""
    integer(SIZE_T) :: a_size(1)
    integer(HID_T) :: a_type"""
        if (t_k=='l'):
            if (d_k=='s'):
                s+="""
    integer :: data_int
    if (data) then
        data_int = 1
    else
        data_int = 0
    end if
"""
            elif(d_k=='1'):
                s+="""
    integer, allocatable :: data_int%s
    allocate(data_int(size(data)))
    where (data)
        data_int = 1
    elsewhere
        data_int=0
    endwhere
""" % (d_v,)
            elif(d_k=='2'):
                s+="""
    integer, allocatable :: data_int%s
    allocate(data_int(size(data,dim=1),size(data,dim=2)))
    where (data)
        data_int = 1
    elsewhere
        data_int=0
    endwhere
""" % (d_v,)

        if (t_k=='c'):
            s+="""
    a_size(1) = len(data)
    call h5tcopy_f(H5T_NATIVE_CHARACTER, a_type, h5_error)
    call h5tset_size_f(a_type, a_size(1), h5_error)"""

        if (d_k=='s'):
            s+="""
    call h5screate_f(H5S_SCALAR_F, par_s, h5_error)
"""
        else:
            s+="""
    dims = shape(data)
    call h5screate_simple_f(%i, dims, par_s, h5_error)
""" % (rank, )


        s+="""
    call h5dcreate_f(file_id, 'parameters/'//name, %s, par_s, par_d, h5_error)
""" % (H5T[t_k], )
        
        if (t_k=='l'):
            s+="""
    call h5dwrite_f(par_d, %s, data_int, dims, h5_error)
""" % (H5T[t_k],)
            if (d_k!='s'):
                s+="""
    deallocate(data_int)"""
        else:
            s+="""
    call h5dwrite_f(par_d, %s, data, dims, h5_error)
""" % (H5T[t_k],)

        s+="""
    call h5sclose_f(par_s, h5_error)

       
  end subroutine h5md_write_par_%s%s
""" % (t_k, d_k)
        print s