光标画图程序

【CSDN平台停更，请移往知乎或者我的blog】
https://hk-shao.github.io/
https://www.zhihu.com/people/shao_feng

写这个程序的缘由是现在疫情期间我们学生在家用在线办公软件上网络课，老师在投屏时学生可以在老师屏幕上“注释”（做记号，写字，画画等），这样可以很好的与老师互动。但是“注释”并不能显示图片，为了达成我的一个骚主意，在老师屏幕上画一个卢本伟，我面向百度编程写了这个Python程序。

视频

水了两个视频：

1. 当你在老师网络课的屏幕上画画

2. 我竟然用windows自带画图工具画了一个炮姐！

以后决定干啥都水个视频

源代码

项目地址：https://github.com/HK-SHAO/cursor_painter

其中dist文件夹有exe程序可以直接运行，不需要安装Python等环境。

其实代码很简单，这也多亏了Python超简单的语法和强大的库（自从用了python以后编程能力节节败退，Lua，Java，C已经忘完了）。

from pynput.mouse import Button, Controller
from pynput import keyboard
from PIL import Image, ImageFilter, ImageGrab
import time

mouse = Controller()
p = None
img = None
isB = False

def click(x, y):
    mouse.position = (x, y)
    mouse.click(Button.left, 1)

def draw(s, pt = p, n = 1, m = '00', r = 1, t = 220, st = 0.01):
    global img
    img = Image.open(s)
    img = img.resize((int(img.width*n), int(img.height*n)))
    pt = p

    if m[0] == '0':
        img = img.filter(ImageFilter.SMOOTH).filter(ImageFilter.CONTOUR).convert('L')
    else:
        img = img.convert('L')

    pixels = img.load()
    for x in range(img.width):
        for y in range(img.height):

            if isB:
                print('已中止')
                return

            pix = pixels[x, y]
            if r*pix < r*t:
                if m[1] == '0' or (x%2==0 and y%2==0):
                    click(pt[0]+x, pt[1]+y)
                    time.sleep(st)

def color(s, n = 1, st = 0.01):
    global img
    img = Image.open(s)
    img = img.resize((int(img.width*n), int(img.height*n))).convert('RGB')

    colors, pos = [], []
    print("将光标移动到颜色按钮，回车添加，输入s并回车从当前光标位置开始绘制，输入q并回车退出")
    while True:
        input_str = input()
        if input_str == 's':
            mp = mouse.position
            break
        elif input_str == 'q':
            return
        mp = mouse.position
        c = ImageGrab.grab().getpixel(mp)
        pos.append(mp)
        colors.append(c)
        print('{} {}'.format(c, mp))

    pixels = img.load()
    for x in range(img.width):
        for y in range(img.height):

            if isB:
                print('已中止')
                return

            pix = pixels[x, y]
            if (pix[0]+pix[1]+pix[2])/3 < 240:
                ds = []
                for c in colors:
                    d = ((pix[0]-c[0])**2+(pix[1]-c[1])**2+(pix[2]-c[2])**2)**0.5
                    ds.append(d)
                index = ds.index(min(ds))

                click(pos[index][0], pos[index][1])
                click(mp[0]+x, mp[1]+y)
                time.sleep(st)

def on_release(key):
    if key == keyboard.Key.esc:
        global isB
        isB = True

listener = keyboard.Listener(on_release=on_release)
listener.start()

print("""对象：
    p：光标所在位置             mouse：鼠标
    time：时间                 img：图像
命令：
    draw(s图片地址，pt点坐标，n缩放倍数，m模式，r是否反转，t下笔阈值，st延迟时间)：绘制黑白图片
        例如 draw('img.png') 或 draw('img.png', pt=[200, 300], n=1, m='00', r=-1, t=220, st=0)
    color(s图片地址，n缩放倍数，st延迟时间)：绘制彩色图片
        例如 color('img.png', n=0.1)
    click(横坐标，纵坐标)：点击某位置
        例如 click(200, 300)
模式：
    第1个数字：             第2个数字：
        0：去噪边缘灰度化        0：正常
        1：只进行灰度化          1：稀疏
注意：
    1.使用 CTRL+C 退出程序，绘制中按下ESC键中止
    2.还可执行其它的python命令，如 img.show() 显示图片
""")

while True:
    isB = False
    i = input('>>> ')
    p = mouse.position
    try:
        o = eval(i)
        if o != None:
            print(o)
    except:
        print('error')