Changeset 3709:43c54e411713 in orange


Ignore:
Timestamp:
05/28/07 11:54:50 (7 years ago)
Author:
miha <miha.stajdohar@…>
Branch:
default
Convert:
ada7affd6d24057ecc2ee54b3cb86efd15017394
Message:

improved select hubs

File:
1 edited

Legend:

Unmodified
Added
Removed
  • orange/OrangeWidgets/Prototypes/OWGraphDrawerCanvas.py

    r3706 r3709  
    4141    def addSelection(self, ndx): 
    4242        #print("add selection") 
    43         if not ndx in self.selection: 
    44             (key, neighbours) = self.vertices[ndx] 
    45             self.selectionStyles[ndx] = self.curve(key).symbol().brush().color().name() 
    46             newSymbol = QwtSymbol(QwtSymbol.Ellipse, QBrush(QColor(self.selectionStyles[ndx])), QPen(Qt.yellow, 3), QSize(10, 10)) 
    47             self.setCurveSymbol(key, newSymbol) 
    48             self.selection.append(ndx); 
    49             #self.visualizer.filter[ndx] = True 
    50             #self.replot() 
    51             return True 
     43        if isinstance(ndx, list): 
     44            change = False 
     45            for v in ndx: 
     46                if not v in self.selection: 
     47                    (key, neighbours) = self.vertices[v] 
     48                    self.selectionStyles[v] = self.curve(key).symbol().brush().color().name() 
     49                    newSymbol = QwtSymbol(QwtSymbol.Ellipse, QBrush(QColor(self.selectionStyles[v])), QPen(Qt.yellow, 3), QSize(10, 10)) 
     50                    self.setCurveSymbol(key, newSymbol) 
     51                    self.selection.append(v); 
     52                    change = True 
     53            if change: 
     54                self.replot() 
     55                return True 
     56        else: 
     57            if not ndx in self.selection: 
     58                (key, neighbours) = self.vertices[ndx] 
     59                self.selectionStyles[ndx] = self.curve(key).symbol().brush().color().name() 
     60                newSymbol = QwtSymbol(QwtSymbol.Ellipse, QBrush(QColor(self.selectionStyles[ndx])), QPen(Qt.yellow, 3), QSize(10, 10)) 
     61                self.setCurveSymbol(key, newSymbol) 
     62                self.selection.append(ndx); 
     63                #self.visualizer.filter[ndx] = True 
     64                self.replot() 
     65                return True 
    5266         
    5367        return False 
     
    91105         
    92106    def selectHubs(self, no): 
    93         print "start selecting..." 
    94         if self.vertexDegree == []: 
    95             self.generateVertexPower() 
    96         print "generated generateVertexPower" 
    97         count = 0 
    98         old_power = -1 
    99         next_power = -1      
    100         #print "no: " + str(no) 
    101         while count < no or old_power == next_power: 
    102             (v,old_power) = self.vertexDegree[count] 
    103              
    104             next_power = -1 
    105             if count < (len(self.vertexDegree) - 1): 
    106                 (next_v, next_power) = self.vertexDegree[count + 1] 
    107                  
    108             self.addSelection(v) 
    109             #print "old_power: " + str(old_power) + " new_power: " + str(next_power) 
    110             count += 1 
    111         print "selected." 
     107        #print "get hubs..." 
     108        hubs = self.visualizer.getHubs(no) 
     109        #print hubs 
     110        #print "start selecting..." 
     111        self.addSelection(hubs) 
     112        #print "done." 
    112113             
    113114    def selectConnectedNodes(self, distance): 
     
    362363            y2 = self.visualizer.coors[j][1] 
    363364 
    364             key = self.addCurve(str(e), fillColor, edgeColor, 0, style = QwtCurve.Lines, xData = [x1, x2], yData = [y1, y2]) 
     365            #key = self.addCurve(str(e), fillColor, edgeColor, 0, style = QwtCurve.Lines, xData = [x1, x2], yData = [y1, y2]) 
    365366            #self.edges[e] = (key,i,j) 
    366367            xData.append(x1) 
     
    381382            edgesCount += 1 
    382383         
    383 #        edgesCurveObject = UnconnectedLinesCurve(self, QPen(QColor(192,192,192)), xData, yData) 
    384 #        edgesCurveObject.xData = xData 
    385 #        edgesCurveObject.yData = yData 
    386         #self.edgesKey = self.insertCurve(edgesCurveObject) 
     384        edgesCurveObject = UnconnectedLinesCurve(self, QPen(QColor(192,192,192)), xData, yData) 
     385        edgesCurveObject.xData = xData 
     386        edgesCurveObject.yData = yData 
     387        self.edgesKey = self.insertCurve(edgesCurveObject) 
    387388         
    388389        selectionX = [] 
Note: See TracChangeset for help on using the changeset viewer.