コード例 #1
0
ファイル: table.py プロジェクト: CallumJHays/AirHockeyRobot
    def __init__(self):
        self.surface = pygame.Surface(constants.TABLE_SIZE)

        self.topGoal = Goal()
        self.topGoal.setColor((255, 100, 100))
        self.bottomGoal = Goal()
        self.bottomGoal.setColor((100, 100, 255))

        self.puck = Puck()

        self.arm = Arm()

        self.joy = pygame.joystick.Joystick(0)
        self.joy.init()
コード例 #2
0
ファイル: table.py プロジェクト: CallumJHays/AirHockeyRobot
class Table(SimObject):
    def __init__(self):
        self.surface = pygame.Surface(constants.TABLE_SIZE)

        self.topGoal = Goal()
        self.topGoal.setColor((255, 100, 100))
        self.bottomGoal = Goal()
        self.bottomGoal.setColor((100, 100, 255))

        self.puck = Puck()

        self.arm = Arm()

        self.joy = pygame.joystick.Joystick(0)
        self.joy.init()

    def tick(self):
        self.arm.tick()

        self.topGoal.tick()
        self.bottomGoal.tick()

    def draw(self):
        self.surface.fill((255, 255, 255))
        self.drawQuadrants()

        topGoalSurf = self.topGoal.draw()
        topGoalPos = self.toPygame((-topGoalSurf.get_width()/2, self.surface.get_height()-topGoalSurf.get_height()))
        self.surface.blit(topGoalSurf, topGoalPos)

        bottomGoalSurf = self.bottomGoal.draw()
        bottomGoalPos = self.toPygame((-bottomGoalSurf.get_width()/2, 0))
        self.surface.blit(bottomGoalSurf, bottomGoalPos)


        x = (self.joy.get_axis(4))*constants.TABLE_SIZE[0]
        y = (300+350)*(-self.joy.get_axis(1))
        self.arm.mousePos = (x, y)#self.puck.calculateTrajectory(200), 200#= self.getMousePos()
        #print(x, y)


        armSurf = self.arm.draw()
        armPos = self.toPygame((-armSurf.get_width()/2, 0))
        self.surface.blit(armSurf, armPos)

        mPos = self.getMousePos()
        diff = self.puck.pos.getDifference(Point(mPos[0], mPos[1]))
        self.puck.vel = diff
        self.puck.draw(self.surface)

        return self.surface

    def drawQuadrants(self):
        pygame.draw.line(self.surface, (255, 0, 0), self.toPygame((-constants.TABLE_SIZE[0]/2, constants.TABLE_SIZE[1]/2)), self.toPygame((constants.TABLE_SIZE[0]/2, constants.TABLE_SIZE[1]/2)))
        pygame.draw.line(self.surface, (255, 0, 0), self.toPygame((0, 0)), self.toPygame((0, constants.TABLE_SIZE[1])))

    def toPygame(self, pos):
        return pos[0]+constants.TABLE_SIZE[0]/2, pos[1]

    def getMousePos(self):
        pos = pygame.mouse.get_pos()
        pos = (pos[0]-constants.BORDER_SIZE*constants.SCALE), (pos[1]-constants.BORDER_SIZE*constants.SCALE)
        pos = pos[0]/constants.SCALE, pos[1]/constants.SCALE
        pos = int(pos[0]-constants.TABLE_SIZE[0]/2), int(constants.TABLE_SIZE[1]-pos[1])
        return pos