Fixed frame position when seeking past the end of file

This commit is contained in:
Andrew Murray 2015-06-19 00:49:18 +10:00
parent a09da242fd
commit 1111e9fb35
9 changed files with 105 additions and 5 deletions

View File

@ -127,8 +127,14 @@ class FliImageFile(ImageFile.ImageFile):
return
if frame < self.__frame:
self._seek(0)
last_frame = self.__frame
for f in range(self.__frame + 1, frame + 1):
try:
self._seek(f)
except EOFError:
self.seek(last_frame)
raise EOFError("no more images in FLI file")
def _seek(self, frame):
if frame == 0:

View File

@ -107,8 +107,14 @@ class GifImageFile(ImageFile.ImageFile):
return
if frame < self.__frame:
self._seek(0)
last_frame = self.__frame
for f in range(self.__frame + 1, frame + 1):
try:
self._seek(f)
except EOFError:
self.seek(last_frame)
raise EOFError("no more images in GIF file")
def _seek(self, frame):
@ -241,7 +247,7 @@ class GifImageFile(ImageFile.ImageFile):
if not self.tile:
# self.__fp = None
raise EOFError("no more images in GIF file")
raise EOFError
self.mode = "L"
if self.palette:

View File

@ -34,6 +34,18 @@ class TestFileDcx(PillowTestCase):
im = Image.open(TEST_FILE)
self.assertEqual(im.n_frames, 1)
def test_eoferror(self):
im = Image.open(TEST_FILE)
n_frames = im.n_frames
while True:
n_frames -= 1
try:
im.seek(n_frames)
break
except EOFError:
self.assertTrue(im.tell() < n_frames)
def test_seek_too_far(self):
# Arrange
im = Image.open(TEST_FILE)

View File

@ -20,7 +20,19 @@ class TestFileFli(PillowTestCase):
def test_n_frames(self):
im = Image.open(test_file)
self.assertEqual(im.n_frames, 2)
self.assertEqual(im.n_frames, 1)
def test_eoferror(self):
im = Image.open(test_file)
n_frames = im.n_frames
while True:
n_frames -= 1
try:
im.seek(n_frames)
break
except EOFError:
self.assertTrue(im.tell() < n_frames)
if __name__ == '__main__':

View File

@ -135,8 +135,23 @@ class TestFileGif(PillowTestCase):
self.assertEqual(framecount, 5)
def test_n_frames(self):
im = Image.open(TEST_GIF)
self.assertEqual(im.n_frames, 1)
im = Image.open("Tests/images/iss634.gif")
self.assertEqual(im.n_frames, 43)
self.assertEqual(im.n_frames, 42)
def test_eoferror(self):
im = Image.open(TEST_GIF)
n_frames = im.n_frames
while True:
n_frames -= 1
try:
im.seek(n_frames)
break
except EOFError:
self.assertTrue(im.tell() < n_frames)
def test_dispose_none(self):
img = Image.open("Tests/images/dispose_none.gif")

View File

@ -19,6 +19,18 @@ class TestFileIm(PillowTestCase):
im = Image.open(TEST_IM)
self.assertEqual(im.n_frames, 1)
def test_eoferror(self):
im = Image.open(TEST_IM)
n_frames = im.n_frames
while True:
n_frames -= 1
try:
im.seek(n_frames)
break
except EOFError:
self.assertTrue(im.tell() < n_frames)
def test_roundtrip(self):
out = self.tempfile('temp.im')
im = hopper()

View File

@ -99,6 +99,18 @@ class TestFileMpo(PillowTestCase):
im = Image.open("Tests/images/sugarshack.mpo")
self.assertEqual(im.n_frames, 2)
def test_eoferror(self):
im = Image.open("Tests/images/sugarshack.mpo")
n_frames = im.n_frames
while True:
n_frames -= 1
try:
im.seek(n_frames)
break
except EOFError:
self.assertTrue(im.tell() < n_frames)
def test_image_grab(self):
for test_file in test_files:
im = Image.open(test_file)

View File

@ -23,6 +23,19 @@ class TestImagePsd(PillowTestCase):
im = Image.open(test_file)
self.assertEqual(im.n_frames, 2)
def test_eoferror(self):
im = Image.open(test_file)
n_frames = im.n_frames
while True:
n_frames -= 1
try:
# PSD seek index starts at 1 rather than 0
im.seek(n_frames+1)
break
except EOFError:
self.assertTrue(im.tell() < n_frames)
if __name__ == '__main__':
unittest.main()

View File

@ -157,6 +157,18 @@ class TestFileTiff(PillowTestCase):
im = Image.open('Tests/images/multipage.tiff')
self.assertEqual(im.n_frames, 3)
def test_eoferror(self):
im = Image.open('Tests/images/multipage-lastframe.tif')
n_frames = im.n_frames
while True:
n_frames -= 1
try:
im.seek(n_frames)
break
except EOFError:
self.assertTrue(im.tell() < n_frames)
def test_multipage(self):
# issue #862
im = Image.open('Tests/images/multipage.tiff')