new logarithmic double spin box, real and imag fit data are shown, proper cursor position found

This commit is contained in:
Markus Rosenstihl 2014-03-18 19:48:39 +01:00
parent 8673bdd7ff
commit dfa12bfc3c
11 changed files with 153 additions and 83 deletions

View File

@ -2,7 +2,7 @@
# Form implementation generated from reading ui file 'ConductivityGroupBox.ui' # Form implementation generated from reading ui file 'ConductivityGroupBox.ui'
# #
# Created: Fri Feb 21 17:15:59 2014 # Created: Sat Mar 15 21:26:00 2014
# by: PyQt4 UI code generator 4.10.3 # by: PyQt4 UI code generator 4.10.3
# #
# WARNING! All changes made in this file will be lost! # WARNING! All changes made in this file will be lost!
@ -26,7 +26,7 @@ except AttributeError:
class Ui_ConductivityGroupBox(object): class Ui_ConductivityGroupBox(object):
def setupUi(self, ConductivityGroupBox): def setupUi(self, ConductivityGroupBox):
ConductivityGroupBox.setObjectName(_fromUtf8("ConductivityGroupBox")) ConductivityGroupBox.setObjectName(_fromUtf8("ConductivityGroupBox"))
ConductivityGroupBox.resize(255, 150) ConductivityGroupBox.resize(254, 150)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Fixed) sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Fixed)
sizePolicy.setHorizontalStretch(0) sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0) sizePolicy.setVerticalStretch(0)
@ -196,7 +196,7 @@ class Ui_ConductivityGroupBox(object):
self.pushButton_6.setText(_translate("ConductivityGroupBox", "+", None)) self.pushButton_6.setText(_translate("ConductivityGroupBox", "+", None))
self.lineEdit_2.setText(_translate("ConductivityGroupBox", "0.00e+00", None)) self.lineEdit_2.setText(_translate("ConductivityGroupBox", "0.00e+00", None))
self.lineEdit_1.setText(_translate("ConductivityGroupBox", "0.00e+00", None)) self.lineEdit_1.setText(_translate("ConductivityGroupBox", "0.00e+00", None))
self.label_1.setText(_translate("ConductivityGroupBox", "ε<sub>static</sub>", None)) self.label_1.setText(_translate("ConductivityGroupBox", "<html><head/><body><p>ε<span style=\" vertical-align:sub;\">infty</span></p></body></html>", None))
self.pushButton_3.setText(_translate("ConductivityGroupBox", "-", None)) self.pushButton_3.setText(_translate("ConductivityGroupBox", "-", None))
self.pushButton_4.setText(_translate("ConductivityGroupBox", "+", None)) self.pushButton_4.setText(_translate("ConductivityGroupBox", "+", None))
self.pushButton_1.setText(_translate("ConductivityGroupBox", "-", None)) self.pushButton_1.setText(_translate("ConductivityGroupBox", "-", None))

View File

@ -6,7 +6,7 @@
<rect> <rect>
<x>0</x> <x>0</x>
<y>0</y> <y>0</y>
<width>255</width> <width>254</width>
<height>150</height> <height>150</height>
</rect> </rect>
</property> </property>
@ -219,7 +219,7 @@
</sizepolicy> </sizepolicy>
</property> </property>
<property name="text"> <property name="text">
<string>ε&lt;sub&gt;static&lt;/sub&gt;</string> <string>&lt;html&gt;&lt;head/&gt;&lt;body&gt;&lt;p&gt;ε&lt;span style=&quot; vertical-align:sub;&quot;&gt;infty&lt;/span&gt;&lt;/p&gt;&lt;/body&gt;&lt;/html&gt;</string>
</property> </property>
<property name="alignment"> <property name="alignment">
<set>Qt::AlignCenter</set> <set>Qt::AlignCenter</set>

37
CustomWidgets.py Normal file
View File

@ -0,0 +1,37 @@
__author__ = 'markusro'
from PyQt4.QtGui import *
from PyQt4.QtCore import QRegExp
import PeakWidget
class LogFSpinBox(QDoubleSpinBox):
scientificNotationValidator = QRegExpValidator(QRegExp("[+-]?(?:0|[1-9]\\d*)(?:\\.\\d*)?(?:[eE][+-]?\\d+)?"))
def __init__(self, parent = None):
super(LogFSpinBox, self).__init__(parent)
self.setRange(1e-16,1e16)
self.setMinimum(1e-16)
self.setDecimals(17)
self.setValue(1.0)
def stepBy(self, up_down):
print self.minimum(),self.maximum()
if self.value() != 0.0:
self.setValue(self.value()*10**(up_down/9.0)) # 19 steps per decade
def textFromValue(self, value):
return "%.3e"%value
def valueFromText(self, str_value):
return str_value.toDouble()[0]
def validate(self, str_value, p_int):
return self.scientificNotationValidator.validate(str_value, p_int)
class ParameterWidget(QWidget):
def __init__(self, parent = None):
super(ParameterWidget, self).__init__(parent)
layout = QGridLayout(self)
if __name__ == "__main__":
app = QApplication([])

View File

@ -2,7 +2,7 @@
# Form implementation generated from reading ui file 'PeakGroupBox.ui' # Form implementation generated from reading ui file 'PeakGroupBox.ui'
# #
# Created: Fri Feb 21 17:15:59 2014 # Created: Sat Mar 15 21:26:00 2014
# by: PyQt4 UI code generator 4.10.3 # by: PyQt4 UI code generator 4.10.3
# #
# WARNING! All changes made in this file will be lost! # WARNING! All changes made in this file will be lost!

View File

@ -74,14 +74,14 @@ class PeakWidget(QGroupBox):
def peakParameter(self): def peakParameter(self):
tmp = [i.text().toDouble()[0] # selects the number, ignores the status tmp = [i.text().toDouble()[0] # selects the number, ignores the status
for i in self.lineEdits] for i in self.lineEdits]
print "peakParams:", tmp #print "peakParams:", tmp
return tmp return tmp
def update(self): def update(self):
self.changedTable.emit() self.changedTable.emit()
def updateTable(self, *args): def updateTable(self, *args):
print "updateTable", args #print "updateTable", args
for i, arg in enumerate(args): for i, arg in enumerate(args):
self.lineEdits[i].setText("%g" % (args[i])) self.lineEdits[i].setText("%g" % (args[i]))

88
QDS.py
View File

@ -28,6 +28,7 @@ import pyqtgraph as pg
#import yaff #import yaff
USE_CROSSH=False
class AppWindow(QMainWindow): class AppWindow(QMainWindow):
def __init__(self, parent=None): def __init__(self, parent=None):
@ -94,20 +95,48 @@ class AppWindow(QMainWindow):
self.ui.graphicsView.addItem(self.data.data_curve_real) self.ui.graphicsView.addItem(self.data.data_curve_real)
self.ui.graphicsView.addItem(self.data.fitted_curve_imag) self.ui.graphicsView.addItem(self.data.fitted_curve_imag)
self.ui.graphicsView.addItem(self.data.fitted_curve_real) self.ui.graphicsView.addItem(self.data.fitted_curve_real)
# cross hair
if USE_CROSSH:
#self.poslabel = pg.LabelItem(justify='right')
self.horizontalLine = pg.InfiniteLine(angle=0)
self.verticalLine = pg.InfiniteLine(angle=90)
#self.ui.graphicsView.plotItem.addItem(self.poslabel)
self.ui.graphicsView.addItem(self.horizontalLine)
self.ui.graphicsView.addItem(self.verticalLine)
self.ui.graphicsView.addItem(self.fit_boundary) self.ui.graphicsView.addItem(self.fit_boundary)
self.ui.graphicsView.setLogMode(x=True, y=True) self.ui.graphicsView.setLogMode(x=True, y=True)
self.ui.graphicsView.showGrid(x=True, y=True) self.ui.graphicsView.showGrid(x=True, y=True)
self.ui.graphicsView.setLabel("bottom","Frequency",units="Hz") self.ui.graphicsView.setLabel("bottom","Frequency",units="Hz")
self.ui.graphicsView.setLabel("left", "Dielectric loss", units="Debye") self.ui.graphicsView.setLabel("left", "Dielectric loss", units="Debye")
self.ui.graphicsView.disableAutoRange()
self.peak_box_layout = QGridLayout() self.peak_box_layout = QGridLayout()
self.ui.scrollAreaWidgetContents.setLayout(self.peak_box_layout) self.ui.scrollAreaWidgetContents.setLayout(self.peak_box_layout)
##self.huh = pg.SignalProxy(, slot=self.mPE)
sc = self.ui.graphicsView.scene()
sc.sigMouseClicked.connect(self.mousePress)
sc.sigMouseMoved.connect(self.updateCrosshair)
def mousePressEvent(self, evt): def updateCrosshair(self,evt):
pos = evt.pos() vb = self.ui.graphicsView.getPlotItem().vb
if self.ui.graphicsView.sceneRect().contains(pos): pos = vb.mapSceneToView(evt)
# translate position to data coordinates self.last_pos = pos
data_pos = self.ui.graphicsView.plotItem.vb.mapSceneToView(pos) if USE_CROSSH:
self.horizontalLine.setBounds([self.data.frequency.min(), self.data.frequency.max()])
self.horizontalLine.setPos(pos.y())
self.verticalLine.setPos(pos.x())
#self.poslabel.setText("<span style='font-size: 12pt'>f=%0.1f Hz <span style='color: red'>e=%0.1f</span>"\
# % (pos.x(),pos.y()))
def mousePress(self, evt):
vb = self.ui.graphicsView.getPlotItem().vb
data_pos = self.last_pos
if self.ui.actionAdd_Peak.isChecked() and not self.ui.actionAdd_Cond.isChecked(): if self.ui.actionAdd_Peak.isChecked() and not self.ui.actionAdd_Cond.isChecked():
self.addPeak(data_pos) self.addPeak(data_pos)
self.ui.actionAdd_Peak.setChecked(False) self.ui.actionAdd_Peak.setChecked(False)
@ -186,7 +215,8 @@ class AppWindow(QMainWindow):
self.statusBar().showMessage("Click on graph") self.statusBar().showMessage("Click on graph")
self.Conductivity = Conductivity(mpl=self.ui.graphicsView, limits=self.data.fit_limits) self.Conductivity = Conductivity(mpl=self.ui.graphicsView, limits=self.data.fit_limits)
self.Conductivity.changedData.connect(self.updatePlot) self.Conductivity.changedData.connect(self.updatePlot)
self.Conductivity.setParameter(0, 1 / (10**pos.x() / 10**pos.y() / 2 / N.pi), 1.0) print pos.x(), pos.y(),"test", 10**pos.x(),10**pos.y()
self.Conductivity.setParameter(0, 10**(pos.y() + pos.x())*2*N.pi , 1.0)
self.ui.scrollAreaWidgetContents.layout().addWidget(self.Conductivity.widget) self.ui.scrollAreaWidgetContents.layout().addWidget(self.Conductivity.widget)
self.Conductivity.widget.ui.removeButton.clicked.connect(self.delCond) self.Conductivity.widget.ui.removeButton.clicked.connect(self.delCond)
@ -208,7 +238,8 @@ class AppWindow(QMainWindow):
# connect to delPeak # connect to delPeak
peak.widget.ui.removeButton.clicked.connect(self.delPeak) peak.widget.ui.removeButton.clicked.connect(self.delPeak)
peak.changedData.connect(self.updatePlot) peak.changedData.connect(self.updatePlot)
peak.setParameter(delta_eps=10**pos.y(), tau=1 / (10**pos.x()/2/N.pi), a=1, b=1) print pos, 10**pos.x(), 10**pos.y()
peak.setParameter(delta_eps=(10**pos.y())*2, tau=1 / (2*N.pi*10**pos.x()), a=1, b=1)
self.peakBoxes[peak] = None self.peakBoxes[peak] = None
wdgt_num = self.peak_box_layout.rowCount() wdgt_num = self.peak_box_layout.rowCount()
@ -261,7 +292,7 @@ class AppWindow(QMainWindow):
print "Set fit data" print "Set fit data"
self.data.set_fit(newres.beta, funcs) self.data.set_fit(newres.beta, funcs)
print newres.beta,newres.sd_beta print newres.beta,newres.sd_beta
result = newres.beta
self.fitresult = result self.fitresult = result
for i, pb in enumerate(self.peakBoxes.keys()): for i, pb in enumerate(self.peakBoxes.keys()):
@ -276,10 +307,10 @@ class AppWindow(QMainWindow):
self.updatePlot() self.updatePlot()
def openFile(self): def openFile(self):
path = unicode(QFileDialog.getOpenFileName(self, "Open file")) #path = unicode(QFileDialog.getOpenFileName(self, "Open file"))
path = "MCM42PG0_199.96K.dat"
self.filepath=path self.filepath=path
#path = "MCM42PG0_199.96K.dat" # TODO analyize file (LF,MF, HF) and act accordingly
# TODO anaylize file (LF,MF, HF) and act accordingly
data = N.loadtxt(path, skiprows=4) data = N.loadtxt(path, skiprows=4)
self.setWindowTitle(os.path.basename(path)) self.setWindowTitle(os.path.basename(path))
numpat = re.compile('\d+\.\d+') numpat = re.compile('\d+\.\d+')
@ -302,39 +333,30 @@ class AppWindow(QMainWindow):
self.data.set_data(_freq, _die_stor, _die_loss) self.data.set_data(_freq, _die_stor, _die_loss)
self.data.meta["T"] = Temp self.data.meta["T"] = Temp
self.fit_boundary.setRegion([N.log10(_freq.min()), N.log10(_freq.max())]) self.fit_boundary.setRegion([N.log10(_freq.min()), N.log10(_freq.max())])
self.ui.graphicsView.enableAutoRange()
self.updatePlot() self.updatePlot()
self.ui.graphicsView.disableAutoRange()
def updatePlot(self): def updatePlot(self):
nu = self.data.frequency
fit = N.zeros(len(nu))
if self.Conductivity != None:
print "Cond. given"
params = self.Conductivity.getParameter()[1:]
fit += conductivity(params, nu)
fit += self.Conductivity.getParameter()[0] # eps static
funcs = ["static", "power"]
p0 = self.Conductivity.getParameter() if self.Conductivity != None else [0.0, 0.0, 1.0]
# assemble fit function
funcs = ["static", "power"]
p0 = self.Conductivity.getParameter() if self.Conductivity is not None else [0.0, 0.0, 1.0]
for peak in self.peakBoxes.keys(): for peak in self.peakBoxes.keys():
params = peak.getParameter() params = peak.getParameter()
fit += hn(params, nu)
#fit += peak.get_data()[1]
p0.extend(params) p0.extend(params)
funcs.append("hn") funcs.append("hn")
#self.data.epsilon_fit = fit[:] # calculate fit
print "p0",p0
self.data.set_fit(p0, funcs) self.data.set_fit(p0, funcs)
fit = self.data.epsilon_fit[:]
self.data.data_curve_imag.setData(self.data.frequency, self.data.epsilon.imag)
self.data.data_curve_real.setData(self.data.frequency, self.data.epsilon.real)
if len(self.peakBoxes) > 0 or self.Conductivity != None:
self.data.fitted_curve_imag.setData(nu, fit.imag)
self.data.fitted_curve_real.setData(nu, fit.real)
# replot data and fit, TODO: replot only if data changed
self.data.data_curve_real.setData(self.data.frequency, self.data.epsilon.real)
self.data.data_curve_imag.setData(self.data.frequency, self.data.epsilon.imag)
if len(self.peakBoxes) > 0 or self.Conductivity != None:
self.data.fitted_curve_real.setData(self.data.frequency, self.data.epsilon_fit.real)
self.data.fitted_curve_imag.setData(self.data.frequency, self.data.epsilon_fit.imag)
def sigint_handler(*args): def sigint_handler(*args):
"""Handler for the SIGINT signal (CTRL + C). """Handler for the SIGINT signal (CTRL + C).
@ -342,7 +364,7 @@ def sigint_handler(*args):
sys.stderr.write('\r') sys.stderr.write('\r')
if QMessageBox.question(None, '', "Are you sure you want to quit?", if QMessageBox.question(None, '', "Are you sure you want to quit?",
QMessageBox.Yes | QMessageBox.No, QMessageBox.Yes | QMessageBox.No,
QMessageBox.No) == QMessageBox.Yes: QMessageBox.Yes) == QMessageBox.Yes:
QApplication.quit() QApplication.quit()
if __name__ == '__main__': if __name__ == '__main__':

View File

@ -2,7 +2,7 @@
# Form implementation generated from reading ui file 'QDSMain.ui' # Form implementation generated from reading ui file 'QDSMain.ui'
# #
# Created: Fri Feb 21 17:15:59 2014 # Created: Sat Mar 15 21:26:00 2014
# by: PyQt4 UI code generator 4.10.3 # by: PyQt4 UI code generator 4.10.3
# #
# WARNING! All changes made in this file will be lost! # WARNING! All changes made in this file will be lost!
@ -39,8 +39,8 @@ class Ui_MainWindow(object):
sizePolicy.setHeightForWidth(self.centralwidget.sizePolicy().hasHeightForWidth()) sizePolicy.setHeightForWidth(self.centralwidget.sizePolicy().hasHeightForWidth())
self.centralwidget.setSizePolicy(sizePolicy) self.centralwidget.setSizePolicy(sizePolicy)
self.centralwidget.setObjectName(_fromUtf8("centralwidget")) self.centralwidget.setObjectName(_fromUtf8("centralwidget"))
self.verticalLayout = QtGui.QVBoxLayout(self.centralwidget) self.horizontalLayout = QtGui.QHBoxLayout(self.centralwidget)
self.verticalLayout.setObjectName(_fromUtf8("verticalLayout")) self.horizontalLayout.setObjectName(_fromUtf8("horizontalLayout"))
self.splitter = QtGui.QSplitter(self.centralwidget) self.splitter = QtGui.QSplitter(self.centralwidget)
self.splitter.setOrientation(QtCore.Qt.Horizontal) self.splitter.setOrientation(QtCore.Qt.Horizontal)
self.splitter.setObjectName(_fromUtf8("splitter")) self.splitter.setObjectName(_fromUtf8("splitter"))
@ -65,7 +65,7 @@ class Ui_MainWindow(object):
sizePolicy.setHeightForWidth(self.graphicsView.sizePolicy().hasHeightForWidth()) sizePolicy.setHeightForWidth(self.graphicsView.sizePolicy().hasHeightForWidth())
self.graphicsView.setSizePolicy(sizePolicy) self.graphicsView.setSizePolicy(sizePolicy)
self.graphicsView.setObjectName(_fromUtf8("graphicsView")) self.graphicsView.setObjectName(_fromUtf8("graphicsView"))
self.verticalLayout.addWidget(self.splitter) self.horizontalLayout.addWidget(self.splitter)
MainWindow.setCentralWidget(self.centralwidget) MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QtGui.QMenuBar(MainWindow) self.menubar = QtGui.QMenuBar(MainWindow)
self.menubar.setGeometry(QtCore.QRect(0, 0, 956, 22)) self.menubar.setGeometry(QtCore.QRect(0, 0, 956, 22))

View File

@ -26,7 +26,7 @@
<verstretch>0</verstretch> <verstretch>0</verstretch>
</sizepolicy> </sizepolicy>
</property> </property>
<layout class="QVBoxLayout" name="verticalLayout"> <layout class="QHBoxLayout" name="horizontalLayout">
<item> <item>
<widget class="QSplitter" name="splitter"> <widget class="QSplitter" name="splitter">
<property name="orientation"> <property name="orientation">
@ -63,6 +63,15 @@
<height>572</height> <height>572</height>
</rect> </rect>
</property> </property>
<layout class="QVBoxLayout" name="verticalLayout">
<item>
<widget class="QGroupBox" name="groupBox">
<property name="title">
<string>GroupBox</string>
</property>
</widget>
</item>
</layout>
</widget> </widget>
</widget> </widget>
<widget class="PlotWidget" name="graphicsView"> <widget class="PlotWidget" name="graphicsView">

29
data.py
View File

@ -19,7 +19,7 @@ class Data:
self.data_curve_imag = pg.PlotDataItem(x=[N.nan], y=[N.nan],pen=QColor(0,0,0,0), symbol='o', self.data_curve_imag = pg.PlotDataItem(x=[N.nan], y=[N.nan],pen=QColor(0,0,0,0), symbol='o',
symbolBrush=(255,127,0,127)) symbolBrush=(255,127,0,127))
self.data_curve_real = pg.PlotDataItem(x=[N.nan], y=[N.nan],pen=QColor(0,0,0,0), symbol='s', self.data_curve_real = pg.PlotDataItem(x=[N.nan], y=[N.nan],pen=QColor(0,0,0,0), symbol='s',
symbolBrush=(53,159,50,127)) symbolBrush=(119,202,92,127))
self.fitted_curve_imag = pg.PlotDataItem(N.array([N.nan]), N.array([N.nan]), pen=myPen_imag) self.fitted_curve_imag = pg.PlotDataItem(N.array([N.nan]), N.array([N.nan]), pen=myPen_imag)
self.fitted_curve_real = pg.PlotDataItem(N.array([N.nan]), N.array([N.nan]), pen=myPen_real) self.fitted_curve_real = pg.PlotDataItem(N.array([N.nan]), N.array([N.nan]), pen=myPen_real)
self.length = len(frequency) self.length = len(frequency)
@ -63,8 +63,8 @@ class Data:
""" """
mask = N.ones(len(self.frequency), dtype='bool') mask = N.ones(len(self.frequency), dtype='bool')
mask = (self.frequency > self.fit_limits[0]) & (self.frequency < self.fit_limits[1]) mask = (self.frequency > self.fit_limits[0]) & (self.frequency < self.fit_limits[1])
mask &= (self.epsilon.imag > self.fit_limits[2]) & (self.epsilon.imag < self.fit_limits[3]) #mask &= (self.epsilon.imag > self.fit_limits[2]) & (self.epsilon.imag < self.fit_limits[3])
mask &= (self.epsilon.real > self.fit_limits[2]) & (self.epsilon.real < self.fit_limits[3]) #mask &= (self.epsilon.real > self.fit_limits[2]) & (self.epsilon.real < self.fit_limits[3])
return self.frequency[mask], self.epsilon[mask] return self.frequency[mask], self.epsilon[mask]
def remove_curves(self): def remove_curves(self):
@ -117,15 +117,16 @@ class Conductivity(QObject):
y = conductivity([sigma, sigma_N], nu) y = conductivity([sigma, sigma_N], nu)
y_static = N.ones(len(nu)) * eps_static y_static = N.ones(len(nu)) * eps_static
# clip data to axes limits # clip data to axes limits
mask_static = (y_static < y_max) & (y_static > y_min)
# clip data to axes limits
mask = (y < y_max) & (y > y_min)
self.frequency = nu[mask] #mask_static = (y_static < y_max) & (y_static > y_min)
self.conductivity = y[mask] # clip data to axes limits
self.epsilon_static = y_static[mask] #mask = (y < y_max) & (y > y_min)
self.mpl_line.setData(x=nu[mask], y=y[mask], label="Cond.")
self.mpl_line_static.setData(x=nu[mask_static], y=y_static[mask_static]) self.frequency = nu
self.conductivity = y
self.epsilon_static = y_static
self.mpl_line.setData(x=nu, y=y, label="Cond.")
self.mpl_line_static.setData(x=nu, y=y_static)
self.changedData.emit() self.changedData.emit()
@ -184,9 +185,9 @@ class Peak(QObject):
nu = N.logspace(N.log10(x_min), N.log10(x_max), 2048) nu = N.logspace(N.log10(x_min), N.log10(x_max), 2048)
y = hn(self.getParameter(), nu) y = hn(self.getParameter(), nu)
# clip data to axes limits # clip data to axes limits
mask = (y < y_max) & (y > y_min) #mask = (y < y_max) & (y > y_min)
y = y[mask] #y = y[mask]
nu = nu[mask] #nu = nu[mask]
self.frequency = nu[:] self.frequency = nu[:]
self.epsilon = y[:] self.epsilon = y[:]
self.mpl_line.setData(x=nu, y=y) self.mpl_line.setData(x=nu, y=y)

View File

@ -2,7 +2,7 @@
# Resource object code # Resource object code
# #
# Created: Fr. Feb. 21 17:15:59 2014 # Created: Sa. Mär. 15 21:26:00 2014
# by: The Resource Compiler for PyQt (Qt v4.8.5) # by: The Resource Compiler for PyQt (Qt v4.8.5)
# #
# WARNING! All changes made in this file will be lost! # WARNING! All changes made in this file will be lost!

View File

@ -74,7 +74,7 @@ def hn(p, nu):
-b / 2.) * N.sin(b * Phi) -b / 2.) * N.sin(b * Phi)
e_stor = delta_eps * (1 + 2 * (om * tau) ** a * N.cos(N.pi * a / 2.) + (om * tau) ** (2. * a) ) ** ( e_stor = delta_eps * (1 + 2 * (om * tau) ** a * N.cos(N.pi * a / 2.) + (om * tau) ** (2. * a) ) ** (
-b / 2.) * N.cos(b * Phi) -b / 2.) * N.cos(b * Phi)
return 2 * e_loss return e_loss # 2* oder nicht?
def id_to_color(id): def id_to_color(id):
@ -139,32 +139,30 @@ class Functions:
def hn_cmplx(self, p, x): def hn_cmplx(self, p, x):
om = 2*N.pi*x om = 2*N.pi*x
hn = om*1j #hn = om*1j
eps,t,a,b = p eps,t,a,b = p
print p hn = eps/(1+(1j*om*t)**a)**b
hn = eps/(1-(1j*om*t)**a)**b cplx = N.array([hn.real, -hn.imag])
cplx = N.array([hn.real, hn.imag])
print cplx
return cplx return cplx
def cond_cmplx(self, p, x): def cond_cmplx(self, p, x):
om = 2*N.pi*x om = 2*N.pi*x
sgma = p[0] sgma = p[0]
cond = -sgma/(1j*om) cond = sgma/(1j*om)
cplx = N.array([cond.real, cond.imag]) cplx = N.array([cond.real, -cond.imag])
return cplx return cplx
def power_cmplx(self, p, x): def power_cmplx(self, p, x):
om = 2*N.pi*x om = 2*N.pi*x
sgma,n = p sgma,n = p
power = -sgma/(om*1j)**n power = sgma/(om*1j)**n
cplx = N.array([power.real, power.imag]) cplx = N.array([power.real, -power.imag])
return cplx return cplx
def static_cmplx(self, p, x): def static_cmplx(self, p, x):
eps_inf = p[0] eps_inf = p[0]
static = N.ones( (2,x.size) )*eps_inf static = N.ones( (2,x.size) )*eps_inf
static[:,1] *= 0 # set imag part zero static[1,:] *= 0 # set imag part zero
#cplx = N.array([static.real, static.imag]) #cplx = N.array([static.real, static.imag])
return static return static
@ -184,7 +182,6 @@ class FitFunctionCreator:
self.data = N.zeros( (2,x.size) ) self.data = N.zeros( (2,x.size) )
ndx = 0 ndx = 0
for fn in funcs: # loop over functions and add the results up for fn in funcs: # loop over functions and add the results up
f,num_p = self.functions.get(fn) f,num_p = self.functions.get(fn)
p = p0[ndx:ndx + num_p] p = p0[ndx:ndx + num_p]
if x.ndim == 2: if x.ndim == 2:
@ -195,16 +192,20 @@ class FitFunctionCreator:
def fit_odr_cmplx(x, y, p0, fixed, fcns): def fit_odr_cmplx(x, y, p0, fixed, fcns):
f = FitFunctionCreator() f = FitFunctionCreator()
if x.ndim < 2: #if x.ndim < 2:
x = N.resize(x, (2,x.size)) # x = N.resize(x, (2,x.size))
if N.iscomplexobj(y) and y.ndim == 1: if N.iscomplexobj(y) and y.ndim == 1:
weights = 1/N.abs(y)**2
we = N.resize(weights, (2, weights.size))
#we = 1/N.array([y.real**2, y.imag**2])
y = N.array([y.real, y.imag]) y = N.array([y.real, y.imag])
else: else:
raise NotImplementedError,"need complex input for now" raise NotImplementedError,"need complex input for now"
dat = odr.Data(x, y, 1.0 / y**2) dat = odr.Data(x, y, we=we)
mod = odr.Model(f.fitfcn, extra_args=fcns) mod = odr.Model(f.fitfcn, extra_args=fcns)
fit = odr.ODR(dat, mod, p0, ifixx=N.zeros(x.ndim), ifixb=fixed, maxit=5000) fit = odr.ODR(dat, mod, p0, ifixx=(0,), ifixb=fixed, maxit=8000)
fit.run() fit.run()
print fit.output.pprint()
return fit.output return fit.output