source: TI02-CSML/trunk/Scanner/csmllibs/csmlfileextracts.py @ 1123

#!/usr/bin/env python
import os
import csmllibs
import sys
class fileExtractBuilder:
    def __init__(self, dataset_element, ffmap):
        self.dataset_element=dataset_element
        self.ffmap=ffmap
        self.createFileExtracts()
    
    def createFileExtracts(self):
        #print "REPRESENTATIVE FILES:"
        #print self.ffmap.getRepresentativeFiles()
        
        #given a featurefilemap object, create file extracts from the "representativeFiles"     
        #used to store extract/filename/variable relationship
        #referenced when creating features
        self.fileExtractDictionary={}
    
        fileid=0  #used to distinguish extract names for similar variables from different files
        #adlist is an empty list object to hold CSML ArrayDescriptors eg. NetCDFExtract, GRIBExtract etc
        adlist=[]
        for repfile in self.ffmap.getRepresentativeFiles():
            fileid=fileid+1
            filename=repfile.getRepresentativeFileName()
            #print 'representative file:'
            #print repfile.getRepresentativeFileName()
            #print 'feature type:'
            #print repfile.getFeatureType()     
            DI = csmllibs.csmldataiface.DataInterface()
            DI = DI.getUnknownInterfaceType(filename)
            DI.openFile(filename)
            #creates fileextracts for each variable of each representative file
            self.extractType = DI.extractType
            self.extractPrefix = str(fileid)+'_'+DI.extractPrefix
            #print dir(DI)
            allDimNames=DI.getListOfAxes()
            numDomains=len(allDimNames)
            for j in range (0, len(allDimNames)):
                print self.extractType
                print self.extractPrefix
                if self.extractType=='NASAAmesExtract':
                    arrayDescriptor=csmllibs.Parser.NASAAmesExtract()
                if self.extractType=='NetCDFExtract':
                    arrayDescriptor=csmllibs.Parser.NetCDFExtract()
                if self.extractType=='GRIBExtract':
                    arrayDescriptor=csmllibs.Parser.GRIBExtract()
                if self.extractType=='PPExtract':
                    arrayDescriptor=csmllibs.Parser.PPExtract()
                #print self.extractType
                arrayDescriptor.id=str(self.extractPrefix+allDimNames[j])
    
                #SET variableName attribute of file extract
                arrayDescriptor.variableName=str(allDimNames[j])
                
                
                #SET arraySize attribute of file extract
                dimValue =DI.getSizeOfAxis(allDimNames[j])
                if dimValue == None:
                    dimValue ='1'  #temporary to fix unlimited problem
                arrayDescriptor.arraySize=str(dimValue)
                #SET filename attribute of file extract
                arrayDescriptor.fileName = filename
            
    #                   #keep record of extracts/filenames/variables in dictionary
                idstr = filename + allDimNames[j]
                self.fileExtractDictionary[idstr] = self.extractPrefix+allDimNames[j]
                adlist.append(arrayDescriptor)
            DI.closeFile()      
        #Add all the file extracts to the dataset element
        setattr(self.dataset_element, 'arrayDescriptors', adlist)
    

def createSingleExtract(extractType, fileName, variableName, arraySize):
    #creates a single file extract on demand. (not part of general builder class)
    #need to incorporate check that a file extract of the same name doesn't already exist.
    if extractType=='NASAAmesExtract':
        arrayDescriptor=csmllibs.Parser.NASAAmesExtract()
    if extractType=='NetCDFExtract':
        arrayDescriptor=csmllibs.Parser.NetCDFExtract()
    if extractType=='GRIBExtract':
        arrayDescriptor=csmllibs.Parser.GRIBExtract()
    if extractType=='PPExtract':
        arrayDescriptor=csmllibs.Parser.PPExtract()
    #need to change this so it does the right thing for grib extracts too..
    arrayDescriptor.id='_%s' % variableName   #need prefix system here..
    arrayDescriptor.fileName = fileName
    arrayDescriptor.arraySize=[arraySize]
    arrayDescriptor.variableName=variableName
    return arrayDescriptor
            
        #    sys.exit()