Exemplo n.º 1
0
def rasterize(encoded_image_data):
    for i in range(0, len(encoded_image_data), CHUNK_SIZE):
        buffer = bytearray()
        chunk = encoded_image_data[i:i + CHUNK_SIZE]

        if chunk == ZERO_CHUNK:
            buffer += ZERO_COMMAND
        else:
            packed_chunk = packbits.encode(chunk)

            buffer += RASTER_COMMAND
            buffer += len(packed_chunk).to_bytes(2, "little")
            buffer += packbits.encode(chunk)

        yield buffer
Exemplo n.º 2
0
def printImg(img, end=False):
    "Sends a PIL image object to printer after converting it to the proper format."
    imgW, imgH = img.size
    img = img.transpose(Image.ROTATE_270)  ##Rotate to fit in the printer
    img = img.transpose(Image.FLIP_LEFT_RIGHT)
    ilist = list(img.getdata())
    img.save("rot.png")
    img.close()
    ##Below is the starting string for a Brother 710-W printer with no cut after
    kid_b1 = b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x001B\x40\x1B\x69\x61\x01\x1B\x69\x55\x4A\x00\x0C\xD4\x3D\x7E\xBE\x4D\x36\x00\x00\x0A\x00\x00\x00\x1B\x69\x7A\x8E\x0B\x3E\x64\x55\x04\x00\x00\x00\x00\x1B\x69\x4D\x00\x1B\x69\x4B\x00\x1B\x69\x64\x00\x00\x4D\x02'
    ##This is the starting string with a cut after
    parent_b1 = b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x001B\x40\x1B\x69\x61\x01\x1B\x69\x55\x4A\x00\x0C\xD4\x3D\x7E\xBE\x4D\x36\x00\x00\x09\x00\x00\x00\x1B\x69\x7A\x8E\x0B\x3E\x64\x55\x04\x00\x00\x00\x00\x1B\x69\x4D\x00\x1B\x69\x4B\x08\x1B\x69\x64\x00\x00\x4D\x02'
    b2 = b""
    b3 = b"\x1a"  ##Finish and cut paper
    header = b"\x67\x00"
    ##This loop converts the pixels into the bytes the printer can read
    ##hopefully as fast as possible.
    for x in range(0, imgW):
        r = ilist[x * 720:((x * 720) + 720)]
        row = __convert(r)
        row = packbits.encode(row)
        ##This replaces an empty row with the code for an empty road
        if row == b"\xA7\x00":
            b2 += b'Z'
        else:
            b2 += header + chr(len(row)).encode() + row
    if end:
        b1 = parent_b1
    else:
        b1 = kid_b1
    return (b''.join([b1, b2, b3]))
Exemplo n.º 3
0
 def add_raster_data(self, image, second_image=None):
     """ image: Pillow Image() """
     logger.info("raster_image_size: {0}x{1}".format(*image.size))
     if image.size[0] != self.get_pixel_width():
         fmt = 'Wrong pixel width: {}, expected {}'
         raise BrotherQLRasterError(fmt.format(image.size[0], self.get_pixel_width()))
     images = [image]
     if second_image:
         if image.size != second_image.size:
             fmt = "First and second image don't have the same dimesions: {} vs {}."
             raise BrotherQLRasterError(fmt.format(image.size, second_image.size))
         images.append(second_image)
     frames = []
     for image in images:
         image = image.transpose(Image.FLIP_LEFT_RIGHT)
         image = image.convert("1")
         frames.append(bytes(image.tobytes(encoder_name='raw')))
     frame_len = len(frames[0])
     row_len = images[0].size[0]//8
     start = 0
     file_str = BytesIO()
     while start + row_len <= frame_len:
         for i, frame in enumerate(frames):
             row = frame[start:start+row_len]
             if second_image:
                 file_str.write(b'\x77\x01' if i == 0 else b'\x77\x02')
             else:
                 file_str.write(b'\x67\x00')
             if self._compression:
                 row = packbits.encode(row)
             file_str.write(bytes([len(row)]))
             file_str.write(row)
         start += row_len
     self.data += file_str.getvalue()
Exemplo n.º 4
0
def compress_constant_rle(
        fd,  # type: BinaryIO
        value,  # type: int
        width,  # type: int
        rows,  # type: int
        depth,  # type: int
        version  # type: int
):  # type: (...) -> None
    """
    Write a virtual image containing a constant to a runlength-encoded
    stream.

{}
    """
    if depth == 1:  # pragma: no cover
        raise ValueError("rle compression is not supported for 1-bit images")

    row = _make_constant_row(value, width, depth)
    packed = packbits.encode(row.tobytes())

    if version == 1:
        lengths = np.full((rows, ), len(packed), dtype='>u2')
    else:
        lengths = np.full((rows, ), len(packed), dtype='>u4')
    fd.write(lengths.tobytes())

    for i in range(rows):
        fd.write(packed)
Exemplo n.º 5
0
def encode_raster_transfer(data):
    """ Encode 1 bit per pixel image data for transfer over serial to the printer """
    buf = bytearray()

    # Send in chunks of 1 line (128px @ 1bpp = 16 bytes)
    # This mirrors the official app from Brother. Other values haven't been tested.
    chunk_size = 16

    for i in range(0, len(data), chunk_size):
        chunk = data[i:i + chunk_size]

        # Encode as tiff
        packed_chunk = packbits.encode(chunk)

        # Write header
        buf.append(TRANSFER_COMMAND)

        # Write number of bytes to transfer (n1 + n2*256)
        length = len(packed_chunk)
        buf.append(int(length % 256))
        buf.append(int(length / 256))

        # Write data
        buf.extend(packed_chunk)

    return buf
Exemplo n.º 6
0
def encode_packbits(data, width, height, depth, version):
    row_size = width * depth // 8
    with io.BytesIO(data) as fp:
        rows = [packbits.encode(fp.read(row_size)) for _ in range(height)]
    bytes_counts = array.array(('H', 'I')[version - 1], map(len, rows))
    encoded = b''.join(rows)

    with io.BytesIO() as fp:
        write_be_array(fp, bytes_counts)
        fp.write(encoded)
        result = fp.getvalue()

    return result
Exemplo n.º 7
0
def compress_rle(
        fd,  # type: BinaryIO
        image,  # type: np.ndarray
        depth,  # type: int
        version  # type: int
):  # type: (...) -> None
    """
    Write a Numpy array to a run length encoded stream.

{}
    """
    if depth == 1:  # pragma: no cover
        raise ValueError("rle compression is not supported for 1-bit images")

    start = fd.tell()
    if version == 1:
        fd.seek(image.shape[0] * 2, 1)
        lengths = np.empty((len(image), ), dtype='>u2')
    else:
        fd.seek(image.shape[0] * 4, 1)
        lengths = np.empty((len(image), ), dtype='>u4')

    if util.needs_byteswap(image):
        for i, row in enumerate(image):
            row = util.do_byteswap(row)
            packed = packbits.encode(row)
            lengths[i] = len(packed)
            fd.write(packed)
    else:
        for i, row in enumerate(image):
            packed = packbits.encode(row)
            lengths[i] = len(packed)
            fd.write(packed)

    end = fd.tell()
    fd.seek(start)
    fd.write(lengths.tobytes())
    fd.seek(end)
Exemplo n.º 8
0
 def add_raster_data(self, np_array):
     """ np_array: numpy array of 1-bit values """
     np_array = np.fliplr(np_array)
     logger.info("raster_image_size: {1}x{0}".format(*np_array.shape))
     if np_array.shape[1] != self.get_pixel_width():
         fmt = 'Wrong pixel width: {}, expected {}'
         raise BrotherQLRasterError(fmt.format(np_array.shape[0], self.get_pixel_width()))
     for row in np_array:
         self.data += b'\x67\x00'
         row = bytes(np.packbits(row))
         if self._compression:
             row = packbits.encode(row)
         self.data += bytes([len(row)])
         self.data += row
Exemplo n.º 9
0
    def add_raster_data(self, image, second_image=None):
        """
        Add the image data to the instructions.
        The provided image has to be binary (every pixel
        is either black or white).

        :param PIL.Image.Image image: The image to be converted and added to the raster instructions
        :param PIL.Image.Image second_image: A second image with a separate color layer (red layer for the QL-800 series)
        """
        logger.debug("raster_image_size: {0}x{1}".format(*image.size))
        if image.size[0] != self.get_pixel_width():
            fmt = 'Wrong pixel width: {}, expected {}'
            raise BrotherQLRasterError(
                fmt.format(image.size[0], self.get_pixel_width()))
        images = [image]
        if second_image:
            if image.size != second_image.size:
                fmt = "First and second image don't have the same dimesions: {} vs {}."
                raise BrotherQLRasterError(
                    fmt.format(image.size, second_image.size))
            images.append(second_image)
        frames = []
        for image in images:
            image = image.transpose(Image.FLIP_LEFT_RIGHT)
            image = image.convert("1")
            frames.append(bytes(image.tobytes(encoder_name='raw')))
        frame_len = len(frames[0])
        row_len = images[0].size[0] // 8
        start = 0
        file_str = BytesIO()
        while start + row_len <= frame_len:
            for i, frame in enumerate(frames):
                row = frame[start:start + row_len]
                if self._compression:
                    row = packbits.encode(row)
                translen = len(row)  # number of bytes to be transmitted
                if self.model.startswith('PT'):
                    file_str.write(b'\x47')
                    file_str.write(bytes([translen % 256, translen // 256]))
                else:
                    if second_image:
                        file_str.write(b'\x77\x01' if i == 0 else b'\x77\x02')
                    else:
                        file_str.write(b'\x67\x00')
                    file_str.write(bytes([translen]))
                file_str.write(row)
            start += row_len
        self.data += file_str.getvalue()
Exemplo n.º 10
0
    def processFrame(self, start, row_len, frame_len, frame, index, g_data):

        file_str = BytesIO()
        while start + row_len <= frame_len:
            row = frame[start:start+row_len]
            start += row_len
            #self.data += b'\x67\x00' # g 0x00
            file_str.write(b'\x67\x00')
            if self._compression:
                row = packbits.encode(row)
            #self.data += bytes([len(row)])
            #self.data += row
            file_str.write(bytes([len(row)]))
            file_str.write(row)

        g_data[index] = file_str.getvalue()
Exemplo n.º 11
0
def encode_packbits(data, width, height, depth, version):
    bytes_counts = array.array(('H', 'I')[version - 1])
    encoded = b''
    with io.BytesIO(data) as fp:
        row_size = width * depth // 8
        for index in range(height):
            row = packbits.encode(fp.read(row_size))
            bytes_counts.append(len(row))
            encoded += row

    with io.BytesIO() as fp:
        write_be_array(fp, bytes_counts)
        fp.write(encoded)
        result = fp.getvalue()

    return result
Exemplo n.º 12
0
def encode_line(bitmap_line: bytes, tape_info: TapeInfo) -> bytes:
    # The number of bits we need to add left or right is not always a multiple
    # of 8, so we need to convert our line into an int, shift it over by the
    # left margin and convert it to back again, padding to 16 bytes.

    line_int = int.from_bytes(bitmap_line, byteorder='big')
    line_int <<= tape_info.rmargin
    padded = line_int.to_bytes(16, byteorder='big')

    # pad to 16 bytes
    compressed = packbits.encode(padded)
    logger.debug("original bitmap: %s", bitmap_line)
    logger.debug("padded bitmap %s", padded)
    logger.debug("packbits compressed %s", compressed)
    # <h: big endian short (2 bytes)
    prefix = struct.pack("<H", len(compressed))

    return prefix + compressed
Exemplo n.º 13
0
 def add_raster_data(self, image):
     """ image: Pillow Image() """
     logger.info("raster_image_size: {0}x{1}".format(*image.size))
     image = image.transpose(Image.FLIP_LEFT_RIGHT)
     image = image.convert("1")
     if image.size[0] != self.get_pixel_width():
         fmt = 'Wrong pixel width: {}, expected {}'
         raise BrotherQLRasterError(
             fmt.format(image.size[0], self.get_pixel_width()))
     frame = bytes(image.tobytes(encoder_name='raw'))
     frame_len = len(frame)
     row_len = image.size[0] // 8
     start = 0
     file_str = BytesIO()
     while start + row_len <= frame_len:
         row = frame[start:start + row_len]
         start += row_len
         file_str.write(b'\x67\x00')
         if self._compression:
             row = packbits.encode(row)
         file_str.write(bytes([len(row)]))
         file_str.write(row)
     self.data += file_str.getvalue()
Exemplo n.º 14
0
def test_encode_long_raw2():
    data = b'12345678' * 16
    encoded = packbits.encode(data)
    assert packbits.decode(encoded) == data
Exemplo n.º 15
0
def test_encode_switching_rle():
    encoded = packbits.encode(b'1122')
    assert packbits.decode(encoded) == b'1122'
Exemplo n.º 16
0
def test_encode_long_raw():
    data = b'12345678' * 17
    encoded = packbits.encode(data)
    print(encoded)
    assert packbits.decode(encoded) == data
Exemplo n.º 17
0
def test_encode2():
    encoded = packbits.encode(b"112112")
    assert packbits.decode(encoded) == b"112112"
Exemplo n.º 18
0
def test_encode_single():
    encoded = packbits.encode(b"X")
    assert encoded == b"\x00X"
    assert packbits.decode(encoded) == b"X"
Exemplo n.º 19
0
def test_encode_decode():
    encoded = packbits.encode(RESULT)
    decoded = packbits.decode(encoded)
    assert decoded == RESULT
Exemplo n.º 20
0
def test_restart_rle():
    data = b"1" * 127 + b"foo"
    encoded = packbits.encode(data)
    assert packbits.decode(encoded) == data
Exemplo n.º 21
0
def test_encode_single():
    encoded = packbits.encode(b'X')
    assert encoded == b'\x00X'
    assert packbits.decode(encoded) == b'X'
Exemplo n.º 22
0
def test_encode_single():
    encoded = packbits.encode(b'X')
    assert encoded == b'\x00X'
    assert packbits.decode(encoded) == b'X'
Exemplo n.º 23
0
def test_encode_long():
    data = b'1' * 128 + b'12345678' * 17
    encoded = packbits.encode(data)
    assert packbits.decode(encoded) == data
Exemplo n.º 24
0
def test_raw():
    encoded = packbits.encode(b'123')
    assert encoded == b'\x02123'
    assert packbits.decode(encoded) == b'123'
Exemplo n.º 25
0
def test_encode_long_raw():
    data = b'12345678' * 16
    encoded = packbits.encode(data)
    assert packbits.decode(encoded) == data
Exemplo n.º 26
0
def test_restart_rle():
    data = b'1' * 127 + b'foo'
    encoded = packbits.encode(data)
    assert packbits.decode(encoded) == data
Exemplo n.º 27
0
def test_encode_long_rle3():
    data = b'1' * 128
    encoded = packbits.encode(data)
    assert packbits.decode(encoded) == data
Exemplo n.º 28
0
def test_encode_long():
    data = b'1' * 128 + b'12345678' * 17
    encoded = packbits.encode(data)
    assert packbits.decode(encoded) == data
Exemplo n.º 29
0
def test_raw():
    encoded = packbits.encode(b'123')
    assert encoded == b'\x02123'
    assert packbits.decode(encoded) == b'123'
Exemplo n.º 30
0
def test_encode_empty():
    assert packbits.encode(b'') == b''
Exemplo n.º 31
0
def test_encode2():
    encoded = packbits.encode(b'112112')
    assert packbits.decode(encoded) == b'112112'
Exemplo n.º 32
0
def test_encode_long_raw():
    data = b"12345678" * 17
    encoded = packbits.encode(data)
    print(encoded)
    assert packbits.decode(encoded) == data
Exemplo n.º 33
0
def test_encode_switching_rle():
    encoded = packbits.encode(b'1122')
    assert packbits.decode(encoded) == b'1122'
Exemplo n.º 34
0
def test_encode_long_rle2():
    data = b"1" * 127
    encoded = packbits.encode(data)
    assert packbits.decode(encoded) == data
Exemplo n.º 35
0
def test_encode_long_rle3():
    data = b'1' * 128
    encoded = packbits.encode(data)
    assert packbits.decode(encoded) == data
Exemplo n.º 36
0
def test_raw():
    encoded = packbits.encode(b"123")
    assert encoded == b"\x02123"
    assert packbits.decode(encoded) == b"123"
Exemplo n.º 37
0
def test_restart_rle():
    data = b'1' * 127 + b'foo'
    encoded = packbits.encode(data)
    assert packbits.decode(encoded) == data
Exemplo n.º 38
0
def test_encode_decode():
    encoded = packbits.encode(RESULT)
    decoded = packbits.decode(encoded)
    assert decoded == RESULT
Exemplo n.º 39
0
def test_encode2():
    encoded = packbits.encode(b'112112')
    assert packbits.decode(encoded) == b'112112'
Exemplo n.º 40
0
def test_encode_empty():
    assert packbits.encode(b'') == b''
Exemplo n.º 41
0
    def add_raster_data(self, image, cores=1):
        """ image: Pillow Image() """
        logger.info("raster_image_size: {0}x{1}".format(*image.size))
        image = image.transpose(Image.FLIP_LEFT_RIGHT)
        image = image.convert("1")
        if image.size[0] != self.get_pixel_width():
            fmt = 'Wrong pixel width: {}, expected {}'
            raise BrotherQLRasterError(fmt.format(image.size[0], self.get_pixel_width()))
        frame = bytes(image.tobytes(encoder_name='raw'))

        frame_len = len(frame)
        row_len = image.size[0]//8
        start = 0

        file_str = BytesIO()
         
        if cores > 1:

            len_prop = float(float(frame_len) / (float(row_len) * float(cores)))
            len_check = len_prop.is_integer()
            logger.debug("Using " + str(cores) + " cores in add_raster_data.")
            processes = {}
            manager = Manager()
            g_data = manager.list(['']*cores)

            #Check if we can equally assign subframes to processes, or adjust their length to fit.
            if len_check:
                subframe_len = int(frame_len/cores)
                for i in range(0, cores):
                    
                    subframe_start =int(subframe_len * i)
                    subframe_limit = int(subframe_len * (i+1))
                    subframe = frame[subframe_start:subframe_limit]
                    processes[i] = Process(target=self.processFrame, args=(0, row_len, subframe_len, subframe, i, g_data,))
                    processes[i].start()
            else:
                num_rows = int(float(frame_len) / (float(row_len) * float(cores)))
                last_limit = 0
                subframe_len = int(row_len * num_rows)
                for i in range(0, cores - 1):
                    
                    subframe_start =int(subframe_len * i)
                    subframe_limit = int(subframe_len * (i+1))
                    last_limit = subframe_limit
                    subframe = frame[subframe_start:subframe_limit]
                    processes[i] = Process(target=self.processFrame, args=(0, row_len, subframe_len, subframe, i, g_data,))
                    processes[i].start()

                subframe = frame[last_limit:frame_len]
                processes[cores - 1] = Process(target=self.processFrame, args=(0, row_len, frame_len - last_limit, subframe, cores - 1, g_data,))
                processes[cores-1].start()

            for i in range(0, cores):
                processes[i].join()

            for i in range(0, cores):
                self.data += g_data[i]

        else:

            logger.debug("Using 1 core in add_raster_data (" + str(cores) + " given).")

            while start + row_len <= frame_len:
                row = frame[start:start+row_len]
                start += row_len
                file_str.write(b'\x67\x00')
                if self._compression:
                    row = packbits.encode(row)
                file_str.write(bytes([len(row)]))
                file_str.write(row)

            self.data += file_str.getvalue()