-
Notifications
You must be signed in to change notification settings - Fork 0
/
screenshot.py
38 lines (32 loc) · 873 Bytes
/
screenshot.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import numpy as np
import subprocess
import pyautogui
import psutil
import time
import cv2
def takeScreenShot(imageName, low, high):
ss = pyautogui.screenshot()
img = np.array(ss)
img = img[:, :, ::-1].copy()
img = img[low[0]:low[0] + high[0], low[1]:low[1] + high[1]]
cv2.imwrite("./Images/{}.png".format(imageName), img)
def runMarchingCubesProgram(cubeSize):
subprocess.Popen(["./CG2", "bun_zipper.xyz", "lut.txt", "{0:2f}".format(value)])
def closeMarchingCubesProgram():
for proc in psutil.process_iter():
if proc.name() == "CG2":
proc.kill()
if __name__ == "__main__":
value = 1.0
decay = 2
i = 68
while value > 0.01:
value = 1 * pow(1 - 0.06, i)
runMarchingCubesProgram(value)
time.sleep(20)
takeScreenShot(str(i), (80, 80), (630, 650))
time.sleep(0.5)
closeMarchingCubesProgram()
print("{0:2f}".format(value))
i += 1
print(i)