source: python/genericio.py @ dd56945

Revision dd56945, 5.3 KB checked in by Hal Finkel <hfinkel@…>, 6 years ago (diff)

import uint16 fix by Biwer, Christopher Michael <cmbiwer@…>
  • Property mode set to 100644
Line 
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
30# THE SOFTWARE IS SUPPLIED "AS IS" WITHOUT WARRANTY OF ANY KIND.  NEITHER THE
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
40from __future__ import print_function
41import numpy as np
42import ctypes as ct
43import os,sys
44
45#Define where the library is and load it
46_path = os.path.dirname(__file__)
47libpygio = ct.CDLL(_path + '/../frontend/libpygio.so')
48#we need to define the return type ("restype") and
49#the argument types
50libpygio.get_elem_num.restype=ct.c_int64
51libpygio.get_elem_num.argtypes=[ct.c_char_p]
52
53libpygio.get_variable_type.restype=ct.c_int
54libpygio.get_variable_type.argtypes=[ct.c_char_p,ct.c_char_p]
55
56libpygio.get_variable_field_count.restype=ct.c_int
57libpygio.get_variable_field_count.argtypes=[ct.c_char_p,ct.c_char_p]
58
59libpygio.read_gio_uint16.restype=None
60libpygio.read_gio_uint16.argtypes=[ct.c_char_p,ct.c_char_p,ct.POINTER(ct.c_uint16),ct.c_int]
61
62libpygio.read_gio_int32.restype=None
63libpygio.read_gio_int32.argtypes=[ct.c_char_p,ct.c_char_p,ct.POINTER(ct.c_int),ct.c_int]
64
65libpygio.read_gio_int64.restype=None
66libpygio.read_gio_int64.argtypes=[ct.c_char_p,ct.c_char_p,ct.POINTER(ct.c_int64),ct.c_int]
67
68libpygio.read_gio_float.restype=None
69libpygio.read_gio_float.argtypes=[ct.c_char_p,ct.c_char_p,ct.POINTER(ct.c_float),ct.c_int]
70
71libpygio.read_gio_double.restype=None
72libpygio.read_gio_double.argtypes=[ct.c_char_p,ct.c_char_p,ct.POINTER(ct.c_double),ct.c_int]
73
74libpygio.inspect_gio.restype=None
75libpygio.inspect_gio.argtypes=[ct.c_char_p]
76
77def gio_read(file_name,var_name):
78    if sys.version_info[0] == 3:
79        file_name = bytes(file_name,'ascii')
80        var_name = bytes(var_name,'ascii')
81    var_size = libpygio.get_elem_num(file_name)
82    var_type = libpygio.get_variable_type(file_name,var_name)
83    field_count = libpygio.get_variable_field_count(file_name,var_name)
84    if(var_type==10):
85        print("Variable not found")
86        return
87    elif(var_type==9):
88        print("variable type not known (not uint16/int32/int64/float/double)")
89    elif(var_type==0):
90        #float
91        result = np.ndarray((var_size,field_count),dtype=np.float32)
92        libpygio.read_gio_float(file_name,var_name,result.ctypes.data_as(ct.POINTER(ct.c_float)),field_count)
93        return result
94    elif(var_type==1):
95        #double
96        result = np.ndarray((var_size,field_count),dtype=np.float64)
97        libpygio.read_gio_double(file_name,var_name,result.ctypes.data_as(ct.POINTER(ct.c_double)),field_count)
98        return result
99    elif(var_type==2):
100        #int32
101        result = np.ndarray((var_size,field_count),dtype=np.int32)
102        libpygio.read_gio_int32(file_name,var_name,result.ctypes.data_as(ct.POINTER(ct.c_int32)),field_count)
103        return result
104    elif(var_type==3):
105        #int64
106        result = np.ndarray((var_size,field_count),dtype=np.int64)
107        libpygio.read_gio_int64(file_name,var_name,result.ctypes.data_as(ct.POINTER(ct.c_int64)),field_count)
108        return result       
109    elif(var_type==4):
110        #uint16
111        result = np.ndarray((var_size,field_count),dtype=np.uint16)
112        libpygio.read_gio_uint16(file_name,var_name,result.ctypes.data_as(ct.POINTER(ct.c_uint16)),field_count)
113        return result
114
115def gio_has_variable(file_name,var_name):
116    if sys.version_info[0] == 3:
117        file_name=bytes(file_name,'ascii')
118        var_name=bytes(var_name,'ascii')
119    var_size = libpygio.get_elem_num(file_name)
120    var_type = libpygio.get_variable_type(file_name,var_name)
121    return var_type!=10
122
123def gio_inspect(file_name):
124    if sys.version_info[0] == 3:
125        file_name=bytes(file_name,'ascii')
126    libpygio.inspect_gio(file_name)
127
Note: See TracBrowser for help on using the repository browser.