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

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

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

revision 696 by jonathan, Wed Apr 16 16:39:49 2003 UTC revision 1189 by jonathan, Thu Jun 12 17:01:13 2003 UTC
# Line 6  Line 6 
6  # Read the file COPYING coming with Thuban for details.  # Read the file COPYING coming with Thuban for details.
7    
8  import os  import os
9    import os.path
10    
11  import Thuban  import Thuban
12  from Thuban.Model.load import XMLProcessor, load_xmlfile  from Thuban import _
13  from Thuban.Model.save import XMLSaver  
14    from Thuban.Lib.fileutil import get_application_dir
15    
16    from Thuban.Model.xmlreader import XMLReader
17    from Thuban.Model.xmlwriter import XMLWriter
18  from Thuban.Model.proj import Projection, ProjFile  from Thuban.Model.proj import Projection, ProjFile
19    from xml.sax import SAXParseException
20    
21  projdir = \  projdir = \
22          os.path.join(Thuban.__path__[0], os.pardir, "Resources", "Projections")          os.path.join(Thuban.__path__[0], os.pardir, "Resources", "Projections")
23    
 usrdir  = os.path.expanduser("~/.thuban")  
24    
25  PROJ_EXT = ".proj"  PROJ_EXT = ".proj"
26    
27  def ReadProjFile(filename):  has_gdal_support = lambda: False
28      """Read a single .proj file and return a ProjFile object."""  try:
29        import gdal
30        """Return True if the gdal library is available."""
31        has_gdal_support = lambda: True
32    except ImportError:
33        pass
34    
35    def read_proj_file(filename):
36        """Read a single .proj file and return a ProjFile object.
37        
38        Raises IOError if the file cannot be opened.
39        Raises OSError if the file cannot be read.
40        Raises SAXParseException if the file is not valid XML.
41        """
42    
43      projs = []      handler = ProjFileReader()
44      try:      handler.read(filename)
45          handler = ProjFileReader(filename)      return handler.GetProjFile()
46    
47          load_xmlfile(filename, handler)  def write_proj_file(pf):
48          pf = handler.GetProjFile()      """Write a single .proj file
     except OSError:  
         pass  
49    
50      return pf      Raises IOError if the file cannot be written.
51        """
52    
 def WriteProjFile(pf):  
53      saver = ProjFileSaver(pf)      saver = ProjFileSaver(pf)
54      saver.write(pf.GetFileName())      saver.write(pf.GetFilename())
55    
56  def GetProjections(dir):  def get_proj_files(dir):
57      """Read all .proj files in the given directory and      """Read all .proj files in the given directory and
58      return a list of ProjFile objects.      return a list of ProjFile objects.
59      """      """
60    
61      list = []      list = []
62      try:      try:
63          for file in filter(lambda s: s.endswith(PROJ_EXT), os.listdir(dir)):          dirlist = os.listdir(dir)
             filename = os.path.join(dir, file)  
             list.extend(ReadProjFile(filename))  
64      except OSError:      except OSError:
65          pass          pass # if we can't get a directory listing just return []
66        else:
67            for file in filter(lambda s: s.endswith(PROJ_EXT), dirlist):
68                try:
69                    filename = os.path.join(dir, file)
70                    list.append(read_proj_file(filename))
71                except (OSError, IOError, SAXParseException):
72                    pass # just move onto the next file
73    
74      return list      return list
75    
76  def GetSuppliedProjections():  def get_system_proj_files():
77      """Return a list of ProjFile objects from files that are      """Return a list of ProjFile objects from files that are
78      supplied by Thuban.      supplied by Thuban.
79    
80        If no files could not be opened return a list with one
81        empty projection file set to store data in the default file.
82      """      """
83      return GetProjections(projdir)      filename = os.path.join(projdir, "defaults.proj")
84        try:
85            return [read_proj_file(filename)]
86        except (OSError, IOError, SAXParseException):
87            return [ProjFile(filename)]
88    
89  def GetUserProjections():  def get_user_proj_files():
90      """Return a list of ProjFile objects from files that are user-defined."""      """Return a list of ProjFile objects from files that are user-defined.
     return GetProjections(usrdir)  
91    
92  class ProjFileReader(XMLProcessor):      If no files could not be opened return a list with one
93        empty projection file set to store data in the default file.
94        """
95    
96      def __init__(self, filename):      usrdir  = get_application_dir()
97          XMLProcessor.__init__(self, os.path.dirname(filename))      filename = os.path.join(usrdir, "user.proj")
98                try:
99          self.pf = ProjFile(filename)          return [read_proj_file(filename)]
100        except (OSError, IOError, SAXParseException):
101            return [ProjFile(filename)]
102    
103    class ProjFileReader(XMLReader):
104    
105        def __init__(self):
106            XMLReader.__init__(self)
107            self.__pf = ProjFile("")
108    
109            XMLReader.AddDispatchers(self,
110                {'projection': ("start_projection", "end_projection"),
111                 'parameter':  ("start_parameter", None)})
112                
113        def read(self, file_or_filename):
114            XMLReader.read(self, file_or_filename)
115    
116            self.__pf.SetFilename(XMLReader.GetFilename(self))
117                    
118      def start_projection(self, name, qname, attrs):      def start_projection(self, name, qname, attrs):
119          self.params = []          self.params = []
120          self.name = attrs.get((None, 'name'), "Unknown")          self.name = self.encode(attrs.get((None, 'name'), _("Unknown")))
     XMLProcessor.start_dispatcher['projection'] = "start_projection"  
121    
122      def end_projection(self, name, qname):      def end_projection(self, name, qname):
123          self.pf.Add(Projection(self.params, self.name))          self.__pf.Add(Projection(self.params, self.name))
     XMLProcessor.end_dispatcher['projection'] = "end_projection"  
124    
125      def start_parameter(self, name, qname, attrs):      def start_parameter(self, name, qname, attrs):
126          s = attrs.get((None, 'value'))          s = attrs.get((None, 'value'))
127          s = str(s) # we can't handle unicode in proj          s = str(s) # we can't handle unicode in proj
128          self.params.append(s)          self.params.append(s)
     XMLProcessor.start_dispatcher['parameter'] = "start_parameter"  
129    
130      def GetProjFile(self):      def GetProjFile(self):
131          return self.pf          return self.__pf
   
132    
133  class ProjFileSaver(XMLSaver):  class ProjFileSaver(XMLWriter):
134    
135      def __init__(self, pf):      def __init__(self, pf):
136          XMLSaver.__init__(self)          XMLWriter.__init__(self)
137          self.pf = pf          self.__pf = pf
138    
139      def write(self, file_or_filename):      def write(self, file_or_filename):
140          XMLSaver.write(self, file_or_filename)          XMLWriter.write(self, file_or_filename)
141    
142          self.write_header("projfile", "thuban.dtd")          self.write_header("projfile", "projfile.dtd")
143          self.write_projfile(self.pf)          self.write_projfile(self.__pf)
144          self.close()          self.close()
145    
146      def write_projfile(self, pf):      def write_projfile(self, pf):
# Line 107  class ProjFileSaver(XMLSaver): Line 150  class ProjFileSaver(XMLSaver):
150          for p in pf.GetProjections():          for p in pf.GetProjections():
151              self.open_element("projection", {"name": p.GetName()})              self.open_element("projection", {"name": p.GetName()})
152    
153              for param in p.GetParameters():              for param in p.GetAllParameters():
154                  self.write_element("parameter", {"value": param})                  self.write_element("parameter", {"value": param})
155    
156              self.close_element("projection")              self.close_element("projection")
Legend:
Removed from v.696  
changed lines
  Added in v.1189
[email protected]
 ViewVC Help
Powered by ViewVC 1.1.26