import go_canada_rio
import numpy as np

# initialize library
r = go_canada_rio.Riometer()
site_uid = "fsim"
year = 2017
month = 1
day = 1
hour = 0
downsample = 5
file_path = "/mnt/nfs/go/go-rio/raw/%d/%02d/%02d/%s_xrio" % (year, month, day, site_uid)

# get hour of data
rio_data = r.read_raw_hour(file_path, site_uid, year, month, day, hour, downsample=downsample)

# print data values
print "Shape: ", rio_data.shape
print "Max value: ", rio_data.max()