forked from charlottejiminnam/StrokeCollaborativeDrawing
-
Notifications
You must be signed in to change notification settings - Fork 0
/
demoUI.py
42 lines (40 loc) · 2.07 KB
/
demoUI.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
39
40
41
42
import argparse
import CapUI.MousePainter as MP
from CapUI.utils import misc
from lmser import stroke
parser = argparse.ArgumentParser()
# Evaluation input filepaths
parser.add_argument('--save_file_name', type=str, help='set output file name for original input',
default='mouse_deltas.npy')
parser.add_argument('--rdp_file_name', type=str, help='set output file name for rdp', default='rdp_deltas.npy')
parser.add_argument('--ai_file_name', type=str, help='set output file name for AI drawing', default='ai_deltas.npy')
args = parser.parse_args()
if __name__ == "__main__":
painter = MP.MousePainter(args)
while not painter.exit:
painter.run()
# self.current_frame_index == 1 : Mouse Drawing -> RDP
# self.current_frame_index == 2 : RDP algorithm -> AI
# self.current_frame_index == 0 : AI -> Mouse Drawing
if painter.current_frame_index == 0:
# print("wait for drawing reconstruction")
# print(painter.save_file_name)
painter.reflect_ai()
painter.load_and_reconstruct(filename=painter.save_file_name)
# painter.deltas_draw = painter.load_and_reconstruct(filename=painter.save_file_name)
elif painter.current_frame_index == 1:
# print("wait for RDP")
misc.rdp_final(painter.save_file_name, painter.rdp_file_name)
# print(painter.rdp_file_name)
painter.load_and_reconstruct(filename=painter.rdp_file_name)
# painter.deltas_rdp = painter.load_and_reconstruct(filename=painter.rdp_file_name)
else:
# print("wait for AI")
# print(painter.ai_file_name)
misc.npy2npz(painter.rdp_file_name, painter.ai_file_name)
stroke.run(painter.ai_index, misc.just_name(painter.ai_file_name))
painter.load_and_reconstruct(filename=painter.ai_file_name)
# painter.deltas_ai = painter.load_and_reconstruct(filename=painter.ai_file_name)
painter.ai_index = painter.ai_index + 1
painter.running = True
# print("Loop termination check")