package com.se.nsl.service; import cn.hutool.core.io.FileUtil; import com.alibaba.fastjson.JSON; import com.se.nsl.config.PropertiesConfig; import com.se.nsl.domain.dto.*; import com.se.nsl.domain.po.DataPo; import com.se.nsl.domain.po.PondingPo; import com.se.nsl.domain.vo.BuildingDepthVo; import com.se.nsl.helper.ComHelper; import com.se.nsl.helper.GdalHelper; import com.se.nsl.helper.ShpHelper; import lombok.extern.slf4j.Slf4j; import org.gdal.gdal.Band; import org.gdal.gdal.Dataset; import org.gdal.gdal.gdal; import org.gdal.gdalconst.gdalconstConstants; import org.gdal.ogr.*; import org.springframework.stereotype.Service; import org.springframework.util.CollectionUtils; import javax.annotation.Resource; import java.awt.*; import java.awt.image.BufferedImage; import java.io.*; import java.math.BigDecimal; import java.math.RoundingMode; import java.text.ParseException; import java.text.SimpleDateFormat; import java.util.*; import java.util.List; import java.util.concurrent.CopyOnWriteArrayList; @Slf4j @Service @SuppressWarnings("ALL") public class ResultService { @Resource PropertiesConfig config; public static final double MAX_X_OFFSET = 0; public static final SimpleDateFormat SDF = new SimpleDateFormat("yyyy-MM-dd HH:mm"); public void process(DataPo data) throws Exception { String basePath = config.getInPath() + File.separator + data.getInPath() + File.separator; ResultDto dto = new ResultDto( data.getInPath(), basePath + config.getTerrainFile(), basePath + config.getBuildingFile(), basePath + config.getWaterPath(), basePath + config.getFlowPath(), config.getInPath(), config.getOutPath(), data.getEpsg()); LayerDto layer = new LayerDto(config.getVer(), data.getEpsg(), config.getSizes()); process(dto, layer); } public void process(ResultDto dto, LayerDto layer) throws Exception { try { copeTerrain(dto, layer); copeBuilding(dto); List buildings = copeWater(dto, layer); copeFlow(dto, layer); copeLayerJson(dto, layer); copeRainFallJson(dto); copeBuildingDepthJson(dto, buildings); } finally { File dir = new File(dto.getTemp()); if (dir.exists()) { FileUtil.del(dir); } } } public void copeTerrain(ResultDto dto, LayerDto layer) { Dataset ds = null; try { ds = gdal.Open(dto.getTerrainFile(), gdalconstConstants.GA_ReadOnly); if (null == ds || 0 == ds.getRasterCount()) return; setTerrainInfo(ds, layer); createTerrainPng(dto, ds, layer); setWaterInfo(dto, layer); } finally { if (null != ds) ds.delete(); } } public void setTerrainInfo(Dataset ds, LayerDto layer) { Geometry minPoint = GdalHelper.getMinPoint(ds); Geometry maxPoint = GdalHelper.getMaxPoint(ds); double minx = ComHelper.getMinVal(minPoint.GetX(0), 10000000); double miny = ComHelper.getMinVal(minPoint.GetY(0), 10000000); double maxx = ComHelper.getMaxVal(maxPoint.GetX(0) + MAX_X_OFFSET, 10000000); double maxy = ComHelper.getMaxVal(maxPoint.GetY(0), 10000000); //layer.setExtension(new ExtensionDto(minx, miny, maxx, maxy, Double.MAX_VALUE, Double.MIN_VALUE)); Band band = ds.GetRasterBand(1); double[] mm = new double[2]; band.ComputeRasterMinMax(mm, 0); //layer.getTerrain().setHeight(getMinVal(mm[0], 1000), getMaxVal(mm[1], 1000)); layer.setExtension(new ExtensionDto(minx, miny, maxx, maxy, mm[0], mm[1])); } public void createTerrainPng(ResultDto dto, Dataset ds, LayerDto layer) { String terrainPath = dto.getOutPath() + File.separator + "terrain"; File f = new File(terrainPath); if (!f.exists() || !f.isDirectory()) f.mkdirs(); for (int[] sizes : layer.getTerrain().getSize()) { String tif = dto.getTemp() + File.separator + "terrain_" + sizes[0] + "_" + sizes[1] + ".tif"; ComHelper.cutAndResample(ds, tif, sizes[0], sizes[1], layer, false); if (!new File(tif).exists()) continue; String png = terrainPath + File.separator + sizes[0] + "_" + sizes[1] + ".png"; terrain2Png(layer, tif, png, sizes[0], sizes[1]); } } public void terrain2Png(LayerDto layer, String tif, String png, int width, int height) { Dataset ds = null; try { ds = gdal.Open(tif, gdalconstConstants.GA_ReadOnly); if (null == ds || 0 == ds.getRasterCount()) return; Band band = ds.GetRasterBand(1); float[] buffer = new float[width * height]; band.ReadRaster(0, 0, width, height, buffer); layer.getTerrain().getVals().put(width + "_" + height, buffer); BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB); double differ = layer.getExtension().getMaxHeight() - layer.getExtension().getMinHeight(); double minHeight = layer.getExtension().getMinHeight(); for (int x = 0; x < width; x++) { for (int y = 0; y < height; y++) { int offset = x + y * width; if (Float.isNaN(buffer[offset]) || buffer[offset] < -999 || buffer[offset] < minHeight) continue; int r = 0; int g; int b; if (buffer[offset] - layer.getExtension().getMaxHeight() > 0) { g = b = 255; } else { int val = (int) ((buffer[offset] - minHeight) / differ * 65535); g = val / 256; b = val % 256; } Color color = new Color(r, g, b, 127); image.setRGB(x, y, color.getRGB()); } } ComHelper.savePng(image, png); } finally { if (null != ds) ds.delete(); } } public void copeBuilding(ResultDto dto) { Driver driver = null; DataSource dataSource = null; Layer shpLayer = null; try { driver = ogr.GetDriverByName("ESRI shapefile"); if (null == driver) return; dataSource = driver.Open(dto.getBuildingFile()); if (null == dataSource) return; shpLayer = dataSource.GetLayer(0); if (shpLayer.GetFeatureCount() < 1) return; do { Feature f = shpLayer.GetNextFeature(); if (null == f) break; String id = f.GetFieldAsString(config.getBuildingKey()); BuildingDto building = new BuildingDto(id, f.GetGeometryRef()); dto.getBuildingList().add(building); } while (true); } finally { if (null != shpLayer) shpLayer.delete(); if (null != dataSource) dataSource.delete(); if (null != driver) driver.delete(); } } public void setWaterInfo(ResultDto dto, LayerDto layer) { List files = getFiles(dto.getWaterPath(), ".tif"); layer.getWaters().setFiles(files); if (null == files || files.isEmpty()) return; setWaterData(layer, files); setWaterHeight(layer, files); } public List copeWater(ResultDto dto, LayerDto layer) { List files = layer.getWaters().getFiles(); if (files.isEmpty() || files.size() != layer.getWaters().getData().size()) return Collections.emptyList(); processWaters(dto, files, layer); return processBuilding(dto, files, layer); } public List getFiles(String path, String suffix) { List files = new ArrayList<>(); ComHelper.getFiles(files, new File(path), suffix); files.sort(String::compareToIgnoreCase); return files; } public void setWaterData(LayerDto layer, List files) { Calendar calendar = Calendar.getInstance(); calendar.setTime(new Date()); calendar.set(Calendar.MILLISECOND, 0); for (String file : files) { String fileName = ComHelper.getNameWithExt(file); int year = Integer.parseInt(fileName.substring(0, 4)); int month = Integer.parseInt(fileName.substring(4, 6)); int day = Integer.parseInt(fileName.substring(6, 8)); int hour = Integer.parseInt(fileName.substring(8, 10)); int minute = Integer.parseInt(fileName.substring(10, 12)); int second = Integer.parseInt(fileName.substring(12, 14)); calendar.set(Calendar.YEAR, year); calendar.set(Calendar.MONTH, month); calendar.set(Calendar.DAY_OF_MONTH, day); calendar.set(Calendar.HOUR_OF_DAY, hour); calendar.set(Calendar.MINUTE, minute); calendar.set(Calendar.SECOND, second); layer.getWaters().getData().add(calendar.getTime().getTime()); } layer.getDuration().setStart(layer.getWaters().getData().get(0)); layer.getDuration().setEnd(layer.getWaters().getData().get(layer.getWaters().getData().size() - 1)); } public void setWaterHeight(LayerDto layer, List files) { files.parallelStream().forEach(file -> { Dataset ds = null; try { ds = gdal.Open(file, gdalconstConstants.GA_ReadOnly); if (null == ds || 0 == ds.getRasterCount()) return; double[] mm = new double[2]; ds.GetRasterBand(1).ComputeRasterMinMax(mm, 0); //layer.getExtension().setHeight(mm[0], mm[1]); layer.getWaters().setHeight(mm[0], mm[1]); } finally { if (null != ds) ds.delete(); } }); layer.getExtension().setMaxHeight(layer.getExtension().getMaxHeight() + layer.getWaters().getMaxHeight()); layer.getExtension().setMaxHeight(ComHelper.getMaxVal(layer.getExtension().getMaxHeight(), 1000000)); layer.getExtension().setMinHeight(ComHelper.getMaxVal(layer.getExtension().getMinHeight(), 1000000)); layer.getExtension().setDiffer(); } public void processWaters(ResultDto dto, List files, LayerDto layer) { for (int i = 0, c = files.size(); i < c; i++) { Dataset ds = null; try { ds = gdal.Open(files.get(i), gdalconstConstants.GA_ReadOnly); if (null == ds || 0 == ds.getRasterCount()) return; if (null == ds.GetSpatialRef()) ds.SetSpatialRef(dto.getSpatialReference()); createWaterPng(dto, ds, layer, layer.getWaters().getData().get(i)); if (config.getCopyTif()) copyWaterTif(dto, ds, layer.getWaters().getData().get(i)); createVectors(dto, ds, layer.getWaters().getData().get(i)); } finally { if (null != ds) ds.delete(); } } } public void createWaterPng(ResultDto dto, Dataset ds, LayerDto layer, long ticks) { String waterPath = dto.getOutPath() + File.separator + "waters" + File.separator + ticks; File dir = new File(waterPath); if (!dir.exists() || !dir.isDirectory()) dir.mkdirs(); for (int[] sizes : layer.getTerrain().getSize()) { String fileName = ComHelper.getNameWithExt(ds.GetDescription()) + "_" + sizes[0] + "_" + sizes[1]; String tif = dto.getTemp() + File.separator + fileName + ".tif"; ComHelper.cutAndResample(ds, tif, sizes[0], sizes[1], layer, false); if (!new File(tif).exists()) continue; String png = waterPath + File.separator + sizes[0] + "_" + sizes[1] + ".png"; water2Png(layer, tif, png, sizes[0], sizes[1]); } } public void water2Png(LayerDto layer, String tif, String png, int width, int height) { Dataset ds = null; try { ds = gdal.Open(tif, gdalconstConstants.GA_ReadOnly); if (null == ds || 0 == ds.getRasterCount()) return; Band band = ds.GetRasterBand(1); float[] buffer = new float[width * height]; band.ReadRaster(0, 0, width, height, buffer); BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB); //double differ = layer.getWaters().getMaxHeight() - layer.getWaters().getMinHeight(), minHeight = layer.getWaters().getMinHeight(); double differ = layer.getExtension().getDiffer(); double maxHeight = layer.getExtension().getMaxHeight(); double minHeight = layer.getExtension().getMinHeight(); float[] ts = layer.getTerrain().getVals().get(width + "_" + height); for (int x = 0; x < width; x++) { for (int y = 0; y < height; y++) { int offset = x + y * width; //if (Float.isNaN(buffer[offset]) || buffer[offset] < -999 || buffer[offset] < minHeight) continue; if (Float.isNaN(buffer[offset]) || buffer[offset] <= 0 || Float.isNaN(ts[offset])) continue; //double X = transform[0] + x * transform[1] + y * transform[2]; //double Y = transform[3] + x * transform[4] + y * transform[5]; //BuildingDto building = intersects(dto, X, Y); //if (null != building) continue; int r = 0; int g; int b; if (buffer[offset] + ts[offset] > maxHeight) { g = b = 255; } else { int val = (int) ((buffer[offset] + ts[offset] - minHeight) / differ * 65535); g = val / 256; b = val % 256; } Color color = new Color(r, g, b, 127); image.setRGB(x, y, color.getRGB()); } } ComHelper.savePng(image, png); } finally { if (null != ds) ds.delete(); } } public List processBuilding(ResultDto dto, List files, LayerDto layer) { List list = new CopyOnWriteArrayList<>(); for (int i = 0, c = files.size(); i < c; i++) { Dataset ds = null; try { ds = gdal.Open(files.get(i), gdalconstConstants.GA_ReadOnly); if (null == ds || 0 == ds.getRasterCount()) continue; if (null == ds.GetSpatialRef()) ds.SetSpatialRef(dto.getSpatialReference()); copeBuildingDepth(dto, ds, layer.getWaters().getData().get(i), list); } finally { if (null != ds) ds.delete(); } } return list; } public void copeBuildingDepth(ResultDto dto, Dataset ds, long ticks, List list) { double[] transform = ds.GetGeoTransform(); double minX = transform[0]; double pixelWidth = transform[1]; double maxY = transform[3]; double pixelHeight = Math.abs(transform[5]); for (BuildingDto building : dto.getBuildingList()) { Double val = getValue(ds, building, minX, maxY, pixelWidth, pixelHeight); list.add(new BuildingDepthVo(building.getId(), ticks, val)); } } public Double getValue(Dataset ds, BuildingDto building, double minX, double maxY, double pixelWidth, double pixelHeight) { double[] env = new double[4]; building.getGeom().GetEnvelope(env); int startX = (int) Math.floor((env[0] - minX) / pixelWidth); int endX = (int) Math.floor((env[1] - minX) / pixelWidth); int startY = (int) Math.floor((maxY - env[3]) / Math.abs(pixelHeight)); int endY = (int) Math.floor((maxY - env[2]) / Math.abs(pixelHeight)); if (startX < 0) startX = 0; if (startY < 0) startY = 0; if (endX > ds.getRasterXSize()) endX = ds.getRasterXSize(); if (endY > ds.getRasterYSize()) endY = ds.getRasterYSize(); if (endX - startX < 1 || endY - startY < 1) return null; int width = endX - startX; int height = endY - startY; double[] pixelValues = new double[width * height]; ds.GetRasterBand(1).ReadRaster(startX, startY, width, height, pixelValues); Double val = Arrays.stream(pixelValues).max().getAsDouble(); return ComHelper.isValid(val) ? val : null; } public void copyWaterTif(ResultDto dto, Dataset ds, long ticks) { String source = ds.GetDescription(); String target = dto.getOutPath() + File.separator + "waters" + File.separator + ticks + File.separator + "water.tif"; FileUtil.copyFile(source, target); } public void createVectors(ResultDto dto, Dataset ds, Long ticks) { String filePath = dto.getOutPath() + File.separator + "waters" + File.separator + ticks + File.separator + "polygonize.geojson"; ShpHelper.polygonize2Geojson(ds, filePath); List geometries = getGeometries(filePath); if (CollectionUtils.isEmpty(geometries)) return; List list = copePonding(ds, geometries); if (CollectionUtils.isEmpty(list)) return; try { filePath = dto.getOutPath() + File.separator + "waters" + File.separator + ticks + File.separator + "water.json"; ComHelper.writeJson(filePath, JSON.toJSONString(list)); } catch (Exception ex) { log.error(ex.getMessage(), ex); } } public List getGeometries(String filePath) { if (!FileUtil.exist(filePath)) return Collections.emptyList(); Driver driver = null; DataSource dataSource = null; org.gdal.ogr.Layer layer = null; try { driver = ogr.GetDriverByName("GeoJSON"); if (null == driver) return Collections.emptyList(); DataSource ds = driver.Open(filePath); if (null == ds) return Collections.emptyList(); layer = ds.GetLayer(0); List list = new ArrayList<>(); for (long i = 0, d = layer.GetFeatureCount(); i < d; i++) { Feature f = layer.GetFeature(i); if (f.GetFieldAsDouble("val") < -999) continue; list.add(f.GetGeometryRef()); } return list; } catch (Exception ex) { log.error(ex.getMessage(), ex); return Collections.emptyList(); } finally { GdalHelper.delete(layer, dataSource, driver); } } public List copePonding(Dataset ds, List geometries) { double[] transform = ds.GetGeoTransform(); List list = new ArrayList<>(); for (Geometry geometry : geometries) { PointDto point = getValues(ds, geometry, transform); if (null == point) continue; list.add(new PondingPo(geometry, point)); } return list; } public PointDto getValues(Dataset ds, Geometry g, double[] transform) { double[] env = new double[4]; g.GetEnvelope(env); int startX = (int) Math.floor((env[0] - transform[0]) / transform[1]); int endX = (int) Math.floor((env[1] - transform[0]) / transform[1]); int startY = (int) Math.floor((transform[3] - env[3]) / Math.abs(transform[5])); int endY = (int) Math.floor((transform[3] - env[2]) / Math.abs(transform[5])); if (startX < 0) startX = 0; if (startY < 0) startY = 0; if (endX > ds.getRasterXSize()) endX = ds.getRasterXSize(); if (endY > ds.getRasterYSize()) endY = ds.getRasterYSize(); if (endX - startX < 1 || endY - startY < 1) return null; int width = endX - startX; int height = endY - startY; double[] pixelValues = new double[width * height]; ds.GetRasterBand(1).ReadRaster(startX, startY, width, height, pixelValues); Double val = Arrays.stream(pixelValues).max().getAsDouble(); return new PointDto(g.Centroid().GetX(), g.Centroid().GetY(), ComHelper.isValid(val) ? val : 0.0); } public void copeFlow(ResultDto dto, LayerDto layer) { List files = getFiles(dto.getFlowPath(), ".tif"); if (null == files || files.size() != layer.getWaters().getData().size()) return; for (int i = 0, c = files.size(); i < c; i++) { Dataset ds = null; try { ds = gdal.Open(files.get(i), gdalconstConstants.GA_ReadOnly); if (null == ds || 0 == ds.getRasterCount()) return; if (null == ds.GetSpatialRef()) ds.SetSpatialRef(dto.getSpatialReference()); createFlowPng(dto, ds, layer, layer.getWaters().getData().get(i)); } finally { if (null != ds) ds.delete(); } } } public void createFlowPng(ResultDto dto, Dataset ds, LayerDto layer, long ticks) { String flowPath = dto.getOutPath() + File.separator + "flows" + File.separator + ticks; File dir = new File(flowPath); if (!dir.exists() || !dir.isDirectory()) dir.mkdirs(); for (int[] sizes : layer.getTerrain().getSize()) { String name = ComHelper.getNameWithExt(ds.GetDescription()) + "_" + sizes[0] + "_" + sizes[1]; String tif = dto.getTemp() + File.separator + name + ".tif"; ComHelper.cutAndResample(ds, tif, sizes[0], sizes[1], layer, false); if (!new File(tif).exists()) continue; String png = flowPath + File.separator + sizes[0] + "_" + sizes[1] + ".png"; vxyTif2Png(tif, png, sizes[0], sizes[1]); } } public void vxyTif2Png(String tif, String png, int width, int height) { Dataset ds = null; try { ds = gdal.Open(tif, gdalconstConstants.GA_ReadOnly); if (null == ds || 0 == ds.getRasterCount()) return; float[] vxBuffer = new float[width * height]; float[] vyBuffer = new float[width * height]; ds.GetRasterBand(1).ReadRaster(0, 0, width, height, vxBuffer); ds.GetRasterBand(2).ReadRaster(0, 0, width, height, vyBuffer); createFlowPng(vxBuffer, vyBuffer, png, width, height); } finally { if (null != ds) ds.delete(); } } public void createFlowPng(float[] vxBuffer, float[] vyBuffer, String png, int width, int height) { BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB); for (int x = 0; x < width; x++) { for (int y = 0; y < height; y++) { int offset = x + y * width; float fx = ComHelper.getFloatValue(vxBuffer[offset]); float fy = ComHelper.getFloatValue(vyBuffer[offset]); if (Float.isNaN(fx) && Float.isNaN(fy) || (fx == 0 && fy == 0)) continue; fx = Float.isNaN(fx) ? 0 : fx; fy = Float.isNaN(fy) ? 0 : fy; double dr = Math.sqrt(Math.pow(fx, 2) + Math.pow(fy, 2)); int r = (int) (dr / 4 * 255); int g = (int) ((fx / dr * 0.5 + 0.5) * 255); int b = (int) ((fy / dr * 0.5 + 0.5) * 255); Color color = new Color(ComHelper.getSafeValue(r), ComHelper.getSafeValue(g), ComHelper.getSafeValue(b), 127); image.setRGB(x, y, color.getRGB()); } } ComHelper.savePng(image, png); } public void copeLayerJson(ResultDto dto, LayerDto layer) throws IOException { layer.getWaters().setFiles(null); layer.getTerrain().setEpsg(null); layer.getExtension().setDiffer(null); String json = JSON.toJSONString(layer); String filePath = dto.getOutPath() + File.separator + "layer.json"; ComHelper.writeJson(filePath, json); } public void copeRainFallJson(ResultDto dto) throws IOException, ParseException { String rainGageFilePath = config.getInPath() + File.separator + dto.getServiceName() + File.separator + "RainGage.dat"; String filePath = dto.getOutPath() + File.separator + "rainfall.json"; Map map = new LinkedHashMap<>(); try (FileReader fr = new FileReader(rainGageFilePath); BufferedReader br = new BufferedReader(fr)) { String line = br.readLine(); while ((line = br.readLine()) != null) { String[] rainFall = line.split(" "); if (rainFall.length < 7) continue; String sdt = rainFall[1] + "-" + rainFall[2] + "-" + rainFall[3] + " " + rainFall[4] + ":" + rainFall[5]; BigDecimal num = new BigDecimal(rainFall[6]); map.put("" + SDF.parse(sdt).getTime(), num.setScale(2, RoundingMode.HALF_UP).doubleValue()); } } String json = JSON.toJSONString(map); ComHelper.writeJson(filePath, json); } public void copeBuildingDepthJson(ResultDto dto, List list) throws IOException { if (CollectionUtils.isEmpty(list)) return; String json = JSON.toJSONString(list); String filePath = dto.getOutPath() + File.separator + "building.json"; ComHelper.writeJson(filePath, json); } }