Skip to content

GeoSentinel
GeoSentinel
Commit 4e1631d2 authored by Mario Chirinos Colunga's avatar Mario Chirinos Colunga 💬
Browse files
Options

wkt raster

parent daa735f0
Show whitespace changes
Inline Side-by-side
Showing with 29 additions and 29 deletions
geosentinel/rasterWkt.py
View file @ 4e1631d2
...@@ -83,44 +83,44 @@ def rasterWkt(wkt, inputfile, outputfile): ...@@ -83,44 +83,44 @@ def rasterWkt(wkt, inputfile, outputfile):
print("driver") print("driver")
driverTiff = gdal.GetDriverByName('GTiff') driverTiff = gdal.GetDriverByName('GTiff')
output = driverTiff.Create(outputfile, cols, rows, inputImage.RasterCount, gdal.GDT_Byte) output = driverTiff.Create(outputfile, cols, rows, inputImage.RasterCount, gdal.GDT_Byte)
# polygonRaster = driverTiff.Create("polygonTmp.tif", cols, rows, 1, gdal.GDT_Byte) polygonRaster = driverTiff.Create("polygonTmp.tif", cols, rows, 1, gdal.GDT_Byte)
# #Set up polygon raster #Set up polygon raster
# polygonRaster.GetRasterBand(1).SetNoDataValue(-99) polygonRaster.GetRasterBand(1).SetNoDataValue(-99)
# polygonRaster.GetRasterBand(1).WriteArray(np.zeros((rows, cols))) polygonRaster.GetRasterBand(1).WriteArray(np.zeros((rows, cols)))
# polygonRaster.GetRasterBand(1).FlushCache() polygonRaster.GetRasterBand(1).FlushCache()
# polygonRaster.SetGeoTransform(geotransform) polygonRaster.SetGeoTransform(geotransform)
# polygonRaster.SetGeoTransform(geotransform) polygonRaster.SetGeoTransform(geotransform)
# polygonRaster.SetProjection(sr.ExportToWkt()) polygonRaster.SetProjection(sr.ExportToWkt())
# print("Shapefile") print("Shapefile")
# # shapefile # shapefile
# driver = ogr.GetDriverByName("ESRI Shapefile") driver = ogr.GetDriverByName("ESRI Shapefile")
# data_source = driver.CreateDataSource("myShape.shp") data_source = driver.CreateDataSource("myShape.shp")
# print("Burn Polygon") print("Burn Polygon")
# #Burn Polygon #Burn Polygon
# srs = osr.SpatialReference() srs = osr.SpatialReference()
# srs.ImportFromEPSG(4326) srs.ImportFromEPSG(4326)
# layer = data_source.GetLayer() layer = data_source.GetLayer()
## layer = data_source.CreateLayer("wkt", srs, ogr.wkbPolygon) # layer = data_source.CreateLayer("wkt", srs, ogr.wkbPolygon)
## geometry = ogr.CreateGeometryFromWkt(wkt) # geometry = ogr.CreateGeometryFromWkt(wkt)
## feature = ogr.Feature(layer.GetLayerDefn()) # feature = ogr.Feature(layer.GetLayerDefn())
# print("features") print("features")
## feature.SetStyleString("PEN(c:#FF0000,w:5px);") # feature.SetStyleString("PEN(c:#FF0000,w:5px);")
## feature.SetGeometry(geometry) # feature.SetGeometry(geometry)
## layer.CreateFeature(feature) # layer.CreateFeature(feature)
# gdal.RasterizeLayer(polygonRaster, [1], layer, burn_values=[255]) gdal.RasterizeLayer(polygonRaster, [1], layer, burn_values=[255])
# polygonRaster.GetRasterBand(1).FlushCache() polygonRaster.GetRasterBand(1).FlushCache()
## Draw output image # Draw output image
# np_polygon = polygonRaster.GetRasterBand(1).ReadAsArray(0,0,cols,rows) np_polygon = polygonRaster.GetRasterBand(1).ReadAsArray(0,0,cols,rows)
np_polygon = getPolygonArray(inputImage, wkt) # np_polygon = getPolygonArray(inputImage, wkt)
print("Setup") print("Setup")
for b in range(1, output.RasterCount+1): for b in range(1, output.RasterCount+1):
np_input = np_polygon & inputImage.GetRasterBand(b).ReadAsArray(0,0,cols,rows) np_input = np_polygon & inputImage.GetRasterBand(b).ReadAsArray(0,0,cols,rows)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment