pebble/tools/generate_watchface/generate_watchface.py
2025-01-27 11:38:16 -08:00

189 lines
7.1 KiB
Python

# Copyright 2024 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import json
import argparse, os, sys
import shutil
import watch_templates as temps
class WatchGenerator:
def __init__(self):
self._static_defs = []
self._init_func = []
self._update_func = []
self.watch_name = None
self.version = None
self.company_name = None
self.visible_name = None
def h_file_name(self):
return 'pebble_gen_header.h'
def c_file_name(self):
return self.watch_name + '.c'
def defs_file_name(self):
return 'pebble_gen_defs.h'
def generate_c_code(self):
fs = temps.FileScope
res_strings = []
res_strings.append(fs.includes.format(header_name=self.h_file_name()))
res_strings.append('\n'.join(self._static_defs))
res_strings.append(fs.update_function.format(watch_name=self.watch_name, update_content='\n'.join(self._update_func)))
res_strings.append(fs.init_function.format(watch_name=self.watch_name, init_content='\n'.join(self._init_func)))
return '\n'.join(res_strings)
def generate_h_code(self):
return temps.FileScope.header.format(watch_name=self.watch_name, app_version=self.version)
def generate_defs_file(self):
return temps.FileScope.defs_file.format(watch_name=self.watch_name, visible_name=self.visible_name, company_name=self.company_name)
def load_config(self, conf):
self.watch_name = conf["name"]
self.version = conf["friendlyVersion"]
self.company_name = conf["company-name"]
self.visible_name = conf["visible-name"]
for elem in conf["elements"]:
t = elem["type"]
if t == "static-image":
self.parse_static_image(elem)
elif t == "time-text":
self.parse_time_text(elem)
elif t == "hand":
self.parse_hand(elem)
else:
raise "Unknown element type: " + t
def parse_static_image(self, elem):
si = temps.StaticImage
self._static_defs.append(si.static_defs.format(name=elem["name"]))
self._init_func.append(si.init_lines.format(res_def_name=elem["image"]["def-name"],
name=elem["name"]))
def parse_date_format(self, fmt_list):
### parses a json object list into a c strftime format string and required buffer size
buffer_size = 1
fmt_string = ""
for fmt in fmt_list:
# TODO: implement the other operations for strftime
t = fmt["type"]
if t == "string":
fmt_string += fmt.val
buffer_size += len(fmt.val)
elif t == "day-name-abbreviated":
fmt_string += "%a"
buffer_size += 4
elif t == "day-of-month":
fmt_string += "%d"
buffer_size += 2
else:
raise "Unknown date format type: " + t
return buffer_size, fmt_string
def parse_color(self, color_string):
if color_string == "black":
return "GColorBlack"
elif color_string == "white":
return "GColorWhite"
elif color_string == "transparent":
return "GColorTrans"
else:
raise "Unknown color: " + color_string
def parse_font_info(self, font_obj):
f = font_obj["font"]
if f == "gothic":
bold = "modifiers" in font_obj and "bold" in font_obj["modifiers"]
size = font_obj["size"]
return "graphics_text_get_system_font({size}, GFontStyle{style})".format(size=size, style=("Bold" if bold else "Regular"))
else:
raise "Unknown font: " + f
def parse_time_text(self, elem):
tt = temps.TimeText
buffer_len, date_fmt = self.parse_date_format(elem["format"])
color = self.parse_color(elem["color"])
bg_color = self.parse_color(elem["background-color"])
font_func = self.parse_font_info(elem["font"])
self._static_defs.append(tt.static_defs.format(name=elem["name"], buffer_length=buffer_len))
self._init_func.append(tt.init_lines.format(name=elem["name"],
x=elem["position"][0], y=elem["position"][1],
w=elem["size"][0], h=elem["size"][1],
color=color, background_color=bg_color,
font_func=font_func))
self._update_func.append(tt.update_lines.format(name=elem["name"], date_format_string=date_fmt))
def parse_time_element(self, te):
if te == "hour":
return "t.tm_hour / 12"
elif te == "hour24":
return "t.tm_hour / 24"
elif te == "minute":
return "t.tm_min / 60"
elif te == "second":
return "t.tm_sec / 60"
else:
raise "Unknown time element: " + te
def parse_hand(self, elem):
h = temps.Hand
ratio = self.parse_time_element(elem["time-element"])
self._static_defs.append(h.static_defs.format(name=elem["name"]))
self._init_func.append(h.init_lines.format(res_def_name=elem["trans-image"]["def-name"],
name=elem["name"],
pivot_x=elem["pivot"][0], pivot_y=elem["pivot"][1]))
self._update_func.append(h.update_lines.format(name=elem["name"], angle_ratio=ratio))
def generate(watch_config, src_output_dir):
with open(watch_config, 'r') as conf:
config_data = json.loads(conf.read())
gen = WatchGenerator()
gen.load_config(config_data)
if not os.path.isdir(src_output_dir):
os.makedirs(src_output_dir)
with open(os.path.join(src_output_dir, gen.c_file_name()), 'w') as output:
output.write(gen.generate_c_code())
with open(os.path.join(src_output_dir, gen.h_file_name()), 'w') as output:
output.write(gen.generate_h_code())
with open(os.path.join(src_output_dir, gen.defs_file_name()), 'w') as output:
output.write(gen.generate_defs_file())
def main():
parser = argparse.ArgumentParser()
parser.add_argument('watch_config')
parser.add_argument('src_output_dir')
args = parser.parse_args()
generate(args.watch_config, args.src_output_dir)
shutil.copy("watch_boilerplate.template.c", os.path.join(args.src_output_dir, "watch_boilerplate.template.c"))
if __name__ == "__main__":
main()