blob: fcdd3ba97b11f255afd2783e6ee9b99af715e2cd [file] [log] [blame]
# Copyright 2020 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
import os
import math
from base_generator import Color, Modes, BaseGenerator, VariableType
class ViewsStyleGenerator(BaseGenerator):
'''Generator for Views Variables'''
@staticmethod
def GetName():
return 'Views'
def Render(self):
self.Validate()
return self.ApplyTemplate(self, 'views_generator_h.tmpl',
self.GetParameters())
def GetParameters(self):
return {
'colors': self._CreateColorList(),
'opacities': self.model[VariableType.OPACITY],
}
def GetFilters(self):
return {
'to_const_name': self._ToConstName,
'cpp_color': self._CppColor,
'alpha_to_hex': self._AlphaToHex,
}
def GetGlobals(self):
globals = {
'Modes': Modes,
'out_file_path': None,
'namespace_name': None,
'in_files': self.in_file_to_context.keys(),
}
if self.out_file_path:
globals['out_file_path'] = self.out_file_path
globals['namespace_name'] = os.path.splitext(
os.path.basename(self.out_file_path))[0]
return globals
def _CreateColorList(self):
color_list = []
for name, mode_values in self.model[VariableType.COLOR].items():
color_list.append({'name': name, 'mode_values': mode_values})
return color_list
def _ToConstName(self, var_name):
return 'k%s' % var_name.title().replace('_', '')
def _AlphaToHex(self, a):
return '0x%X' % math.floor(a * 255)
def _CppColor(self, c):
'''Returns the C++ color representation of |c|'''
assert (isinstance(c, Color))
def CppOpacity(color):
if c.a != -1:
return self._AlphaToHex(c.a)
elif c.opacity_var:
return 'GetOpacity(OpacityName::%s)' % self._ToConstName(
c.opacity_var)
raise ValueError('Color with invalid opacity: ' + repr(color))
if c.var:
return ('ResolveColor(ColorName::%s, is_dark_mode)' %
self._ToConstName(c.var))
if c.rgb_var:
return (
'SkColorSetA(ResolveColor(ColorName::%s, is_dark_mode), %s)' %
(self._ToConstName(c.RGBVarToVar()), CppOpacity(c)))
if c.a != 1:
return 'SkColorSetARGB(%s, 0x%X, 0x%X, 0x%X)' % (CppOpacity(c),
c.r, c.g, c.b)
else:
return 'SkColorSetRGB(0x%X, 0x%X, 0x%X)' % (c.r, c.g, c.b)