auto-accept/restrict_cursor.py

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()