-
Notifications
You must be signed in to change notification settings - Fork 5
/
blueearth.py
149 lines (118 loc) · 5 KB
/
blueearth.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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
blueearth.py
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
blueearth.py从日本的地球同步卫星向日葵8号(Himawari 8 (ひまわり8号)获取高清地球图像，设置为桌面壁纸
关于向日葵8号卫星
维基百科: https://zh.wikipedia.org/wiki/%E5%90%91%E6%97%A5%E8%91%B58%E8%99%9F
由H-IIA202火箭在日本时间2014年10月7日14时16分00秒于种子岛宇宙中心发射升空。其后14时43分57秒卫星与火箭成功分离，
同月16日19时00分确认其进入静止轨道（东经104.7度, 笔者目测大致在澳大利亚中部上空）
制造厂商 三菱电机
卫星平台 DS2000
姿态控制 3轴姿态控制
寿命 卫星：15年
观察仪器：8年
卫星重量 发射时：3.5吨
进入轨道后：1.3吨
观察传感器 AHI (Advanced Himawari Imager) 16频段 宇宙环境观察 (SEDA)
观察频度 全球观察：10分钟/1次
日本附近：2.5分钟/1次
静止经纬度 赤道, 东经140.7度附近
图像传送 使用商用通信卫星传送
"""
import os
import sys
import requests
from concurrent.futures import ThreadPoolExecutor
from PIL import Image
from loguru import logger
save_img_file = "earth.png"
proportion = 1.78 # width / heith
zoom_level = 4 # image zoom level of himawari8 , curent supported: 1, 2, 4, 8, ..., 20(MAX)
if(len(sys.argv) > 1) and sys.argv[1].startswith("--level"):
try:
zoom_level = int(sys.argv[1].split("=")[1])
logger.debug("Using zoom level: %d" % zoom_level)
except ValueError as e:
pass
png_unit_size = 550 # per earth fragment size is 550x550
y_offset = int(png_unit_size / 2.0) # y offset when we splice the fragments
png_height = png_unit_size * zoom_level + 2 * y_offset
png_width = int(png_height * proportion)
x_offset = int((png_width - png_unit_size * zoom_level) / 2.0)
latest_json_url = "https://himawari8.nict.go.jp/img/D531106/latest.json"
earth_templ_url = "https://himawari8.nict.go.jp/img/D531106/{}d/550/{}/{}_{}_{}.png"
def safe_urlopen(url):
try:
response = requests.get(url)
if response.status_code != 200:
logger.fatal(f"fail to get form {url}, status code is {response.status_code}")
return response
except requests.exceptions.ConnectionError as error:
logger.fatal(f"ConnectionError: {error}")
def download(url):
res = safe_urlopen(url)
with open(os.path.basename(url), "wb") as fp:
fp.write(res.content)
logger.info("success %s" % os.path.basename(url))
def update_pilimage_list(datalist, fpath):
fp = open(fpath, "rb")
datalist.append((Image.open(fp), fp, fpath))
def stitching(urls, zoomlv=zoom_level):
datalist, index = [], 0
for url in urls:
fpath = os.path.basename(url)
try:
update_pilimage_list(datalist, fpath)
except IOError as error:
logger.error(f"IOError for {fpath}: {error}")
res = safe_urlopen(url)
with open(fpath, "wb") as fp:
fp.write(res.content)
update_pilimage_list(datalist, fpath)
logger.debug("repaired:)")
target = Image.new('RGB', (int(png_width), int(png_height)))
x, y = 0, 0
for i in range(len(urls)):
col = int(i / zoomlv)
target.paste(datalist[i][0], (int(x + x_offset + col * png_unit_size), int(y + y_offset),
int(x + x_offset + png_unit_size + col * png_unit_size),
int(y + png_unit_size + y_offset)))
y += png_unit_size
if (i + 1) % zoomlv == 0:
y = 0
datalist[i][1].close()
try:
os.remove(datalist[i][2])
except Exception as error:
logger.debug(f"remove tmp file exception: {error}")
target.save(save_img_file, quality=100)
def get_fragments(date, time, zoomlv=zoom_level):
urls = []
executor = ThreadPoolExecutor(max_workers=10)
for i in range(zoomlv):
for j in range(zoomlv):
url = earth_templ_url.format(zoomlv, date, time, i, j)
urls.append(url)
executor.submit(download, url)
executor.shutdown(wait=True)
return urls
def get_latest_fragments(zoomlv=zoom_level):
res = safe_urlopen(latest_json_url)
if res.status_code != 200:
logger.fatal(f"{res}")
# json data format: {"date":"2017-02-27 01:20:00","file":"PI_H08_20170227_0120_TRC_FLDK_R10_PGPFD.png"}
json_data = res.json()
date_str = json_data.get("date", "")
if date_str == "":
logger.fatal("bad date string")
date, t = date_str.split(" ")
date, t = date.replace("-", "/"), t.replace(":", "")
return get_fragments(date, t, zoomlv)
def main():
urls = get_latest_fragments(zoom_level)
stitching(urls)
logger.info("done")
if __name__ == "__main__":
main()