CLI-ify rgblight_breathing_table_calc.c (#11174)
Co-authored-by: Takeshi ISHII <2170248+mtei@users.noreply.github.com> Co-authored-by: Zach White <skullydazed@drpepper.org>kle2json_hyper 0.11.9
Ryan
4 years ago
committed by
GitHub
parent
3925ff5342
commit
37fb14f1b5
@ -1,2 +1,3 @@ |
||||
from . import api |
||||
from . import docs |
||||
from . import rgb_breathe_table |
||||
|
||||
@ -0,0 +1,79 @@ |
||||
"""Generate rgblight_breathe_table.h |
||||
""" |
||||
import math |
||||
from argparse import ArgumentTypeError |
||||
|
||||
from milc import cli |
||||
|
||||
import qmk.path |
||||
|
||||
|
||||
def breathing_center(value): |
||||
value = float(value) |
||||
if value >= 1 and value <= 2.7: |
||||
return value |
||||
else: |
||||
raise ArgumentTypeError('Breathing center must be between 1 and 2.7') |
||||
|
||||
|
||||
def breathing_max(value): |
||||
value = int(value) |
||||
if value in range(0, 256): |
||||
return value |
||||
else: |
||||
raise ArgumentTypeError('Breathing max must be between 0 and 255') |
||||
|
||||
|
||||
@cli.argument('-c', '--center', arg_only=True, type=breathing_center, default=1.85, help='The breathing center value, from 1 to 2.7. Default: 1.85') |
||||
@cli.argument('-m', '--max', arg_only=True, type=breathing_max, default=255, help='The breathing maximum value, from 0 to 255. Default: 255') |
||||
@cli.argument('-o', '--output', arg_only=True, type=qmk.path.normpath, help='File to write to') |
||||
@cli.argument('-q', '--quiet', arg_only=True, action='store_true', help='Quiet mode, only output error messages') |
||||
@cli.subcommand('Generates an RGB Light breathing table header.') |
||||
def generate_rgb_breathe_table(cli): |
||||
"""Generate a rgblight_breathe_table.h file containing a breathing LUT for RGB Lighting (Underglow) feature. |
||||
""" |
||||
breathe_values = [0] * 256 |
||||
for pos in range(0, 256): |
||||
breathe_values[pos] = (int)((math.exp(math.sin((pos/255) * math.pi)) - cli.args.center / math.e) * (cli.args.max / (math.e - 1 / math.e))) # noqa: yapf insists there be no whitespace around / |
||||
|
||||
values_template = '' |
||||
for s in range(0, 3): |
||||
step = 1 << s |
||||
|
||||
values_template += '#if RGBLIGHT_BREATHE_TABLE_SIZE == {}\n'.format(256 >> s) |
||||
|
||||
for pos in range(0, 256, step): |
||||
values_template += ' ' if pos % 8 == 0 else '' |
||||
values_template += '0x{:02X}'.format(breathe_values[pos]) |
||||
values_template += ',' if (pos + step) < 256 else '' |
||||
values_template += '\n' if (pos+step) % 8 == 0 else ' ' # noqa: yapf insists there be no whitespace around + |
||||
|
||||
values_template += '#endif' |
||||
values_template += '\n\n' if s < 2 else '' |
||||
|
||||
table_template = '''#pragma once |
||||
|
||||
#define RGBLIGHT_EFFECT_BREATHE_TABLE |
||||
|
||||
// clang-format off |
||||
|
||||
// Breathing center: {0:.2f} |
||||
// Breathing max: {1:d} |
||||
|
||||
const uint8_t PROGMEM rgblight_effect_breathe_table[] = {{ |
||||
{2} |
||||
}}; |
||||
|
||||
static const int table_scale = 256 / sizeof(rgblight_effect_breathe_table); |
||||
'''.format(cli.args.center, cli.args.max, values_template) |
||||
|
||||
if cli.args.output: |
||||
cli.args.output.parent.mkdir(parents=True, exist_ok=True) |
||||
if cli.args.output.exists(): |
||||
cli.args.output.replace(cli.args.output.name + '.bak') |
||||
cli.args.output.write_text(table_template) |
||||
|
||||
if not cli.args.quiet: |
||||
cli.log.info('Wrote header to %s.', cli.args.output) |
||||
else: |
||||
print(table_template) |
||||
@ -1,49 +0,0 @@ |
||||
//
|
||||
// calculate rgblight_effect_breathe_table[] values
|
||||
//
|
||||
// this is host program for quantum/rgblight.c:void rgblight_effect_breathing();
|
||||
//
|
||||
// example:
|
||||
// $ edit util/rgblight_breathing_table_calc.c
|
||||
// $ cc -o util/rgblight_breathing_table_calc util/rgblight_breathing_table_calc.c
|
||||
// $ ./util/rgblight_breathing_table_calc > keyboards/KEYBOARD_NAME/keymaps/KEYMAP_NAME/rgblight_breathe_table.h
|
||||
//
|
||||
#include <stdio.h> |
||||
#include <math.h> |
||||
#include <stdint.h> |
||||
|
||||
/// customize breeathing effect part ///////////////////////////
|
||||
#define RGBLIGHT_EFFECT_BREATHE_CENTER 1.85 // 1 to 2.7
|
||||
#define RGBLIGHT_EFFECT_BREATHE_MAX 255 // 0 to 255
|
||||
////////////////////////////////////////////////////////////////
|
||||
|
||||
int main(void) { |
||||
int pos, step; |
||||
int table[256]; |
||||
for (pos = 0; pos < 256; pos ++ ) { |
||||
table[pos] = (uint8_t)( |
||||
(exp(sin((pos/255.0)*M_PI))- RGBLIGHT_EFFECT_BREATHE_CENTER/M_E) |
||||
* (RGBLIGHT_EFFECT_BREATHE_MAX/(M_E-1/M_E)) |
||||
); |
||||
} |
||||
printf("#ifndef RGBLIGHT_EFFECT_BREATHE_TABLE\n"); |
||||
printf("#define RGBLIGHT_EFFECT_BREATHE_TABLE\n\n"); |
||||
printf("const uint8_t rgblight_effect_breathe_table[] PROGMEM = {\n"); |
||||
printf(" /* #define RGBLIGHT_EFFECT_BREATHE_CENTER %.2f */\n", RGBLIGHT_EFFECT_BREATHE_CENTER); |
||||
printf(" /* #define RGBLIGHT_EFFECT_BREATHE_MAX %d */\n", RGBLIGHT_EFFECT_BREATHE_MAX); |
||||
|
||||
for (int s = 0, step = (1<<s); s < 3 ; s += 1, step = (1<<s) ) { |
||||
printf("\n #if RGBLIGHT_BREATHE_TABLE_SIZE == %d\n", |
||||
s == 0 ? 256:(s== 1 ? 128: 64)); |
||||
for (pos = 0; pos < 256; pos += step ) { |
||||
printf(" 0x%x%s", table[pos], (pos+step)>=256?"":"," ); |
||||
if ((pos+step) % 8 == 0) |
||||
printf("\n"); |
||||
} |
||||
printf(" #endif /* %d bytes table */\n", s == 0 ? 256:(s== 1 ? 128: 64)); |
||||
} |
||||
printf("};\n"); |
||||
printf("\nstatic const int table_scale = 256/sizeof(rgblight_effect_breathe_table);\n"); |
||||
printf("\n#endif /* RGBLIGHT_EFFECT_BREATHE_TABLE */\n"); |
||||
return 0; |
||||
} |
||||
Loading…
Reference in new issue