#!/usr/bin/env python3

# usage: test_leak.py <image path>

import resource
import sys

import pyvips

fname = sys.argv[1]
out_path = "/tmp/test.tiff"

pyvips.cache_set_max(0)

for i in range(100):
    with open(fname, "rb") as fh:
        img = pyvips.Image.new_from_buffer(fh.read(), "", access="sequential")

    kw = {
        "compression": "jpeg",
        "Q": 90,
        "tile": True,
        "tile_width": 256,
        "tile_height": 256,
        "pyramid": True,
        "bigtiff": True,
    }

    img.tiffsave(out_path, **kw)
    print(f"[{i:03d}] Memory usage: {resource.getrusage(resource.RUSAGE_SELF).ru_maxrss}")

del(img)
del(fh)
del(kw)
print(f"[END] Memory usage after freeing file handle and image: {resource.getrusage(resource.RUSAGE_SELF).ru_maxrss}")