135 lines
3.1 KiB
Python
135 lines
3.1 KiB
Python
import random
|
|
from collections import namedtuple, defaultdict
|
|
from enum import Enum, auto
|
|
from typing import Set
|
|
from math import cos, sin, sqrt, pi
|
|
|
|
import matplotlib.pyplot as plt
|
|
import numpy as np
|
|
|
|
Point = namedtuple('Point', 'x y')
|
|
Vector = namedtuple('Vector', 'x y')
|
|
|
|
|
|
def gen_point(lower: int = 0, upper: int = 10) -> Point:
|
|
a = random.uniform(lower, upper)
|
|
b = random.uniform(lower, upper)
|
|
|
|
#x_i = random.uniform(lower, upper)
|
|
#p_i = Point(x_i, a * x_i + b)
|
|
return Point(a, b)
|
|
|
|
|
|
def display(points: Set[Point], hull: Set[Point]):
|
|
x = [point.x for point in points]
|
|
y = [point.y for point in points]
|
|
|
|
h_x = [point.x for point in hull]
|
|
h_y = [point.y for point in hull]
|
|
|
|
plt.plot(h_x, h_y, 'ro')
|
|
|
|
plt.scatter(x, y)
|
|
plt.show()
|
|
|
|
|
|
def gen_circular_point(lower : int = 0, upper: int = 10, radius: int = 5) -> Point:
|
|
|
|
a = random.uniform(lower, upper) * 2 * pi
|
|
|
|
r = radius * sqrt(random.uniform(lower, upper))
|
|
|
|
x = r * cos(a)
|
|
y = r * sin(a)
|
|
|
|
return Point(x,y)
|
|
|
|
def gen_weird_point(lower : int = 0, upper: int = 10) -> Point:
|
|
x = random.uniform(lower, upper)
|
|
y = x**2
|
|
|
|
if x < 0:
|
|
return Point(random.uniform(x, -x), y)
|
|
return Point(random.uniform(-x, x), y)
|
|
|
|
|
|
def read_and_prep_data(filename):
|
|
|
|
data = open(filename).read()
|
|
lines = data.split('\n')
|
|
data = defaultdict(list)
|
|
|
|
for line in lines[1:]:
|
|
all_vars = line.split("\t\t")
|
|
name, points, time = all_vars
|
|
data[name.strip()].append([points, time[:8]])
|
|
|
|
return data
|
|
|
|
|
|
def gen_graph(data):
|
|
|
|
graham = data['graham']
|
|
quick = data['quick']
|
|
mbch = data['mbch']
|
|
mbch2 = data['mbch2']
|
|
gift = data['gift']
|
|
|
|
graham_x = [p[0] for p in graham]
|
|
graham_y = [p[1] for p in graham]
|
|
|
|
quick_x = [p[0] for p in quick]
|
|
quick_y = [p[1] for p in quick]
|
|
|
|
mbch_x = [p[0] for p in mbch]
|
|
mbch_y = [p[1] for p in mbch]
|
|
|
|
mbch2_x = [p[0] for p in mbch2]
|
|
mbch2_y = [p[1] for p in mbch2]
|
|
|
|
gift_x = [p[0] for p in gift]
|
|
gift_y = [p[1] for p in gift]
|
|
|
|
plt.plot(graham_x, graham_y)
|
|
plt.plot(quick_x, quick_y)
|
|
plt.plot(mbch_x, mbch_y)
|
|
plt.plot(mbch2_x, mbch2_y)
|
|
plt.plot(gift_x, gift_y)
|
|
|
|
plt.legend(['graham', 'quick', 'mbch', 'mbch2', 'gift'], loc='upper left')
|
|
|
|
plt.show()
|
|
|
|
|
|
def gen_triangular_point(left : Point, right : Point, top : Point):
|
|
r1 = random.uniform(0,1)
|
|
r2 = random.uniform(0,1)
|
|
|
|
return Point((1 - sqrt(r1)) * left.x + (sqrt(r1) * (1 - r2)) * right.x + (sqrt(r1) * r2) * top.x,
|
|
(1 - sqrt(r1)) * left.y + (sqrt(r1) * (1 - r2)) * right.y + (sqrt(r1) * r2) * top.y)
|
|
|
|
|
|
def display_line_only(points: Set[Point], slope: float, intercept: float, line_points: Set[Point]):
|
|
x = [point.x for point in points]
|
|
y = [point.y for point in points]
|
|
|
|
plt.scatter(x, y)
|
|
|
|
# Plot a line from slope and intercept
|
|
axes = plt.gca()
|
|
x_vals = np.array(axes.get_xlim())
|
|
y_vals = intercept + slope * x_vals
|
|
|
|
for point in line_points:
|
|
plt.plot(point.x, point.y, 'go')
|
|
|
|
plt.plot(x_vals, y_vals, '--')
|
|
|
|
plt.show()
|
|
|
|
|
|
class Side(Enum):
|
|
ON = auto()
|
|
ABOVE = auto()
|
|
BELOW = auto()
|