compute_disk_size.py

import numpy as np

def compute_disk_size(user_radius, upscale_factor=1.2):
    return int(np.ceil(upscale_factor * 2 * user_radius + 1) // 2 * 2 + 1)