import numpy as np
from flask import render_template, request, session
from . import blueprint
from numpy import *
from functions import PlaneSheetAnalytical

@blueprint.get("/efficiency_module/COU")
def COU():
    return render_template(
        "efficiency_COU.html",
        tool_title="CHRIS - Extraction efficiency",
        nav_links=[
            ("Tool", "/efficiency"),
            ("CoU", "/efficiency_module/COU"),
            ("About", "/efficiency_module/RST"),
        ],
        footer_text="Office of Science and Engineering Laboratories · ",
        contact_email="david.saylor@fda.hhs.gov",
    )
@blueprint.get("/efficiency_module/RST")
def RST():
    return render_template("efficiency_RST.html",
        tool_title="CHRIS - Extraction efficiency",
        nav_links=[
            ("Tool", "/efficiency"),
            ("CoU", "/efficiency_module/COU"),
            ("About", "/efficiency_module/RST"),
        ],
        footer_text="Office of Science and Engineering Laboratories · ",
        contact_email="david.saylor@fda.hhs.gov",
    )

# load the index page for the exposure module
@blueprint.route('/efficiency', methods=['GET','POST'])
def efficiency():
    sysnum = 2
    mass = ones((sysnum))*1.0
    density = ones((sysnum))*1.0
    area = ones((sysnum)) * 6.0
    svol = ones((sysnum)) * 1.0
    iterations = ones((sysnum)).astype('int')
    time = ones((sysnum)) * 24.0
    D = ones((sysnum)) * 1e-08
    K = ones((sysnum)) * 1.0
    release = zeros((sysnum))
    conc = zeros((sysnum))

    if request.method == 'POST':
        sysnum = int(request.form["sysnum"])
        mass = ones((sysnum))
        density = ones((sysnum))
        area = ones((sysnum))
        svol = ones((sysnum))
        iterations = ones((sysnum)).astype('int')
        time = ones((sysnum))
        D = ones((sysnum))
        K = ones((sysnum))
        release = ones((sysnum))
        conc = ones((sysnum))

        for i in range(sysnum):
            mass[i] = float(request.form["mass"+str(i)])
            density[i] = float(request.form["density" + str(i)])
            area[i] = float(request.form["area" + str(i)])
            svol[i] = float(request.form["svol" + str(i)])
            iterations[i] = int(request.form["iterations" + str(i)])
            time[i] = float(request.form["time" + str(i)])
            D[i] = float(request.form["D" + str(i)])
            K[i] = float(request.form["K" + str(i)])

            PolymerVolume = mass[i] / density[i]
            ExtractionTime = time[i] * 3600. * iterations[i]
            SolventVolume = svol[i] * iterations[i]
            nterms = 5
            M0 = 1.
            release[i] = PlaneSheetAnalytical(M0, D[i], K[i], PolymerVolume, area[i], SolventVolume, ExtractionTime, nterms)
            conc[i] = release[i] * PolymerVolume / SolventVolume

    return render_template('efficiency_index.html', sysnum=sysnum, mass=mass, density=density, area=area, svol=svol, iterations=iterations,
                           time=time, D=D, K=K, release=release, conc=conc
                           ,
                           tool_title="CHRIS - Extraction efficiency",
                           nav_links=[
                               ("Tool", "/efficiency"),
                               ("CoU", "/efficiency_module/COU"),
                               ("About", "/efficiency_module/RST"),
                           ],
                           footer_text="Office of Science and Engineering Laboratories · ",
                           contact_email="david.saylor@fda.hhs.gov",
                           )