pyx/graph/key.py

   1 #!/usr/bin/env python
   2 # -*- coding: ISO-8859-1 -*-
   3 #
   4 #
   5 # Copyright (C) 2002-2004 Jörg Lehmann <joergl@users.sourceforge.net>
   6 # Copyright (C) 2003-2004 Michael Schindler <m-schindler@users.sourceforge.net>
   7 # Copyright (C) 2002-2004 André Wobst <wobsta@users.sourceforge.net>
   8 #
   9 # This file is part of PyX (http://pyx.sourceforge.net/).
  10 #
  11 # PyX is free software; you can redistribute it and/or modify
  12 # it under the terms of the GNU General Public License as published by
  13 # the Free Software Foundation; either version 2 of the License, or
  14 # (at your option) any later version.
  15 #
  16 # PyX is distributed in the hope that it will be useful,
  17 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19 # GNU General Public License for more details.
  20 #
  21 # You should have received a copy of the GNU General Public License
  22 # along with PyX; if not, write to the Free Software
  23 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  24
  25
  26 from pyx import box, canvas, text, unit
  27
  28
  29 class key:
  30
  31     defaulttextattrs = [text.vshift.mathaxis]
  32
  33     def __init__(self, dist=0.2*unit.v_cm, pos="tr", hinside=1, vinside=1, hdist=0.6*unit.v_cm, vdist=0.4*unit.v_cm,
  34                  symbolwidth=0.5*unit.v_cm, symbolheight=0.25*unit.v_cm, symbolspace=0.2*unit.v_cm,
  35                  textattrs=[]):
  36         self.dist = dist
  37         self.pos = pos
  38         self.hinside = hinside
  39         self.vinside = vinside
  40         self.hdist = hdist
  41         self.vdist = vdist
  42         self.symbolwidth = symbolwidth
  43         self.symbolheight = symbolheight
  44         self.symbolspace = symbolspace
  45         self.textattrs = textattrs
  46         if self.pos in ("tr", "rt"):
  47             self.right = 1
  48             self.top = 1
  49         elif self.pos in ("br", "rb"):
  50             self.right = 1
  51             self.top = 0
  52         elif self.pos in ("tl", "lt"):
  53             self.right = 0
  54             self.top = 1
  55         elif self.pos in ("bl", "lb"):
  56             self.right = 0
  57             self.top = 0
  58         else:
  59             raise RuntimeError("invalid pos attribute")
  60
  61     def paint(self, plotdata):
  62         "creates the layout of the key"
  63         plotdata = [plotdat for plotdat in plotdata if plotdat.title is not None]
  64         c = canvas.canvas()
  65         self.dist_pt = unit.topt(self.dist)
  66         self.hdist_pt = unit.topt(self.hdist)
  67         self.vdist_pt = unit.topt(self.vdist)
  68         self.symbolwidth_pt = unit.topt(self.symbolwidth)
  69         self.symbolheight_pt = unit.topt(self.symbolheight)
  70         self.symbolspace_pt = unit.topt(self.symbolspace)
  71         for plotdat in plotdata:
  72             plotdat.temp_titlebox = c.texrunner.text_pt(0, 0, plotdat.title, self.defaulttextattrs + self.textattrs)
  73         box.tile_pt([plotdat.temp_titlebox for plotdat in plotdata], self.dist_pt, 0, -1)
  74         box.linealignequal_pt([plotdat.temp_titlebox for plotdat in plotdata], self.symbolwidth_pt + self.symbolspace_pt, 1, 0)
  75         for plotdat in plotdata:
  76             plotdat.style.key_pt(c, 0, -0.5 * self.symbolheight_pt + plotdat.temp_titlebox.center[1],
  77                                  self.symbolwidth_pt, self.symbolheight_pt, plotdat)
  78             c.insert(plotdat.temp_titlebox)
  79
  80         # for plotdat in plotdata:
  81         #     del plotdat.temp_titlebox
  82         return c