Marlin/buildroot/share/scripts/createSpeedLookupTable.py
Scott Lahteine 143a8701d5 ️ Optimize speed lookup for AVR
Co-Authored-By: tombrazier <68918209+tombrazier@users.noreply.github.com>
2023-03-13 16:49:33 -05:00

56 lines
1.7 KiB
Python
Executable file

#!/usr/bin/env python
from __future__ import print_function
from __future__ import division
""" Generate the stepper delay lookup table for Marlin firmware. """
import argparse
__author__ = "Ben Gamari <bgamari@gmail.com>"
__copyright__ = "Copyright 2012, Ben Gamari"
__license__ = "GPL"
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument('-f', '--cpu-freq', type=int, default=16, help='CPU clockrate in MHz (default=16)')
parser.add_argument('-d', '--divider', type=int, default=8, help='Timer/counter pre-scale divider (default=8)')
args = parser.parse_args()
cpu_freq = args.cpu_freq * 1000000
timer_freq = cpu_freq / args.divider
print("#pragma once")
print()
print("#if F_CPU == %d" % cpu_freq)
print()
print(" const struct { uint16_t base; uint8_t gain; } speed_lookuptable_fast[256] PROGMEM = {")
a = [0 for i in range(8)] + [ int(0.5 + float(timer_freq) / (i*256)) for i in range(8, 256) ]
b = [0 for i in range(8)] + [ a[i] - a[i+1] for i in range(8, 255) ]
b.append(b[-1])
for i in range(32):
print(" ", end='')
for j in range(8):
print("{ %5d, %5d }," % (a[8*i+j], b[8*i+j]), end='')
if j < 7: print(" ", end='')
if i == 0: print(" // dummy first row")
else: print()
print(" };")
print()
print(" const uint16_t speed_lookuptable_slow[256][2] PROGMEM = {")
a = [ int(0.5 + float(timer_freq) / ((i*8)+(args.cpu_freq*2))) for i in range(256) ]
b = [ a[i] - a[i+1] for i in range(255) ]
b.append(b[-1])
for i in range(32):
print(" ", end='')
for j in range(8):
print("{ %5d, %5d }," % (a[8*i+j], b[8*i+j]), end='')
if j < 7: print(" ", end='')
print()
print(" };")
print()
print("#endif")