Initital commit
This commit is contained in:
commit
4d96d3ca60
GPG Key ID:
8905E8CE5CC3000D
1
.gitignore
vendored
Normal file
1
.gitignore
vendored
Normal file
@ -0,0 +1 @@
|
|||||||
|
__pycache__/
|
||||||
8
Dockerfile
Normal file
8
Dockerfile
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
FROM python:3.11-bullseye
|
||||||
|
WORKDIR /app
|
||||||
|
RUN apt-get update && apt-get install -y libglib2.0-dev usbutils
|
||||||
|
RUN pip3 install 'radiacode[examples]' --upgrade
|
||||||
|
RUN apt-get install -y libusb-1.0-0-dev
|
||||||
|
COPY . .
|
||||||
|
RUN pip3 install pyusb
|
||||||
|
CMD ["python3", "webserver.py"]
|
||||||
27
docker-compose.yml
Normal file
27
docker-compose.yml
Normal file
@ -0,0 +1,27 @@
|
|||||||
|
version: '3.7'
|
||||||
|
services:
|
||||||
|
radia:
|
||||||
|
environment:
|
||||||
|
- UDEV=1
|
||||||
|
devices:
|
||||||
|
- '/dev:/dev'
|
||||||
|
# restart: always
|
||||||
|
privileged: true
|
||||||
|
ports:
|
||||||
|
- 8080:8080
|
||||||
|
build: .
|
||||||
|
container_name: radia
|
||||||
|
volumes:
|
||||||
|
- /dev/bus/usb:/dev/bus/usb
|
||||||
|
- /run/udev/control:/run/udev/control
|
||||||
|
- ./:/app
|
||||||
|
networks:
|
||||||
|
- caddy
|
||||||
|
labels:
|
||||||
|
caddy: kyiv.dead.guru
|
||||||
|
caddy.tls: "assada+ua@gmail.com"
|
||||||
|
caddy.reverse_proxy: "{{upstreams 8080}}"
|
||||||
|
|
||||||
|
networks:
|
||||||
|
caddy:
|
||||||
|
external: true
|
||||||
5
usb_test.py
Normal file
5
usb_test.py
Normal file
@ -0,0 +1,5 @@
|
|||||||
|
import usb.core
|
||||||
|
|
||||||
|
|
||||||
|
d = usb.core.find(find_all=1)
|
||||||
|
print(list(d))
|
||||||
146
webserver.html
Normal file
146
webserver.html
Normal file
@ -0,0 +1,146 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1">
|
||||||
|
<title>Kyiv Radioactive!</title>
|
||||||
|
<script src="https://cdn.jsdelivr.net/npm/vue/dist/vue.min.js"></script>
|
||||||
|
<script src="https://cdn.jsdelivr.net/npm/apexcharts"></script>
|
||||||
|
<script src="https://cdn.jsdelivr.net/npm/vue-apexcharts"></script>
|
||||||
|
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/bulma@0.9.4/css/bulma.min.css">
|
||||||
|
<style>
|
||||||
|
#app > div {
|
||||||
|
margin: 5px auto;
|
||||||
|
width: 80%;
|
||||||
|
text-align: center;
|
||||||
|
padding: 5px;
|
||||||
|
border: 1px #aaa dashed;
|
||||||
|
}
|
||||||
|
#app fieldset {
|
||||||
|
display: inline-block;
|
||||||
|
border: 0;
|
||||||
|
padding: 0;
|
||||||
|
margin-left: 20px;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<section class="section">
|
||||||
|
<div class="container">
|
||||||
|
<h1 class="title">
|
||||||
|
неРадіоактивна ситуація в Києві.
|
||||||
|
</h1>
|
||||||
|
<h2 class="subtitle">В реальному часі.</h2>
|
||||||
|
<div id="app">
|
||||||
|
<div>
|
||||||
|
<apexchart type="bar" height="350" :options="spectrumChartOptions" :series="spectrum_series"></apexchart>
|
||||||
|
<div>
|
||||||
|
<fieldset>
|
||||||
|
<input type="checkbox" id="spectrum_x_accum" v-model="spectrum_accum">
|
||||||
|
<label for="spectrum_x_accum">Акумульований</label>
|
||||||
|
</fieldset>
|
||||||
|
<fieldset>
|
||||||
|
<input type="radio" id="spectrum_x_channel" v-bind:value="false" v-model="spectrum_energy">
|
||||||
|
<label for="spectrum_x_channel">Канал</label>
|
||||||
|
|
||||||
|
<input type="radio" id="spectrum_x_energy" v-bind:value="true" v-model="spectrum_energy">
|
||||||
|
<label for="spectrum_x_energy">Енергія</label>
|
||||||
|
</fieldset>
|
||||||
|
<fieldset>
|
||||||
|
<input type="radio" id="spectrum_linear" v-bind:value="false" v-model="spectrum_logarithmic">
|
||||||
|
<label for="spectrum_linear">Лінійне</label>
|
||||||
|
<input type="radio" id="spectrum_log" v-bind:value="true" v-model="spectrum_logarithmic">
|
||||||
|
<label for="spectrum_log">Логарифмічне</label>
|
||||||
|
</fieldset>
|
||||||
|
</div>
|
||||||
|
<button @click="updateSpectrum">Оновити спектр</button>
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<apexchart type="line" height="350" :options="ratesChartOptions" :series="rates_series"></apexchart>
|
||||||
|
<button @click="rates_autoupdate = !rates_autoupdate">Автооновлення: {{ rates_autoupdate ? "ВКЛ" : "ВИКЛ" }}</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</section>
|
||||||
|
<script>
|
||||||
|
const common_options = {
|
||||||
|
chart: {
|
||||||
|
animations: {enabled: false},
|
||||||
|
zoom: {autoScaleYaxis: true},
|
||||||
|
},
|
||||||
|
tooltip: {intersect: false},
|
||||||
|
grid: {xaxis: {lines: {show: true}}},
|
||||||
|
dataLabels: {enabled: false},
|
||||||
|
};
|
||||||
|
var app = new Vue({
|
||||||
|
el: '#app',
|
||||||
|
components: {
|
||||||
|
apexchart: VueApexCharts,
|
||||||
|
},
|
||||||
|
data: function() {
|
||||||
|
return {
|
||||||
|
ws: null,
|
||||||
|
spectrum_duration: 0,
|
||||||
|
rates_autoupdate: true,
|
||||||
|
rates_series: [],
|
||||||
|
spectrum_accum: false,
|
||||||
|
spectrum_series: [],
|
||||||
|
spectrum_coef: [0, 0, 0],
|
||||||
|
spectrum_logarithmic: true,
|
||||||
|
spectrum_energy: true,
|
||||||
|
ratesChartOptions: {
|
||||||
|
...common_options,
|
||||||
|
title: {text: 'Активнсть подій (подій в секунду) і доза'},
|
||||||
|
xaxis: {type: 'datetime'},
|
||||||
|
yaxis: [
|
||||||
|
{seriesName: 'Подій', title: {text: 'ПНС'}, labels: {formatter:(v) => v.toFixed(2) + ' ПНС'}},
|
||||||
|
{seriesName: 'Доза', title: {text: 'мк3в/г'}, labels: {formatter:(v) => v.toFixed(4) + ' мк3в/г'}, opposite: true},
|
||||||
|
],
|
||||||
|
},
|
||||||
|
};
|
||||||
|
},
|
||||||
|
watch: {
|
||||||
|
spectrum_accum() {
|
||||||
|
this.updateSpectrum();
|
||||||
|
}
|
||||||
|
},
|
||||||
|
computed: {
|
||||||
|
spectrumChartOptions() {
|
||||||
|
const a0 = this.spectrum_coef[0], a1 = this.spectrum_coef[1], a2 = this.spectrum_coef[2];
|
||||||
|
const fmt = this.spectrum_energy ? ((c) => (a0 + a1*c + a2*c*c).toFixed(0)) : undefined;
|
||||||
|
const title = this.spectrum_energy ? 'кеВ' : 'канал';
|
||||||
|
return{
|
||||||
|
...common_options,
|
||||||
|
title: {text: `Спектр, ${this.spectrum_duration} секунд`},
|
||||||
|
xaxis: {type: 'numeric', title: {text: title}, tickAmount: 25, labels: {formatter:fmt}},
|
||||||
|
yaxis: {logarithmic: this.spectrum_logarithmic, decimalsInFloat: 0},
|
||||||
|
plotOptions: {bar: {columnWidth: '95%'}},
|
||||||
|
};
|
||||||
|
},
|
||||||
|
},
|
||||||
|
created() {
|
||||||
|
this.ws = new WebSocket('wss://' + window.location.host + '/ws')
|
||||||
|
this.ws.onmessage = this.onmessage;
|
||||||
|
this.updateSpectrum();
|
||||||
|
},
|
||||||
|
beforeDestroy: function() {
|
||||||
|
this.ws.close();
|
||||||
|
},
|
||||||
|
methods: {
|
||||||
|
onmessage(ev) {
|
||||||
|
if (!this.rates_autoupdate) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const d = JSON.parse(ev.data);
|
||||||
|
this.rates_series = d.series;
|
||||||
|
},
|
||||||
|
updateSpectrum() {
|
||||||
|
fetch(`/spectrum?accum=${this.spectrum_accum}`)
|
||||||
|
.then(response => response.json())
|
||||||
|
.then(data => (this.spectrum_duration=data.duration, this.spectrum_coef=data.coef, this.spectrum_series=data.series));
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
</script>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
113
webserver.py
Normal file
113
webserver.py
Normal file
@ -0,0 +1,113 @@
|
|||||||
|
import argparse
|
||||||
|
import asyncio
|
||||||
|
import json
|
||||||
|
import pathlib
|
||||||
|
|
||||||
|
from aiohttp import web
|
||||||
|
|
||||||
|
from radiacode import CountRate, DoseRate, RadiaCode
|
||||||
|
|
||||||
|
|
||||||
|
print('LOL')
|
||||||
|
|
||||||
|
file = open(pathlib.Path(__file__).parent.absolute() / 'webserver.html', 'r')
|
||||||
|
contecnt = file.read()
|
||||||
|
|
||||||
|
|
||||||
|
async def handle_index(request):
|
||||||
|
return web.Response(content_type='text/html', text=contecnt)
|
||||||
|
|
||||||
|
|
||||||
|
async def handle_ws(request):
|
||||||
|
ws = web.WebSocketResponse()
|
||||||
|
await ws.prepare(request)
|
||||||
|
request.app.ws_clients.append(ws)
|
||||||
|
async for _ in ws: # noqa: WPS328
|
||||||
|
pass
|
||||||
|
request.app.ws_clients.remove(ws)
|
||||||
|
return ws
|
||||||
|
|
||||||
|
|
||||||
|
async def handle_spectrum(request):
|
||||||
|
cn = request.app.rc_conn
|
||||||
|
accum = request.query.get('accum') == 'true'
|
||||||
|
spectrum = cn.spectrum_accum() if accum else cn.spectrum()
|
||||||
|
# apexcharts can't handle 0 in logarithmic view
|
||||||
|
spectrum_data = [
|
||||||
|
(channel, cnt if cnt > 0 else 0.5)
|
||||||
|
for channel, cnt in enumerate(spectrum.counts)
|
||||||
|
]
|
||||||
|
print('Spectrum updated')
|
||||||
|
return web.json_response(
|
||||||
|
{
|
||||||
|
'coef': [spectrum.a0, spectrum.a1, spectrum.a2],
|
||||||
|
'duration': spectrum.duration.total_seconds(),
|
||||||
|
'series': [{'name': 'spectrum', 'data': spectrum_data}],
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def process(app):
|
||||||
|
max_history_size = 128
|
||||||
|
countrate_history, doserate_history = [], []
|
||||||
|
while True: # noqa: WPS457
|
||||||
|
databuf = app.rc_conn.data_buf()
|
||||||
|
for v in databuf:
|
||||||
|
if isinstance(v, CountRate):
|
||||||
|
countrate_history.append(v)
|
||||||
|
elif isinstance(v, DoseRate):
|
||||||
|
doserate_history.append(v)
|
||||||
|
|
||||||
|
countrate_history.sort(key=lambda x: x.dt)
|
||||||
|
countrate_history = countrate_history[-max_history_size:]
|
||||||
|
doserate_history.sort(key=lambda x: x.dt)
|
||||||
|
doserate_history = doserate_history[-max_history_size:]
|
||||||
|
jdata = json.dumps(
|
||||||
|
{
|
||||||
|
'series': [
|
||||||
|
{
|
||||||
|
'name': 'Події',
|
||||||
|
'data': [(int(1000 * x.dt.timestamp()), x.count_rate) for x in countrate_history],
|
||||||
|
},
|
||||||
|
{
|
||||||
|
'name': 'Доза',
|
||||||
|
'data': [(int(1000 * x.dt.timestamp()), 10000 * x.dose_rate) for x in doserate_history],
|
||||||
|
},
|
||||||
|
],
|
||||||
|
},
|
||||||
|
)
|
||||||
|
print(f'Rates updated, sending to {len(app.ws_clients)} connected clients')
|
||||||
|
await asyncio.gather(*[ws.send_str(jdata) for ws in app.ws_clients], asyncio.sleep(1.0))
|
||||||
|
|
||||||
|
|
||||||
|
async def on_startup(app):
|
||||||
|
asyncio.create_task(process(app))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument('--bluetooth-mac', type=str, required=False, help='bluetooth MAC address of radiascan device')
|
||||||
|
parser.add_argument('--listen-host', type=str, required=False, default='0.0.0.0', help='listen host for webserver')
|
||||||
|
parser.add_argument('--listen-port', type=int, required=False, default=8080, help='listen port for webserver')
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
app = web.Application()
|
||||||
|
|
||||||
|
|
||||||
|
app.ws_clients = []
|
||||||
|
if args.bluetooth_mac:
|
||||||
|
print('will use Bluetooth connection')
|
||||||
|
app.rc_conn = RadiaCode(bluetooth_mac=args.bluetooth_mac)
|
||||||
|
else:
|
||||||
|
print('will use USB connection')
|
||||||
|
app.rc_conn = RadiaCode()
|
||||||
|
|
||||||
|
app.on_startup.append(on_startup)
|
||||||
|
app.add_routes(
|
||||||
|
[
|
||||||
|
web.get('/', handle_index),
|
||||||
|
web.get('/spectrum', handle_spectrum),
|
||||||
|
web.get('/ws', handle_ws),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
web.run_app(app, host=args.listen_host, port=args.listen_port)
|
||||||
Loading…
Reference in New Issue
Block a user