random_generator.py

import inspect
import os
import sys

current_dir = os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))
parent_dir = os.path.dirname(current_dir)
sys.path.insert(0, parent_dir)

import utilities as ut


def main(args):
    # Argument parsing
    parser = ut.ArgParser(description='Generate a random array and save it to a file.')
    parser.add_argument('rows', type=int, help='Number of rows for the array')
    parser.add_argument('cols', type=int, help='Number of columns for the array')
    parser.add_argument('output_file', type=str, help='Path to the output file')
    parser.add_argument('output_mask_file', type=str, help='Path to the output mask file')
    parser.add_argument('--chance', type=float, help='Chance of a pixel being on, only for non-uniform', default=0.2)
    parser.add_argument('--uniform', type=bool, help='Whether to use a uniform distribution', default=False)

    parsed_args = parser.parse_args(args)

    # Apply the function to create the new array
    array, mask = ut.create_array(parsed_args.rows, parsed_args.cols, ut.ArrayType.RANDOM_UNIFORM if parsed_args.uniform else ut.ArrayType.RANDOM, chance=parsed_args.chance)

    # Save the output
    ut.save_array_to_file(array, parsed_args.output_file)
    ut.save_array_to_file(mask, parsed_args.output_mask_file)


if __name__ == "__main__":
    main(sys.argv[1:])