jyapayne/glymur
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use memoryview instead of np.getbuffer. #210

Browse source 
np.getbuffer is not available in python3.
This commit is contained in:
John Evans 2014-04-02 12:21:56 -04:00
commit 63d203796b
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
glymur/jp2box.py
View file

@ -1885,11 +1885,11 @@ class PaletteBox(Jp2kBox):
if all(b == bps[0] for b in bps): if all(b == bps[0] for b in bps):
# All components are the same. Writing is straightforward. # All components are the same. Writing is straightforward.
if self.bits_per_component[0] <= 8: if self.bits_per_component[0] <= 8:
write_buffer = np.getbuffer(self.palette.astype(np.uint8)) write_buffer = memoryview(self.palette.astype(np.uint8))
elif self.bits_per_component[0] <= 16: elif self.bits_per_component[0] <= 16:
write_buffer = np.getbuffer(self.palette.astype(np.uint16)) write_buffer = memoryview(self.palette.astype(np.uint16))
elif self.bits_per_component[0] <= 32: elif self.bits_per_component[0] <= 32:
write_buffer = np.getbuffer(self.palette.astype(np.uint32)) write_buffer = memoryview(self.palette.astype(np.uint32))
fptr.write(write_buffer) fptr.write(write_buffer)
else: else:
# Not all the components are the same. More general, but much rarer # Not all the components are the same. More general, but much rarer