demos/py_simple/gfx.py

   1 #!/usr/bin/env python
   2
   3 import sys
   4
   5 import gfxprim.core as core
   6 import gfxprim.gfx as gfx
   7 import gfxprim.backends as backends
   8 import gfxprim.input as input
   9
  10 def fill(bk):
  11     color = bk.pixmap.RGBToPixel(0xee, 0xee, 0xee)
  12     bk.pixmap.gfx.Fill(color)
  13     bk.Flip()
  14
  15 def hline(bk):
  16     fg = bk.pixmap.RGBToPixel(0xee, 0xee, 0xee)
  17     bg = bk.pixmap.RGBToPixel(0, 0, 0);
  18
  19     bk.pixmap.gfx.Fill(bg)
  20     for i in range(0, bk.pixmap.h, 10):
  21         bk.pixmap.gfx.HLine(0, bk.pixmap.w, i, fg)
  22     bk.Flip()
  23
  24 def vline(bk):
  25     fg = bk.pixmap.RGBToPixel(0xee, 0xee, 0xee)
  26     bg = bk.pixmap.RGBToPixel(0, 0, 0);
  27
  28     bk.pixmap.gfx.Fill(bg)
  29
  30     for i in range(0, bk.pixmap.w, 10):
  31         bk.pixmap.gfx.VLine(i, 0, bk.pixmap.h, fg)
  32
  33     bk.Flip()
  34
  35 def line(bk):
  36     fg = bk.pixmap.RGBToPixel(0xee, 0xee, 0xee)
  37     bg = bk.pixmap.RGBToPixel(0, 0, 0);
  38
  39     bk.pixmap.gfx.Fill(bg)
  40
  41     for i in range(0, 2 * max(bk.pixmap.w, bk.pixmap.h), 13):
  42            bk.pixmap.gfx.Line(0, i, i, 0, fg)
  43
  44     bk.Flip()
  45
  46 def rect(bk):
  47     fg = bk.pixmap.RGBToPixel(0xee, 0xee, 0xee)
  48     bg = bk.pixmap.RGBToPixel(0, 0, 0);
  49
  50     bk.pixmap.gfx.Fill(bg)
  51
  52     for i in range(10, 130, 10):
  53         bk.pixmap.gfx.Rect(i, i, bk.pixmap.w - i, bk.pixmap.h - i, fg)
  54
  55     bk.Flip()
  56
  57 def triangle(bk):
  58     fg = bk.pixmap.RGBToPixel(0xee, 0xee, 0xee)
  59     bg = bk.pixmap.RGBToPixel(0, 0, 0);
  60
  61     bk.pixmap.gfx.Fill(bg)
  62
  63     w = bk.pixmap.w
  64     h = bk.pixmap.h
  65
  66     for i in range(10, 90, 10):
  67         bk.pixmap.gfx.Triangle(2*i, i, w - 2*i, i, w//2, h - 2*i, fg)
  68
  69     bk.Flip()
  70
  71 def tetragon(bk):
  72     fg = bk.pixmap.RGBToPixel(0xee, 0xee, 0xee)
  73     bg = bk.pixmap.RGBToPixel(0, 0, 0);
  74
  75     bk.pixmap.gfx.Fill(bg)
  76
  77     w = bk.pixmap.w
  78     h = bk.pixmap.h
  79
  80     for i in range(10, 70, 10):
  81         bk.pixmap.gfx.Tetragon(i, i, w-2*i, i, w-i, h-i, 2*i, h-i, fg)
  82
  83     bk.Flip()
  84
  85 def polygon(bk):
  86     fg = bk.pixmap.RGBToPixel(0xee, 0xee, 0xee)
  87     bg = bk.pixmap.RGBToPixel(0, 0, 0);
  88
  89     bk.pixmap.gfx.Fill(bg)
  90
  91     w = bk.pixmap.w
  92     h = bk.pixmap.h
  93
  94     polygon = [(10, 10), (10, (h-10)//3), ((w-10)//3, (h-10)//2),
  95                (10, 2*(h-10)//3), (10, h-10), ((w-10)//3, h-10),
  96                ((w-10)//2, 2*(h-10)//3), (2*(w-10)//3, h-10),
  97                (w-10, h-10), (w-10, 2*(h-10)//3), (2*(w-10)//3, (h-10)//2),
  98                (w-10, (h-10)//3), (w-10, 10), (2*(w-10)//3, 10),
  99                ((w-10)//2, (h-10)//3), ((w-10)//3, 10)]
 100
 101     bk.pixmap.gfx.Polygon(polygon, fg)
 102
 103     bk.Flip()
 104
 105 def next(bk, i):
 106
 107     if (i == 0):
 108         fill(bk)
 109
 110     if (i == 1):
 111         hline(bk)
 112
 113     if (i == 2):
 114         vline(bk)
 115
 116     if (i == 3):
 117         line(bk)
 118
 119     if (i == 4):
 120         rect(bk)
 121
 122     if (i == 5):
 123         triangle(bk)
 124
 125     if (i == 6):
 126         tetragon(bk)
 127
 128     if (i == 7):
 129         polygon(bk)
 130
 131     i = i + 1;
 132
 133     if (i >= 8):
 134         i = 0
 135
 136     return i
 137
 138 def main():
 139     # Create X11 window
 140     bk = backends.BackendX11Init(None, 0, 0, 320, 240, "GFX demo", 0)
 141     assert(bk)
 142
 143     bk.Flip()
 144
 145     i = 0
 146
 147     print("Press SPACE to draw different shape, ESC to exit")
 148
 149     # Event loop
 150     while True:
 151         ev = bk.WaitEvent()
 152
 153         input.EventDump(ev)
 154
 155         if (ev.type == input.EV_KEY and ev.code == input.EV_KEY_DOWN):
 156
 157             if (ev.val.val == input.KEY_ESC):
 158                 sys.exit(0)
 159
 160             i = next(bk, i)
 161
 162         if (ev.type == input.EV_SYS):
 163            if (ev.code == input.EV_SYS_QUIT):
 164                sys.exit(0)
 165            elif (ev.code == input.EV_SYS_RESIZE):
 166                bk.ResizeAck()
 167                fill(bk)
 168                bk.Flip()
 169
 170 if __name__ == '__main__':
 171     main()