John McCardle
5499aa5e49
Rocks, lives, and missiles are identified by their rectangles from cv2.matchTemplate. Clusters from SIFT are checked against these detected rectangles. The remaining objects (like the rotated ship) are classified as "mysteries". These will be target for further analysis (like ship angle determination), or we can just shoot at them :)
19 lines
595 B
Python
19 lines
595 B
Python
from math import sqrt
|
|
|
|
def squared_distance(vec1, vec2):
|
|
"""returns distance-squared between two x, y point tuples"""
|
|
return (vec1[0] - vec2[0])**2 + (vec1[1] - vec2[1])**2
|
|
|
|
def rect_radius_squared(w, h):
|
|
"""Returns the radius^2 of the circle inscribed in a rectangle of w * h"""
|
|
return (w/2)**2 + (h/2)**2
|
|
|
|
def point_in_rect(pt, rect):
|
|
"""Returns True if the (x,y) point is within the ((x,y),(w,h)) rectangle."""
|
|
px, py = pt
|
|
tl, wh = rect
|
|
rx, ry = tl
|
|
rw, rh = wh
|
|
rx2 = rx + rw
|
|
ry2 = ry + rh
|
|
return all([px >= rx, py >= ry, px <= rx2, py <= ry2])
|