2022-11-04 18:42:22 -07:00
|
|
|
from flask import Flask, render_template, request, send_file
|
2022-10-11 13:46:53 -07:00
|
|
|
import numpy as np
|
2022-11-14 22:47:43 -08:00
|
|
|
from plotly_util import generate_plotly_temperature_pdf
|
2022-10-11 13:46:53 -07:00
|
|
|
from ratio_pyrometry import ratio_pyrometry_pipeline
|
2022-10-28 20:11:50 -07:00
|
|
|
from size_projection import get_projected_area
|
2022-10-11 13:46:53 -07:00
|
|
|
import base64
|
2022-10-11 16:03:00 -07:00
|
|
|
import cv2 as cv
|
2022-10-11 11:19:48 -07:00
|
|
|
|
2022-10-11 16:03:00 -07:00
|
|
|
app = Flask(
|
|
|
|
__name__,
|
2022-10-20 17:33:15 -07:00
|
|
|
static_folder='static',
|
2022-10-11 16:03:00 -07:00
|
|
|
static_url_path='/s/'
|
|
|
|
)
|
|
|
|
|
2022-10-11 11:19:48 -07:00
|
|
|
@app.route('/', methods=['GET'])
|
|
|
|
def index():
|
2022-10-28 15:18:51 -07:00
|
|
|
return render_template('index.html')
|
2022-10-11 11:19:48 -07:00
|
|
|
|
|
|
|
@app.route('/ratio_pyro', methods=['POST'])
|
|
|
|
def ratio_pyro():
|
|
|
|
f = request.files['file']
|
2022-10-11 13:46:53 -07:00
|
|
|
f_bytes = np.fromstring(f.read(), np.uint8)
|
2022-11-14 22:47:43 -08:00
|
|
|
img_orig, img_res, key, ptemps, indiv_firebrands = ratio_pyrometry_pipeline(
|
2022-10-11 13:46:53 -07:00
|
|
|
f_bytes,
|
|
|
|
ISO=float(request.form['iso']),
|
|
|
|
I_Darkcurrent=float(request.form['i_darkcurrent']),
|
|
|
|
exposure_time=float(request.form['exposure_time']),
|
|
|
|
f_stop=float(request.form['f_stop']),
|
|
|
|
MAX_TEMP=float(request.form['max_temp']),
|
2022-10-12 17:37:31 -07:00
|
|
|
MIN_TEMP=float(request.form['min_temp']),
|
|
|
|
smoothing_radius=int(request.form['smoothing_radius']),
|
2022-10-20 17:33:15 -07:00
|
|
|
key_entries=int(request.form['legend_entries']),
|
2022-11-14 22:47:43 -08:00
|
|
|
eqn_scaling_factor=float(request.form['equation_scaling_factor']),
|
|
|
|
firebrand_min_intensity_threshold=float(request.form['intensity_threshold']),
|
|
|
|
firebrand_min_area=float(request.form['min_area']),
|
2022-10-11 13:46:53 -07:00
|
|
|
)
|
|
|
|
|
2022-10-27 12:01:35 -07:00
|
|
|
# get base64 encoded images
|
2022-10-12 12:27:52 -07:00
|
|
|
img_orig_b64 = base64.b64encode(cv.imencode('.png', img_orig)[1]).decode(encoding='utf-8')
|
|
|
|
img_res_b64 = base64.b64encode(cv.imencode('.png', img_res)[1]).decode(encoding='utf-8')
|
|
|
|
|
2022-11-14 22:47:43 -08:00
|
|
|
ptemps_list = [ptemps]
|
2022-11-04 18:42:22 -07:00
|
|
|
|
2022-11-14 22:47:43 -08:00
|
|
|
for i in range(len(indiv_firebrands)):
|
|
|
|
# base64 encode image data
|
|
|
|
brand_data = indiv_firebrands[i]
|
|
|
|
unencoded = brand_data["img_data"]
|
|
|
|
brand_data["img_data"] = base64.b64encode(cv.imencode('.png', unencoded)[1]).decode(encoding='utf-8')
|
|
|
|
indiv_firebrands[i] = brand_data
|
|
|
|
|
|
|
|
# add ptemp data to list
|
|
|
|
ptemps_list.append(brand_data["ptemps"])
|
2022-11-04 18:42:22 -07:00
|
|
|
|
2022-11-14 22:47:43 -08:00
|
|
|
freq_plot, csvstrs = generate_plotly_temperature_pdf(ptemps_list)
|
2022-11-04 18:42:22 -07:00
|
|
|
|
2022-10-11 13:46:53 -07:00
|
|
|
return render_template(
|
2022-10-28 17:52:47 -07:00
|
|
|
'pyrometry-results.html',
|
2022-10-12 12:27:52 -07:00
|
|
|
img_orig_b64=img_orig_b64,
|
|
|
|
img_res_b64=img_res_b64,
|
2022-10-27 12:01:35 -07:00
|
|
|
legend=key,
|
2022-11-04 18:42:22 -07:00
|
|
|
freq_plot=freq_plot,
|
2022-11-14 22:47:43 -08:00
|
|
|
csv_data=csvstrs[0],
|
|
|
|
individual_firebrands=indiv_firebrands,
|
2022-10-11 13:46:53 -07:00
|
|
|
)
|
2022-10-28 17:52:47 -07:00
|
|
|
|
|
|
|
|
|
|
|
@app.route('/projected_area')
|
|
|
|
def projected_area():
|
|
|
|
return render_template('projected-area.html')
|
|
|
|
|
|
|
|
|
2022-10-28 20:11:50 -07:00
|
|
|
@app.route('/projected_area_results', methods=['POST'])
|
2022-10-28 17:52:47 -07:00
|
|
|
def projected_area_results():
|
2022-10-28 20:11:50 -07:00
|
|
|
f = request.files['file']
|
|
|
|
f_bytes = np.fromstring(f.read(), np.uint8)
|
|
|
|
|
|
|
|
img, dtable = get_projected_area(
|
|
|
|
f_bytes,
|
|
|
|
int(request.form['area_threshold']),
|
|
|
|
int(request.form['min_display_threshold']),
|
2022-11-04 18:42:22 -07:00
|
|
|
float(request.form['paper_width']),
|
|
|
|
float(request.form['paper_width'])
|
2022-10-28 20:11:50 -07:00
|
|
|
)
|
|
|
|
|
|
|
|
return render_template(
|
|
|
|
'projected-area-results.html',
|
|
|
|
img_b64=img,
|
|
|
|
dtable=dtable
|
|
|
|
)
|
2022-11-04 18:42:22 -07:00
|
|
|
|
|
|
|
# @app.route("/download_pyrometry_temps")
|
|
|
|
# def download_pyrometry_temps():
|
|
|
|
# return send_file()
|