/
eqxml2geojson.py
101 lines (85 loc) · 3.96 KB
/
eqxml2geojson.py
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
import sys, re
import csv, json
import sqlite3
from bs4 import BeautifulSoup
argvs = sys.argv
argc = len(argvs)
if (argc != 4):
print('Usage: # python %s [centroid csvs dir] [input file(uuid.xml)] [output dir] ' % argvs[0])
quit()
# JMAXMLのオープン
xmlFile = open(argvs[2],'r')
xml = BeautifulSoup(xmlFile, "xml")
csvDir = argvs[1]
outputDir = argvs[3]
# sqlite3にcsvを都度展開する
connection = sqlite3.connect(":memory:")
cursor = connection.cursor()
# 細分区域レベルのテーブルを生成
cursor.execute("CREATE TABLE jma_area_centroid (jma_code INTEGER PRIMARY KEY, jma_name TEXT, lat REAL, lon REAL);")
with open(csvDir + "jma_area_centroid.csv",'r') as fin:
dr = csv.DictReader(fin, fieldnames = ('jma_code','jma_name','lat','lon'))
to_db = [(c['jma_code'], c['jma_name'], c['lat'], c['lon']) for c in dr]
cursor.executemany("INSERT INTO jma_area_centroid (jma_code, jma_name, lat, lon) VALUES (?, ?, ?, ?);", to_db)
connection.commit()
# 市区町村レベルのテーブルを生成
cursor.execute("CREATE TABLE jma_city_centroid (jma_code INTEGER PRIMARY KEY, jma_name TEXT, lat REAL, lon REAL);")
with open(csvDir + "jma_city_centroid.csv",'r') as fin:
dr = csv.DictReader(fin, fieldnames = ('jma_code','jma_name','lat','lon'))
to_db = [(c['jma_code'], c['jma_name'], c['lat'], c['lon']) for c in dr]
cursor.executemany("INSERT INTO jma_city_centroid (jma_code, jma_name, lat, lon) VALUES (?, ?, ?, ?);", to_db)
connection.commit()
# 震源の座標を取得
epicenterStr = xml.Report.Body.Earthquake.Hypocenter.Area.find('jmx_eb:Coordinate').string
# 分割
epicenterCoord = re.split(r'(\-|\+|\/)', epicenterStr)
# epicenterに入れる
epicenter = {"type":"Feature",
"properties": {"class":"epicenter"},
"geometry":{
"type":"Point",
"coordinates":[float(epicenterCoord[3] + epicenterCoord[4]), float(epicenterCoord[1] + epicenterCoord[2])]
}
}
# epicenterをfeaturesに入れる
areaLevelFeatures = [epicenter]
cityLevelFeatures = [epicenter]
# 各震度ごとにポイントを格納するリストpointsを定義
areaLevelPoints = [[] for i in range(0,9)]
cityLevelPoints = [[] for i in range(0,9)]
if xml.Report.Body.find('Intensity'):
# 各震度ごとにpointsにまとめる
pref = xml.Report.Body.Intensity.Observation.findAll('Pref')
for p in pref:
area = p.findAll('Area')
for a in area:
cursor.execute("SELECT lat,lon,jma_name FROM jma_area_centroid WHERE jma_code=" + a.Code.string)
row = cursor.fetchone()
if a.MaxInt is not None:
maxint = a.MaxInt.string
else:
maxint = "5?" # 震度５弱以上未入電
areaLevelFeatures.append({"type":"Feature","properties":{"class": maxint, "name": row[2]},"geometry":{"type":"Point","coordinates": [row[1],row[0]]}})
city = a.findAll('City')
for c in city:
cursor.execute("SELECT lat,lon,jma_name FROM jma_city_centroid WHERE jma_code=" + c.Code.string)
row = cursor.fetchone()
if c.MaxInt is not None:
maxint = c.MaxInt.string
else:
maxint = "5?" # 震度５弱以上未入電
cityLevelFeatures.append({"type":"Feature","properties":{"class": maxint, "name": row[2]},"geometry":{"type":"Point","coordinates": [row[1],row[0]]}})
# featuresをfeatureCollectionに追加
areaLevelFeatureCollection = {"type":"FeatureCollection","features":areaLevelFeatures}
cityLevelFeatureCollection = {"type":"FeatureCollection","features":cityLevelFeatures}
# jsonを出力
f = open(outputDir+'smallScalePoints.json', 'w')
f.write(json.dumps(areaLevelFeatureCollection, ensure_ascii=False))
f.close()
f = open(outputDir+'largeScalePoints.json', 'w')
f.write(json.dumps(cityLevelFeatureCollection, ensure_ascii=False))
f.close()
# 後処理
cursor.close()
connection.close()
xmlFile.close()