zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add experimental code for saving data into a file

Browse Source
This commit is contained in:
usov_i 2020-09-29 13:41:24 +02:00
parent 7afd41ea8b
commit e3dd010c94
1 changed files with 28 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
pyzebra/app/panel_1D_detector.py
View File

@ -9,6 +9,7 @@ from bokeh.models import (
Button, Button,
Circle, Circle,
ColumnDataSource, ColumnDataSource,
CustomJS,
DataRange1d, DataRange1d,
Div, Div,
FileInput, FileInput,
@ -26,8 +27,25 @@ from bokeh.models import (
import pyzebra import pyzebra
javaScript = """
setTimeout(function() {
const filename = 'output' + js_data.data['ext']
const blob = new Blob([js_data.data['cont']], {type: 'text/plain'})
const link = document.createElement('a');
document.body.appendChild(link);
const url = window.URL.createObjectURL(blob);
link.href = url;
link.download = filename;
link.click();
window.URL.revokeObjectURL(url);
document.body.removeChild(link);
}, 500);
"""
def create(): def create():
det_data = {} det_data = {}
js_data = ColumnDataSource(data=dict(cont=[], ext=[]))
def upload_button_callback(_attr, _old, new): def upload_button_callback(_attr, _old, new):
nonlocal det_data nonlocal det_data
@ -119,13 +137,7 @@ def create():
process_button = Button(label="Process All", button_type="primary") process_button = Button(label="Process All", button_type="primary")
process_button.on_click(process_button_callback) process_button.on_click(process_button_callback)
output_textinput = TextAreaInput( def export_results(det_data):
title="Content of comm/incomm file: (once fixed, Export button will save to user machine)",
width=600,
height=400,
)
def save_button_callback():
if det_data["meta"]["indices"] == "hkl": if det_data["meta"]["indices"] == "hkl":
ext = ".comm" ext = ".comm"
elif det_data["meta"]["indices"] == "real": elif det_data["meta"]["indices"] == "real":
@ -136,15 +148,22 @@ def create():
pyzebra.export_comm(det_data, temp_file) pyzebra.export_comm(det_data, temp_file)
with open(f"{temp_file}{ext}") as f: with open(f"{temp_file}{ext}") as f:
output_textinput.value = f.read() output_content = f.read()
return output_content, ext
def save_button_callback():
cont, ext = export_results(det_data)
js_data.data.update(cont=[cont], ext=[ext])
save_button = Button(label="Export to .comm/.incomm file:") save_button = Button(label="Export to .comm/.incomm file:")
save_button.on_click(save_button_callback) save_button.on_click(save_button_callback)
save_button.js_on_click(CustomJS(args={"js_data": js_data}, code=javaScript))
upload_div = Div(text="Upload .ccl file:") upload_div = Div(text="Upload .ccl file:")
tab_layout = column( tab_layout = column(
row(column(Spacer(height=5), upload_div), upload_button, meas_select), row(column(Spacer(height=5), upload_div), upload_button, meas_select),
row(plot, Spacer(width=30), fit_output_textinput, output_textinput), row(plot, Spacer(width=30), fit_output_textinput),
row(column(smooth_toggle, process_button, save_button)), row(column(smooth_toggle, process_button, save_button)),
) )