source: TI02-CSML/branches/CSML2/csParser.py @ 1863

Subversion URL: http://proj.badc.rl.ac.uk/svn/ndg/TI02-CSML/branches/CSML2/csParser.py@1863
Revision 1863, 26.2 KB checked in by domlowe, 13 years ago (diff)

GridSeries Feature and coverage added

Line 
1import cElementTree as ET
2import elementtree.ElementTree as etree
3import parser_extra
4import sys
5'''CSML v2 Parser ''' 
6
7#this map needs updating for V2
8etree._namespace_map.update({'http://www.opengis.net/om': 'om', 'http://www.opengis.net/swe': 'swe',  'http://www.opengis.net/gml': 'gml','http://ndg.nerc.ac.uk/csml' : 'csml', 'http://www.w3.org/1999/xlink':'xlink'})
9
10nsCSML = 'http://ndg.nerc.ac.uk/csml'
11nsGML = 'http://www.opengis.net/gml'
12nsOM = 'http://www.opengis.net/om'
13nsXLINK = 'http://www.w3.org/1999/xlink'
14nsXML = 'http://ndg.nerc.ac.uk/csml'
15nsMOLES='http://ndg.nerc.ac.uk/moles'
16nsSWE='http://www.opengis.net/swe'
17   
18def myQName(uri,tag):
19    return "{"+uri+"}"+tag
20
21def CSML(tag):
22    return myQName(nsCSML,tag)
23
24def GML(tag):
25    return myQName(nsGML,tag)
26
27def SWE(tag):
28    return myQName(nsSWE,tag)
29
30def XLINK(tag):
31    return myQName(nsXLINK,tag)
32
33def addchildren(obj, dict):
34    ''' merges (adds) dictionary to existing self.CHILDREN dictionary '''
35    dict1 = dict
36    if hasattr(obj, 'CHILDREN'):
37      dict2=obj.CHILDREN
38    else:
39      dict2={}
40    dict3={}
41    for item in dict1:
42        dict3[item] =dict1[item]
43    for item in dict2:
44        dict3[item] =dict2[item]
45    obj.CHILDREN=dict3
46   
47
48def addatts(obj, atts):
49  ''' merges self.ATTRIBUTES'''
50  if hasattr(obj, 'ATTRIBUTES'):
51    for att in atts:
52        obj.ATTRIBUTES.append(att)
53  else:
54    obj.ATTRIBUTES=atts
55
56
57#Some variable definitions: these things are often repeated so store in variables.
58FILEFORMATS=[CSML('NetCDFExtract'),CSML('NASAAmesExtract'), CSML('GRIBExtract'),CSML('CDMLExtract'), CSML('RawFileExtract')]
59 
60
61class csElement(object):
62    ''' main csElement class - all other elements inherit from this baseclass 
63    all the from/to XML conversion is done by this class'''
64   
65    def __init__(self, **kwargs):
66        if not hasattr(self, 'ATTRIBUTES'):
67            self.ATTRIBUTES=[]
68           
69       
70    def __removeURI(self, qname):
71        try:
72            attname = qname.split('}')[1]
73        except IndexError:
74            attname = qname
75        return attname
76       
77    def _getSubstitutionType(self,tag):
78        print 'splitting tag'
79        return tag.split('}')[1]
80       
81   
82    def _getReverseSubsType(self, typename):
83        return typename
84   
85    def toXML(self, csmlfrag):
86        #process self and convert to XML
87        if hasattr(self, 'CONTENT'):
88            if self.CONTENT is not None: csmlfrag.text=self.CONTENT
89        if hasattr(self, 'ATTRIBUTES'):
90            for item in self.__dict__:
91                if item in self.ATTRIBUTES:
92                    csmlfrag.set(item, self.__dict__[item])
93            for item in self.__dict__:
94                if GML(item) in self.ATTRIBUTES:
95                    csmlfrag.set(GML(item), self.__dict__[item])       
96    # self.CHILDREN (recursive - calls the toXML method of children
97        for att in self.__dict__:
98            if att not in ['ATTRIBUTES', 'CHILDREN', 'CONTENT']:
99                for child in self.CHILDREN:
100                    if child == att:
101                        parserobjects=[]
102                        if type(self.__dict__[att]) is list:
103                            for a in self.__dict__[att]:
104                                parserobjects.append(a)
105                        else:
106                            parserobjects.append(self.__dict__[att])
107                        parentfrag=None
108                        if len(self.CHILDREN[child])>=3:
109                                ename2=self.CHILDREN[child][2]
110                                parentfrag=ET.Element(ename2)
111                        for po in parserobjects:
112                            if type(self.CHILDREN[child][0]) is not list:
113                                ename=self.CHILDREN[child][0]
114                            else:
115                                ename = self._getReverseSubsType(type(po).__name__)
116                            if len(self.CHILDREN[child])==3:
117                                frag=ET.Element(ename)
118                                po.toXML(frag)
119                                parentfrag.append(frag)
120                                appendLater=True
121                            elif len(self.CHILDREN[child])>=4:
122                                if self.CHILDREN[child][3]==1:
123                                    frag=ET.Element(ename)
124                                    po.toXML(frag)
125                                    parentfrag.append(frag)
126                                    csmlfrag.append(parentfrag)
127                                    parentfrag=ET.Element(parentfrag.tag)
128                                    appendLater=False
129                            else:
130                                frag=ET.Element(ename)
131                                po.toXML(frag)
132                                csmlfrag.append(frag)
133                                appendLater=True
134                        if appendLater==True and parentfrag != None:
135                            csmlfrag.append(parentfrag)
136        return csmlfrag
137       
138    def fromXML(self,csmlfrag):
139        # deal with attributes, e.g. gml id's
140        if csmlfrag.text is not None:
141                self.CONTENT = csmlfrag.text
142        for item in csmlfrag.items():
143            if item[0] in self.ATTRIBUTES:
144                setattr(self, item[0], item[1])
145        # self.CHILDREN (recursive - calls the fromXML method of children
146        for frag in csmlfrag[:]:
147            #for each child element
148            #iterate through expected CHILDREN
149            for child in self.CHILDREN:
150                #get element name
151                if len(self.CHILDREN[child])>=3:
152                    ename = self.CHILDREN[child][2]
153                    ename2 = self.CHILDREN[child][0]
154                   
155                else:
156                    ename = self.CHILDREN[child][0] 
157                    ename2=None
158                #if there are options, then find the name that matches the frag
159                if ename2 is not None:
160                    if frag[:] != []:
161                        for subfrag in frag[:]:
162                            etype=None
163                            if type(ename2) is list:
164                                if subfrag.tag in ename2:
165                                    etype=self._getSubstitutionType(subfrag.tag)
166                            if subfrag.tag==ename2:
167                                etype=self.CHILDREN[child][1]
168                            if etype:
169                                childobj=eval(etype)()
170                                if len(self.CHILDREN[child])>=3:
171                                    if frag[:] != []:
172                                        childobj.fromXML(subfrag)
173                                else:
174                                    childobj.fromXML(frag)
175                                #set this object to be an attribute of the 'parent' (self) object
176                                if hasattr(self, child):
177                                    if type(self.__dict__[child]) is not list:
178                                        tmp=self.__dict__[child]
179                                        setattr(self, child, [tmp]) #convert to list
180                                    self.__dict__[child].append(childobj)
181                                else:
182                                    setattr(self, child, childobj)
183                else:
184                    etype=None
185                    if type(ename) is list:
186                        if frag.tag in ename:
187                            etype=self._getSubstitutionType(frag.tag)
188                    elif frag.tag==ename:
189                        etype=self.CHILDREN[child][1]
190                    if etype:
191                        childobj=eval(etype)()
192                        if len(self.CHILDREN[child])>=3:
193                            if frag[:] != []:
194                                childobj.fromXML(frag[0])
195                        else:
196                            childobj.fromXML(frag)
197                        #set this object to be an attribute of the 'parent' (self) object
198                        if hasattr(self, child):
199                            if type(self.__dict__[child]) is not list:
200                                tmp=self.__dict__[child]
201                                setattr(self, child, [tmp]) #convert to list
202                            self.__dict__[child].append(childobj)
203                        else:
204                            setattr(self, child, childobj)
205
206               
207class csString(csElement):
208    def __init__(self, **kwargs):
209        children={}
210        addchildren(self,children)
211   
212class AbstractGML(csElement):
213    def __init__(self, **kwargs):
214        a=[GML('id'), GML('description'), GML('name'), GML('MetaDataProperty')]
215        addatts(self,a)
216
217class AssociationAttributeGroup(csElement):
218    def __init__(self, **kwargs):
219        a =[XLINK('href'),XLINK('role'), XLINK('arcrole'),XLINK('title'), XLINK('show'), XLINK('actuate')] 
220        addatts(self,a)
221
222class SRSReferenceGroup(csElement):
223    def __init__(self, **kwargs):
224        a =['srsName','srsDimension'] 
225        addatts(self,a)
226       
227class SRSInformationGroup(csElement):
228    def __init__(self, **kwargs):
229        a =['uomLabels','axisLabels'] 
230        addatts(self,a)
231
232class ArrayDescriptor(AbstractGML, csElement):
233    def __init__(self,**kwargs):
234        AbstractGML.__init__(self,**kwargs)
235        children={'arraySize':[CSML('arraySize'), 'csString'], 'uom':[CSML('uom'),'csString'], 'numericType':[CSML('numericType'),'csString'], 'regExpTransform':[CSML('regExpTransform'),'csString'], 'numericTransform':[CSML('numericTransform'),'csString']}
236        addchildren(self,children)
237
238class DirectPosition(AbstractGML, SRSReferenceGroup,SRSInformationGroup,csElement):
239    def __init__(self, **kwargs):
240        AbstractGML.__init__(self,**kwargs)
241
242
243class Envelope(AbstractGML,SRSReferenceGroup, csElement):
244    def __init__(self, **kwargs):
245        SRSReferenceGroup.__init__(self,**kwargs)
246        AbstractGML.__init__(self,**kwargs)
247        children={'lowerCorner':[GML('lowerCorner'), 'DirectPosition'],'upperCorner':[GML('upperCorner'), 'DirectPosition']}
248        addchildren(self,children)
249
250
251class AbstractFeature(AbstractGML,csElement):
252    def __init__(self, **kwargs):
253        AbstractGML.__init__(self,**kwargs)
254        children={'boundedBy':[GML('boundedBy'), 'Envelope']}
255        addchildren(self,children)
256
257class AbstractFeatureCollection(AbstractFeature,csElement):
258    def __init__(self, **kwargs):
259        AbstractFeature.__init__(self,**kwargs)
260
261class DomainSet(AbstractGML,AssociationAttributeGroup,csElement):
262    def __init__(self, **kwargs):
263        AbstractGML.__init__(self,**kwargs)
264        addchildren(self,{})
265
266class AggregatedArray(ArrayDescriptor,csElement):
267    def __init__(self, **kwargs):
268        ArrayDescriptor.__init__(self,**kwargs)
269        children={'aggType':[CSML('aggType'),'csString'], 'aggIndex':[CSML('aggIndex'),'csString'],'components':[FILEFORMATS, 'ArrayDescriptor',CSML('component')]}
270        addchildren(self,children)
271#'component':[CSML:('component'), 'ArrayDescriptor', CSML('component')],
272class MeasureOrNullList(AbstractGML,csElement):
273    def __init__(self, **kwargs):
274        AbstractGML.__init__(self,**kwargs)
275        children={}
276        addchildren(self,children)
277       
278class CompositeValue(AbstractGML,csElement):
279    def __init__(self, **kwargs):
280        AbstractGML.__init__(self,**kwargs)
281        children={'measures':[GML('measure'),'csString',GML('valueComponents')]}
282        addchildren(self,children)
283       
284class DataBlock(AbstractGML,csElement):
285    def __init__(self, **kwargs):
286        AbstractGML.__init__(self,**kwargs)
287        children={'doubleOrNullTupleList':[GML('doubleOrNullTupleList'),'csString'],'rangeParameters':[GML('CompositeValue'), 'CompositeValue', GML('rangeParameters')]}
288        addchildren(self,children)
289
290class RangeSet(AbstractGML,AssociationAttributeGroup,csElement):
291    def __init__(self, **kwargs):
292        AbstractGML.__init__(self,**kwargs)
293        children={'quantityList':[GML('QuantityList'), 'MeasureOrNullList'],        'dataBlock':[GML('DataBlock'),'DataBlock'],'arrayDescriptor':[FILEFORMATS, 'ArrayDescriptor'],  'aggregatedArray':[CSML('AggregatedArray'), 'AggregatedArray']}
294        addchildren(self,children)
295
296class MultiPoint(AbstractGML, SRSReferenceGroup,csElement):
297    def __init__(self, **kwargs):
298        AbstractGML.__init__(self,**kwargs)
299        SRSReferenceGroup.__init__(self,**kwargs)
300        children={'pointMember':[GML('pointMember'), 'csString'],'pointMembers':[GML('pointMember'), 'csString']}
301        addchildren(self,children)
302
303class Point(AbstractGML,SRSReferenceGroup,csElement):
304    def __init__(self, **kwargs):
305        AbstractGML.__init__(self,**kwargs)
306        SRSReferenceGroup.__init__(self,**kwargs)
307        children={'pos':[GML('pos'), 'csString'],'coordinates':[GML('coordinates'), 'csString']}
308        addchildren(self,children)
309
310class PointDomain(DomainSet, MultiPoint,csElement):
311    def __init__(self, **kwargs):
312        DomainSet.__init__(self,**kwargs)
313        MultiPoint.__init__(self,**kwargs)
314        children={}
315        addchildren(self,children)
316
317
318class AbstractCoverage(AbstractFeature, csElement):
319    def __init__(self, **kwargs):
320        AbstractFeature.__init__(self,**kwargs)
321
322class AbstractDiscreteCoverage(AbstractCoverage, csElement):
323    def __init__(self, **kwargs):
324        AbstractCoverage.__init__(self,**kwargs)
325        addchildren(self,{})
326       
327class Definition(AbstractGML):
328    def __init__(self, **kwargs):
329        AbstractGML.__init__(self,**kwargs)
330        addchildren(self,{})
331
332class Phenomenon(Definition,AssociationAttributeGroup):
333    def __init__(self, **kwargs):
334        Definition.__init__(self,**kwargs)
335        AssociationAttributeGroup.__init__(self,**kwargs)
336        children = {'':[CSML(''), '']}   
337        addchildren(self,children)
338       
339class SpatialOrTemporalPositionList(AbstractGML,csElement):
340    def __init__(self, **kwargs):
341        AbstractGML.__init__(self,**kwargs)
342        children={'coordinateList':[CSML('coordinateList'),'csString'], 'timePositionList':[CSML('timePositionList'),'csString']}
343        addchildren(self,children)
344
345class GridOrdinateDescription(AbstractGML,csElement):
346    def __init__(self, **kwargs):
347        AbstractGML.__init__(self,**kwargs)
348        children={'coordAxisLabel':[CSML('coordAxisLabel'), 'csString'], 'coordAxisValues':[CSML('SpatialOrTemporalPositionList'),'SpatialOrTemporalPositionList',CSML('coordAxisValues')], 'gridAxesSpanned':[CSML('gridAxesSpanned'), 'csString'], 'sequenceRule':[CSML('sequenceRule'),'SequenceRuleType']}
349        addchildren(self,children)
350
351class SequenceRuleType(csElement):
352    def __init__(self, **kwargs):
353        a=['axisOrder']
354        addatts(self,a)
355        children={}
356        addchildren(self,children)
357
358class GridPointDescription(AbstractGML,csElement):
359    def __init__(self, **kwargs):
360        AbstractGML.__init__(self,**kwargs)
361        children={'posList':[CSML('posList'),'csString'],'sequenceRule':[CSML('sequenceRule'),'SequenceRuleType']}
362        addchildren(self,children)
363
364
365class GridCoordinatesTable(AbstractGML,csElement):
366    def __init__(self,**kwargs):
367        '''the additional value '1' in the children dictionary is used to signify that the elements must be nested as:
368        <gml:ordinate>
369            <gml:GridOrdinateDescription>
370        </gml:ordinate>
371        <gml:ordinate>
372            <gml:GridOrdinateDescription>
373        </gml:ordinate>
374       
375        not as:
376        <gml:ordinate>
377            <gml:GridOrdinateDescription>
378            <gml:GridOrdinateDescription>
379        </gml:ordinate> '''
380        AbstractGML.__init__(self,**kwargs)
381        children={'gridOrdinates':[CSML('GridOrdinateDescription'), 'GridOrdinateDescription',CSML('gridOrdinate'),1], 'gridPoints':[CSML('GridPointDescription'),'GridPointDescription',CSML('gridPoints')]}
382        addchildren(self,children)
383       
384class ReferenceableGrid(AbstractGML, AssociationAttributeGroup, csElement):
385    def __init__(self, **kwargs):
386        AbstractGML.__init__(self,**kwargs)
387        AssociationAttributeGroup.__init__(self,**kwargs)
388        children={'coordTransformTable':[CSML('GridCoordinatesTable'), 'GridCoordinatesTable', CSML('coordTransformTable')]}
389        addchildren(self,children)
390        a=['dimension']
391        addatts(self,a)
392
393class ReferenceableGridCoverage(AbstractDiscreteCoverage, csElement):
394    def __init__(self, **kwargs):
395        AbstractDiscreteCoverage.__init__(self,**kwargs)
396        children={'referenceableGridDomain':[CSML('ReferenceableGrid'),'ReferenceableGrid' ,CSML('referenceableGridDomain') ]}
397        addchildren(self,children)
398       
399class AlternatePointCoverage(AbstractDiscreteCoverage, csElement):
400    def __init__(self, **kwargs):
401        AbstractDiscreteCoverage.__init__(self,**kwargs)
402        children={'alternatePointDomain':[GML('Point'),'Point', CSML('alternatePointDomain')], 'rangeSet':[GML('rangeSet'), 'RangeSet'],'coverageFunction':[GML('coverageFunction'),'csString']}
403        addchildren(self,children)
404        print self.CHILDREN
405
406class PointCoverage(AbstractDiscreteCoverage, csElement):
407    def __init__(self, **kwargs):
408        AbstractDiscreteCoverage.__init__(self,**kwargs)
409        children={'pointDomain':[CSML('PointDomain'),'PointDomain' ,CSML('pointDomain') ], 'rangeSet':[GML('rangeSet'), 'RangeSet'],'coverageFunction':[GML('coverageFunction'),'csString']}
410        addchildren(self,children)
411
412class GridSeriesDomain(ReferenceableGrid, DomainSet, csElement):
413    def __init__(self, **kwargs):
414        DomainSet.__init__(self,**kwargs)
415        ReferenceableGrid.__init__(self,**kwargs)
416        children={}
417        addchildren(self,children)
418       
419class GridSeriesCoverage(AbstractDiscreteCoverage,csElement):
420    def __init__(self, **kwargs):
421        AbstractDiscreteCoverage.__init__(self,**kwargs)
422        children={'gridSeriesDomain':[CSML('GridSeriesDomain'),'GridSeriesDomain' ,CSML('gridSeriesDomain') ], 'rangeSet':[GML('rangeSet'), 'RangeSet'],'coverageFunction':[GML('coverageFunction'),'csString']}
423        addchildren(self,children)
424
425class AlternatePointFeature(AbstractFeature, csElement):
426    def __init__(self, **kwargs):
427        AbstractFeature.__init__(self,**kwargs)
428        children={'location':[CSML('location'), 'csString'],'time':[CSML('time'), 'csString'], 'value':[CSML('AlternatePointCoverage'), 'AlternatePointCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
429        addchildren(self,children)
430
431class PointFeature(AbstractFeature, csElement):
432    def __init__(self, **kwargs):
433        AbstractFeature.__init__(self,**kwargs)
434        children={'location':[CSML('location'), 'csString'],'time':[CSML('time'), 'csString'], 'value':[CSML('PointCoverage'), 'PointCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
435        addchildren(self,children)
436
437class PointCollectionFeature(AbstractFeature, csElement):
438    def __init__(self, **kwargs):
439        AbstractFeature.__init__(self,**kwargs)
440        children={'time':[CSML('time'), 'csString'], 'value':[CSML('MultiPointCoverage'), 'MultiPointCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
441        addchildren(self,children)
442
443class PointSeriesFeature(AbstractFeature, csElement):
444    def __init__(self, **kwargs):
445        AbstractFeature.__init__(self,**kwargs)
446        children={'location':[CSML('location'), 'csString'], 'value':[CSML('PointSeriesCoverage'), 'PointSeriesCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
447        addchildren(self,children)
448
449
450class GridFeature(AbstractFeature, csElement):
451    def __init__(self, **kwargs):
452        AbstractFeature.__init__(self,**kwargs)
453        children={'time':[CSML('time'), 'csString'], 'value':[CSML('ReferenceableGridCoverage'), 'ReferenceableGridCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
454        addchildren(self,children)
455
456class GridSeriesFeature(AbstractFeature, csElement):
457    def __init__(self, **kwargs):
458        AbstractFeature.__init__(self,**kwargs)
459        children={'value':[CSML('GridSeriesCoverage'), 'GridSeriesCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
460        addchildren(self,children)
461
462class ProfileFeature(AbstractFeature, csElement):
463    def __init__(self, **kwargs):
464        AbstractFeature.__init__(self,**kwargs)
465        children={'time':[CSML('time'), 'csString'],'location':[CSML('location'), 'csString'], 'value':[CSML('ProfileCoverage'), 'ProfileCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
466        addchildren(self,children)
467
468class ProfileSeriesFeature(AbstractFeature, csElement):
469    def __init__(self, **kwargs):
470        AbstractFeature.__init__(self,**kwargs)
471        children={'location':[CSML('location'), 'csString'], 'value':[CSML('ProfileSeriesCoverage'), 'ProfileSeriesCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
472        addchildren(self,children)
473
474class RaggedProfileSeriesFeature(AbstractFeature, csElement):
475    def __init__(self, **kwargs):
476        AbstractFeature.__init__(self,**kwargs)
477        children={'location':[CSML('location'), 'csString'], 'profileLength':[CSML('profileLength'), 'csString'],'value':[CSML('ProfileSeriesCoverage'), 'ProfileSeriesCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
478        addchildren(self,children)
479
480class RaggedSectionFeature(AbstractFeature, csElement):
481    def __init__(self, **kwargs):
482        AbstractFeature.__init__(self,**kwargs)
483        children={'stationLocations':[CSML('stationLocations'), 'csString'], 'stationTimes':[CSML('stationTimes'), 'csString'],'profileLength':[CSML('profileLength'),'csString'],'value':[CSML('RaggedSectionCoverage'), 'RaggedSectionCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
484        addchildren(self,children)
485
486class SectionFeature(AbstractFeature, csElement):
487    def __init__(self, **kwargs):
488        AbstractFeature.__init__(self,**kwargs)
489        children={'stationLocations':[CSML('stationLocations'), 'csString'], 'stationTimes':[CSML('stationTimes'), 'csString'],'value':[CSML('SectionCoverage'), 'SectionCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
490        addchildren(self,children)
491
492class ScanningRadarFeature(AbstractFeature, csElement):
493    def __init__(self, **kwargs):
494        AbstractFeature.__init__(self,**kwargs)
495        children={'elevation':[CSML('elevation'), 'csString'], 'value':[CSML('ScanningRadarCoverage'), 'ScanningRadarCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
496        addchildren(self,children)
497
498class SwathFeature(AbstractFeature, csElement):
499    def __init__(self, **kwargs):
500        AbstractFeature.__init__(self,**kwargs)
501        children={'eqCrossLon':[CSML('eqCrossLon'), 'csString'],'eqCrossTime':[CSML('eqCrossTime'), 'csString'], 'value':[CSML('ReferenceableGridCoverage'), 'ReferenceableGridCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
502        addchildren(self,children)
503
504class TrajectoryFeature(AbstractFeature, csElement):
505    def __init__(self, **kwargs):
506        AbstractFeature.__init__(self,**kwargs)
507        children={'value':[CSML('TrajectoryCoverage'), 'TrajectoryCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
508        addchildren(self,children)
509
510class FileExtract(ArrayDescriptor, csElement):
511    def __init__(self,**kwargs):
512        ArrayDescriptor.__init__(self,**kwargs)
513        children= {'fileName':[CSML('fileName'), 'csString']}
514        addchildren(self,children)
515
516class NetCDFExtract(FileExtract, csElement):
517    def __init__(self,**kwargs):
518        FileExtract.__init__(self, **kwargs)
519        children={'variableName':[CSML('variableName'), 'csString']}
520        addchildren(self,children)
521
522class NASAAmesExtract(FileExtract, csElement):
523    def __init__(self,**kwargs):
524        FileExtract.__init__(self, **kwargs)
525        children={'variableName':[CSML('variableName'), 'csString'], 'index':[CSML('index'),'csString']}
526        addchildren(self,children)
527
528class FeatureCollection(AbstractFeatureCollection,csElement):
529    def __init__(self,**kwargs):
530        AbstractFeatureCollection.__init__(self,**kwargs)
531        children={'members':[[CSML('GridFeature'), CSML('GridSeriesFeature'),CSML('PointFeature'),CSML('AlternatePointFeature')], 'AbstractFeature', GML('featureMembers')]}
532        addchildren(self,children)
533    def _getSubstitutionType(self,tag):
534        if tag==CSML('GridFeature'):
535            return 'GridFeature'
536        elif tag==CSML('GridSeriesFeature'):
537            return 'GridSeriesFeature'
538        elif tag==CSML('PointFeature'):
539            return 'PointFeature'
540        elif tag==CSML('AlternatePointFeature'):
541            return 'AlternatePointFeature'
542        else: return 'AbstractFeature'
543    def _getReverseSubsType(self, typename):
544        if typename== 'GridFeature':
545            return CSML('GridFeature')
546        elif typename == 'PointFeature':
547            return CSML('PointFeature')
548        elif typename == 'AlternatePointFeature':
549            return CSML('AlternatePointFeature')
550        else: return CSML('AbstractFeature')
551
552       
553class Dataset(AbstractGML, csElement):   
554    ''' Dataset class, needed as root of tree'''
555    def __init__(self, **kwargs):
556        AbstractGML.__init__(self,**kwargs)
557        children = {'featureCollection':[GML('FeatureCollection') ,'FeatureCollection'],'fileExtracts':[FILEFORMATS, 'ArrayDescriptor']}
558        addchildren(self,children)
559    def toXML(self):
560        csmlfrag=ET.Element(CSML('Dataset'))
561        csElement.toXML(self, csmlfrag)
562        return csmlfrag
563    def _getSubstitutionType(self,tag):
564        if tag==CSML('NetCDFExtract'):
565            return 'NetCDFExtract'
566        elif tag==CSML('NASAAmesExtract'):
567            return 'NASAAmesExtract'
568        else: return 'ArrayDescriptor'
569    def _getReverseSubsType(self, typename):
570        if typename== 'NetCDFExtract':
571            return CSML('NetCDFExtract')
572        elif typename == 'NASAAmesExtract':
573            return CSML('NASAAmesExtract')
574        else: return CSML('ArrayDescriptor')
575       
576        return typename
577
578def main():
579    '''round trip for testing purposes:'''
580    print '\n'
581    tree=ET.ElementTree(file='test.xml')
582    ds=Dataset()
583    ds.fromXML(tree.getroot())
584    csmltree=ds.toXML()
585    #print csmltree
586   
587    csmlout=parser_extra.PrettyPrint(csmltree)
588    csmlout=parser_extra.removeInlineNS(csmlout)
589    print '\n %s'% csmlout
590    #for member in ds.featureCollection.members:
591      #print dir(member.value)
592       
593
594if __name__=='__main__':
595    main()
Note: See TracBrowser for help on using the repository browser.