1135 lines
39 KiB
Python
Executable File
1135 lines
39 KiB
Python
Executable File
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import sys, os, random, copy, struct
|
|
|
|
from collections import deque
|
|
from operator import attrgetter
|
|
|
|
import xml.etree.ElementTree as etree
|
|
|
|
import pylab
|
|
from PyQt4 import QtGui, QtCore
|
|
|
|
from numpy import arange, sin, pi, array, linspace, arange
|
|
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
|
|
from matplotlib.figure import Figure
|
|
from matplotlib.lines import Line2D
|
|
from matplotlib.path import Path
|
|
import matplotlib.patches as patches
|
|
|
|
from mpltools import annotation
|
|
|
|
progname = os.path.basename(sys.argv[0])
|
|
progversion = "0.1"
|
|
|
|
def calc_colors(count):
|
|
if count == 1:
|
|
return [QtGui.QColor(0, 255, 0),]
|
|
r = 0
|
|
g = 255
|
|
step = int(512. / (count-1))
|
|
colors = list()
|
|
for i in range(count):
|
|
colors.append(QtGui.QColor(r, g, 0))
|
|
if r < 255:
|
|
r += step
|
|
if r > 255:
|
|
g -= (r - 256)
|
|
r = 255
|
|
g = max(0, g)
|
|
else:
|
|
g -= step
|
|
g = max(0, g)
|
|
return colors
|
|
|
|
def set_colors(temp_levels):
|
|
colors = calc_colors(len(temp_levels) - 1)
|
|
ix = 0
|
|
for temp_level in temp_levels:
|
|
if not temp_level.is_env:
|
|
temp_level.color = colors[ix]
|
|
ix += 1
|
|
else:
|
|
temp_level.color = QtGui.QColor("black")
|
|
|
|
|
|
|
|
def create_profile_header(x_list, y_list):
|
|
tpl = "#ifndef _H_SOLDER_PROFILE\n" \
|
|
"#define _H_SOLDER_PROFILE\n" \
|
|
"\n" \
|
|
"%s\n" \
|
|
"\n" \
|
|
"};\n" \
|
|
"\n" \
|
|
"#endif\n"
|
|
|
|
data_tpl = "unsigned int profile_%d[%d][2] = \n%s\n" \
|
|
|
|
s = list()
|
|
|
|
ix = 0
|
|
for x, y in zip(x_list, y_list):
|
|
tmp = list()
|
|
for a, b in zip(x, y):
|
|
tmp.append("{%d, %d}" % (a, b))
|
|
s.append(data_tpl % (ix, len(x), "\n".join(tmp)))
|
|
ix += 1
|
|
|
|
open("test.h", "w").write(tpl % ("\n".join(s)))
|
|
|
|
def create_profile_packet(x, y):
|
|
# msg = [length, x_0, y_0,x_1, y_1, ..., x_length-1, y_length-1]
|
|
|
|
tmp = [len(x) * 2 * 2,]
|
|
|
|
for a, b in zip(x, y):
|
|
tmp.append(a)
|
|
tmp.append(b)
|
|
|
|
tpl = "H" + (len(x) * 2) * "H"
|
|
res = struct.pack(tpl, *tmp)
|
|
return res
|
|
|
|
def getTemperature():
|
|
return 20.
|
|
|
|
|
|
class TempLevel(object):
|
|
def __init__(self, name, temp, is_env=False):
|
|
self.name = name
|
|
self.temp = temp
|
|
self.is_env = is_env
|
|
self.color = None
|
|
|
|
|
|
class Solder(object):
|
|
|
|
def __init__(self, name=str(), description=str()):
|
|
self.name = name
|
|
self.description = description
|
|
self.temp_levels = list()
|
|
self.durations = list()
|
|
self.rates = list()
|
|
self.changed = False
|
|
|
|
def __unicode__(self):
|
|
return unicode(self.name)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def add_temp_level(self, name, temp, is_env):
|
|
s = TempLevel(name, temp, is_env)
|
|
self.temp_levels.append(s)
|
|
return s
|
|
|
|
def add_rate(self, states, rate):
|
|
self.rates.append([states, rate])
|
|
|
|
def add_duration(self, states, duration):
|
|
self.durations.append([states, duration])
|
|
|
|
def get_temp_level_by_name(self, name):
|
|
assert isinstance(name, basestring)
|
|
for i in self.temp_levels:
|
|
if i.name == name:
|
|
return i
|
|
return None
|
|
|
|
def get_temp_level(self, ix):
|
|
assert isinstance(ix, int)
|
|
return self.temp_levels[ix]
|
|
|
|
def calc_profile(self):
|
|
|
|
x = list()
|
|
y = list()
|
|
duration_points = dict()
|
|
rate_points = dict()
|
|
self.time = 0
|
|
used_temp_levels = set()
|
|
for ix, temp_level in enumerate(self.temp_levels):
|
|
if temp_level != self.temp_levels[0] and temp_level not in used_temp_levels:
|
|
ix = self.temp_levels.index(temp_level)
|
|
raise ValueError("TempLevel %r not connected to %r" % (self.temp_levels[ix-1].name, self.temp_levels[ix].name))
|
|
|
|
temp_levels = None
|
|
duration = None
|
|
for sts, dur in self.durations:
|
|
if sts and sts[0] == temp_level:
|
|
duration = dur
|
|
temp_levels = sts
|
|
break
|
|
|
|
if temp_level not in used_temp_levels:
|
|
used_temp_levels.add(temp_level)
|
|
x.append(self.time)
|
|
y.append(temp_level.temp)
|
|
|
|
if duration is not None:
|
|
if len(temp_levels) == 3:
|
|
used_temp_levels.add(temp_levels[1])
|
|
used_temp_levels.add(temp_levels[2])
|
|
|
|
self.time += duration / 2
|
|
x.append(self.time)
|
|
y.append(temp_levels[1].temp)
|
|
|
|
self.time += duration / 2
|
|
x.append(self.time)
|
|
y.append(temp_levels[2].temp)
|
|
|
|
duration_points[ix] = (x[-3:], y[-3:])
|
|
else:
|
|
y.append(temp_levels[1].temp)
|
|
used_temp_levels.add(temp_levels[1])
|
|
self.time += duration
|
|
x.append(self.time)
|
|
duration_points[ix] = (x[-2:], y[-2:])
|
|
else:
|
|
for ex, (sts, rate) in enumerate(self.rates):
|
|
if sts and sts[0] == temp_level:
|
|
used_temp_levels.add(sts[1])
|
|
duration = (sts[1].temp - temp_level.temp) / rate
|
|
self.time += duration
|
|
x.append(self.time)
|
|
y.append(sts[1].temp)
|
|
rate_points[ex] = (x[-2:], y[-2:])
|
|
|
|
return array(map(float, x)), array(map(float, y)), max(x), max(y), duration_points, rate_points
|
|
|
|
|
|
@staticmethod
|
|
def unpack(filename):
|
|
xmltree = etree.parse(filename)
|
|
root = xmltree.getroot()
|
|
solder_node = root[0]
|
|
s = Solder(solder_node.attrib["name"], solder_node.attrib["description"])
|
|
env_count = 0
|
|
for temp_level in solder_node.findall("state"):
|
|
tstr = temp_level.attrib["temperature"]
|
|
is_env = False
|
|
try:
|
|
temp = int(tstr)
|
|
except ValueError:
|
|
if tstr == "$ENV":
|
|
temp = getTemperature()
|
|
is_env = True
|
|
env_count += 1
|
|
s.add_temp_level(temp_level.attrib["name"], temp, is_env)
|
|
|
|
set_colors(s.temp_levels)
|
|
|
|
for duration in solder_node.findall("duration"):
|
|
temp_levels = list()
|
|
for temp_level in duration:
|
|
temp_levels.append(s.get_temp_level_by_name(temp_level.attrib["name"]))
|
|
s.add_duration(temp_levels, int(duration.attrib["value"]))
|
|
for rate in solder_node.findall("rate"):
|
|
temp_levels = list()
|
|
for temp_level in rate:
|
|
temp_levels.append(s.get_temp_level_by_name(temp_level.attrib["name"]))
|
|
s.add_rate(temp_levels, int(rate.attrib["value"]))
|
|
|
|
return s
|
|
|
|
def save(self, filename):
|
|
if self.changed:
|
|
solder_node = etree.Element("solder_type", {"name" : self.name, "description" : self.description})
|
|
for temp_level in self.temp_levels:
|
|
temp = temp_level.is_env and "$ENV" or str(temp_level.temp)
|
|
solder_node.append(etree.Element("state", {"name" : temp_level.name, "temperature" : temp}))
|
|
for temp_levels, value in self.durations:
|
|
duration = etree.Element("duration", {"value" : str(value)})
|
|
for temp_level in temp_levels:
|
|
duration.append(etree.Element("state", {"name" : temp_level.name}))
|
|
solder_node.append(duration)
|
|
for temp_levels, value in self.rates:
|
|
rate = etree.Element("rate", {"value" : str(value)})
|
|
for temp_level in temp_levels:
|
|
rate.append(etree.Element("state", {"name" : temp_level.name}))
|
|
solder_node.append(rate)
|
|
|
|
dirname = os.path.join(os.path.dirname(__file__), "solder_types")
|
|
root = etree.Element("xml")
|
|
root.append(solder_node)
|
|
etree.ElementTree(root).write(os.path.join(dirname, self.name + ".xml"), "UTF-8", True)
|
|
self.changed = False
|
|
|
|
|
|
|
|
class SolderListModel(QtCore.QAbstractListModel):
|
|
def __init__(self, parent=None, *args):
|
|
""" datain: a list where each item is a row
|
|
"""
|
|
super(SolderListModel, self).__init__(parent, *args)
|
|
|
|
self.reload()
|
|
|
|
def rowCount(self, parent=QtCore.QModelIndex()):
|
|
return len(self.listdata)
|
|
|
|
def reload(self):
|
|
dirname = os.path.join(os.path.dirname(__file__), "solder_types")
|
|
dirlisting = filter(lambda x: os.path.splitext(x)[1] == ".xml", os.listdir(dirname))
|
|
self.listdata = []
|
|
for p in dirlisting:
|
|
self.listdata.append(Solder.unpack(os.path.join(dirname, p)))
|
|
self.listdata.sort(key=lambda x: x.name)
|
|
self.reset()
|
|
|
|
def headerData(self, col, orientation, role):
|
|
if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
|
|
return QtCore.QVariant("Solder Paste")
|
|
return QtCore.QVariant()
|
|
|
|
def data(self, index, role):
|
|
if index.isValid() and role == QtCore.Qt.DisplayRole:
|
|
solder = self.listdata[index.row()]
|
|
if solder.changed:
|
|
return QtCore.QVariant(solder.name + " *")
|
|
else:
|
|
return QtCore.QVariant(solder.name)
|
|
else:
|
|
return QtCore.QVariant()
|
|
|
|
def setData(self, index, variant, role):
|
|
if index.isValid() and role == QtCore.Qt.EditRole:
|
|
new_name = str(variant.toString())
|
|
if new_name and new_name != "":
|
|
self.listdata[index.row()].name = new_name
|
|
self.listdata[index.row()].changed = True
|
|
return True
|
|
return False
|
|
|
|
def flags(self, index):
|
|
if not index.isValid():
|
|
return 0
|
|
return QtCore.Qt.ItemFlags(QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable)
|
|
|
|
def create_solder(self):
|
|
solder = Solder("new", "")
|
|
|
|
tl = solder.add_temp_level("environment temp", getTemperature(), True)
|
|
tl.color = QtGui.QColor(0, 0, 0)
|
|
self.listdata.append(solder)
|
|
self.reset()
|
|
|
|
|
|
class TempLevelModel(QtCore.QAbstractTableModel):
|
|
solder_changed = QtCore.pyqtSignal()
|
|
|
|
def __init__(self, parent):
|
|
super(TempLevelModel, self).__init__(parent)
|
|
self._changed = False
|
|
self.temp_levels = list()
|
|
self.headerdata = [u"Name", u"Temperature (°C)"]
|
|
|
|
def headerData(self, col, orientation, role):
|
|
if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
|
|
return QtCore.QVariant(self.headerdata[col])
|
|
return QtCore.QVariant()
|
|
|
|
def rowCount(self, parent):
|
|
return len(self.temp_levels)
|
|
|
|
def columnCount(self, parent):
|
|
return 2
|
|
|
|
def data(self, index, role):
|
|
if not index.isValid():
|
|
return QtCore.QVariant()
|
|
|
|
if role == QtCore.Qt.DisplayRole:
|
|
col = index.column()
|
|
if col == 0:
|
|
return QtCore.QVariant(self.temp_levels[index.row()].name)
|
|
else:
|
|
return QtCore.QVariant(self.temp_levels[index.row()].temp)
|
|
|
|
if index.column() == 0 and role == QtCore.Qt.DecorationRole:
|
|
p = QtGui.QPixmap(10,10)
|
|
color = self.temp_levels[index.row()].color
|
|
p.fill(color)
|
|
return p
|
|
|
|
return QtCore.QVariant()
|
|
|
|
def flags(self, index):
|
|
if not index.isValid():
|
|
return 0
|
|
return QtCore.Qt.ItemFlags(QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable)
|
|
|
|
def setData(self, index, variant, role):
|
|
if index.isValid() and role == QtCore.Qt.EditRole:
|
|
col = index.column()
|
|
if col == 0:
|
|
self.temp_levels[index.row()].name = str(variant.toString())
|
|
elif col == 1:
|
|
self.temp_levels[index.row()].temp = variant.toInt()[0]
|
|
self.solder_changed.emit()
|
|
return True
|
|
return False
|
|
|
|
def remove_temp_level(self, index):
|
|
tmp = self.temp_levels[index]
|
|
del self.temp_levels[index]
|
|
self.reset()
|
|
self.solder_changed.emit()
|
|
return tmp
|
|
|
|
def add_temp_level(self, index, temp_level):
|
|
self.temp_levels.insert(index.row() + 1, temp_level)
|
|
set_colors(self.temp_levels)
|
|
self.reset()
|
|
self.solder_changed.emit()
|
|
|
|
def setTempLevels(self, temp_levels):
|
|
assert isinstance(temp_levels, list)
|
|
self.temp_levels = temp_levels
|
|
self.reset()
|
|
|
|
|
|
def clear(self):
|
|
self.temp_levels = list()
|
|
self.reset()
|
|
|
|
|
|
class Plotter(FigureCanvas):
|
|
"""A canvas that updates itself every second with a new plot."""
|
|
|
|
def __init__(self, parent, myapp, width, height, dpi):
|
|
self.fig = Figure(figsize=(width, height), dpi=dpi)
|
|
super(Plotter, self).__init__(self.fig)
|
|
self.axes = self.fig.add_subplot(111)
|
|
|
|
self.axes.set_axis_bgcolor('white')
|
|
self.axes.set_title(u'reflow profile', size=12)
|
|
self.axes.set_xlabel(u'time (seconds)', size=12)
|
|
self.axes.set_ylabel(u'temperature (°C)', size=12)
|
|
self.axes.set_ymargin(0)
|
|
self.axes.set_xmargin(0)
|
|
|
|
self.setParent(parent)
|
|
self.myapp = myapp
|
|
self.solder = None
|
|
|
|
self.plot_data, = self.axes.plot([], linewidth=1.0, color=(0,0,1), zorder=10)
|
|
#self.selection_data, = self.axes.plot([], linewidth=1.0, color=(1,1,1), zorder=5)
|
|
|
|
FigureCanvas.setSizePolicy(self,
|
|
QtGui.QSizePolicy.Expanding,
|
|
QtGui.QSizePolicy.Expanding)
|
|
FigureCanvas.updateGeometry(self)
|
|
|
|
timer = QtCore.QTimer(self)
|
|
|
|
QtCore.QObject.connect(timer, QtCore.SIGNAL("timeout()"), self.update_figure)
|
|
timer.start(1000)
|
|
self.updated = True
|
|
|
|
def update_figure(self):
|
|
if self.updated:
|
|
updated = False
|
|
self.axes.patches = list()
|
|
self.axes.texts = list()
|
|
self.x = list()
|
|
self.y = list()
|
|
|
|
try:
|
|
self.x, self.y, self.xmax, self.ymax, self.duration_points, self.rate_points = self.solder.calc_profile()
|
|
|
|
for ix, (a, b) in enumerate(zip(self.x[:-1], self.y[:-1])):
|
|
annotation.slope_marker((a + 10, b), (self.y[ix+1] - b) / (self.x[ix+1] - a), ax=self.axes)
|
|
except Exception, e:
|
|
self.xmax = 500
|
|
self.ymax = 300
|
|
|
|
self.plot_data.set_xdata(self.x)
|
|
self.plot_data.set_ydata(self.y)
|
|
|
|
self.axes.set_xbound(lower=0, upper=self.xmax + 20)
|
|
self.axes.set_ybound(lower=0, upper=self.ymax + 20)
|
|
|
|
self.axes.set_yticks([state.temp for state in self.solder.temp_levels])
|
|
self.axes.set_xticks(self.x)
|
|
|
|
#duration_widget = self.myapp.duration_widget
|
|
|
|
#self.selection_data.set_xdata(array(da))
|
|
#self.selection_data.set_ydata(array(db))
|
|
|
|
self.fig.lines = lines = list()
|
|
for temp_level in self.solder.temp_levels:
|
|
if not temp_level.is_env:
|
|
line = Line2D([0, self.xmax + 20], [temp_level.temp, temp_level.temp],
|
|
transform=self.axes.transData, figure=self.fig, color=str(temp_level.color.name()), label="name", zorder=1)
|
|
lines.append(line)
|
|
|
|
self.axes.legend(("Estimated profile",))
|
|
self.draw()
|
|
|
|
def solder_changed(self):
|
|
self.solder.changed = True
|
|
|
|
def setData(self, solder):
|
|
self.solder = solder
|
|
self.updated = True
|
|
|
|
|
|
class AddRemoveWidget(QtGui.QWidget):
|
|
def __init__(self, parent, with_upown=True):
|
|
#super(AddRemoveWidget, self).__init__(parent)
|
|
QtGui.QWidget.__init__(self, parent)
|
|
self.add_button = QtGui.QPushButton(QtGui.QIcon.fromTheme("list-add"), "")
|
|
self.remove_button = QtGui.QPushButton(QtGui.QIcon.fromTheme("list-remove"), "")
|
|
|
|
sh = "QPushButton:disabled { background-color: #555555; }"
|
|
self.remove_button.setStyleSheet(sh)
|
|
self.add_button.setStyleSheet(sh)
|
|
|
|
self._layout = QtGui.QVBoxLayout(self)
|
|
|
|
self._layout.addWidget(self.add_button)
|
|
self._layout.addWidget(self.remove_button)
|
|
if with_upown:
|
|
self.up_button = QtGui.QPushButton(QtGui.QIcon.fromTheme("go-up"), "")
|
|
self.down_button = QtGui.QPushButton(QtGui.QIcon.fromTheme("go-down"), "")
|
|
self._layout.addWidget(self.up_button)
|
|
self._layout.addWidget(self.down_button)
|
|
self.up_button.setStyleSheet(sh)
|
|
self.down_button.setStyleSheet(sh)
|
|
self._layout.addStretch(4)
|
|
|
|
|
|
class AddRemoveValueWidget(AddRemoveWidget):
|
|
def __init__(self, parent):
|
|
super(AddRemoveValueWidget, self).__init__(parent)
|
|
self.value = QtGui.QSpinBox(self)
|
|
self.value.setRange(-500, 500)
|
|
self._layout.addWidget(self.value)
|
|
|
|
|
|
class ConstraintListModel(QtCore.QAbstractListModel):
|
|
def __init__(self, parent=None, *args):
|
|
""" datain: a list where each item is a row
|
|
"""
|
|
super(ConstraintListModel, self).__init__(parent, *args)
|
|
self.constraint_list = list()
|
|
|
|
def rowCount(self, parent=QtCore.QModelIndex()):
|
|
return len(self.constraint_list)
|
|
|
|
def headerData(self, col, orientation, role):
|
|
if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
|
|
return QtCore.QVariant("Solder Paste")
|
|
return QtCore.QVariant()
|
|
|
|
def data(self, index, role):
|
|
if index.isValid() and role == QtCore.Qt.DisplayRole:
|
|
return QtCore.QVariant(index.row() + 1)
|
|
else:
|
|
return QtCore.QVariant()
|
|
|
|
def append_constraint(self):
|
|
self.constraint_list.append([[], 0])
|
|
self.reset()
|
|
|
|
def remove_constraint(self):
|
|
del self.constraint_list[self.currentIndex().row()]
|
|
self.reset()
|
|
|
|
|
|
class ConstraintWidget(QtGui.QWidget):
|
|
solder_changed = QtCore.pyqtSignal()
|
|
def __init__(self, name):
|
|
super(ConstraintWidget, self).__init__()
|
|
self.name = name
|
|
|
|
self.spinbox_block = False
|
|
|
|
self.constraint_model = ConstraintListModel(self) # constraint selection
|
|
self.all_temp_levels = TempLevelModel(self) # temp_level selection pool
|
|
self.selected_temp_levels = TempLevelModel(self) # selected temp_levels
|
|
|
|
self.controls = AddRemoveValueWidget(self)
|
|
self.constraint_controls = AddRemoveWidget(self, False)
|
|
|
|
self.controls.add_button.setText(u"")
|
|
self.controls.remove_button.setText(u"")
|
|
|
|
self.constraint_controls.add_button.setText(u"")
|
|
self.constraint_controls.remove_button.setText(u"")
|
|
|
|
self.constraint_view = QtGui.QListView(self)
|
|
self.constraint_view.setModel(self.constraint_model)
|
|
|
|
self.all_temp_levels_view = QtGui.QListView(self)
|
|
self.all_temp_levels_view.setModel(self.all_temp_levels)
|
|
|
|
self.selected_temp_levels_view = QtGui.QListView(self)
|
|
self.selected_temp_levels_view.setModel(self.selected_temp_levels)
|
|
|
|
h = QtGui.QHBoxLayout()
|
|
h.addWidget(self.constraint_view, 1)
|
|
h.addWidget(self.constraint_controls, 1)
|
|
h.addWidget(self.all_temp_levels_view, 3)
|
|
h.addWidget(self.controls, 1)
|
|
h.addWidget(self.selected_temp_levels_view, 3)
|
|
|
|
self.controls.add_button.setIcon(QtGui.QIcon.fromTheme("go-next"))
|
|
self.controls.remove_button.setIcon(QtGui.QIcon.fromTheme("go-previous"))
|
|
self.setLayout(h)
|
|
|
|
self.connect(
|
|
self.constraint_view,
|
|
QtCore.SIGNAL("clicked(QModelIndex)"),
|
|
self._constraint_selected)
|
|
|
|
self.connect(
|
|
self.controls.add_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.add_temp_level_to_constraint)
|
|
|
|
self.connect(
|
|
self.controls.remove_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.remove_temp_level_from_constraint)
|
|
|
|
self.connect(
|
|
self.controls.up_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.temp_level_up)
|
|
|
|
self.connect(
|
|
self.controls.down_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.temp_level_down)
|
|
|
|
self.connect(
|
|
self.constraint_controls.add_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.add_constraint)
|
|
|
|
self.connect(
|
|
self.constraint_controls.remove_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.remove_constraint)
|
|
|
|
self.connect(
|
|
self.controls.value,
|
|
QtCore.SIGNAL("valueChanged(int)"),
|
|
self.constraint_value_changed)
|
|
|
|
self.connect(
|
|
self.selected_temp_levels_view,
|
|
QtCore.SIGNAL("clicked(QModelIndex)"),
|
|
self.selected_temp_levels_clicked)
|
|
|
|
def setData(self, solder):
|
|
self.solder = solder
|
|
self.all_temp_levels.setTempLevels(solder.temp_levels)
|
|
self._set_data(solder)
|
|
self.set_controls()
|
|
|
|
self.controls.up_button.setEnabled(False)
|
|
self.controls.down_button.setEnabled(False)
|
|
|
|
def set_controls(self):
|
|
if not self.constraint_model.constraint_list:
|
|
self.controls.value.setEnabled(False)
|
|
self.constraint_controls.remove_button.setEnabled(False)
|
|
self.controls.add_button.setEnabled(False)
|
|
self.controls.remove_button.setEnabled(False)
|
|
self.controls.up_button.setEnabled(False)
|
|
self.controls.down_button.setEnabled(False)
|
|
else:
|
|
self.controls.value.setEnabled(True)
|
|
self.constraint_controls.remove_button.setEnabled(True)
|
|
self.controls.add_button.setEnabled(True)
|
|
self.controls.remove_button.setEnabled(True)
|
|
self.controls.up_button.setEnabled(True)
|
|
self.controls.down_button.setEnabled(True)
|
|
|
|
|
|
def selected_temp_levels_clicked(self, index):
|
|
if index.isValid():
|
|
self.controls.up_button.setEnabled(True)
|
|
self.controls.down_button.setEnabled(True)
|
|
|
|
|
|
def add_constraint(self):
|
|
self.constraint_model.append_constraint()
|
|
self.set_controls()
|
|
self.solder_changed.emit()
|
|
|
|
def _constraint_selected(self, index):
|
|
raise NotImplementedError()
|
|
|
|
def _set_data(self, solder):
|
|
raise NotImplementedError()
|
|
|
|
def add_temp_level_to_constraint(self):
|
|
src_row = self.all_temp_levels_view.currentIndex().row()
|
|
temp_level = self.all_temp_levels.temp_levels[src_row]
|
|
self.selected_temp_levels.temp_levels.append(temp_level)
|
|
self.selected_temp_levels.reset()
|
|
self.selected_temp_levels_view.setCurrentIndex(self.selected_temp_levels.index(len(self.selected_temp_levels.temp_levels)-1,0))
|
|
self.solder_changed.emit()
|
|
|
|
def remove_temp_level_from_constraint(self):
|
|
del self.selected_temp_levels.temp_levels[self.selected_temp_levels_view.currentIndex().row()]
|
|
self.selected_temp_levels.reset()
|
|
self.selected_temp_levels_view.clearSelection()
|
|
self.solder_changed.emit()
|
|
|
|
def remove_constraint(self):
|
|
src_row = self.all_temp_levels_view.currentIndex().row()
|
|
del self.constraint_model.constraint_list[src_row]
|
|
self.constraint_model.reset()
|
|
self.selected_temp_levels.clear()
|
|
self.set_controls()
|
|
self.solder_changed.emit()
|
|
|
|
def constraint_value_changed(self, value):
|
|
if self.spinbox_block:
|
|
self.spinbox_block = False
|
|
return
|
|
src_index = self.constraint_view.currentIndex().row()
|
|
self.constraint_model.constraint_list[src_index][1] = value
|
|
self.solder_changed.emit()
|
|
|
|
def slot_temp_level_removed(self, temp_level):
|
|
deletes = deque()
|
|
ix = 0
|
|
for temp_levels, v in self.constraint_model.constraint_list:
|
|
if temp_level in temp_levels:
|
|
deletes.appendleft(ix)
|
|
ix += 1
|
|
for i in deletes:
|
|
del self.constraint_model.constraint_list[i]
|
|
self.reset()
|
|
self.solder_changed.emit()
|
|
|
|
def temp_level_up(self):
|
|
dst_row = self.selected_temp_levels_view.currentIndex().row()
|
|
self.selected_temp_levels.temp_levels[dst_row - 1], self.selected_temp_levels.temp_levels[dst_row] = self.selected_temp_levels.temp_levels[dst_row], self.selected_temp_levels.temp_levels[dst_row - 1]
|
|
self.selected_temp_levels.reset()
|
|
self.solder_changed.emit()
|
|
|
|
def temp_level_down(self):
|
|
dst_row = self.selected_temp_levels_view.currentIndex().row()
|
|
self.selected_temp_levels.temp_levels[dst_row], self.selected_temp_levels.temp_levels[dst_row + 1] = self.selected_temp_levels.temp_levels[dst_row + 1], self.selected_temp_levels.temp_levels[dst_row]
|
|
self.selected_temp_levels.reset()
|
|
self.solder_changed.emit()
|
|
|
|
|
|
class DurationConstraintWidget(ConstraintWidget):
|
|
|
|
def _set_data(self, solder):
|
|
self.spinbox_block = True
|
|
self.constraint_model.constraint_list = solder.durations
|
|
ix = self.constraint_model.index(0, 0)
|
|
self._constraint_selected(ix)
|
|
self.constraint_view.setCurrentIndex(ix)
|
|
|
|
def _constraint_selected(self, index):
|
|
if index.isValid():
|
|
self.spinbox_block = True
|
|
temp_levels, value = self.constraint_model.constraint_list[index.row()]
|
|
self.selected_temp_levels.setTempLevels(temp_levels)
|
|
self.controls.value.setValue(value)
|
|
else:
|
|
self.spinbox_block = True
|
|
self.selected_temp_levels.setTempLevels([])
|
|
self.controls.value.setValue(0)
|
|
|
|
|
|
class OvenControlsWidget(QtGui.QWidget):
|
|
def __init__(self, parent=None):
|
|
super(OvenControlsWidget, self).__init__(parent)
|
|
|
|
self.left = QtGui.QWidget(self)
|
|
|
|
self.temp_lcd = QtGui.QLCDNumber(self)
|
|
self.time_lcd = QtGui.QLCDNumber(self)
|
|
|
|
self.time_lcd.setDigitCount(3)
|
|
self.temp_lcd.setDigitCount(3)
|
|
|
|
palette = QtGui.QPalette()
|
|
palette.setColor(QtGui.QPalette.WindowText, QtCore.Qt.black)
|
|
self.temp_lcd.setPalette(palette)
|
|
self.time_lcd.setPalette(palette)
|
|
self.time_lcd.setSmallDecimalPoint(False)
|
|
self.temp_lcd.setSmallDecimalPoint(False)
|
|
|
|
self.temp_lcd.setSegmentStyle(2)
|
|
self.time_lcd.setSegmentStyle(2)
|
|
self.time_lcd.display("000")
|
|
self.temp_lcd.display("142")
|
|
|
|
self.temp_label = QtGui.QLabel("Time", self)
|
|
self.time_label = QtGui.QLabel("Temp", self)
|
|
self.temp_label.setBuddy(self.temp_lcd)
|
|
self.time_label.setBuddy(self.time_lcd)
|
|
|
|
self.sicon = QtGui.QIcon.fromTheme("media-playback-start")
|
|
self.picon = QtGui.QIcon.fromTheme("media-playback-pause")
|
|
self.start_button = QtGui.QPushButton(self.sicon, "start")
|
|
self.start_button.setCheckable(True)
|
|
|
|
layout = QtGui.QHBoxLayout(self.left)
|
|
layout.addWidget(self.time_label)
|
|
layout.addWidget(self.time_lcd)
|
|
layout.addWidget(self.temp_label)
|
|
layout.addWidget(self.temp_lcd)
|
|
layout.addWidget(self.start_button)
|
|
|
|
layout2 = QtGui.QHBoxLayout()
|
|
self.temp_level_widget = TempLevelWidget(self, True)
|
|
layout2.addWidget(self.temp_level_widget, 2)
|
|
layout2.addWidget(self.left, 4)
|
|
self.setLayout(layout2)
|
|
|
|
|
|
self.connect(
|
|
self.start_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.toggle_button)
|
|
|
|
def toggle_button(self):
|
|
if self.start_button.isChecked():
|
|
self.start_button.setIcon(self.picon)
|
|
else:
|
|
self.start_button.setIcon(self.sicon)
|
|
|
|
|
|
class RateConstraintWidget(ConstraintWidget):
|
|
|
|
def _set_data(self, solder):
|
|
self.spinbox_block = True
|
|
self.constraint_model.constraint_list = solder.rates
|
|
|
|
ix = self.constraint_model.index(0, 0)
|
|
self._constraint_selected(ix)
|
|
self.constraint_view.setCurrentIndex(ix)
|
|
|
|
|
|
def _constraint_selected(self, index):
|
|
if index.isValid():
|
|
self.spinbox_block = True
|
|
temp_levels, value = self.constraint_model.constraint_list[index.row()]
|
|
self.selected_temp_levels.setTempLevels(temp_levels)
|
|
self.controls.value.setValue(value)
|
|
else:
|
|
self.spinbox_block = True
|
|
self.selected_temp_levels.setTempLevels([])
|
|
self.controls.value.setValue(0)
|
|
|
|
class SolderWidget(QtGui.QWidget):
|
|
def __init__(self, parent, readonly=False):
|
|
super(SolderWidget, self).__init__(parent)
|
|
self.solder_model = SolderListModel(self)
|
|
self.solder_view = QtGui.QListView()
|
|
self.solder_view.setModel(self.solder_model)
|
|
self.solder_controls = AddRemoveWidget(self, False)
|
|
|
|
layout = QtGui.QHBoxLayout(self)
|
|
layout.addWidget(self.solder_view, 3)
|
|
layout.addWidget(self.solder_controls, 1)
|
|
|
|
|
|
self.connect(
|
|
self.solder_controls.add_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.solder_model.create_solder)
|
|
|
|
self.solder_view.setCurrentIndex(self.solder_model.index(0,0))
|
|
|
|
|
|
class TempLevelWidget(QtGui.QWidget):
|
|
temp_level_removed = QtCore.pyqtSignal(TempLevel)
|
|
solder_changed = QtCore.pyqtSignal()
|
|
|
|
def __init__(self, parent, readonly=False):
|
|
super(TempLevelWidget, self).__init__(parent)
|
|
self.temp_level_model = TempLevelModel(self)
|
|
|
|
self.temp_level_view = QtGui.QTableView()
|
|
self.temp_level_view.setModel(self.temp_level_model)
|
|
self.temp_level_view.verticalHeader().setVisible(False)
|
|
self.temp_level_view.resizeColumnsToContents()
|
|
self.temp_level_view.horizontalHeader().setStretchLastSection(True)
|
|
self.temp_level_view.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
|
|
|
|
h = QtGui.QHBoxLayout()
|
|
h.addWidget(self.temp_level_view)
|
|
self.setLayout(h)
|
|
|
|
self.connect(
|
|
self.temp_level_view,
|
|
QtCore.SIGNAL("clicked(QModelIndex)"),
|
|
self.temp_level_selected)
|
|
|
|
if not readonly:
|
|
self.controls = AddRemoveWidget(self)
|
|
h.addWidget(self.controls)
|
|
|
|
self.connect(
|
|
self.controls.add_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.add_temp_level)
|
|
|
|
self.connect(
|
|
self.controls.remove_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.remove_temp_level)
|
|
|
|
self.connect(
|
|
self.controls.up_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.temp_level_up)
|
|
|
|
self.connect(
|
|
self.controls.down_button,
|
|
QtCore.SIGNAL("clicked()"),
|
|
self.temp_level_down)
|
|
|
|
self.connect(
|
|
self.temp_level_model,
|
|
QtCore.SIGNAL("solder_changed()"),
|
|
self._solder_changed)
|
|
|
|
def _solder_changed(self):
|
|
self.solder_changed.emit()
|
|
|
|
|
|
def setData(self, solder):
|
|
self.temp_level_model.setTempLevels(solder.temp_levels)
|
|
self.temp_level_view.resizeColumnsToContents()
|
|
|
|
|
|
def add_temp_level(self):
|
|
index = self.temp_level_view.currentIndex()
|
|
self.temp_level_model.add_temp_level(index, TempLevel("new " + str(self.temp_level_model.rowCount(None)), 0))
|
|
self.temp_level_view.setCurrentIndex(self.temp_level_model.index(index.row() + 1, 0))
|
|
self.plotter.solder.changed = True
|
|
self.solder_changed.emit()
|
|
|
|
|
|
def remove_temp_level(self):
|
|
self.temp_level_removed.emit(
|
|
self.temp_level_model.remove_temp_level(
|
|
self.temp_level_view.currentIndex().row()))
|
|
self.solder_changed.emit()
|
|
|
|
def temp_level_up(self):
|
|
dst_row = self.temp_level_view.currentIndex().row()
|
|
self.temp_level_model.temp_levels[dst_row - 1], self.temp_level_model.temp_levels[dst_row] = self.temp_level_model.temp_levels[dst_row], self.temp_level_model.temp_levels[dst_row - 1]
|
|
self.temp_level_model.reset()
|
|
self.solder_changed.emit()
|
|
|
|
def temp_level_down(self):
|
|
dst_row = self.selected_temp_levels_view.currentIndex().row()
|
|
self.temp_level_model.temp_levels[dst_row], self.temp_level_model.temp_levels[dst_row + 1] = self.temp_level_model.temp_levels[dst_row + 1], self.temp_level_model.temp_levels[dst_row]
|
|
self.temp_level_model.reset()
|
|
self.solder_changed.emit()
|
|
|
|
|
|
def temp_level_selected(self, index):
|
|
if index.isValid():
|
|
row = index.row()
|
|
if not self.readonly:
|
|
is_env = self.temp_level_model.temp_levels[row].is_env
|
|
#is_end = row == len(self.temp_level_model.temp_levels) - 1
|
|
#self.controls.add_button.setEnabled(not is_end)
|
|
self.controls.remove_button.setEnabled(not is_env)
|
|
|
|
class ApplicationWindow(QtGui.QMainWindow):
|
|
def __init__(self):
|
|
QtGui.QMainWindow.__init__(self)
|
|
self.dpi = 100
|
|
|
|
self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
|
|
self.setWindowTitle("application main window")
|
|
|
|
self.file_menu = QtGui.QMenu('&File', self)
|
|
self.file_menu.addAction('&Create profile header', self.create_header,
|
|
QtCore.Qt.CTRL + QtCore.Qt.Key_C)
|
|
self.file_menu.addAction('&Save solder', self.save_solder,
|
|
QtCore.Qt.CTRL + QtCore.Qt.Key_S)
|
|
self.file_menu.addAction('S&ave plot', self.save_plot,
|
|
QtCore.Qt.CTRL + QtCore.Qt.Key_A)
|
|
self.file_menu.addAction('&Quit', self.fileQuit,
|
|
QtCore.Qt.CTRL + QtCore.Qt.Key_Q)
|
|
|
|
self.menuBar().addMenu(self.file_menu)
|
|
|
|
self.view_menu = QtGui.QMenu('&View', self)
|
|
self.profile_view_action = self.view_menu.addAction("&Profile View", self.open_profile_view,
|
|
QtCore.Qt.CTRL + QtCore.Qt.Key_P)
|
|
self.controls_view_action = self.view_menu.addAction("&Oven Controls View", self.open_controls_view,
|
|
QtCore.Qt.CTRL + QtCore.Qt.Key_O)
|
|
|
|
|
|
self.view_group = QtGui.QActionGroup(self)
|
|
self.view_group.setExclusive(True)
|
|
|
|
self.view_group.addAction(self.controls_view_action)
|
|
self.view_group.addAction(self.profile_view_action)
|
|
|
|
self.profile_view_action.setCheckable(True)
|
|
self.controls_view_action.setCheckable(True)
|
|
|
|
#self.connect(self.profile_view_action,
|
|
#QtCore.SIGNAL("triggered()"),
|
|
#self.open_profile_view)
|
|
|
|
#self.connect(self.controls_view_action,
|
|
#QtCore.SIGNAL("triggered()"),
|
|
#self.open_controls_view)
|
|
|
|
self.profile_view_action.setChecked(True)
|
|
|
|
self.menuBar().addMenu(self.view_menu)
|
|
|
|
self.help_menu = QtGui.QMenu('&Help', self)
|
|
self.menuBar().addSeparator()
|
|
self.menuBar().addMenu(self.help_menu)
|
|
|
|
self.help_menu.addAction('&About', self.about)
|
|
|
|
|
|
|
|
self.tab_widget = QtGui.QTabWidget(self)
|
|
self.temp_level_widget = TempLevelWidget(self)
|
|
self.duration_widget = DurationConstraintWidget(u"Duration (s)")
|
|
self.rate_widget = RateConstraintWidget(u"Rate (°C/s)")
|
|
self.tab_widget.addTab(self.temp_level_widget, u"Temperature Levels")
|
|
self.tab_widget.addTab(self.duration_widget, u"Duration (s)")
|
|
self.tab_widget.addTab(self.rate_widget, u"Rate (°C/s)")
|
|
|
|
self.plotter = Plotter(self, self, width=5, height=4, dpi=self.dpi)
|
|
self.controls_widget = OvenControlsWidget(self)
|
|
self.controls_widget.setVisible(False)
|
|
|
|
|
|
|
|
self.connect(
|
|
self.duration_widget,
|
|
QtCore.SIGNAL("solder_changed()"),
|
|
self.plotter.solder_changed)
|
|
|
|
self.connect(
|
|
self.rate_widget,
|
|
QtCore.SIGNAL("solder_changed()"),
|
|
self.plotter.solder_changed)
|
|
|
|
self.connect(
|
|
self.temp_level_widget,
|
|
QtCore.SIGNAL("solder_changed()"),
|
|
self.plotter.solder_changed)
|
|
|
|
self.solder_widget = SolderWidget(self)
|
|
|
|
self.connect(
|
|
self.solder_widget.solder_view,
|
|
QtCore.SIGNAL("clicked(QModelIndex)"),
|
|
self.solder_selected)
|
|
|
|
|
|
self.settings_widget = QtGui.QWidget(self)
|
|
pl = QtGui.QHBoxLayout(self.settings_widget)
|
|
pl.addWidget(self.solder_widget, 1)
|
|
pl.addWidget(self.tab_widget, 3)
|
|
|
|
self.splitter = QtGui.QSplitter(QtCore.Qt.Vertical, self)
|
|
|
|
self.solder_selected(self.solder_widget.solder_model.index(0,0))
|
|
|
|
self.splitter.addWidget(self.settings_widget)
|
|
self.splitter.addWidget(self.controls_widget)
|
|
self.splitter.addWidget(self.plotter)
|
|
self.splitter.setStretchFactor(0, 2)
|
|
self.splitter.setStretchFactor(1, 2)
|
|
self.splitter.setStretchFactor(2, 8)
|
|
|
|
self.setCentralWidget(self.splitter)
|
|
|
|
self.statusBar().showMessage("I'm in reflow heaven", 2000)
|
|
|
|
self.temp_level_widget.temp_level_removed.connect(self.duration_widget.slot_temp_level_removed)
|
|
self.temp_level_widget.temp_level_removed.connect(self.rate_widget.slot_temp_level_removed)
|
|
|
|
def getPlotter(self):
|
|
return self.plotter
|
|
|
|
def create_header(self):
|
|
x_list = list()
|
|
y_list = list()
|
|
for solder in self.solder_model.listdata:
|
|
x, y, xmax, ymax, duration_points, rate_points = solder.calc_profile()
|
|
x_list.append(x)
|
|
y_list.append(y)
|
|
print "packet", repr(create_profile_packet(x, y))
|
|
|
|
create_profile_header(x_list, y_list)
|
|
|
|
def solder_selected(self, index):
|
|
if index.isValid():
|
|
solder = self.solder_widget.solder_model.listdata[index.row()]
|
|
self.temp_level_widget.setData(solder)
|
|
self.duration_widget.setData(solder)
|
|
self.rate_widget.setData(solder)
|
|
self.plotter.setData(solder)
|
|
self.controls_widget.temp_level_widget.setData(solder)
|
|
|
|
def open_controls_view(self):
|
|
print self.open_controls_view
|
|
self.controls_widget.show()
|
|
self.settings_widget.hide()
|
|
|
|
def open_profile_view(self):
|
|
print self.open_profile_view
|
|
self.controls_widget.hide()
|
|
self.settings_widget.show()
|
|
|
|
def save_plot(self):
|
|
file_choices = "PNG (*.png)|*.png"
|
|
|
|
filename = QtGui.QFileDialog.getSaveFileName(self, 'Save File', 'qtplot.png')
|
|
self.plotter.print_figure(str(filename), dpi=self.dpi)
|
|
|
|
def save_solder(self):
|
|
self.plotter.solder.save("foo")
|
|
self.solder_widget.solder_model.reload()
|
|
|
|
def fileQuit(self):
|
|
self.close()
|
|
|
|
def closeEvent(self, ce):
|
|
self.fileQuit()
|
|
|
|
def about(self):
|
|
QtGui.QMessageBox.about(self, "About %s" % progname,
|
|
u"%(prog)s version %(version)s\n" \
|
|
u"Copyright \N{COPYRIGHT SIGN} 2012 Stefan Kögl\n\n" \
|
|
u"reflowctl frontend" % {"prog": progname, "version": progversion})
|
|
|
|
|
|
def main():
|
|
|
|
qApp = QtGui.QApplication(sys.argv)
|
|
|
|
aw = ApplicationWindow()
|
|
aw.setWindowTitle("%s" % progname)
|
|
aw.show()
|
|
sys.exit(qApp.exec_())
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|