You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
522 lines
18 KiB
522 lines
18 KiB
using System; |
|
using System.Collections.Generic; |
|
using System.Linq; |
|
using System.Text; |
|
using OSGeo.OGR; |
|
using OSGeo.GDAL; |
|
using System.Runtime.InteropServices; |
|
using System.Threading.Tasks; |
|
using GIS.HPU.ZYZ.SHP.DBF; |
|
using System.IO; |
|
|
|
namespace Kingo.Plugin.RasterToKOTilesApp.KoDataBase |
|
{ |
|
public class GdalShapeOperate |
|
{ |
|
|
|
private static GdalShapeOperate _instance = null; |
|
public static GdalShapeOperate Instance |
|
{ |
|
get |
|
{ |
|
if (_instance == null) |
|
_instance = new GdalShapeOperate(); |
|
return _instance; |
|
} |
|
} |
|
|
|
public object DbfColumn { get; private set; } |
|
|
|
|
|
public Layer CreateShp(string strVectorFile, string layerName, OSGeo.OSR.SpatialReference pSRS) |
|
{ |
|
try |
|
{ |
|
string strDriverName = "Memory"; |
|
int count = Ogr.GetDriverCount(); |
|
OSGeo.OGR.Driver oDriver = Ogr.GetDriverByName(strDriverName); |
|
if (oDriver == null) |
|
{ |
|
return null; |
|
} |
|
|
|
// 创建数据源 |
|
DataSource oDS = oDriver.CreateDataSource(strVectorFile + layerName, null); |
|
if (oDS == null) |
|
{ |
|
return null; |
|
} |
|
Layer oLayer = oDS.CreateLayer(layerName, pSRS, wkbGeometryType.wkbPolygon, null); |
|
if (oLayer == null) |
|
{ |
|
return null; |
|
} |
|
else |
|
return oLayer; |
|
} |
|
catch (Exception ex) |
|
{ |
|
return null; |
|
|
|
} |
|
} |
|
|
|
System.Data.DataTable tableRst = null; |
|
public System.Data.DataTable ShapeToDataTable(string shapePath, string layerName = null) |
|
{ |
|
try |
|
{ |
|
string dbfPathS = Path.GetDirectoryName(shapePath) + "\\" + Path.GetFileNameWithoutExtension(shapePath) + ".dbf"; |
|
string dbfPath = AppDomain.CurrentDomain.BaseDirectory + Guid.NewGuid().ToString() + ".dbf"; |
|
File.Copy(dbfPathS, dbfPath, true); |
|
DbfFile dbfFile = new DbfFile(Encoding.GetEncoding(936));//编码 |
|
dbfFile.Open(dbfPath, FileMode.Open); |
|
//打开数据 |
|
DataSource ds = Ogr.Open(shapePath, 0); |
|
if (ds == null || ds.GetLayerCount() <= 0) return null; |
|
Layer layer = null; |
|
if (string.IsNullOrWhiteSpace(layerName)) |
|
{ |
|
layer = ds.GetLayerByIndex(0); |
|
} |
|
else |
|
{ |
|
layer = ds.GetLayerByName(layerName); |
|
} |
|
|
|
|
|
if (layer == null) return null; |
|
OSGeo.OGR.Feature feature = null; |
|
//tableRst = GetDataTable(layer.GetLayerDefn()); |
|
string fidName = layer.GetFIDColumn(); |
|
if (string.IsNullOrWhiteSpace(fidName)) |
|
fidName = "OBJECTID"; |
|
tableRst = GetDataTable(dbfFile.Header); |
|
|
|
for (int i = 0; i < dbfFile.Header.RecordCount; i++) |
|
{ |
|
DbfRecord dbfRecord = dbfFile.Read(i); |
|
|
|
fidName = "FID"; |
|
int fidIndex = dbfFile.Header.FindColumn(fidName); |
|
if (fidIndex < 0) |
|
{ |
|
fidName = "OBJECTID"; |
|
fidIndex = dbfFile.Header.FindColumn(fidName); |
|
} |
|
if (fidIndex < 0) |
|
{ |
|
fidName = "BSM"; |
|
fidIndex = dbfFile.Header.FindColumn(fidName); |
|
} |
|
string bsm = string.Empty; |
|
if (fidIndex > -1) |
|
{ |
|
bsm = dbfRecord[fidIndex].Trim(); |
|
layer.ResetReading(); |
|
layer.SetAttributeFilter(string.Format("{1}='{0}'", bsm, fidName)); |
|
} |
|
feature = layer.GetNextFeature(); |
|
if (feature == null) continue; |
|
System.Data.DataRow drNewRow = NewDataRow(dbfRecord, dbfFile.Header, tableRst); |
|
OSGeo.OGR.Geometry geometry = feature.GetGeometryRef(); |
|
if (geometry == null) continue; |
|
drNewRow["SHAPE"] = geometry.Clone(); |
|
if (fidIndex > -1) |
|
{ |
|
drNewRow[fidName] = bsm; |
|
} |
|
tableRst.Rows.Add(drNewRow); |
|
} |
|
dbfFile.Close(); |
|
return tableRst; |
|
} |
|
catch (Exception ex) |
|
{ |
|
throw ex; |
|
} |
|
|
|
} |
|
|
|
public System.Data.DataTable GDBToDataTable(string shapePath, string layerName) |
|
{ |
|
try |
|
{ |
|
//打开数据 |
|
DataSource ds = null; |
|
if (shapePath.ToLower().EndsWith(".gdb") || shapePath.ToLower().EndsWith(".mdb")) |
|
{ |
|
ds = Ogr.Open(shapePath, 0); |
|
} |
|
else |
|
{ |
|
ds = Ogr.Open(shapePath.Substring(0, shapePath.LastIndexOf("\\")), 0); |
|
} |
|
if (ds == null || ds.GetLayerCount() <= 0) return null; |
|
Layer layer = null; |
|
layer = ds.GetLayerByName(layerName); |
|
|
|
|
|
if (layer == null) return null; |
|
tableRst = ShapeToDataTable(layer); |
|
return tableRst; |
|
} |
|
catch (Exception ex) |
|
{ |
|
throw ex; |
|
} |
|
} |
|
|
|
private System.Data.DataRow NewDataRow(DbfRecord record, DbfHeader header, System.Data.DataTable tableRst) |
|
{ |
|
if (tableRst == null) return null; |
|
System.Data.DataRow newRow = tableRst.NewRow(); |
|
|
|
for (int iField = 0; iField < header.ColumnCount; iField++) |
|
{ |
|
string name = header[iField].Name.ToUpper(); |
|
if (name == "SHAPE" || name == "OBJECTID" || name == "FID") continue; |
|
if (name == "SHAPE_AREA") |
|
name = "SHAPEAREA"; |
|
else if (name == "SHAPE_LENGTH") |
|
{ |
|
name = "SHAPELENGTH"; |
|
} |
|
|
|
newRow[name] = record[iField].Trim(); |
|
|
|
} |
|
return newRow; |
|
} |
|
|
|
private System.Data.DataTable GetDataTable(DbfHeader header) |
|
{ |
|
System.Data.DataTable dtNew = new System.Data.DataTable(); |
|
|
|
for (int i = 0; i < header.ColumnCount; i++) |
|
{ |
|
|
|
|
|
string name = header[i].Name.ToUpper(); |
|
if (name.ToUpper() == "SHAPE") continue; |
|
//if (name.ToUpper() == "BSM") |
|
//{ |
|
// dtNew.Columns.Add("BSM", typeof(Int32)); |
|
// continue; |
|
//} |
|
string colName = name.ToUpper(); |
|
if (colName == "SHAPE_AREA") |
|
colName = "SHAPEAREA"; |
|
else if (colName == "SHAPE_LENGTH") |
|
{ |
|
colName = "SHAPELENGTH"; |
|
} |
|
switch (header[i].ColumnType) |
|
{ |
|
case GIS.HPU.ZYZ.SHP.DBF.DbfColumn.DbfColumnType.Character: |
|
dtNew.Columns.Add(colName, typeof(string)); |
|
break; |
|
case GIS.HPU.ZYZ.SHP.DBF.DbfColumn.DbfColumnType.Number: |
|
dtNew.Columns.Add(colName, typeof(double)); |
|
break; |
|
case GIS.HPU.ZYZ.SHP.DBF.DbfColumn.DbfColumnType.Boolean: |
|
dtNew.Columns.Add(colName, typeof(bool)); |
|
break; |
|
case GIS.HPU.ZYZ.SHP.DBF.DbfColumn.DbfColumnType.Date: |
|
dtNew.Columns.Add(colName, typeof(DateTime)); |
|
break; |
|
case GIS.HPU.ZYZ.SHP.DBF.DbfColumn.DbfColumnType.Memo: |
|
dtNew.Columns.Add(colName, typeof(string)); |
|
break; |
|
case GIS.HPU.ZYZ.SHP.DBF.DbfColumn.DbfColumnType.Binary: |
|
dtNew.Columns.Add(colName, typeof(byte[])); |
|
break; |
|
case GIS.HPU.ZYZ.SHP.DBF.DbfColumn.DbfColumnType.Integer: |
|
dtNew.Columns.Add(colName, typeof(Int32)); |
|
break; |
|
default: |
|
dtNew.Columns.Add(colName, typeof(string)); |
|
break; |
|
} |
|
|
|
} |
|
dtNew.Columns.Add("SHAPE", typeof(object)); |
|
return dtNew; |
|
} |
|
|
|
public System.Data.DataTable ShapeToDataTable(Layer layer) |
|
{ |
|
|
|
|
|
if (layer == null) return null; |
|
OSGeo.OGR.Feature feature = null; |
|
tableRst = GetDataTable(layer.GetLayerDefn()); |
|
wkbGeometryType layerGeometryType = layer.GetGeomType(); |
|
while ((feature = layer.GetNextFeature()) != null) |
|
{ |
|
System.Data.DataRow drNewRow = NewDataRow(feature); |
|
OSGeo.OGR.Geometry geometry = feature.GetGeometryRef(); |
|
if (geometry == null) continue; |
|
drNewRow["SHAPE"] = geometry.Clone(); |
|
tableRst.Rows.Add(drNewRow); |
|
} |
|
return tableRst; |
|
} |
|
public bool CheckIsExistFeature(string sourcePath, string layerName) |
|
{ |
|
DataSource ds = null; |
|
try |
|
{ |
|
//打开数据 |
|
ds = Ogr.Open(sourcePath, 0); |
|
if (ds == null || ds.GetLayerCount() <= 0) return false; |
|
Layer layer = null; |
|
if (string.IsNullOrWhiteSpace(layerName)) |
|
{ |
|
layer = ds.GetLayerByIndex(0); |
|
} |
|
else |
|
{ |
|
layer = ds.GetLayerByName(layerName); |
|
} |
|
if (layer == null) return false; |
|
//return layer.GetFeatureCount(0)>0; |
|
return true; |
|
} |
|
catch |
|
{ |
|
return false; |
|
} |
|
finally |
|
{ |
|
if (ds != null) |
|
{ |
|
ds.Dispose(); |
|
ds = null; |
|
} |
|
} |
|
} |
|
|
|
/// <summary> |
|
/// 创建table |
|
/// </summary> |
|
/// <param name="oDefn"></param> |
|
/// <returns></returns> |
|
private System.Data.DataTable GetDataTable(FeatureDefn oDefn) |
|
{ |
|
System.Data.DataTable dtNew = new System.Data.DataTable(); |
|
int iFieldCount = oDefn.GetFieldCount(); |
|
for (int i = 0; i < iFieldCount; i++) |
|
{ |
|
FieldDefn oField = oDefn.GetFieldDefn(i); |
|
FieldType type = oField.GetFieldType(); |
|
string name = oField.GetNameRef(); |
|
if (name.ToUpper() == "SHAPE" || name.ToUpper() == "OBJECTID" || name.ToUpper() == "FID") continue; |
|
if (name.ToUpper() == "SHAPE") continue; |
|
string colName = name.ToUpper(); |
|
if (colName == "SHAPE_AREA") |
|
colName = "SHAPEAREA"; |
|
else if (colName == "SHAPE_LENGTH") |
|
{ |
|
colName = "SHAPELENGTH"; |
|
} |
|
switch (type) |
|
{ |
|
case FieldType.OFTInteger: |
|
dtNew.Columns.Add(colName, typeof(Int32)); |
|
break; |
|
case FieldType.OFTReal: |
|
dtNew.Columns.Add(colName, typeof(double)); |
|
break; |
|
case FieldType.OFTString: |
|
dtNew.Columns.Add(colName, typeof(string)); |
|
break; |
|
default: |
|
dtNew.Columns.Add(colName, typeof(string)); |
|
break; |
|
} |
|
|
|
//Console.WriteLine("{0}:{1} ({2}.{3})", oField.GetNameRef(), |
|
// oField.GetFieldTypeName(oField.GetFieldType()), |
|
// oField.GetWidth(), oField.GetPrecision()); |
|
} |
|
dtNew.Columns.Add("SHAPE", typeof(object)); |
|
dtNew.Columns.Add("FID", typeof(double)); |
|
return dtNew; |
|
} |
|
|
|
/// <summary> |
|
/// 创建新行 |
|
/// </summary> |
|
/// <param name="feature"></param> |
|
/// <returns></returns> |
|
private System.Data.DataRow NewDataRow(Feature feature) |
|
{ |
|
if (tableRst == null) return null; |
|
System.Data.DataRow newRow = tableRst.NewRow(); |
|
FeatureDefn oDefn = feature.GetDefnRef(); |
|
for (int iField = 0; iField < feature.GetFieldCount(); iField++) |
|
{ |
|
FieldDefn oFieldDefn = oDefn.GetFieldDefn(iField); |
|
FieldType type = oFieldDefn.GetFieldType(); |
|
string name = oFieldDefn.GetName().ToUpper(); |
|
if (name == "SHAPE" || name == "OBJECTID" || name == "FID") continue; |
|
if (name == "SHAPE_AREA") |
|
name = "SHAPEAREA"; |
|
else if (name == "SHAPE_LENGTH") |
|
{ |
|
name = "SHAPELENGTH"; |
|
} |
|
switch (type) |
|
{ |
|
//case FieldType.OFTBinary: |
|
// break; |
|
//case FieldType.OFTDate: |
|
|
|
// //feature.GetFieldAsDateTime(i, |
|
// break; |
|
//case FieldType.OFTDateTime: |
|
// break; |
|
//case FieldType.OFTIntegerList: |
|
// break; |
|
//case FieldType.OFTRealList: |
|
// break; |
|
//case FieldType.OFTStringList: |
|
// break; |
|
//case FieldType.OFTTime: |
|
// break; |
|
//case FieldType.OFTWideString: |
|
// break; |
|
//case FieldType.OFTWideStringList: |
|
// break; |
|
case FieldType.OFTInteger: |
|
newRow[name] = feature.GetFieldAsInteger(iField); |
|
break; |
|
case FieldType.OFTReal: |
|
newRow[name] = feature.GetFieldAsDouble(iField); |
|
break; |
|
case FieldType.OFTString: |
|
newRow[name] = feature.GetFieldAsString(iField); |
|
break; |
|
default: |
|
newRow[name] = feature.GetFieldAsString(iField); |
|
break; |
|
} |
|
|
|
|
|
//switch (type) |
|
//{ |
|
// case caseFieldType.OFTString: |
|
// Console.WriteLine("{0}\t",feature.GetFieldAsString(iField)); |
|
// break; |
|
//caseFieldType.OFTReal: |
|
// Console.WriteLine("{0}\t",feature.GetFieldAsDouble(iField)); |
|
// break; |
|
//caseFieldType.OFTInteger: |
|
// Console.WriteLine("{0}\t",feature.GetFieldAsInteger(iField)); |
|
// break; |
|
//default: |
|
// Console.WriteLine("{0}\t",feature.GetFieldAsString(iField)); |
|
// break; |
|
//} |
|
} |
|
return newRow; |
|
} |
|
|
|
|
|
private string GetSingleGeometry(Geometry geo) |
|
{ |
|
StringBuilder sb = new StringBuilder(); |
|
sb.Append("["); |
|
//Geometry ge = geo.GetGeometryRef(i); |
|
int count = geo.GetPointCount(); |
|
for (int j = 0; j < count; j++) |
|
{ |
|
sb.Append("[" + geo.GetX(j) + "," + geo.GetY(j) + "],"); |
|
} |
|
sb.Remove(sb.Length - 1, 1); |
|
sb.Append("],"); |
|
return sb.ToString(); |
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
/// <summary> |
|
/// 获取图层所有字段名称 |
|
/// </summary> |
|
/// <param name="shapePath"></param> |
|
/// <param name="layerName"></param> |
|
/// <returns></returns> |
|
public List<string> GetLayerFieldsName(string shapePath, string layerName = null) |
|
{ |
|
Layer layer = null; |
|
try |
|
{ |
|
|
|
layer = OpenLayer(shapePath, layerName); |
|
if (layer == null) |
|
{ |
|
//KGIS.Framework.Utils.LogAPI.Debug("无法打开图层:" + layerName); |
|
return null; |
|
} |
|
List<string> layerFieldNames = new List<string>(); |
|
FeatureDefn oDefn = layer.GetLayerDefn(); |
|
int iFieldCount = oDefn.GetFieldCount(); |
|
for (int i = 0; i < iFieldCount; i++) |
|
{ |
|
FieldDefn oField = oDefn.GetFieldDefn(i); |
|
layerFieldNames.Add(oField.GetName()); |
|
} |
|
return layerFieldNames; |
|
} |
|
catch (Exception ex) |
|
{ |
|
//KGIS.Framework.Utils.LogAPI.Debug("GetLayerFieldsName方法异常:" + ex); |
|
throw ex; |
|
} |
|
finally |
|
{ |
|
if (layer != null) |
|
layer.Dispose(); |
|
} |
|
} |
|
|
|
/// <summary> |
|
/// 打开图层 |
|
/// </summary> |
|
/// <param name="shapePath"></param> |
|
/// <param name="layerName"></param> |
|
/// <returns></returns> |
|
public Layer OpenLayer(string shapePath, string layerName = null) |
|
{ |
|
Layer layer = null; |
|
try |
|
{ |
|
DataSource ds = Ogr.Open(shapePath, 0); |
|
if (ds == null || ds.GetLayerCount() <= 0) |
|
{ |
|
//KGIS.Framework.Utils.LogAPI.Debug("ds为空"); |
|
return null; |
|
} |
|
if (string.IsNullOrWhiteSpace(layerName)) |
|
{ |
|
layer = ds.GetLayerByIndex(0); |
|
} |
|
else |
|
{ |
|
layer = ds.GetLayerByName(layerName); |
|
} |
|
return layer; |
|
} |
|
catch (Exception ex) |
|
{ |
|
//KGIS.Framework.Utils.LogAPI.Debug("OpenLayer方法异常" + ex); |
|
throw ex; |
|
} |
|
|
|
} |
|
} |
|
}
|
|
|