import numpy as np
from monai.transforms import GridPatchd
from monai.apps.pathology.transforms import TileOnGridd
from numpy.testing import assert_array_equal, assert_equal

# input configs
patch_size = 256
num_patches = 90
h, w = 10000, 1000
img = {"image": np.random.randint(0, 255, (3, h, w))}

# create components objects
patcher = GridPatchd(
    keys=["image"],
    patch_size=(patch_size, patch_size),
    num_patches=num_patches,
    sort_fn="min",
    threshold=0.999 * 3 * 255 * patch_size * patch_size,
    pad_mode=None,
    constant_values=255,
)
tiler = TileOnGridd(
    keys=["image"],
    tile_count=num_patches,
    tile_size=patch_size,
    background_val=255,
    return_list_of_dicts=True,
)

# Make sure the outputs are identical
patch_output = patcher(img)
tile_output = tiler(img)
assert_equal(len(patch_output), len(tile_output))
for p, t in zip(patch_output, tile_output):
    assert_array_equal(p["image"], t["image"])

%%timeit -n 10 -r 10
patch_output = patcher(img)

%%timeit -n 10 -r 10
tile_output = tiler(img)