source: python/genericio.py @ 9796d0a

Revision 9796d0a, 4.5 KB checked in by Hal Finkel <hfinkel@…>, 7 years ago (diff)

Add support for float4 (and similar) to Python bindings

  • Property mode set to 100644
RevLine 
[bd84570]1#                    Copyright (C) 2015, UChicago Argonne, LLC
2#                               All Rights Reserved
3#
4#                               Generic IO (ANL-15-066)
5#                     Hal Finkel, Argonne National Laboratory
6#
7#                              OPEN SOURCE LICENSE
8#
9# Under the terms of Contract No. DE-AC02-06CH11357 with UChicago Argonne,
10# LLC, the U.S. Government retains certain rights in this software.
11#
12# Redistribution and use in source and binary forms, with or without
13# modification, are permitted provided that the following conditions are met:
14#
15#   1. Redistributions of source code must retain the above copyright notice,
16#      this list of conditions and the following disclaimer.
17#
18#   2. Redistributions in binary form must reproduce the above copyright
19#      notice, this list of conditions and the following disclaimer in the
20#      documentation and/or other materials provided with the distribution.
21#
22#   3. Neither the names of UChicago Argonne, LLC or the Department of Energy
23#      nor the names of its contributors may be used to endorse or promote
24#      products derived from this software without specific prior written
25#      permission.
26#
27# *****************************************************************************
28#
29#                                  DISCLAIMER
[ef727a6]30# THE SOFTWARE IS SUPPLIED "AS IS" WITHOUT WARRANTY OF ANY KIND.  NEITHER THE
[bd84570]31# UNTED STATES GOVERNMENT, NOR THE UNITED STATES DEPARTMENT OF ENERGY, NOR
32# UCHICAGO ARGONNE, LLC, NOR ANY OF THEIR EMPLOYEES, MAKES ANY WARRANTY,
33# EXPRESS OR IMPLIED, OR ASSUMES ANY LEGAL LIABILITY OR RESPONSIBILITY FOR THE
34# ACCURACY, COMPLETENESS, OR USEFULNESS OF ANY INFORMATION, DATA, APPARATUS,
35# PRODUCT, OR PROCESS DISCLOSED, OR REPRESENTS THAT ITS USE WOULD NOT INFRINGE
36# PRIVATELY OWNED RIGHTS.
37#
38# *****************************************************************************
39
40import numpy as np
41import ctypes as ct
42import os
43
44#Define where the library is and load it
[c13d973]45_path = os.path.dirname(__file__)
46libpygio = ct.CDLL(_path + '/../frontend/libpygio.so')
[bd84570]47#we need to define the return type ("restype") and
48#the argument types
49libpygio.get_elem_num.restype=ct.c_int64
50libpygio.get_elem_num.argtypes=[ct.c_char_p]
51
52libpygio.get_variable_type.restype=ct.c_int
53libpygio.get_variable_type.argtypes=[ct.c_char_p,ct.c_char_p]
54
[9796d0a]55libpygio.get_variable_field_count.restype=ct.c_int
56libpygio.get_variable_field_count.argtypes=[ct.c_char_p,ct.c_char_p]
57
[bd84570]58libpygio.read_gio_int32.restype=None
[9796d0a]59libpygio.read_gio_int32.argtypes=[ct.c_char_p,ct.c_char_p,ct.POINTER(ct.c_int),ct.c_int]
[bd84570]60
61libpygio.read_gio_int64.restype=None
[9796d0a]62libpygio.read_gio_int64.argtypes=[ct.c_char_p,ct.c_char_p,ct.POINTER(ct.c_int64),ct.c_int]
[bd84570]63
64libpygio.read_gio_float.restype=None
[9796d0a]65libpygio.read_gio_float.argtypes=[ct.c_char_p,ct.c_char_p,ct.POINTER(ct.c_float),ct.c_int]
[bd84570]66
67libpygio.read_gio_double.restype=None
[9796d0a]68libpygio.read_gio_double.argtypes=[ct.c_char_p,ct.c_char_p,ct.POINTER(ct.c_double),ct.c_int]
[bd84570]69
70libpygio.inspect_gio.restype=None
71libpygio.inspect_gio.argtypes=[ct.c_char_p]
72
73def gio_read(file_name,var_name):
74    var_size = libpygio.get_elem_num(file_name)
75    var_type = libpygio.get_variable_type(file_name,var_name)
[9796d0a]76    field_count = libpygio.get_variable_field_count(file_name,var_name)
[bd84570]77    if(var_type==10):
78        print "Variable not found"
79        return
80    elif(var_type==9):
81        print "variable type not known (not int32/int64/float/double)"
82    elif(var_type==0):
83        #float
[9796d0a]84        result = np.ndarray((var_size,field_count),dtype=np.float32)
85        libpygio.read_gio_float(file_name,var_name,result.ctypes.data_as(ct.POINTER(ct.c_float)),field_count)
[bd84570]86        return result
87    elif(var_type==1):
88        #double
[9796d0a]89        result = np.ndarray((var_size,field_count),dtype=np.float64)
90        libpygio.read_gio_double(file_name,var_name,result.ctypes.data_as(ct.POINTER(ct.c_double)),field_count)
[bd84570]91        return result
92    elif(var_type==2):
93        #int32
[9796d0a]94        result = np.ndarray((var_size,field_count),dtype=np.int32)
95        libpygio.read_gio_int32(file_name,var_name,result.ctypes.data_as(ct.POINTER(ct.c_int32)),field_count)
[bd84570]96        return result
97    elif(var_type==3):
98        #int64
[9796d0a]99        result = np.ndarray((var_size,field_count),dtype=np.int64)
100        libpygio.read_gio_int64(file_name,var_name,result.ctypes.data_as(ct.POINTER(ct.c_int64)),field_count)
[bd84570]101        return result       
[9796d0a]102
103def gio_has_variable(file_name,var_name):
104    var_size = libpygio.get_elem_num(file_name)
105    var_type = libpygio.get_variable_type(file_name,var_name)
106    return var_type!=10
107
[bd84570]108def gio_inspect(file_name):
109    libpygio.inspect_gio(file_name)
110
Note: See TracBrowser for help on using the repository browser.