source: TI02-CSML/trunk/csml/API/ops_FileExtract.py @ 2221

Subversion URL: http://proj.badc.rl.ac.uk/svn/ndg/TI02-CSML/trunk/csml/API/ops_FileExtract.py@2221
Revision 2221, 2.5 KB checked in by domlowe, 13 years ago (diff)

removing print statements

Line 
1''' ops_AbstractFileExtract  contains operations for AbstractFileExtracts'''
2import csml.csmllibs.csmldataiface
3import sys
4import Numeric
5import MA
6
7
8class DataContainer(object):
9    #dataArray needs to be the standards data container   
10    def __init__(self, data=None):
11        self.dataArray=None
12        if data is not None:
13            self.data=data
14            self.setup()
15   
16    def setup(self):
17        try: 
18            self.dataArray=Numeric.array(self.data)
19        except MA.MA.MAError:
20            self.dataArray=self.data.tolist()
21            self.dataArray=Numeric.array(self.dataArray)
22         
23def testmethod(self):
24    print 'testmethod for AbstractFileExtract feature'
25    return 'testmethod AbstractFileExtract'
26
27
28def getAxisInfo(self, varName):
29    pass
30
31def getData(self,  fileposition=None,**kwargs):
32    #file position defines the position of the filename if a list of filenames exists
33    #**kwargs can hold subsetting request.
34    #get the right file  ##TO DO< check this is ok with FILE LIST
35    if fileposition is not None:
36        file =self.fileList.fileNames.CONTENT.split()[fileposition]
37        #file=self.fileName.CONTENT.split()[fileposition]
38    else:
39        file=self.fileName.CONTENT
40    #open the file
41    DI = csml.csmllibs.csmldataiface.DataInterface()
42    DI=DI.getUnknownInterfaceType(file)
43    DI.openFile(file)
44    DI.setAxis(self.variableName.CONTENT)
45    try:
46        DI.setAxis(self.variableName.CONTENT)
47        data=DI.getDataForAxis()
48        dataArray=data
49        fillvalue=None
50        axisorder=None
51        units=None
52    except:
53        DI.setVariable(self.variableName.CONTENT)
54        #get fill value (if there is one)   
55        fillvalue=DI.getVariableAttribute('_FillValue')
56        if kwargs:
57            print 'subsetting'
58            data = DI.getSubsetOfDataForVar(**kwargs)
59        else:
60            print 'non subsetting'
61            data = DI.getDataForVar()
62        DC=DataContainer(data)
63        dataArray=DC.dataArray
64        units=[]
65        axisorder=DI.getVariableAxes()
66        for axis in axisorder:   
67            DI.setAxis(axis)
68            units.append(DI.getAxisAttribute('units'))
69    DI.closeFile()
70    return dataArray, fillvalue, axisorder, units
71       
72def __calendar(file,timedim):
73    #open the file
74    DI = csmldataiface.DataInterface()
75    DI=DI.getUnknownInterfaceType(file)
76    DI.openFile(file)
77    DI.setAxis(timedim)
78    cal=DI.getAxisAttribute('calendar')
79    units=DI.getAxisAttribute('units')
80    return cal, units
81   
Note: See TracBrowser for help on using the repository browser.