NJU_CG/cg_cli.py at master · bliss-bug/NJU_CG · GitHub

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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
#!/usr/bin/env python
# -*- coding:utf-8 -*-

import sys
import os
import cg_algorithms as alg
import numpy as np
from PIL import Image


if __name__ == '__main__':
    input_file = sys.argv[1]
    output_dir = sys.argv[2]
    os.makedirs(output_dir, exist_ok=True)

    item_dict = {}
    pen_color = np.zeros(3, np.uint8)
    width = 0
    height = 0

    with open(input_file, 'r') as fp:
        line = fp.readline()
        while line:
            line = line.strip().split(' ')
            if line[0] == 'resetCanvas':
                width = int(line[1])
                height = int(line[2])
                item_dict = {}
            elif line[0] == 'saveCanvas':
                save_name = line[1]
                canvas = np.zeros([height, width, 3], np.uint8)
                canvas.fill(255)
                for item_type, p_list, algorithm, color in item_dict.values():
                    #pixels=[]
                    if item_type == 'line':
                        pixels = alg.draw_line(p_list, algorithm)

                    elif item_type == 'polygon':
                        #pass
                        pixels = alg.draw_polygon(p_list,algorithm)
                    elif item_type == 'ellipse':
                        #pass
                        pixels = alg.draw_ellipse(p_list)
                    elif item_type == 'curve':
                        #pass
                        pixels = alg.draw_curve(p_list,algorithm)
                    for x, y in pixels:
                        canvas[y, x] = color
                Image.fromarray(canvas).save(os.path.join(output_dir, save_name + '.bmp'), 'bmp')
            elif line[0] == 'setColor':
                pen_color[0] = int(line[1])
                pen_color[1] = int(line[2])
                pen_color[2] = int(line[3])
            elif line[0] == 'drawLine':
                item_id = line[1]
                x0 = int(line[2])
                y0 = int(line[3])
                x1 = int(line[4])
                y1 = int(line[5])
                algorithm = line[6]
                item_dict[item_id] = ['line', [[x0, y0], [x1, y1]], algorithm, np.array(pen_color)]
            elif line[0] == 'drawPolygon':
                item_id = line[1]
                algorithm=line[-1]
                l=[]
                for i in range(2,len(line)-2,2):
                    l.append([int(line[i]),int(line[i+1])])
                item_dict[item_id] = ['polygon',l,algorithm,np.array(pen_color)]
            elif line[0] == 'drawEllipse':
                item_id = line[1]
                x0 = int(line[2])
                y0 = int(line[3])
                x1 = int(line[4])
                y1 = int(line[5])
                if x0>x1:
                    x0,x1=x1,x0
                if y0>y1:
                    y0,y1=y1,y0
                item_dict[item_id] = ['ellipse', [[x0, y0], [x1, y1]], None, np.array(pen_color)]
            elif line[0] == 'drawCurve':
                item_id = line[1]
                algorithm=line[-1]
                t=[]
                for i in range(2,len(line)-2,2):
                    t.append([int(line[i]),int(line[i+1])])
                item_dict[item_id] = ['curve',t,algorithm,np.array(pen_color)]
            elif line[0] == 'translate':
                item_id = line[1]
                dx,dy=int(line[2]),int(line[3])
                item_dict[item_id][1]=alg.translate(item_dict[item_id][1],dx,dy)
            elif line[0] == 'rotate':
                item_id = line[1]
                x,y,r=int(line[2]),int(line[3]),int(line[4])
                item_dict[item_id][1]=alg.rotate(item_dict[item_id][1],x,y,r)
            elif line[0] == 'scale':
                item_id = line[1]
                x,y,s=int(line[2]),int(line[3]),float(line[4])
                item_dict[item_id][1]=alg.scale(item_dict[item_id][1],x,y,s)
            elif line[0] == 'clip':
                item_id = line[1]
                x0 = int(line[2])
                y0 = int(line[3])
                x1 = int(line[4])
                y1 = int(line[5])
                algorithm = line[6]
                res=alg.clip(item_dict[item_id][1],x0,y0,x1,y1,algorithm)
                if res==[]:
                    del item_dict[item_id]
                else:
                    item_dict[item_id][1]=res

            line = fp.readline()