206 lines
6.5 KiB
Python
206 lines
6.5 KiB
Python
"""
|
|
|
|
"""
|
|
import os
|
|
import numpy as np
|
|
import datetime
|
|
try:
|
|
from nicegui import app, ui, events, Client
|
|
except ModuleNotFoundError as e:
|
|
print('Module `nicegui` not found.')
|
|
exit(1)
|
|
try:
|
|
import plotly.graph_objects as pgo
|
|
import plotly.io as pio
|
|
import plotly as ply
|
|
import plotly.subplots
|
|
import local_file_picker
|
|
except ModuleNotFoundError as e:
|
|
print('Module `plotly` not found.')
|
|
exit(1)
|
|
|
|
pio.templates["plotly_light"] = pgo.layout.Template(
|
|
layout = pio.templates["plotly_white"].layout)
|
|
#pio.templates["plotly_light"].layout['scene']['xaxis']['gridcolor'] = 'gray'
|
|
pio.templates["plotly_light"].layout['xaxis']['gridcolor'] = 'lightgray'
|
|
pio.templates["plotly_light"].layout['xaxis']['zerolinecolor'] = 'lightgray'
|
|
pio.templates["plotly_light"].layout['yaxis']['gridcolor'] = 'lightgray'
|
|
pio.templates["plotly_light"].layout['yaxis']['zerolinecolor'] = 'lightgray'
|
|
|
|
ELM = {
|
|
'name': 'tempplot',
|
|
'filename': './data/20260420213656.txt',
|
|
'label_file_fmt': 'file: {filename}',
|
|
#'label_data_len_fmt': 'data points (rendered): {num}',
|
|
'label_data_len_fmt': 'data points: {num} (rendered), {numa} (total)',
|
|
'plotly_light': 'plotly_light',
|
|
'plotly_dark': 'plotly_dark',
|
|
}
|
|
|
|
def main_cli():
|
|
import argparse
|
|
parser = argparse.ArgumentParser(
|
|
description=__doc__,
|
|
formatter_class=argparse.RawTextHelpFormatter,
|
|
prefix_chars='-',
|
|
add_help=False,
|
|
)
|
|
parser.add_argument(
|
|
'--host', type=str, default='', help='default 127.0.0.1')
|
|
parser.add_argument(
|
|
'--port', type=str, default='8085', help='default 8080')
|
|
parser.add_argument(
|
|
'-v', '--verbose', action='store_true', help='Verbose output')
|
|
parser.add_argument(
|
|
'-D', '--debug', action='store_true', help=argparse.SUPPRESS)
|
|
parser.add_argument(
|
|
'-h', '--help', action='store_true', help='Show this help message and exit')
|
|
args = parser.parse_args()
|
|
|
|
if args.debug:
|
|
DEBUG = True
|
|
print(f'[DEBUG] args: {args}')
|
|
print(f'[DEBUG] __file__: {__file__}')
|
|
print(f'[DEBUG] cwd: {os.getcwd()}')
|
|
print(f'[DEBUG] host: {args.host}')
|
|
print(f'[DEBUG] port: {args.port}')
|
|
|
|
if args.help:
|
|
parser.print_help()
|
|
exit(0)
|
|
|
|
return args
|
|
|
|
|
|
def np2pgo(x, y, label=None):
|
|
res = {'x': x, 'y': y}
|
|
if label is not None:
|
|
res.update({'name': label})
|
|
return res
|
|
|
|
async def pick_file() -> None:
|
|
result = await local_file_picker.local_file_picker('./data', multiple=False)
|
|
# ui.notify(f'You chose {result}')
|
|
if result:
|
|
filename = result[-1]
|
|
ELM['filename'] = filename
|
|
# update_data() # see timer below
|
|
ELM['label_file'].set_text(ELM['label_file_fmt'].format(filename=os.path.basename(filename)))
|
|
|
|
async def get_data(filename=ELM['filename']):
|
|
data = np.genfromtxt(fname=filename, skip_header=1)
|
|
|
|
if len(data) > 0 and len(data[0]) > 6:
|
|
names = ['time', 'ts', 'ta', 'tb', 'tc', 'td', 'hp', 'hl']
|
|
else:
|
|
names = ['time', 'ts', 'ta', 'tb', 'hp', 'hl']
|
|
|
|
data = np.genfromtxt(
|
|
fname=filename, dtype=None, skip_header=1, converters={
|
|
0: lambda x: datetime.datetime.strptime(x, '%Y%m%d%H%M%S')},
|
|
names=names, encoding='utf-8'
|
|
)
|
|
return data
|
|
|
|
async def update_data():
|
|
filename=ELM['filename']
|
|
data_all = await get_data(filename)
|
|
data = data_all
|
|
# time_range = ELM['number_time_range'].value
|
|
time_range = app.storage.user['time_range']
|
|
if time_range is not None and time_range > 0:
|
|
# delta = datetime.datetime.now() - data['time']
|
|
delta = data['time'][-1] - data['time']
|
|
data = data[np.where(delta <= datetime.timedelta(hours=time_range))]
|
|
print(filename, data_all.shape, data.shape)
|
|
ELM['label_data_len'].set_text(ELM['label_data_len_fmt'].format(num=len(data), numa=len(data_all)))
|
|
|
|
ELM['fig'].data = []
|
|
ELM['plot'].visible = False
|
|
ELM['plot'].update()
|
|
|
|
ELM['fig'].add_trace(pgo.Scatter(**np2pgo(data['time'], data['ts'], 'Setp')))
|
|
ELM['fig'].add_trace(pgo.Scatter(**np2pgo(data['time'], data['ta'], 'Temp A')))
|
|
ELM['fig'].add_trace(pgo.Scatter(**np2pgo(data['time'], data['tb'], 'Temp B')))
|
|
if len(data) > 0 and len(data[0]) > 6:
|
|
ELM['fig'].add_trace(pgo.Scatter(**np2pgo(data['time'], data['tc'], 'Temp C')))
|
|
ELM['fig'].add_trace(pgo.Scatter(**np2pgo(data['time'], data['td'], 'Temp D')))
|
|
|
|
ELM['fig'].add_trace(pgo.Scatter(**np2pgo(data['time'], data['hp'], 'Heat')),
|
|
row=2, col=1)
|
|
|
|
ELM['plot'].visible = True
|
|
ELM['plot'].update()
|
|
|
|
@ui.page('/')
|
|
async def index():
|
|
await ui.context.client.connected()
|
|
|
|
# initialize objects
|
|
ELM['fig'] = ply.subplots.make_subplots(rows=2, cols=1)
|
|
|
|
# render header
|
|
ui.context.client.content.classes('h-[100vh]')
|
|
ui.add_head_html('''<meta name="viewport", content="width=device-width, initial-scale=1">
|
|
<meta name="mobile-web-app-capable" content="yes">''')
|
|
|
|
dark_mode = ui.dark_mode().bind_value(app.storage.user, 'dark_mode')
|
|
|
|
ui.colors(primary='#4888c4')
|
|
|
|
with ui.row(align_items="center").classes('w-full'):
|
|
ui.button('Choose file', on_click=pick_file, icon='folder')
|
|
|
|
ELM['number_time_range'] = ui.number(
|
|
label='time range', value=6, format='%.2f', min=0, step=0.5,
|
|
suffix='h',
|
|
# on_change=lambda e: update_data(), # see timer below
|
|
).props('clearable').bind_value(app.storage.user, 'time_range').classes('w-32')
|
|
|
|
ELM['switch_timer'] = ui.switch('auto-refresh').bind_value(app.storage.user, 'timer')
|
|
|
|
ui.space()
|
|
|
|
ui.switch(
|
|
'Dark mode', on_change=lambda: (
|
|
ELM['fig'].update_layout(
|
|
template=ELM['plotly_dark'] if dark_mode.value else ELM['plotly_light']),
|
|
'plot' in ELM and ELM['plot'].update())
|
|
).bind_value(app.storage.user, 'dark_mode').props('icon="dark_mode"')
|
|
|
|
with ui.row():
|
|
ELM['label_file'] = ui.label(
|
|
ELM['label_file_fmt'].format(filename=os.path.basename(ELM['filename'])))
|
|
ELM['label_data_len'] = ui.label(ELM['label_data_len_fmt'].format(num=0, numa=0))
|
|
|
|
pio.templates.default = 'plotly_dark'
|
|
# ELM['fig'] = pgo.Figure()
|
|
ELM['fig'].update_layout(
|
|
height=600,
|
|
#template='plotly_white'
|
|
template=ELM['plotly_dark'] if dark_mode.value else ELM['plotly_light'],
|
|
)
|
|
ELM['plot'] = ui.plotly(ELM['fig']).classes('w-full')
|
|
|
|
# update_data()
|
|
ui.timer(0, lambda: update_data(), once=True)
|
|
ELM['timer'] = ui.timer(2.0, lambda: update_data(), immediate=False)
|
|
ELM['switch_timer'].bind_value_to(ELM['timer'], 'active')
|
|
|
|
def main_gui(args):
|
|
ui.run(
|
|
host=args.host, port=int(args.port), title=ELM['name'],
|
|
dark=False,
|
|
native=False,
|
|
show=False, storage_secret='tempview')
|
|
|
|
def main(args):
|
|
return 0
|
|
|
|
if __name__ in {'__main__', '__mp_main__'}:
|
|
args = main_cli()
|
|
main_gui(args)
|
|
|
|
|
|
|