-
Notifications
You must be signed in to change notification settings - Fork 0
/
compress.py
72 lines (57 loc) · 1.61 KB
/
compress.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
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
import sys
import os
import math
import time
import cv2
from XIPLUS03 import *
from XIPLUS04 import *
algolist = {
"auto": [],
"xp03": [[XIPLUS03_encode, [-2, 2, 32, True]], [XIPLUS03_decode, [True]]],
"xp04": [[XIPLUS04_encode, [-2, 2, 32, True]], [XIPLUS04_decode, [True]]],
}
ctype = sys.argv[1]
if ctype == "encode":
algo = sys.argv[2].lower()
finname = sys.argv[3]
foutname = sys.argv[4]
if algo not in algolist:
exit("alog not found")
print("encode by {}".format(algo))
oldsize = os.path.getsize(finname)
print("old size: {}".format(oldsize))
start = time.time()
if algo == "auto":
pass
else:
algolist[algo][0][0](finname, foutname, *algolist[algo][0][1])
print("spend {} s".format(time.time()-start))
newsize = os.path.getsize(foutname)
print("new size: {}".format(newsize))
print("compression ratio: {}".format(oldsize/newsize))
elif ctype == "decode":
algo = sys.argv[2].lower()
finname = sys.argv[3]
foutname = sys.argv[4]
if algo not in algolist:
exit("alog not found")
print("decode by {}".format(algo))
oldsize = os.path.getsize(finname)
print("old size: {}".format(oldsize))
start = time.time()
if algo == "auto":
pass
else:
algolist[algo][1][0](finname, foutname, *algolist[algo][1][1])
print("spend {} s".format(time.time()-start))
newsize = os.path.getsize(foutname)
print("new size: {}".format(newsize))
elif ctype == "error":
fname1 = sys.argv[2]
fname2 = sys.argv[3]
img1 = cv2.imread(fname1, cv2.IMREAD_UNCHANGED)
img2 = cv2.imread(fname2, cv2.IMREAD_UNCHANGED)
diff = img2 - img1
print("error", math.sqrt(np.average(np.square(diff))))
else:
print("nothing to do")