Skip to main content

Python wrapper around TinyEXR and stb_image

Project description

Build

Simple Image IO

A lightweight C# and Python wrapper to read and write RGB images from / to various file formats. Supports .exr (with layers) via tinyexr and a number of other formats (including .png, .jpg, and .bmp) via stb_image and stb_image_write. We also implement our own importer and exporter for PFM.

In addition, the package offers some basic image manipulation functionality and error metrics. The C# wrapper further offers utilities for thread-safe atomic splatting of pixel values, and sending image data to the tev viewer via sockets.

The Nuget package contains prebuilt binaries of the C++ wrapper for x86-64 Windows, Ubuntu, and macOS (.github/workflows/build.yml). The Python package is set up to automatically download an adequate CMake version and compile the C++ code on any platform.

Usage example (C#)

The following creates a one pixel image and writes it to various file formats:

RgbImage img = new(width: 1, height: 1);
img.SetPixel(0, 0, new(0.1f, 0.4f, 0.9f));
img.WriteToFile("test.exr");
img.WriteToFile("test.png");
img.WriteToFile("test.jpg");

Reading an image from one of the supported formats is equally simple:

RgbImage img = new("test.exr");
Console.WriteLine(img.GetPixel(0, 0).Luminance);

The pixel coordinate (0,0) corresponds to the top left corner of the image. Coordinates outside the valid range are clamped automatically; no error is raised. The framework also offers a MonochromeImage with a single channel per pixel. Further, the base class ImageBase can be used directly for images with arbitrary channel count (RgbImage and MonochromeImage only add some convenience functions like directly returning an RgbColor object).

As an added bonus, the C# wrapper can connect to the tev HDR viewer and directly display image data via sockets. The following example generates a monochrome image and sends it to tev:

TevIpc tevIpc = new(); // uses tev's default port on localhost

// Create the image and initialize a tev sync
MonochromeImage image = new(width: 20, height: 10);
tevIpc.CreateImageSync("MyAwesomeImage", 20, 10, ("default", image));

// Pretend we are a renderer and write some image data.
image.SetPixel(0, 0, val: 1);
image.SetPixel(10, 0, val: 2);
image.SetPixel(0, 9, val: 5);
image.SetPixel(10, 9, val: 10);

// Tell the TevIpc class to update the image displayed by tev
// (this currently retransmitts all pixel values)
tevIpc.UpdateImage("MyAwesomeImage");

Usage example (Python)

The following creates a one pixel image, writes it to various file formats, reads one of them back in, and prints the red color channel of the pixel:

import simpleimageio as sio
sio.write("test.exr", [[[0.1, 0.4, 0.9]]])
sio.write("test.png", [[[0.1, 0.4, 0.9]]])
sio.write("test.jpg", [[[0.1, 0.4, 0.9]]])
img = sio.read("test.exr")
print(img[0,0,0])

In Python, an image is a 3D row-major array, where [0,0,0] is the red color channel of the top left corner. The convention is compatible with most other libraries that make use of numpy arrays for image represenation, like matplotlib.

Building from source

If you are on an architecture different from x86-64, you will need to compile the C++ wrapper from source. Below, you can find instructions on how to accomplish that.

Dependencies

All dependencies are header-only and included in the repository. Building requires

  • a C++11 (or newer) compiler
  • CMake
  • .NET 5.0 (or newer)
  • Python ≥ 3.6

Building the C# wrapper on x86-64 Windows, Linux, or macOS

Build the C++ low level library with CMake:

mkdir build
cd build
cmake .. -DCMAKE_BUILD_TYPE=Release
cmake --build . --config Release
cd ..

Compile and run the tests (optional):

dotnet test

That's it. Simply add a reference to SimpleImageIO/SimpleImageIO.csproj to your project and you should be up and running.

Building the C# wrapper on other platforms

The SimpleImageIO.csproj file needs to copy the correct .dll / .so / .dylib file to the appropriate runtime folder. Currently, the runtime identifiers (RID) and copy instructions are only set for the x86-64 versions of Windows, Linux, and macOS. To run the framework on other architectures, you will need to add them to the .csproj file. You can find the right RID for your platform here: https://docs.microsoft.com/en-us/dotnet/core/rid-catalog.

Then, you should be able to follow the steps above and proceed as usual.

Building the Python wrapper

Simply running:

python -m build

will automatically fetch an adequate version of CMake, compile the shared library, and build the Python package. You can then simply install the result via:

pip install ./dist/SimpleImageIO-*.whl

The tests can be run via:

cd PyTest
python -m unittest

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

SimpleImageIO-0.5.2.tar.gz (202.8 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page