import sys
import Numeric
from numarray import *
from numarray import strings
from tables import *

# Open a new empty HDF5 file
fileh = openFile("array1.h5", mode = "w")
# Get the root group
root = fileh.root

# Create an Array
a = array([-1, 2, 4], Int16)
# Save it on the HDF5 file
hdfarray = fileh.createArray(root, 'array_1', a, "Signed short array")

# Create a scalar Array
a = array(4, Int16)
# Save it on the HDF5 file
hdfarray = fileh.createArray(root, 'array_s', a, "Scalar signed short array")

# Create a 3-d array of floats
#a = arange(64, type=Float64, shape=(2,4,8))
a = arange(120, type=Float64, shape=(20,3,2))
# Save it on the HDF5 file
hdfarray = fileh.createArray(root, 'array_f', a, "3-D float array")
# Crea a Numeric array
a = Numeric.arange(120, typecode="d")
#Numeric.reshape(a,(20,3,2))
a.shape=(20,3,2)
print "a.shape-->", a.shape
# Save it on the HDF5 file
hdfarray = fileh.createArray(root, 'array_f_Numeric', a, "3-D float array, Numeric")

# Close the file
fileh.close()

# Open the file for reading
fileh = openFile("array1.h5", mode = "r")
# Get the root group
root = fileh.root

a = root.array_1.read()
print "Signed byte array -->",repr(a), a.shape

print "Testing iterator (works even over scalar arrays):",
arr = root.array_s
for x in arr:
    print "nrow-->", arr.nrow
    print "Element-->", repr(x)

# print "Testing getitem:"
# for i in range(root.array_1.nrows):
#     print "array_1["+str(i)+"]", "-->", root.array_1[i]

print "array_f[:,2:3,2::2]", repr(root.array_f[:,2:3,2::2])
print "array_f[1,2:]", repr(root.array_f[1,2:])
print "array_f[1]", repr(root.array_f[1])

# Close the file
fileh.close()