From 082db0c6b5b70df45ef16b7124a982df3054e3c8 Mon Sep 17 00:00:00 2001
From: Vegard Storheil Eriksen <zyp@jvnv.net>
Date: Thu, 7 Jan 2021 12:25:05 +0100
Subject: build: Added hex and size_prefix filters for Jinja2.

---
 build/scons_tools/tool_jinja2.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'build')

diff --git a/build/scons_tools/tool_jinja2.py b/build/scons_tools/tool_jinja2.py
index a927285..c85c559 100644
--- a/build/scons_tools/tool_jinja2.py
+++ b/build/scons_tools/tool_jinja2.py
@@ -1,8 +1,17 @@
 from SCons.Script import *
 import jinja2
 
+loader = jinja2.FileSystemLoader('.')
+
+jinja2_env = jinja2.Environment(
+    loader = loader,
+)
+
+jinja2_env.filters['hex'] = lambda value: '%#x' % value
+jinja2_env.filters['size_prefix'] = lambda value: '%d%s' % next((value / 1024**i, c) for i, c in [(2, 'M'), (1, 'k'), (0, '')] if value % 1024**i == 0)
+
 def jinja2_build(target, source, env):
-    template = jinja2.Template(open(str(source[0])).read())
+    template = jinja2_env.get_template(str(source[0]))
 
     template_env = env.get('TEMPLATE_ENV', {})
 
-- 
cgit v1.2.3