Yet another fixed point fractional prescale calculation

fractional_prescale_bocci.py

# Fixed point fractional prescale calculation using add and sub
# Proposed by Andrea Bocci, Sept. 2019

import argparse

class PrescaleCounter(object):

    precision = 1

    def __init__(self, prescale):
        self.step = 10**self.precision
        self.counter = 0
        self.prescale = int(round(prescale * self.step))

    def next(self):
        result = False
        self.counter += self.step
        if self.counter >= self.prescale:
            self.counter -= self.prescale
            result = True
        return (self.counter / self.step), result

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--prescale', type=float, default=2.3)
    parser.add_argument('--bx-range', type=int, default=256)
    args = parser.parse_args()

    counter = PrescaleCounter(args.prescale)

    for i in range(args.bx_range):

        count, result = counter.next()

        print('{:04} | {} | {}'.format(i, count, int(result)))

if __name__ == '__main__':
    main()