source: TI02-CSML/trunk/Scanner/csmllibs/csmlbuilder.py @ 1403

import xml.dom.ext
from xml.dom.minidom import parseString
from cElementTree import tostring
import csmllibs


import sys


#Class for building the csml document. 


class csmlBuilder:
    #this class contains all the method calls to create and populate a csmldoc
    #it needs to be provided with several bits of information to be able to do this.
    def __init__(self,directory,csmlft, mapping, timedimension ,outputfile,printscreen,timestorage,spatialstorage,valuestorage):
        self.directory = directory # top level directory containing data files 
        self.csmlfeaturetype  = csmlft 
        self.mapping = mapping 
        self.timedimension=timedimension
        self.outputfile=outputfile
        self.printscreen=printscreen
        self.timestorage=timestorage
        self.spatialstorage=spatialstorage
        self.valuestorage=valuestorage
        self.csml = None  # csml object holds the csml document in memory.
        self.ds_element = None #<Dataset> tag, root node.
         
    def build(self):
        #wrapper method to call methods in correct order.
        print "Builder"
        self.createDataset()
        self.addGMLMetadata()
        self.createFeatureFileMap()
        if self.spatialstorage=='fileextract':
           self.makeFileExtracts()
        self.createFeatureCollection()
        self.createFeatures()
        #self.closeFC()
        #self.closeDS()
        self.saveFile()
        self.printToScreen()

    def createDataset(self):
        #*************************************************************************************
        #Creates empty CSML Dataset object with standard attributes
        #*************************************************************************************
              
        ########### The Dataset  ##############
        #Create an Empty Dataset
        self.ds = csmllibs.Parser.Dataset()
        #Set gml:id attribute of dataset to be equal to the directory name. (may want to rethink this later)
        setattr(self.ds,'id',self.directory)
        #setattr(self.ds, 'arrayDescriptors', adlist)
        #setattr(self.ds, 'unitDefinitions', uds)
        #setattr(self.ds, 'phenomenonDefinitions', pds)
        #setattr(self.ds, 'featureCollection',fc)
        ########################################

    def addGMLMetadata(self):
        #*********************************************************************************
        # Adds main GML Metadata elements to dataset element
        #*********************************************************************************
        #Note: TODO: The Parser has a problem with gml:Name from AbstractGML.
        strMetaDataProperty = 'http://ndg.nerc.ac.uk/Metadata' + self.directory
        strCodespace = 'http://ndg/nerc/ac/uk/NDGData'
        strGmlDescription = 'Dataset for directory ' + self.directory
        self.ds.metaDataProperty=[csmllibs.Parser.MetaDataProperty(href=strMetaDataProperty)]
        self.ds.name=csmllibs.Parser.Name( codeSpace=strCodespace, val=[self.directory])
        self.ds.description=csmllibs.Parser.Description(strGmlDescription)
    
    def createCSML(self):
        print "Create CSML"
        #create an empty csml document and dataset tag
        self.csml, self.ds_element = csmllibs.csmlmeta.createDataset()
        #add gml metadata elements
        self.csml, self.ds_element = csmllibs.csmlmeta.addGMLMetadata(self.csml, self.directory, self.ds_element)

    def createFeatureFileMap(self):
        print "Create FFMAP"
         #    print "creating filemap"
        fmm = csmllibs.csmlfiles.FileMapMaker(self.directory, self.csmlfeaturetype)
        if self.mapping=='onetomany':
            self.ffmap = fmm.onetomany()
        elif self.mapping=='onetoone':
            self.ffmap = fmm.onetoone()
        elif self.mapping=='oneonly':
            self.ffmap = fmm.oneonly()

    def makeFileExtracts(self):
        print "make FEs"
        feBuilder =csmllibs.csmlfileextracts.fileExtractBuilder( self.ds, self.ffmap, self.timedimension)
        self.ds = feBuilder.dataset_element
        self.extractType = feBuilder.extractType
        self.extractDictionary=feBuilder.fileExtractDictionary

    def createFeatureCollection(self):
        #creates empty feature collection (list)
        self.gml_FeatureCollection = csmllibs.Parser.FeatureCollection()
 
    def createFeatures(self):
        print "create Features"
        if not hasattr(self, 'extractType'):
            self.extractType=None
        if not hasattr(self, 'extractDictionary'):
            self.extractDictionary=None
        thefeatures = csmllibs.csmlfeaturetypes.featureBuilder( self.ds,self.gml_FeatureCollection, self.ffmap, self.extractDictionary,self.timedimension, self.timestorage, self.spatialstorage,self.valuestorage)
        #self.csml = thefeatures.csml
        self.ds=thefeatures.ds_element
        self.gml_FeatureCollection = thefeatures.gml_FeatureCollection_element
        self.ds.featureCollection=self.gml_FeatureCollection

                    
    def saveFile(self):
        print "Creating CSML document and saving file"
        #call the toXML method of the Dataset object:
        csml = self.ds.toXML()
        #parse and pretty print csml
        self.strCSML =csmllibs.parser_extra.PrettyPrint(csml)
        #tidy up elementtree style namespaces
        self.strCSML=csmllibs.parser_extra.removeInlineNS(self.strCSML)
        #open file, write to it.
        f = open(self.outputfile, "w")
        f.write('<?xml version="1.0" encoding="UTF-8"?>')
        f.write(self.strCSML)
        f.close()

    def printToScreen(self):
        if self.printscreen==1:
            print self.strCSML