mcmodels.py

"""
SpaceCRAFT - Astro Pi competition[http://astro-pi.org/] entry
Conceived by Hannah Belshaw
Created by Martin O'Hanlon[http://www.stuffaboutcode.com]
For the Raspberry Pi Foundation[https://www.raspberrypi.org]

mcmodels.py

A collections of models which can be used in minecraft

"""
from minecraftstuff import MinecraftShape
from minecraftstuff import ShapeBlock
from mcpi.minecraft import Minecraft
from mcpi.minecraft import Vec3
from mcpi import block
from time import sleep
try:
    import thread
except ImportError:
    import _thread as thread

class ISS(MinecraftShape):
    """
    A movable minecraft model of the ISS
    """
    def __init__(self, mc, pos, cleararea = True, makevisible = True):

        self.pos = pos
        self.mc = mc

        #init the MinecraftShape
        MinecraftShape.__init__(self, self.mc, self.pos, visible = False)

        #create the ISS using setBlock(s) commands
        
        #main body
        self.setBlocks(-2, 0, -16, -2, 0, 16, 42)
        self.setBlocks(-1, -1, 16, -1, -1, -16, 42)
        self.setBlocks(0, -2, -16, 0, -2, 16, 42)
        self.setBlocks(1, -1, 16, 1, -1, -16, 42)
        self.setBlocks(2, 0, -16, 2, 0, 16, 42)
        self.setBlocks(1, 1, 16, 1, 1, -16, 42)
        self.setBlocks(0, 2, -16, 0, 2, 16, 42)
        self.setBlocks(-1, 1, 16, -1, 1, -16, 42)
        self.setBlocks(1, 0, -17, -1, 0, -17, 42)
        self.setBlocks(0, 1, -17, 0, -1, -17, 42)
        self.setBlocks(0, 1, 17, 0, -1, 17, 42)
        self.setBlocks(-1, 0, 17, 1, 0, 17, 42)
        #across 
        self.setBlocks(-12, -1, 0, 7, -1, 0, 42)
        self.setBlocks(-12, 0, 1, 8, 0, 1, 42)
        self.setBlocks(7, 1, 0, -12, 1, 0, 42)
        self.setBlocks(-12, 0, -1, 8, 0, -1, 42)
        self.setBlock(-13, 0, 0, 42)
        #bit on top of the across
        self.setBlock(-13, 0, 0, 42)
        self.setBlock(-10, 1, -1, 42)
        self.setBlock(-10, 2, 0, 42)
        self.setBlock(-10, 1, 1, 42)
        self.setBlock(-5, 1, 1, 42)
        self.setBlock(-5, 1, -1, 42)
        #central spire
        self.setBlocks(-1, 2, 0, -1, 4, 0, 42)
        self.setBlocks(0, 3, -1, 0, 4, -1, 42)
        self.setBlocks(1, 2, 0, 1, 4, 0, 42)
        self.setBlocks(0, 3, 1, 0, 4, 1, 42)
        self.setBlock(0, 5, 0, 42)
        #the T
        self.setBlocks(7, 1, -5, 7, 1, 4, 42)
        self.setBlocks(8, 0, 4, 8, 0, -5, 42)
        self.setBlocks(6, 0, -5, 6, 0, 4, 42)
        self.setBlocks(7, -1, 4, 7, -1, -5, 42)
        self.setBlock(7, 0, -6, 42)
        self.setBlock(7, 0, 5, 42)
        #solar panel supports
        self.setBlocks(14, 0, 14, 3, 0, 14, 42)
        self.setBlocks(3, 0, 9, 14, 0, 9, 42)
        self.setBlocks(14, 0, -10, 3, 0, -10, 42)
        self.setBlocks(3, 0, -14, 14, 0, -14, 42)
        self.setBlocks(-3, 0, -14, -15, 0, -14, 42)
        self.setBlocks(-15, 0, -10, -3, 0, -10, 42)
        self.setBlocks(-3, 0, 9, -14, 0, 9, 42)
        self.setBlocks(-14, 0, 14, -3, 0, 14, 42)
        #solar panels
        self.setBlocks(-4, 0, 15, -15, 0, 15, 35, 1)
        self.setBlocks(-15, 0, 13, -4, 0, 13, 35, 1)
        self.setBlocks(-4, 0, 10, -15, 0, 10, 35, 1)
        self.setBlocks(-15, 0, 8, -4, 0, 8, 35, 1)
        self.setBlocks(-4, 0, -9, -16, 0, -9, 35, 1)
        self.setBlocks(-16, 0, -11, -4, 0, -11, 35, 1)
        self.setBlocks(-4, 0, -13, -16, 0, -13, 35, 1)
        self.setBlocks(-16, 0, -15, -4, 0, -15, 35, 1)
        self.setBlocks(4, 0, -15, 15, 0, -15, 35, 1)
        self.setBlocks(15, 0, -13, 4, 0, -13, 35, 1)
        self.setBlocks(4, 0, -11, 15, 0, -11, 35, 1)
        self.setBlocks(15, 0, -9, 4, 0, -9, 35, 1)
        self.setBlocks(4, 0, 8, 15, 0, 8, 35, 1)
        self.setBlocks(15, 0, 10, 4, 0, 10, 35, 1)
        self.setBlocks(4, 0, 13, 15, 0, 13, 35, 1)
        self.setBlocks(15, 0, 15, 4, 0, 15, 35, 1)

        if cleararea:
            self.clearArea()

        #make the model visible
        if makevisible:
            self.draw()

    def clearArea(self):
        """
        Clears an area big enough to put the model in
        """
        self.mc.setBlocks(
            self.pos.x - 17,
            self.pos.y - 3,
            self.pos.z - 17,
            self.pos.x + 17,
            self.pos.y + 3,
            self.pos.z + 17,
            block.AIR.id)
            
class Arrow(MinecraftShape):
    """
    A movable arrow minecraft model
    """
    def __init__(self, mc, pos, cleararea = True, makevisible = True):

        self.pos = pos
        self.mc = mc

        #init the MinecraftShape
        MinecraftShape.__init__(self, self.mc, self.pos, visible = False)

        #create the arrow using setBlock(s) commands
        # an arrow
        self.setBlocks(0, 0, -5, 0, 0, 5, block.WOOL.id, 1)
        self.setBlocks(1, 0, -4, 1, 0, -4, block.WOOL.id, 2)
        self.setBlocks(-1, 0, -4, -1, 0, -4, block.WOOL.id, 2)
        self.setBlocks(2, 0, -3, 2, 0, -3, block.WOOL.id, 3)
        self.setBlocks(-2, 0, -3, -2, 0, -3, block.WOOL.id, 3)
        self.setBlocks(3, 0, -2, 3, 0, -2, block.WOOL.id, 4)
        self.setBlocks(-3, 0, -2, -3, 0, -2, block.WOOL.id, 4)
        
        if cleararea:
            self.clearArea()

        #make the model visible
        if makevisible:
            self.draw()

    def clearArea(self):
        """
        Clears an area big enough to put the model in
        """
        self.mc.setBlocks(
            self.pos.x - 3,
            self.pos.y,
            self.pos.z - 5,
            self.pos.x + 3,
            self.pos.y,
            self.pos.z + 5,
            block.AIR.id)

class MCAstroPi(MinecraftShape):
    """
    A movable model of the astro pi computer
    """
    def __init__(self, mc, pos, cleararea = True, makevisible = True):

        self.pos = pos
        self.mc = mc

        #init the MinecraftShape
        MinecraftShape.__init__(self, self.mc, self.pos, visible = False)

        #create the AstroPi using setBlock(s) commands
        
        #boards
        self.setBlocks(-6, -3, -9, 7, -3, 11, 35, 13, tag = "rpi_board")
        self.setBlocks(-6, 0, -9, 7, 0, 6, 35, 13, tag = "astropi_board")
        #pillars
        self.setBlocks(-6, -2, -9, -6, -1, -9, 42)
        self.setBlocks(7, -2, -9, 7, -1, -9, 42)
        self.setBlocks(-6, -2, 6, -6, -1, 6, 42)
        self.setBlocks(7, -2, 6, 7, -1, 6, 42)
        #gpio headers
        self.setBlocks(7, 1, -8, 7, 1, 5, 35, 15, tag = "astropi_gpio")
        self.setBlocks(7, -2, -8, 7, -1, 5, 35, 15, tag = "rpi_gpio")
        #usb and ethernet port
        self.setBlocks(4, -2, 8, 6, 0, 11, 42, tag = "usb")
        self.setBlocks(0, -2, 8, 2, 0, 11, 42, tag = "usb" )
        self.setBlocks(-5, -2, 8, -2, 0, 11, 42, tag = "ethernet")
        #camera, display, power, hdmi, composite ports
        self.setBlocks(-5, -2, 1, -2, -2, 1, 35, 15, tag = "camera")
        self.setBlocks(2, -2, -9, -1, -2, -9, 35, 15, tag = "display")
        self.setBlocks(-6, -2, -7, -6, -2, -6, 42, tag = "power")
        self.setBlocks(-6, -2, -2, -6, -2, 0, 42, tag = "hdmi")
        self.setBlock(-6, -2, 3, 35, 15, tag = "composite")
        #processor
        self.setBlocks(0, -2, -2, 2, -2, -4, 35, 15, tag = "processor")
        #led grid
        self.setBlocks(-3, 1, -8, 4, 1, -1, 35, 0, tag = "led")
        #other astro pi components
        self.setBlocks(3, 1, 1, 4, 1, 2, 35, 15, tag = "level_shifter")
        self.setBlocks(3, 1, 4, 4, 1, 5, 35, 15, tag = "atmel" )
        self.setBlocks(0, 1, 1, 0, 1, 2, 35, 15, tag = "orientation")
        self.setBlock(1, 1, 5, 35, 15, tag = "humidity")
        self.setBlock(-1, 1, 5, 35, 15, tag = "pressure")
        self.setBlock(-2, 1, 3, 35, 15, tag = "eeprom")
        self.setBlocks(-6, 1, -5, -5, 1, -4, 35, 15, tag = "led_driver")
        #astropi joystick
        self.setBlock(-5, 1, 4, 42, tag = "joy_left")
        self.setBlock(-4, 1, 5, 42, tag = "joy_up")
        self.setBlock(-5, 1, 6, 42, tag = "joy_right")
        self.setBlock(-6, 1, 5, 42, tag = "joy_down")
        self.setBlock(-5, 2, 5, 35, 15, tag = "joy_button")
        #astro pi gaps
        self.setBlocks(-1, 0, -9, 2, 0, -9, 0)
        self.setBlocks(-5, 0, 1, -2, 0, 1, 0)
        
        if cleararea:
            self.clearArea()

        #make the model visible
        if makevisible:
            self.draw()

    def clearArea(self):
        """
        Clears an area big enough to put the model in
        """
        self.mc.setBlocks(
            self.pos.x - 6,
            self.pos.y - 3,
            self.pos.z - 9,
            self.pos.x + 7,
            self.pos.y + 2,
            self.pos.z + 11,
            block.AIR.id)
            
class Rocket(MinecraftShape):
    """
    A movable model of a rocket in minecraft
    """
    def __init__(self, mc, pos, cleararea = True, makevisible = True):

        self.pos = pos
        self.mc = mc

        #init the MinecraftShape
        MinecraftShape.__init__(self, self.mc, self.pos, visible = False)

        #create the arrow using setBlock(s) commands
        #red wool fire!
        self.setBlock(0, 0, 0, 35, 14)
        #block wool wings
        self.setBlocks(-2, 1, 0, 2, 1, 0, 35, 15)
        self.setBlocks(0, 1, -2, 0, 1, 2, 35, 15)
        self.setBlocks(0, 2, -1, 0, 2, 1, 35, 15)
        self.setBlocks(-1, 2, 0, 1, 2, 0, 35, 15)
        #white steps on wings
        self.setBlock(-3, 1, 0, 156)
        self.setBlock(-2, 2, 0, 156)
        self.setBlock(0, 1, -3, 156, 2)
        self.setBlock(0, 2, -2, 156, 2)
        self.setBlock(3, 1, 0, 156, 1)
        self.setBlock(2, 2, 0, 156, 1)
        self.setBlock(0, 1, 3, 156, 3)
        self.setBlock(0, 2, 2, 156, 3)
        #white base
        self.setBlocks(1, 3, 0, -1, 3, 0, 155)
        self.setBlocks(0, 3, 1, 0, 3, -1, 155)
        #black band
        self.setBlocks(1, 4, 0, -1, 4, 0, 35, 15)
        self.setBlocks(0, 4, 1, 0, 4, -1, 35, 15)
        #main body
        self.setBlocks(0, 5, -1, 0, 10, -1, 155)
        self.setBlocks(-1, 10, 0, -1, 5, 0, 155)
        self.setBlocks(0, 5, 1, 0, 10, 1, 155)
        self.setBlocks(1, 10, 0, 1, 5, 0, 155)
        #white steps at the top
        self.setBlock(1, 11, 0, 156, 1)
        self.setBlock(0, 11, 1, 156, 3)
        self.setBlock(-1, 11, 0, 156)
        self.setBlock(0, 11, -1, 156, 2)
        #block on the top
        self.setBlock(0, 12, 0, 155)

        if cleararea:
            self.clearArea()

        #make the model visible
        if makevisible:
            self.draw()

    def clearArea(self):
        """
        Clears an area big enough to put the model in
        """
        self.mc.setBlocks(
            self.pos.x - 3,
            self.pos.y,
            self.pos.z - 3,
            self.pos.x + 3,
            self.pos.y + 12,
            self.pos.z + 3,
            block.AIR.id)

    def launch(self, height):
        for up in range(0, height):
            self.moveBy(0, 1, 0)

class LaunchPad(MinecraftShape):
    """
    A model of a launch pad which the rocket can sit on
    """
    def __init__(self, mc, pos, cleararea = True, makevisible = True):

        self.pos = pos
        self.mc = mc

        #init the MinecraftShape
        MinecraftShape.__init__(self, self.mc, self.pos, visible = False)

        #base
        self.setBlocks(-4, 0, -3, 4, 0, 3, 1, tag = "launch")
        self.setBlocks(3, 0, -4, -3, 0, -4, 1, tag = "launch")
        self.setBlocks(-3, 0, 4, 3, 0, 4, 1, tag = "launch")

        #steel tower
        self.setBlocks(4, 1, 2, 4, 10, 2, 42)
        self.setBlocks(4, 10, 2, 4, 10, -2, 42)
        self.setBlocks(4, 1, -2, 4, 10, -2, 42)

        #glass lift and tunnel
        self.setBlocks(4, 9, 0, 4, 1, 0, 20)
        self.setBlocks(3, 10, 0, 2, 10, 0, 20)

        if cleararea:
            self.clearArea()

        #make the model visible
        if makevisible:
            self.draw()

    def clearArea(self):
        """
        Clears an area big enough to put the model in
        """
        self.mc.setBlocks(
            self.pos.x - 4,
            self.pos.y,
            self.pos.z - 4,
            self.pos.x + 4,
            self.pos.y + 10,
            self.pos.z + 4,
            block.AIR.id)

class Stairs():
    def __init__(self, mc, pos, width, height, blockId, blockData = 0):
        self.mc = mc
        self.pos = pos
        self.width = width
        self.height = height
        self.blockId = blockId
        self.blockData = blockData

        self.draw()

    def _drawStairs(self, blockId, blockData = 0):
        x = self.pos.x
        y = self.pos.y
        z = self.pos.z

        xIncrement = -1
        zIncrement = 1

        currentWidth = 0

        #keep building steps until we have reached the height
        while y < self.pos.y + self.height:
            self.mc.setBlock(x, y, z, blockId, blockData)
            #if we have reached the width change step direction
            currentWidth += 1
            if currentWidth == self.width:
                
                currentWidth = 1
                if xIncrement == -1 and zIncrement == 1:
                    xIncrement, zIncrement = 1, 1
                    
                elif xIncrement == 1 and zIncrement == 1:
                    xIncrement, zIncrement = 1, -1
                    
                elif xIncrement == 1 and zIncrement == -1:
                    xIncrement, zIncrement = -1, -1
                    
                elif xIncrement == -1 and zIncrement == -1:
                    xIncrement, zIncrement = -1, 1
            #move the step on
            x += xIncrement
            y += 1
            z += zIncrement

    def draw(self):
        self._drawStairs(self.blockId, self.blockData)
        
    def clear(self):
        self._drawStairs(block.AIR.id)

#test
if __name__ == "__main__":

    mc = Minecraft.create()

    #isspos = Vec3(10, 70, 0)
    #mcastropipos = Vec3(0, 60, 0)
    #arrowpos = Vec3(0, 65, 0)
    #stairspos = Vec3(0,0,0)
    rocketpos = Vec3(10,60,0)

    #iss = ISS(mc, isspos)
    launchpad = LaunchPad(mc, rocketpos)
    #rocket= Rocket(mc, rocketpos)
    #mcastropi = MCAstroPi(mc, mcastropipos)
    #arrow = Arrow(mc, arrowpos)
    #stairs = Stairs(mc, stairspos, 5, 65, block.IRON_BLOCK.id)
    #stairs.draw()
    
    try:
        #sleep(1)
        #rocket.launch(75)
        #rocket.reset()
        sleep(5)

    finally:
        #iss.clear()
        #rocket.clear()
        launchpad.clear()
        #mcastropi.clear()
        #arrow.clear()
        #stairs.clear()