[Author Prev][Author Next][Thread Prev][Thread Next][Author Index][Thread Index]

[pygame] QuadMap



Here's a program that will map points from one quadrilateral to another quadrilateral:

import pygame, math

WIDTH   = 800
HEIGHT  = 600

BLACK   = (  0,   0,   0, 255)
WHITE   = (255, 255, 255, 255)
RED     = (255,   0,   0, 255)
YELLOW  = (255, 255,   0, 255)
BLUE    = (  0,   0, 255, 255)
GREEN   = (  0, 255,   0, 255)

class Quad(object):
    data = None
    point = None
    def __init__(self, p1, p2, p3, p4):
        self.p1 = list(p1)
        self.p2 = list(p2)
        self.p3 = list(p3)
        self.p4 = list(p4)

    def Render(self, bg):
        pygame.draw.line(bg, RED,    self.p1, self.p2, 1)
        pygame.draw.line(bg, YELLOW, self.p2, self.p3, 1)
        pygame.draw.line(bg, BLUE,   self.p3, self.p4, 1)
        pygame.draw.line(bg, GREEN,  self.p4, self.p1, 1)
        if self.point:
pygame.draw.rect(bg, BLACK, [self.point[0]-2, self.point[1]-2, 4, 4], 0)

    def SetPoint(self, offset):
        if offset == None:
            self.point = None
            return
        s, t = offset
        x1, y1 = self.p1
        x2, y2 = self.p2
        x3, y3 = self.p4
        x4, y4 = self.p3
        xa = x1 + t * (x2 - x1)
        ya = y1 + t * (y2 - y1)
        xb = x3 + t * (x4 - x3)
        yb = y3 + t * (y4 - y3)
        X = xa + s * (xb - xa)
        Y = ya + s * (yb - ya)
        self.point = X, Y

    def HitTest(self, pos):
        x, y = float(pos[0]), float(pos[1])
        x1, y1 = self.p1
        x2, y2 = self.p2
        x3, y3 = self.p4
        x4, y4 = self.p3

        Ax = x2 - x1
        Ay = y2 - y1
        Bx = x4 - x3
        By = y4 - y3
        Cx = x3 - x1
        Cy = y3 - y1
        Dx = x  - x1
        Dy = y  - y1
        Ex = Bx - Ax
        Ey = By - Ay

        a = -Ax * Ey + Ay * Ex
        b = Ey * Dx - Dy * Ex + Ay * Cx - Ax * Cy
        c = Dx * Cy - Dy * Cx

        det = b * b - 4 * a * c
        if det >= 0:
            if abs(a) < 0.001:
                t = -c / float(b)
            else:
                t = (-b - math.sqrt(det)) / float(2 * a)
            denom = (Cx + Ex * t)
            if denom > 0.01:
                s = (Dx - Ax * t) / float(denom)
            else:
                s = (Dy - Ay * t) / float(Cy + Ey * t)

            if (t >= 0) and (t <= 1) and  (s >= 0) and (s <= 1):
                return s, t


class Square(Quad):
    def __init__(self, posx, posy, width, height):
        p1 = posx, posy
        p2 = posx + width, posy
        p3 = posx + width, posy + height
        p4 = posx, posy + height
        super(Square, self).__init__(p1, p2, p3, p4)


def main():
    pygame.init()
    bg = pygame.display.set_mode([WIDTH, HEIGHT], 0, 32)
    quit = 0
    data = []
    data.append( Quad([ 20,  50], [120,  30], [150, 110], [ 10, 130]) )
    data.append( Square(200, 20, 100, 100) )
    data.append( Quad([210, 170], [250, 260], [100, 240], [120, 150]) )
    while not quit:
        bg.fill(WHITE)
        for quad in data:
            quad.Render(bg)
        pygame.display.flip()
        pygame.event.pump()
        moved = 0
        for e in pygame.event.get():
            if e.type == pygame.QUIT:
                quit = 1
                break
            elif e.type == pygame.MOUSEMOTION:
                if not moved:
                    moved = 1
                    for quad in data:
                        offset = quad.HitTest(e.pos)
                        if offset:
                            break
                    for quad in data:
                        quad.SetPoint(offset)
    pygame.quit()


main()