|
1
|
|
-
|
|
2
|
|
-from osgeo.ogr import *
|
|
3
|
|
-from osgeo import ogr
|
|
4
|
|
-from osgeo import gdal
|
|
5
|
|
-import os
|
|
6
|
|
-import uuid
|
|
7
|
|
-import shutil
|
|
8
|
|
-import time
|
|
9
|
|
-from app.modules.data.models import *
|
|
10
|
|
-from app.util.component.PGUtil import PGUtil
|
|
11
|
|
-from app.util.component.StructuredPrint import StructurePrint
|
|
12
|
|
-from sqlalchemy.orm import Session
|
|
13
|
|
-import configure
|
|
14
|
|
-import math
|
|
15
|
|
-from functools import lru_cache
|
|
16
|
|
-import traceback
|
|
17
|
|
-import copy
|
|
18
|
|
-from app.util.component.GeometryAdapter import GeometryAdapter
|
|
19
|
|
-from app.util.component.VacuateConf import VacuateConf
|
|
20
|
|
-import datetime
|
|
21
|
|
-class EntryDataVacuate:
|
|
22
|
|
-
|
|
23
|
|
- def entry(self,parameter):
|
|
24
|
|
- # meta:dict = parameter.get("meta")
|
|
25
|
|
-
|
|
26
|
|
- # 初始化任务
|
|
27
|
|
- this_task = ThisTask(parameter)
|
|
28
|
|
- this_task.write_process("入库任务初始化...")
|
|
29
|
|
-
|
|
30
|
|
- # 数据路径,用作删除
|
|
31
|
|
- _data_path=None
|
|
32
|
|
- try:
|
|
33
|
|
- metas: list = parameter.get("meta")
|
|
34
|
|
- # 总的入库是否成功
|
|
35
|
|
- is_success=True
|
|
36
|
|
-
|
|
37
|
|
- this_task.update({"process": "入库中"})
|
|
38
|
|
-
|
|
39
|
|
- # 开始入库事务
|
|
40
|
|
- this_task.start()
|
|
41
|
|
-
|
|
42
|
|
- # 多个文件依次入库
|
|
43
|
|
- for meta in metas:
|
|
44
|
|
- #设置编码
|
|
45
|
|
- encoding = parameter.get("encoding")
|
|
46
|
|
- if encoding:
|
|
47
|
|
- gdal.SetConfigOption("SHAPE_ENCODING",encoding)
|
|
48
|
|
- else:
|
|
49
|
|
- gdal.SetConfigOption("SHAPE_ENCODING", "GBK")
|
|
50
|
|
-
|
|
51
|
|
- #如果包含cpg文件,优先使用cpg文件中声明的编码
|
|
52
|
|
- encoding_cpg = meta.get("encoding")
|
|
53
|
|
- if encoding_cpg:
|
|
54
|
|
- gdal.SetConfigOption("SHAPE_ENCODING", encoding_cpg)
|
|
55
|
|
-
|
|
56
|
|
- data_path = meta.get("data_path")
|
|
57
|
|
-
|
|
58
|
|
- #设定删除路径
|
|
59
|
|
- if not _data_path:
|
|
60
|
|
- _data_path=data_path
|
|
61
|
|
-
|
|
62
|
|
-
|
|
63
|
|
- if not data_path:
|
|
64
|
|
- raise Exception("数据错误!")
|
|
65
|
|
- # 分为shp和gdb 2种录入形式
|
|
66
|
|
-
|
|
67
|
|
- if data_path.endswith("shp"):
|
|
68
|
|
- is_success_one,new_layer_name = self.entry_shp(data_path,this_task,meta)
|
|
69
|
|
- else:
|
|
70
|
|
- is_success_one,new_layer_names = self.entry_gdb(data_path,this_task,meta)
|
|
71
|
|
-
|
|
72
|
|
- #如果其中一个失败,总的入库就失败
|
|
73
|
|
- if not is_success_one:
|
|
74
|
|
- is_success=False
|
|
75
|
|
-
|
|
76
|
|
- this_task.write_process("数据入库结束。")
|
|
77
|
|
-
|
|
78
|
|
- if is_success:
|
|
79
|
|
- # 更新任务为成功任务
|
|
80
|
|
- this_task.update({"state": 1,"process":"入库完成","update_time": datetime.datetime.now()})
|
|
81
|
|
- this_task.commit()
|
|
82
|
|
- else:
|
|
83
|
|
- # 更新任务为失败任务
|
|
84
|
|
- this_task.update({"state": -1, "process": "入库失败", "update_time": datetime.datetime.now()})
|
|
85
|
|
- # rollback
|
|
86
|
|
- this_task.rollback()
|
|
87
|
|
-
|
|
88
|
|
- except Exception as e:
|
|
89
|
|
- this_task.write_process(e.__str__())
|
|
90
|
|
- this_task.write_process("任务中止!")
|
|
91
|
|
-
|
|
92
|
|
- this_task.update({"state": -1, "process": "入库失败", "update_time": datetime.datetime.now()})
|
|
93
|
|
- StructurePrint().print(e.__str__(),"ERROR")
|
|
94
|
|
- # rollback
|
|
95
|
|
- this_task.rollback()
|
|
96
|
|
- finally:
|
|
97
|
|
- this_task.end()
|
|
98
|
|
- try:
|
|
99
|
|
- file_tmp_path = os.path.join(_data_path.split("file_tmp")[0],"file_tmp")
|
|
100
|
|
- dir_path = os.path.dirname(_data_path)
|
|
101
|
|
- i=0
|
|
102
|
|
- while not os.path.dirname(dir_path).__eq__(file_tmp_path) and i<30:
|
|
103
|
|
- dir_path = os.path.dirname(dir_path)
|
|
104
|
|
- i+=1
|
|
105
|
|
- if i<30:
|
|
106
|
|
- shutil.rmtree(dir_path,True)
|
|
107
|
|
- StructurePrint().print("删除文件成功!")
|
|
108
|
|
- else:
|
|
109
|
|
- raise Exception("找不到文件!")
|
|
110
|
|
-
|
|
111
|
|
- except Exception as e:
|
|
112
|
|
- StructurePrint().print(e.__str__(), "ERROR")
|
|
113
|
|
- StructurePrint().print("删除文件失败!","ERROR")
|
|
114
|
|
-
|
|
115
|
|
-
|
|
116
|
|
- def entry_shp(self,data_path,this_task,meta):
|
|
117
|
|
- '''
|
|
118
|
|
- 录入shp
|
|
119
|
|
- :param data_path:
|
|
120
|
|
- :return:
|
|
121
|
|
- '''
|
|
122
|
|
-
|
|
123
|
|
- driver: Driver = ogr.GetDriverByName("ESRI Shapefile")
|
|
124
|
|
- ds: DataSource = driver.Open(data_path, 1)
|
|
125
|
|
- if not ds:
|
|
126
|
|
- raise Exception("打开数据失败!")
|
|
127
|
|
- layer: Layer = ds.GetLayer(0)
|
|
128
|
|
-
|
|
129
|
|
- is_success_one, new_layer_name =self.entry_one_layer(layer, this_task,meta)
|
|
130
|
|
- ds.Destroy()
|
|
131
|
|
- return is_success_one, new_layer_name
|
|
132
|
|
-
|
|
133
|
|
- def entry_gdb(self,data_path,this_task,meta):
|
|
134
|
|
- '''
|
|
135
|
|
- 录入gdb
|
|
136
|
|
- :param data_path:
|
|
137
|
|
- :return:
|
|
138
|
|
- '''
|
|
139
|
|
-
|
|
140
|
|
- is_successes = []
|
|
141
|
|
- new_layer_names=[]
|
|
142
|
|
- driver: Driver = ogr.GetDriverByName("OpenFileGDB")
|
|
143
|
|
- ds: DataSource = driver.Open(data_path, 0)
|
|
144
|
|
- if not ds:
|
|
145
|
|
- raise Exception("打开数据失败!")
|
|
146
|
|
-
|
|
147
|
|
-
|
|
148
|
|
- for i in range(ds.GetLayerCount()):
|
|
149
|
|
-
|
|
150
|
|
- layer: Layer = ds.GetLayer(i)
|
|
151
|
|
- if layer.GetName().lower() not in meta.get("layer").keys():
|
|
152
|
|
- continue
|
|
153
|
|
- is_success, new_layer_name = self.entry_one_layer(layer,this_task,meta)
|
|
154
|
|
- new_layer_names.append(new_layer_name)
|
|
155
|
|
- is_successes.append(is_success)
|
|
156
|
|
- ds.Destroy()
|
|
157
|
|
- if is_successes.__contains__(False):
|
|
158
|
|
- return False,new_layer_names
|
|
159
|
|
- else:
|
|
160
|
|
- return True,new_layer_names
|
|
161
|
|
-
|
|
162
|
|
- def entry_one_layer(self,layer: Layer,this_task,meta):
|
|
163
|
|
-
|
|
164
|
|
- # this_task.pg_ds.StartTransaction()
|
|
165
|
|
- new_layer_name = None
|
|
166
|
|
- # vacuate_process= None
|
|
167
|
|
- success = True
|
|
168
|
|
- table_guid = uuid.uuid1().__str__()
|
|
169
|
|
- try:
|
|
170
|
|
- # 图层设置
|
|
171
|
|
- parameter = this_task.parameter
|
|
172
|
|
-
|
|
173
|
|
- overwrite = parameter.get("overwrite") if parameter.get("overwrite") is not None and parameter.get("overwrite")=="yes" else "no"
|
|
174
|
|
- geom_name = parameter.get("geom_name") if parameter.get("geom_name") is not None else "geom"
|
|
175
|
|
- fid = parameter.get("fid") if parameter.get("fid") is not None else "fid"
|
|
176
|
|
- options = ["OVERWRITE={}".format(overwrite), "FID={}".format(fid), "GEOMETRY_NAME={}".format(geom_name),"PRECISION=NO"]
|
|
177
|
|
-
|
|
178
|
|
-
|
|
179
|
|
- # 将线/面转多线多面
|
|
180
|
|
- geom_type = GeometryAdapter.change_geom_type(layer.GetGeomType())
|
|
181
|
|
-
|
|
182
|
|
- # 更改图层名
|
|
183
|
|
- change_name = False
|
|
184
|
|
- origin_name = layer.GetName().lower()
|
|
185
|
|
-
|
|
186
|
|
- # 新图层名
|
|
187
|
|
- new_layer_name: str = meta.get("layer").get(origin_name)
|
|
188
|
|
- origin_name = new_layer_name
|
|
189
|
|
- no = 1
|
|
190
|
|
- while overwrite.__eq__("no") and this_task.pg_ds.GetLayerByName(new_layer_name) :
|
|
191
|
|
- change_name=True
|
|
192
|
|
- new_layer_name = origin_name+"_{}".format(no)
|
|
193
|
|
- no+=1
|
|
194
|
|
-
|
|
195
|
|
- if change_name:
|
|
196
|
|
- this_task.write_process("{}图层已存在,更名为{}入库".format(origin_name, new_layer_name))
|
|
197
|
|
-
|
|
198
|
|
-
|
|
199
|
|
- this_task.write_process("{}图层正在入库...".format(new_layer_name))
|
|
200
|
|
-
|
|
201
|
|
- pg_layer: Layer = this_task.pg_ds.CreateLayer(new_layer_name, layer.GetSpatialRef(), geom_type, options)
|
|
202
|
|
-
|
|
203
|
|
- # 复制原图层的属性
|
|
204
|
|
- # 去掉fid的属性
|
|
205
|
|
- schema = [sche for sche in layer.schema if not sche.name.__eq__(fid)]
|
|
206
|
|
- pg_layer.CreateFields(schema)
|
|
207
|
|
-
|
|
208
|
|
- #创建抽稀过程
|
|
209
|
|
- # vacuate_process = VacuateProcess(layer,table_guid,options)
|
|
210
|
|
-
|
|
211
|
|
-
|
|
212
|
|
-
|
|
213
|
|
- count =0
|
|
214
|
|
-
|
|
215
|
|
- for feature in layer:
|
|
216
|
|
- count+=1
|
|
217
|
|
- if count%10000==0:
|
|
218
|
|
- StructurePrint().print("{}图层已入库{}个对象".format(new_layer_name,count))
|
|
219
|
|
- # print(time.time()-this_time)
|
|
220
|
|
- #this_time=time.time()
|
|
221
|
|
- geo :Geometry = feature.GetGeometryRef()
|
|
222
|
|
- # 如果是空对象不录入
|
|
223
|
|
- if geo is not None:
|
|
224
|
|
- if geo.IsEmpty():
|
|
225
|
|
- this_task.write_process("FID:{}要素的空间字段为空,跳过该要素!".format(feature.GetFID()))
|
|
226
|
|
- StructurePrint().print("FID:{}要素的空间字段为空,跳过该要素!".format(feature.GetFID()),"WARN")
|
|
227
|
|
- continue
|
|
228
|
|
-
|
|
229
|
|
- out_feature: Feature = copy.copy(feature)
|
|
230
|
|
- out_geom = None
|
|
231
|
|
- if geo is not None:
|
|
232
|
|
- out_geom:Geometry = GeometryAdapter.change_geom(geo, geom_type)
|
|
233
|
|
- out_feature.SetGeometry(out_geom)
|
|
234
|
|
- # 出现fid为0经常有问题
|
|
235
|
|
- out_feature.SetFID(out_feature.GetFID() + 1)
|
|
236
|
|
- pg_layer.CreateFeature(out_feature)
|
|
237
|
|
-
|
|
238
|
|
- #插入抽稀图层
|
|
239
|
|
- # if out_geom is not None:
|
|
240
|
|
- # vacuate_process.vacuate(out_geom)
|
|
241
|
|
-
|
|
242
|
|
- # 注册图层信息
|
|
243
|
|
- # 是否抽吸过
|
|
244
|
|
- # is_vacuate = 1 if vacuate_process.max_level>0 else 0
|
|
245
|
|
- is_vacuate = 0
|
|
246
|
|
-
|
|
247
|
|
- this_task.register_table(pg_layer,new_layer_name,overwrite,parameter.get("creator"),is_vacuate,table_guid)
|
|
248
|
|
-
|
|
249
|
|
- # 注册抽稀表
|
|
250
|
|
- # this_task.register_table_vacuate(table_guid,vacuate_process.vacuate_layers)
|
|
251
|
|
-
|
|
252
|
|
- this_task.write_process("{}图层入库成功。".format(new_layer_name))
|
|
253
|
|
-
|
|
254
|
|
- except Exception as e:
|
|
255
|
|
-
|
|
256
|
|
- this_task.write_process("{}入库失败,数据回滚!原因:{}".format(new_layer_name,e.__str__()))
|
|
257
|
|
- StructurePrint().print("{}入库失败,数据回滚!原因:{}".format(new_layer_name,e.__str__()), "error")
|
|
258
|
|
- print(traceback.format_exc())
|
|
259
|
|
- # 抽稀回滚
|
|
260
|
|
- # vacuate_process.rollback()
|
|
261
|
|
- success =False
|
|
262
|
|
-
|
|
263
|
|
- finally:
|
|
264
|
|
- # vacuate_process.end()
|
|
265
|
|
- pass
|
|
266
|
|
- return success,new_layer_name
|
|
267
|
|
-
|
|
268
|
|
-
|
|
269
|
|
-
|
|
270
|
|
-class ThisTask:
|
|
271
|
|
-
|
|
272
|
|
- def __init__(self, parameter):
|
|
273
|
|
- try:
|
|
274
|
|
- # 该任务事务的连接
|
|
275
|
|
- self.sys_session: Session = PGUtil.get_db_session(configure.SQLALCHEMY_DATABASE_URI)
|
|
276
|
|
- # 专门的写过程的连接
|
|
277
|
|
- self.process_session: Session = PGUtil.get_db_session(configure.SQLALCHEMY_DATABASE_URI)
|
|
278
|
|
-
|
|
279
|
|
- except Exception as e:
|
|
280
|
|
- raise Exception("打开数据库失败!")
|
|
281
|
|
- self.parameter = parameter
|
|
282
|
|
-
|
|
283
|
|
- self.task = self.process_session.query(Task).filter_by(guid=parameter.get("task_guid"))
|
|
284
|
|
-
|
|
285
|
|
- self.database = self.sys_session.query(Database).filter_by(
|
|
286
|
|
- guid=parameter.get("database_guid")).one_or_none()
|
|
287
|
|
-
|
|
288
|
|
- self.catalog_guid = parameter.get("catalog_guid")
|
|
289
|
|
-
|
|
290
|
|
- self.pg_ds: DataSource = PGUtil.open_pg_data_source(1, DES.decode(self.database.sqlalchemy_uri))
|
|
291
|
|
-
|
|
292
|
|
-
|
|
293
|
|
- def start(self):
|
|
294
|
|
- self.pg_ds.StartTransaction()
|
|
295
|
|
-
|
|
296
|
|
- def update(self, update_dict):
|
|
297
|
|
- self.task.update(update_dict)
|
|
298
|
|
- self.process_session.commit()
|
|
299
|
|
-
|
|
300
|
|
- def write_process(self, message):
|
|
301
|
|
- message = "{} {}".format(datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'), message)
|
|
302
|
|
- task_process_guid = uuid.uuid1().__str__()
|
|
303
|
|
- task_process = Process(guid=task_process_guid, message=message, time=datetime.datetime.now(),
|
|
304
|
|
- task_guid=self.parameter.get("task_guid"))
|
|
305
|
|
- self.process_session.add(task_process)
|
|
306
|
|
- self.process_session.commit()
|
|
307
|
|
-
|
|
308
|
|
- def register_table(self, layer: Layer, new_layer_name, overwrite, creator,is_vacuate,table_guid):
|
|
309
|
|
- '''
|
|
310
|
|
- 注册表
|
|
311
|
|
- :param layer: 图层
|
|
312
|
|
- :param new_layer_name: 图层名
|
|
313
|
|
- :return: 表名
|
|
314
|
|
- '''
|
|
315
|
|
-
|
|
316
|
|
- this_time = datetime.datetime.now()
|
|
317
|
|
-
|
|
318
|
|
- ext = layer.GetExtent()
|
|
319
|
|
- if ext[0] < 360:
|
|
320
|
|
- ext = [round(e, 6) for e in ext]
|
|
321
|
|
- else:
|
|
322
|
|
- ext = [round(e, 2) for e in ext]
|
|
323
|
|
-
|
|
324
|
|
- geom_type = GeometryAdapter.get_geometry_type(layer)
|
|
325
|
|
-
|
|
326
|
|
- extent = "{},{},{},{}".format(ext[0], ext[1], ext[2], ext[3])
|
|
327
|
|
-
|
|
328
|
|
- table = Table(guid=table_guid,
|
|
329
|
|
- database_guid=self.database.guid,
|
|
330
|
|
- creator=creator,
|
|
331
|
|
- name=new_layer_name, create_time=this_time, update_time=this_time,
|
|
332
|
|
- catalog_guid=self.catalog_guid, table_type=GeometryAdapter.get_table_type(geom_type),
|
|
333
|
|
- extent=extent,
|
|
334
|
|
- feature_count=layer.GetFeatureCount(),
|
|
335
|
|
- is_vacuate=is_vacuate
|
|
336
|
|
- )
|
|
337
|
|
- # 删除遗留业务数据
|
|
338
|
|
- try:
|
|
339
|
|
- history_table = self.sys_session.query(Table).filter_by(name=new_layer_name,database_guid=self.database.guid).all()
|
|
340
|
|
- except:
|
|
341
|
|
- self.sys_session: Session = PGUtil.get_db_session(configure.SQLALCHEMY_DATABASE_URI)
|
|
342
|
|
- history_table = self.sys_session.query(Table).filter_by(name=new_layer_name,database_guid=self.database.guid).all()
|
|
343
|
|
-
|
|
344
|
|
- if history_table:
|
|
345
|
|
- for ht in history_table:
|
|
346
|
|
- self.sys_session.delete(ht)
|
|
347
|
|
- self.sys_session.add(table)
|
|
348
|
|
-
|
|
349
|
|
- feature_defn: FeatureDefn = layer.GetLayerDefn()
|
|
350
|
|
-
|
|
351
|
|
- for i in range(feature_defn.GetFieldCount()):
|
|
352
|
|
- field_defn: FieldDefn = feature_defn.GetFieldDefn(i)
|
|
353
|
|
- field_name = field_defn.GetName().lower()
|
|
354
|
|
- field_alias = field_name if field_defn.GetAlternativeName() is None or field_defn.GetAlternativeName().__eq__(
|
|
355
|
|
- "") else field_defn.GetAlternativeName()
|
|
356
|
|
- column = Columns(guid=uuid.uuid1().__str__(), table_guid=table_guid,
|
|
357
|
|
- name=field_name, alias=field_alias, create_time=this_time, update_time=this_time)
|
|
358
|
|
- self.sys_session.add(column)
|
|
359
|
|
- return table_guid
|
|
360
|
|
-
|
|
361
|
|
-
|
|
362
|
|
- def register_table_vacuate(self,table_guid,vacuate_layers:dict):
|
|
363
|
|
-
|
|
364
|
|
- # 抽稀表有固定的命名规则
|
|
365
|
|
- for level,layer in vacuate_layers.items():
|
|
366
|
|
- pixel_distance_str:str=layer.GetName().split("_")[-1]
|
|
367
|
|
- lev = layer.GetName().split("_")[-2]
|
|
368
|
|
-
|
|
369
|
|
- if pixel_distance_str.startswith("0"):
|
|
370
|
|
- pixel_distance_str="0.{}".format(pixel_distance_str)
|
|
371
|
|
-
|
|
372
|
|
- pixel_distance = float(pixel_distance_str)
|
|
373
|
|
-
|
|
374
|
|
- table_vacuate = TableVacuate(guid=uuid.uuid1().__str__(),
|
|
375
|
|
- table_guid=table_guid,
|
|
376
|
|
- level=int(lev),
|
|
377
|
|
- name=layer.GetName(),
|
|
378
|
|
- pixel_distance=pixel_distance)
|
|
379
|
|
- self.sys_session.add(table_vacuate)
|
|
380
|
|
-
|
|
381
|
|
-
|
|
382
|
|
- def commit(self):
|
|
383
|
|
- if self.sys_session:
|
|
384
|
|
- self.sys_session.commit()
|
|
385
|
|
- if self.pg_ds:
|
|
386
|
|
- self.pg_ds.CommitTransaction()
|
|
387
|
|
- if self.process_session:
|
|
388
|
|
- self.process_session.commit()
|
|
389
|
|
-
|
|
390
|
|
-
|
|
391
|
|
- def end(self):
|
|
392
|
|
- if self.sys_session:
|
|
393
|
|
- self.sys_session.close()
|
|
394
|
|
- if self.pg_ds:
|
|
395
|
|
- self.pg_ds.Destroy()
|
|
396
|
|
- if self.process_session:
|
|
397
|
|
- self.process_session.close()
|
|
398
|
|
-
|
|
399
|
|
- def rollback(self):
|
|
400
|
|
- if self.sys_session:
|
|
401
|
|
- self.sys_session.rollback()
|
|
402
|
|
- if self.pg_ds:
|
|
403
|
|
- self.pg_ds.RollbackTransaction()
|
|
404
|
|
-
|
|
405
|
|
-
|
|
406
|
|
-
|
|
407
|
|
-class VacuateProcess:
|
|
408
|
|
-
|
|
409
|
|
- max_level=0
|
|
410
|
|
- fill_dict={}
|
|
411
|
|
- vacuate_layers={}
|
|
412
|
|
- vacuate_layers_gridsize={}
|
|
413
|
|
- pg_ds_dict = {}
|
|
414
|
|
- # 图层要素大于5W才抽稀
|
|
415
|
|
- least_vacuate_count = VacuateConf.least_vacuate_count
|
|
416
|
|
-
|
|
417
|
|
- extent=[]
|
|
418
|
|
- is_spatial=False
|
|
419
|
|
-
|
|
420
|
|
- lonlat_gridsize = VacuateConf.lonlat_gridsize
|
|
421
|
|
- project_gridsize = VacuateConf.project_gridsize
|
|
422
|
|
-
|
|
423
|
|
- # 该抽稀过程使用的grid_size
|
|
424
|
|
- t_grid_size = []
|
|
425
|
|
-
|
|
426
|
|
- # 该抽稀过程的抽稀网格
|
|
427
|
|
- this_gridsize=[]
|
|
428
|
|
-
|
|
429
|
|
-
|
|
430
|
|
- def __init__(self,layer:Layer,table_guid, options,sqlalchemy_uri):
|
|
431
|
|
-
|
|
432
|
|
- #是空间图层才初始化
|
|
433
|
|
- if layer.GetExtent()[0] > 0 or layer.GetExtent()[0] < 0:
|
|
434
|
|
-
|
|
435
|
|
- self.is_spatial=True
|
|
436
|
|
-
|
|
437
|
|
- # 判断需要抽稀多少级
|
|
438
|
|
-
|
|
439
|
|
- lc = layer.GetFeatureCount()
|
|
440
|
|
- extent = layer.GetExtent()
|
|
441
|
|
- self.extent=extent
|
|
442
|
|
-
|
|
443
|
|
- #判断疏密程度
|
|
444
|
|
- p_x = (extent[1]-extent[0])/10.0
|
|
445
|
|
- p_y = (extent[3] - extent[2]) / 10.0
|
|
446
|
|
- fill_precent=0
|
|
447
|
|
- StructurePrint().print("判断疏密")
|
|
448
|
|
- for ix in range(10):
|
|
449
|
|
- for iy in range(10):
|
|
450
|
|
- grid_extent = [extent[0]+ix*p_x,extent[0]+ix*p_x+p_x,extent[2]+iy*p_y,extent[2]+iy*p_y+p_y]
|
|
451
|
|
- poly = GeometryAdapter.envelop_2_polygon(grid_extent)
|
|
452
|
|
-
|
|
453
|
|
- layer.SetSpatialFilter(None)
|
|
454
|
|
- layer.SetSpatialFilter(poly)
|
|
455
|
|
- layer.ResetReading()
|
|
456
|
|
- if layer.GetNextFeature():
|
|
457
|
|
- fill_precent += 1
|
|
458
|
|
-
|
|
459
|
|
- print(fill_precent)
|
|
460
|
|
- StructurePrint().print("判断疏密结束")
|
|
461
|
|
-
|
|
462
|
|
- layer.SetSpatialFilter(None)
|
|
463
|
|
- layer.ResetReading()
|
|
464
|
|
- # 固有疏密程度
|
|
465
|
|
- original_density=8
|
|
466
|
|
-
|
|
467
|
|
-
|
|
468
|
|
- # 额外一层
|
|
469
|
|
- # self.this_gridsize.append(0.000075)
|
|
470
|
|
- # self.max_level += 1
|
|
471
|
|
- ######
|
|
472
|
|
-
|
|
473
|
|
- if extent[0]>180:
|
|
474
|
|
- self.t_grid_size=self.project_gridsize
|
|
475
|
|
- else:
|
|
476
|
|
- self.t_grid_size = self.lonlat_gridsize
|
|
477
|
|
-
|
|
478
|
|
- for grid_size in self.t_grid_size:
|
|
479
|
|
- # 最少抽稀个数
|
|
480
|
|
- if lc > self.least_vacuate_count:
|
|
481
|
|
- # 网格数至少大于
|
|
482
|
|
- if ((extent[1] - extent[0]) * (extent[3] - extent[2])) / (grid_size**2)>self.least_vacuate_count:
|
|
483
|
|
- # 要素数量大于网格数量
|
|
484
|
|
- # 要考虑图层的疏密程度,original_density*(100.0/fill_precent) 为疏密指数
|
|
485
|
|
- if lc * original_density * (100.0/fill_precent)>((extent[1] - extent[0])*(extent[3] - extent[2]))/(grid_size**2) :
|
|
486
|
|
- print(grid_size)
|
|
487
|
|
- self.this_gridsize.append(grid_size)
|
|
488
|
|
- self.max_level += 1
|
|
489
|
|
-
|
|
490
|
|
-
|
|
491
|
|
-
|
|
492
|
|
- # 创建抽稀ds
|
|
493
|
|
- for l in range(self.max_level):
|
|
494
|
|
- # pg_ds_l: DataSource = PGUtil.open_pg_data_source(1, DES.decode(sqlalchemy_uri))
|
|
495
|
|
- if configure.VACUATE_DB_URI:
|
|
496
|
|
- pg_ds_l: DataSource = PGUtil.open_pg_data_source(1, configure.VACUATE_DB_URI)
|
|
497
|
|
- else:
|
|
498
|
|
- pg_ds_l: DataSource = PGUtil.open_pg_data_source(1, DES.decode(sqlalchemy_uri))
|
|
499
|
|
- pg_ds_l.StartTransaction()
|
|
500
|
|
- self.pg_ds_dict[l] = pg_ds_l
|
|
501
|
|
-
|
|
502
|
|
- # 生成抽稀图层
|
|
503
|
|
- options = options[1:]
|
|
504
|
|
- options.append("OVERWRITE=yes")
|
|
505
|
|
- options.append("LAUNDER=no")
|
|
506
|
|
-
|
|
507
|
|
- schema = layer.schema
|
|
508
|
|
- # 增加统计字段
|
|
509
|
|
- schema.append(ogr.FieldDefn("_dcigrid_count_", ogr.OFTInteger))
|
|
510
|
|
- schema.append(ogr.FieldDefn("_dcigrid_name_", ogr.OFTString))
|
|
511
|
|
-
|
|
512
|
|
- for l in range(self.max_level):
|
|
513
|
|
- this_grid_len = self.this_gridsize[l]
|
|
514
|
|
-
|
|
515
|
|
- self.vacuate_layers_gridsize[l] = this_grid_len
|
|
516
|
|
-
|
|
517
|
|
- pg = self.pg_ds_dict[l]
|
|
518
|
|
-
|
|
519
|
|
- grid_name = str(this_grid_len)
|
|
520
|
|
- if this_grid_len<1:
|
|
521
|
|
- grid_name = str(this_grid_len).split(".")[-1]
|
|
522
|
|
- if this_grid_len.__eq__(0.00008):
|
|
523
|
|
- grid_name = "00008"
|
|
524
|
|
-
|
|
525
|
|
- # 抽稀图层是点面混合的
|
|
526
|
|
- # 抽稀表有固定的命名规则
|
|
527
|
|
- # 抽稀表一定要覆盖
|
|
528
|
|
-
|
|
529
|
|
-
|
|
530
|
|
- print("{}:{}".format(self.t_grid_size.index(this_grid_len),this_grid_len))
|
|
531
|
|
-
|
|
532
|
|
-
|
|
533
|
|
- v_ln = "z{}_vacuate_{}_{}".format(table_guid, self.t_grid_size.index(this_grid_len), grid_name)
|
|
534
|
|
- vl = pg.CreateLayer(v_ln, layer.GetSpatialRef(),ogr.wkbUnknown, options)
|
|
535
|
|
- # 抽稀表需要属性
|
|
536
|
|
- vl.CreateFields(schema)
|
|
537
|
|
- self.vacuate_layers[l] = vl
|
|
538
|
|
-
|
|
539
|
|
- else:
|
|
540
|
|
- pass
|
|
541
|
|
-
|
|
542
|
|
-
|
|
543
|
|
- def vacuate(self,g,feature):
|
|
544
|
|
-
|
|
545
|
|
- if self.is_spatial:
|
|
546
|
|
-
|
|
547
|
|
- # 插入到所有抽稀图层中
|
|
548
|
|
- for level in range(self.max_level):
|
|
549
|
|
-
|
|
550
|
|
- center: Geometry = g.Centroid()
|
|
551
|
|
-
|
|
552
|
|
- extent = g.GetEnvelope()
|
|
553
|
|
- long_extent= extent[1]-extent[0]
|
|
554
|
|
- lat_extent = extent[3]-extent[2]
|
|
555
|
|
-
|
|
556
|
|
- this_grid_len =self.vacuate_layers_gridsize[level]
|
|
557
|
|
- #超大的直接加入
|
|
558
|
|
- # if long_extent > 10*this_grid_len or lat_extent >10*this_grid_len:
|
|
559
|
|
- # vacuate_layer: Layer = self.vacuate_layers.get(level)
|
|
560
|
|
- # feat = ogr.Feature(vacuate_layer.GetLayerDefn())
|
|
561
|
|
- # feat.SetGeometry(g)
|
|
562
|
|
- # vacuate_layer.CreateFeature(feat)
|
|
563
|
|
- # else:
|
|
564
|
|
-
|
|
565
|
|
- row = int((center.GetY() - self.extent[2]) / this_grid_len)
|
|
566
|
|
- col = int((center.GetX() - self.extent[0]) / this_grid_len)
|
|
567
|
|
- key = "{}.{}.{}".format(level, row, col)
|
|
568
|
|
-
|
|
569
|
|
- if not self.fill_dict.get(key):
|
|
570
|
|
- self.fill_dict[key] = 0
|
|
571
|
|
- if self.fill_dict[key] == 0:
|
|
572
|
|
-
|
|
573
|
|
- vacuate_layer: Layer = self.vacuate_layers.get(level)
|
|
574
|
|
- feat = ogr.Feature(vacuate_layer.GetLayerDefn())
|
|
575
|
|
- # 如果图形比网格小,直接存储其中心点
|
|
576
|
|
- if this_grid_len>long_extent and this_grid_len>lat_extent:
|
|
577
|
|
- feat.SetGeometry(center)
|
|
578
|
|
- else:
|
|
579
|
|
- feat.SetGeometry(g)
|
|
580
|
|
-
|
|
581
|
|
- # 复制旧feature属性
|
|
582
|
|
- field_dict = feature.items()
|
|
583
|
|
- for field_name in field_dict:
|
|
584
|
|
- feat.SetField(field_name, field_dict[field_name])
|
|
585
|
|
- feat.SetField("_dcigrid_name_",".".join(key.split(".")[1:]))
|
|
586
|
|
-
|
|
587
|
|
- vacuate_layer.CreateFeature(feat)
|
|
588
|
|
- self.fill_dict[key] += 1
|
|
589
|
|
- #超大的还有机会
|
|
590
|
|
- elif (long_extent > 10*this_grid_len or lat_extent >10*this_grid_len) and self.fill_dict[key]<5:
|
|
591
|
|
- vacuate_layer: Layer = self.vacuate_layers.get(level)
|
|
592
|
|
- feat = ogr.Feature(vacuate_layer.GetLayerDefn())
|
|
593
|
|
- feat.SetGeometry(g)
|
|
594
|
|
-
|
|
595
|
|
- # 复制旧feature属性
|
|
596
|
|
- field_dict = feature.items()
|
|
597
|
|
- for field_name in field_dict:
|
|
598
|
|
- feat.SetField(field_name, field_dict[field_name])
|
|
599
|
|
- feat.SetField("_dcigrid_name_",".".join(key.split(".")[1:]))
|
|
600
|
|
-
|
|
601
|
|
- vacuate_layer.CreateFeature(feat)
|
|
602
|
|
- self.fill_dict[key] += 1
|
|
603
|
|
- else:
|
|
604
|
|
- self.fill_dict[key] += 1
|
|
605
|
|
-
|
|
606
|
|
- def set_vacuate_count(self):
|
|
607
|
|
- if self.is_spatial:
|
|
608
|
|
- # 插入到所有抽稀图层中
|
|
609
|
|
- for level in range(self.max_level):
|
|
610
|
|
- vacuate_layer: Layer = self.vacuate_layers.get(level)
|
|
611
|
|
- for feat in vacuate_layer:
|
|
612
|
|
- key = "{}.{}".format(level,feat.GetField("_dcigrid_name_"))
|
|
613
|
|
- feat.SetField("_dcigrid_count_",self.fill_dict.get(key))
|
|
614
|
|
- vacuate_layer.SetFeature(feat)
|
|
615
|
|
-
|
|
616
|
|
- def end(self):
|
|
617
|
|
- for pg in self.pg_ds_dict.values():
|
|
618
|
|
- pg.Destroy()
|
|
619
|
|
-
|
|
620
|
|
- def rollback(self):
|
|
621
|
|
- for pg in self.pg_ds_dict.values():
|
|
622
|
|
- pg.RollbackTransaction() |