package com.moon.server.helper;
|
|
import org.apache.commons.logging.Log;
|
import org.apache.commons.logging.LogFactory;
|
import org.gdal.gdal.Band;
|
import org.gdal.gdal.Driver;
|
import org.gdal.gdal.gdal;
|
import org.gdal.gdal.Dataset;
|
import org.gdal.gdalconst.gdalconstConstants;
|
import org.gdal.ogr.*;
|
import org.gdal.osr.SpatialReference;
|
|
import java.io.File;
|
import java.sql.Time;
|
import java.sql.Timestamp;
|
import java.time.LocalDate;
|
import java.time.LocalDateTime;
|
import java.time.LocalTime;
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
/**
|
* GDAL帮助类
|
* @author WWW
|
*/
|
@SuppressWarnings("ALL")
|
public class GdalHelper {
|
private final static Log log = LogFactory.getLog(GdalHelper.class);
|
|
public static void init(String gdalPath) {
|
// 支持中文路径
|
gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "YES");
|
// 属性表支持中文：CP936
|
gdal.SetConfigOption("SHAPE_ENCODING", "");
|
gdal.SetConfigOption("PGEO_DRIVER_TEMPLATE", "DRIVER=Microsoft Access Driver (*.mdb, *.accdb);DBQ=%s");
|
gdal.SetConfigOption("MDB_DRIVER_TEMPLATE", "DRIVER=Microsoft Access Driver (*.mdb, *.accdb);DBQ=%s");
|
|
// 配置环境变量
|
if (!StringHelper.isEmpty(gdalPath)) {
|
gdal.SetConfigOption("GDAL_DATA", gdalPath + File.separator + "gdal-data");
|
gdal.SetConfigOption("PROJ_LIB", gdalPath + File.separator + "proj7" + File.separator + "share");
|
//System.setProperty("PROJ_LIB", gdalPath + File.separator + "proj7" + File.separator + "share")
|
gdal.SetConfigOption("GDAL_DRIVER_PATH", gdalPath + File.separator + "gdalplugins");
|
|
String path = System.getenv("PATH");
|
if (!path.contains(gdalPath)) {
|
System.setProperty("PATH", path + ";" + gdalPath);
|
}
|
}
|
|
// 注册所有的驱动
|
gdal.AllRegister();
|
ogr.RegisterAll();
|
}
|
|
/**
|
* 读取tif文件
|
*
|
* @param fileName
|
*/
|
public static void readTif(String fileName) {
|
// 读取影像数据
|
Dataset dataset = gdal.Open(fileName, gdalconstConstants.GA_ReadOnly);
|
if (dataset == null) {
|
System.out.println("read fail!");
|
return;
|
}
|
|
// providing various methods for a format specific driver.
|
Driver driver = dataset.GetDriver();
|
System.out.println("driver name: " + driver.getLongName());
|
|
// 读取影像信息
|
int xSize = dataset.getRasterXSize();
|
int ySzie = dataset.getRasterYSize();
|
int rasterCount = dataset.getRasterCount();
|
System.out.println("dataset xSize: " + xSize + ", ySzie = " + ySzie + ", rasterCount = " + rasterCount);
|
|
Band band = dataset.GetRasterBand(1);
|
// the data type of the band.
|
int type = band.GetRasterDataType();
|
System.out.println("data type = " + type + ", " + (type == gdalconstConstants.GDT_Byte));
|
|
// Frees the native resource associated to a Dataset object and close the file.
|
dataset.delete();
|
|
gdal.GDALDestroyDriverManager();
|
}
|
|
/**
|
* 读取shp文件
|
*
|
* @param filePath
|
*/
|
public static void readShp(String filePath) {
|
try {
|
org.gdal.ogr.Driver driver = ogr.GetDriverByName("ESRI shapefile");
|
if (driver == null) {
|
System.out.println(" ESRI shapefile驱动不可用！\n");
|
System.out.println("fail");
|
}
|
|
DataSource dataSource = driver.Open(filePath);
|
Layer layer = dataSource.GetLayer(0);
|
|
String layerName = layer.GetName();
|
System.out.println("图层名称：" + layerName);
|
|
SpatialReference spatialReference = layer.GetSpatialRef();
|
// System.out.println(spatialReference)
|
System.out.println("空间参考坐标系：" + spatialReference.GetAttrValue("AUTHORITY", 0) + spatialReference.GetAttrValue("AUTHORITY", 1));
|
|
double[] layerExtent = layer.GetExtent();
|
System.out.println("图层范围：minx:" + layerExtent[0] + ",maxx:" + layerExtent[1] + ",miny:" + layerExtent[2] + ",maxy:" + layerExtent[3]);
|
|
FeatureDefn featureDefn = layer.GetLayerDefn();
|
Map<String, Object> fieldMap = new HashMap(5);
|
for (int i = 0, count = featureDefn.GetFieldCount(); i < count; i++) {
|
FieldDefn fieldDefn = featureDefn.GetFieldDefn(i);
|
//得到属性字段类型
|
int fieldType = fieldDefn.GetFieldType();
|
String fieldTypeName = fieldDefn.GetFieldTypeName(fieldType);
|
//得到属性字段名称
|
String fieldName = fieldDefn.GetName();
|
|
fieldMap.put(fieldTypeName, fieldName);
|
}
|
|
long featureCount = layer.GetFeatureCount();
|
System.out.println("图层要素个数：" + featureCount);
|
for (int i = 0; i < featureCount; i++) {
|
Feature feature = layer.GetFeature(i);
|
Object[] arr = fieldMap.values().toArray();
|
for (int k = 0; k < arr.length; k++) {
|
String value = feature.GetFieldAsString(arr[k].toString());
|
System.out.print(" 属性名称:" + arr[k].toString() + ",属性值:" + value);
|
}
|
System.out.println();
|
break;
|
}
|
} catch (Exception ex) {
|
log.error(ex.getMessage(), ex);
|
}
|
}
|
|
public static void getLayerDto(String filePath) {
|
org.gdal.ogr.Driver driver = ogr.GetDriverByName("OpenFileGDB");
|
if (driver == null) {
|
return;
|
}
|
|
List<Map> list = new ArrayList<>();
|
List<Map<String, String>> mapList = new ArrayList<>();
|
DataSource dataSource = null;
|
try {
|
dataSource = driver.Open(filePath, 0);
|
int num = dataSource.GetLayerCount();
|
for (int i = 0; i < num; i++) {
|
// 获取图层
|
Layer layer = dataSource.GetLayer(i);
|
// 表名称
|
String strlayerName = layer.GetName();
|
// 获取图层要数个数
|
long count = layer.GetFeatureCount();
|
if (0 != count) {
|
do {
|
// 获取图层下的要素
|
Feature feature = layer.GetNextFeature();
|
if (null == feature) {
|
break;
|
}
|
// 获取边界坐标
|
Geometry geometry = feature.GetGeometryRef();
|
if (geometry != null) {
|
String geometryJson = geometry.ExportToJson();
|
String str = geometryJson.substring(geometryJson.lastIndexOf(":") + 1, geometryJson.length());
|
if (!" [ [ ] ] }".equals(str)) {
|
String s1 = str.substring(7, str.length() - 8);
|
String[] split = s1.replaceAll(" ", "").replaceAll("\\[", "").replaceAll("]", "").split(",");
|
List<String> xList = new ArrayList();
|
List<String> yList = new ArrayList();
|
Map<String, String> map1 = new HashMap(5);
|
for (int j = 0; j < split.length; j++) {
|
if (j != (split.length - 1) && j % 2 == 0) {
|
map1.put(split[j], split[j + 1]);
|
}
|
}
|
mapList.add(map1);
|
}
|
}
|
Map map = new HashMap(5);
|
//获取属性
|
for (int p = 0; p < feature.GetFieldCount(); p++) {
|
map.put(feature.GetFieldDefnRef(p).GetName(), getProperty(feature, p));
|
}
|
list.add(map);
|
feature.delete();
|
} while (true);
|
}
|
//layerDto.setLayerName(strlayerName);
|
//layerDto.setList(list);
|
//layerDto.setCount(count);
|
//layerDto.setMap(mapList);
|
}
|
} catch (Exception ex) {
|
log.error(ex.getMessage(), ex);
|
} finally {
|
if (dataSource != null) {
|
dataSource.delete();
|
}
|
}
|
}
|
|
/**
|
* 读取gdb文件
|
*
|
* @param filePath
|
*/
|
public static void readGdb(String filePath) {
|
try {
|
org.gdal.ogr.Driver driver = ogr.GetDriverByName("OpenFileGDB");
|
if (driver == null) {
|
return;
|
}
|
DataSource dataSource = driver.Open(filePath, 0);
|
|
for (int i = 0, count = dataSource.GetLayerCount(); i < count; i++) {
|
// 获取图层
|
Layer layer = dataSource.GetLayer(i);
|
System.out.println(i + "\t" + layer.GetName());
|
do {
|
// 获取图层下的要素
|
Feature feature = layer.GetNextFeature();
|
if (null == feature) {
|
break;
|
}
|
// 获取样式，这里是空，待确定
|
System.out.println(feature.GetStyleString());
|
// 获取geometry，也可以ExportToWkb() gml等等
|
System.out.println(feature.GetGeometryRef().ExportToWkt());
|
// 获取属性
|
System.out.println("----------------------------------");
|
for (int p = 0; p < feature.GetFieldCount(); p++) {
|
System.out.println(feature.GetFieldDefnRef(p).GetName() + "\t" + getProperty(feature, p));
|
}
|
System.out.println("----------------------------------");
|
break;
|
} while (true);
|
}
|
} catch (Exception ex) {
|
log.error(ex.getMessage(), ex);
|
}
|
}
|
|
public static Object getProperty(Feature feature, int index) {
|
int type = feature.GetFieldType(index);
|
|
PropertyGetter propertyGetter;
|
if (type < 0 || type >= PROPERTY_GETTERS.length) {
|
propertyGetter = STRING_PROPERTY_GETTER;
|
} else {
|
propertyGetter = PROPERTY_GETTERS[type];
|
}
|
|
try {
|
return propertyGetter.get(feature, index);
|
} catch (Exception ex) {
|
// ex.printStackTrace()
|
log.error(ex.getMessage(), ex);
|
|
return null;
|
}
|
}
|
|
/**
|
* 属性获取器
|
*/
|
@FunctionalInterface
|
private interface PropertyGetter {
|
/**
|
* 获取属性
|
*
|
* @param feature
|
* @param index
|
* @return
|
*/
|
Object get(Feature feature, int index);
|
}
|
|
private static final PropertyGetter STRING_PROPERTY_GETTER = (feature, index) -> feature.GetFieldAsString(index);
|
|
/**
|
* feature.GetFieldType(index)得到一个属性类型的int值,该值对应具体类型
|
*/
|
private static final PropertyGetter[] PROPERTY_GETTERS = new PropertyGetter[]{
|
// 0-Integer
|
(feature, index) -> feature.GetFieldAsInteger(index),
|
// 1-IntegerList
|
(feature, index) -> feature.GetFieldAsIntegerList(index),
|
// 2-Real
|
(feature, index) -> feature.GetFieldAsDouble(index),
|
// 3-RealList
|
(feature, index) -> feature.GetFieldAsDoubleList(index),
|
// 4-String
|
STRING_PROPERTY_GETTER,
|
// 5-StringList
|
(feature, index) -> feature.GetFieldAsStringList(index),
|
// 6-unknown
|
STRING_PROPERTY_GETTER,
|
// 7-unknown
|
STRING_PROPERTY_GETTER,
|
// 8-Binary
|
(feature, index) -> feature.GetFieldAsBinary(index),
|
// 9-Date
|
(feature, index) -> {
|
int[] pnYear = new int[1];
|
int[] pnMonth = new int[1];
|
int[] pnDay = new int[1];
|
int[] pnHour = new int[1];
|
int[] pnMinute = new int[1];
|
float[] pfSecond = new float[1];
|
int[] pnTZFlag = new int[1];
|
feature.GetFieldAsDateTime(index, pnYear, pnMonth, pnDay, pnHour, pnMinute, pfSecond, pnTZFlag);
|
java.sql.Date date = java.sql.Date.valueOf(LocalDate.of(pnYear[0], pnMonth[0], pnDay[0]));
|
return date;
|
},
|
// 10-Time
|
(feature, index) -> {
|
int[] pnYear = new int[1];
|
int[] pnMonth = new int[1];
|
int[] pnDay = new int[1];
|
int[] pnHour = new int[1];
|
int[] pnMinute = new int[1];
|
float[] pfSecond = new float[1];
|
int[] pnTZFlag = new int[1];
|
feature.GetFieldAsDateTime(index, pnYear, pnMonth, pnDay, pnHour, pnMinute, pfSecond, pnTZFlag);
|
float fSecond = pfSecond[0];
|
int s = (int) fSecond;
|
int ns = (int) (1000000000 * fSecond - s);
|
Time time = Time.valueOf(LocalTime.of(pnHour[0], pnMinute[0], s, ns));
|
return time;
|
},
|
// 11-DateTime
|
(feature, index) -> {
|
int[] pnYear = new int[1];
|
int[] pnMonth = new int[1];
|
int[] pnDay = new int[1];
|
int[] pnHour = new int[1];
|
int[] pnMinute = new int[1];
|
float[] pfSecond = new float[1];
|
int[] pnTZFlag = new int[1];
|
feature.GetFieldAsDateTime(index, pnYear, pnMonth, pnDay, pnHour, pnMinute, pfSecond, pnTZFlag);
|
float fSecond = pfSecond[0];
|
int s = (int) fSecond;
|
int ns = (int) (1000000000 * fSecond - s);
|
LocalDateTime localDateTime = LocalDateTime.of(
|
LocalDate.of(pnYear[0], pnMonth[0], pnDay[0]),
|
LocalTime.of(pnHour[0], pnMinute[0], s, ns)
|
);
|
Timestamp timestamp = Timestamp.valueOf(localDateTime);
|
return timestamp;
|
},
|
// 12-Integer64
|
(feature, index) -> feature.GetFieldAsInteger64(index),
|
// 13 Integer64List
|
(feature, index) -> feature.GetFieldAsIntegerList(index)
|
// >=14 (unknown)
|
};
|
}
|
