import os
import unittest
import numpy as np
import ck2cti

import utilities
import Cantera as ct

class chemkinConverterTest(utilities.CanteraTest):
    def checkConversion(self, refFile, testFile):
        ref = ct.IdealGasMix(refFile)
        gas = ct.IdealGasMix(testFile)

        self.assertEqual(ref.elementNames(), gas.elementNames())
        self.assertEqual(ref.speciesNames(), gas.speciesNames())
        self.assertTrue((ref.reactantStoichCoeffs() == gas.reactantStoichCoeffs()).all())

        compositionA = [[ref.nAtoms(i,j) for j in range(ref.nElements())]
                        for i in range(ref.nSpecies())]
        compositionB = [[gas.nAtoms(i,j) for j in range(gas.nElements())]
                        for i in range(gas.nSpecies())]
        self.assertEqual(compositionA, compositionB)

    def test_gri30(self):
        if os.path.exists('gri30_test.cti'):
            os.remove('gri30_test.cti')

        ck2cti.convertMech('../../data/inputs/gri30.inp',
                           transportFile='../../data/transport/gri30_tran.dat',
                           outName='gri30_test.cti', quiet=True)

        self.checkConversion('gri30.xml', 'gri30_test.cti')

    def test_soot(self):
        if os.path.exists('soot_test.cti'):
            os.remove('soot_test.cti')

        ck2cti.convertMech('../data/soot.inp',
                           thermoFile='../data/soot-therm.dat',
                           outName='soot_test.cti', quiet=True)

        self.checkConversion('../data/soot.xml', 'soot_test.cti')

    def test_missingElement(self):
        if os.path.exists('h2o2_missingElement.cti'):
            os.remove('h2o2_missingElement.cti')

        self.assertRaises(ck2cti.InputParseError,
                          lambda: ck2cti.convertMech('../data/h2o2_missingElement.inp',
                                                    quiet=True))

    def test_missingThermo(self):
        if os.path.exists('h2o2_missingThermo.cti'):
            os.remove('h2o2_missingThermo.cti')

        self.assertRaises(ck2cti.InputParseError,
                          lambda: ck2cti.convertMech('../data/h2o2_missingThermo.inp',
                                                    quiet=True))

    def test_nasa9(self):
        if os.path.exists('nasa9_test.cti'):
            os.remove('nasa9_test.cti')
        ck2cti.convertMech('../data/nasa9-test.inp',
                           thermoFile='../data/nasa9-test-therm.dat',
                           outName='nasa9_test.cti', quiet=True)

        self.checkConversion('../data/nasa9-test.xml', 'nasa9_test.cti')

        ref = ct.IdealGasMix('../data/nasa9-test.xml')
        gas = ct.IdealGasMix('nasa9_test.cti')

        for T in [300, 500, 1200, 5000]:
            ref_cp = ref.cp_R()
            gas_cp = gas.cp_R()
            ref_h = ref.enthalpies_RT()
            gas_h = gas.enthalpies_RT()
            ref_s = ref.entropies_R()
            gas_s = gas.entropies_R()
            for i in range(gas.nSpecies()):
                self.assertNear(ref_cp[i], gas_cp[i], 1e-7)
                self.assertNear(ref_h[i], gas_h[i], 1e-7)
                self.assertNear(ref_s[i], gas_s[i], 1e-7)