77 lines
1.8 KiB
Python
77 lines
1.8 KiB
Python
|
import math
|
||
|
|
from pynput import mouse
|
||
|
|
from pynput.mouse import Button, Controller
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
ct = Controller()
|
||
|
|
|
||
|
|
# mouse position
|
||
|
|
start_pos = (500, 200)
|
||
|
|
ct.position = start_pos
|
||
|
|
|
||
|
|
# displacement margin (px)
|
||
|
|
d_margin = 2
|
||
|
|
|
||
|
|
# hexagon parameters
|
||
|
|
centerX=400
|
||
|
|
centerY=300
|
||
|
|
radius=200
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def hexa_points(cx, cy, r):
|
||
|
|
points=[None]*6
|
||
|
|
points[0]=(cx+r*math.cos(0), cy+r*math.sin(0))
|
||
|
|
for i in range(1,6):
|
||
|
|
points[i]=( int(cx+r*math.cos(i*2*math.pi/6)), int(cy+r*math.sin(i*2*math.pi/6)) )
|
||
|
|
return points
|
||
|
|
|
||
|
|
|
||
|
|
hexagon = hexa_points(centerX, centerY, radius)
|
||
|
|
# print(hexagon)
|
||
|
|
|
||
|
|
|
||
|
|
def update_position(p, pi, pj):
|
||
|
|
relative=(pj[0]-pi[0], pj[1]-pi[1])
|
||
|
|
normal=(relative[1],-relative[0])
|
||
|
|
magnitude=math.sqrt((normal[0]*normal[0])+(normal[1]*normal[1]))
|
||
|
|
normal_unit=(normal[0]/magnitude, normal[1]/magnitude)
|
||
|
|
displacement=((p[0]-pi[0])*normal_unit[0])+((p[1]-pi[1])*normal_unit[1])
|
||
|
|
# print(displacement)
|
||
|
|
if (displacement > d_margin):
|
||
|
|
return ( int(p[0]-normal_unit[0]*displacement), int(p[1]-normal_unit[1]*displacement) )
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
def keep_in_hexagon(point, hexagon):
|
||
|
|
for i in range(0,6):
|
||
|
|
j=i+1
|
||
|
|
if j==6:
|
||
|
|
j=0
|
||
|
|
# print("point: "+str(point)+" pi: "+str(hexagon[i])+" pj: "+str(hexagon[j]))
|
||
|
|
updated_point=update_position(point, hexagon[i], hexagon[j])
|
||
|
|
if updated_point:
|
||
|
|
print("[OOB] "+str(point)+' => '+str(updated_point))
|
||
|
|
ct.position = updated_point
|
||
|
|
|
||
|
|
|
||
|
|
def on_move(x, y):
|
||
|
|
keep_in_hexagon((x, y), hexagon)
|
||
|
|
|
||
|
|
|
||
|
|
# click to stop listening
|
||
|
|
def on_click(x, y, button, pressed):
|
||
|
|
print('{0} at {1}'.format(
|
||
|
|
'Pressed' if pressed else 'Released',
|
||
|
|
(x, y)))
|
||
|
|
if not pressed:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
with mouse.Listener(
|
||
|
|
on_move=on_move,
|
||
|
|
on_click=on_click) as listener:
|
||
|
|
listener.join()
|