Ejemplo n.º 1
0
    def send_packet(self, name, data=b""):
        """ Sends a packet """

        if self.closed:
            return

        self.log_packet("# send", name)

        # Prepend ident
        key = (self.protocol_version, self.protocol_mode, self.send_direction, name)
        try:
            ident = packets.packet_idents[key]
        except KeyError:
            raise ProtocolError("No ID known for packet: %s" % (key,))
        data = Buffer.pack_varint(ident) + data

        if self.compression_enabled:
            # Compress data and prepend uncompressed data length
            if len(data) >= self.compression_threshold:
                data = Buffer.pack_varint(len(data)) + zlib.compress(data)
            else:
                data = Buffer.pack_varint(0) + data

        # Prepend packet length
        data = self.buff_type.pack_varint(len(data)) + data

        # Encrypt
        data = self.cipher.encrypt(data)

        # Send
        self.transport.write(data)
Ejemplo n.º 2
0
    def send_packet(self, name, data=b""):
        """ Sends a packet """

        if self.closed:
            return

        self.log_packet("# send", name)

        # Prepend ident
        key = (self.protocol_version, self.protocol_mode, self.send_direction,
               name)
        try:
            ident = packets.packet_idents[key]
        except KeyError:
            raise ProtocolError("No ID known for packet: %s" % (key, ))
        data = Buffer.pack_varint(ident) + data

        if self.compression_enabled:
            # Compress data and prepend uncompressed data length
            if len(data) >= self.compression_threshold:
                data = Buffer.pack_varint(len(data)) + zlib.compress(data)
            else:
                data = Buffer.pack_varint(0) + data

        # Prepend packet length
        data = self.buff_type.pack_varint(len(data)) + data

        # Encrypt
        data = self.cipher.encrypt(data)

        # Send
        self.transport.write(data)
Ejemplo n.º 3
0
    def data_received(self, data):
        # Decrypt data
        data = self.cipher.decrypt(data)

        # Add it to our buffer
        self.recv_buff.add(data)

        # Read some packets
        while not self.closed:
            # Save the buffer, in case we read an incomplete packet
            self.recv_buff.save()

            # Try to read a packet
            try:
                max_bits = 32 if self.protocol_mode == "play" else 21
                packet_length = self.recv_buff.unpack_varint(max_bits=max_bits)
                packet_body = self.recv_buff.read(packet_length)

            # Incomplete packet read, restore the buffer.
            except BufferUnderrun:
                self.recv_buff.restore()
                break

            # Load the packet body into a buffer
            packet_buff = self.buff_type()
            packet_buff.add(packet_body)

            try:  # Catch protocol errors
                try:  # Catch buffer overrun/underrun
                    if self.compression_enabled:
                        uncompressed_length = packet_buff.unpack_varint()

                        if uncompressed_length > 0:
                            data = zlib.decompress(packet_buff.read())
                            packet_buff = Buffer()
                            packet_buff.add(data)
                    ident = packet_buff.unpack_varint()
                    key = (self.protocol_version, self.protocol_mode,
                           self.recv_direction, ident)
                    try:
                        name = packets.packet_names[key]
                    except KeyError:
                        raise ProtocolError("No name known for packet: %s" %
                                            (key, ))
                    self.packet_received(packet_buff, name)

                except BufferUnderrun:
                    raise ProtocolError("Packet is too short!")

                if len(packet_buff) > 0:
                    raise ProtocolError("Packet is too long!")

            except ProtocolError as e:
                self.protocol_error(e)
                break

            # We've read a complete packet, so reset the inactivity timeout
            self.connection_timer.restart()
Ejemplo n.º 4
0
def test_save_discard_restore():
    buffer = Buffer()
    buffer.add(b"spam")
    buffer.save()
    assert len(buffer) == 4
    buffer.discard()
    assert len(buffer) == 0
    buffer.restore()
    assert len(buffer) == 4
Ejemplo n.º 5
0
def test_read():
    buffer = Buffer()
    buffer.add(b"spam")
    assert buffer.read(2) == b"sp"
    assert buffer.read(2) == b"am"
    buffer.add(b"eggs")
    assert buffer.read() == b"eggs"
    with pytest.raises(BufferUnderrun):
        buffer.read(1)
Ejemplo n.º 6
0
    def data_received(self, data):
        # Decrypt data
        data = self.cipher.decrypt(data)

        # Add it to our buffer
        self.recv_buff.add(data)

        # Read some packets
        while not self.closed:
            # Save the buffer, in case we read an incomplete packet
            self.recv_buff.save()

            # Try to read a packet
            try:
                packet_length = self.recv_buff.unpack_varint()
                packet_body = self.recv_buff.read(packet_length)

            # Incomplete packet read, restore the buffer.
            except BufferUnderrun:
                self.recv_buff.restore()
                break

            # Load the packet body into a buffer
            packet_buff = self.buff_type()
            packet_buff.add(packet_body)

            try:  # Catch protocol errors
                try:  # Catch buffer overrun/underrun
                    if self.compression_enabled:
                        uncompressed_length = packet_buff.unpack_varint()

                        if uncompressed_length > 0:
                            data = zlib.decompress(packet_buff.read())
                            packet_buff = Buffer()
                            packet_buff.add(data)
                    ident = packet_buff.unpack_varint()
                    key = (self.protocol_version, self.protocol_mode, self.recv_direction, ident)
                    try:
                        name = packets.packet_names[key]
                    except ValueError:
                        raise ProtocolError("No name known for packet: %s" % (key,))
                    self.packet_received(packet_buff, name)

                except BufferUnderrun:
                    raise ProtocolError("Packet is too short!")

                if len(packet_buff) > 0:
                    raise ProtocolError("Packet is too long!")

            except ProtocolError as e:
                self.protocol_error(e)
                break

            # We've read a complete packet, so reset the inactivity timeout
            self.connection_timer.restart()
Ejemplo n.º 7
0
 def __init__(self,
              topleft=(-1000, -1000),
              bottomright=(1000, 1000),
              cuty=0,
              world=1,
              norenderblocks=["none"],
              multymatblocks=[
                  35, 43, 44, 95, 97, 98, 125, 126, 139, 155, 159, 160, 171
              ],
              colormats={}):
     self.colormats = colormats
     self.topleft = topleft
     self.bottomright = bottomright
     self.cuty = cuty
     self.world = world
     self.norenderblocks = norenderblocks
     self.multymatblocks = multymatblocks
     self.chunks = {}
     self.materials = {}
     self.neightbors = {}
     self.worldnum = 0
     self.chunkbuffer = Buffer()
Ejemplo n.º 8
0
 def __init__(self, topleft=(-1000, -1000), bottomright=(1000, 1000), cuty=0, world=1, norenderblocks=["none"], multymatblocks=[35, 43, 44, 95, 97, 98, 125, 126, 139, 155, 159, 160, 171], colormats={}):
     self.colormats = colormats
     self.topleft = topleft
     self.bottomright = bottomright
     self.cuty = cuty
     self.world = world
     self.norenderblocks = norenderblocks
     self.multymatblocks = multymatblocks
     self.chunks = {}
     self.materials= {}
     self.neightbors = {}
     self.worldnum = 0
     self.chunkbuffer = Buffer()
Ejemplo n.º 9
0
def test_pack_json():
    assert Buffer.pack_json({"spam": "eggs"}) == b'\x10{"spam": "eggs"}'
Ejemplo n.º 10
0
def test_pack_string():
    assert Buffer.pack_string("spam") == b"\x04spam"
Ejemplo n.º 11
0
def test_pack():
    for fmt, data, values in pack_unpack_vectors:
        if not isinstance(values, tuple):
            values = (values,)
        assert Buffer.pack(fmt, *values) == data
Ejemplo n.º 12
0
 def testVarInt(self):
     value = 5533
     buff = Buffer()
     buff.buff = buff.pack_varint(value)
     self.assertTrue(buff.unpack_varint() == value)
Ejemplo n.º 13
0
Archivo: tests.py Proyecto: aaps/quarry
    def testString(self):
        text = "more tests, and stuff"

        buff = Buffer()
        buff.buff = buff.pack_string(text)
        self.assertTrue(text == buff.unpack_string())
Ejemplo n.º 14
0
class ChunkParser(object):
    def __init__(self,
                 topleft=(-1000, -1000),
                 bottomright=(1000, 1000),
                 cuty=0,
                 world=1,
                 norenderblocks=["none"],
                 multymatblocks=[
                     35, 43, 44, 95, 97, 98, 125, 126, 139, 155, 159, 160, 171
                 ],
                 colormats={}):
        self.colormats = colormats
        self.topleft = topleft
        self.bottomright = bottomright
        self.cuty = cuty
        self.world = world
        self.norenderblocks = norenderblocks
        self.multymatblocks = multymatblocks
        self.chunks = {}
        self.materials = {}
        self.neightbors = {}
        self.worldnum = 0
        self.chunkbuffer = Buffer()

    def set_top_left(self, topleft=(-1000, -1000)):
        self.topleft = topleft

    def set_bottom_right(self, bottomright=(1000, 1000)):
        self.bottomright = bottomright

    def set_cut_y(self, cuty=0):
        self.cuty = cuty

    def set_world(self, world=1):
        self.world = world

    def set_no_render_blocks(self, norenderblocks=["none"]):
        self.norenderblocks = norenderblocks

    def get_chunks(self, row):
        chunkposses = []
        chunkxz = ast.literal_eval(row[1])
        chunkdata = base64.standard_b64decode(row[5])
        self.chunkbuffer.add(chunkdata)
        blocks = []
        self.chunkbuffer.save()

        matsamples = []
        if chunkxz not in self.chunks:
            self.chunks.update({chunkxz: {'blocks': []}})

        rightcounter = 0
        if chunkxz not in chunkposses and chunkxz[0] >= self.topleft[
                0] and chunkxz[1] >= self.topleft[1] and chunkxz[
                    0] <= self.bottomright[0] and chunkxz[
                        1] <= self.bottomright[1]:
            chunkposses.append(chunkxz)

            for index1 in range(16):

                if (int(row[3]) & (1 << index1)):

                    bitsperblock = self.chunkbuffer.unpack('B')
                    usespalette = True

                    if bitsperblock == 0:
                        bitsperblock = 13
                        usespalette = False

                    palletelength = self.chunkbuffer.unpack_varint()

                    palette = []
                    blockdata = []
                    maxvalue = (1 << bitsperblock) - 1
                    for palindex in range(palletelength):
                        palette.append(self.chunkbuffer.unpack_varint())

                    dataarrlength = self.chunkbuffer.unpack_varint()

                    # print "DAlength: " + str(dataarrlength) + " index: " +  str(index1) + " bitsperblock: " + str(bitsperblock)
                    for dataind in range(dataarrlength):
                        blockdata.append(self.chunkbuffer.unpack('Q'))

                    unneededlength = (256 * 16)
                    for notneed in range(unneededlength):
                        self.chunkbuffer.unpack('B')

                    for ypos in xrange(0, 16):
                        for zpos in xrange(0, 16):
                            for xpos in xrange(0, 16):
                                goodindex = xpos + (zpos * 16) + (ypos * 256)
                                startlong = (goodindex * bitsperblock) // 64
                                startoffset = (goodindex * bitsperblock) % 64
                                endlong = (
                                    (goodindex + 1) * bitsperblock - 1) // 64
                                if startlong == endlong and len(blockdata) > 0:
                                    temp = (blockdata[startlong] >>
                                            startoffset) & maxvalue

                                elif len(blockdata) > 0:
                                    endoffset = 64 - startoffset
                                    temp = (blockdata[startlong] >> startoffset
                                            | blockdata[endlong] << endoffset
                                            ) & maxvalue
                                else:
                                    temp = -1

                                if usespalette and temp > -1:  # convert to global palette
                                    btype = palette[temp] >> 4
                                    bmeta = palette[temp] & 15
                                    bbtest = ((xpos + (chunkxz[0] * 16),
                                               zpos + (chunkxz[1] * 16),
                                               ypos + (index1 * 16)),
                                              int(btype), int(bmeta))
                                    # blocks.append(bbtest)
                                elif temp > -1:
                                    btype = temp >> 4
                                    bmeta = temp & 15
                                    bbtest = ((xpos + (chunkxz[0] * 16),
                                               zpos + (chunkxz[1] * 16),
                                               ypos + (index1 * 16)),
                                              int(btype), int(bmeta))
                                    # blocks.append(bbtest)
                                else:
                                    bbtest = ((xpos + (chunkxz[0] * 16),
                                               zpos + (chunkxz[1] * 16),
                                               ypos + (index1 * 16)), 0, 0)
                                    # blocks.append(bbtest)

                                if index1 > self.cuty:
                                    self.chunks[chunkxz]['blocks'].append(
                                        bbtest)

        self.chunkbuffer.discard()
        # return blocks

    def fill_mat_indexes(self):
        for chunk in self.chunks:
            for block in self.chunks[chunk]:
                if len(self.chunks[chunk][block]) > 0:
                    for idindex in self.chunks[chunk][block]:
                        position = idindex[0][0], idindex[0][1] * -1, idindex[
                            0][2]
                        if idindex[1] > 0 and idindex[1] < 256 and idindex[
                                1] not in self.norenderblocks:
                            blockinfo = {
                                position: {
                                    'meta': idindex[2],
                                    'faces': []
                                }
                            }
                            if idindex[1] in self.multymatblocks:
                                if idindex[1] in [182, 126, 44
                                                  ] and idindex[2] > 7:
                                    idindex = idindex[0], idindex[
                                        1], idindex[2] - 8
                                if idindex[1] in [125, 181, 43
                                                  ] and idindex[2] > 7:
                                    idindex = idindex[0], idindex[
                                        1], idindex[2] - 8
                                if (idindex[1],
                                        idindex[2]) not in self.materials:
                                    self.materials.update({
                                        (idindex[1], idindex[2]):
                                        blockinfo
                                    })
                                else:
                                    self.materials[(
                                        idindex[1],
                                        idindex[2])].update(blockinfo)
                            else:
                                if (idindex[1], 0) not in self.materials:
                                    self.materials.update({
                                        (idindex[1], 0):
                                        blockinfo
                                    })
                                else:
                                    self.materials[(idindex[1],
                                                    0)].update(blockinfo)
                self.chunks[chunk][block] = None
        self.chunks = None

    def gen_faces_neighbors(self, agressiveremoval=False):
        allmaterials = {}
        print 'find material neightbors for ' + str(len(
            self.materials)) + ' materials'
        # todo below is somewhat sloppy refactor material

        for mat in self.materials:
            if mat in self.materials and mat in self.colormats:
                for blockindex in self.materials[mat]:
                    self.materials[mat][blockindex][
                        'interneighbor'] = self.colormats[mat]['interneighbor']
                    self.materials[mat][blockindex][
                        'extraneighbor'] = self.colormats[mat]['extraneighbor']
                allmaterials.update(self.materials[mat])

        for mat in self.materials:
            if mat in self.materials and mat in self.colormats:
                self.neightbors[mat] = {}
        for mat in self.materials:
            if mat in self.materials and mat in self.colormats:
                for block in self.materials[mat]:
                    if agressiveremoval:
                        blockstocheck = allmaterials
                    else:
                        blockstocheck = self.materials[mat]
                    self.neightbors[mat][block] = {
                        'meta': self.materials[mat][block]['meta'],
                        'faces': []
                    }
                    if (block[0] - 1, block[1], block[2]) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(1)
                    elif blockstocheck[(block[0] - 1, block[1],
                                        block[2])]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(1)
                    if (block[0] + 1, block[1], block[2]) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(2)
                    elif blockstocheck[(block[0] + 1, block[1],
                                        block[2])]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(2)
                    if (block[0], block[1] - 1, block[2]) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(3)
                    elif blockstocheck[(block[0], block[1] - 1,
                                        block[2])]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(3)
                    if (block[0], block[1] + 1, block[2]) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(4)
                    elif blockstocheck[(block[0], block[1] + 1,
                                        block[2])]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(4)
                    if (block[0], block[1], block[2] - 1) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(5)
                    elif blockstocheck[(block[0], block[1],
                                        block[2] - 1)]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(5)
                    if (block[0], block[1], block[2] + 1) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(6)
                    elif blockstocheck[(block[0], block[1],
                                        block[2] + 1)]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(6)
                self.materials[mat] = None

    def remove_super_cosy(self):
        loneneighbors = {}
        for mat in self.neightbors:
            if mat in self.colormats and 'interneighbor' in self.colormats[
                    mat] and self.colormats[mat]['interneighbor']:
                removeneibors = False
            else:
                removeneibors = True
            loneneighbors[mat] = {}
            for block in self.neightbors[mat]:
                if len(self.neightbors[mat][block]
                       ['faces']) > 0 or not removeneibors:
                    loneneighbors[mat][block[0] + 0.5, block[1] + 0.5,
                                       block[2] +
                                       0.5] = self.neightbors[mat][block]
            self.neightbors[mat] = None
        return loneneighbors

    def world_from_sample(self, sample):
        overworldblocks = [
            1, 2, 3, 4, 6, 8, 9, 12, 13, 15, 16, 17, 18, 37, 38, 39, 40
        ]
        netherblocks = [10, 11, 87, 88, 112, 113, 114, 115, 153]
        theendblocks = [121]
        overscore = 0
        endscore = 0
        netherscore = 0
        for block in sample:
            if block in overworldblocks:
                overscore += 1
            if block in netherblocks:
                netherscore += 1
            if block in theendblocks:
                endscore += 1
        if endscore > overscore and endscore > netherscore:
            return 3
        elif netherscore > endscore and netherscore > overscore:
            return 2
        else:
            return 1
Ejemplo n.º 15
0
def test_pack_uuid():
    assert Buffer.pack_uuid(UUID.from_bytes(uuid_vector)) == uuid_vector
Ejemplo n.º 16
0
def test_unpack():
    buffer = Buffer()
    for fmt, data, values in pack_unpack_vectors:
        buffer.add(data)
        assert buffer.unpack(fmt) == values
Ejemplo n.º 17
0
class ChunkParser(object):

    def __init__(self, topleft=(-1000, -1000), bottomright=(1000, 1000), cuty=0, world=1, norenderblocks=["none"], multymatblocks=[35, 43, 44, 95, 97, 98, 125, 126, 139, 155, 159, 160, 171], colormats={}):
        self.colormats = colormats
        self.topleft = topleft
        self.bottomright = bottomright
        self.cuty = cuty
        self.world = world
        self.norenderblocks = norenderblocks
        self.multymatblocks = multymatblocks
        self.chunks = {}
        self.materials= {}
        self.neightbors = {}
        self.worldnum = 0
        self.chunkbuffer = Buffer()

    def set_top_left(self, topleft=(-1000, -1000)):
        self.topleft = topleft

    def set_bottom_right(self, bottomright=(1000, 1000)):
        self.bottomright = bottomright

    def set_cut_y(self, cuty=0):
        self.cuty = cuty

    def set_world(self, world=1):
        self.world = world

    def set_no_render_blocks(self, norenderblocks=["none"]):
        self.norenderblocks = norenderblocks


    def get_chunks(self, row):
        chunkposses = []
        chunkxz = ast.literal_eval(row[1])
        chunkdata = base64.standard_b64decode(row[5])
        self.chunkbuffer.add(chunkdata)
        blocks = []
        self.chunkbuffer.save()

        matsamples = []
        if chunkxz not in self.chunks:
            self.chunks.update({chunkxz:{'blocks':[]}})

        rightcounter = 0
        if chunkxz not in chunkposses and chunkxz[0] >= self.topleft[0] and chunkxz[1] >= self.topleft[1] and chunkxz[0] <= self.bottomright[0] and chunkxz[1] <= self.bottomright[1]:
            chunkposses.append(chunkxz)

            for index1 in range(16):
                

                if (int(row[3]) & (1 << index1)):
                    
                    bitsperblock = self.chunkbuffer.unpack('B')
                    usespalette = True
                    
                    if bitsperblock == 0:
                        bitsperblock = 13
                        usespalette = False

                    palletelength = self.chunkbuffer.unpack_varint()

                    palette = []
                    blockdata = []
                    maxvalue = (1 << bitsperblock) - 1
                    for palindex in range(palletelength):
                        palette.append(self.chunkbuffer.unpack_varint())

                    dataarrlength = self.chunkbuffer.unpack_varint()
                    
                    # print "DAlength: " + str(dataarrlength) + " index: " +  str(index1) + " bitsperblock: " + str(bitsperblock)
                    for dataind in range(dataarrlength):
                        blockdata.append(self.chunkbuffer.unpack('Q'))


                    unneededlength = (256*16)
                    for notneed in range(unneededlength):
                       self.chunkbuffer.unpack('B')

                    for ypos in xrange(0, 16):
                            for zpos in xrange(0, 16):
                                for xpos in xrange(0, 16):
                                    goodindex = xpos+(zpos*16)+(ypos*256)
                                    startlong = (goodindex * bitsperblock) // 64
                                    startoffset = (goodindex * bitsperblock) % 64
                                    endlong = ((goodindex + 1) * bitsperblock - 1) // 64
                                    if startlong == endlong and len(blockdata) > 0:
                                        temp = (blockdata[startlong] >> startoffset) & maxvalue

                                    elif len(blockdata) > 0:
                                        endoffset = 64 - startoffset
                                        temp = (blockdata[startlong] >> startoffset
                                                 | blockdata[endlong] << endoffset
                                                 ) & maxvalue
                                    else:
                                        temp = -1

                                    
                                    if usespalette and temp > -1:  # convert to global palette
                                        btype = palette[temp] >> 4
                                        bmeta = palette[temp] & 15
                                        bbtest = ((xpos + (chunkxz[0]*16), zpos + (chunkxz[1]*16), ypos+(index1*16)), int(btype), int(bmeta))
                                        # blocks.append(bbtest)
                                    elif temp > -1:
                                        btype = temp >> 4
                                        bmeta = temp & 15
                                        bbtest = ((xpos + (chunkxz[0]*16), zpos + (chunkxz[1]*16), ypos+(index1*16)), int(btype), int(bmeta))
                                        # blocks.append(bbtest)
                                    else:
                                        bbtest = ((xpos + (chunkxz[0]*16), zpos + (chunkxz[1]*16), ypos+(index1*16)), 0, 0)
                                        # blocks.append(bbtest)
                                    
                                    if index1 > self.cuty:
                                        self.chunks[chunkxz]['blocks'].append(bbtest)
        
                    
        self.chunkbuffer.discard()
        # return blocks



    def fill_mat_indexes(self):
        for chunk in self.chunks:
            for block in self.chunks[chunk]:
                if len(self.chunks[chunk][block]) > 0:
                    for idindex in self.chunks[chunk][block]:
                        position = idindex[0][0], idindex[0][1]*-1, idindex[0][2]
                        if idindex[1] > 0 and idindex[1] < 256 and idindex[1] not in self.norenderblocks:
                            blockinfo = {position:{'meta':idindex[2], 'faces':[]}}
                            if idindex[1] in self.multymatblocks:
                                if idindex[1] in [182, 126, 44] and idindex[2] > 7:
                                    idindex = idindex[0], idindex[1], idindex[2] - 8
                                if idindex[1] in [125, 181, 43] and idindex[2] > 7:
                                    idindex = idindex[0], idindex[1], idindex[2] - 8
                                if (idindex[1], idindex[2]) not in self.materials:
                                    self.materials.update({(idindex[1], idindex[2]):blockinfo})
                                else:
                                    self.materials[(idindex[1], idindex[2])].update(blockinfo)
                            else:
                                if (idindex[1], 0) not in self.materials:
                                    self.materials.update({(idindex[1], 0):blockinfo})
                                else:
                                    self.materials[(idindex[1], 0)].update(blockinfo)
                self.chunks[chunk][block] = None
        self.chunks = None


    def gen_faces_neighbors(self, agressiveremoval=False):
        allmaterials = {}
        print 'find material neightbors for ' + str(len(self.materials)) + ' materials'
        # todo below is somewhat sloppy refactor material
        


        for mat in self.materials:
            if mat in self.materials and mat in self.colormats:
                for blockindex in self.materials[mat]:
                    self.materials[mat][blockindex]['interneighbor'] = self.colormats[mat]['interneighbor']
                    self.materials[mat][blockindex]['extraneighbor'] = self.colormats[mat]['extraneighbor']
                allmaterials.update(self.materials[mat])

        for mat in self.materials:
            if mat in self.materials and mat in self.colormats:
                self.neightbors[mat] = {}
        for mat in self.materials:
            if mat in self.materials and mat in self.colormats:
                for block in self.materials[mat]:
                    if agressiveremoval:
                        blockstocheck = allmaterials
                    else:
                        blockstocheck = self.materials[mat]
                    self.neightbors[mat][block] = {'meta': self.materials[mat][block]['meta'], 'faces':[]}
                    if (block[0]-1, block[1], block[2]) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(1)
                    elif blockstocheck[(block[0]-1, block[1], block[2])]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(1)
                    if (block[0]+1, block[1], block[2]) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(2)
                    elif blockstocheck[(block[0]+1, block[1], block[2])]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(2)
                    if (block[0], block[1]-1, block[2]) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(3)
                    elif blockstocheck[(block[0], block[1]-1, block[2])]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(3)
                    if (block[0], block[1]+1, block[2]) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(4)
                    elif blockstocheck[(block[0], block[1]+1, block[2])]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(4)
                    if (block[0], block[1], block[2]-1) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(5)
                    elif blockstocheck[(block[0], block[1], block[2]-1)]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(5)
                    if (block[0], block[1], block[2]+1) not in blockstocheck:
                        self.neightbors[mat][block]['faces'].append(6)
                    elif blockstocheck[(block[0], block[1], block[2]+1)]['extraneighbor']:
                        self.neightbors[mat][block]['faces'].append(6)
                self.materials[mat] = None


    def remove_super_cosy(self):
        loneneighbors = {}
        for mat in self.neightbors:
            if mat in self.colormats and 'interneighbor' in self.colormats[mat] and self.colormats[mat]['interneighbor']:
                removeneibors = False
            else:
                removeneibors = True
            loneneighbors[mat] = {}
            for block in self.neightbors[mat]:
                if len(self.neightbors[mat][block]['faces']) > 0 or not removeneibors:
                    loneneighbors[mat][block[0]+0.5, block[1]+0.5, block[2]+0.5] = self.neightbors[mat][block]
            self.neightbors[mat] = None
        return loneneighbors

    def world_from_sample(self, sample):
        overworldblocks = [1, 2, 3, 4, 6, 8, 9, 12, 13, 15, 16, 17, 18, 37, 38, 39, 40]
        netherblocks = [10, 11, 87, 88, 112, 113, 114, 115, 153]
        theendblocks = [121]
        overscore = 0
        endscore = 0
        netherscore = 0
        for block in sample:
            if block in overworldblocks:
                overscore += 1
            if block in netherblocks:
                netherscore += 1
            if block in theendblocks:
                endscore += 1
        if endscore > overscore and endscore > netherscore:
            return 3
        elif netherscore > endscore and netherscore > overscore:
            return 2
        else:
            return 1
Ejemplo n.º 18
0
Archivo: tests.py Proyecto: aaps/quarry
 def testBlockPosition(self):
     blockpos = (345, -100, -2005)
     buff = Buffer()
     buff.buff = buff.pack_blockposition(blockpos)
     self.assertTrue(blockpos == buff.unpack_blockposition())
Ejemplo n.º 19
0
Archivo: tests.py Proyecto: aaps/quarry
 def testVarInt(self):
     value = 5533
     buff = Buffer()
     buff.buff = buff.pack_varint(value)
     self.assertTrue(buff.unpack_varint() == value)
Ejemplo n.º 20
0
Archivo: tests.py Proyecto: aaps/quarry
 def testUuid(self):
     uuid = types.UUID.from_offline_player("KLonter")
     buff = Buffer()
     buff.buff = buff.pack_uuid(uuid)
     self.assertTrue(uuid == buff.unpack_uuid())
Ejemplo n.º 21
0
def test_pack_chat():
    assert Buffer.pack_chat("spam") == b'\x10{"text": "spam"}'
Ejemplo n.º 22
0
def test_pack_varint():
    for value, data in varint_vectors:
        assert Buffer.pack_varint(value, signed=True) == data
Ejemplo n.º 23
0
def test_unpack_json():
    buffer = Buffer()
    buffer.add(b'\x10{"spam": "eggs"}')
    assert buffer.unpack_json() == {"spam": "eggs"}
Ejemplo n.º 24
0
def test_add():
    buffer = Buffer()
    buffer.add(b"spam")
    assert len(buffer) == 4
    assert buffer.read() == b"spam"
Ejemplo n.º 25
0
def test_unpack_varint():
    buffer = Buffer()
    for value, data in varint_vectors:
        buffer.add(data)
        assert buffer.unpack_varint(signed=True) == value
        assert len(buffer) == 0
Ejemplo n.º 26
0
 def testBlockPosition(self):
     blockpos = (345, -100, -2005)
     buff = Buffer()
     buff.buff = buff.pack_blockposition(blockpos)
     self.assertTrue(blockpos == buff.unpack_blockposition())
Ejemplo n.º 27
0
Archivo: tests.py Proyecto: aaps/quarry
 def testChat(self):
     text = "Hi there !"
     buff = Buffer()
     buff.buff = buff.pack_chat(text)
     self.assertTrue(text == buff.unpack_chat())
Ejemplo n.º 28
0
def test_unpack_string():
    buffer = Buffer()
    buffer.add(b"\x04spam")
    assert buffer.unpack_string() == "spam"
Ejemplo n.º 29
0
 def testChat(self):
     text = "Hi there !"
     buff = Buffer()
     buff.buff = buff.pack_chat(text)
     self.assertTrue(text == buff.unpack_chat())
Ejemplo n.º 30
0
def test_unpack_chat():
    buffer = Buffer()
    buffer.add(b'\x11["spam", " eggs"]')
    assert buffer.unpack_chat() == "spam eggs"
    buffer.add(b'\x22{"text": "spam", "extra": " eggs"}')
    assert buffer.unpack_chat() == "spam eggs"
    buffer.add(b'\x14{"translate": "foo"}')
    assert buffer.unpack_chat() == "foo"
    buffer.add(b'\x2E{"translate": "foo", "with": ["spam", "eggs"]}')
    assert buffer.unpack_chat() == "foo{spam, eggs}"
Ejemplo n.º 31
0
    def testString(self):
        text = "more tests, and stuff"

        buff = Buffer()
        buff.buff = buff.pack_string(text)
        self.assertTrue(text == buff.unpack_string())
Ejemplo n.º 32
0
def test_unpack_uuid():
    buffer = Buffer()
    buffer.add(uuid_vector)
    assert buffer.unpack_uuid().to_bytes() == uuid_vector
Ejemplo n.º 33
0
 def testUuid(self):
     uuid = types.UUID.from_offline_player("KLonter")
     buff = Buffer()
     buff.buff = buff.pack_uuid(uuid)
     self.assertTrue(uuid == buff.unpack_uuid())