diff --git a/PIL/ImImagePlugin.py b/PIL/ImImagePlugin.py index 3371b303f..388015b7f 100644 --- a/PIL/ImImagePlugin.py +++ b/PIL/ImImagePlugin.py @@ -310,6 +310,7 @@ SAVE = { "I;16": ("L 16", "I;16"), "I;16L": ("L 16L", "I;16L"), "I;16B": ("L 16B", "I;16B"), + "I;16S": ("L 16S", "I;16S"), "F": ("L 32F", "F;32F"), "RGB": ("RGB", "RGB;L"), "RGBA": ("RGBA", "RGBA;L"), diff --git a/PIL/Image.py b/PIL/Image.py index c22ab3955..5205bbf42 100644 --- a/PIL/Image.py +++ b/PIL/Image.py @@ -223,7 +223,7 @@ _MODEINFO = { "LAB": ("RGB", "L", ("L", "A", "B")), "HSV": ("RGB", "L", ("H", "S", "V")), - # Experimental modes include I;16, I;16L, I;16B, RGBa, BGR;15, and + # Experimental modes include I;16, I;16L, I;16B, I;16S, RGBa, BGR;15, and # BGR;24. Use these modes only if you know exactly what you're # doing... diff --git a/PIL/ImageMode.py b/PIL/ImageMode.py index b227f2127..3f638d5f5 100644 --- a/PIL/ImageMode.py +++ b/PIL/ImageMode.py @@ -50,6 +50,7 @@ def getmode(mode): modes["I;16"] = ModeDescriptor("I;16", "I", "L", "L") modes["I;16L"] = ModeDescriptor("I;16L", "I", "L", "L") modes["I;16B"] = ModeDescriptor("I;16B", "I", "L", "L") + modes["I;16S"] = ModeDescriptor("I;16S", "I", "L", "L") # set global mode cache atomically _modes = modes return _modes[mode] diff --git a/Tests/test_mode_i16.py b/Tests/test_mode_i16.py index d51847199..b150126f0 100644 --- a/Tests/test_mode_i16.py +++ b/Tests/test_mode_i16.py @@ -79,6 +79,7 @@ class TestModeI16(PillowTestCase): basic("I;16") basic("I;16B") basic("I;16L") + basic("I;16S") basic("I") @@ -92,6 +93,7 @@ class TestModeI16(PillowTestCase): self.assertEqual(tobytes("L"), b"\x01") self.assertEqual(tobytes("I;16"), b"\x01\x00") self.assertEqual(tobytes("I;16B"), b"\x00\x01") + self.assertEqual(tobytes("I;16S"), b"\x01\x00") self.assertEqual(tobytes("I"), b"\x01\x00\x00\x00"[::order]) def test_convert(self): @@ -106,6 +108,10 @@ class TestModeI16(PillowTestCase): self.verify(im.convert("I;16B").convert("L")) self.verify(im.convert("I;16B").convert("I")) + self.verify(im.convert("I;16S")) + self.verify(im.convert("I;16S").convert("L")) + self.verify(im.convert("I;16S").convert("I")) + if __name__ == '__main__': unittest.main() diff --git a/libImaging/Convert.c b/libImaging/Convert.c index 0ec5ef864..19337a3e4 100644 --- a/libImaging/Convert.c +++ b/libImaging/Convert.c @@ -693,6 +693,17 @@ I_I16B(UINT8* out, const UINT8* in_, int xsize) } } +static void +I_I16S(UINT8* out, const UINT8* in_, int xsize) +{ + int x, v; + INT32* in = (INT32*) in_; + for (x = 0; x < xsize; x++, in++) { + v = CLIP16(*in); + *out++ = (UINT8) v; + *out++ = (UINT8) (v >> 8); + } +} static void I16L_I(UINT8* out_, const UINT8* in, int xsize) @@ -713,6 +724,15 @@ I16B_I(UINT8* out_, const UINT8* in, int xsize) *out++ = in[1] + ((int) in[0] << 8); } +static void +I16S_I(UINT8* out_, const UINT8* in, int xsize) +{ + int x; + INT32* out = (INT32*) out_; + for (x = 0; x < xsize; x++, in += 2) + *out++ = (INT16)(in[0] + ((int) in[1] << 8)); +} + static void I16L_F(UINT8* out_, const UINT8* in, int xsize) { @@ -732,6 +752,15 @@ I16B_F(UINT8* out_, const UINT8* in, int xsize) *out++ = (FLOAT32) (in[1] + ((int) in[0] << 8)); } +static void +I16S_F(UINT8* out_, const UINT8* in, int xsize) +{ + int x; + FLOAT32* out = (FLOAT32*) out_; + for (x = 0; x < xsize; x++, in += 2) + *out++ = (FLOAT32) (INT16) (in[0] + ((int) in[1] << 8)); +} + static void L_I16L(UINT8* out, const UINT8* in, int xsize) { @@ -752,6 +781,16 @@ L_I16B(UINT8* out, const UINT8* in, int xsize) } } +static void +L_I16S(UINT8* out, const UINT8* in, int xsize) +{ + int x; + for (x = 0; x < xsize; x++, in++) { + *out++ = *in; + *out++ = 0; + } +} + static void I16L_L(UINT8* out, const UINT8* in, int xsize) { @@ -774,6 +813,19 @@ I16B_L(UINT8* out, const UINT8* in, int xsize) *out++ = in[1]; } +static void +I16S_L(UINT8* out, const UINT8* in, int xsize) +{ + int x; + for (x = 0; x < xsize; x++, in += 2) + if (in[1] & 0x80) + *out++ = 0; /* Negative -> 0 */ + else if (in[1] != 0) + *out++ = 255; /* Greater than 255 -> 255 */ + else + *out++ = in[0]; +} + static struct { const char* from; const char* to; @@ -872,9 +924,15 @@ static struct { { "L", "I;16B", L_I16B }, { "I;16B", "L", I16B_L }, + { "I", "I;16S", I_I16S }, + { "I;16S", "I", I16S_I }, + { "L", "I;16S", L_I16S }, + { "I;16S", "L", I16S_L }, + { "I;16", "F", I16L_F }, { "I;16L", "F", I16L_F }, { "I;16B", "F", I16B_F }, + { "I;16S", "F", I16S_F }, { NULL } };