class HimawariRX:
def __init__(self):
self.version = "0.1.1"
try:
print("┌──────────────────────────────────────────────┐")
print("│ himawari-rx │")
print("│ HimawariCast Downlink Processor │")
print("├──────────────────────────────────────────────┤")
print("│ @sam210723 vksdr.com/himawari-rx │")
print("└──────────────────────────────────────────────┘\n")
except UnicodeEncodeError:
print(f"himawari-rx v{self.version}\n")
# Initialise Colorama
colorama.init(autoreset=True)
self.args = self.parse_args()
self.config = self.parse_config()
self.print_config()
self.config_dirs()
self.config_input()
if self.args.dump != None:
self.dumpf = open(self.args.dump, "wb")
print(Fore.GREEN + Style.BRIGHT +
f"Writing packets to: \"{self.args.dump}\"")
else:
self.dumpf = None
assembler_config = namedtuple('assembler_config',
'verbose dump path format ignored')
self.assembler = Assembler(
assembler_config(self.args.v, self.dumpf,
self.config['rx']['path'],
self.config['rx']['format'],
self.config['rx']['ignored_channels']))
# Check assembler thread is ready
if not self.assembler.ready:
print(Fore.WHITE + Back.RED + Style.BRIGHT +
"ASSEMBLER THREAD FAILED TO START")
self.safe_stop()
try:
print(
"──────────────────────────────────────────────────────────────────────────────────"
)
except UnicodeEncodeError:
pass
# Reset stop flag
self.stop = False
# Get processing start time
self.stime = time()
# Enter main loop
try:
self.loop()
except KeyboardInterrupt:
self.safe_stop()
def loop(self):
"""
Handle data from UDP socket
"""
while not self.stop:
if self.args.file == None:
try:
# Read packet from socket
data, addr = self.sck.recvfrom(1427)
except Exception as e:
print(e)
self.safe_stop()
# Push to assembler
self.assembler.push(data)
else:
if not self.packetf.closed:
# Read packet header from file
header = self.packetf.read(6)
# No more data to read from file
if header == b'':
self.packetf.close()
continue
# Get length of packet
length = int.from_bytes(header[2:4], 'little') - 6
# Read remaining bytes of packet from file and combine with header bytes
packet = header + self.packetf.read(length)
# Push to assembler
self.assembler.push(packet)
else:
# Assembler has all packets from file, wait for processing
if self.assembler.complete():
run_time = round(time() - self.stime, 3)
print(f"\nFINISHED PROCESSING FILE ({run_time}s)")
self.safe_stop()
else:
# Limit loop speed when waiting for assembler to finish processing
sleep(0.5)
def config_input(self):
"""
Configure UDP socket or file input
"""
if self.args.file == None:
self.sck = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
ip = self.config['udp']['ip']
port = self.config['udp']['port']
print(f"Binding UDP socket ({ip}:{port})...", end='')
try:
# Bind socket
self.sck.bind(('', port))
# Setup multicast
mreq = struct.pack("=4sl", socket.inet_aton(ip),
socket.INADDR_ANY)
self.sck.setsockopt(socket.IPPROTO_IP,
socket.IP_ADD_MEMBERSHIP, mreq)
print(Fore.GREEN + Style.BRIGHT + "SUCCESS")
except socket.error as e:
print(Fore.WHITE + Back.RED + Style.BRIGHT + "FAILED")
print(e)
self.safe_stop()
else:
print(f"Opening packet file...", end='')
if Path(self.args.file).exists():
self.packetf = open(self.args.file, 'rb')
print(Fore.GREEN + Style.BRIGHT + "SUCCESS")
else:
print(Fore.WHITE + Back.RED + Style.BRIGHT +
"FILE DOES NOT EXIST")
self.safe_stop()
def config_dirs(self):
"""
Configure output directory structure
"""
# Create output root directory
if not self.config['rx']['path'].exists():
self.config['rx']['path'].mkdir()
def parse_args(self):
"""
Parse command line arguments
"""
argp = ArgumentParser()
argp.description = "Receive weather images from geostationary satellite Himawari-8 (140.7˚E) via the HimawariCast service."
argp.add_argument("--config",
action="store",
help="Configuration file path (.ini)",
default="himawari-rx.ini")
argp.add_argument("--file",
action="store",
help="Path to packet file",
default=None)
argp.add_argument(
"-v",
action="store_true",
help="Enable verbose console output (only useful for debugging)",
default=True)
argp.add_argument("--dump",
action="store",
help="Path to packet output file")
return argp.parse_args()
def parse_config(self):
"""
Parse configuration file
"""
cfgp = ConfigParser()
cfgp.read(self.args.config)
opts = {
"rx": {
"path": Path(cfgp.get('rx', 'path')),
"format": cfgp.get('rx', 'format'),
"ignored_channels": cfgp.get('rx', 'ignored_channels')
},
"udp": {
"ip": cfgp.get('udp', 'ip'),
"port": cfgp.getint('udp', 'port')
}
}
# Check output format is valid
if opts['rx']['format'] not in ['bz2', 'xrit', 'png', 'jpg',
'bmp']: #TODO: Handle image formats
print(Fore.WHITE + Back.RED + Style.BRIGHT +
f"INVALID OUTPUT FORMAT \"{opts['rx']['format']}\"")
self.safe_stop()
# If VCID blacklist is not empty
if opts['rx']['ignored_channels'] != "":
# Parse blacklist string into int or list
ignored_channels = ast.literal_eval(opts['rx']['ignored_channels'])
# If parsed into int, wrap int in list
if type(ignored_channels) == str:
ignored_channels = [ignored_channels]
opts['rx']['ignored_channels'] = ignored_channels
return opts
def print_config(self):
"""
Print configuration information
"""
print(f"CONFIG FILE: {self.args.config}")
if self.args.file == None:
print(
f"UDP INPUT: {self.config['udp']['ip']}:{self.config['udp']['port']}"
)
else:
print(f"FILE INPUT: {self.args.file}")
print(f"OUTPUT PATH: {self.config['rx']['path'].absolute()}")
print(f"OUTPUT FORMAT: {self.config['rx']['format']}")
if (len(self.config['rx']['ignored_channels']) == 0):
print("IGNORED CHANNELS: None")
else:
ignored_channels = ""
for i, c in enumerate(self.config['rx']['ignored_channels']):
if i > 0: ignored_channels += ", "
ignored_channels += c
print(f"IGNORED CHANNELS: {ignored_channels}")
print(f"VERSION: {self.version}\n")
def safe_stop(self, message=True):
"""
Safely kill threads and exit
"""
self.stop = True
try:
self.assembler.stop = True
except AttributeError:
pass
if self.args.dump != None:
self.dumpf.close()
if message: print("\nExiting...")
exit()
