1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51
|
#!/usr/bin/python
from __future__ import absolute_import
import dballe
import volnd
#import rpy, MA, numpy, rconvert
import numpy, rconvert, rpy2
from datetime import *
numpy.seterr(divide="raise", over="raise", under="raise", invalid="raise")
#def f(x,y,z):
# return (x+1)*100+(y+1)*10+(z+1)
#a = MA.array(numpy.fromfunction(f, (4,3,2)))
##a = MA.array(numpy.array([[[1,2,3],[4,5,6]],[[7,8,9],[10,11,12]]], dtype='int8'))
#a[1,1,1]=MA.masked
#print "verypre", a
##rconvert.makeConvertible(a)
#rpy.r.print_(rconvert.ma_to_r(a, dimnames=[['a','b','c','d'],['e','f','g'],['h', 'i']]))
# Test from volnd
db = dballe.DB.connect_test()
query = dballe.Record(date=datetime(2007, 1, 1, 0, 0, 0))
#query.set("var", "B10004")
#query.settimerange(dballe.TimeRange(0,0,0))
vars = volnd.read(db.query_data(query), (volnd.AnaIndex(), volnd.NetworkIndex(), volnd.LevelIndex(), volnd.TimeRangeIndex()))
#print "ana:", vars["B10004"].dims[0]
#print "net:", vars["B10004"].dims[1]
#print vars["B10004"]
#rpy.r.print_(rconvert.vnddata_to_r(vars['B10004']))
#rpy.r.assign('pippo', rconvert.vnddata_to_r(vars['B10004']))
#rpy.r.save('pippo', file='/tmp/pippo')
rconvert.volnd_save_to_r(vars, "/tmp/pippo")
#
# Second test case reported by Paolo
#
query = dballe.Record()
#query.setd("latmin", 10.)
#query.setd("latmax", 60.)
#query.setd("lonmin", -10.)
#query.setd("lonmax", 40.)
query["var"] = "B13011"
vars = volnd.read(db.query_data(query), (volnd.AnaIndex(),volnd.DateTimeIndex()), checkConflicts=False)
rconvert.volnd_save_to_r(vars, "/tmp/pippo")
|