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