/[thuban]/branches/WIP-pyshapelib-bramz/Thuban/Model/load.py
ViewVC logotype

Diff of /branches/WIP-pyshapelib-bramz/Thuban/Model/load.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 737 by jonathan, Fri Apr 25 09:12:17 2003 UTC revision 874 by jonathan, Fri May 9 16:31:12 2003 UTC
# Line 20  import xml.sax.handler Line 20  import xml.sax.handler
20  from xml.sax import make_parser, ErrorHandler, SAXNotRecognizedException  from xml.sax import make_parser, ErrorHandler, SAXNotRecognizedException
21    
22  from Thuban import _  from Thuban import _
 from Thuban.common import *  
23    
24  from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \  from Thuban.Model.table import FIELDTYPE_INT, FIELDTYPE_DOUBLE, \
25       FIELDTYPE_STRING       FIELDTYPE_STRING
# Line 30  from Thuban.Model.map import Map Line 29  from Thuban.Model.map import Map
29  from Thuban.Model.layer import Layer  from Thuban.Model.layer import Layer
30  from Thuban.Model.color import Color  from Thuban.Model.color import Color
31  from Thuban.Model.proj import Projection  from Thuban.Model.proj import Projection
32    from Thuban.Model.range import Range
33  from Thuban.Model.classification import Classification, \  from Thuban.Model.classification import Classification, \
34      ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \      ClassGroupDefault, ClassGroupSingleton, ClassGroupRange, ClassGroupMap, \
35      ClassGroupProperties      ClassGroupProperties
# Line 76  class XMLReader(xml.sax.handler.ContentH Line 76  class XMLReader(xml.sax.handler.ContentH
76    
77      def __init__(self):      def __init__(self):
78          self.chars = ''          self.chars = ''
         self.__parser = None  
79          self.__directory = ""          self.__directory = ""
80          self.__dispatchers = {}          self.__dispatchers = {}
81    
# Line 91  class XMLReader(xml.sax.handler.ContentH Line 90  class XMLReader(xml.sax.handler.ContentH
90              self.__directory = os.path.dirname(filename)              self.__directory = os.path.dirname(filename)
91              self.__file = open(filename)              self.__file = open(filename)
92    
93          if self.__parser is None:          parser = make_parser()
94              self.__parser = make_parser()          parser.setContentHandler(self)
95              self.__parser.setContentHandler(self)          parser.setErrorHandler(ErrorHandler())
96              self.__parser.setErrorHandler(ErrorHandler())          parser.setFeature(xml.sax.handler.feature_namespaces, 1)
97              self.__parser.setFeature(xml.sax.handler.feature_namespaces, 1)  
98            #
99              #          # Well, this isn't pretty, but it appears that if you
100              # Well, this isn't pretty, but it appears that if you          # use Python 2.2 without the site-package _xmlplus then
101              # use Python 2.2 without the site-package _xmlplus then          # the following will fail, and without them it will work.
102              # the following will fail, and without them it will work.          # However, if you do have the site-package and you don't
103              # However, if you do have the site-package and you don't          # call these functions, the reader raises an exception
104              # call these functions, the reader raises an exception          #
105              #          # The reason we set these to 0 in the first place is
106              # The reason we set these to 0 in the first place is          # because there is an unresolved issue with external
107              # because there is an unresolved issue with external          # entities causing an exception in the reader
108              # entities causing an exception in the reader          #
109              #          try:
110              try:              parser.setFeature(xml.sax.handler.feature_validation,0)
111                  self.__parser.setFeature(xml.sax.handler.feature_validation,0)              parser.setFeature(xml.sax.handler.feature_external_ges,0)
112                  self.__parser.setFeature(xml.sax.handler.feature_external_ges,0)              parser.setFeature(xml.sax.handler.feature_external_pes,0)
113                  self.__parser.setFeature(xml.sax.handler.feature_external_pes,0)          except SAXNotRecognizedException:
114              except SAXNotRecognizedException:              pass
                 pass  
115    
116          self.__parser.parse(self.__file)          parser.parse(self.__file)
117    
118          self.close()          self.close()
119    
# Line 167  class XMLReader(xml.sax.handler.ContentH Line 165  class XMLReader(xml.sax.handler.ContentH
165          if method_name is not None and method_name[1] is not None:          if method_name is not None and method_name[1] is not None:
166              getattr(self, method_name[1])(name, qname)              getattr(self, method_name[1])(name, qname)
167    
168        def encode(self, str):
169            """Assume that str is in Unicode and encode it into Latin1.
170            
171            If str is None, return None
172            """
173    
174            if str is not None:
175                return str.encode("latin1")
176            else:
177                return None
178    
179  class SessionLoader(XMLReader):  class SessionLoader(XMLReader):
180    
181      def __init__(self):      def __init__(self):
# Line 193  class SessionLoader(XMLReader): Line 202  class SessionLoader(XMLReader):
202               'label'         : ("start_label",          None)})               'label'         : ("start_label",          None)})
203    
204      def start_session(self, name, qname, attrs):      def start_session(self, name, qname, attrs):
205          self.theSession = Session(attrs.get((None, 'title'), None))          self.theSession = Session(self.encode(attrs.get((None, 'title'), None)))
206    
207      def end_session(self, name, qname):      def end_session(self, name, qname):
208          pass          pass
# Line 201  class SessionLoader(XMLReader): Line 210  class SessionLoader(XMLReader):
210      def start_map(self, name, qname, attrs):      def start_map(self, name, qname, attrs):
211          """Start a map."""          """Start a map."""
212          self.aMap = Map(attrs.get((None, 'title'), None))          self.aMap = Map(attrs.get((None, 'title'), None))
         self.__projReceiver = self.aMap  
213    
214      def end_map(self, name, qname):      def end_map(self, name, qname):
215          self.theSession.AddMap(self.aMap)          self.theSession.AddMap(self.aMap)
216          self.__projReceiver = None          self.aMap = None
217    
218      def start_projection(self, name, qname, attrs):      def start_projection(self, name, qname, attrs):
219            self.ProjectionName = self.encode(attrs.get((None, 'name'), None))
220          self.ProjectionParams = [ ]          self.ProjectionParams = [ ]
221    
222      def end_projection(self, name, qname):      def end_projection(self, name, qname):
223          self.__projReceiver.SetProjection(Projection(self.ProjectionParams))          if self.aLayer is not None:
224                obj = self.aLayer
225            elif self.aMap is not None:
226                obj = self.aMap
227            else:
228                assert False, "projection tag out of context"
229                pass
230    
231            obj.SetProjection(
232                Projection(self.ProjectionParams, self.ProjectionName))
233    
234      def start_parameter(self, name, qname, attrs):      def start_parameter(self, name, qname, attrs):
235          s = attrs.get((None, 'value'))          s = attrs.get((None, 'value'))
# Line 225  class SessionLoader(XMLReader): Line 243  class SessionLoader(XMLReader):
243          attrs which may be a dictionary as well as the normal SAX attrs          attrs which may be a dictionary as well as the normal SAX attrs
244          object and bind it to self.aLayer.          object and bind it to self.aLayer.
245          """          """
246          title = attrs.get((None, 'title'), "")          title = self.encode(attrs.get((None, 'title'), ""))
247          filename = attrs.get((None, 'filename'), "")          filename = attrs.get((None, 'filename'), "")
248          filename = os.path.join(self.GetDirectory(), filename)          filename = os.path.join(self.GetDirectory(), filename)
249            visible  = attrs.get((None, 'visible'), "true")
250          fill = parse_color(attrs.get((None, 'fill'), "None"))          fill = parse_color(attrs.get((None, 'fill'), "None"))
251          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))          stroke = parse_color(attrs.get((None, 'stroke'), "#000000"))
252          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))          stroke_width = int(attrs.get((None, 'stroke_width'), "1"))
253          self.aLayer = layer_class(title,          self.aLayer = layer_class(title,
254                                    self.theSession.OpenShapefile(filename),                                    self.theSession.OpenShapefile(filename),
255                                    fill = fill, stroke = stroke,                                    fill = fill, stroke = stroke,
256                                    lineWidth = stroke_width)                                    lineWidth = stroke_width,
257                                      visible = visible != "false")
         self.__projReceiver = self.aLayer  
258    
259      def end_layer(self, name, qname):      def end_layer(self, name, qname):
260          self.aMap.AddLayer(self.aLayer)          self.aMap.AddLayer(self.aLayer)
261          self.__projReceiver = None          self.aLayer = None
262    
263      def start_classification(self, name, qname, attrs):      def start_classification(self, name, qname, attrs):
264          field = attrs.get((None, 'field'), None)          field = attrs.get((None, 'field'), None)
# Line 262  class SessionLoader(XMLReader): Line 280  class SessionLoader(XMLReader):
280    
281          self.aLayer.GetClassification().SetField(field)          self.aLayer.GetClassification().SetField(field)
282    
   
283      def end_classification(self, name, qname):      def end_classification(self, name, qname):
284          pass          pass
285    
286      def start_clnull(self, name, qname, attrs):      def start_clnull(self, name, qname, attrs):
287          self.cl_group = ClassGroupDefault()          self.cl_group = ClassGroupDefault()
288          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))
289          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
290    
291      def end_clnull(self, name, qname):      def end_clnull(self, name, qname):
# Line 279  class SessionLoader(XMLReader): Line 296  class SessionLoader(XMLReader):
296      def start_clpoint(self, name, qname, attrs):      def start_clpoint(self, name, qname, attrs):
297          attrib_value = attrs.get((None, 'value'), "0")          attrib_value = attrs.get((None, 'value'), "0")
298    
         #try:  
             #value  = Str2Num(attrib_value)  
         #except:  
             #value  = attrib_value  
   
299          value = self.conv(attrib_value)          value = self.conv(attrib_value)
300    
301          self.cl_group = ClassGroupSingleton(value)          self.cl_group = ClassGroupSingleton(value)
302          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(self.encode(attrs.get((None, 'label'), "")))
303          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
304    
305    
# Line 298  class SessionLoader(XMLReader): Line 310  class SessionLoader(XMLReader):
310    
311      def start_clrange(self, name, qname, attrs):      def start_clrange(self, name, qname, attrs):
312    
313            range = attrs.get((None, 'range'), None)
314            # for backward compatibility (min/max are not saved)
315            min   = attrs.get((None, 'min'), None)
316            max   = attrs.get((None, 'max'), None)
317    
318          try:          try:
319              min = self.conv(attrs.get((None, 'min'), "0"))              if range is not None:
320              max = self.conv(attrs.get((None, 'max'), "0"))                  self.cl_group = ClassGroupRange(Range(range))
321              #min = Str2Num(attrs.get((None, 'min'), "0"))              elif min is not None and max is not None:
322              #max = Str2Num(attrs.get((None, 'max'), "0"))                  self.cl_group = ClassGroupRange(self.conv(min), self.conv(max))
323                else:
324                    self.cl_group = ClassGroupRange(Range(None))
325    
326          except ValueError:          except ValueError:
327              raise ValueError(_("Classification range is not a number!"))              raise ValueError(_("Classification range is not a number!"))
328    
         self.cl_group = ClassGroupRange(min, max)  
329          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))          self.cl_group.SetLabel(attrs.get((None, 'label'), ""))
330          self.cl_prop = ClassGroupProperties()          self.cl_prop = ClassGroupProperties()
331    
# Line 339  class SessionLoader(XMLReader): Line 358  class SessionLoader(XMLReader):
358      def start_label(self, name, qname, attrs):      def start_label(self, name, qname, attrs):
359          x = float(attrs[(None, 'x')])          x = float(attrs[(None, 'x')])
360          y = float(attrs[(None, 'y')])          y = float(attrs[(None, 'y')])
361          text = attrs[(None, 'text')]          text = self.encode(attrs[(None, 'text')])
362          halign = attrs[(None, 'halign')]          halign = attrs[(None, 'halign')]
363          valign = attrs[(None, 'valign')]          valign = attrs[(None, 'valign')]
364          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)          self.aLayer.AddLabel(x, y, text, halign = halign, valign = valign)
Legend:
Removed from v.737  
changed lines
  Added in v.874
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26