def test_can_unset_attributes_evaluated_values(self):
"""Because the content of pt is changing per iteration, the left of
square should change as well (top should change due to p in people)."""
content_list = ["hello world", "people"]
square_lefts = [] # stores left values per iteration
square_tops = []
pt = PTL("pt", "Arial", 15, "Black", left=NA(0), top=NA(0))
square = ColorLayer("square",
content="Red",
left=AA(SA("pt.right")),
top=SA("pt.bottom"),
width=NA(20),
height=NA(20))
bg = ColorBackgroundLayer("bg", content="White")
temp = Template("temp",
pt,
square,
bg,
left=NA(0),
top=NA(0),
height=NA(100),
width=NA(100))
for i, content in enumerate(content_list):
pt.pre_render = None
pt.content = content
temp.unset_bounds_and_attributes()
temp.update_bounds()
temp.render().save(
filename=
f"tests/images/{i}_test_can_unset_attributes_evaluated_values.png"
)
square_lefts.append(temp.get_layer("square")["left"])
square_tops.append(temp.get_layer("square")["top"])
assert square_lefts[0] != square_lefts[1]
assert square_tops[0] != square_tops[1]
def main():
RESOURCE_DIR = join(os.getcwd(), "resources")
SETS = join(RESOURCE_DIR, "card_data", "sets.json")
TEST_DIR = "test_images"
MPLANTIN = join(RESOURCE_DIR, "fonts", "MPlantin.ttf")
BELEREN_SC = join(RESOURCE_DIR, "fonts", "Beleren_small_caps.ttf")
BELEREN = join(RESOURCE_DIR, "fonts", "Jace_Beleren_bold.ttf")
MPLANTIN_BOLD = join(RESOURCE_DIR, "fonts", "MPlantin_bold.ttf")
MPLANTIN_ITAL = join(RESOURCE_DIR, "fonts", "MPlantin_italic.ttf")
RELAY = join(RESOURCE_DIR, "fonts", "Relay_medium.ttf")
FC = "White"
FLAVOR_SPLIT_COLOR = "RGBA(255, 255, 255, 0.6)"
FLAVOR_SPLIT_OFFSET = 40
if os.path.isfile(SETS):
with open(SETS, "r") as f:
sets = json.load(f)
else:
raise ValueError("sets.json not found.")
myset = "WAR"
# JSON = join(RESOURCE_DIR, "card_data", f"{myset}.json")
# JSON = join(RESOURCE_DIR, "card_data", f"mardu_aristocrats_M20.json")
# TOKENS = join(RESOURCE_DIR, "card_data", f"tokens.json")
# ANGELS = join(RESOURCE_DIR, "card_data", f"mardu_angels_M20.json")
JSON = join(RESOURCE_DIR, "set_data", f"{myset}.json")
# SULTAI_FLASH = join(RESOURCE_DIR, "card_data", f"sultai_flash.json")
SAVE_LOCATION = myset
# SAVE_LOCATION = "sultai_flash"
# JSON = SULTAI_FLASH
# JSON = TOKENS
"""make directory in art"""
# if not os.path.isdir(join(RESOURCE_DIR, "art", myset)): # dir does not exist
# os.mkdir(join(RESOURCE_DIR, "art", myset))
"""make directory in render dir"""
if not os.path.isdir(join("test_images", "all_render",
SAVE_LOCATION)): # dir does not exist
os.mkdir(join("test_images", "all_render", SAVE_LOCATION))
"""load cards"""
with open(JSON, "r") as f:
cards = json.load(f)
"""load tokens"""
# with open(TOKENS, "r") as f:
# tokens = json.load(f)
BORDER = 45
RULES_BORDER = 52
HEIGHT = 1050
WIDTH = 752
SET_DOT_LANG_WIDTH = 5
INFO_SIZE = 18
# FONT_SIZE = 40
NAME_SIZE = 38
TYPE_SIZE = 33
PT_LOYAL_SIZE = 40
RULES_TEXT_SIZE = 25
la = AA(MA(SA("language.right"), SA("number.right")), NA(3))
lmiddle = AA(
NA(WIDTH / 2),
DivA(AA(SA("artist_brush.width"), NA(3), SA("artist.width")),
NA(2),
negative=True))
art_layers = {
"bg":
ColorBackgroundLayer("bg", content=Color("Black")),
"art":
FillIL("art",
order=-5,
XP50=NA(WIDTH / 2),
YP50=NA(HEIGHT / 2),
width=NA(WIDTH),
height=NA(HEIGHT)),
"shadow1":
ColorLayer("shadow1",
content="RGBA(0,0,0,0.4)",
left=NA(0),
width=NA(WIDTH),
top=NA(0),
bottom=SA("shadow2.top")),
"shadow2":
GradL("shadow2",
start="RGBA(0,0,0,0.4)",
end="RGBA(0,0,0,0.0)",
left=NA(0),
width=NA(WIDTH),
top=SA("name.bottom"),
height=NA(200)),
"shadow3":
GradL("shadow3",
start="RGBA(0,0,0,0.0)",
end="RGBA(0,0,0,0.7)",
left=NA(0),
width=NA(WIDTH),
bottom=SA("shadow4.top"),
top=AA(SA("type.cap"), NA(-150))),
"shadow4":
ColorLayer("shadow4",
content="RGBA(0,0,0,0.7)",
left=NA(0),
width=NA(WIDTH),
bottom=NA(HEIGHT + 1),
top=SA("rules.YP70"))
}
no_content_reset = {
"dot":
PTL("dot",
RELAY,
25,
FC,
content=".",
left=AA(SA("set.right"), NA(SET_DOT_LANG_WIDTH)),
ycenter=SA("set.ycenter")),
"language":
PTL("language",
RELAY,
INFO_SIZE,
FC,
content="EN",
left=AA(SA("dot.right"), NA(SET_DOT_LANG_WIDTH)),
base=NA(HEIGHT - BORDER)),
"artist_brush":
ResizeIL("artist_brush",
content=join(RESOURCE_DIR, "artist_brush_white.png"),
width=NA(20),
left=lmiddle,
height=SA("set.height"),
bottom=NA(HEIGHT - BORDER)),
"copyright":
PTL("copyright",
MPLANTIN,
INFO_SIZE - 5,
FC,
right=NA(WIDTH - BORDER),
bottom=SA("set.bottom")),
}
layers = {
"name":
PTL("name", BELEREN, NAME_SIZE, FC, left=NA(BORDER), base=NA(80)),
"type":
PTL("type",
BELEREN,
TYPE_SIZE,
FC,
left=NA(BORDER),
base=AA(SA("rules.top"), NA(-10))),
"PT":
PTL("PT",
BELEREN,
PT_LOYAL_SIZE,
FC,
right=NA(WIDTH - BORDER),
base=SA("set.base")),
"loyalty":
PTL("loyalty",
BELEREN,
PT_LOYAL_SIZE,
FC,
right=NA(WIDTH - BORDER),
base=SA("set.base")),
"set":
PTL("set",
RELAY,
INFO_SIZE,
FC,
left=NA(BORDER),
base=NA(HEIGHT - BORDER)),
"number":
PTL("number",
RELAY,
INFO_SIZE,
FC,
left=NA(BORDER),
base=AA(SA("set.cap"), NA(-3))),
"rarity":
PTL("rarity",
RELAY,
INFO_SIZE,
FC,
left=SA("artist_brush.left"),
base=SA("number.base")),
"artist":
PTL("artist",
BELEREN_SC,
INFO_SIZE,
FC,
left=AA(SA("artist_brush.right"), NA(3)),
base=SA("set.base")),
"mana_cost":
ManaCost("mana_cost",
font=BELEREN,
font_size=NAME_SIZE,
font_color=FC,
right=NA(WIDTH - BORDER),
bottom=AA(SA("name.base"), NA(4))),
"rules":
RulesText("rules",
MPLANTIN,
MPLANTIN_ITAL,
RULES_TEXT_SIZE,
FC,
RULES_TEXT_SIZE - 4,
left=NA(RULES_BORDER),
right=NA(WIDTH - RULES_BORDER),
bottom=AA(SA("PT.bottom"), NA(-PT_LOYAL_SIZE), NA(-5))),
# "flavor_split": ColorLayer("flavor_split", left=NA(RULES_BORDER + FLAVOR_SPLIT_OFFSET),
# right=NA(WIDTH - RULES_BORDER - FLAVOR_SPLIT_OFFSET), height=NA(2),
# bottom=AA(SA("flavor.top"), NA(-5))),
# "flavor": RulesText("flavor", MPLANTIN, MPLANTIN_ITAL, RULES_TEXT_SIZE, FC,
# RULES_TEXT_SIZE - 4, left=NA(RULES_BORDER), right=NA(WIDTH-RULES_BORDER),
# bottom=AA(SA("PT.bottom"), NA(-PT_LOYAL_SIZE), NA(-5))),
}
text_template = Template("text_temp",
*layers.values(),
*no_content_reset.values(),
left=NA(0),
width=NA(WIDTH),
top=NA(0),
height=NA(HEIGHT))
art_template = Template("art_temp",
*art_layers.values(),
order=-5,
left=NA(0),
width=NA(WIDTH),
top=NA(0),
height=NA(HEIGHT))
name = text_template.get_layer("name")
with_art = list(os.listdir(join(RESOURCE_DIR, "art", myset)))
cards = [c for c in cards if f"{c['name']}_{c['id']}.jpg" in with_art]
cards = [c for c in cards if c['name'] == "Nicol Bolas, Dragon-God"]
print(len(cards))
# no_content_reset["copyright"].content = f"™ & © {datetime.now().year} Wizards of the Coast"
# no_content_reset["copyright"].content = f"™ & © {datetime.now().year} WOTC"
loga = math.ceil(math.log10(len(cards)))
temp = Template("template",
text_template,
art_template,
left=NA(0),
width=NA(WIDTH),
top=NA(0),
height=NA(HEIGHT))
temp.mana_image_format = "svg"
temp.resource_dir = RESOURCE_DIR
max_card_length = max(len(c['name']) for c in cards)
row = f"| {{:0{loga}}}/{{:0{loga}}} | {{}} | {{}} | {{:07.3f}} |"
total = 0
for i, card in enumerate(sorted(cards, key=lambda x: x['name'])):
start_time = time.time()
for shadow in ["shadow3", "shadow4"]:
l = temp.get_layer(shadow)
l.pre_render = None
reset_layers = list(layers) + ["art"]
for name in reset_layers:
layer = temp.get_layer(name)
layer.unset_content_and_pre_render()
# layer.content = None
# layer.pre_render = None
if layer.name in card:
layer.content = card[layer.name]
temp.get_layer("artist_brush").pre_shadow = None
if "Creature" in card["types"]:
layers["PT"].content = f"{card['power']}/{card['toughness']}"
# print(card)
# print(repr(temp.get_layer("name").content))
count = 999
sset = [s for s in sets if s['code'] == card['set']]
if len(sset) == 1:
count = sset[0]["count"]
number = card['number'].upper().zfill(3)
layers["number"].content = f"{number}/{count:03}"
rarity_colors = {
"M": "#D15003",
"R": "#DFBD6C",
"U": "#C8C8C8",
"C": FC,
"L": FC,
"T": FC,
}
rarity = card["rarity"][0].upper()
layers["rarity"].color = rarity_colors[rarity]
layers["rarity"].content = rarity
rules = ""
text_to_use = "text"
if card[text_to_use] is not None:
rules = card[text_to_use]
if card["flavor"] is not None:
flavor = "\n".join(
[f"<i>{f}</i>" for f in card['flavor'].split('\n')])
if rules == "":
rules = flavor
else:
rules += "\n" + flavor
if rules != "":
temp.get_layer("rules").content = rules
art_path = join(RESOURCE_DIR, "art", card['set'], f"{card['name']}_{card['id']}.jpg") \
.replace("//", "__")
temp.get_layer("art").content = art_path if os.path.isfile(
art_path) else None
temp.update_bounds()
render_bg = temp.get_layer("art_temp").render()
image = render_bg.clone()
if temp.get_layer("art").content is not None:
render_text_shadow = temp.get_layer("text_temp").shadow(
-4, 4, sigma=2, radius=4, color="Black")
image.composite(render_text_shadow, left=0, top=0)
render_text = temp.get_layer("text_temp").render()
render_text.save(filename="test.png")
image.composite(render_text, left=0, top=0)
# xb = Bounds(start=BORDER - 10, end=WIDTH - BORDER + 10)
# yb = Bounds(start=temp.get_layer("type")["top"] - 10,
# end=temp.get_layer("rules")['bottom'] + 10)
# exp = render_bg.export_pixels(x=int(xb['start']), y=int(yb['start']),
# width=int(xb['full']), height=int(yb['full']))
# with Image(width=int(xb['full']), height=int(yb['full']),
# background=Color("Transparent")) as blur_image:
# blur_image.import_pixels(width=int(xb['full']), height=int(yb['full']),
# channel_map="RGBA", storage="char", data=exp)
# with Image(width=blur_image.width, height=blur_image.height,
# background=Color("RGBA(0,0,0,0.2)")) as dark:
# blur_image.composite(dark, 0, 0)
# blur_image.blur(radius=10, sigma=5)
# with Image(width=blur_image.width, height=blur_image.height,
# background=Color("Black")) as mask:
# with Drawing() as ctx:
# ctx.fill_color = Color("White")
# ctx.rectangle(0, 0, width=mask.width,
# height=mask.height, radius=15)
# ctx(mask)
# apply_mask(blur_image, mask)
# image.composite(blur_image, left=int(xb['start']), top=int(yb['start']))
image.save(filename=join(
"test_images", "all_render", f"{SAVE_LOCATION}",
f"{card['name']}_{card['id']}.jpg").replace("//", "__"))
# image.save(filename=join("test_images", "all_render", "Printing", f"{SAVE_LOCATION}", f"{card['name']}_{card['id']}.jpg").replace("//", "__"))
# image.save(filename=join("test_images", "all_render", "Printing", f"{SAVE_LOCATION}", f"{card['name']}.jpg").replace("//", "__"))
temp.unset_bounds_and_attributes()
delta = time.time() - start_time
total += delta
if delta < .250:
color = "green"
elif delta < .500:
color = "yellow"
else:
color = "red"
print(f"\r{row}".format(
i,
len(cards) - 1,
colored(f"{card['name']: <{max_card_length}}", color),
colored(f"{delta:03.3f}", color), total))