-
Notifications
You must be signed in to change notification settings - Fork 169
/
Copy pathbase.py
198 lines (155 loc) · 7.58 KB
/
base.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
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
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
import os
import sys
import json
import codecs
import plistlib
import textwrap
from jinja2 import Template
from glue.helpers import round_up, nearest_fration
from glue import __version__
class BaseFormat(object):
extension = None
build_per_ratio = False
def __init__(self, sprite):
self.sprite = sprite
def output_dir(self, *args, **kwargs):
return self.sprite.config['{0}_dir'.format(self.format_label)]
def output_filename(self, ratio=None, *args, **kwargs):
if self.build_per_ratio:
if ratio is None:
raise AttributeError("Format {0} output_filename requires a ratio.".format(self.__class__))
ratio_suffix = '@%.1fx' % ratio if int(ratio) != ratio else '@%ix' % ratio
if ratio_suffix == '@1x':
ratio_suffix = ''
return '{0}{1}'.format(self.sprite.name, ratio_suffix)
return self.sprite.name
def output_path(self, *args, **kwargs):
return os.path.join(self.output_dir(*args, **kwargs), '{0}.{1}'.format(self.output_filename(*args, **kwargs), self.extension))
def build(self):
if self.build_per_ratio:
for ratio in self.sprite.config['ratios']:
self.save(ratio=ratio)
else:
self.save()
def save(self, *args, **kwargs):
raise NotImplementedError
def needs_rebuild(self):
return True
def validate(self):
pass
@property
def format_label(self):
from glue.formats import formats
return dict((v,k) for k, v in formats.iteritems())[self.__class__]
@classmethod
def populate_argument_parser(cls, parser):
pass
@classmethod
def apply_parser_contraints(cls, parser, options):
pass
def fix_windows_path(self, path):
if os.name == 'nt':
path = path.replace('\\', '/')
return path
class BaseTextFormat(BaseFormat):
def get_context(self, *args, **kwargs):
sprite_path = os.path.relpath(self.sprite.sprite_path(), self.output_dir())
sprite_path = self.fix_windows_path(sprite_path)
context = {'version': __version__,
'hash': self.sprite.hash,
'name': self.sprite.name,
'sprite_path': sprite_path,
'sprite_filename': os.path.basename(sprite_path),
'width': round_up(self.sprite.canvas_size[0] / self.sprite.max_ratio),
'height': round_up(self.sprite.canvas_size[1] / self.sprite.max_ratio),
'images': [],
'ratios': {}}
for i, img in enumerate(self.sprite.images):
base_x = img.x * -1 - img.margin[3] * self.sprite.max_ratio
base_y = img.y * -1 - img.margin[0] * self.sprite.max_ratio
base_abs_x = img.x + img.margin[3] * self.sprite.max_ratio
base_abs_y = img.y + img.margin[0] * self.sprite.max_ratio
image = dict(filename=img.filename,
last=i == len(self.sprite.images) - 1,
x=round_up(base_x / self.sprite.max_ratio),
y=round_up(base_y / self.sprite.max_ratio),
abs_x=round_up(base_abs_x / self.sprite.max_ratio),
abs_y=round_up(base_abs_y / self.sprite.max_ratio),
height=round_up((img.height / self.sprite.max_ratio) + img.padding[0] + img.padding[2]),
width=round_up((img.width / self.sprite.max_ratio) + img.padding[1] + img.padding[3]),
original_width=img.original_width,
original_height=img.original_height,
ratios={})
for r in self.sprite.ratios:
image['ratios'][r] = dict(filename=img.filename,
last=i == len(self.sprite.images) - 1,
x=round_up(base_x / self.sprite.max_ratio * r),
y=round_up(base_y / self.sprite.max_ratio * r),
abs_x=round_up(base_abs_x / self.sprite.max_ratio * r),
abs_y=round_up(base_abs_y / self.sprite.max_ratio * r),
height=round_up((img.height + img.padding[0] + img.padding[2]) / self.sprite.max_ratio * r),
width=round_up((img.width + img.padding[1] + img.padding[3]) / self.sprite.max_ratio * r))
context['images'].append(image)
# Ratios
for r in self.sprite.ratios:
ratio_sprite_path = os.path.relpath(self.sprite.sprite_path(ratio=r), self.output_dir())
ratio_sprite_path = self.fix_windows_path(ratio_sprite_path)
context['ratios'][r] = dict(ratio=r,
fraction=nearest_fration(r),
sprite_path=ratio_sprite_path,
sprite_filename=os.path.basename(ratio_sprite_path),
width=round_up(self.sprite.canvas_size[0] / self.sprite.max_ratio * r),
height=round_up(self.sprite.canvas_size[1] / self.sprite.max_ratio * r))
return context
def render(self, *args, **kwargs):
raise NotImplementedError
def save(self, *args, **kwargs):
# Create the destination directory if required
if not os.path.exists(self.output_dir(*args, **kwargs)):
os.makedirs(self.output_dir(*args, **kwargs))
with codecs.open(self.output_path(*args, **kwargs), 'w', 'utf-8') as f:
f.write(self.render(*args, **kwargs))
class BaseJSONFormat(BaseTextFormat):
meta_key = 'meta'
def needs_rebuild(self):
for ratio in self.sprite.config['ratios']:
json_path = self.output_path(ratio)
if os.path.exists(json_path):
with codecs.open(json_path, 'r', 'utf-8') as f:
try:
data = json.loads(f.read())
assert data[self.meta_key]['hash'] == self.sprite.hash
except Exception:
continue
return True
return False
def render(self, *args, **kwargs):
return json.dumps(self.get_context(*args, **kwargs), indent=4)
class BasePlistFormat(BaseTextFormat):
meta_key = 'metadata'
def render(self, *args, **kwargs):
context = self.get_context(*args, **kwargs)
if sys.version < '3':
return plistlib.writePlistToString(context)
return plistlib.writePlistToBytes(context).decode('unicode_escape')
def needs_rebuild(self):
for ratio in self.sprite.config['ratios']:
cocos2d_path = self.output_path(ratio)
if os.path.exists(cocos2d_path):
try:
data = plistlib.readPlist(cocos2d_path)
assert data[self.meta_key]['hash'] == self.sprite.hash
except Exception:
continue
return True
return False
class JinjaTextFormat(BaseTextFormat):
template = ''
def render(self, *args, **kwargs):
context = self.get_context(*args, **kwargs)
template = self.template
custom_template_config = '{0}_template'.format(self.format_label)
if self.sprite.config.get(custom_template_config):
with open(self.sprite.config[custom_template_config]) as f:
template = f.read()
return Template(textwrap.dedent(template).strip()).render(**context)