/
Shapes.py
59 lines (53 loc) · 1.39 KB
/
Shapes.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
__author__ = 'Mandar'
import pygame
import numpy as np
import math as m
from pygame import gfxdraw
import Vertices
import copy
class Creates(object):
def __init__(self):
pass
#Using Bressenham's
def line(self, surface, X1, Y1, X2, Y2, color):
x1 = copy.copy(X1)
y1= copy.copy (Y1)
x2 = copy.copy(X2)
y2 = copy.copy(Y2)
gfxdraw.pixel(surface, x1, y1, color)
dx = abs(x2 - x1)
dy = abs(y2 - y1)
if (x2 > x1):
xinc = 1
else:
xinc = -1
if (y2 > y1):
yinc = 1
else:
yinc = -1
x1t = x1
y1t = y1
if (dx > dy):
p = 2 * dy - dx
k = 0
while k <= dx:
x1t += xinc
if p < 0:
p += 2 * dy
else:
y1t += yinc
p += 2 * dy - 2 * dx
gfxdraw.pixel(surface, x1t, y1t, color)
k += 1
else:
p = 2 * dx - dy
k = 0
while k <= dy:
y1t += yinc
if p < 0:
p += 2 * dx
else:
x1t += xinc
p += 2 * dx - 2 * dy
gfxdraw.pixel(surface, x1t, y1t, color)
k += 1