Generate 3D LUT

gen-lut.py

#!/usr/bin/env python

class Lut3DGenerator:
    def __init__(self):
        pass

    def gen_identity_3dlut(self, lut_size, output_fname):
        lut_size_f = float(lut_size)

        # Generate an identity 3D LUT and write to output file.
        with open(output_fname, 'w') as f:
            f.write('TITLE "Identity 3D LUT"\n')
            f.write('LUT_3D_SIZE {}\n\n'.format(lut_size))

            for b in range(lut_size):
                for g in range(lut_size):
                    for r in range(lut_size):
                        f.write('{} {} {}\n'.format(r / lut_size_f, g / lut_size_f, b / lut_size_f))


if __name__ == '__main__':
    Lut3DGenerator().gen_identity_3dlut(33, 'identity_33.cube')