Changeset 1046

Timestamp:

Mar 9, 2005, 4:37:00 PM (20 years ago)

Author:

duncan

Message:

added reading and writing to ASCII files

Location:

inundation/ga/storm_surge/pyvolution/coordinate_transforms

Files:

• geo_reference.py (modified) (3 diffs)
• test_geo_reference.py (modified) (1 diff)

inundation/ga/storm_surge/pyvolution/coordinate_transforms/geo_reference.py

@@ -14 +14 @@
                  xllcorner=None,
                  yllcorner=None,
-                 NetCDFObject = None):
+                 NetCDFObject = None,
+                 ASCIIFile = None):
         if not xllcorner == None:
             self.xllcorner = xllcorner #Easting of lower left corner
@@ -24 +25 @@
         if not NetCDFObject == None:
             self.read_NetCDF(NetCDFObject)
+
+        #FIXME think if this should be here    
+        if not ASCIIFile == None:
+            self.read_ASCII(ASCIIFile)
         
     def write_NetCDF(self, outfile):
@@ -46 +51 @@
         assert (type(self.yllcorner) == types.FloatType)
         assert (type(self.zone) == types.IntType)
+
+        
+    def write_ASCII(self, fd):
+        fd.write('# Geo reference' + "\n")
+        fd.write(str(self.zone) + "\n") 
+        fd.write(str(self.xllcorner) + "\n") 
+        fd.write(str(self.yllcorner) + "\n")
+
+    def read_ASCII(self, fd):
+        fd.readline() # remove the title line
+        self.zone = int(fd.readline())
+        self.xllcorner = float(fd.readline())
+        self.yllcorner = float(fd.readline())
+            
+        
+        assert (type(self.xllcorner) == types.FloatType)
+        assert (type(self.yllcorner) == types.FloatType)
+        assert (type(self.zone) == types.IntType)
         
         #false_easting = infile.false_easting[0]
         #false_northing = infile.false_northing[0]
+
+    def change_geo_ref(self, primary_geo_ref,
+                       secondary_geo_ref, secondary_points):
+        #convert into array
+        #add secondary geo ref to points
+        #subtract secondary geo ref from points
+        # return points
+        pass
 
     def __repr__(self):

inundation/ga/storm_surge/pyvolution/coordinate_transforms/test_geo_reference.py

@@ -32 +32 @@
         self.failUnless(g == new_g, 'test_read_write_NetCDF failed')  
         
+    def test_read_write_ASCII(self):
+        from Scientific.IO.NetCDF import NetCDFFile
+        g = Geo_reference(56,1.9,1.9)
+        file_name = tempfile.mktemp(".geo_referenceTest")
+        print "file_name",file_name 
+        fd = open(file_name,'w')
+        g.write_ASCII(fd)
+        fd.close()
+        
+        fd = open(file_name,'r')
+        new_g = Geo_reference(ASCIIFile=fd)
+        fd.close()
+        os.remove(file_name)
+
+        self.failUnless(g == new_g, 'test_read_write_ASCII failed')  
+        
     def test___cmp__(self):
         g = Geo_reference(56,1.9,1.9,)