apr_io_demo.py

import pyapr
import numpy as np
from skimage import io as skio


"""
This demo converts a selected TIFF image to an APR, writes the result to file and then reads the file.
"""

io_int = pyapr.utils.InteractiveIO()

# Read in an image
fpath = io_int.get_tiff_file_name()
img = skio.imread(fpath)

# convert image to APR (with default parameters)
apr, parts = pyapr.converter.get_apr(img)

# Compute and display the computational ratio
numParts = apr.total_number_particles()
numPix = img.size
cr = numPix / numParts
print('Input image size: {} pixels, APR size: {} particles --> Computational Ratio = {}'.format(numPix, numParts, cr))

# Save the APR to file
fpath_apr = io_int.save_apr_file_name()  # get save path from gui
pyapr.io.write(fpath_apr, apr, parts)    # write apr and particles to file

# Read the newly written file
apr2, parts2 = pyapr.io.read(fpath_apr)

# check that particles are equal at a single, random index
ri = np.random.randint(0, numParts-1)
assert parts[ri] == parts2[ri]

# check some APR properties
assert apr.total_number_particles() == apr2.total_number_particles()
assert apr.level_max() == apr2.level_max()
assert apr.level_min() == apr2.level_min()