UI/loadProtocol.py at master · Nick3869/UI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
# -*- coding: utf-8 -*-

"""

@author: Nicolas Fanjat
 Scientific Computing and Imaging Institute
 University of Utah
 02/17/2015

"""
from PyQt5.QtWidgets import QFileDialog, QWidget, QPushButton, QLabel, QLineEdit, QGridLayout, QRadioButton
from PyQt5.QtCore import QObject, QDir, Qt

from lxml import etree

class MainWidget(QWidget):
	def __init__(self, parent):

		super(MainWidget, self).__init__(parent)
		self.initWidget()

	def initWidget(self):

		self.browseb = QPushButton('Browse')
		self.browseinput = QPushButton('Browse')
		self.browseb.clicked.connect(self.browse)
		self.browseinput.clicked.connect(self.browseInput)

		self.inEdit = QLineEdit()
		self.proEdit = QLineEdit()

		self.source = QLabel('Input')
		self.protocol = QLabel('Protocol')

		grid1 = QGridLayout()

		grid1.setVerticalSpacing(20)

		grid1.setRowStretch(3,1)
		grid1.setRowStretch(0,1)

		grid1.addWidget(self.source,1,0)
		grid1.addWidget(self.inEdit,1,1)
		grid1.addWidget(self.browseinput,1,2)

		grid1.addWidget(self.protocol,2,0)
		grid1.addWidget(self.proEdit,2,1)
		grid1.addWidget(self.browseb,2,2)

		self.setLayout(grid1)

	def browse(self):

		fileName, _ = QFileDialog.getOpenFileName(self, "Open File",".","*.xml")

		if fileName:
			self.proEdit.setText(fileName)

	def browseInput(self):

		fileName, _ = QFileDialog.getOpenFileName(self, "Choose File",".","*.nrrd")

		if fileName:
			self.inEdit.setText(fileName)