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;
using Kingo.Plugin.MakeTaskPackage.Entity;

namespace Kingo.Plugin.MakeTaskPackage.Common
{
    public class GdalShapeOperate
    {

        private static GdalShapeOperate _instance = null;
        public static GdalShapeOperate Instance
        {
            get
            {
                if (_instance == null)
                    _instance = new GdalShapeOperate();
                return _instance;
            }
        }

        public GdalShapeOperate()
        {
            //// 注册所有的驱动
            //Ogr.RegisterAll();
            //// 为了支持中文路径，请添加下面这句代码
            //Gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "NO");
            //// 为了使属性表字段支持中文，请添加下面这句
            //Gdal.SetConfigOption("SHAPE_ENCODING", "NO");
        }
        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
            {
                ////注册所有的驱动
                //Ogr.RegisterAll();
                //// 为了支持中文路径，请添加下面这句代码
                //Gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "NO");
                //// 为了使属性表字段支持中文，请添加下面这句
                //Gdal.SetConfigOption("SHAPE_ENCODING", "NO");
                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
            {
                //// 注册所有的驱动
                //Ogr.RegisterAll();
                //// 为了支持中文路径，请添加下面这句代码
                //Gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "NO");
                //// 为了使属性表字段支持中文，请添加下面这句
                //Gdal.SetConfigOption("SHAPE_ENCODING", "NO");
                //打开数据
                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 DbfColumn.DbfColumnType.Character:
                        dtNew.Columns.Add(colName, typeof(string));
                        break;
                    case DbfColumn.DbfColumnType.Number:
                        dtNew.Columns.Add(colName, typeof(double));
                        break;
                    case DbfColumn.DbfColumnType.Boolean:
                        dtNew.Columns.Add(colName, typeof(bool));
                        break;
                    case DbfColumn.DbfColumnType.Date:
                        dtNew.Columns.Add(colName, typeof(DateTime));
                        break;
                    case DbfColumn.DbfColumnType.Memo:
                        dtNew.Columns.Add(colName, typeof(string));
                        break;
                    case DbfColumn.DbfColumnType.Binary:
                        dtNew.Columns.Add(colName, typeof(byte[]));
                        break;
                    case 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)
        {
            //// 注册所有的驱动
            //Ogr.RegisterAll();
            //// 为了支持中文路径，请添加下面这句代码
            //Gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "NO");
            //// 为了使属性表字段支持中文，请添加下面这句
            //Gdal.SetConfigOption("SHAPE_ENCODING", "NO");

            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 GetPoints(OSGeo.OGR.Geometry geometry)
        {

            StringBuilder sb = new StringBuilder("{");
            switch (geometry.GetGeometryType())
            {
                case wkbGeometryType.wkbGeometryCollection:
                    break;
                case wkbGeometryType.wkbGeometryCollection25D:
                    break;
                case wkbGeometryType.wkbLineString:
                    break;
                case wkbGeometryType.wkbLineString25D:
                    break;
                case wkbGeometryType.wkbLinearRing:
                    break;
                case wkbGeometryType.wkbMultiLineString:
                    break;
                case wkbGeometryType.wkbMultiLineString25D:
                    break;
                case wkbGeometryType.wkbMultiPoint:
                    break;
                case wkbGeometryType.wkbMultiPoint25D:
                    break;

                case wkbGeometryType.wkbMultiPolygon25D:
                    break;
                case wkbGeometryType.wkbNone:
                    break;
                case wkbGeometryType.wkbPoint:
                    string json = @"{""x"":" + geometry.GetX(0) + @",""y"":" + geometry.GetY(0);
                    sb.Append(@"""point"":" + json);
                    break;
                case wkbGeometryType.wkbPoint25D:
                    break;
                case wkbGeometryType.wkbMultiPolygon:
                case wkbGeometryType.wkbPolygon:
                    int geometryCount = geometry.GetGeometryCount();
                    if (geometryCount > 0)
                    {
                        sb.Append(@"""rings"":[");

                        for (int i = 0; i < geometryCount; i++)
                        {
                            Geometry ge = geometry.GetGeometryRef(i);
                            int subGeoCount = ge.GetGeometryCount();
                            if (subGeoCount > 0)
                            {
                                for (int j = 0; j < subGeoCount; j++)
                                {
                                    sb.Append(GetSingleGeometry(ge.GetGeometryRef(j)));
                                }
                            }
                            else
                            {
                                sb.Append(GetSingleGeometry(ge));
                            }
                            //sb.Append("[");
                            //Geometry ge = geometry.GetGeometryRef(i);
                            //int count = ge.GetPointCount();
                            //for (int j = 0; j < count; j++)
                            //{
                            //    sb.Append("[" + ge.GetX(j) + "," + ge.GetY(j) + "],");
                            //}
                            //sb.Remove(sb.Length - 1, 1);
                            //sb.Append("],");
                        }
                        sb.Replace(',', ']', sb.Length - 1, 1);

                    }
                    break;
                case wkbGeometryType.wkbPolygon25D:
                    break;
                case wkbGeometryType.wkbUnknown:
                    break;
                default:
                    break;
            }
            sb.Append("}");
            return sb.ToString();
        }

        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();
        }

        public string CheckedDataValid(string sourcePath, Entity.LayerEntity entity, out List<string> listfield, string keyField = "")
        {
            try
            {
                listfield = new List<string>();
                // 注册所有的驱动
                Ogr.RegisterAll();
                // 为了支持中文路径，请添加下面这句代码
                Gdal.SetConfigOption("GDAL_FILENAME_IS_UTF8", "NO");
                // 为了使属性表字段支持中文，请添加下面这句
                Gdal.SetConfigOption("SHAPE_ENCODING", "NO");

                //打开数据
                DataSource ds = Ogr.Open(sourcePath, 0);
                if (ds == null || ds.GetLayerCount() <= 0) return "无效数据，请检查数据";
                Layer layer = ds.GetLayerByIndex(0);
                if (layer == null) return "无效数据，请检查数据";
                if (entity != null)
                {

                    listfield.Clear();
                    var task = Task.Factory.StartNew(() =>
                    {
                        var listconfigfields = new List<string>();
                        foreach (var item in entity.Fields)
                        {
                            listconfigfields.Add(item.Name.ToUpper());
                        }
                        return listconfigfields;
                    });
                    OSGeo.OGR.Feature feature = null;
                    FeatureDefn feacture = layer.GetLayerDefn();
                    wkbGeometryType layerGeometryType = layer.GetGeomType();
                    int iFieldCount = feacture.GetFieldCount();
                    for (int iAttr = 0; iAttr < iFieldCount; iAttr++)
                    {
                        FieldDefn oField = feacture.GetFieldDefn(iAttr);
                        if (oField != null)
                        {
                            listfield.Add(oField.GetName().ToUpper());
                        }
                    }

                    Task.WaitAll(task);
                    if (listfield.Count != entity.Fields.Count || task.Result.Intersect(listfield).Count() != task.Result.Count)
                    {
                        return "数据模板不一致";
                    }

                }
                if (layer.GetGeomType() != wkbGeometryType.wkbMultiPolygon && layer.GetGeomType() != wkbGeometryType.wkbPolygon)
                {
                    return "无效数据，请选择面数据";
                }
                else if (layer.GetFeatureCount(0) <= 0)
                {
                    return "选择数据源不存在数据";
                }
                else if (!string.IsNullOrWhiteSpace(keyField))
                {
                    if (layer.FindFieldIndex(keyField, 1) < 0)
                    {
                        return "无效数据，不存在字段" + keyField + ",请参照模板";
                    }
                }
                return null;
            }
            catch (Exception ex)
            {
                listfield = new List<string>();
                return null;
            }

        }

        public string CheckedGeometry(Layer layer)
        {
            OSGeo.OGR.Feature feature = null;
            string rstMsg = null;
            while ((feature = layer.GetNextFeature()) != null)
            {
                OSGeo.OGR.Geometry geometry = feature.GetGeometryRef();
                if (!geometry.IsValid())
                {
                    rstMsg = "选择数据中有错误数据，请先修复";
                }
                else if (!geometry.IsSimple())
                {
                    rstMsg = "选择数据中有错误数据，请先修复";
                }
                else if (geometry.IsEmpty())
                {
                    rstMsg = "选择数据中有错误数据，请先修复";
                }
            }
            return rstMsg;
        }

        /// <summary>
        /// Shape转WKT
        /// </summary>
        /// <param name="obejctID">主键</param>
        /// <param name="layer">图形</param>
        /// <returns>WKT</returns>
        public GeometryEntity GetWKT(string ringsPolygon)
        {
            string wktstring = "";
            GeometryEntity result = new GeometryEntity();
            OSGeo.OGR.Geometry geometry = GetPolygon(ringsPolygon);
            if (geometry != null)
            {
                OSGeo.OGR.Geometry g = geometry.Centroid();
                result.x = Math.Round(g.GetX(0), 3);
                result.y = Math.Round(g.GetY(0), 3);
                geometry.ExportToWkt(out wktstring);
                result.wktStr = wktstring;
            }
            return result;
        }

        /// <summary>
        /// 获取坐标点
        /// </summary>
        /// <param name="pointsStr"></param>
        /// <returns></returns>
        private OSGeo.OGR.Geometry GetPolygon(string pointsStr)
        {
            try
            {
                if (string.IsNullOrEmpty(pointsStr)) return null;
                JsonRings p = JsonRings.FromJson(pointsStr);
                //ESRI.ArcGIS.Client.Geometry.Polygon polygon = new ESRI.ArcGIS.Client.Geometry.Polygon();
                OSGeo.OGR.Geometry geometry = new Geometry(wkbGeometryType.wkbPolygon);
                foreach (var ring in p.rings)
                {

                    OSGeo.OGR.Geometry geometryTmp = new Geometry(wkbGeometryType.wkbLinearRing);
                    foreach (var point in ring)
                    {
                        geometryTmp.AddPoint_2D(point[0], point[1]);
                        //.Add(new ESRI.ArcGIS.Client.Geometry.MapPoint(point[0], point[1]));
                    }
                    geometryTmp.CloseRings();
                    geometry.AddGeometry(geometryTmp);
                }
                return geometry;
            }
            catch (Exception ex)
            {
                return null;
            }

        }

        /// <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;
            }

        }
    }
}