From c381ab8acda22f8d3b42d084d59b9b097962c4a0 Mon Sep 17 00:00:00 2001 From: 13693261870 <252740454@qq.com> Date: 星期一, 23 九月 2024 15:23:55 +0800 Subject: [PATCH] 解决编辑bug --- src/main/java/com/se/simu/service/GedbService.java | 96 +++++++++++++++++++++++++++-------------------- 1 files changed, 55 insertions(+), 41 deletions(-) diff --git a/src/main/java/com/se/simu/service/GedbService.java b/src/main/java/com/se/simu/service/GedbService.java index 29e086e..f1b2110 100644 --- a/src/main/java/com/se/simu/service/GedbService.java +++ b/src/main/java/com/se/simu/service/GedbService.java @@ -8,6 +8,7 @@ import com.se.simu.domain.dto.GeField; import com.se.simu.domain.dto.GeFile; import com.se.simu.domain.dto.GeLayer; +import com.se.simu.domain.po.DataPo; import com.se.simu.helper.RsaHelper; import com.se.simu.helper.ShpHelper; import com.se.simu.helper.StringHelper; @@ -82,31 +83,21 @@ @Resource RestTemplate restTemplate; - public boolean test(String bbox, String taskName) throws Exception { + public boolean test(DataPo data) throws Exception { + createPath(inPath + File.separator + data.getInPath()); + String token = getToken(); - GeDb db = getSeDb(token); - db.setBbox(bbox); + GeDb db = connectGedb(token, data); - List<GeLayer> layers = getLayers(token, db); - queryData(token, db, layers); + copeVectors(token, data, db); - String basePath = inPath + File.separator + taskName; - createPath(basePath); - createShps(basePath, layers); - - GeDb fileDb = getFileDb(token); - String fileId = getFileId(token, fileDb.getDbid()); - List<GeFile> files = getFileNames(token, fileDb.getDbid(), fileId); - - String filePath = inPath + File.separator + fileDb.getDbid(); - downloadFiles(token, filePath, files, fileDb.getDbid(), fileId); - clipDemFile(filePath, files, basePath, bbox, 4528); + copeDem(token, data); return true; } - private void createPath(String basePath) { - File f = new File(basePath); + private void createPath(String path) { + File f = new File(path); if (f.exists() && f.isDirectory()) { FileUtil.del(f); } @@ -125,7 +116,7 @@ return data.getStr("token"); } - public String getPassword() throws Exception { + private String getPassword() throws Exception { if (StringHelper.isEmpty(password)) { String key = getPublicKey(); RsaHelper.setPublicKey(key); @@ -135,14 +126,22 @@ return password; } - public String getPublicKey() { + private String getPublicKey() { //{"datetime":"2024-09-12 17:24:38","code":200,"data":"MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQCtFwJCh2taVTEi05C8qT2oG7Y+rDmJhlO4zicpSeRtiro9LsytePeWI7BXM6sfDU0WeKun1izawcfgGkZgnoJuMBluAOKI1tL0uCrR+DreNLqMVtnXHwoWEIk/hGJedDWaf3q22aGDyEB5h9qCq0JklSShP1Ih4ppap4LmgxdTPQIDAQAB"} JSONObject obj = restTemplate.getForObject(host + "account-service/security/publickey", JSONObject.class); return obj.getStr("data"); } - public GeDb getSeDb(String token) { + public GeDb connectGedb(String token, DataPo data) { + GeDb db = getSeDb(token); + db.setBbox(data.getBbox()); + data.setEpsg(db.getEpsg()); + + return db; + } + + private GeDb getSeDb(String token) { Map<String, Object> map = new HashMap<>(1); map.put("token", token); @@ -245,7 +244,15 @@ return data.getJSONArray("features"); } - public void createShps(String basePath, List<GeLayer> layers) throws Exception { + public void copeVectors(String token, DataPo data, GeDb db) throws Exception { + String basePath = inPath + File.separator + data.getInPath(); + + List<GeLayer> layers = getLayers(token, db); + queryData(token, db, layers); + createShps(basePath, layers); + } + + private void createShps(String basePath, List<GeLayer> layers) throws Exception { for (GeLayer layer : layers) { String path = String.format("%s\\%s.shp", basePath, layer.getShpName()); if (!ShpHelper.createShp(path, layer)) { @@ -254,7 +261,17 @@ } } - public GeDb getFileDb(String token) { + public void copeDem(String token, DataPo data) throws Exception { + GeDb fileDb = getFileDb(token); + String fileId = getFileId(token, fileDb.getDbid()); + List<GeFile> files = getFileNames(token, fileDb.getDbid(), fileId); + + String filePath = inPath + File.separator + fileDb.getDbid(); + downloadFiles(token, filePath, files, fileDb.getDbid(), fileId); + clipDemFile(filePath, files, data); + } + + private GeDb getFileDb(String token) { String uri = String.format("%sfile-service/docdb/query/canview?token=%s", host, token); JSONObject obj = restTemplate.getForObject(uri, JSONObject.class); JSONArray data = obj.getJSONArray("data"); @@ -265,7 +282,7 @@ return list.stream().filter(db -> null != db.getName() && db.getName().contains(dbName)).findFirst().orElse(null); } - public String getFileId(String token, String dbid) { + private String getFileId(String token, String dbid) { String uri = String.format("%sfile-service/doc/catagory/file/query?token=%s&dbid=%s&catagory=%s&count=%d&start=%d&like=", host, token, dbid, "image", 9999, 1); @@ -282,7 +299,7 @@ return null; } - public List<GeFile> getFileNames(String token, String dbid, String fileId) { + private List<GeFile> getFileNames(String token, String dbid, String fileId) { String uri = String.format("%sfile-service/doc/cluster/struct/list?token=%s&dbid=%s&cluster_fileid=%s&onlychild=true&folder_stairs=", host, token, dbid, fileId); @@ -292,7 +309,7 @@ return JSONUtil.toList(data, GeFile.class); } - public void downloadFiles(String token, String path, List<GeFile> files, String dbid, String fileId) throws IOException { + private void downloadFiles(String token, String path, List<GeFile> files, String dbid, String fileId) throws IOException { File f = new File(path); if (!f.exists() || !f.isDirectory()) { f.mkdirs(); @@ -333,48 +350,45 @@ is.close(); } - private void clipDemFile(String filePath, List<GeFile> files, String basePath, String bbox, int epsg) throws Exception { - String target = basePath + File.separator + demName; + private void clipDemFile(String filePath, List<GeFile> files, DataPo data) throws Exception { + String target = inPath + File.separator + data.getInPath() + File.separator + demFile; for (GeFile file : files) { if (file.getName().toLowerCase().endsWith(demType)) { String source = filePath + File.separator + file.getName(); - clipDem(source, target, bbox, epsg); + clipDem(source, target, data); break; } } } - private void clipDem(String source, String target, String bbox, int epsg) throws Exception { + private void clipDem(String source, String target, DataPo data) throws Exception { Dataset ds = null; try { ds = gdal.Open(source, gdalconst.GA_ReadOnly); if (null == ds || ds.getRasterCount() < 1 || null == ds.GetSpatialRef()) throw new Exception("DEM鏁版嵁鏃犳晥"); // String bbox = "116.64388473935195,39.884315914604464,116.64754729082588,39.887069143903496"; - String[] bb = bbox.split(","); - - // gdalwarp -ot UInt16 -s_srs EPSG:4326 -t_srs EPSG:2382 -r bilinear -of GTiff -te 116.526854182 40.0481829856 116.532848182 40.0541769856 - // -te_srs EPSG:4326 -co COMPRESS=DEFLATE -co PREDICTOR=1 -co ZLEVEL=6 -co TILED=YES -wo OPTIMIZE_SIZE=TRUE E:\GDALhomework\000002.tif E:/CSDN/warped1.tif - Vector<String> vector = new Vector<>(); //vector.add("-s_srs"); //vector.add("EPSG:" + 4326); vector.add("-t_srs"); - vector.add("EPSG:" + epsg); + vector.add("EPSG:" + data.getEpsg()); vector.add("-r"); vector.add("bilinear"); vector.add("-of"); vector.add("GTiff"); vector.add("-te"); - vector.add(bb[0]); - vector.add(bb[1]); - vector.add(bb[2]); - vector.add(bb[3]); + vector.add(data.getMinx().toString()); + vector.add(data.getMiny().toString()); + vector.add(data.getMaxx().toString()); + vector.add(data.getMaxy().toString()); vector.add("-te_srs"); vector.add("EPSG:" + 4326); - WarpOptions warpOptions = new WarpOptions(vector); + // gdalwarp -ot UInt16 -s_srs EPSG:4326 -t_srs EPSG:2382 -r bilinear -of GTiff -te 116.526854182 40.0481829856 116.532848182 40.0541769856 + // -te_srs EPSG:4326 -co COMPRESS=DEFLATE -co PREDICTOR=1 -co ZLEVEL=6 -co TILED=YES -wo OPTIMIZE_SIZE=TRUE E:\GDALhomework\000002.tif E:/CSDN/warped1.tif + Dataset destDs = gdal.Warp(target, new Dataset[]{ds}, warpOptions); destDs.delete(); } finally { -- Gitblit v1.9.3