6cb4bd93ce
4 spaces per indentation level, no tabs, no trailing whitespace, and a single newline at end of each file.
193 lines
6.6 KiB
Python
193 lines
6.6 KiB
Python
from Tkinter import *
|
|
|
|
from ElementFrame import getElements
|
|
from utilities import handleError
|
|
from Cantera import *
|
|
from config import *
|
|
from SpeciesFrame import getSpecies
|
|
|
|
def testit():
|
|
pass
|
|
|
|
class EditFrame(Frame):
|
|
|
|
def redraw(self):
|
|
try:
|
|
self.eframe.destroy()
|
|
self.sframe.destroy()
|
|
self.rframe.destroy()
|
|
except:
|
|
pass
|
|
self.addElementFrame()
|
|
self.addSpeciesFrame()
|
|
self.addReactionFrame()
|
|
|
|
def __init__(self, master, app):
|
|
Frame.__init__(self, master)
|
|
self.mix = app.mix
|
|
print self.mix, dir(self.mix)
|
|
self.app = app
|
|
self.master = master
|
|
self.master.title("Cantera Mechanism Editor")
|
|
self.redraw()
|
|
|
|
def addReactionFrame(self):
|
|
self.rframe = Frame(self)
|
|
self.rframe.config(relief=GROOVE,bd=4)
|
|
self.rframe.grid(row=2,column=0,columnspan=10,sticky=E+W)
|
|
b=Button(self.rframe,text='Reactions',command=testit)
|
|
b.grid(column=5, row=0)
|
|
|
|
def addElementFrame(self):
|
|
self.eframe = Frame(self)
|
|
self.eframe.config(relief=GROOVE,bd=4)
|
|
self.eframe.grid(row=0,column=0,columnspan=10,sticky=E+W)
|
|
self.element_labels = []
|
|
n = 0
|
|
for el in self.mix._mech.elementNames():
|
|
x = Label(self.eframe,text=el,fg='darkblue')
|
|
x.grid(column = n, row=0)
|
|
self.element_labels.append(x)
|
|
n = n + 1
|
|
b=Button(self.eframe,text='Element',command=self.chooseElements, default=ACTIVE)
|
|
b.grid(column=0, row=1, columnspan=10)
|
|
|
|
|
|
def addSpeciesFrame(self):
|
|
self.sframe = Frame(self)
|
|
self.sframe.config(relief=GROOVE,bd=4)
|
|
self.sframe.grid(row=1,column=0,columnspan=10,sticky=E+W)
|
|
r = 0
|
|
c = 0
|
|
splist = self.app.species
|
|
self.spcheck = []
|
|
self.spec = []
|
|
for i in range(self.app.mech.nSpecies()):
|
|
self.spec.append(IntVar())
|
|
self.spec[i].set(1)
|
|
self.spcheck.append( Checkbutton(self.sframe,
|
|
text=splist[i].name,
|
|
variable=self.spec[i],
|
|
onvalue = 1, offvalue = 0) )
|
|
self.spcheck[i].grid(row = r, column = c, sticky = N+W)
|
|
self.spcheck[i].bind("<Button-3>", self.editSpecies)
|
|
c = c + 1
|
|
if c > 4:
|
|
c, r = 0, r + 1
|
|
|
|
def getspecies(self):
|
|
print getSpecies(self.mix.speciesNames(),
|
|
self.mix.speciesNames())
|
|
|
|
def editSpecies(self, event=None):
|
|
e = Toplevel(event.widget.master)
|
|
w = event.widget
|
|
txt = w.cget('text')
|
|
sp = self.app.mix.species[txt]
|
|
|
|
# name, etc.
|
|
e1 = Frame(e, relief=FLAT)
|
|
self.addEntry(e1,'Name',0,0,sp.name)
|
|
self.addEntry(e1,'ID Tag',1,0,sp.id)
|
|
self.addEntry(e1,'Phase',2,0,sp.phase)
|
|
e1.grid(row=0,column=0)
|
|
|
|
# elements
|
|
elframe = Frame(e)
|
|
elframe.grid(row=1,column=0)
|
|
Label(elframe,text='Elemental Composition').grid(row=0,column=0,columnspan=2,sticky=E+W)
|
|
|
|
i = 0
|
|
for el in self.app.mech.elementNames():
|
|
self.addEntry(elframe,el,i,0,self.mech.nAtoms(sp, el))
|
|
i = i + 1
|
|
|
|
# thermo
|
|
thframe = Frame(e)
|
|
thframe.grid(row=0,rowspan=2,column=1)
|
|
thframe.config(relief=GROOVE,bd=4)
|
|
i = 0
|
|
Label(thframe,text='Thermodynamic Properties').grid(row=0,
|
|
column=0, columnspan=4, sticky=E+W)
|
|
if isinstance(sp.thermoParam(),NasaPolynomial):
|
|
Label(thframe,text='Parametrization:').grid(row=1,column=1)
|
|
self.addEntry(thframe,'',2,0,'NasaPolynomial')
|
|
Label(thframe,text='Temperatures (min, mid, max):').grid(row=3,column=1)
|
|
self.addEntry(thframe,'',4,0,`sp.minTemp`)
|
|
self.addEntry(thframe,'',5,0,`sp.midTemp`)
|
|
self.addEntry(thframe,'',6,0,`sp.maxTemp`)
|
|
low = Frame(thframe)
|
|
low.config(relief=GROOVE,bd=4)
|
|
low.grid(row=1,rowspan=6,column=3,columnspan=2)
|
|
Label(low,text='Coefficients for the Low\n Temperature Range').grid(row=0,column=0,columnspan=2,sticky=E+W)
|
|
c = sp.thermoParam().coefficients(sp.minTemp)
|
|
for j in range(7):
|
|
self.addEntry(low,'a'+`j`,j+3,0,`c[j]`)
|
|
high = Frame(thframe)
|
|
high.config(relief=GROOVE,bd=4)
|
|
high.grid(row=1,rowspan=6,column=5,columnspan=2)
|
|
Label(high,text='Coefficients for the High\n Temperature Range').grid(row=0,column=0,columnspan=2,sticky=E+W)
|
|
c = sp.thermoParam().coefficients(sp.maxTemp)
|
|
for j in range(7):
|
|
self.addEntry(high,'a'+`j`,j+3,0,`c[j]`)
|
|
|
|
com = Frame(e)
|
|
com.grid(row=10,column=0,columnspan=5)
|
|
ok = Button(com,text='OK',default=ACTIVE)
|
|
ok.grid(row=0,column=0)
|
|
ok.bind('<1>',self.modifySpecies)
|
|
Button(com,text='Cancel',command=e.destroy).grid(row=0,column=1)
|
|
self.especies = e
|
|
|
|
def modifySpecies(self,event=None):
|
|
button = event.widget
|
|
e = self.especies
|
|
for fr in e.children.values():
|
|
for item in fr.children.values():
|
|
try:
|
|
print item.cget('selection')
|
|
except:
|
|
pass
|
|
e.destroy()
|
|
|
|
def addEntry(self,master,name,row,column,text):
|
|
if name:
|
|
Label(master, text=name).grid(row=row, column=column)
|
|
nm = Entry(master)
|
|
nm.grid(row=row, column=column+1)
|
|
nm.insert(END,text)
|
|
|
|
def chooseElements(self):
|
|
oldel = self.mix.g.elementNames()
|
|
newel = getElements(self.mix.g.elementNames())
|
|
removeList = []
|
|
for el in oldel:
|
|
if not el in newel:
|
|
removeList.append(el)
|
|
#self.app.mech.removeElements(removeList)
|
|
addList = []
|
|
for el in newel:
|
|
if not el in oldel:
|
|
addList.append(el)
|
|
#self.app.mech.addElements(addList)
|
|
try:
|
|
self.redraw()
|
|
self.app.makeWindows()
|
|
except:
|
|
handleError('Edit err')
|
|
|
|
self.app.mix = IdealGasMixture(self.app.mech)
|
|
self.mix = self.app.mix
|
|
nn = self.mix.speciesList[0].name
|
|
self.mix.set(temperature = 300.0, pressure = 101325.0, moles = {nn:1.0})
|
|
for label in self.element_labels:
|
|
label.destroy()
|
|
self.element_labels = []
|
|
n = 0
|
|
for el in self.mix._mech.elementList():
|
|
x = Label(self.eframe,text=el.symbol(),fg='darkblue')
|
|
x.grid(column = n, row=0)
|
|
self.element_labels.append(x)
|
|
n = n + 1
|
|
|
|
self.app.makeWindows()
|