예제 #1
0
  def draw(self):

    dpi = self.prefs['dpi']
    ax_xsize = self.ax.get_window_extent().width
    ax_ysize = self.ax.get_window_extent().height
    nLabels = len(self.labels)
    nColumns = min(self.prefs['legend_max_columns'], int(ax_xsize / self.column_width))

    maxRows = self.prefs['legend_max_rows']
    nRows_ax = int(ax_ysize / 1.6 / self.prefs['text_size'])
    nRows_label = nLabels / nColumns + (nLabels % nColumns != 0)
    nRows = max(1, min(min(nRows_label, maxRows), nRows_ax))
    maxLabels = nColumns * nRows - 1
    self.ax.set_xlim(0., float(ax_xsize))
    self.ax.set_ylim(-float(ax_ysize), 0.)

    legend_text_size, legend_text_padding = self.__get_legend_text_size()
    legend_text_size_point = pixelToPoint(legend_text_size, dpi)

    box_width = legend_text_size
    legend_offset = (ax_xsize - nColumns * self.column_width) / 2

    nc = 0
    # self.labels.reverse()

    for label, num in self.labels:
      num_flag = self.prefs.get('legend_numbers', True)
      percent_flag = self.prefs.get('legend_unit', '')
      if num_flag:
        if percent_flag == "%":
          num = "%.1f" % num + '%'
        else:
          num = "%.1f" % num
      else:
        num = None
      color = self.palette.getColor(label)
      row = nc % nRows
      column = nc / nRows
      if row == nRows - 1 and column == nColumns - 1 and nc != nLabels - 1:
        last_text = '... plus %d more' % (nLabels - nc)
        self.ax.text(float(column * self.column_width) + legend_offset, -float(row * 1.6 * box_width),
                     last_text, horizontalalignment='left',
                     verticalalignment='top', size=legend_text_size_point)
        break
      else:
        self.ax.text(float(column * self.column_width) + 2. * box_width + legend_offset, -row * 1.6 * box_width,
                     str(label), horizontalalignment='left',
                     verticalalignment='top', size=legend_text_size_point)
        if num is not None:
          self.ax.text(float((column + 1) * self.column_width) - 2 * box_width + legend_offset,
                       -float(row * 1.6 * box_width),
                       str(num), horizontalalignment='right',
                       verticalalignment='top', size=legend_text_size_point)
        box = Rectangle((float(column * self.column_width) + legend_offset, -float(row * 1.6 * box_width) - box_width),
                        box_width, box_width)
        box.set_ec('black')
        box.set_linewidth(pixelToPoint(0.5, dpi))
        box.set_fc(color)
        self.ax.add_patch(box)
        nc += 1
예제 #2
0
 def draw(self):
 
   dpi = self.prefs['dpi']
   ax_xsize = self.ax.get_window_extent().width
   ax_ysize = self.ax.get_window_extent().height  
   nLabels = len(self.labels)
   nColumns = min(self.prefs['legend_max_columns'],int(ax_xsize/self.column_width))
   
   maxRows = self.prefs['legend_max_rows']
   nRows_ax = int(ax_ysize/1.6/self.prefs['text_size'])
   nRows_label = nLabels/nColumns + (nLabels%nColumns != 0)
   nRows = max(1,min(min(nRows_label,maxRows),nRows_ax ))
   maxLabels = nColumns*nRows - 1
   self.ax.set_xlim(0.,float(ax_xsize))
   self.ax.set_ylim(-float(ax_ysize),0.)
  
   legend_text_size,legend_text_padding = self.__get_legend_text_size()
   legend_text_size_point = pixelToPoint(legend_text_size,dpi)
       
   box_width = legend_text_size
   legend_offset = (ax_xsize - nColumns*self.column_width)/2 
   
   nc = 0
   #self.labels.reverse()
   
   for label,num in self.labels:
     num_flag = self.prefs.get('legend_numbers',True)
     percent_flag = self.prefs.get('legend_unit','')
     if num_flag:
       if percent_flag == "%":
         num = "%.1f" % num +'%'
       else:  
         num = "%.1f" % num
     else:
       num = None  
     color = self.palette.getColor(label)
     row = nc%nRows 
     column = nc/nRows          
     if row == nRows-1 and column == nColumns-1 and nc != nLabels-1:
       last_text = '... plus %d more' % (nLabels-nc)
       self.ax.text(float(column*self.column_width)+legend_offset,-float(row*1.6*box_width),
                    last_text,horizontalalignment='left',
                    verticalalignment='top',size=legend_text_size_point)  
       break   
     else:
       self.ax.text(float(column*self.column_width)+2.*box_width+legend_offset,-row*1.6*box_width,
                    str(label),horizontalalignment='left',
                    verticalalignment='top',size=legend_text_size_point)
       if num is not None:
         self.ax.text(float((column+1)*self.column_width)-2*box_width+legend_offset,-float(row*1.6*box_width),
                      str(num),horizontalalignment='right',
                      verticalalignment='top',size=legend_text_size_point)             
       box = Rectangle((float(column*self.column_width)+legend_offset,-float(row*1.6*box_width)-box_width),
                       box_width,box_width)            
       box.set_ec('black')
       box.set_linewidth(pixelToPoint(0.5,dpi))
       box.set_fc(color)
       self.ax.add_patch(box)
       nc += 1    
예제 #3
0
class CellObject:

    def __init__(self, row, col, value, fig, axes):
        self.row = row
        self.col = col
        self.value = value
        self.fig, self.axes = fig, axes
        self.start()

    def start(self):
        if self.value == 0:
            self.ec = (0,0,0,0.1)
            self.text_color = (0,0,0,0)
        else:
            self.ec = (0,0,0,1)
            self.text_color = (1,1,1,1)

        self.fc = color[self.value]
        
        self.center = [self.col, self.row]

        self.rect = Rectangle([self.col-0.5, self.row-0.5], \
                              width = 1, height = 1, fc = self.fc, ec = self.ec, \
                              linewidth = 2)

        self.text = self.axes.text(self.col, self.row, str(self.value), \
                                   color = self.text_color, \
                                   ha = 'center', va = 'center', \
                                   fontweight = 'bold')

        self.axes.add_patch(self.rect)
        self.fig.canvas.draw()

    def update(self, draw = False):
        if self.value == 0:
            self.ec = (0,0,0,0.1)
            self.text_color = (0,0,0,0)
        else:
            self.ec = (0,0,0,1)
            self.text_color = (1,1,1,1)

        self.fc = color[min(self.value, 2048)]
        self.rect.set_fc(self.fc)
        self.rect.set_ec(self.ec)
        self.text.set_color(self.text_color)
        self.text.set_text(str(self.value))

        if draw:
            self.fig.canvas.draw()