pebble/tools/resources/resource_map/resource_generator_pbi.py
2025-01-27 11:38:16 -08:00

62 lines
2.2 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.
from resources.types.resource_object import ResourceObject
from resources.resource_map.resource_generator import ResourceGenerator
from pebble_sdk_platform import pebble_platforms
import bitmapgen
def _pbi_generator(task, definition, format_str):
pb = bitmapgen.PebbleBitmap(task.inputs[0].abspath(), bitmap_format=format_str)
return ResourceObject(definition, pb.convert_to_pbi())
class PbiResourceGenerator(ResourceGenerator):
type = 'pbi'
@staticmethod
def generate_object(task, definition):
return _pbi_generator(task, definition, 'bw')
class Pbi8ResourceGenerator(ResourceGenerator):
type = 'pbi8'
@staticmethod
def generate_object(task, definition):
env = task.generator.env
format = 'color' if 'color' in pebble_platforms[env.PLATFORM_NAME]['TAGS'] else 'bw'
return _pbi_generator(task, definition, format)
# This implementation is in the "pbi" file because it's implemented using pbis, even though it's
# named with "png" prefix and we have a "png" file.
class PngTransResourceGenerator(ResourceGenerator):
type = 'png-trans'
@staticmethod
def generate_object(task, definition):
if 'WHITE' in definition.name:
color_map = bitmapgen.WHITE_COLOR_MAP
elif 'BLACK' in definition.name:
color_map = bitmapgen.BLACK_COLOR_MAP
else:
task.generator.bld.fatal('png-trans with neither white nor black in the name: ' +
resource_definition.name)
pb = bitmapgen.PebbleBitmap(task.inputs[0].abspath(), color_map=color_map)
return ResourceObject(definition, pb.convert_to_pbi())