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

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

added AlternatePointFeature and fixed Point Feature

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
317class AbstractCoverage(AbstractFeature, csElement):
318    def __init__(self, **kwargs):
319        AbstractFeature.__init__(self,**kwargs)
320
321class AbstractDiscreteCoverage(AbstractCoverage, csElement):
322    def __init__(self, **kwargs):
323        AbstractCoverage.__init__(self,**kwargs)
324        addchildren(self,{})
325       
326class Definition(AbstractGML):
327    def __init__(self, **kwargs):
328        AbstractGML.__init__(self,**kwargs)
329        addchildren(self,{})
330
331class Phenomenon(Definition,AssociationAttributeGroup):
332    def __init__(self, **kwargs):
333        Definition.__init__(self,**kwargs)
334        AssociationAttributeGroup.__init__(self,**kwargs)
335        children = {'':[CSML(''), '']}   
336        addchildren(self,children)
337       
338class SpatialOrTemporalPositionList(AbstractGML,csElement):
339    def __init__(self, **kwargs):
340        AbstractGML.__init__(self,**kwargs)
341        children={'coordinateList':[GML('coordinateList'),'csString'], 'timePositionList':[CSML('timePositionList'),'csString']}
342        addchildren(self,children)
343
344class GridOrdinateDescription(AbstractGML,csElement):
345    def __init__(self, **kwargs):
346        AbstractGML.__init__(self,**kwargs)
347        children={'coordAxisLabel':[GML('coordAxisLabel'), 'csString'], 'coordAxisValues':[GML('SpatialOrTemporalPositionList'),'SpatialOrTemporalPositionList',GML('coordAxisValues')], 'gridAxesSpanned':[GML('gridAxesSpanned'), 'csString'], 'sequenceRule':[GML('sequenceRule'),'SequenceRuleType']}
348        addchildren(self,children)
349
350class SequenceRuleType(csElement):
351    def __init__(self, **kwargs):
352        a=['axisOrder']
353        addatts(self,a)
354        children={}
355        addchildren(self,children)
356
357class GridPointDescription(AbstractGML,csElement):
358    def __init__(self, **kwargs):
359        AbstractGML.__init__(self,**kwargs)
360        children={'posList':[GML('posList'),'csString'],'sequenceRule':[GML('sequenceRule'),'SequenceRuleType']}
361        addchildren(self,children)
362
363
364class GridCoordinatesTable(AbstractGML,csElement):
365    def __init__(self,**kwargs):
366        '''the additional value '1' in the children dictionary is used to signify that the elements must be nested as:
367        <gml:ordinate>
368            <gml:GridOrdinateDescription>
369        </gml:ordinate>
370        <gml:ordinate>
371            <gml:GridOrdinateDescription>
372        </gml:ordinate>
373       
374        not as:
375        <gml:ordinate>
376            <gml:GridOrdinateDescription>
377            <gml:GridOrdinateDescription>
378        </gml:ordinate> '''
379        AbstractGML.__init__(self,**kwargs)
380        children={'gridOrdinates':[GML('GridOrdinateDescription'), 'GridOrdinateDescription',GML('gridOrdinate'),1], 'gridPoints':[GML('GridPointDescription'),'GridPointDescription',GML('gridPoints')]}
381        addchildren(self,children)
382       
383class ReferenceableGrid(AbstractGML, AssociationAttributeGroup, csElement):
384    def __init__(self, **kwargs):
385        AbstractGML.__init__(self,**kwargs)
386        AssociationAttributeGroup.__init__(self,**kwargs)
387        children={'coordTransformTable':[GML('GridCoordinatesTable'), 'GridCoordinatesTable', GML('coordTransformTable')]}
388        addchildren(self,children)
389        a=['dimension']
390        addatts(self,a)
391
392class ReferenceableGridCoverage(AbstractDiscreteCoverage, csElement):
393    def __init__(self, **kwargs):
394        AbstractDiscreteCoverage.__init__(self,**kwargs)
395        children={'referenceableGridDomain':[CSML('ReferenceableGrid'),'ReferenceableGrid' ,CSML('referenceableGridDomain') ]}
396        addchildren(self,children)
397       
398class AlternatePointCoverage(AbstractDiscreteCoverage, csElement):
399    def __init__(self, **kwargs):
400        AbstractDiscreteCoverage.__init__(self,**kwargs)
401        children={'alternatePointDomain':[GML('Point'),'Point', CSML('alternatePointDomain')], 'rangeSet':[GML('rangeSet'), 'RangeSet'],'coverageFunction':[GML('coverageFunction'),'csString']}
402        addchildren(self,children)
403        print self.CHILDREN
404
405class PointCoverage(AbstractDiscreteCoverage, csElement):
406    def __init__(self, **kwargs):
407        AbstractDiscreteCoverage.__init__(self,**kwargs)
408        children={'pointDomain':[CSML('PointDomain'),'PointDomain' ,CSML('pointDomain') ], 'rangeSet':[GML('rangeSet'), 'RangeSet'],'coverageFunction':[GML('coverageFunction'),'csString']}
409        addchildren(self,children)
410
411class AlternatePointFeature(AbstractFeature, csElement):
412    def __init__(self, **kwargs):
413        AbstractFeature.__init__(self,**kwargs)
414        children={'location':[CSML('location'), 'csString'],'time':[CSML('time'), 'csString'], 'value':[CSML('AlternatePointCoverage'), 'AlternatePointCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
415        addchildren(self,children)
416
417class PointFeature(AbstractFeature, csElement):
418    def __init__(self, **kwargs):
419        AbstractFeature.__init__(self,**kwargs)
420        children={'location':[CSML('location'), 'csString'],'time':[CSML('time'), 'csString'], 'value':[CSML('PointCoverage'), 'PointCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
421        addchildren(self,children)
422
423class PointCollectionFeature(AbstractFeature, csElement):
424    def __init__(self, **kwargs):
425        AbstractFeature.__init__(self,**kwargs)
426        children={'time':[CSML('time'), 'csString'], 'value':[CSML('MultiPointCoverage'), 'MultiPointCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
427        addchildren(self,children)
428
429class PointSeriesFeature(AbstractFeature, csElement):
430    def __init__(self, **kwargs):
431        AbstractFeature.__init__(self,**kwargs)
432        children={'location':[CSML('location'), 'csString'], 'value':[CSML('PointSeriesCoverage'), 'PointSeriesCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
433        addchildren(self,children)
434
435
436class GridFeature(AbstractFeature, csElement):
437    def __init__(self, **kwargs):
438        AbstractFeature.__init__(self,**kwargs)
439        children={'time':[CSML('time'), 'csString'], 'value':[CSML('ReferenceableGridCoverage'), 'ReferenceableGridCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
440        addchildren(self,children)
441
442class GridSeriesFeature(AbstractFeature, csElement):
443    def __init__(self, **kwargs):
444        AbstractFeature.__init__(self,**kwargs)
445        children={'value':[CSML('GridSeriesCoverage'), 'GridSeriesCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
446        addchildren(self,children)
447
448class ProfileFeature(AbstractFeature, csElement):
449    def __init__(self, **kwargs):
450        AbstractFeature.__init__(self,**kwargs)
451        children={'time':[CSML('time'), 'csString'],'location':[CSML('location'), 'csString'], 'value':[CSML('ProfileCoverage'), 'ProfileCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
452        addchildren(self,children)
453
454class ProfileSeriesFeature(AbstractFeature, csElement):
455    def __init__(self, **kwargs):
456        AbstractFeature.__init__(self,**kwargs)
457        children={'location':[CSML('location'), 'csString'], 'value':[CSML('ProfileSeriesCoverage'), 'ProfileSeriesCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
458        addchildren(self,children)
459
460class RaggedProfileSeriesFeature(AbstractFeature, csElement):
461    def __init__(self, **kwargs):
462        AbstractFeature.__init__(self,**kwargs)
463        children={'location':[CSML('location'), 'csString'], 'profileLength':[CSML('profileLength'), 'csString'],'value':[CSML('ProfileSeriesCoverage'), 'ProfileSeriesCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
464        addchildren(self,children)
465
466class RaggedSectionFeature(AbstractFeature, csElement):
467    def __init__(self, **kwargs):
468        AbstractFeature.__init__(self,**kwargs)
469        children={'stationLocations':[CSML('stationLocations'), 'csString'], 'stationTimes':[CSML('stationTimes'), 'csString'],'profileLength':[CSML('profileLength'),'csString'],'value':[CSML('RaggedSectionCoverage'), 'RaggedSectionCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
470        addchildren(self,children)
471
472class SectionFeature(AbstractFeature, csElement):
473    def __init__(self, **kwargs):
474        AbstractFeature.__init__(self,**kwargs)
475        children={'stationLocations':[CSML('stationLocations'), 'csString'], 'stationTimes':[CSML('stationTimes'), 'csString'],'value':[CSML('SectionCoverage'), 'SectionCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
476        addchildren(self,children)
477
478class ScanningRadarFeature(AbstractFeature, csElement):
479    def __init__(self, **kwargs):
480        AbstractFeature.__init__(self,**kwargs)
481        children={'elevation':[CSML('elevation'), 'csString'], 'value':[CSML('ScanningRadarCoverage'), 'ScanningRadarCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
482        addchildren(self,children)
483
484class SwathFeature(AbstractFeature, csElement):
485    def __init__(self, **kwargs):
486        AbstractFeature.__init__(self,**kwargs)
487        children={'eqCrossLon':[CSML('eqCrossLon'), 'csString'],'eqCrossTime':[CSML('eqCrossTime'), 'csString'], 'value':[CSML('ReferenceableGridCoverage'), 'ReferenceableGridCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
488        addchildren(self,children)
489
490class TrajectoryFeature(AbstractFeature, csElement):
491    def __init__(self, **kwargs):
492        AbstractFeature.__init__(self,**kwargs)
493        children={'value':[CSML('TrajectoryCoverage'), 'TrajectoryCoverage', CSML('value')], 'parameter':[CSML('parameter'), 'Phenomenon']}
494        addchildren(self,children)
495
496class FileExtract(ArrayDescriptor, csElement):
497    def __init__(self,**kwargs):
498        ArrayDescriptor.__init__(self,**kwargs)
499        children= {'fileName':[CSML('fileName'), 'csString']}
500        addchildren(self,children)
501
502class NetCDFExtract(FileExtract, csElement):
503    def __init__(self,**kwargs):
504        FileExtract.__init__(self, **kwargs)
505        children={'variableName':[CSML('variableName'), 'csString']}
506        addchildren(self,children)
507
508class NASAAmesExtract(FileExtract, csElement):
509    def __init__(self,**kwargs):
510        FileExtract.__init__(self, **kwargs)
511        children={'variableName':[CSML('variableName'), 'csString'], 'index':[CSML('index'),'csString']}
512        addchildren(self,children)
513
514class FeatureCollection(AbstractFeatureCollection,csElement):
515    def __init__(self,**kwargs):
516        AbstractFeatureCollection.__init__(self,**kwargs)
517        children={'members':[[CSML('GridFeature'),CSML('PointFeature'),CSML('AlternatePointFeature')], 'AbstractFeature', GML('featureMembers')]}
518        addchildren(self,children)
519    def _getSubstitutionType(self,tag):
520        if tag==CSML('GridFeature'):
521            return 'GridFeature'
522        elif tag==CSML('PointFeature'):
523            return 'PointFeature'
524        elif tag==CSML('AlternatePointFeature'):
525            return 'AlternatePointFeature'
526        else: return 'AbstractFeature'
527    def _getReverseSubsType(self, typename):
528        if typename== 'GridFeature':
529            return CSML('GridFeature')
530        elif typename == 'PointFeature':
531            return CSML('PointFeature')
532        elif typename == 'AlternatePointFeature':
533            return CSML('AlternatePointFeature')
534        else: return CSML('AbstractFeature')
535
536       
537class Dataset(AbstractGML, csElement):   
538    ''' Dataset class, needed as root of tree'''
539    def __init__(self, **kwargs):
540        AbstractGML.__init__(self,**kwargs)
541        children = {'featureCollection':[GML('FeatureCollection') ,'FeatureCollection'],'fileExtracts':[FILEFORMATS, 'ArrayDescriptor']}
542        addchildren(self,children)
543    def toXML(self):
544        csmlfrag=ET.Element(CSML('Dataset'))
545        csElement.toXML(self, csmlfrag)
546        return csmlfrag
547    def _getSubstitutionType(self,tag):
548        if tag==CSML('NetCDFExtract'):
549            return 'NetCDFExtract'
550        elif tag==CSML('NASAAmesExtract'):
551            return 'NASAAmesExtract'
552        else: return 'ArrayDescriptor'
553    def _getReverseSubsType(self, typename):
554        if typename== 'NetCDFExtract':
555            return CSML('NetCDFExtract')
556        elif typename == 'NASAAmesExtract':
557            return CSML('NASAAmesExtract')
558        else: return CSML('ArrayDescriptor')
559       
560        return typename
561
562def main():
563    '''round trip for testing purposes:'''
564    print '\n'
565    tree=ET.ElementTree(file='test.xml')
566    ds=Dataset()
567    ds.fromXML(tree.getroot())
568    csmltree=ds.toXML()
569    #print csmltree
570   
571    csmlout=parser_extra.PrettyPrint(csmltree)
572    csmlout=parser_extra.removeInlineNS(csmlout)
573    print '\n %s'% csmlout
574    #for member in ds.featureCollection.members:
575      #print dir(member.value)
576       
577
578if __name__=='__main__':
579    main()
Note: See TracBrowser for help on using the repository browser.