sourceProduct = ProductIO.readProduct(sys.argv[1]) b1 = sourceProduct.getBand('reflec_5') b2 = sourceProduct.getBand('reflec_7') b3 = sourceProduct.getBand('reflec_9') w1 = b1.getSpectralWavelength() w2 = b2.getSpectralWavelength() w3 = b3.getSpectralWavelength() a = (w2 - w1) / (w3 - w1) k = 1.03 width = sourceProduct.getSceneRasterWidth() height = sourceProduct.getSceneRasterHeight() targetProduct = Product('FLH_Product', 'FLH_Type', width, height) targetBand = targetProduct.addBand('FLH', ProductData.TYPE_FLOAT32) ProductUtils.copyGeoCoding(sourceProduct, targetProduct) targetProduct.setProductWriter(ProductIO.getProductWriter('GeoTIFF')) targetProduct.writeHeader(String('beampy_flh_output.tif')) r1 = numpy.zeros(width, dtype=numpy.float32) r2 = numpy.zeros(width, dtype=numpy.float32) r3 = numpy.zeros(width, dtype=numpy.float32) print("Writing...") for y in range(height): b1.readPixels(0, y, width, 1, r1) b2.readPixels(0, y, width, 1, r2) b2.readPixels(0, y, width, 1, r3) print("processing line ", y, " of ", height) FLH = r2 - k * (r1 + a * (r3 - r1))
ndviBand = ndviProduct.addBand("ndvi", ProductData.TYPE_FLOAT32) ndviFlagsBand = ndviProduct.addBand("ndvi_flags", ProductData.TYPE_UINT8) writer = ProductIO.getProductWriter("BEAM-DIMAP") ProductUtils.copyGeoCoding(product, ndviProduct) ndviFlagCoding = FlagCoding("ndvi_flags") ndviFlagCoding.addFlag("NDVI_LOW", 1, "NDVI below 0") ndviFlagCoding.addFlag("NDVI_HIGH", 2, "NDVI above 1") group = ndviProduct.getFlagCodingGroup() # print(dir(group)) group.add(ndviFlagCoding) ndviFlagsBand.setSampleCoding(ndviFlagCoding) ndviProduct.setProductWriter(writer) ndviProduct.writeHeader("beampy_ndvi_output.dim") r7 = numpy.zeros(width, dtype=numpy.float32) r10 = numpy.zeros(width, dtype=numpy.float32) print("Writing...") for y in range(height): print("processing line ", y, " of ", height) r7 = b7.readPixels(0, y, width, 1, r7) r10 = b10.readPixels(0, y, width, 1, r10) ndvi = (r10 - r7) / (r10 + r7) ndviBand.writePixels(0, y, width, 1, ndvi) ndviLow = ndvi < 0.0
ndviBand = ndviProduct.addBand('ndvi', ProductData.TYPE_FLOAT32) ndviFlagsBand = ndviProduct.addBand('ndvi_flags', ProductData.TYPE_UINT8) writer = ProductIO.getProductWriter('BEAM-DIMAP') ProductUtils.copyGeoCoding(product, ndviProduct) ndviFlagCoding = FlagCoding('ndvi_flags') ndviFlagCoding.addFlag("NDVI_LOW", 1, "NDVI below 0") ndviFlagCoding.addFlag("NDVI_HIGH", 2, "NDVI above 1") group = ndviProduct.getFlagCodingGroup() #print(dir(group)) group.add(ndviFlagCoding) ndviFlagsBand.setSampleCoding(ndviFlagCoding) ndviProduct.setProductWriter(writer) ndviProduct.writeHeader('beampy_ndvi_output.dim') r7 = numpy.zeros(width, dtype=numpy.float32) r10 = numpy.zeros(width, dtype=numpy.float32) print("Writing...") for y in range(height): print("processing line ", y, " of ", height) r7 = b7.readPixels(0, y, width, 1, r7) r10 = b10.readPixels(0, y, width, 1, r10) ndvi = (r10 - r7) / (r10 + r7) ndviBand.writePixels(0, y, width, 1, ndvi) ndviLow = ndvi < 0.0