Advertisement
Guest User

Untitled

a guest
Mar 23rd, 2017
54
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.88 KB | None | 0 0
  1. from osgeo import gdal
  2. import os, glob, osr, numpy as np
  3. path = 'D:TesisImagenesL8 OLI TIRS'
  4. VRT = "C:/DUMP_PYTHON"
  5. pathout = "C:UsersppDocumentsLANDSATNUEVAS IMGS LANDSATInput ETa"
  6.  
  7. def save_raster ( output_name, dataset, raster_data, driver ,NaN_Value):
  8. """
  9. A function to save a 1-band raster using GDAL to the file indicated
  10. by ``output_name``. It requires a GDAL-accesible dataset to collect
  11. the projection and geotransform.
  12. """
  13. # Open the reference dataset
  14. g = ( dataset )
  15. # Get the Geotransform vector
  16. if raster_data is False:
  17. raster_data=g.ReadAsArray()
  18. geo_transform = g.GetGeoTransform ()
  19. x_size = g.RasterXSize # Raster xsize
  20. y_size = g.RasterYSize # Raster ysize
  21. srs = g.GetProjectionRef () # Projection
  22. NaN_rast = NaN_Value
  23. # raster_data[raster_data == NaN_rast] = 'NaN'
  24. raster_data[raster_data == NaN_rast] = np.NaN
  25. # Need a driver object. By default, we use GeoTIFF
  26. driver = gdal.GetDriverByName ( driver )
  27. dataset_out = driver.Create ( output_name, x_size, y_size, 1,
  28. gdal.GDT_Float32 )
  29. dataset_out.SetGeoTransform ( geo_transform )
  30. dataset_out.SetProjection ( srs )
  31. dataset_out.GetRasterBand ( 1 ).WriteArray (
  32. raster_data.astype(np.float32) )
  33.  
  34.  
  35. #READ DATA
  36. i=672
  37. files= glob.glob(os.path.join(path,'*.tif'))
  38. albedo = gdal.Open(VRT+"/dummy_alb"+str(i)+'.vrt')
  39. save_raster(pathout+'/Albedo_L8_'+files[i][39:46]+'.tif',albedo1,False,"GTiff",99999)
  40.  
  41. i=672
  42. albedo = gdal.Open(VRT+"/dummy_alb"+str(i)+'.vrt')
  43. src_ds = gdal.Open(VRT+"/dummy_alb"+str(i)+'.vrt')
  44. #Open output format driver, see gdal_translate --formats for list
  45. format = "GTiff"
  46. driver = gdal.GetDriverByName( format )
  47. #Output to new format
  48. dst_ds = driver.CreateCopy( pathout+'/Albedo_L8.tif', src_ds, -10 )
  49.  
  50. #Properly close the datasets to flush to disk
  51. dst_ds = None
  52. src_ds = None
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement