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
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
|
# python import
import gtk, logging, os, struct, StringIO
# atoidejouerbeta import
from atoidejouerbeta.tools import registry, storage
# get application logger
logger = logging.getLogger('atoidejouerbeta')
def get_pixbuf(file_path, max_width, max_height, use_max=False):
# little check
_pixbuf = registry.PixRegistry().get_pix(file_path, max_width, max_height)
# already exist
if _pixbuf is not None:
return _pixbuf
elif os.path.exists(file_path):
pass
else:
return None
# get image info
_info, _width, _height = get_image_info(file_path)
# check content
if _info in ['image/png']:
# update size
_width, _height = compute_width_height(_width, _height,
max_width, max_height, use_max=use_max)
# prepare preview
_pixbuf = gtk.gdk.pixbuf_new_from_file(file_path)
# ..
_pixbuf = _pixbuf.scale_simple(_width, _height,
gtk.gdk.INTERP_BILINEAR)
# udpate registry
registry.PixRegistry().set_pix(file_path, max_width, max_height,
_pixbuf)
# ..
return _pixbuf
else:
# TODO log something
return None
def get_sequence_first_graphic(type_, sequence_name, size=None):
# ..
_filename = storage.get_sequence_first_graphic_name(type_, sequence_name)
if _filename is None:
_path = storage.get_image_path('blank', dir_='data')
if type_ == 'graphics':
_path = storage.get_image_path(_filename)
elif type_ == 'sounds':
_path = storage.get_image_path('sound', dir_='data')
else:
_path = storage.get_image_path('blank', dir_='data')
# ensure size
if size is None:
size = (64, 48)
else:
pass
# ..
return get_pixbuf(_path, *size)
def compute_width_height(width, height, max_width, max_height, use_max=False):
# compute ratio
_ratio_scr = max_width / float(max_height)
_ratio_img = width / float(height)
# ..
if width > max_width\
or height > max_height:
if _ratio_img > _ratio_scr:
width = max_width
height = int(max_width / _ratio_img)
elif _ratio_img < _ratio_scr:
width = int(max_height * _ratio_img)
height = max_height
else:
width = max_width
height = max_height
# ..
return width, height
# ..
elif use_max is True:
return max_width, max_height
# ..
else:
return width, height
def get_image_info(path):
"""Tricky method found on Internet that returns the image info from a given
raw image data.
"""
# little check
_info = registry.InfoRegistry().get_info(path)
# already exist
if _info is not None:
return _info
elif os.path.exists(path):
pass
else:
return None, 0, 0
# read file
_f = open(path)
_data = _f.read()
_f.close()
#
size = len(_data)
height = 0
width = 0
content_type = None
# handle GIFs
if (size >= 10) and _data[:6] in ('GIF87a', 'GIF89a'):
# Check to see if content_type is correct
content_type = 'image/gif'
w, h = struct.unpack("<HH", _data[6:10])
width = int(w)
height = int(h)
# See PNG 2. Edition spec (http://www.w3.org/TR/PNG/)
# Bytes 0-7 are below, 4-byte chunk length, then 'IHDR'
# and finally the 4-byte width, height
elif ((size >= 24) and _data.startswith('\211PNG\r\n\032\n')
and (_data[12:16] == 'IHDR')):
content_type = 'image/png'
w, h = struct.unpack(">LL", _data[16:24])
width = int(w)
height = int(h)
# Maybe this is for an older PNG version.
elif (size >= 16) and _data.startswith('\211PNG\r\n\032\n'):
# Check to see if we have the right content type
content_type = 'image/png'
w, h = struct.unpack(">LL", _data[8:16])
width = int(w)
height = int(h)
# handle JPEGs
elif (size >= 2) and _data.startswith('\377\330'):
content_type = 'image/jpeg'
jpeg = StringIO.StringIO(_data)
jpeg.read(2)
b = jpeg.read(1)
try:
while (b and ord(b) != 0xDA):
while (ord(b) != 0xFF): b = jpeg.read(1)
while (ord(b) == 0xFF): b = jpeg.read(1)
if (ord(b) >= 0xC0 and ord(b) <= 0xC3):
jpeg.read(3)
h, w = struct.unpack(">HH", jpeg.read(4))
break
else:
jpeg.read(int(struct.unpack(">H", jpeg.read(2))[0])-2)
b = jpeg.read(1)
width = int(w)
height = int(h)
except struct.error:
pass
except ValueError:
pass
# udpate registry
registry.InfoRegistry().set_info(path, (content_type, width, height))
# ..
return content_type, width, height
|
