Surface NMR processing and inversion GUI
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

akvoGUI.py 67KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491
  1. #/usr/bin/env python
  2. import sys
  3. #import readline
  4. try:
  5. from akvo.gui.main_ui import Ui_MainWindow
  6. uicerr = False
  7. except: # Fallback
  8. from akvo.gui.mainui import Ui_MainWindow
  9. uicerr = """
  10. USING THE DEFAULT GUI FILES, AKVO MAY NOT WORK CORRECTLY!
  11. See INSTALL.txt for details regarding GUI configuration
  12. if you are encountering problems.
  13. Clicking ignore will prevent this warning from showing
  14. each time you launch Akvo.
  15. """
  16. import matplotlib
  17. #matplotlib.use("QT4Agg")
  18. from PyQt5 import QtCore, QtGui, QtWidgets
  19. import numpy as np
  20. import time
  21. import os
  22. from copy import deepcopy
  23. from matplotlib.backends.backend_qt4 import NavigationToolbar2QT #as NavigationToolbar
  24. import datetime, time
  25. from akvo.tressel import mrsurvey
  26. #from akvo.lemma import pyLemmaCore # Looking ahead!
  27. import pkg_resources # part of setuptools
  28. version = pkg_resources.require("Akvo")[0].version
  29. import yaml
  30. # Writes out numpy arrays into Eigen vectors as serialized by Lemma
  31. class MatrixXr(yaml.YAMLObject):
  32. yaml_tag = u'MatrixXr'
  33. def __init__(self, rows, cols, data):
  34. self.rows = rows
  35. self.cols = cols
  36. self.data = np.zeros((rows,cols))
  37. def __repr__(self):
  38. return "%s(rows=%r, cols=%r, data=%r)" % (self.__class__.__name__, self.rows, self.cols, self.data)
  39. class VectorXr(yaml.YAMLObject):
  40. yaml_tag = r'VectorXr'
  41. def __init__(self, array):
  42. self.size = np.shape(array)[0]
  43. self.data = array.tolist()
  44. def __repr__(self):
  45. # Converts to numpy array on import
  46. return "np.array(%r)" % (self.data)
  47. from collections import OrderedDict
  48. #def represent_ordereddict(dumper, data):
  49. # print("representing IN DA HOUSE!!!!!!!!!!!!!!!!!!!!!")
  50. # value = []
  51. # for item_key, item_value in data.items():
  52. # node_key = dumper.represent_data(item_key)
  53. # node_value = dumper.represent_data(item_value)
  54. # value.append((node_key, node_value))
  55. # return yaml.nodes.MappingNode(u'tag:yaml.org,2002:map', value)
  56. #yaml.add_representer(OrderedDict, represent_ordereddict)
  57. def setup_yaml():
  58. """ https://stackoverflow.com/a/8661021 """
  59. represent_dict_order = lambda self, data: self.represent_mapping('tag:yaml.org,2002:map', data.items())
  60. yaml.add_representer(OrderedDict, represent_dict_order)
  61. setup_yaml()
  62. class AkvoYamlNode(yaml.YAMLObject):
  63. yaml_tag = u'AkvoData'
  64. def __init__(self):
  65. self.Akvo_VERSION = version
  66. self.Import = OrderedDict() # {}
  67. self.Processing = OrderedDict()
  68. #def __init__(self, node):
  69. # self.Akvo_VERSION = node["version"]
  70. # self.Import = OrderedDict( node["Import"] ) # {}
  71. # self.Processing = OrderedDict( node["Processing"] )
  72. def __repr__(self):
  73. return "%s(name=%r, Akvo_VERSION=%r, Import=%r, Processing=%r)" % (
  74. #self.__class__.__name__, self.Akvo_VERSION, self.Import, self.Processing )
  75. self.__class__.__name__, self.Akvo_VERSION, self.Import, OrderedDict(self.Processing) )
  76. #self.__class__.__name__, self.Akvo_VERSION, self.Import, OrderedDict(self.Processing))
  77. try:
  78. import thread
  79. except ImportError:
  80. import _thread as thread #Py3K compatibility
  81. class MyPopup(QtWidgets.QWidget):
  82. def __init__(self, name):
  83. super().__init__()
  84. self.name = name
  85. self.initUI()
  86. def initUI(self):
  87. lblName = QtWidgets.QLabel(self.name, self)
  88. class ApplicationWindow(QtWidgets.QMainWindow):
  89. def __init__(self):
  90. QtWidgets.QMainWindow.__init__(self)
  91. self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
  92. akvohome = os.path.expanduser("~") + "/.akvo"
  93. if not os.path.exists(akvohome):
  94. os.makedirs(akvohome)
  95. self.ui = Ui_MainWindow()
  96. self.ui.setupUi(self)
  97. if uicerr != False and not os.path.exists(akvohome+"/pyuic-warned"):
  98. reply = QtGui.QMessageBox.warning(self, 'Warning', uicerr, QtGui.QMessageBox.Ok, QtGui.QMessageBox.Ignore)
  99. if reply == 1024: # "0x400" in hex
  100. pass
  101. elif reply == 1048576: # "0x100000" in hex
  102. warn = open( akvohome+"/pyuic-warned" ,"w" )
  103. warn.write("Gui files were not compiled locally using pyuic! Further warnings have been supressed")
  104. warn.close()
  105. self.RAWDataProc = None
  106. self.YamlNode = AkvoYamlNode()
  107. # initialise some stuff
  108. self.ui.lcdNumberTauPulse2.setEnabled(0)
  109. self.ui.lcdNumberTauPulse1.setEnabled(0)
  110. self.ui.lcdNumberNuTx.setEnabled(0)
  111. self.ui.lcdNumberTuneuF.setEnabled(0)
  112. self.ui.lcdNumberSampFreq.setEnabled(0)
  113. self.ui.lcdNumberTauDelay.setEnabled(0)
  114. self.ui.lcdNumberNQ.setEnabled(0)
  115. #MAK 20170126: add in a list to hold processing steps
  116. self.logText = []
  117. ####################
  118. # Make connections #
  119. ####################
  120. # Menu items
  121. self.ui.actionOpen_GMR.triggered.connect(self.openGMRRAWDataset)
  122. self.ui.actionSave_Preprocessed_Dataset.triggered.connect(self.SavePreprocess)
  123. self.ui.actionExport_Preprocessed_Dataset.triggered.connect(self.ExportPreprocess)
  124. self.ui.actionExport_Preprocessed_Dataset.setEnabled(False)
  125. self.ui.actionOpen_Preprocessed_Dataset.triggered.connect(self.OpenPreprocess)
  126. self.ui.actionAboutAkvo.triggered.connect(self.about)
  127. # Buttons
  128. # #QtCore.QObject.connect(self.ui.fullWorkflowPushButton, QtCore.SIGNAL("clicked()"), self.preprocess )
  129. self.ui.loadDataPushButton.pressed.connect(self.loadRAW)
  130. self.ui.sumDataGO.pressed.connect( self.sumDataChans )
  131. self.ui.bandPassGO.pressed.connect( self.bandPassFilter )
  132. self.ui.filterDesignPushButton.pressed.connect( self.designFilter )
  133. self.ui.fdDesignPushButton.pressed.connect( self.designFDFilter )
  134. self.ui.downSampleGO.pressed.connect( self.downsample )
  135. self.ui.windowFilterGO.pressed.connect( self.windowFilter )
  136. # self.ui.despikeGO.pressed.connect( self.despikeFilter ) # use smart stack instead
  137. self.ui.adaptGO.pressed.connect( self.adaptFilter )
  138. self.ui.adaptFDGO.pressed.connect( self.adaptFilterFD )
  139. self.ui.qdGO.pressed.connect( self.quadDet )
  140. self.ui.gateIntegrateGO.pressed.connect( self.gateIntegrate )
  141. self.ui.calcQGO.pressed.connect( self.calcQ )
  142. self.ui.FDSmartStackGO.pressed.connect( self.FDSmartStack )
  143. self.ui.harmonicGO.pressed.connect( self.harmonicModel )
  144. self.ui.f0K1Spin.valueChanged.connect( self.LCDHarmonics )
  145. self.ui.f0KNSpin.valueChanged.connect( self.LCDHarmonics )
  146. self.ui.f0KsSpin.valueChanged.connect( self.LCDHarmonics )
  147. self.ui.f0Spin.valueChanged.connect( self.LCDHarmonics )
  148. self.ui.NHarmonicsFreqsSpin.valueChanged.connect( self.LCDHarmonics2 )
  149. self.ui.f1K1Spin.valueChanged.connect( self.LCDHarmonics2 )
  150. self.ui.f1KNSpin.valueChanged.connect( self.LCDHarmonics2 )
  151. self.ui.f1KsSpin.valueChanged.connect( self.LCDHarmonics2 )
  152. self.ui.f1Spin.valueChanged.connect( self.LCDHarmonics2 )
  153. self.ui.plotQD.setEnabled(False)
  154. self.ui.plotQD.pressed.connect( self.plotQD )
  155. self.ui.plotGI.setEnabled(False)
  156. self.ui.plotGI.pressed.connect( self.plotGI )
  157. # hide header info box
  158. #self.ui.headerFileBox.setVisible(False)
  159. self.ui.headerFileBox.clicked.connect( self.headerBoxShrink )
  160. self.ui.headerBox2.setVisible(False)
  161. # Clean up the tab widget
  162. self.ui.actionPreprocessing.triggered.connect(self.addPreProc)
  163. self.ui.actionModelling.triggered.connect(self.addModelling)
  164. self.ui.actionInversion.triggered.connect(self.addInversion)
  165. # tabs
  166. #self.ui.ProcTabs.tabCloseRequested.connect( self.closeTabs )
  167. #self.ui.ProcTabs.tabBar().setTabButton(7, QtWidgets.QTabBar.RightSide,None)
  168. self.ui.ProcTabs.removeTab(4)
  169. self.ui.ProcTabs.removeTab(4)
  170. self.ui.ProcTabs.removeTab(4)
  171. self.ui.ProcTabs.removeTab(4)
  172. #self.ui.LoadTab.close( )
  173. # Add progressbar to statusbar
  174. self.ui.barProgress = QtWidgets.QProgressBar()
  175. self.ui.statusbar.addPermanentWidget(self.ui.barProgress, 0);
  176. self.ui.barProgress.setMaximumSize(100, 16777215);
  177. self.ui.barProgress.hide();
  178. self.ui.mplwidget_navigator.setCanvas(self.ui.mplwidget)
  179. #self.ui.mplwidget_navigator_2.setCanvas(self.ui.mplwidget)
  180. ##########################################################################
  181. # Loop Table
  182. self.ui.loopTableWidget.setRowCount(80)
  183. self.ui.loopTableWidget.setColumnCount(6)
  184. self.ui.loopTableWidget.setHorizontalHeaderLabels( ["ch. tag", \
  185. "Northing [m]","Easting [m]","Height [m]", "Radius","Tx"] )
  186. for ir in range(0, self.ui.loopTableWidget.rowCount() ):
  187. for ic in range(1, self.ui.loopTableWidget.columnCount() ):
  188. pCell = QtWidgets.QTableWidgetItem()
  189. #pCell.setFlags(QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable)
  190. pCell.setFlags(QtCore.Qt.NoItemFlags) # not selectable
  191. pCell.setBackground( QtGui.QColor("lightgrey").lighter(110) )
  192. self.ui.loopTableWidget.setItem(ir, ic, pCell)
  193. self.ui.loopTableWidget.cellChanged.connect(self.loopCellChanged)
  194. #self.ui.loopTableWidget.cellPressed.connect(self.loopCellChanged)
  195. self.ui.loopTableWidget.itemClicked.connect(self.loopCellClicked)
  196. #self.ui.loopTableWidget.cellPressed.connect(self.loopCellClicked)
  197. self.ui.loopTableWidget.setDragDropOverwriteMode(False)
  198. self.ui.loopTableWidget.setDragEnabled(False)
  199. #self.ui.loopTableWidget.setDragDropMode(QtWidgets.QAbstractItemView.InternalMove)
  200. self.loops = {}
  201. ##########################################################################
  202. # layer Table
  203. self.ui.layerTableWidget.setRowCount(80)
  204. self.ui.layerTableWidget.setColumnCount(3)
  205. self.ui.layerTableWidget.setHorizontalHeaderLabels( [r"top [m]", r"bottom [m]", "σ [ Ωm]" ] )
  206. # do we want this
  207. self.ui.layerTableWidget.setDragDropOverwriteMode(False)
  208. self.ui.layerTableWidget.setDragEnabled(True)
  209. self.ui.layerTableWidget.setDragDropMode(QtWidgets.QAbstractItemView.InternalMove)
  210. pCell = QtWidgets.QTableWidgetItem()
  211. pCell.setFlags(QtCore.Qt.NoItemFlags) # not selectable
  212. pCell.setBackground( QtGui.QColor("lightgrey").lighter(110) )
  213. self.ui.layerTableWidget.setItem(0, 1, pCell)
  214. for ir in range(1, self.ui.layerTableWidget.rowCount() ):
  215. for ic in range(0, self.ui.layerTableWidget.columnCount() ):
  216. pCell = QtWidgets.QTableWidgetItem()
  217. #pCell.setFlags(QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable)
  218. pCell.setFlags(QtCore.Qt.NoItemFlags) # not selectable
  219. pCell.setBackground( QtGui.QColor("lightgrey").lighter(110) )
  220. self.ui.layerTableWidget.setItem(ir, ic, pCell)
  221. self.ui.layerTableWidget.cellChanged.connect(self.sigmaCellChanged)
  222. def LCDHarmonics(self):
  223. self.ui.lcdH1F.setEnabled(True)
  224. self.ui.lcdH1F.display( self.ui.f0Spin.value() * self.ui.f0K1Spin.value() )
  225. self.ui.lcdHNF.setEnabled(True)
  226. self.ui.lcdHNF.display( self.ui.f0Spin.value() * self.ui.f0KNSpin.value() )
  227. self.ui.lcdf0NK.setEnabled(True)
  228. self.ui.lcdf0NK.display( (self.ui.f0KNSpin.value()+1-self.ui.f0K1Spin.value()) * self.ui.f0KsSpin.value() )
  229. def LCDHarmonics2(self):
  230. if self.ui.NHarmonicsFreqsSpin.value() == 2:
  231. self.ui.lcdH1F2.setEnabled(True)
  232. self.ui.lcdH1F2.display( self.ui.f1Spin.value() * self.ui.f1K1Spin.value() )
  233. self.ui.lcdHNF2.setEnabled(True)
  234. self.ui.lcdHNF2.display( self.ui.f1Spin.value() * self.ui.f1KNSpin.value() )
  235. self.ui.lcdf0NK2.setEnabled(True)
  236. self.ui.lcdf0NK2.display( (self.ui.f1KNSpin.value()+1-self.ui.f1K1Spin.value()) * self.ui.f1KsSpin.value() )
  237. else:
  238. self.ui.lcdH1F2.setEnabled(False)
  239. self.ui.lcdHNF2.setEnabled(False)
  240. self.ui.lcdf0NK2.setEnabled(False)
  241. def closeTabs(self):
  242. #self.ui.ProcTabs.removeTab(idx)
  243. self.ui.ProcTabs.clear( )
  244. def addPreProc(self):
  245. if self.ui.actionPreprocessing.isChecked():
  246. self.ui.actionModelling.setChecked(False)
  247. self.ui.actionInversion.setChecked(False)
  248. self.ui.ProcTabs.clear( )
  249. self.ui.ProcTabs.insertTab( 0, self.ui.LoadTab, "Load" )
  250. self.ui.ProcTabs.insertTab( 1, self.ui.NCTab, "NC" )
  251. self.ui.ProcTabs.insertTab( 2, self.ui.QCTab, "QC" )
  252. self.ui.ProcTabs.insertTab( 3, self.ui.METATab, "META" )
  253. self.ui.ProcTabs.insertTab( 4, self.ui.LogTab, "Log" )
  254. else:
  255. self.ui.ProcTabs.removeTab(0)
  256. self.ui.ProcTabs.removeTab(0)
  257. self.ui.ProcTabs.removeTab(0)
  258. self.ui.ProcTabs.removeTab(0)
  259. def addModelling(self):
  260. if self.ui.actionModelling.isChecked():
  261. self.ui.actionPreprocessing.setChecked(False)
  262. self.ui.actionInversion.setChecked(False)
  263. self.ui.ProcTabs.clear( )
  264. self.ui.ProcTabs.insertTab( 0, self.ui.KernTab, "Kernel" )
  265. self.ui.ProcTabs.insertTab( 1, self.ui.ModelTab, "Modelling" )
  266. self.ui.ProcTabs.insertTab( 2, self.ui.LogTab, "Log" )
  267. else:
  268. self.ui.ProcTabs.removeTab(0)
  269. self.ui.ProcTabs.removeTab(0)
  270. def addInversion(self, idx):
  271. if self.ui.actionInversion.isChecked():
  272. self.ui.actionPreprocessing.setChecked(False)
  273. self.ui.actionModelling.setChecked(False)
  274. self.ui.ProcTabs.clear( )
  275. self.ui.ProcTabs.insertTab( 0, self.ui.InvertTab, "Inversion" )
  276. self.ui.ProcTabs.insertTab( 1, self.ui.AppraiseTab, "Appraisal" )
  277. self.ui.ProcTabs.insertTab( 2, self.ui.LogTab, "Log" )
  278. else:
  279. self.ui.ProcTabs.removeTab(0)
  280. self.ui.ProcTabs.removeTab(0)
  281. def headerBoxShrink(self):
  282. #self.ui.headerFileBox.setVisible(False)
  283. if self.ui.headerFileBox.isChecked( ):
  284. #self.ui.headerFileBox.setMinimumSize(460,250)
  285. self.ui.headerBox2.setVisible(True)
  286. else:
  287. #self.ui.headerFileBox.setMinimumSize(460,50)
  288. self.ui.headerBox2.setVisible(False)
  289. def sigmaCellChanged(self):
  290. self.ui.layerTableWidget.cellChanged.disconnect(self.sigmaCellChanged)
  291. # TODO consider building the model whenever this is called. Would be nice to be able to
  292. # do that. Would require instead dist of T2 I guess.
  293. jj = self.ui.layerTableWidget.currentColumn()
  294. ii = self.ui.layerTableWidget.currentRow()
  295. val = "class 'NoneType'>"
  296. try:
  297. val = eval (str( self.ui.layerTableWidget.item(ii, jj).text() ))
  298. except:
  299. #if jj != 0:
  300. # Error = QtWidgets.QMessageBox()
  301. # Error.setWindowTitle("Error!")
  302. # Error.setText("Non-numeric value encountered")
  303. self.ui.layerTableWidget.cellChanged.connect(self.sigmaCellChanged)
  304. return
  305. if jj == 1:
  306. #item.setFlags(QtCore.Qt.ItemIsEnabled)
  307. pCell = self.ui.layerTableWidget.item(ii, jj)
  308. pCell.setBackground( QtGui.QColor("white"))
  309. pCell = self.ui.layerTableWidget.item(ii+1, jj-1)
  310. if str(type(pCell)) == "<class 'NoneType'>":
  311. pCell = QtWidgets.QTableWidgetItem()
  312. pCell.setFlags(QtCore.Qt.ItemIsEnabled)
  313. self.ui.layerTableWidget.setItem(ii+1, jj-1, pCell)
  314. if ii == 0:
  315. pCell.setText(str(val))
  316. #pCell3 = self.ui.layerTableWidget.item(ii+1, jj)
  317. #print ("setting", ii, jj, type(pCell3))
  318. #print ( "setting", ii, jj, type(pCell3))
  319. #pCell3.setFlags( QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled )
  320. #pCell3.setFlags( QtCore.Qt.ItemIsEditable )
  321. elif ii > 0:
  322. val2 = eval (str( self.ui.layerTableWidget.item(ii-1, jj).text() ))
  323. #print ("val2", val2, val, type(val))
  324. #if str(type(pCell)) == "<class 'NoneType'>":
  325. if type(val) == str or val > val2:
  326. pCell.setText(str(val))
  327. else:
  328. Error = QtWidgets.QMessageBox()
  329. Error.setWindowTitle("Error!")
  330. Error.setText("Non-increasing layer detected")
  331. Error.setDetailedText("Each layer interface must be below the one above it.")
  332. Error.exec_()
  333. pCell2 = self.ui.layerTableWidget.item(ii, jj)
  334. pCell2.setText(str(""))
  335. self.ui.layerTableWidget.cellChanged.connect(self.sigmaCellChanged)
  336. return
  337. # enable next layer
  338. pCell4 = self.ui.layerTableWidget.item(ii+1, jj)
  339. pCell4.setBackground( QtGui.QColor("lightblue") ) #.lighter(110))
  340. pCell4.setFlags( QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled )
  341. pCell5 = self.ui.layerTableWidget.item(ii+1, jj+1)
  342. pCell5.setBackground( QtGui.QColor("white"))
  343. pCell5.setFlags( QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled )
  344. print("ii", ii, "jj", jj)
  345. if ii == 0 and jj == 0:
  346. pCell = self.ui.layerTableWidget.item(0, 1)
  347. pCell.setBackground( QtGui.QColor("lightblue")) #.lighter(110) )
  348. pCell.setFlags( QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled )
  349. self.ui.layerTableWidget.cellChanged.connect(self.sigmaCellChanged)
  350. def loopCellClicked(self, item):
  351. print("checkstate", item.checkState(),item.row())
  352. #self.ui.loopTableWidget.itemClicked.disconnect(self.loopCellClicked)
  353. jj = item.column()
  354. ii = item.row()
  355. tp = type(self.ui.loopTableWidget.item(ii, 0))
  356. print("tp", tp, ii, jj)
  357. if str(tp) == "<class 'NoneType'>":
  358. return
  359. #print("Clicked", ii, jj)
  360. if jj == 5 and self.ui.loopTableWidget.item(ii, 0).text() in self.loops.keys():
  361. #print("jj=5")
  362. self.loops[ self.ui.loopTableWidget.item(ii, 0).text() ]["Tx"] = self.ui.loopTableWidget.item(ii, 5).checkState()
  363. # update surrogates
  364. print("updating surrogates")
  365. for point in self.loops[ self.ui.loopTableWidget.item(ii, 0).text() ]["points"][1:]:
  366. pCell = self.ui.loopTableWidget.item(point, 5)
  367. if self.ui.loopTableWidget.item(ii, 5).checkState():
  368. pCell.setCheckState(QtCore.Qt.Checked);
  369. else:
  370. pCell.setCheckState(QtCore.Qt.Unchecked);
  371. #print( "loops", self.loops[ self.ui.loopTableWidget.item(ii, 0).text() ]["Tx"])
  372. #self.ui.loopTableWidget.itemClicked.connect(self.loopCellClicked)
  373. def loopCellChanged(self):
  374. self.ui.loopTableWidget.cellChanged.disconnect(self.loopCellChanged)
  375. jj = self.ui.loopTableWidget.currentColumn()
  376. ii = self.ui.loopTableWidget.currentRow()
  377. if jj == 0 and len( self.ui.loopTableWidget.item(ii, jj).text().strip()) == 0:
  378. for jjj in range(jj+1,jj+6):
  379. pCell = self.ui.loopTableWidget.item(ii, jjj)
  380. pCell.setBackground( QtGui.QColor("white") )
  381. pCell.setFlags( QtCore.Qt.NoItemFlags | QtCore.Qt.ItemIsUserCheckable ) # not selectable
  382. elif jj == 0 and len( self.ui.loopTableWidget.item(ii, jj).text().strip() ): # ch. tag modified
  383. for jjj in range(jj+1,jj+5):
  384. pCell = self.ui.loopTableWidget.item(ii, jjj)
  385. pCell.setFlags( QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled )
  386. pCell.setBackground( QtGui.QColor("lightblue") )
  387. if self.ui.loopTableWidget.item(ii, jj).text() not in self.loops.keys():
  388. # This is a new loop ID
  389. self.loops[ self.ui.loopTableWidget.item(ii, jj).text() ] = {}
  390. self.loops[ self.ui.loopTableWidget.item(ii, jj).text() ]["Tx"] = self.ui.loopTableWidget.item(ii, 5).checkState()
  391. self.loops[ self.ui.loopTableWidget.item(ii, jj).text() ]["points"] = [ii]
  392. # Transmitter cell
  393. pCell = self.ui.loopTableWidget.item(ii, jj+5)
  394. pCell.setCheckState(QtCore.Qt.Unchecked)
  395. pCell.setFlags( QtCore.Qt.ItemIsUserCheckable | QtCore.Qt.ItemIsEnabled )
  396. pCell.setBackground( QtGui.QColor("lightblue") )
  397. else:
  398. # This is an existing loop ID
  399. self.loops[ self.ui.loopTableWidget.item(ii, jj).text() ]["points"].append( ii )
  400. pCell = self.ui.loopTableWidget.item(ii, jj+5)
  401. pCell.setFlags(QtCore.Qt.NoItemFlags) # not selectable
  402. if self.loops[ self.ui.loopTableWidget.item(ii, 0).text() ]["Tx"]:
  403. pCell.setCheckState(QtCore.Qt.Checked);
  404. else:
  405. pCell.setCheckState(QtCore.Qt.Unchecked);
  406. #pCell.setFlags( )
  407. pCell.setBackground( QtGui.QColor("lightblue") )
  408. self.plotLoops()
  409. self.ui.loopTableWidget.cellChanged.connect(self.loopCellChanged)
  410. def plotLoops(self):
  411. print("Plotting loopz")
  412. self.ui.mplwidget.reAxH(1)
  413. #self.ui.mplwidget.ax1.clear()
  414. #self.ui.mplwidget.ax2.clear()
  415. nor = dict()
  416. eas = dict()
  417. dep = dict()
  418. for ii in range( self.ui.loopTableWidget.rowCount() ):
  419. for jj in range( self.ui.loopTableWidget.columnCount() ):
  420. tp = type(self.ui.loopTableWidget.item(ii, jj))
  421. if str(tp) == "<class 'NoneType'>":
  422. pass
  423. elif not len(self.ui.loopTableWidget.item(ii, jj).text()):
  424. pass
  425. else:
  426. if jj == 0:
  427. idx = self.ui.loopTableWidget.item(ii, 0).text()
  428. if idx not in nor.keys():
  429. nor[idx] = list()
  430. eas[idx] = list()
  431. dep[idx] = list()
  432. if jj == 1:
  433. nor[idx].append( eval(self.ui.loopTableWidget.item(ii, 1).text()) )
  434. elif jj == 2:
  435. eas[idx].append( eval(self.ui.loopTableWidget.item(ii, 2).text()) )
  436. elif jj == 3:
  437. dep[idx].append( eval(self.ui.loopTableWidget.item(ii, 3).text()) )
  438. for ii in nor.keys():
  439. try:
  440. self.ui.mplwidget.ax1.plot( np.array(nor[ii]), np.array(eas[ii]) )
  441. except:
  442. pass
  443. #self.ui.mplwidget.figure.axes().set
  444. self.ui.mplwidget.ax1.set_aspect('equal') #, adjustable='box')
  445. self.ui.mplwidget.draw()
  446. def about(self):
  447. # TODO proper popup with info
  448. #self.w = MyPopup("""About Akvo \n
  449. # Akvo is an open source project developed primarily by Trevor Irons.
  450. #""")
  451. #self.w.setGeometry(100, 100, 400, 200)
  452. #self.w.show()
  453. # Just a splash screen for now
  454. logo = pkg_resources.resource_filename(__name__, 'akvo_about.png')
  455. pixmap = QtGui.QPixmap(logo)
  456. self.splash = QtWidgets.QSplashScreen(pixmap, QtCore.Qt.WindowStaysOnTopHint)
  457. self.splash.show()
  458. def connectGMRDataProcessor(self):
  459. self.RAWDataProc = mrsurvey.GMRDataProcessor()
  460. self.RAWDataProc.progressTrigger.connect(self.updateProgressBar)
  461. self.RAWDataProc.enableDSPTrigger.connect(self.enableDSP)
  462. self.RAWDataProc.doneTrigger.connect(self.doneStatus)
  463. self.RAWDataProc.updateProcTrigger.connect(self.updateProc)
  464. def openGMRRAWDataset(self):
  465. """ Opens a GMR header file
  466. """
  467. try:
  468. with open('.gmr.last.path') as f:
  469. fpath = f.readline()
  470. pass
  471. except IOError as e:
  472. fpath = '.'
  473. self.headerstr = QtWidgets.QFileDialog.getOpenFileName(self, 'Open File', fpath)[0] # arg2 = File Type 'All Files (*)'
  474. self.ui.headerFileTextBrowser.clear()
  475. self.ui.headerFileTextBrowser.append(self.headerstr)
  476. if len(self.headerstr) == 0:
  477. return
  478. # clear the processing log
  479. self.ui.logTextBrowser.clear()
  480. self.logText = [] #MAK 20170126
  481. path,filen=os.path.split(str(self.headerstr))
  482. f = open('.gmr.last.path', 'w')
  483. f.write( str(self.headerstr) ) # prompt last file
  484. self.connectGMRDataProcessor()
  485. self.RAWDataProc.readHeaderFile(str(self.headerstr))
  486. # If we got this far, enable all the widgets
  487. self.ui.lcdNumberTauPulse1.setEnabled(True)
  488. self.ui.lcdNumberNuTx.setEnabled(True)
  489. self.ui.lcdNumberTuneuF.setEnabled(True)
  490. self.ui.lcdNumberSampFreq.setEnabled(True)
  491. self.ui.lcdNumberNQ.setEnabled(True)
  492. self.ui.headerFileBox.setEnabled(True)
  493. self.ui.headerFileBox.setChecked( True )
  494. self.ui.headerBox2.setVisible(True)
  495. self.ui.inputRAWParametersBox.setEnabled(True)
  496. self.ui.loadDataPushButton.setEnabled(True)
  497. # make plots as you import the dataset
  498. self.ui.plotImportCheckBox.setEnabled(True)
  499. self.ui.plotImportCheckBox.setChecked(True)
  500. # Update info from the header into the GUI
  501. self.ui.pulseTypeTextBrowser.clear()
  502. self.ui.pulseTypeTextBrowser.append(self.RAWDataProc.pulseType)
  503. self.ui.lcdNumberNuTx.display(self.RAWDataProc.transFreq)
  504. self.ui.lcdNumberTauPulse1.display(1e3*self.RAWDataProc.pulseLength[0])
  505. self.ui.lcdNumberTuneuF.display(self.RAWDataProc.TuneCapacitance)
  506. self.ui.lcdNumberSampFreq.display(self.RAWDataProc.samp)
  507. self.ui.lcdNumberNQ.display(self.RAWDataProc.nPulseMoments)
  508. self.ui.DeadTimeSpinBox.setValue(1e3*self.RAWDataProc.deadTime)
  509. self.ui.CentralVSpinBox.setValue( self.RAWDataProc.transFreq )
  510. if self.RAWDataProc.pulseType != "FID":
  511. self.ui.lcdNumberTauPulse2.setEnabled(1)
  512. self.ui.lcdNumberTauPulse2.display(1e3*self.RAWDataProc.pulseLength[1])
  513. self.ui.lcdNumberTauDelay.setEnabled(1)
  514. self.ui.lcdNumberTauDelay.display(1e3*self.RAWDataProc.interpulseDelay)
  515. self.ui.FIDProcComboBox.clear()
  516. if self.RAWDataProc.pulseType == "4PhaseT1" or self.RAWDataProc.pulseType == "T1":
  517. self.ui.FIDProcComboBox.insertItem(0, "Pulse 1")
  518. self.ui.FIDProcComboBox.insertItem(1, "Pulse 2")
  519. self.ui.FIDProcComboBox.insertItem(2, "Both")
  520. self.ui.FIDProcComboBox.setCurrentIndex (1)
  521. elif self.RAWDataProc.pulseType == "FID":
  522. self.ui.FIDProcComboBox.insertItem(0, "Pulse 1")
  523. self.ui.FIDProcComboBox.setCurrentIndex (0)
  524. def ExportPreprocess(self):
  525. """ This method export to YAML
  526. """
  527. try:
  528. with open('.akvo.last.yaml.path') as f:
  529. fpath = f.readline()
  530. pass
  531. except IOError as e:
  532. fpath = '.'
  533. fdir = os.path.dirname(fpath)
  534. # Pickle the preprocessed data dictionary
  535. SaveStr = QtWidgets.QFileDialog.getSaveFileName(self, "Save as", fdir, r"Processed data (*.yaml)")[0]
  536. spath,filen=os.path.split(str(SaveStr))
  537. f = open('.akvo.last.yaml.path', 'w')
  538. f.write( str(spath) ) # prompt last file
  539. INFO = {}
  540. INFO["headerstr"] = str(self.headerstr)
  541. INFO["pulseType"] = self.RAWDataProc.pulseType
  542. INFO["transFreq"] = self.RAWDataProc.transFreq.tolist()
  543. INFO["pulseLength"] = self.RAWDataProc.pulseLength.tolist()
  544. INFO["TuneCapacitance"] = self.RAWDataProc.TuneCapacitance.tolist()
  545. #INFO["samp"] = self.RAWDataProc.samp
  546. INFO["nPulseMoments"] = self.RAWDataProc.nPulseMoments
  547. #INFO["deadTime"] = self.RAWDataProc.deadTime
  548. INFO["processed"] = "Akvo v. 1.0, on " + time.strftime("%d/%m/%Y")
  549. # Pulse current info
  550. ip = 0
  551. INFO["Pulses"] = {}
  552. for pulse in self.RAWDataProc.DATADICT["PULSES"]:
  553. qq = []
  554. qv = []
  555. for ipm in range(self.RAWDataProc.DATADICT["nPulseMoments"]):
  556. #for istack in self.RAWDataProc.DATADICT["stacks"]:
  557. # print ("stack q", self.RAWDataProc.DATADICT[pulse]["Q"][ipm,istack-1])
  558. qq.append(np.mean( self.RAWDataProc.DATADICT[pulse]["Q"][ipm,:]) )
  559. qv.append(np.std( self.RAWDataProc.DATADICT[pulse]["Q"][ipm,:]/self.RAWDataProc.pulseLength[ip] ))
  560. INFO["Pulses"][pulse] = {}
  561. INFO["Pulses"][pulse]["units"] = "A"
  562. INFO["Pulses"][pulse]["current"] = VectorXr(np.array(qq)/self.RAWDataProc.pulseLength[ip])
  563. INFO["Pulses"][pulse]["variance"] = VectorXr(np.array(qv))
  564. ip += 1
  565. # Data
  566. if self.RAWDataProc.gated == True:
  567. INFO["Gated"] = {}
  568. INFO["Gated"]["abscissa units"] = "ms"
  569. INFO["Gated"]["data units"] = "nT"
  570. for pulse in self.RAWDataProc.DATADICT["PULSES"]:
  571. INFO["Gated"][pulse] = {}
  572. INFO["Gated"][pulse]["abscissa"] = VectorXr( self.RAWDataProc.GATEDABSCISSA )
  573. INFO["Gated"][pulse]["windows"] = VectorXr( self.RAWDataProc.GATEDWINDOW )
  574. for ichan in self.RAWDataProc.DATADICT[pulse]["chan"]:
  575. INFO["Gated"][pulse]["Chan. " + str(ichan)] = {}
  576. INFO["Gated"][pulse]["Chan. " + str(ichan)]["STD"] = VectorXr( np.std(self.RAWDataProc.GATED[ichan]["NR"], axis=0) )
  577. for ipm in range(self.RAWDataProc.DATADICT["nPulseMoments"]):
  578. INFO["Gated"][pulse]["Chan. " + str(ichan)]["Q-"+str(ipm) + " CA"] = VectorXr(self.RAWDataProc.GATED[ichan]["CA"][ipm])
  579. INFO["Gated"][pulse]["Chan. " + str(ichan)]["Q-"+str(ipm) + " RE"] = VectorXr(self.RAWDataProc.GATED[ichan]["RE"][ipm])
  580. INFO["Gated"][pulse]["Chan. " + str(ichan)]["Q-"+str(ipm) + " IM"] = VectorXr(self.RAWDataProc.GATED[ichan]["IM"][ipm])
  581. #INFO["Gated"][pulse]["Chan. " + str(ichan)]["Q-"+str(ipm) + " IP"] = VectorXr(self.RAWDataProc.GATED[ichan]["IP"][ipm])
  582. #INFO["Gated"][pulse]["Chan. " + str(ichan)]["Q-"+str(ipm) + " NR"] = VectorXr(self.RAWDataProc.GATED[ichan]["NR"][ipm])
  583. #INFO["Gated"][pulse]["Chan. " + str(ichan)]["Q-"+str(ipm) + " STD" ] = VectorXr(self.RAWDataProc.GATED[ichan]["SIGMA"][ipm])
  584. # we have gated data
  585. # Window edges
  586. # Window centres
  587. with open(SaveStr, 'w') as outfile:
  588. #for line in self.logText:
  589. # outfile.write(line+"\n")
  590. yaml.dump(self.YamlNode, outfile)
  591. yaml.dump(INFO, outfile, default_flow_style=False)
  592. def SavePreprocess(self):
  593. #if "Saved" not in self.YamlNode.Processing.keys():
  594. # self.YamlNode.Processing["Saved"] = []
  595. #self.YamlNode.Processing["Saved"].append(datetime.datetime.now().isoformat())
  596. #self.Log()
  597. import pickle, os
  598. try:
  599. with open('.akvo.last.path') as f:
  600. fpath = f.readline()
  601. pass
  602. except IOError as e:
  603. fpath = '.'
  604. fdir = os.path.dirname(fpath)
  605. # Pickle the preprocessed data dictionary
  606. SaveStr = QtWidgets.QFileDialog.getSaveFileName(self, "Save as", fdir, r"Pickle (*.dmp)")
  607. print(SaveStr)
  608. spath,filen=os.path.split(str(SaveStr[0]))
  609. f = open('.akvo.last.path', 'w')
  610. f.write( str(spath) ) # prompt last file
  611. save = open(SaveStr[0], 'wb')
  612. # Add some extra info
  613. INFO = {}
  614. INFO["pulseType"] = self.RAWDataProc.pulseType
  615. INFO["interpulseDelay"] = self.RAWDataProc.interpulseDelay
  616. INFO["transFreq"] = self.RAWDataProc.transFreq
  617. INFO["pulseLength"] = self.RAWDataProc.pulseLength
  618. INFO["TuneCapacitance"] = self.RAWDataProc.TuneCapacitance
  619. INFO["samp"] = self.RAWDataProc.samp
  620. INFO["nPulseMoments"] = self.RAWDataProc.nPulseMoments
  621. INFO["deadTime"] = self.RAWDataProc.deadTime
  622. INFO["transFreq"] = self.RAWDataProc.transFreq
  623. INFO["headerstr"] = str(self.headerstr)
  624. INFO["nDAQVersion"] = self.RAWDataProc.nDAQVersion
  625. INFO["log"] = yaml.dump( self.YamlNode ) #self.logText #MAK 20170127
  626. self.RAWDataProc.DATADICT["INFO"] = INFO
  627. pickle.dump(self.RAWDataProc.DATADICT, save)
  628. save.close()
  629. # Export XML file suitable for USGS ScienceBase Data Release
  630. def ExportXML(self):
  631. """ This is a filler function for use by USGS collaborators
  632. """
  633. return 42
  634. def OpenPreprocess(self):
  635. import pickle
  636. try:
  637. with open('.akvo.last.path') as f:
  638. fpath = f.readline()
  639. pass
  640. except IOError as e:
  641. fpath = '.'
  642. #filename = QtWidgets.QFileDialog.getOpenFileName(self, 'Open File', '.')
  643. fpath = QtWidgets.QFileDialog.getOpenFileName(self, 'Open preprocessed file', fpath, r"Pickle Files (*.dmp)")[0]
  644. f = open('.akvo.last.path', 'w')
  645. f.write( str(fpath) ) # prompt last file
  646. self.ui.logTextBrowser.clear()
  647. self.logText = []
  648. if len(fpath) == 0:
  649. return
  650. pfile = open(fpath,'rb')
  651. unpickle = pickle.Unpickler(pfile)
  652. self.connectGMRDataProcessor()
  653. self.RAWDataProc.DATADICT = unpickle.load()
  654. # This line causes Akvo to crash, if the header file is no longer there. We don't need to load the
  655. # file. TODO, need to disable "Load Data" in Load command though, as that is no longer possible.
  656. #self.RAWDataProc.readHeaderFile(self.RAWDataProc.DATADICT["INFO"]["headerstr"])
  657. self.headerstr = self.RAWDataProc.DATADICT["INFO"]["headerstr"]
  658. self.RAWDataProc.pulseType = self.RAWDataProc.DATADICT["INFO"]["pulseType"]
  659. self.RAWDataProc.transFreq = self.RAWDataProc.DATADICT["INFO"]["transFreq"]
  660. self.RAWDataProc.pulseLength = self.RAWDataProc.DATADICT["INFO"]["pulseLength"]
  661. self.RAWDataProc.TuneCapacitance = self.RAWDataProc.DATADICT["INFO"]["TuneCapacitance"]
  662. self.RAWDataProc.samp = self.RAWDataProc.DATADICT["INFO"]["samp"]
  663. self.RAWDataProc.nPulseMoments = self.RAWDataProc.DATADICT["INFO"]["nPulseMoments"]
  664. self.RAWDataProc.deadTime = self.RAWDataProc.DATADICT["INFO"]["deadTime"]
  665. self.RAWDataProc.transFreq = self.RAWDataProc.DATADICT["INFO"]["transFreq"]
  666. self.RAWDataProc.nDAQVersion = self.RAWDataProc.DATADICT["INFO"]["nDAQVersion"]
  667. self.RAWDataProc.dt = 1./self.RAWDataProc.samp
  668. self.dataChan = self.RAWDataProc.DATADICT[ self.RAWDataProc.DATADICT["PULSES"][0] ]["chan"]
  669. # Keep backwards compatibility with prior saved pickles???
  670. #self.ui.logTextBrowser.clear()
  671. #self.ui.logTextBrowser.append( yaml.dump(self.YamlNode)) #, default_flow_style=False) )
  672. #for a in self.logText:
  673. # self.ui.logTextBrowser.append(str(a))
  674. #self.ui.logTextBrowser
  675. #self.ui.logTextBrowser.clear()
  676. #print ( self.RAWDataProc.DATADICT["INFO"]["log"] )
  677. self.logText = self.RAWDataProc.DATADICT["INFO"]["log"] # YAML
  678. self.YamlNode = AkvoYamlNode( ) #self.logText )
  679. self.YamlNode.Akvo_VERSION = (yaml.load( self.logText, Loader=yaml.Loader )).Akvo_VERSION
  680. AKVO_VERSION = np.array(self.YamlNode.Akvo_VERSION.split("."), dtype=int)
  681. if AKVO_VERSION[0] >= 1 and AKVO_VERSION[1] >= 2 and AKVO_VERSION[2] >= 3:
  682. self.RAWDataProc.interpulseDelay = self.RAWDataProc.DATADICT["INFO"]["interpulseDelay"]
  683. self.YamlNode.Import = OrderedDict((yaml.load( self.logText, Loader=yaml.Loader )).Import)
  684. self.YamlNode.Processing = OrderedDict((yaml.load( self.logText, Loader=yaml.Loader )).Processing)
  685. self.Log()
  686. #self.ui.logTextBrowser.append( yaml.dump(self.YamlNode)) #, default_flow_style=False) )
  687. #except KeyError:
  688. # pass
  689. # Remove "Saved" and "Loaded" from processing flow
  690. #if "Loaded" not in self.YamlNode.Processing.keys():
  691. # self.YamlNode.Processing["Loaded"] = []
  692. #self.YamlNode.Processing["Loaded"].append(datetime.datetime.now().isoformat())
  693. #self.Log()
  694. # If we got this far, enable all the widgets
  695. self.ui.lcdNumberTauPulse1.setEnabled(True)
  696. self.ui.lcdNumberNuTx.setEnabled(True)
  697. self.ui.lcdNumberTuneuF.setEnabled(True)
  698. self.ui.lcdNumberSampFreq.setEnabled(True)
  699. self.ui.lcdNumberNQ.setEnabled(True)
  700. self.ui.headerFileBox.setEnabled(True)
  701. self.ui.headerFileBox.setChecked( True )
  702. self.headerBoxShrink()
  703. #self.ui.headerBox2.setVisible(True)
  704. self.ui.inputRAWParametersBox.setEnabled(False)
  705. self.ui.loadDataPushButton.setEnabled(True)
  706. # make plots as you import the datasetmost
  707. self.ui.plotImportCheckBox.setEnabled(True)
  708. self.ui.plotImportCheckBox.setChecked(True)
  709. # enable the LCDs
  710. self.ui.lcdNumberFID1Length.setEnabled(1)
  711. self.ui.lcdNumberFID2Length.setEnabled(1)
  712. self.ui.lcdNumberResampFreq.setEnabled(1)
  713. self.ui.lcdTotalDeadTime.setEnabled(1)
  714. self.ui.lcdTotalDeadTime.display( 1e3*self.RAWDataProc.DATADICT["INFO"]["deadTime"] )
  715. self.ui.headerFileTextBrowser.clear( )
  716. self.ui.headerFileTextBrowser.append( self.RAWDataProc.DATADICT["INFO"]["headerstr"] )
  717. if u"Pulse 1" in self.RAWDataProc.DATADICT.keys():
  718. self.ui.lcdNumberFID1Length.display(self.RAWDataProc.DATADICT["Pulse 1"]["TIMES"][-1]- self.RAWDataProc.DATADICT["Pulse 1"]["TIMES"][0])
  719. if u"Pulse 2" in self.RAWDataProc.DATADICT.keys():
  720. self.ui.lcdNumberFID1Length.display(self.RAWDataProc.DATADICT["Pulse 2"]["TIMES"][-1]- self.RAWDataProc.DATADICT["Pulse 2"]["TIMES"][0])
  721. # Update info from the header into the GUI
  722. self.ui.pulseTypeTextBrowser.clear()
  723. self.ui.pulseTypeTextBrowser.append(self.RAWDataProc.pulseType)
  724. self.ui.lcdNumberNuTx.display(self.RAWDataProc.transFreq)
  725. self.ui.lcdNumberTauPulse1.display(1e3*self.RAWDataProc.pulseLength[0])
  726. self.ui.lcdNumberTuneuF.display(self.RAWDataProc.TuneCapacitance)
  727. self.ui.lcdNumberResampFreq.display(self.RAWDataProc.samp)
  728. self.ui.lcdNumberSampFreq.display(50000) # TODO, if non GMR is supported, query
  729. self.ui.lcdNumberNQ.display(self.RAWDataProc.nPulseMoments)
  730. self.ui.DeadTimeSpinBox.setValue(1e3*self.RAWDataProc.deadTime)
  731. self.ui.CentralVSpinBox.setValue( self.RAWDataProc.transFreq )
  732. if self.RAWDataProc.pulseType != "FID":
  733. self.ui.lcdNumberTauPulse2.setEnabled(1)
  734. self.ui.lcdNumberTauPulse2.display(1e3*self.RAWDataProc.pulseLength[1])
  735. self.ui.lcdNumberTauDelay.setEnabled(1)
  736. self.ui.lcdNumberTauDelay.display(1e3*self.RAWDataProc.interpulseDelay)
  737. self.ui.FIDProcComboBox.clear()
  738. if self.RAWDataProc.pulseType == "4PhaseT1" or self.RAWDataProc.pulseType == "T1":
  739. self.ui.FIDProcComboBox.insertItem(0, "Pulse 1") #, const QVariant & userData = QVariant() )
  740. self.ui.FIDProcComboBox.insertItem(1, "Pulse 2") #, const QVariant & userData = QVariant() )
  741. self.ui.FIDProcComboBox.insertItem(2, "Both") #, const QVariant & userData = QVariant() )
  742. if len( self.RAWDataProc.DATADICT["PULSES"]) == 2:
  743. self.ui.FIDProcComboBox.setCurrentIndex (2)
  744. elif self.RAWDataProc.DATADICT["PULSES"][0] == "Pulse 1":
  745. self.ui.FIDProcComboBox.setCurrentIndex (0)
  746. else:
  747. self.ui.FIDProcComboBox.setCurrentIndex (1)
  748. elif self.RAWDataProc.pulseType == "FID":
  749. self.ui.FIDProcComboBox.insertItem(0, "Pulse 1") #, const QVariant & userData = QVariant() )
  750. self.ui.FIDProcComboBox.setCurrentIndex (0)
  751. # QtCore.QObject.connect(self.RAWDataProc, QtCore.SIGNAL("updateProgress(int)"), self.updateProgressBar)
  752. # QtCore.QObject.connect(self.RAWDataProc, QtCore.SIGNAL("enableDSP()"), self.enableDSP)
  753. # QtCore.QObject.connect(self.RAWDataProc, QtCore.SIGNAL("doneStatus()"), self.doneStatus)
  754. self.RAWDataProc.progressTrigger.connect(self.updateProgressBar)
  755. self.RAWDataProc.enableDSPTrigger.connect(self.enableDSP)
  756. self.RAWDataProc.doneTrigger.connect(self.doneStatus)
  757. self.enableAll()
  758. def loadRAW(self):
  759. #################################################
  760. # Check to make sure we are ready to process
  761. # Header
  762. if self.RAWDataProc == None:
  763. err_msg = "You need to load a header first."
  764. reply = QtGui.QMessageBox.critical(self, 'Error',
  765. err_msg) #, QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
  766. return
  767. # Stacks
  768. try:
  769. self.procStacks = np.array(eval(str("np.r_["+self.ui.stacksLineEdit.text())+"]"))
  770. except:
  771. err_msg = "You need to set your stacks correctly.\n" + \
  772. "This should be a Python Numpy interpretable list\n" + \
  773. "of stack indices. For example 1:24 or 1:4,8:24"
  774. QtGui.QMessageBox.critical(self, 'Error', err_msg)
  775. return
  776. # Data Channels
  777. #Chan = np.arange(0,9,1)
  778. try:
  779. self.dataChan = np.array(eval(str("np.r_["+self.ui.dataChanLineEdit.text())+"]"))
  780. except:
  781. #QMessageBox messageBox;
  782. #messageBox.critical(0,"Error","An error has occured !");
  783. #messageBox.setFixedSize(500,200);
  784. #quit_msg = "Are you sure you want to exit the program?"
  785. #reply = QtGui.QMessageBox.question(self, 'Message',
  786. # quit_msg, QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
  787. err_msg = "You need to set your data channels correctly.\n" + \
  788. "This should be a Python Numpy interpretable list\n" + \
  789. "of indices. For example 1 or 1:3 or 1:3 5\n\n" + \
  790. "valid GMR data channels fall between 1 and 8. Note that\n" +\
  791. "1:3 is not inclusive of 3 and is the same as 1,2 "
  792. reply = QtGui.QMessageBox.critical(self, 'Error',
  793. err_msg) #, QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
  794. return
  795. #############################
  796. # Reference Channels
  797. # TODO make sure no overlap between data and ref channels
  798. self.refChan = np.array( () )
  799. if str(self.ui.refChanLineEdit.text()): # != "none":
  800. try:
  801. self.refChan = np.array(eval(str("np.r_["+self.ui.refChanLineEdit.text())+"]"))
  802. except:
  803. err_msg = "You need to set your reference channels correctly.\n" + \
  804. "This should be a Python Numpy interpretable list\n" + \
  805. "of indices. For example 1 or 1:3 or 1:3 5\n\n" + \
  806. "valid GMR data channels fall between 1 and 8. Note that\n" +\
  807. "1:3 is not inclusive of 3 and is the same as 1,2 "
  808. QtGui.QMessageBox.critical(self, 'Error', err_msg)
  809. return
  810. #####################################################
  811. # Load data
  812. self.lock("loading RAW GMR dataset")
  813. if self.RAWDataProc.pulseType == "FID":
  814. self.procThread = thread.start_new_thread(self.RAWDataProc.loadFIDData, \
  815. (str(self.headerstr), self.procStacks, self.dataChan, self.refChan, \
  816. str(self.ui.FIDProcComboBox.currentText()), self.ui.mplwidget, \
  817. 1e-3 * self.ui.DeadTimeSpinBox.value( ), self.ui.plotImportCheckBox.isChecked() )) #, self))
  818. elif self.RAWDataProc.pulseType == "4PhaseT1":
  819. self.procThread = thread.start_new_thread(self.RAWDataProc.load4PhaseT1Data, \
  820. (str(self.headerstr), self.procStacks, self.dataChan, self.refChan, \
  821. str(self.ui.FIDProcComboBox.currentText()), self.ui.mplwidget, \
  822. 1e-3 * self.ui.DeadTimeSpinBox.value( ), self.ui.plotImportCheckBox.isChecked() )) #, self))
  823. elif self.RAWDataProc.pulseType == "T1":
  824. self.procThread = thread.start_new_thread(self.RAWDataProc.loadT1Data, \
  825. (str(self.headerstr), self.procStacks, self.dataChan, self.refChan, \
  826. str(self.ui.FIDProcComboBox.currentText()), self.ui.mplwidget, \
  827. 1e-3 * self.ui.DeadTimeSpinBox.value( ), self.ui.plotImportCheckBox.isChecked() )) #, self))
  828. #self.procThread = thread.start_new_thread(self.RAWDataProc.load4PhaseT1Data, \
  829. # (str(self.headerstr), self.procStacks, self.dataChan, self.refChan, \
  830. # str(self.ui.FIDProcComboBox.currentText()), self.ui.mplwidget, \
  831. # 1e-3 * self.ui.DeadTimeSpinBox.value( ), self.ui.plotImportCheckBox.isChecked() )) #, self))
  832. self.YamlNode.Import["GMR Header"] = self.headerstr
  833. self.YamlNode.Import["opened"] = datetime.datetime.now().isoformat()
  834. self.YamlNode.Import["pulse Type"] = str(self.RAWDataProc.pulseType)
  835. self.YamlNode.Import["stacks"] = self.procStacks.tolist()
  836. self.YamlNode.Import["data channels"] = self.dataChan.tolist()
  837. self.YamlNode.Import["reference channels"] = self.refChan.tolist()
  838. self.YamlNode.Import["pulse records"] = str(self.ui.FIDProcComboBox.currentText())
  839. self.YamlNode.Import["instrument dead time"] = (1e-3 * self.ui.DeadTimeSpinBox.value( ))
  840. self.Log ( )
  841. # should be already done
  842. # QtCore.QObject.connect(self.RAWDataProc, QtCore.SIGNAL("updateProgress(int)"), self.updateProgressBar)
  843. # QtCore.QObject.connect(self.RAWDataProc, QtCore.SIGNAL("enableDSP()"), self.enableDSP)
  844. # QtCore.QObject.connect(self.RAWDataProc, QtCore.SIGNAL("doneStatus()"), self.doneStatus)
  845. #self.ui.ProcessedBox.setEnabled(True)
  846. self.ui.lcdNumberFID1Length.setEnabled(1)
  847. self.ui.lcdNumberFID2Length.setEnabled(1)
  848. self.ui.lcdNumberResampFreq.setEnabled(1)
  849. self.ui.lcdTotalDeadTime.setEnabled(1)
  850. self.ui.lcdTotalDeadTime.display( self.ui.DeadTimeSpinBox.value( ) )
  851. #self.ui.lcdNumberFID1Length.display(0)
  852. #self.ui.lcdNumberFID2Length.display(0)
  853. #self.ui.lcdNumberResampFreq.display( self.RAWDataProc.samp )
  854. self.mpl_toolbar = NavigationToolbar2QT(self.ui.mplwidget, self.ui.mplwidget)
  855. self.ui.mplwidget.draw()
  856. def Log(self):
  857. #for line in yaml.dump(self.YamlNode, default_flow_style=False):
  858. #for line in nlogText:
  859. # self.ui.logTextBrowser.append( line )
  860. # self.logText.append( line )
  861. self.ui.logTextBrowser.clear()
  862. self.ui.logTextBrowser.append( yaml.dump(self.YamlNode)) #, default_flow_style=False) )
  863. def disable(self):
  864. self.ui.inputRAWParametersBox.setEnabled(False)
  865. self.ui.BandPassBox.setEnabled(False)
  866. self.ui.downSampleGroupBox.setEnabled(False)
  867. self.ui.windowFilterGroupBox.setEnabled(False)
  868. # self.ui.despikeGroupBox.setEnabled(False)
  869. self.ui.adaptBox.setEnabled(False)
  870. self.ui.adaptFDBox.setEnabled(False)
  871. self.ui.qCalcGroupBox.setEnabled(False)
  872. self.ui.FDSmartStackGroupBox.setEnabled(False)
  873. self.ui.sumDataBox.setEnabled(False)
  874. self.ui.qdGroupBox.setEnabled(False)
  875. self.ui.gateBox.setEnabled(False)
  876. def enableAll(self):
  877. self.enableDSP()
  878. self.enableQC()
  879. def enableDSP(self):
  880. # Bandpass filter
  881. self.ui.BandPassBox.setEnabled(True)
  882. self.ui.BandPassBox.setChecked(True)
  883. self.ui.bandPassGO.setEnabled(False) # need to design first
  884. self.ui.plotBP.setEnabled(True)
  885. self.ui.plotBP.setChecked(True)
  886. # downsample
  887. self.ui.downSampleGroupBox.setEnabled(True)
  888. self.ui.downSampleGroupBox.setChecked(True)
  889. # window
  890. self.ui.windowFilterGroupBox.setEnabled(True)
  891. self.ui.windowFilterGroupBox.setChecked(True)
  892. # Despike
  893. # self.ui.despikeGroupBox.setEnabled(True)
  894. # self.ui.despikeGroupBox.setChecked(False)
  895. # Adaptive filtering
  896. self.ui.adaptBox.setEnabled(True)
  897. self.ui.adaptBox.setChecked(True)
  898. # FD Adaptive filtering
  899. self.ui.adaptFDBox.setEnabled(True)
  900. self.ui.adaptFDBox.setChecked(False)
  901. # Harmonic
  902. self.ui.harmonicBox.setEnabled(True)
  903. self.ui.harmonicBox.setChecked(True)
  904. # sum group box
  905. try:
  906. if len(self.dataChan) > 1:
  907. self.ui.sumDataBox.setEnabled(True)
  908. self.ui.sumDataBox.setChecked(True)
  909. except:
  910. pass
  911. # Quadrature Detect
  912. self.ui.qdGroupBox.setEnabled(True)
  913. self.ui.qdGroupBox.setChecked(True)
  914. self.enableQC()
  915. def enableQC(self):
  916. # Q calc
  917. self.ui.qCalcGroupBox.setEnabled(True)
  918. self.ui.qCalcGroupBox.setChecked(True)
  919. # FD SmartStack
  920. self.ui.FDSmartStackGroupBox.setEnabled(True)
  921. self.ui.FDSmartStackGroupBox.setChecked(True)
  922. # Quadrature detect
  923. try:
  924. for pulse in self.RAWDataProc.DATADICT["PULSES"]:
  925. np.shape(self.RAWDataProc.DATADICT[pulse]["Q"])
  926. self.RAWDataProc.DATADICT["stack"]
  927. self.ui.qdGroupBox.setEnabled(True)
  928. self.ui.qdGroupBox.setChecked(True)
  929. except:
  930. self.ui.qdGroupBox.setEnabled(False)
  931. self.ui.qdGroupBox.setChecked(False)
  932. # Gating
  933. try:
  934. self.RAWDataProc.DATADICT["CA"]
  935. self.ui.gateBox.setEnabled(True)
  936. self.ui.gateBox.setChecked(True)
  937. except:
  938. self.ui.gateBox.setEnabled(False)
  939. self.ui.gateBox.setChecked(False)
  940. def despikeFilter(self):
  941. self.lock("despike filter")
  942. thread.start_new_thread(self.RAWDataProc.despike, \
  943. (self.ui.windowSpinBox.value(), \
  944. self.ui.thresholdSpinBox.value(), \
  945. str(self.ui.replComboBox.currentText()), \
  946. self.ui.rollOnSpinBox.value(), \
  947. self.ui.despikeInterpWinSpinBox.value(),
  948. self.ui.mplwidget))
  949. def calcQ(self):
  950. if "Calc Q" not in self.YamlNode.Processing.keys():
  951. #print("In CalcQ", yaml.dump(self.YamlNode.Processing) )
  952. self.YamlNode.Processing["Calc Q"] = True
  953. #print( yaml.dump(self.YamlNode.Processing) )
  954. self.Log()
  955. else:
  956. err_msg = "Q values have already been calculated"
  957. reply =QtWidgets.QMessageBox.critical(self, 'Error',
  958. err_msg)
  959. return
  960. self.lock("pulse moment calculation")
  961. thread.start_new_thread(self.RAWDataProc.effectivePulseMoment, \
  962. (self.ui.CentralVSpinBox.value(), \
  963. self.ui.mplwidget))
  964. def harmonicModel(self):
  965. if "Harmonic modelling" not in self.YamlNode.Processing.keys():
  966. self.YamlNode.Processing["Harmonic modelling"] = {}
  967. self.YamlNode.Processing["Harmonic modelling"]["NF"] = str( self.ui.NHarmonicsFreqsSpin.value() )
  968. self.YamlNode.Processing["Harmonic modelling"]["Segments"] = str( self.ui.NSegments.value() )
  969. self.YamlNode.Processing["Harmonic modelling"]["f0K1"] = str( self.ui.f0K1Spin.value() )
  970. self.YamlNode.Processing["Harmonic modelling"]["f0KN"] = str( self.ui.f0KNSpin.value() )
  971. self.YamlNode.Processing["Harmonic modelling"]["f0Ks"] = str( self.ui.f0KsSpin.value() )
  972. self.YamlNode.Processing["Harmonic modelling"]["f0"] = str( self.ui.f0Spin.value() )
  973. if self.ui.NHarmonicsFreqsSpin.value() > 1:
  974. self.YamlNode.Processing["Harmonic modelling"]["f1K1"] = str( self.ui.f1K1Spin.value() )
  975. self.YamlNode.Processing["Harmonic modelling"]["f1KN"] = str( self.ui.f1KNSpin.value() )
  976. self.YamlNode.Processing["Harmonic modelling"]["f1Ks"] = str( self.ui.f1KsSpin.value() )
  977. self.YamlNode.Processing["Harmonic modelling"]["f1"] = str( self.ui.f1Spin.value() )
  978. self.Log()
  979. else:
  980. err_msg = "Harmonic modelling noise cancellation has already been applied!"
  981. reply = QtWidgets.QMessageBox.critical(self, 'Error',
  982. err_msg)
  983. return
  984. self.lock("harmonic noise modelling")
  985. thread.start_new_thread(self.RAWDataProc.harmonicModel, \
  986. ( \
  987. self.ui.NHarmonicsFreqsSpin.value(), \
  988. self.ui.f0Spin.value(), \
  989. self.ui.f0K1Spin.value(), \
  990. self.ui.f0KNSpin.value(), \
  991. self.ui.f0KsSpin.value(), \
  992. self.ui.NSegments.value(), \
  993. self.ui.f1Spin.value(), \
  994. self.ui.f1K1Spin.value(), \
  995. self.ui.f1KNSpin.value(), \
  996. self.ui.f1KsSpin.value(), \
  997. self.ui.plotHarmonic.isChecked(), \
  998. self.ui.mplwidget \
  999. ) \
  1000. )
  1001. def FDSmartStack(self):
  1002. if "TD stack" not in self.YamlNode.Processing.keys():
  1003. self.YamlNode.Processing["TD stack"] = {}
  1004. self.YamlNode.Processing["TD stack"]["outlier"] = str( self.ui.outlierTestCB.currentText() )
  1005. self.YamlNode.Processing["TD stack"]["cutoff"] = str( self.ui.MADCutoff.value() )
  1006. self.Log()
  1007. else:
  1008. err_msg = "TD noise cancellation has already been applied!"
  1009. reply =QtWidgets.QMessageBox.critical(self, 'Error',
  1010. err_msg)
  1011. return
  1012. self.lock("time-domain smart stack")
  1013. thread.start_new_thread(self.RAWDataProc.TDSmartStack, \
  1014. (str(self.ui.outlierTestCB.currentText()), \
  1015. self.ui.MADCutoff.value(),
  1016. self.ui.mplwidget))
  1017. def adaptFilter(self):
  1018. if "TD noise cancellation" not in self.YamlNode.Processing.keys():
  1019. self.YamlNode.Processing["TD noise cancellation"] = {}
  1020. self.YamlNode.Processing["TD noise cancellation"]["n_Taps"] = str(self.ui.MTapsSpinBox.value())
  1021. self.YamlNode.Processing["TD noise cancellation"]["lambda"] = str(self.ui.adaptLambdaSpinBox.value())
  1022. self.YamlNode.Processing["TD noise cancellation"]["truncate"] = str(self.ui.adaptTruncateSpinBox.value())
  1023. self.YamlNode.Processing["TD noise cancellation"]["mu"] = str(self.ui.adaptMuSpinBox.value())
  1024. self.YamlNode.Processing["TD noise cancellation"]["PCA"] = str(self.ui.PCAComboBox.currentText())
  1025. self.Log()
  1026. else:
  1027. err_msg = "TD noise cancellation has already been applied!"
  1028. reply =QtWidgets.QMessageBox.critical(self, 'Error', err_msg)
  1029. #return
  1030. self.lock("TD noise cancellation filter")
  1031. thread.start_new_thread(self.RAWDataProc.adaptiveFilter, \
  1032. (self.ui.MTapsSpinBox.value(), \
  1033. self.ui.adaptLambdaSpinBox.value(), \
  1034. self.ui.adaptTruncateSpinBox.value(), \
  1035. self.ui.adaptMuSpinBox.value(), \
  1036. str(self.ui.PCAComboBox.currentText()), \
  1037. self.ui.mplwidget))
  1038. def sumDataChans(self):
  1039. if "Data sum" not in self.YamlNode.Processing.keys():
  1040. self.YamlNode.Processing["Data sum"] = True
  1041. self.Log()
  1042. else:
  1043. err_msg = "Data channels have already been summed!"
  1044. reply =QtWidgets.QMessageBox.critical(self, 'Error',
  1045. err_msg)
  1046. return
  1047. self.lock("Summing data channels")
  1048. self.dataChan = [self.dataChan[0]]
  1049. self.ui.sumDataBox.setEnabled(False)
  1050. thread.start_new_thread( self.RAWDataProc.sumData, ( self.ui.mplwidget, 7 ) )
  1051. def adaptFilterFD(self):
  1052. self.lock("FD noise cancellation filter")
  1053. thread.start_new_thread(self.RAWDataProc.adaptiveFilterFD, \
  1054. (str(self.ui.windowTypeComboBox.currentText()), \
  1055. self.ui.windowBandwidthSpinBox.value(), \
  1056. self.ui.CentralVSpinBox.value(), \
  1057. self.ui.mplwidget))
  1058. def bandPassFilter(self):
  1059. if "Bandpass filter" not in self.YamlNode.Processing.keys():
  1060. self.YamlNode.Processing["Bandpass filter"] = {}
  1061. self.YamlNode.Processing["Bandpass filter"]["central_nu"] = str(self.ui.CentralVSpinBox.value())
  1062. self.YamlNode.Processing["Bandpass filter"]["passband"] = str(self.ui.passBandSpinBox.value())
  1063. self.YamlNode.Processing["Bandpass filter"]["stopband"] = str(self.ui.stopBandSpinBox.value())
  1064. self.YamlNode.Processing["Bandpass filter"]["gpass"] = str(self.ui.gpassSpinBox.value())
  1065. self.YamlNode.Processing["Bandpass filter"]["gstop"] = str(self.ui.gstopSpinBox.value())
  1066. self.YamlNode.Processing["Bandpass filter"]["type"] = str(self.ui.fTypeComboBox.currentText())
  1067. self.Log()
  1068. else:
  1069. err_msg = "Bandpass filter has already been applied!"
  1070. reply =QtWidgets.QMessageBox.critical(self, 'Error',
  1071. err_msg)
  1072. return
  1073. self.lock("bandpass filter")
  1074. nv = self.ui.lcdTotalDeadTime.value( ) + self.ui.lcdNumberFTauDead.value()
  1075. self.ui.lcdTotalDeadTime.display( nv )
  1076. thread.start_new_thread(self.RAWDataProc.bandpassFilter, \
  1077. (self.ui.mplwidget, 0, self.ui.plotBP.isChecked() ))
  1078. def downsample(self):
  1079. self.lock("resampling")
  1080. if "Resample" not in self.YamlNode.Processing.keys():
  1081. self.YamlNode.Processing["Resample"] = {}
  1082. self.YamlNode.Processing["Resample"]["downsample factor"] = []
  1083. self.YamlNode.Processing["Resample"]["truncate length"] = []
  1084. self.YamlNode.Processing["Resample"]["downsample factor"].append( str(self.ui.downSampleSpinBox.value() ) )
  1085. self.YamlNode.Processing["Resample"]["truncate length"].append( str( self.ui.truncateSpinBox.value() ) )
  1086. self.Log( )
  1087. thread.start_new_thread(self.RAWDataProc.downsample, \
  1088. (self.ui.truncateSpinBox.value(), \
  1089. self.ui.downSampleSpinBox.value(), \
  1090. self.ui.dsPlot.isChecked(), \
  1091. self.ui.mplwidget))
  1092. def quadDet(self):
  1093. method = ['trf','dogbox','lm'][int(self.ui.QDMethod.currentIndex())]
  1094. loss = ['linear','soft_l1','cauchy','huber'][int(self.ui.QDLoss.currentIndex())]
  1095. # allow overwrite of Quad Det.
  1096. self.YamlNode.Processing["Quadrature detection"] = {}
  1097. self.YamlNode.Processing["Quadrature detection"]["trim"] = str( self.ui.trimSpin.value() )
  1098. self.YamlNode.Processing["Quadrature detection"]["method"] = method
  1099. self.YamlNode.Processing["Quadrature detection"]["loss"] = loss
  1100. self.Log()
  1101. #if "Quadrature detection" not in self.YamlNode.Processing.keys():
  1102. # self.YamlNode.Processing["Quadrature detection"] = {}
  1103. # self.YamlNode.Processing["Quadrature detection"]["trim"] = str( self.ui.trimSpin.value() )
  1104. # self.Log()
  1105. #else:
  1106. # self.YamlNode.Processing["Quadrature detection"] = {}
  1107. # self.YamlNode.Processing["Quadrature detection"]["trim"] = str( self.ui.trimSpin.value() )
  1108. # self.Log()
  1109. #err_msg = "Quadrature detection has already been done!"
  1110. #reply =QtWidgets.QMessageBox.critical(self, 'Error',
  1111. # err_msg)
  1112. #return
  1113. self.lock("quadrature detection")
  1114. thread.start_new_thread(self.RAWDataProc.quadDet, \
  1115. (self.ui.trimSpin.value(), method, loss, self.ui.mplwidget))
  1116. self.ui.plotQD.setEnabled(True)
  1117. def plotQD(self):
  1118. self.lock("plot QD")
  1119. thread.start_new_thread(self.RAWDataProc.plotQuadDet, \
  1120. (self.ui.trimSpin.value(), int(self.ui.QDType.currentIndex()), self.ui.mplwidget))
  1121. def gateIntegrate(self):
  1122. if "Gate integrate" not in self.YamlNode.Processing.keys():
  1123. self.YamlNode.Processing["Gate integrate"] = {}
  1124. self.YamlNode.Processing["Gate integrate"]["gpd"] = str(self.ui.GPDspinBox.value( ) )
  1125. self.Log()
  1126. self.lock("gate integration")
  1127. thread.start_new_thread(self.RAWDataProc.gateIntegrate, \
  1128. (self.ui.GPDspinBox.value(), self.ui.trimSpin.value(), self.ui.mplwidget))
  1129. self.ui.actionExport_Preprocessed_Dataset.setEnabled(True)
  1130. self.ui.plotGI.setEnabled(True)
  1131. def plotGI(self):
  1132. self.lock("plot gate integrate")
  1133. thread.start_new_thread(self.RAWDataProc.plotGateIntegrate, \
  1134. (self.ui.GPDspinBox.value(), self.ui.trimSpin.value(), \
  1135. self.ui.QDType_2.currentIndex(), self.ui.mplwidget))
  1136. def designFilter(self):
  1137. [bord, fe] = self.RAWDataProc.designFilter( \
  1138. self.ui.CentralVSpinBox.value(), \
  1139. self.ui.passBandSpinBox.value(), \
  1140. self.ui.stopBandSpinBox.value(), \
  1141. self.ui.gpassSpinBox.value(), \
  1142. self.ui.gstopSpinBox.value(), \
  1143. str(self.ui.fTypeComboBox.currentText()),
  1144. self.ui.mplwidget
  1145. )
  1146. self.ui.lcdNumberFilterOrder.display(bord)
  1147. self.ui.lcdNumberFTauDead.display(1e3*fe)
  1148. self.ui.bandPassGO.setEnabled(1)
  1149. # Hack for MacOS to force refresh of group box and plot
  1150. self.ui.mplwidget.hide()
  1151. self.ui.mplwidget.show()
  1152. self.ui.BandPassBox.hide()
  1153. self.ui.BandPassBox.show()
  1154. def windowFilter(self):
  1155. if "Window filter" not in self.YamlNode.Processing.keys():
  1156. self.YamlNode.Processing["Window filter"] = {}
  1157. self.YamlNode.Processing["Window filter"]["type"] = str(self.ui.windowTypeComboBox.currentText())
  1158. self.YamlNode.Processing["Window filter"]["width"] = str(self.ui.windowBandwidthSpinBox.value())
  1159. self.YamlNode.Processing["Window filter"]["centre"] = str(self.ui.CentralVSpinBox.value() )
  1160. self.Log()
  1161. else:
  1162. err_msg = "FD window has already been applied!"
  1163. reply =QtWidgets.QMessageBox.critical(self, 'Error',
  1164. err_msg)
  1165. return
  1166. self.lock("window filter")
  1167. thread.start_new_thread(self.RAWDataProc.windowFilter, \
  1168. (str(self.ui.windowTypeComboBox.currentText()), \
  1169. self.ui.windowBandwidthSpinBox.value(), \
  1170. self.ui.CentralVSpinBox.value(), \
  1171. self.ui.mplwidget))
  1172. def designFDFilter(self):
  1173. # thread.start_new_thread(self.RAWDataProc.computeWindow, ( \
  1174. # "Pulse 1",
  1175. # self.ui.windowBandwidthSpinBox.value(), \
  1176. # self.ui.CentralVSpinBox.value(), \
  1177. # str(self.ui.windowTypeComboBox.currentText()), \
  1178. # self.ui.mplwidget ))
  1179. mPulse = "None"
  1180. if u"Pulse 1" in self.RAWDataProc.DATADICT.keys():
  1181. mPulse = u"Pulse 1"
  1182. elif u"Pulse 2" in self.RAWDataProc.DATADICT.keys():
  1183. mPulse = u"Pulse 2"
  1184. a,b,c,d,dead = self.RAWDataProc.computeWindow( \
  1185. mPulse,
  1186. self.ui.windowBandwidthSpinBox.value(), \
  1187. self.ui.CentralVSpinBox.value(), \
  1188. str(self.ui.windowTypeComboBox.currentText()), \
  1189. self.ui.mplwidget )
  1190. self.ui.lcdWinDead.display(dead)
  1191. # Hack for MacOS to force refresh of group box and plot
  1192. self.ui.mplwidget.hide()
  1193. self.ui.mplwidget.show()
  1194. self.ui.windowFilterGroupBox.hide()
  1195. self.ui.windowFilterGroupBox.show()
  1196. def updateProgressBar(self, percent):
  1197. self.ui.barProgress.setValue(percent)
  1198. def updateProc(self):
  1199. if str(self.ui.FIDProcComboBox.currentText()) == "Pulse 1":
  1200. self.ui.lcdNumberFID1Length.display(self.RAWDataProc.DATADICT["Pulse 1"]["TIMES"][-1]- self.RAWDataProc.DATADICT["Pulse 1"]["TIMES"][0])
  1201. elif str(self.ui.FIDProcComboBox.currentText()) == "Pulse 2":
  1202. self.ui.lcdNumberFID2Length.display(self.RAWDataProc.DATADICT["Pulse 2"]["TIMES"][-1]- self.RAWDataProc.DATADICT["Pulse 2"]["TIMES"][0])
  1203. else:
  1204. self.ui.lcdNumberFID1Length.display(self.RAWDataProc.DATADICT["Pulse 1"]["TIMES"][-1]- self.RAWDataProc.DATADICT["Pulse 1"]["TIMES"][0])
  1205. self.ui.lcdNumberFID2Length.display(self.RAWDataProc.DATADICT["Pulse 2"]["TIMES"][-1]- self.RAWDataProc.DATADICT["Pulse 2"]["TIMES"][0])
  1206. self.ui.lcdNumberResampFreq.display( self.RAWDataProc.samp )
  1207. def doneStatus(self): # unlocks GUI
  1208. self.ui.statusbar.clearMessage ( )
  1209. self.ui.barProgress.hide()
  1210. self.updateProc()
  1211. self.enableAll()
  1212. def lock(self, string):
  1213. self.ui.statusbar.showMessage ( string )
  1214. self.ui.barProgress.show()
  1215. self.ui.barProgress.setValue(0)
  1216. self.disable()
  1217. def unlock(self):
  1218. self.ui.statusbar.clearMessage ( )
  1219. self.ui.barProgress.hide()
  1220. self.enableAll()
  1221. def done(self):
  1222. self.ui.statusbar.showMessage ( "" )
  1223. ################################################################
  1224. ################################################################
  1225. # Boiler plate main function
  1226. import pkg_resources
  1227. from pkg_resources import resource_string
  1228. import matplotlib.image as mpimg
  1229. import matplotlib.pyplot as plt
  1230. def main():
  1231. # splash screen logo
  1232. logo = pkg_resources.resource_filename(__name__, 'akvo.png')
  1233. logo2 = pkg_resources.resource_filename(__name__, 'akvo2.png')
  1234. qApp = QtWidgets.QApplication(sys.argv)
  1235. ssplash = True
  1236. if ssplash:
  1237. pixmap = QtGui.QPixmap(logo)
  1238. splash = QtWidgets.QSplashScreen(pixmap, QtCore.Qt.WindowStaysOnTopHint)
  1239. splash.show()
  1240. aw = ApplicationWindow()
  1241. img=mpimg.imread(logo)
  1242. for ax in [ aw.ui.mplwidget ]:
  1243. ax.fig.clear()
  1244. subplot = ax.fig.add_subplot(111)
  1245. #ax.fig.patch.set_facecolor( None )
  1246. #ax.fig.patch.set_alpha( .0 )
  1247. subplot.imshow(img)
  1248. subplot.xaxis.set_major_locator(plt.NullLocator())
  1249. subplot.yaxis.set_major_locator(plt.NullLocator())
  1250. ax.draw()
  1251. if ssplash:
  1252. splash.showMessage("Loading modules")
  1253. splash.finish(aw)
  1254. #time.sleep(1)
  1255. aw.setWindowTitle("Akvo v"+str(version))
  1256. aw.show()
  1257. qApp.setWindowIcon(QtGui.QIcon(logo2))
  1258. sys.exit(qApp.exec_())
  1259. if __name__ == "__main__":
  1260. main()