from pathlib import Path
import cantera as ct

p = ct.one_atm  # pressure
tin_f = 300.0  # fuel inlet temperature
tin_o = 300.0  # oxidizer inlet temperature
mdot_o = 0.72  # kg/m^2/s
mdot_f = 0.24  # kg/m^2/s

comp_o = "O2:0.21, N2:0.78, AR:0.01"  # air composition
comp_f = "C2H6:1"  # fuel composition

width = 0.02  # Distance between inlets is 2 cm

loglevel = 1  # amount of diagnostic output (0 to 5)

gas = ct.Solution("gri30.yaml")
gas.TP = gas.T, p

for i in range(2):
    f = ct.CounterflowDiffusionFlame(gas, width=width)

    # Set the state of the two inlets
    f.fuel_inlet.mdot = mdot_f
    f.fuel_inlet.X = comp_f
    f.fuel_inlet.T = tin_f

    f.oxidizer_inlet.mdot = mdot_o
    f.oxidizer_inlet.X = comp_o
    f.oxidizer_inlet.T = tin_o

    # Set the boundary emissivities
    f.boundary_emissivities = 0.0, 0.0
    # Turn radiation off
    f.radiation_enabled = False

    f.set_refine_criteria(ratio=4, slope=0.2, curve=0.3, prune=0.04)

    # Use the prior solution as an intial guess
    if i > 0:
        f.set_initial_guess(data=restart_file_name)

    # Solve the problem
    f.solve(loglevel, auto=True)

    # Save the result to .csv to be used as the next initial guess
    restart_file_name = Path() / "restart.csv"
    f.save(
        restart_file_name,
        overwrite=True,
    )