Skip to content

Commit

Permalink
Added signed option when saving JPEG2000 images
Browse files Browse the repository at this point in the history
  • Loading branch information
radarhere committed Nov 1, 2022
1 parent a4ec9f3 commit 1e9314b
Show file tree
Hide file tree
Showing 6 changed files with 29 additions and 2 deletions.
14 changes: 14 additions & 0 deletions Tests/test_file_jpeg2k.py
Expand Up @@ -252,6 +252,20 @@ def test_mct():
assert_image_similar(im, jp2, 1.0e-3)


def test_sgnd(tmp_path):
outfile = str(tmp_path / "temp.jp2")

im = Image.new("L", (1, 1))
im.save(outfile)
with Image.open(outfile) as reloaded:
assert reloaded.getpixel((0, 0)) == 0

im = Image.new("L", (1, 1))
im.save(outfile, signed=True)
with Image.open(outfile) as reloaded_signed:
assert reloaded_signed.getpixel((0, 0)) == 128


def test_rgba():
# Arrange
with Image.open("Tests/images/rgb_trns_ycbc.j2k") as j2k:
Expand Down
5 changes: 5 additions & 0 deletions docs/handbook/image-file-formats.rst
Expand Up @@ -563,6 +563,11 @@ The :py:meth:`~PIL.Image.Image.save` method supports the following options:
encoded using RLCP mode will have increasing resolutions decoded as they
arrive, and so on.

**signed**
If true, then tell the encoder to save the image as signed.

.. versionadded:: 9.4.0

**cinema_mode**
Set the encoder to produce output compliant with the digital cinema
specifications. The options here are ``"no"`` (the default),
Expand Down
2 changes: 2 additions & 0 deletions src/PIL/Jpeg2KImagePlugin.py
Expand Up @@ -321,6 +321,7 @@ def _save(im, fp, filename):
progression = info.get("progression", "LRCP")
cinema_mode = info.get("cinema_mode", "no")
mct = info.get("mct", 0)
signed = info.get("signed", False)
fd = -1

if hasattr(fp, "fileno"):
Expand All @@ -342,6 +343,7 @@ def _save(im, fp, filename):
progression,
cinema_mode,
mct,
signed,
fd,
)

Expand Down
5 changes: 4 additions & 1 deletion src/encode.c
Expand Up @@ -1188,11 +1188,12 @@ PyImaging_Jpeg2KEncoderNew(PyObject *self, PyObject *args) {
char *cinema_mode = "no";
OPJ_CINEMA_MODE cine_mode;
char mct = 0;
int sgnd = 0;
Py_ssize_t fd = -1;

if (!PyArg_ParseTuple(
args,
"ss|OOOsOnOOOssbn",
"ss|OOOsOnOOOssbbn",
&mode,
&format,
&offset,
Expand All @@ -1207,6 +1208,7 @@ PyImaging_Jpeg2KEncoderNew(PyObject *self, PyObject *args) {
&progression,
&cinema_mode,
&mct,
&sgnd,
&fd)) {
return NULL;
}
Expand Down Expand Up @@ -1305,6 +1307,7 @@ PyImaging_Jpeg2KEncoderNew(PyObject *self, PyObject *args) {
context->progression = prog_order;
context->cinema_mode = cine_mode;
context->mct = mct;
context->sgnd = sgnd;

return (PyObject *)encoder;
}
Expand Down
3 changes: 3 additions & 0 deletions src/libImaging/Jpeg2K.h
Expand Up @@ -85,6 +85,9 @@ typedef struct {
/* Set multiple component transformation */
char mct;

/* Signed */
int sgnd;

/* Progression order (LRCP/RLCP/RPCL/PCRL/CPRL) */
OPJ_PROG_ORDER progression;

Expand Down
2 changes: 1 addition & 1 deletion src/libImaging/Jpeg2KEncode.c
Expand Up @@ -343,7 +343,7 @@ j2k_encode_entry(Imaging im, ImagingCodecState state) {
image_params[n].x0 = image_params[n].y0 = 0;
image_params[n].prec = prec;
image_params[n].bpp = bpp;
image_params[n].sgnd = 0;
image_params[n].sgnd = context->sgnd == 1 ? 1 : 0;
}

image = opj_image_create(components, image_params, color_space);
Expand Down

0 comments on commit 1e9314b

Please sign in to comment.