#!/usr/bin/env python # -*- encoding: utf-8 -*- import os import sys import re import signal from PyQt4.QtCore import * from PyQt4.QtGui import * from mathlib import fit_anneal, fit_lbfgsb, fit_odr, hn, id_to_color from matplotlibWidget import PlotWidget #matplotlib.rc_file("default.mplrc") import numpy as N import QDSMain import PeakWidget from data import Data, Conductivity, conductivity #import yaff def sigint_handler(*args): """Handler for the SIGINT signal (CTRL + C). """ sys.stderr.write('\r') if QMessageBox.question(None, '', "Are you sure you want to quit?", QMessageBox.Yes | QMessageBox.No, QMessageBox.No) == QMessageBox.Yes: QApplication.quit() def tau_peak(f, a, b): tau = (N.sin(N.pi * a / 2. / (b + 1)) / N.sin(N.pi * a * b / 2. / (b + 1))) ** (1 / a) tau /= 2 * N.pi * f return tau class Peak(QObject): changedData = pyqtSignal() def __init__(self, id=None, mpl=None): QObject.__init__(self) super(Peak, self).__init__() self.color = id_to_color(id) self.widget = PeakWidget.PeakWidget() self.widget.setId(id) self.widget.setColor(map(int, [255 * i for i in self.color])) self.widget.changedTable.connect(self.updatePeak) self.mpl = mpl self.mpl_line = None def getParameter(self): p = self.widget.peakParameter() return p def getFixed(self): p = self.widget.fixedParameter() return p def setParameter(self, delta_eps=None, tau=None, a=None, b=None): self.widget.updateTable(delta_eps, tau, a, b) self.updatePeak() def updatePeak(self): # get current axis limits x_min, x_max = self.mpl.canvas.axes.get_xlim() y_min, y_max = self.mpl.canvas.axes.get_ylim() nu = N.logspace(N.log10(x_min), N.log10(x_max), 2048) y = hn(self.getParameter(), nu) # clip data to axes limits mask = (y < y_max) & (y > y_min) y = y[mask] nu = nu[mask] if self.mpl_line == None: self.mpl_line, = self.mpl.canvas.axes.loglog(nu, y, '--', label="Peak %i" % (self.widget.id), animated=True) # peak self.mpl_line.set_color(self.color) else: self.mpl_line.set_xdata(nu) self.mpl_line.set_ydata(y) self.changedData.emit() class AppWindow(QMainWindow): def __init__(self, parent=None): super(AppWindow, self).__init__(parent) self.ui = QDSMain.Ui_MainWindow() self.ui.setupUi(self) self.picked_artist = None self.data = None self.Conductivity = None self._lines = dict() self.peakId = 0 self.peakBoxes = {} ## menubar fileMenu = self.menuBar().addMenu("File") openFile = QAction("&Open", self) openFile.setShortcut(QKeySequence.Open) openFile.triggered.connect(self.openFile) fileMenu.addAction(openFile) # fitting methods fitMenu = self.menuBar().addMenu("Standard Fits") # lm fit_lmAction = QAction("&Levenberg-Marquardt", self) fit_lmAction.setShortcut(QKeySequence("Ctrl+F")) fitMenu.addAction(fit_lmAction) # lbfgsb fit_lbfgsbAction = QAction("&L-BFGS-B", self) fitMenu.addAction(fit_lbfgsbAction) # Simulated Annealing fit_annealAction = QAction("&Simulated Annealing", self) fitMenu.addAction(fit_annealAction) # YAFF yaffMenu = self.menuBar().addMenu("YAFF") start_yaff = QAction("&Startparam.", self) yaffMenu.addAction(start_yaff) fit_yaff = QAction("&Fit", self) yaffMenu.addAction(fit_yaff) self.signalMapper = QSignalMapper(self) for i, fit_action in enumerate([fit_lmAction, fit_lbfgsbAction, fit_annealAction ]): self.signalMapper.setMapping(fit_action, i) fit_action.triggered.connect(self.signalMapper.map) self.signalMapper.mapped.connect(self.fitData) # save fitted values self.ui.actionSave_FitResult.triggered.connect(self.saveFitResult) # the plot area, a matplotlib widget self.mplWidget = PlotWidget(self.ui.mplWidget) self.mplWidget.canvas.draw() self.mplWidget.updateGeometry() # what to do with CIDs? self.cid = [] self.cid.append(self.mplWidget.canvas.mpl_connect("button_press_event", self.mpl_button_press)) self.cid.append(self.mplWidget.canvas.mpl_connect("pick_event", self.mpl_button_pick)) self.cid.append(self.mplWidget.canvas.mpl_connect("button_release_event", self.mpl_button_release)) #self.cid.append(self.mplWidget.canvas.mpl_connect("resize_event", self.myresizeEvent)) self.mplWidget.toolbar.spanSelectedTrigger.connect(self.set_fit_xlimits) def resizeEvent(self, evt): """resizing the main window executes this method TODO: FIX RESIZEING keeping mainwindow maximised for now :param evt: """ self.mplWidget.canvas.draw() self.mplWidget._bg_cache = self.mplWidget.canvas.copy_from_bbox(self.mplWidget.canvas.axes.bbox) for line in self.mplWidget.canvas.axes.get_lines(): line.set_animated(False) self.mplWidget.canvas.draw() for line in self.mplWidget.canvas.axes.get_lines(): line.set_animated(True) def saveFitResult(self): """ Saving fit parameters to fitresults.log including temperature """ if not os.path.exists("fitresults.log"): f = open("fitresults.log", "w") else: f = open("fitresults.log", "a") # write header f.write('# T ') parfmt = "%.2f" # T formatting # if self.Conductivity != None: pass# always true f.write("%8s %8s %8s " % ("e_s", "sig", "pow_sig")) parfmt += " %.3g %.3g %.2f " # conductivity formatting for i, pb in enumerate(self.peakBoxes): enum_peak = ("e_inf_%i" % i, "tau_%i" % i, "alpha_%i" % i, "beta_%i" % i) f.write("%8s %8s %8s %8s " % enum_peak) parfmt += " %.3g %.3g %.2f %.2f" # peak formatting f.write("high_lim lower_lim") # TODO: store limits f.write('\n') f.flush() #f.write("%3.2f "%(self.data.meta["T"])) pars = list(self.fitresult) pars.insert(0, self.data.meta["T"]) N.savetxt(f, N.array([pars, ]), fmt=parfmt, delimiter=" ") f.close() def set_fit_xlimits(self, xmin, xmax): self.data.fit_limits = (xmin, xmax, None, None) self.updatePlot() def mpl_button_release(self, event): ax = self.mplWidget.canvas.axes if self.picked_artist: if not event.inaxes: # moved outside the plot, add back to original position ax.add_artist(self.picked_artist) else: # we move one of the three points determinig the peak self.picked_artist.set_xdata(event.xdata) self.picked_artist.set_ydata(event.ydata) ax.add_artist(self.picked_artist) for peak in self.peakBoxes.keys(): peak.updatePeak() self.picked_artist = None self.mplWidget.canvas.draw_idle() def mpl_button_pick(self, event): self.picked_artist = event.artist event.artist.remove() self.mplWidget.canvas.draw_idle() def addCond(self, pos): if self.Conductivity != None: return self.statusBar().showMessage("Click on graph") self.Conductivity = Conductivity(mpl=self.mplWidget) self.Conductivity.changedData.connect(self.updatePlot) self.Conductivity.setParameter(0, 1 / (pos[0] / pos[1] / 2 / N.pi), 1.0) self.ui.scrollAreaWidgetContents.layout().addWidget(self.Conductivity.widget) self.Conductivity.widget.ui.removeButton.clicked.connect(self.delCond) def delCond(self): self.cond_param = None self.cond = None self.Conductivity.mpl_line.remove() self.Conductivity.mpl_line_static.remove() del self.Conductivity self.Conductivity = None self.updatePlot() def addPeak(self, pos): self.peakId += 1 self.statusBar().showMessage("Select Peak Position") peak = Peak(id=self.peakId, mpl=self.mplWidget) # connect to delPeak peak.widget.ui.removeButton.clicked.connect(self.delPeak) peak.setParameter(delta_eps=pos[1], tau=1 / (2. * N.pi * pos[0]), a=1, b=1) peak.changedData.connect(self.updatePlot) self.peakBoxes[peak] = None for pb in self.peakBoxes.keys(): self.ui.scrollAreaWidgetContents.layout().addWidget(pb.widget) self.updatePlot() def delPeak(self): deletePeaks = [] for i in xrange(self.ui.scrollAreaWidgetContents.layout().count()): print i for i, peak in enumerate(self.peakBoxes.keys()): if peak.widget.isHidden(): peak.mpl_line.remove() deletePeaks.append(peak) for peak in deletePeaks: self.peakBoxes.pop(peak) self.updatePlot() def fitData(self, method): if self.Conductivity != None: start_parameter = list(self.Conductivity.getParameter()) fixed_params = [i for i in self.Conductivity.getFixed()] else: start_parameter = [0, 0, 1] fixed_params = [0, 0, 0] for pb in self.peakBoxes.keys(): [start_parameter.append(i) for i in pb.getParameter()] [fixed_params.append(i) for i in pb.getFixed()] fit_methods = [fit_odr, fit_lbfgsb, fit_anneal] print "StartParameter", start_parameter print "FixedParameter", fixed_params print "Limits (xmin, xmax, ymin, ymax)", self.data.fit_limits _freq, _fit = self.data.get_data() result = fit_methods[method](_freq, _fit.imag, start_parameter, fixed_params) self.fitresult = result for i, pb in enumerate(self.peakBoxes.keys()): delta_eps, tau, a, b = result[3 + i * 4:3 + (i + 1) * 4] pb.setParameter(delta_eps, tau, a, b) e_static, sigma, sigma_N = result[:3] if self.Conductivity != None: self.Conductivity.setParameter(e_static, sigma, sigma_N) #print "*** FIT RESULTS ***" #print u"\u03c3" #print u"\u0394\u03b5" self.updatePlot() def mpl_button_press(self, event): """ Handles the clicks on the matplotlib figure canvas """ if self.ui.actionAdd_Peak.isChecked() and event.inaxes: x, y = event.xdata, event.ydata self.addPeak((x, y)) self.ui.actionAdd_Peak.setChecked(False) self.statusBar().clear() if self.ui.actionAdd_Cond.isChecked() and event.inaxes: x, y = event.xdata, event.ydata self.addCond((x, y)) self.ui.actionAdd_Cond.setChecked(False) self.statusBar().clear() def openFile(self): ax = self.mplWidget.canvas.axes ax.clear() path = unicode(QFileDialog.getOpenFileName(self, "Open file")) # TODO anaylize file (LF,MF, HF) and act accordingly data = N.loadtxt(path, skiprows=4) numpat = re.compile('\d+\.\d+') try: Temp = None for line in open(path).readlines(): if re.search("Fixed", line) or re.search("Temp", line): print "Found line with Fixed or Temp" Temp = float(re.search(numpat, line).group()) print "Temperature found in file:", Temp break if Temp == None: raise ValueError except: Temp = QInputDialog.getDouble(self, "No temperature found in data set", "Temperature/K:", value=0.00)[0] # mask the data to values > 0 (loglog plot) mask = (data[:, 1] > 0) & (data[:, 2] > 0) #& (data[:,2]>1e-3) & (data[:,0] > 1e-2) _freq = data[mask, 0] _die_stor = data[mask, 1] _die_loss = data[mask, 2] # clear the figure ax.clear() #if self.data != None: # self.data.remove_curves() self.data = Data(_freq, _die_stor, _die_loss) self.data.meta["T"] = Temp self.data.data_curve, = ax.loglog(self.data.frequency, self.data.epsilon.imag, 'b.', markersize=4, label="Data", animated=True) ax.set_xlabel("Frequency/Hz", fontsize=16) #ax.set_ylabel(u'\u03B5"', fontsize=16) ax.set_ylabel(u'e"', fontsize=16) ax.autoscale(True) ax.set_xlim(_freq.min()/3, _freq.max()*3) self.legend = ax.legend(title="T=%.2f" % Temp) for line in ax.get_lines(): line.set_animated(False) ax.grid() self.mplWidget.canvas.draw() # weird behaviour need to draw all first, then set_animated the redraw for line in ax.get_lines(): line.set_animated(True) self.legend.set_animated(True) ax.autoscale(False) self.mplWidget._bg_cache = self.mplWidget.canvas.copy_from_bbox(ax.bbox) self.updatePlot() def updatePlot(self): ax = self.mplWidget.canvas.axes nu = self.data.frequency fit = N.zeros(len(nu)) for peak in self.peakBoxes.keys(): params = peak.getParameter() fit += hn(params, nu) if self.Conductivity != None: print "Cond. given" params = self.Conductivity.getParameter()[1:] fit += conductivity(params, nu) fit += self.Conductivity.getParameter()[0] # eps static # clip data to axes limits y_min, y_max = ax.get_ylim() mask = (fit < y_max) & (fit > y_min) #mask = N.ones(len(fit), dtype="bool") if self.data.fitted_curve == None: self.data.fitted_curve, = ax.loglog(nu[mask], fit[mask], 'k-', alpha=0.5, label="Sum", animated=True) else: self.data.fitted_curve.set_xdata(nu[mask]) self.data.fitted_curve.set_ydata(fit[mask]) # handling of fit limit bars # if self.mplWidget._axvlims != []: # [axv.remove() for axv in self.mplWidget._axvlims] # self.mplWidget._axvlims = [] # if self.mplWidget._axvname != []: # [axvname.remove() for axvname in self.mplWidget._axvname] # self.mplWidget._axvname = [] #print self.mplWidget._axvlims if self.mplWidget._axvlims == []: for i,xlim in enumerate(self.data.fit_limits[:2]): self.mplWidget._axvlims.append(ax.axvline(xlim, color="k", ls="--", alpha=0.5, animated=True)) self.mplWidget._axvname.append(ax.text(xlim, y_min*3. , "%.3g"%xlim, horizontalalignment='center', verticalalignment='center', animated=True) ) else: for i,xlim in enumerate(self.data.fit_limits[:2]): self.mplWidget._axvlims[i].set_xdata(xlim) self.mplWidget._axvname[i].set_x(xlim) self.mplWidget._axvname[i].set_text("%.3g"%xlim) self.mplWidget.canvas.restore_region(self.mplWidget._bg_cache) self.legend = ax.legend(title="T=%.2f" % (self.data.meta["T"])) self.legend.set_animated(True) for animated_artist in ax.findobj(match=lambda x: x.get_animated()): #print "updatePlot: animated artist:",animated_artist ax.draw_artist(animated_artist) self.mplWidget.canvas.blit(ax.bbox) if __name__ == '__main__': signal.signal(signal.SIGINT, sigint_handler) app = QApplication(sys.argv) timer = QTimer() timer.start(1000) # Check every second for Strg-c on Cmd line timer.timeout.connect(lambda: None) main = AppWindow() main.showMaximized() main.raise_() sys.exit(app.exec_())