import cantera as ct
import numpy as np

gas = ct.Solution('CH4.yaml')

maxTs = []
Us = np.linspace(1.0,20.0,26)
for U in Us:

    Xfuel = "CH4:0.5,H2:0.5"
    Xoxidizer = "O2:0.21,N2:0.79"

    gas.TPX = 300., ct.one_atm, Xfuel
    density_fuel = gas.density
    gas.TPX = 300., ct.one_atm, Xoxidizer
    density_oxidizer = gas.density

    f = ct.CounterflowDiffusionFlame(gas, width=0.02)
    f.fuel_inlet.mdot = density_fuel * U
    f.fuel_inlet.X = Xfuel
    f.fuel_inlet.T = 300.

    f.oxidizer_inlet.mdot = density_oxidizer * U
    f.oxidizer_inlet.X = Xoxidizer
    f.oxidizer_inlet.T = 300.

    f.set_refine_criteria(ratio=4, slope=0.2, curve=0.3, prune=0.04)
    f.solve(1, auto=True)
    maxTs.append(np.max(f.T))

np.savetxt("direct.txt",np.vstack([Us,np.array(maxTs)]).T)