gyorgyf@37: #Copyright 2006 DR0ID <dr0id@bluewin.ch> http://mypage.bluewin.ch/DR0ID
gyorgyf@37: #
gyorgyf@37: #
gyorgyf@37: #
gyorgyf@37: """
gyorgyf@37: Allow to draw some gradients relatively easy.
gyorgyf@37: """
gyorgyf@37: 
gyorgyf@37: __author__ = "$Author: DR0ID $"
gyorgyf@37: __version__= "$Revision: 109 $"
gyorgyf@37: __date__   = "$Date: 2007-08-09 20:33:32 +0200 (Do, 09 Aug 2007) $"
gyorgyf@37: 
gyorgyf@37: import pygame
gyorgyf@37: import math
gyorgyf@37: 
gyorgyf@37: BLEND_MODES_AVAILABLE = False
gyorgyf@37: vernum = pygame.vernum
gyorgyf@37: if vernum[0]>=1 and vernum[1]>=8:
gyorgyf@37:     BLEND_MODES_AVAILABLE = True
gyorgyf@37:     
gyorgyf@37: 
gyorgyf@37: class ColorInterpolator(object):
gyorgyf@37:     '''
gyorgyf@37:     ColorInterpolator(distance, color1, color2, rfunc, gfunc, bfunc, afunc)
gyorgyf@37:     
gyorgyf@37:     interpolates a color over the distance using different functions for r,g,b,a
gyorgyf@37:     separately (a= alpha).
gyorgyf@37:     '''
gyorgyf@37:     def __init__(self, distance, color1, color2, rfunc, gfunc, bfunc, afunc):
gyorgyf@37:         object.__init__(self)
gyorgyf@37:         
gyorgyf@37:         self.rInterpolator = FunctionInterpolator(color1[0], color2[0], distance, rfunc)
gyorgyf@37:         self.gInterpolator = FunctionInterpolator(color1[1], color2[1], distance, gfunc)
gyorgyf@37:         self.bInterpolator = FunctionInterpolator(color1[2], color2[2], distance, bfunc)
gyorgyf@37:         if len(color1)==4 and len(color2)==4:
gyorgyf@37:             self.aInterpolator = FunctionInterpolator(color1[3], color2[3], distance, afunc)
gyorgyf@37:         else:
gyorgyf@37:             self.aInterpolator = FunctionInterpolator(255, 255, distance, afunc)
gyorgyf@37:             
gyorgyf@37:     def eval(self, x):
gyorgyf@37:         '''
gyorgyf@37:         eval(x) -> color
gyorgyf@37:         
gyorgyf@37:         returns the color at the position 0<=x<=d (actually not bound to this interval).
gyorgyf@37:         '''
gyorgyf@37: ##        print "colorInterp x", x, self.rInterpolator.eval(x), self.gInterpolator.eval(x), self.bInterpolator.eval(x)
gyorgyf@37:         return [self.rInterpolator.eval(x), 
gyorgyf@37:                 self.gInterpolator.eval(x), 
gyorgyf@37:                 self.bInterpolator.eval(x), 
gyorgyf@37:                 self.aInterpolator.eval(x)]
gyorgyf@37:             
gyorgyf@37: 
gyorgyf@37: 
gyorgyf@37: class FunctionInterpolator(object):
gyorgyf@37:     '''
gyorgyf@37:     FunctionINterpolator(startvalue, endvalue, trange, func)
gyorgyf@37:     
gyorgyf@37:     interpolates a function y=f(x) in the range trange with
gyorgyf@37:     startvalue = f(0)
gyorgyf@37:     endvalue   = f(trange)
gyorgyf@37:     using the function func
gyorgyf@37:     '''
gyorgyf@37:     def __init__(self, startvalue, endvalue, trange, func):
gyorgyf@37:         object.__init__(self)
gyorgyf@37:         # function
gyorgyf@37:         self.func = func
gyorgyf@37:         # y-scaling
gyorgyf@37:         self.a = endvalue-startvalue
gyorgyf@37:         if self.a == 0:
gyorgyf@37:             self.a = 1.
gyorgyf@37:         # x-scaling
gyorgyf@37:         if trange!=0:
gyorgyf@37:             self.b = 1./abs(trange)
gyorgyf@37:         else:
gyorgyf@37:             self.b = 1.
gyorgyf@37:         # x-displacement
gyorgyf@37:         self.c = 0
gyorgyf@37:         # y-displacement
gyorgyf@37:         self.d = min(max(startvalue,0),255)
gyorgyf@37:         
gyorgyf@37:     def eval(self, x):
gyorgyf@37:         ''' 
gyorgyf@37:         eval(x)->float
gyorgyf@37:         
gyorgyf@37:         return value at position x
gyorgyf@37:         '''
gyorgyf@37:         # make sure that the returned value is in [0,255]
gyorgyf@37: ##        return int(round(min(max(self.a*self.func(self.b*(x+self.c))+self.d, 0), 255)))
gyorgyf@37:         return int(min(max(self.a*self.func(self.b*(x+self.c))+self.d, 0), 255))
gyorgyf@37: 
gyorgyf@37: 
gyorgyf@37: 
gyorgyf@37: ##def gradient(surface, 
gyorgyf@37: ##                startpoint, 
gyorgyf@37: ##                endpoint, 
gyorgyf@37: ##                startcolor, 
gyorgyf@37: ##                endcolor,
gyorgyf@37: ##                Rfunc = (lambda x:x), 
gyorgyf@37: ##                Gfunc = (lambda x:x), 
gyorgyf@37: ##                Bfunc = (lambda x:x), 
gyorgyf@37: ##                Afunc = (lambda x:1), 
gyorgyf@37: ##                type  = "line", 
gyorgyf@37: ##                mode  = None ):
gyorgyf@37: ##    '''
gyorgyf@37: ##    surface   : surface to draw on
gyorgyf@37: ##    startpoint: (x,y) point on surface
gyorgyf@37: ##    endpoint  : (x,y) point on surface
gyorgyf@37: ##    startcolor: (r,g,b,a) color at startpoint
gyorgyf@37: ##    endcolor  : (r,g,b,a) color at endpoint
gyorgyf@37: ##    Rfunc     : function y = f(x) with  startcolor =f(0) and endcolor = f(1) where 0 is at startpoint and 1 at endpoint
gyorgyf@37: ##    Gfunc     :  ---  "  ---
gyorgyf@37: ##    Bfunc     :  ---  "  ---
gyorgyf@37: ##    Afunc     :  ---  "  ---
gyorgyf@37: ##                these functions are evaluated in the range 0 <= x <= 1 and 0<= y=f(x) <= 1
gyorgyf@37: ##    type      : "line", "circle" or "rect"
gyorgyf@37: ##    mode      : "+", "-", "*", None (how the pixels are drawen)
gyorgyf@37: ##    
gyorgyf@37: ##    returns   : surface with the color characteristics w,h = (d, 256) and d = length of endpoint-startpoint
gyorgyf@37: ##    
gyorgyf@37: ##    '''
gyorgyf@37: ##    dx = endpoint[0]-startpoint[0]
gyorgyf@37: ##    dy = endpoint[1]-startpoint[1]
gyorgyf@37: ##    d = int(round(math.hypot(dx, dy)))
gyorgyf@37: ##    angle = math.degrees( math.atan2(dy, dx) )
gyorgyf@37: ##    
gyorgyf@37: ##    color = ColorInterpolator(d, startcolor, endcolor, Rfunc, Gfunc, Bfunc, Afunc)
gyorgyf@37: ##    
gyorgyf@37: ##    if type=="line":
gyorgyf@37: ##        h = int(2.*math.hypot(*surface.get_size()))
gyorgyf@37: ###        bigSurf = pygame.Surface((d, h)).convert_alpha()
gyorgyf@37: ##        bigSurf = pygame.Surface((d, h), pygame.SRCALPHA)#.convert_alpha()
gyorgyf@37: ###        bigSurf = pygame.Surface((d, 1), pygame.SRCALPHA)#.convert_alpha()
gyorgyf@37: ##        bigSurf.lock()
gyorgyf@37: ##        bigSurf.fill((0,0,0,0))
gyorgyf@37: ##        bigSurf.set_colorkey((0,0,0,0))
gyorgyf@37: ##        for x in range(d):
gyorgyf@37: ##            pygame.draw.line(bigSurf, color.eval(x), (x,0), (x,h), 1)
gyorgyf@37: ###        for x in range(d):
gyorgyf@37: ###            bigSurf.set_at((x, 0), color.eval(x))
gyorgyf@37: ###        bigSurf = pygame.transform.scale(bigSurf, (d, h))
gyorgyf@37: ##            
gyorgyf@37: ##        bigSurf = pygame.transform.rotate(bigSurf, -angle) #rotozoom(bigSurf, -angle, 1)
gyorgyf@37: ##        bigSurf.set_colorkey((0,0,0, 0))
gyorgyf@37: ##        rect = bigSurf.get_rect()
gyorgyf@37: ##        srect = pygame.Rect(rect)
gyorgyf@37: ##        dx = d/2. * math.cos(math.radians(angle))
gyorgyf@37: ##        dy = d/2. * math.sin(math.radians(angle))
gyorgyf@37: ##        rect.center = startpoint
gyorgyf@37: ##        rect.move_ip(dx, dy)
gyorgyf@37: ##        bigSurf.unlock()
gyorgyf@37: ##        
gyorgyf@37: ##    elif type=="circle":
gyorgyf@37: ##        bigSurf = pygame.Surface((2*d, 2*d)).convert_alpha()
gyorgyf@37: ##        bigSurf.fill((0,0,0,0))
gyorgyf@37: ##        bigSurf.lock()
gyorgyf@37: ##        for x in range(d, 0, -1):
gyorgyf@37: ##            pygame.draw.circle(bigSurf, color.eval(x), (d,d), x)
gyorgyf@37: ##        bigSurf.unlock()
gyorgyf@37: ##        rect = bigSurf.get_rect()
gyorgyf@37: ##        srect = pygame.Rect(rect)
gyorgyf@37: ##        rect.center = (startpoint[0], startpoint[1])
gyorgyf@37: ##        
gyorgyf@37: ##    elif type=="rect":
gyorgyf@37: ##        bigSurf = pygame.Surface((2*d, 2*d)).convert_alpha()
gyorgyf@37: ##        bigSurf.fill((0,0,0,0))
gyorgyf@37: ##        c = bigSurf.get_rect().center
gyorgyf@37: ##        bigSurf.lock()
gyorgyf@37: ##        for x in range(d,-1,-1):
gyorgyf@37: ##            r = pygame.Rect(0,0,2*x,2*x)
gyorgyf@37: ##            r.center = c
gyorgyf@37: ##            pygame.draw.rect(bigSurf, color.eval(x), r)
gyorgyf@37: ##        bigSurf.unlock()
gyorgyf@37: ##        bigSurf = pygame.transform.rotozoom(bigSurf, -angle, 1)
gyorgyf@37: ##        bigSurf.set_colorkey((0,0,0, 0))
gyorgyf@37: ##        
gyorgyf@37: ##        rect = bigSurf.get_rect()
gyorgyf@37: ##        srect = pygame.Rect(rect)
gyorgyf@37: ##        rect.center = startpoint
gyorgyf@37: ##    else:
gyorgyf@37: ##        raise NameError("type must be one of \"line\",\"circle\" or \"rect\"")
gyorgyf@37: ##    
gyorgyf@37: ##    if mode is None:
gyorgyf@37: ##        surface.blit(bigSurf, rect, srect)
gyorgyf@37: ##    else:
gyorgyf@37: ##        if mode=="+":
gyorgyf@37: ##            cf = pygame.color.add
gyorgyf@37: ##        elif mode=="*":
gyorgyf@37: ##            cf = pygame.color.multiply
gyorgyf@37: ##        elif mode=="-":
gyorgyf@37: ##            cf = pygame.color.subtract
gyorgyf@37: ##        else:
gyorgyf@37: ##            raise NameError("type must be one of \"+\", \"*\", \"-\" or None")
gyorgyf@37: ##        irect = surface.get_clip().clip(rect)
gyorgyf@37: ##        surface.lock()
gyorgyf@37: ##        for x in range(irect.left, irect.left+irect.width):
gyorgyf@37: ##            for y in range(irect.top, irect.top+irect.height):
gyorgyf@37: ##                surface.set_at((x,y), cf(surface.get_at((x,y)), bigSurf.get_at((x-rect.left, y-rect.top)) ) )
gyorgyf@37: ##        surface.unlock()
gyorgyf@37: ##    
gyorgyf@37: ##    del bigSurf   
gyorgyf@37: ##    char = pygame.Surface((d+1, 257))
gyorgyf@37: ###    char.fill((0,0,0))
gyorgyf@37: ###    ox = 0
gyorgyf@37: ###    oldcol = color.eval(0)
gyorgyf@37: ###    for x in range(d):
gyorgyf@37: ###        col = color.eval(x)
gyorgyf@37: ###        pygame.draw.line(char, (255,0,0), (x, 256-col[0]), (ox, 256-oldcol[0]))
gyorgyf@37: ###        pygame.draw.line(char, (0,255,0), (x, 256-col[1]), (ox, 256-oldcol[1]))
gyorgyf@37: ###        pygame.draw.line(char, (0,0,255), (x, 256-col[2]), (ox, 256-oldcol[2]))
gyorgyf@37: ###        pygame.draw.line(char, (255,255,255), (x, 256-col[3]), (ox, 256-oldcol[3]))
gyorgyf@37: ###        ox = x
gyorgyf@37: ###        oldcol = col
gyorgyf@37: ###     
gyorgyf@37: ##    return char
gyorgyf@37:         
gyorgyf@37:     
gyorgyf@37:     
gyorgyf@37: 
gyorgyf@37: def vertical(size, startcolor, endcolor):
gyorgyf@37:     """
gyorgyf@37:     Draws a vertical linear gradient filling the entire surface. Returns a
gyorgyf@37:     surface filled with the gradient (numeric is only 2-3 times faster).
gyorgyf@37:     """
gyorgyf@37:     height = size[1]
gyorgyf@37:     bigSurf = pygame.Surface((1,height)).convert_alpha()
gyorgyf@37:     dd = 1.0/height
gyorgyf@37:     sr, sg, sb, sa = startcolor
gyorgyf@37:     er, eg, eb, ea = endcolor
gyorgyf@37:     rm = (er-sr)*dd
gyorgyf@37:     gm = (eg-sg)*dd
gyorgyf@37:     bm = (eb-sb)*dd
gyorgyf@37:     am = (ea-sa)*dd
gyorgyf@37:     for y in range(height):
gyorgyf@37:         bigSurf.set_at((0,y),
gyorgyf@37:                         (int(sr + rm*y),
gyorgyf@37:                          int(sg + gm*y),
gyorgyf@37:                          int(sb + bm*y),
gyorgyf@37:                          int(sa + am*y))
gyorgyf@37:                       )
gyorgyf@37:     return pygame.transform.scale(bigSurf, size)
gyorgyf@37: 
gyorgyf@37: 
gyorgyf@37: def horizontal(size, startcolor, endcolor):
gyorgyf@37:     """
gyorgyf@37:     Draws a horizontal linear gradient filling the entire surface. Returns a
gyorgyf@37:     surface filled with the gradient (numeric is only 2-3 times faster).
gyorgyf@37:     """
gyorgyf@37:     width = size[0]
gyorgyf@37:     bigSurf = pygame.Surface((width, 1)).convert_alpha()
gyorgyf@37:     dd = 1.0/width
gyorgyf@37:     sr, sg, sb, sa = startcolor
gyorgyf@37:     er, eg, eb, ea = endcolor
gyorgyf@37:     rm = (er-sr)*dd
gyorgyf@37:     gm = (eg-sg)*dd
gyorgyf@37:     bm = (eb-sb)*dd
gyorgyf@37:     am = (ea-sa)*dd
gyorgyf@37:     for y in range(width):
gyorgyf@37:         bigSurf.set_at((y,0),
gyorgyf@37:                         (int(sr + rm*y),
gyorgyf@37:                          int(sg + gm*y),
gyorgyf@37:                          int(sb + bm*y),
gyorgyf@37:                          int(sa + am*y))
gyorgyf@37:                       )
gyorgyf@37:     return pygame.transform.scale(bigSurf, size)
gyorgyf@37: 
gyorgyf@37: 
gyorgyf@37: def radial(radius, startcolor, endcolor):
gyorgyf@37:     """
gyorgyf@37:     Draws a linear raidal gradient on a square sized surface and returns
gyorgyf@37:     that surface.
gyorgyf@37:     """
gyorgyf@37:     bigSurf = pygame.Surface((2*radius, 2*radius)).convert_alpha()
gyorgyf@37:     bigSurf.fill((0,0,0,0))
gyorgyf@37:     dd = -1.0/radius
gyorgyf@37:     sr, sg, sb, sa = endcolor
gyorgyf@37:     er, eg, eb, ea = startcolor
gyorgyf@37:     rm = (er-sr)*dd
gyorgyf@37:     gm = (eg-sg)*dd
gyorgyf@37:     bm = (eb-sb)*dd
gyorgyf@37:     am = (ea-sa)*dd
gyorgyf@37:     
gyorgyf@37:     draw_circle = pygame.draw.circle
gyorgyf@37:     for rad in range(radius, 0, -1):
gyorgyf@37:         draw_circle(bigSurf, (er + int(rm*rad),
gyorgyf@37:                               eg + int(gm*rad),
gyorgyf@37:                               eb + int(bm*rad),
gyorgyf@37:                               ea + int(am*rad)), (radius, radius), rad)
gyorgyf@37:     return bigSurf
gyorgyf@37: 
gyorgyf@37: def squared(width, startcolor, endcolor):
gyorgyf@37:     """
gyorgyf@37:     Draws a linear sqared gradient on a square sized surface and returns
gyorgyf@37:     that surface.
gyorgyf@37:     """
gyorgyf@37:     bigSurf = pygame.Surface((width, width)).convert_alpha()
gyorgyf@37:     bigSurf.fill((0,0,0,0))
gyorgyf@37:     dd = -1.0/(width/2)
gyorgyf@37:     sr, sg, sb, sa = endcolor
gyorgyf@37:     er, eg, eb, ea = startcolor
gyorgyf@37:     rm = (er-sr)*dd
gyorgyf@37:     gm = (eg-sg)*dd
gyorgyf@37:     bm = (eb-sb)*dd
gyorgyf@37:     am = (ea-sa)*dd
gyorgyf@37:     
gyorgyf@37:     draw_rect = pygame.draw.rect
gyorgyf@37:     for currentw in range((width/2), 0, -1):
gyorgyf@37:         pos = (width/2)-currentw
gyorgyf@37:         draw_rect(bigSurf, (er + int(rm*currentw),
gyorgyf@37:                             eg + int(gm*currentw),
gyorgyf@37:                             eb + int(bm*currentw),
gyorgyf@37:                             ea + int(am*currentw)), pygame.Rect(pos, pos, 2*currentw, 2*currentw ))
gyorgyf@37:     return bigSurf
gyorgyf@37: 
gyorgyf@37: 
gyorgyf@37: def vertical_func(size, startcolor, endcolor, Rfunc = (lambda x:x), Gfunc = (lambda x:x), Bfunc = (lambda x:x), Afunc = (lambda x:1)):
gyorgyf@37:     """
gyorgyf@37:     Draws a vertical linear gradient filling the entire surface. Returns a
gyorgyf@37:     surface filled with the gradient (numeric is only 2x faster).
gyorgyf@37:     Rfunc, Gfunc, Bfunc and Afunc are function like y = f(x). They define
gyorgyf@37:     how the color changes.
gyorgyf@37:     """
gyorgyf@37:     height = size[1]
gyorgyf@37:     bigSurf = pygame.Surface((1,height)).convert_alpha()
gyorgyf@37:     color = ColorInterpolator(height, startcolor, endcolor, Rfunc, Gfunc, Bfunc, Afunc)
gyorgyf@37:     for y in range(0, height):
gyorgyf@37:         bigSurf.set_at((0,y), color.eval(y+0.1))
gyorgyf@37:     return pygame.transform.scale(bigSurf, size)
gyorgyf@37: 
gyorgyf@37: 
gyorgyf@37: def horizontal_func(size, startcolor, endcolor, Rfunc = (lambda x:x), Gfunc = (lambda x:x), Bfunc = (lambda x:x), Afunc = (lambda x:1)):
gyorgyf@37:     """
gyorgyf@37:     Draws a horizontal linear gradient filling the entire surface. Returns a
gyorgyf@37:     surface filled with the gradient (numeric is only 2x faster).
gyorgyf@37:     Rfunc, Gfunc, Bfunc and Afunc are function like y = f(x). They define
gyorgyf@37:     how the color changes.
gyorgyf@37:     """
gyorgyf@37:     width = size[0]
gyorgyf@37:     bigSurf = pygame.Surface((width, 1)).convert_alpha()
gyorgyf@37:     color = ColorInterpolator(width, startcolor, endcolor, Rfunc, Gfunc, Bfunc, Afunc)
gyorgyf@37:     for y in range(0, width):
gyorgyf@37:         bigSurf.set_at((y, 0), color.eval(y+0.1))
gyorgyf@37:     return pygame.transform.scale(bigSurf, size)
gyorgyf@37: 
gyorgyf@37: def radial_func(radius, startcolor, endcolor, Rfunc = (lambda x:x), Gfunc = (lambda x:x), Bfunc = (lambda x:x), Afunc = (lambda x:1), colorkey=(0,0,0,0)):
gyorgyf@37:     """
gyorgyf@37:     Draws a linear raidal gradient on a square sized surface and returns
gyorgyf@37:     that surface.
gyorgyf@37:     """
gyorgyf@37:     bigSurf = pygame.Surface((2*radius, 2*radius)).convert_alpha()
gyorgyf@37:     if len(colorkey)==3:
gyorgyf@37:         colorkey += (0,)
gyorgyf@37:     bigSurf.fill(colorkey)
gyorgyf@37:     color = ColorInterpolator(radius, startcolor, endcolor, Rfunc, Gfunc, Bfunc, Afunc)
gyorgyf@37:     draw_circle = pygame.draw.circle
gyorgyf@37:     for rad in range(radius, 0, -1):
gyorgyf@37:         draw_circle(bigSurf, color.eval(rad), (radius, radius), rad)
gyorgyf@37:     return bigSurf
gyorgyf@37: 
gyorgyf@37: def radial_func_offset(radius, startcolor, endcolor, Rfunc = (lambda x:x), Gfunc = (lambda x:x), Bfunc = (lambda x:x), Afunc = (lambda x:1), colorkey=(0,0,0,0), offset=(0,0)):
gyorgyf@37:     """
gyorgyf@37:     Draws a linear raidal gradient on a square sized surface and returns
gyorgyf@37:     that surface.
gyorgyf@37:     offset is the amount the center of the gradient is displaced of the center of the image.
gyorgyf@37:     Unfotunately this function ignores alpha.
gyorgyf@37:     """
gyorgyf@37:     bigSurf = pygame.Surface((2*radius, 2*radius))#.convert_alpha()
gyorgyf@37:     
gyorgyf@37:     mask = pygame.Surface((2*radius, 2*radius), pygame.SRCALPHA)#.convert_alpha()
gyorgyf@37:     mask.fill(colorkey)
gyorgyf@37:     mask.set_colorkey((255,0,255))
gyorgyf@37:     pygame.draw.circle(mask, (255,0,255), (radius, radius), radius)
gyorgyf@37:     
gyorgyf@37:     if len(colorkey)==3:
gyorgyf@37:         colorkey += (0,)
gyorgyf@37:     bigSurf.fill(colorkey)
gyorgyf@37:     
gyorgyf@37:     color = ColorInterpolator(radius, startcolor, endcolor, Rfunc, Gfunc, Bfunc, Afunc)
gyorgyf@37:     draw_circle = pygame.draw.circle
gyorgyf@37:     radi = radius + int(math.hypot(offset[0], offset[1])+1)
gyorgyf@37:     for rad in range(radi, 0, -1):
gyorgyf@37:         draw_circle(bigSurf, color.eval(rad), (radius+offset[0], radius+offset[1]), rad)
gyorgyf@37:         
gyorgyf@37:     bigSurf.blit(mask, (0,0))
gyorgyf@37:     bigSurf.set_colorkey(colorkey)
gyorgyf@37:     return bigSurf
gyorgyf@37: 
gyorgyf@37: 
gyorgyf@37: def squared_func(width, startcolor, endcolor, Rfunc = (lambda x:x), Gfunc = (lambda x:x), Bfunc = (lambda x:x), Afunc = (lambda x:1), offset=(0,0)):
gyorgyf@37:     """
gyorgyf@37:     Draws a linear sqared gradient on a square sized surface and returns
gyorgyf@37:     that surface.
gyorgyf@37:     """
gyorgyf@37:     bigSurf = pygame.Surface((width, width)).convert_alpha()
gyorgyf@37:     bigSurf.fill((0,0,0,0))
gyorgyf@37:     color = ColorInterpolator(width/2, startcolor, endcolor, Rfunc, Gfunc, Bfunc, Afunc)
gyorgyf@37:     draw_rect = pygame.draw.rect
gyorgyf@37:     widthh = width+2*int(max(abs(offset[0]),abs(offset[1])))
gyorgyf@37:     for currentw in range((widthh/2), 0, -1):
gyorgyf@37: ##        pos = (width/2)-currentw
gyorgyf@37:         rect = pygame.Rect(0, 0, 2*currentw, 2*currentw )
gyorgyf@37:         rect.center = (width/2+offset[0], width/2+offset[1])
gyorgyf@37:         draw_rect(bigSurf, color.eval(currentw), rect)
gyorgyf@37:     return bigSurf
gyorgyf@37: 
gyorgyf@37: def draw_gradient(surface, startpoint, endpoint, startcolor, endcolor, Rfunc = (lambda x:x), Gfunc = (lambda x:x), Bfunc = (lambda x:x), Afunc = (lambda x:1), mode=0):
gyorgyf@37:     """
gyorgyf@37:     Instead of returning an Surface, this function draw it directy onto the 
gyorgyf@37:     given Surface and returns the rect.
gyorgyf@37:     """
gyorgyf@37:     dx = endpoint[0]-startpoint[0]
gyorgyf@37:     dy = endpoint[1]-startpoint[1]
gyorgyf@37:     d = int(round(math.hypot(dx, dy)))
gyorgyf@37:     angle = math.degrees( math.atan2(dy, dx) )
gyorgyf@37:     
gyorgyf@37:     h = int(2.*math.hypot(*surface.get_size()))
gyorgyf@37:     
gyorgyf@37:     bigSurf = horizontal_func((d,h), startcolor, endcolor, Rfunc, Gfunc, Bfunc, Afunc)
gyorgyf@37:     
gyorgyf@37: ##    bigSurf = pygame.transform.rotate(bigSurf, -angle) #rotozoom(bigSurf, -angle, 1)
gyorgyf@37:     bigSurf = pygame.transform.rotozoom(bigSurf, -angle, 1)
gyorgyf@37: ##    bigSurf.set_colorkey((0,0,0, 0))
gyorgyf@37:     rect = bigSurf.get_rect()
gyorgyf@37:     srect = pygame.Rect(rect)
gyorgyf@37:     dx = d/2. * math.cos(math.radians(angle))
gyorgyf@37:     dy = d/2. * math.sin(math.radians(angle))
gyorgyf@37:     rect.center = startpoint
gyorgyf@37:     rect.move_ip(dx, dy)
gyorgyf@37:     if BLEND_MODES_AVAILABLE:
gyorgyf@37:         return surface.blit(bigSurf, rect, None, mode)
gyorgyf@37:     else:
gyorgyf@37:         return surface.blit(bigSurf, rect)
gyorgyf@37: 
gyorgyf@37: 
gyorgyf@37: def draw_circle(surface, startpoint, endpoint, startcolor, endcolor, Rfunc = (lambda x:x), Gfunc = (lambda x:x), Bfunc = (lambda x:x), Afunc = (lambda x:1), mode=0):
gyorgyf@37:     """
gyorgyf@37:     Instead of returning an Surface, this function draw it directy onto the 
gyorgyf@37:     given Surface and returns the rect.
gyorgyf@37:     """
gyorgyf@37:     dx = endpoint[0]-startpoint[0]
gyorgyf@37:     dy = endpoint[1]-startpoint[1]
gyorgyf@37:     radius = int(round(math.hypot(dx, dy)))
gyorgyf@37:     pos = (startpoint[0]-radius, startpoint[1]-radius)
gyorgyf@37:     if BLEND_MODES_AVAILABLE:
gyorgyf@37:         return surface.blit(radial_func(radius, startcolor, endcolor, Rfunc, Gfunc, Bfunc, Afunc), pos, None, mode)
gyorgyf@37:     else:
gyorgyf@37:         return surface.blit(radial_func(radius, startcolor, endcolor, Rfunc, Gfunc, Bfunc, Afunc), pos)
gyorgyf@37: 
gyorgyf@37: def draw_squared(surface, startpoint, endpoint, startcolor, endcolor, Rfunc = (lambda x:x), Gfunc = (lambda x:x), Bfunc = (lambda x:x), Afunc = (lambda x:1), mode=0):
gyorgyf@37:     """
gyorgyf@37:     Instead of returning an Surface, this function draw it directy onto the 
gyorgyf@37:     given Surface and returns the rect.
gyorgyf@37:     """
gyorgyf@37:     dx = endpoint[0]-startpoint[0]
gyorgyf@37:     dy = endpoint[1]-startpoint[1]
gyorgyf@37:     angle = math.degrees( math.atan2(dy, dx) )
gyorgyf@37:     width = 2*int(round(math.hypot(dx, dy)))
gyorgyf@37:     
gyorgyf@37:     bigSurf = squared_func(width, startcolor, endcolor, Rfunc, Gfunc, Bfunc, Afunc)
gyorgyf@37:     
gyorgyf@37:     bigSurf = pygame.transform.rotozoom(bigSurf, -angle, 1)
gyorgyf@37: ##    bigSurf.set_colorkey((0,0,0, 0))
gyorgyf@37:     rect = bigSurf.get_rect()
gyorgyf@37:     rect.center = startpoint
gyorgyf@37:     if BLEND_MODES_AVAILABLE:
gyorgyf@37:         return surface.blit(bigSurf, rect, None, mode)
gyorgyf@37:     else:
gyorgyf@37:         return surface.blit(bigSurf, rect)
gyorgyf@37:     
gyorgyf@37:     
gyorgyf@37: def chart(startpoint, endpoint, startcolor, endcolor, Rfunc = (lambda x:x), Gfunc = (lambda x:x), Bfunc = (lambda x:x), Afunc = (lambda x:1), scale=None):
gyorgyf@37:     """
gyorgyf@37:     This returns a Surface where the change of the colors over the distance 
gyorgyf@37:     (the width of the image) is showen as a line.
gyorgyf@37:     scale: a float, 1 is not scaling
gyorgyf@37:     """
gyorgyf@37:     dx = endpoint[0]-startpoint[0]
gyorgyf@37:     dy = endpoint[1]-startpoint[1]
gyorgyf@37:     distance = int(round(math.hypot(dx, dy)))
gyorgyf@37:     color = ColorInterpolator(distance, startcolor, endcolor, Rfunc, Gfunc, Bfunc, Afunc)
gyorgyf@37:     bigSurf = pygame.Surface((distance, 256))
gyorgyf@37:     bigSurf.fill((0,)*3)
gyorgyf@37:     oldcol = color.eval(0)
gyorgyf@37:     for x in range(distance):
gyorgyf@37:         r, g, b, a = color.eval(x)
gyorgyf@37:         pygame.draw.line(bigSurf, (255, 0, 0, 128), (x-1, oldcol[0]), (x, r))
gyorgyf@37:         pygame.draw.line(bigSurf, (0, 255, 0, 128), (x-1, oldcol[1]), (x, g))
gyorgyf@37:         pygame.draw.line(bigSurf, (0, 0, 255, 128), (x-1, oldcol[2]), (x, b))
gyorgyf@37:         pygame.draw.line(bigSurf, (255, 255, 255, 128), (x-1, oldcol[3]), (x, a))
gyorgyf@37:         oldcol = (r,g,b,a)
gyorgyf@37:     if scale:
gyorgyf@37: ##        return pygame.transform.scale(bigSurf, size)
gyorgyf@37:         return pygame.transform.rotozoom(bigSurf, 0, scale)
gyorgyf@37:     return pygame.transform.flip(bigSurf, 0, 1)
gyorgyf@37: #------------------------------------------------------------------------------
gyorgyf@37: 
gyorgyf@37: 
gyorgyf@37:     
gyorgyf@37: 
gyorgyf@37: def genericFxyGradient(surf, clip, color1, color2, func, intx, yint, zint=None):
gyorgyf@37:     """
gyorgyf@37:     genericFxyGradient(size, color1, color2,func, intx, yint, zint=None)
gyorgyf@37:     
gyorgyf@37:     some sort of highfield drawer :-)
gyorgyf@37:     
gyorgyf@37:     surf   : surface to draw
gyorgyf@37:     clip   : rect on surf to draw in
gyorgyf@37:     color1 : start color
gyorgyf@37:     color2 : end color
gyorgyf@37:     func   : function z = func(x,y)
gyorgyf@37:     xint   : interval in x direction where the function is evaluated
gyorgyf@37:     yint   : interval in y direction where the function is evaluated
gyorgyf@37:     zint   : if not none same as yint or xint, if None then the max and min value
gyorgyf@37:              of func is taken as z-interval
gyorgyf@37:     
gyorgyf@37:     color = a*func(b*(x+c), d*(y+e))+f
gyorgyf@37:     """
gyorgyf@37:     # make shure that x1<x2 and y1<y2 and z1<z2
gyorgyf@37:     w,h = clip.size
gyorgyf@37:     x1 = min(intx)
gyorgyf@37:     x2 = max(intx)
gyorgyf@37:     y1 = min(yint)
gyorgyf@37:     y2 = max(yint)
gyorgyf@37:     if zint: # if user give us z intervall, then use it
gyorgyf@37:         z1 = min(zint)
gyorgyf@37:         z2 = max(zint)
gyorgyf@37:     else: # look for extrema of function (not best algorithme)
gyorgyf@37:         z1 = func(x1,y1)
gyorgyf@37:         z2 = z1
gyorgyf@37:         for i in range(w):
gyorgyf@37:             for j in range(h):
gyorgyf@37:                 r = func(i,j)
gyorgyf@37:                 z1 = min(z1, r)
gyorgyf@37:                 z2 = max(z2, r)
gyorgyf@37:                 
gyorgyf@37:     x1 = float(x1)
gyorgyf@37:     x2 = float(x2)
gyorgyf@37:     y1 = float(y1)
gyorgyf@37:     y2 = float(y2)
gyorgyf@37:     z1 = float(z1)
gyorgyf@37:     z2 = float(z2)
gyorgyf@37:     if len(color1)==3:
gyorgyf@37:         color1 = list(color1)
gyorgyf@37:         color1.append(255)
gyorgyf@37:     if len(color2)==3:
gyorgyf@37:         color2 = list(color2)
gyorgyf@37:         color2.append(255)
gyorgyf@37:     
gyorgyf@37:     # calculate streching and displacement variables
gyorgyf@37:     a = ((color2[0]-color1[0])/(z2-z1), \
gyorgyf@37:          (color2[1]-color1[1])/(z2-z1), \
gyorgyf@37:          (color2[2]-color1[2])/(z2-z1), \
gyorgyf@37:          (color2[3]-color1[3])/(z2-z1) ) # streching in z direction
gyorgyf@37:     b = (x2-x1)/float(w) # streching in x direction
gyorgyf@37:     d = (y2-y1)/float(h) # streching in y direction
gyorgyf@37:     f = ( color1[0]-a[0]*z1, \
gyorgyf@37:           color1[1]-a[1]*z1, \
gyorgyf@37:           color1[2]-a[2]*z1, \
gyorgyf@37:           color1[3]-a[3]*z1 )# z displacement
gyorgyf@37:     c = x1/b
gyorgyf@37:     e = y1/d
gyorgyf@37:     
gyorgyf@37:     surff = pygame.surface.Surface((w,h)).convert_alpha()
gyorgyf@37:     # generate values
gyorgyf@37:     for i in range(h):
gyorgyf@37:         for j in range(w):
gyorgyf@37:             val = func(b*(j+c), d*(i+e))
gyorgyf@37:             #clip color
gyorgyf@37:             color = (   max(min(a[0]*val+f[0],255),0), \
gyorgyf@37:                         max(min(a[1]*val+f[1],255),0), \
gyorgyf@37:                         max(min(a[2]*val+f[2],255),0), \
gyorgyf@37:                         max(min(a[3]*val+f[3],255),0) )
gyorgyf@37:             surff.set_at( (j,i), color )
gyorgyf@37:     surf.blit(surff, clip)
gyorgyf@37: 
gyorgyf@37: 
gyorgyf@37: