rgb l2

geosentinel/rasterWkt.py

@@ -35,14 +35,25 @@ def rasterWkt(wkt, inputfile, outputfile):
 	intput = gdal.Open(inputfile)
 	rows, cols, geotransform = intput.RasterYSize, intput.RasterXSize, intput.GetGeoTransform()
 	# Read the input bands as numpy arrays.
-	np_intput = intput.GetRasterBand(1).ReadAsArray(0,0,cols,rows)
-
+	np_intput_1 = intput.GetRasterBand(1).ReadAsArray(0,0,cols,rows)
+	np_intput_2 = intput.GetRasterBand(2).ReadAsArray(0,0,cols,rows)
+	np_intput_3 = intput.GetRasterBand(3).ReadAsArray(0,0,cols,rows)
 
 	driverTiff = gdal.GetDriverByName('GTiff')
 	output = driverTiff.Create(outputfile, cols, rows, 1, gdal.GDT_Byte)
+
 	output.GetRasterBand(1).SetNoDataValue(-99)
-	output.GetRasterBand(1).WriteArray(np_intput)
+	output.GetRasterBand(1).WriteArray(np_intput_1)
 	output.GetRasterBand(1).FlushCache()
+
+	output.GetRasterBand(2).SetNoDataValue(-99)
+	output.GetRasterBand(2).WriteArray(np_intput_2)
+	output.GetRasterBand(2).FlushCache()
+
+	output.GetRasterBand(3).SetNoDataValue(-99)
+	output.GetRasterBand(3).WriteArray(np_intput_3)
+	output.GetRasterBand(3).FlushCache()
+
 	output.SetGeoTransform(geotransform)
 
 	sr = osr.SpatialReference()