source: TI02-CSML/trunk/csml/API/ops_GridSeriesFeature.py @ 2105

''' ops_GridSeriesFeature  contains operations for GridSeriesFeatures'''

import csml.parser
import csml.csmllibs.csmltime
import csml.csmllibs.csmldocument
import csml.API.ops_AbstractFeature
import csml.API.genSubset
import csml.csmllibs.netCDFWriter
import csmlutils


import sys  #remove later

def testmethod(self):
    #print 'testmethod for gridseries feature'
    return 'testmethod - gridseries'

def getAllowedSubsettings(self):
    return ['subsetToGridSeries']  #other operations

def getDomain(self):
    #returns domain as a dictionary of ordinates {name: [values], ...}
    self.domain={}
    for gridOrd in self.value.gridSeriesDomain.coordTransformTable.gridOrdinates:
        name=gridOrd.coordAxisLabel.CONTENT
        if hasattr(gridOrd.coordAxisValues, '__insertedExtract'):
            self.domain[name]=gridOrd.coordAxisValues.__insertedExtract.getData()
        else:
            vals=gridOrd.coordAxisValues.coordinateList.CONTENT
            valList=[]
            for val in vals.split():  
                valList.append(val)
            self.domain[name]=valList
    return self.domain

    
def subsetToGridSeries(self, csmlpath=None, ncpath=None,**kwargs):
    #set self.domain:
    self.getDomain()
    
    #get the CRS from a  the  catalogue
    cat=csml.csmllibs.csmlcrs.CRSCatalogue()
    crs=cat.getCRS(self.value.gridSeriesDomain.srsName)  
    
    #non-feature specific setup code, mainly handles the time dimension/calendar
    pathToSubsetNetCDF, kwargs, timeName, calunits, caltype, times=csml.API.genSubset.genericSubset(self, csmlpath, ncpath, self.domain, kwargs)
      
    #Find the original time name (timeAxis) for the file.
    for gridOrd in self.value.gridSeriesDomain.coordTransformTable.gridOrdinates:
        if gridOrd.coordAxisLabel.CONTENT==timeName:
            timeAxis = gridOrd.gridAxesSpanned.CONTENT  #only works for regular grids atm
        else: 
            print 'crs not supported'
 
    ##Get names of variables in file and relate them to the subset selection 
    selection={}
    for gridOrd in self.value.gridSeriesDomain.coordTransformTable.gridOrdinates:
        selection[gridOrd.gridAxesSpanned.CONTENT]=kwargs[gridOrd.coordAxisLabel.CONTENT]
   
    
    strTimes, axisorder, fulldata=csml.API.genSubset.getTheData(self, selection, times, timeName)
   
    
    #Okay, got the data now. Need to write CSML feature and NetCDF files.
    #Writing out the CSML feature
    
    # define domain/coverage  to use in 'value' attribute   
    domain=csml.parser.GridSeriesDomain()
    domainSubset, totalArraySize=csml.API.genSubset.subsetDomain(timeName,strTimes,self.domain, **kwargs)
    cTT=csml.API.genSubset.getCoordTransformTable(domainSubset, crs)
    domain.coordTransformTable=cTT
    rangeSet=csml.parser.RangeSet()
    rangeSet.arrayDescriptor=csml.parser.NetCDFExtract(id=self.id,fileName=csml.parser.csString(pathToSubsetNetCDF),variableName=csml.parser.csString(self.id),arraySize=csml.parser.csString(totalArraySize))
    
    #gridseries coverage
    cvg=csml.parser.GridSeriesCoverage()
    cvg.rangeSet=rangeSet
    cvg.gridSeriesDomain=domain    
    csmlWrap=csml.csmllibs.csmlfeaturewrap.CSMLWrapper()
    
    #create a stand alone gridseries feature containing this coverage
    subsettedFeature=csmlWrap.createGridSeriesFeature(value=cvg,featureID=self.id,description=self.description)
 
    
    ### write netcdf using NCWriter class (wraps cdms) ###
    nc=csml.csmllibs.netCDFWriter.NCwriter(pathToSubsetNetCDF)
    ords=cTT.gridOrdinates
    nc.genWriteVar(self.id,ords, times, calunits, caltype, axisorder, fulldata)
    nc.closeFinishedFile()
    
    return subsettedFeature, pathToSubsetNetCDF