# SPDX-License-Identifier: MPL-2.0
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import make_axes_locatable

def random_pixels(n_pixels, xmin=0, xmax=512, ymin=0, ymax=1024):
    """Return a list of random pixels.
    
    Args:
        n_pixels (int): Number of pixels to return.
        rows (int): Number of rows in the image.
        cols (int): Number of columns in the image.
        
    Returns:
        list: List of (row, col) tuples.
    """
    return [(np.random.randint(ymin, ymax), np.random.randint(xmin, xmax)) for _ in range(n_pixels)]


def random_pixel(xmin=0, xmax=512, ymin=0, ymax=1024):
    """Return a random pixel.
    
    Returns:
        tuple: (row, col)
    """
    return random_pixels(1, xmin, xmax, ymin, ymax)[0]

def flat_list(xss):
    """Flatten a list of lists."""
    return [x for xs in xss for x in xs]

def add_colorbar(ax, im, size="5%", pad=0.05):
    """Add a colorbar with the same height as the image."""
    divider = make_axes_locatable(ax)
    cax = divider.append_axes("right", size=size, pad=pad)
    plt.colorbar(im, cax=cax)
    return ax, im, cax