Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
import sys
from shapely.ops import unary_union
from main import ENV_targetProj
from utils import *
def computeData(df):
"""
Specific treatments
"""
# Explode MULTIPOLYGON and keep geometry
currentGeoSerie = df.loc[:,'geometry']
allGeoSerie = currentGeoSerie.explode(index_parts=False)
# Make GDF and buffer 0.1
df = gp.GeoDataFrame(allGeoSerie)
df = makeBufferFromGDF(df, 0.1)
# Regroup
unionGeom = unary_union(df)
# Make GDF
dataUnion = {'geometry': unionGeom}
unionDF = gp.GeoDataFrame(dataUnion, crs=ENV_targetProj)
return unionDF
def wrongArguments():
print(style.RED + '/!\ Subscript error : Please specify a source and result filename as arguments to launch this script \n', style.RESET)
sys.exit(1)
if __name__ == "__main__":
# Init timer
subTimer = startTimerLog('Plan d\'eau subscript process')
# Get data with temp filename in argv
argv = sys.argv[1:]
firstArgv = None
secArgv = None
# Argv exist ?
if argv:
if len(sys.argv[1:]) > 0:
firstArgv = sys.argv[1:][0]
else:
wrongArguments()
if len(sys.argv[1:]) > 1:
secArgv = sys.argv[1:][1]
else:
wrongArguments()
# Load file Data (geoJSON)
currentGDF = createGDFfromGeoJSON(firstArgv)
# Log & Launch treatment
currentGDF = computeData(currentGDF)
# Write Result in temp file
currentGDF.to_file(secArgv)
# End timer
endTimerLog(subTimer)
else:
wrongArguments()