pyx/color.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 import attr, style
  26
  27 class color(attr.exclusiveattr, style.strokestyle, style.fillstyle):
  28
  29     """base class for all colors"""
  30
  31     def __init__(self):
  32         attr.exclusiveattr.__init__(self, color)
  33
  34
  35 clear = attr.clearclass(color)
  36
  37
  38 class grey(color):
  39
  40     """grey tones"""
  41
  42     def __init__(self, gray):
  43         color.__init__(self)
  44         if gray<0 or gray>1: raise ValueError
  45         self.color = {"gray": gray}
  46
  47     def write(self, file):
  48         file.write("%(gray)f setgray\n" % self.color)
  49
  50 grey.black = grey(0.0)
  51 grey.white = grey(1.0)
  52 gray = grey
  53
  54
  55 class rgb(color):
  56
  57     """rgb colors"""
  58
  59     def __init__(self, r=0.0, g=0.0, b=0.0):
  60         color.__init__(self)
  61         if r<0 or r>1 or g<0 or g>1 or b<0 or b>1: raise ValueError
  62         self.color = {"r": r, "g": g, "b": b}
  63
  64     def write(self, file):
  65         file.write("%(r)f %(g)f %(b)f setrgbcolor\n" % self.color)
  66
  67 rgb.red   = rgb(1 ,0, 0)
  68 rgb.green = rgb(0 ,1, 0)
  69 rgb.blue  = rgb(0 ,0, 1)
  70 rgb.white = rgb(1 ,1, 1)
  71 rgb.black = rgb(0 ,0, 0)
  72
  73
  74 class hsb(color):
  75
  76     """hsb colors"""
  77
  78     def __init__(self, h=0.0, s=0.0, b=0.0):
  79         color.__init__(self)
  80         if h<0 or h>1 or s<0 or s>1 or b<0 or b>1: raise ValueError
  81         self.color = {"h": h, "s": s, "b": b}
  82
  83     def write(self, file):
  84         file.write("%(h)f %(s)f %(b)f sethsbcolor\n" % self.color)
  85
  86
  87 class cmyk(color):
  88
  89     """cmyk colors"""
  90
  91     def __init__(self, c=0.0, m=0.0, y=0.0, k=0.0):
  92         color.__init__(self)
  93         if c<0 or c>1 or m<0 or m>1 or y<0 or y>1 or k<0 or k>1: raise ValueError
  94         self.color = {"c": c, "m": m, "y": y, "k": k}
  95
  96     def write(self, file):
  97         file.write("%(c)f %(m)f %(y)f %(k)f setcmykcolor\n" % self.color)
  98
  99
 100 cmyk.GreenYellow    = cmyk(0.15, 0, 0.69, 0)
 101 cmyk.Yellow         = cmyk(0, 0, 1, 0)
 102 cmyk.Goldenrod      = cmyk(0, 0.10, 0.84, 0)
 103 cmyk.Dandelion      = cmyk(0, 0.29, 0.84, 0)
 104 cmyk.Apricot        = cmyk(0, 0.32, 0.52, 0)
 105 cmyk.Peach          = cmyk(0, 0.50, 0.70, 0)
 106 cmyk.Melon          = cmyk(0, 0.46, 0.50, 0)
 107 cmyk.YellowOrange   = cmyk(0, 0.42, 1, 0)
 108 cmyk.Orange         = cmyk(0, 0.61, 0.87, 0)
 109 cmyk.BurntOrange    = cmyk(0, 0.51, 1, 0)
 110 cmyk.Bittersweet    = cmyk(0, 0.75, 1, 0.24)
 111 cmyk.RedOrange      = cmyk(0, 0.77, 0.87, 0)
 112 cmyk.Mahogany       = cmyk(0, 0.85, 0.87, 0.35)
 113 cmyk.Maroon         = cmyk(0, 0.87, 0.68, 0.32)
 114 cmyk.BrickRed       = cmyk(0, 0.89, 0.94, 0.28)
 115 cmyk.Red            = cmyk(0, 1, 1, 0)
 116 cmyk.OrangeRed      = cmyk(0, 1, 0.50, 0)
 117 cmyk.RubineRed      = cmyk(0, 1, 0.13, 0)
 118 cmyk.WildStrawberry = cmyk(0, 0.96, 0.39, 0)
 119 cmyk.Salmon         = cmyk(0, 0.53, 0.38, 0)
 120 cmyk.CarnationPink  = cmyk(0, 0.63, 0, 0)
 121 cmyk.Magenta        = cmyk(0, 1, 0, 0)
 122 cmyk.VioletRed      = cmyk(0, 0.81, 0, 0)
 123 cmyk.Rhodamine      = cmyk(0, 0.82, 0, 0)
 124 cmyk.Mulberry       = cmyk(0.34, 0.90, 0, 0.02)
 125 cmyk.RedViolet      = cmyk(0.07, 0.90, 0, 0.34)
 126 cmyk.Fuchsia        = cmyk(0.47, 0.91, 0, 0.08)
 127 cmyk.Lavender       = cmyk(0, 0.48, 0, 0)
 128 cmyk.Thistle        = cmyk(0.12, 0.59, 0, 0)
 129 cmyk.Orchid         = cmyk(0.32, 0.64, 0, 0)
 130 cmyk.DarkOrchid     = cmyk(0.40, 0.80, 0.20, 0)
 131 cmyk.Purple         = cmyk(0.45, 0.86, 0, 0)
 132 cmyk.Plum           = cmyk(0.50, 1, 0, 0)
 133 cmyk.Violet         = cmyk(0.79, 0.88, 0, 0)
 134 cmyk.RoyalPurple    = cmyk(0.75, 0.90, 0, 0)
 135 cmyk.BlueViolet     = cmyk(0.86, 0.91, 0, 0.04)
 136 cmyk.Periwinkle     = cmyk(0.57, 0.55, 0, 0)
 137 cmyk.CadetBlue      = cmyk(0.62, 0.57, 0.23, 0)
 138 cmyk.CornflowerBlue = cmyk(0.65, 0.13, 0, 0)
 139 cmyk.MidnightBlue   = cmyk(0.98, 0.13, 0, 0.43)
 140 cmyk.NavyBlue       = cmyk(0.94, 0.54, 0, 0)
 141 cmyk.RoyalBlue      = cmyk(1, 0.50, 0, 0)
 142 cmyk.Blue           = cmyk(1, 1, 0, 0)
 143 cmyk.Cerulean       = cmyk(0.94, 0.11, 0, 0)
 144 cmyk.Cyan           = cmyk(1, 0, 0, 0)
 145 cmyk.ProcessBlue    = cmyk(0.96, 0, 0, 0)
 146 cmyk.SkyBlue        = cmyk(0.62, 0, 0.12, 0)
 147 cmyk.Turquoise      = cmyk(0.85, 0, 0.20, 0)
 148 cmyk.TealBlue       = cmyk(0.86, 0, 0.34, 0.02)
 149 cmyk.Aquamarine     = cmyk(0.82, 0, 0.30, 0)
 150 cmyk.BlueGreen      = cmyk(0.85, 0, 0.33, 0)
 151 cmyk.Emerald        = cmyk(1, 0, 0.50, 0)
 152 cmyk.JungleGreen    = cmyk(0.99, 0, 0.52, 0)
 153 cmyk.SeaGreen       = cmyk(0.69, 0, 0.50, 0)
 154 cmyk.Green          = cmyk(1, 0, 1, 0)
 155 cmyk.ForestGreen    = cmyk(0.91, 0, 0.88, 0.12)
 156 cmyk.PineGreen      = cmyk(0.92, 0, 0.59, 0.25)
 157 cmyk.LimeGreen      = cmyk(0.50, 0, 1, 0)
 158 cmyk.YellowGreen    = cmyk(0.44, 0, 0.74, 0)
 159 cmyk.SpringGreen    = cmyk(0.26, 0, 0.76, 0)
 160 cmyk.OliveGreen     = cmyk(0.64, 0, 0.95, 0.40)
 161 cmyk.RawSienna      = cmyk(0, 0.72, 1, 0.45)
 162 cmyk.Sepia          = cmyk(0, 0.83, 1, 0.70)
 163 cmyk.Brown          = cmyk(0, 0.81, 1, 0.60)
 164 cmyk.Tan            = cmyk(0.14, 0.42, 0.56, 0)
 165 cmyk.Gray           = cmyk(0, 0, 0, 0.50)
 166 cmyk.Grey           = cmyk.Gray
 167 cmyk.Black          = cmyk(0, 0, 0, 1)
 168 cmyk.White          = cmyk(0, 0, 0, 0)
 169 cmyk.white          = cmyk.White
 170 cmyk.black          = cmyk.Black
 171
 172
 173 # TODO: - a palette is not a style (as it is not an PSOp)
 174 #       - should we have a changeable attribute instead
 175 #         those may define a method get etc.
 176
 177 class palette:
 178
 179     """palette is a collection of two colors for calculating transitions between them"""
 180
 181     def __init__(self, mincolor, maxcolor, min=0, max=1):
 182         if mincolor.__class__ != maxcolor.__class__:
 183             raise ValueError
 184         self.colorclass = mincolor.__class__
 185         self.mincolor = mincolor
 186         self.maxcolor = maxcolor
 187         self.min = min
 188         self.max = max
 189
 190     def getcolor(self, index):
 191         color = {}
 192         for key in self.mincolor.color.keys():
 193             color[key] = ((index - self.min) * self.maxcolor.color[key] +
 194                           (self.max - index) * self.mincolor.color[key]) / float(self.max - self.min)
 195         return self.colorclass(**color)
 196
 197 #palette.clear = attr._classclear(palette)
 198
 199 palette.Gray           = palette(gray.white, gray.black)
 200 palette.Grey           = palette.Gray
 201 palette.ReverseGray    = palette(gray.black, gray.white)
 202 palette.ReverseGrey    = palette.ReverseGray
 203 palette.RedGreen       = palette(rgb.red, rgb.green)
 204 palette.RedBlue        = palette(rgb.red, rgb.blue)
 205 palette.GreenRed       = palette(rgb.green, rgb.red)
 206 palette.GreenBlue      = palette(rgb.green, rgb.blue)
 207 palette.BlueRed        = palette(rgb.blue, rgb.red)
 208 palette.BlueGreen      = palette(rgb.blue, rgb.green)
 209 palette.RedBlack       = palette(rgb.red, rgb.black)
 210 palette.BlackRed       = palette(rgb.black, rgb.red)
 211 palette.RedWhite       = palette(rgb.red, rgb.white)
 212 palette.WhiteRed       = palette(rgb.white, rgb.red)
 213 palette.GreenBlack     = palette(rgb.green, rgb.black)
 214 palette.BlackGreen     = palette(rgb.black, rgb.green)
 215 palette.GreenWhite     = palette(rgb.green, rgb.white)
 216 palette.WhiteGreen     = palette(rgb.white, rgb.green)
 217 palette.BlueBlack      = palette(rgb.blue, rgb.black)
 218 palette.BlackBlue      = palette(rgb.black, rgb.blue)
 219 palette.BlueWhite      = palette(rgb.blue, rgb.white)
 220 palette.WhiteBlue      = palette(rgb.white, rgb.blue)
 221 palette.Rainbow        = palette(hsb(0, 1, 1), hsb(2.0/3.0, 1, 1))
 222 palette.ReverseRainbow = palette(hsb(2.0/3.0, 1, 1), hsb(0, 1, 1))
 223 palette.Hue            = palette(hsb(0, 1, 1), hsb(1, 1, 1))
 224 palette.ReverseHue     = palette(hsb(1, 1, 1), hsb(0, 1, 1))
 225