def addEnvelope(self, etype="Coulomb", name="Env"):
item = pg.TreeWidgetItem([name])
self.envelopes.addChild(item)
self.eNames.append(name)
self.eTypes[name] = etype
typeLabel = QtGui.QLabel(etype)
item.setWidget(2, typeLabel)
colorButton = ColorButton()
self.eCButtons[name] = colorButton
color = get_color()
colorButton.setColor(color)
item.setExpanded(True)
self.eBoxes[name] = {}
colorItem = pg.TreeWidgetItem(["Color"])
if etype == "Coulomb":
item1 = pg.TreeWidgetItem(["Friction Angle"])
item2 = pg.TreeWidgetItem(["Cohesion"])
step1 = 1
step2 = 50
elif etype == "Brown":
item1 = pg.TreeWidgetItem(["m"])
item2 = pg.TreeWidgetItem(["UCS"])
step1 = 1
step2 = 1
else:
print etype
return 0
item.addChild(colorItem)
item.addChild(item1)
item.addChild(item2)
frictionBox = pg.SpinBox(value=50, step=step1)
cohesionBox = pg.SpinBox(value=1e3, step=step2)
frictionBox.sigValueChanged.connect(self.plot)
cohesionBox.sigValueChanged.connect(self.plot)
colorItem.setWidget(2, colorButton)
item1.setWidget(2, frictionBox)
item2.setWidget(2, cohesionBox)
self.fBoxes[name] = frictionBox
self.cBoxes[name] = cohesionBox
for dname in self.dNames:
child = pg.TreeWidgetItem([dname])
item.addChild(child)
box = CheckBox()
child.setWidget(2, box)
self.eBoxes[name][dname] = box
box.click()
box.clicked.connect(lambda: self.getEnvelope(name))
removeEnvelopeItem = pg.TreeWidgetItem([""])
item.addChild(removeEnvelopeItem)
removeButton = QtGui.QPushButton("Remove")
removeEnvelopeItem.setWidget(2, removeButton)
removeButton.clicked.connect(lambda: self.removeEnvelope(item))
colorButton.sigColorChanged.connect(self.plot)
self.nEnvelopes += 1
self.getEnvelope(eName=name)
def addEnvelope(self,etype='Coulomb',name='Env'):
item = pg.TreeWidgetItem([name])
self.envelopes.addChild(item)
self.eNames.append(name)
self.eTypes[name] = etype
typeLabel = QtGui.QLabel(etype)
item.setWidget(2,typeLabel)
colorButton = ColorButton()
self.eCButtons[name] = colorButton
color = get_color()
colorButton.setColor(color)
item.setExpanded(True)
self.eBoxes[name] = {}
colorItem = pg.TreeWidgetItem(['Color'])
if etype == 'Coulomb':
item1 = pg.TreeWidgetItem(['Friction Angle'])
item2 = pg.TreeWidgetItem(['Cohesion'])
step1 = 1
step2 = 50
elif etype == 'Brown':
item1 = pg.TreeWidgetItem(['m'])
item2 = pg.TreeWidgetItem(['UCS'])
step1 = 1
step2 = 1
else:
print etype
return 0
item.addChild(colorItem)
item.addChild(item1)
item.addChild(item2)
frictionBox = pg.SpinBox(value=50, step=step1)
cohesionBox = pg.SpinBox(value=1e3, step=step2)
frictionBox.sigValueChanged.connect(self.plot)
cohesionBox.sigValueChanged.connect(self.plot)
colorItem.setWidget(2,colorButton)
item1.setWidget(2,frictionBox)
item2.setWidget(2,cohesionBox)
self.fBoxes[name] = frictionBox
self.cBoxes[name] = cohesionBox
for dname in self.dNames:
child = pg.TreeWidgetItem([dname])
item.addChild(child)
box = CheckBox()
child.setWidget(2,box)
self.eBoxes[name][dname] = box
box.click()
box.clicked.connect(lambda:self.getEnvelope(name))
removeEnvelopeItem = pg.TreeWidgetItem([''])
item.addChild(removeEnvelopeItem)
removeButton = QtGui.QPushButton('Remove')
removeEnvelopeItem.setWidget(2,removeButton)
removeButton.clicked.connect(lambda:self.removeEnvelope(item))
colorButton.sigColorChanged.connect(self.plot)
self.nEnvelopes += 1
self.getEnvelope(eName=name)
def addData(self, s1, s3, name=None):
if name is None:
name = "Untitled_%d" % (self.nData)
self.s1.append(s1)
self.s3.append(s3)
item = pg.TreeWidgetItem([name])
self.dNames.append(name)
self.fpoints.addChild(item)
# self.tree.addTopLevelItem(item)
color = (0, 0, 0)
colorButton = ColorButton()
item.setWidget(2, colorButton)
self.dCButtons[name] = colorButton
colorButton.setColor(color)
colorButton.sigColorChanged.connect(self.plot)
self.nData += 1
def addData(self,s1,s3,name=None):
if name is None:
name = 'Untitled_%d'%(self.nData)
self.s1.append(s1)
self.s3.append(s3)
item = pg.TreeWidgetItem([name])
self.dNames.append(name)
self.fpoints.addChild(item)
# self.tree.addTopLevelItem(item)
color = (0,0,0)
colorButton = ColorButton()
item.setWidget(2,colorButton)
self.dCButtons[name] = colorButton
colorButton.setColor(color)
colorButton.sigColorChanged.connect(self.plot)
self.nData += 1
def addItem(self, name, xdata, ydata):
item = pg.TreeWidgetItem([name])
self.names.append(name)
self.items[name] = item
self.tree.addTopLevelItem(item)
# self.insertTopLevelItem(self.nitems,item)
color = (rand() * 230, rand() * 230, rand() * 230)
colorButton = ColorButton()
self.colorButtons[name] = colorButton
colorButton.setColor(color)
removeButton = QtGui.QPushButton("Remove", self)
self.removeButtons[name] = removeButton
item.setWidget(1, colorButton)
item.setWidget(2, removeButton)
removeButton.clicked.connect(self.removeItem)
self.xData[name] = xdata
self.yData[name] = ydata
def addItem(self,name,xdata,ydata):
item = pg.TreeWidgetItem([name])
self.names.append(name)
self.items[name] = item
self.tree.addTopLevelItem(item)
# self.insertTopLevelItem(self.nitems,item)
color = (rand()*230,rand()*230,rand()*230)
colorButton = ColorButton()
self.colorButtons[name] = colorButton
colorButton.setColor(color)
removeButton = QtGui.QPushButton("Remove", self)
self.removeButtons[name] = removeButton
item.setWidget(1,colorButton)
item.setWidget(2,removeButton)
removeButton.clicked.connect(self.removeItem)
self.xData[name] = xdata
self.yData[name] = ydata
