class BitmapCoreHeader(BitmapInfo):
"""docstring for BitmapCoreHeader"""
def __init__(self, byte_array):
super(BitmapCoreHeader, self).__init__(byte_array)
self._width = Info(
"bcWidth", byte_array[self._offset:self._offset + 2], self._offset,
2, "The bitmap width in pixels (signed integer)")
self._offset += 2
self._height = Info(
"bcHeight", byte_array[self._offset:self._offset + 2],
self._offset, 2, "The bitmap height in pixels (signed integer)")
self._offset += 2
self._planes = Info(
"bcPlanes", byte_array[self._offset:self._offset + 2], 2,
self._offset, "The number of color planes must be 1")
self._offset += 2
self._bit_count = Info(
"bcBitCount", byte_array[self._offset:self._offset + 2], 2,
self._offset, "The number of bits per pixel, which is the color "
"depth of the image. Typical values are 1, 4, 8, 16, 24 and 32")
self._offset += 2
def get_list_info(self):
return(super(BitmapCoreHeader, self).get_list_info())
def get_all_info(self):
return(super(BitmapCoreHeader, self).get_all_info())
def fill_color_table(self, byte_array):
bytes_per_cell = 3
if self._bit_count.get_unpack_value() <= 8:
value_cell = 2**self._bit_count.get_unpack_value() * bytes_per_cell
self._color_table = \
byte_array[self._offset:self._offset + value_cell]
self._color_table = \
list(zip(*[iter(self._color_table)] * bytes_per_cell))
else:
self._color_table = None
class BitmapInfoHeader(BitmapInfo):
"""docstring for BitmapInfoHeader"""
def __init__(self, byte_array):
super(BitmapInfoHeader, self).__init__(byte_array)
self._width = Info(
"biWidth", byte_array[self._offset:self._offset + 4],
self._offset, 4, "The bitmap width in pixels (signed integer)")
self._offset += 4
self._height = Info(
"biHeight", byte_array[self._offset:self._offset + 4],
self._offset, 4, "The bitmap height in pixels (signed integer)")
self._offset += 4
self._planes = Info(
"biPlanes", byte_array[self._offset:self._offset + 2],
self._offset, 2, "The number of color planes must be 1")
self._offset += 2
self._bit_count = Info(
"biBitCount", byte_array[self._offset:self._offset + 2],
self._offset, 2, "The number of bits per pixel, which is the "
"color depth of the image. Typical values are 1, 4, 8, 16, 24"
" and 32")
self._offset += 2
self._compression = Info(
"biCompression", byte_array[self._offset:self._offset + 4],
self._offset, 4, "The compression method being used. See the next "
"table for a list of possible values")
self._offset += 4
self._size_image = Info(
"biSizeImage", byte_array[self._offset:self._offset + 4],
self._offset, 4, "The image size. This is the size of the"
" raw bitmap data; a dummy 0 can be given for BI_RGB bitmaps")
self._offset += 4
self._x_pels_per_meter = Info(
"biXPelsPerMeter", byte_array[self._offset:self._offset + 4],
self._offset, 4, "The horizontal resolution of the image. "
"(pixel per meter, signed integer)")
self._offset += 4
self._y_pels_per_meter = Info(
"biYPelsPerMeter", byte_array[self._offset:self._offset + 4],
self._offset, 4, "The vertical resolution of the "
"image. (pixel per meter, signed integer)")
self._offset += 4
self._clr_used = Info(
"biClrUsed", byte_array[self._offset:self._offset + 4],
self._offset, 4, "The number of colors in the color "
"palette, or 0 to default to 2^n")
self._offset += 4
self._clr_important = Info(
"biClrImportant", byte_array[self._offset:self._offset + 4],
self._offset, 4, "The number of important colors used, "
"or 0 when every color is important; generally ignored")
self._offset += 4
def get_list_info(self):
list_fields = [
self._compression, self._size_image, self._x_pels_per_meter,
self._y_pels_per_meter, self._clr_used, self._clr_important]
return(super(BitmapInfoHeader, self).get_list_info() + list_fields)
def get_all_info(self):
info_field = ''.join(
map(lambda x: x.get_all_data(), self.get_list_info()))
return(info_field)
def get_compression(self):
return(self._compression)
def get_size_bitmap(self):
return(self._size_image)
def get_x_pels_per_meter(self):
return(self._x_pels_per_meter)
def get_y_pels_per_meter(self):
return(self._y_pels_per_meter)
def get_clr_used(self):
return(self._clr_used)
def get_clr_important(self):
return(self._clr_important)
def fill_color_table(self, byte_array):
bytes_per_cell = 4
value_cell = 0
if self._bit_count.get_unpack_value() <= 8 or \
self._bit_count.get_unpack_value() >= 8 and \
self._clr_used.get_unpack_value() != 0:
if self._bit_count.get_unpack_value() <= 8:
if self._clr_used.get_unpack_value() == 0:
value_cell = \
2**self._bit_count.get_unpack_value() * \
bytes_per_cell
else:
value_cell = \
self._clr_used.get_unpack_value() * bytes_per_cell
elif self._bit_count.get_unpack_value() >= 8 and \
self._clr_used.get_unpack_value() != 0:
value_cell = \
self._clr_used.get_unpack_value() * bytes_per_cell
self._color_table = \
byte_array[self._offset:self._offset + value_cell]
self._color_table = \
list(zip(*[iter(self._color_table)] * bytes_per_cell))
if self._clr_important.get_unpack_value() != 0:
self._color_table = self._color_table[:self._clr_important]
else:
self._color_table = None
class BitmapInfo(Info):
"""docstring for BitmapInfo"""
__metaclass__ = ABCMeta
def __init__(self, byte_array):
self._offset = 14
super(BitmapInfo, self).__init__(
"BitmapInfo",
None,
self._offset,
None,
"To store detailed information about the " "bitmap image and define the pixel format",
)
self._size_header = Info(
"bcSize", byte_array[self._offset : self._offset + 4], self._offset, 4, "The size of this header"
)
self._offset += 4
version = {
12: "BITMAPCORE||OS21XHEADER",
40: "BITMAPINFOHEADER",
52: "BITMAPV2INFOHEADER",
56: "BITMAPV3INFOHEADER",
64: "OS22XBITMAPHEADER",
108: "BITMAPV4HEADER",
124: "BITMAPV5HEADER",
}[self._size_header.get_unpack_value()]
self._version = Info("Version", version, None, None, None)
def get_list_info(self):
return [self, self._version, self._size_header, self._width, self._height, self._planes, self._bit_count]
def get_all_info(self):
info_field = "".join(map(lambda x: x.get_all_data(), self.get_list_info()))
return info_field
@abstractmethod
def fill_color_table(self, byte_array):
pass
def get_color_table(self):
return self._color_table
def get_size_header(self):
return self._size_header
def get_version(self):
return self._version
def get_width(self):
return self._width
def get_height(self):
return self._height
def get_planes(self):
return self._planes
def get_bit_count(self):
return self._bit_count