engine.py

__author__ = 'Nathan'

import pygame
import graphics
import random
import map
from action import Action
from tribute import Tribute
from goal import Goal
from mapReader import readMap
from random import randint
from pygame.locals import *
import json

class GameEngine(object):
    """
    here we will essentially manage everything and probably handle the controls
    """
    constants = range(1)
    is_looping = False
    tributes = []
    PAUSED = False
    curTrib = None
    tributes_by_district = []
    map_dims = []
    FIGHT_MESSAGES = False
    @staticmethod
    def start():
        me = GameEngine
        d = json.load(open('./distributions/action_values.json'))
        #create actions (will be overwritten in a minute)
        move_up = Action([], '', 1, 0, (0, -1), 'move_up')
        move_down = Action([], '', 1, 1, (0, 1), 'move_down')
        move_right = Action([], '', 1, 2, (1, 0), 'move_right')
        move_left = Action([], '', 1, 3, (-1, 0), 'move_left')
        hunt = Action([d['hunt']], ["hunger"], 1, 4,(0, 0), 'hunt')
        fight = Action([d['fight']], ["kill"], 1, 5, (0, 0), 'fight')
        scavenge = Action([d['scavenge']], ["getweapon"], 1, 6, (0, 0), 'scavenge')
        craft = Action([d['craft']], ["getweapon"], 1, 7, (0, 0), 'craft')
        hide = Action([d['hide']], ["hide"], 1, 8, (0, 0), 'hide')
        getwater = Action([d['get_water']], ["thirst"], 1, 9, (0, 0), 'get_water')
        rest = Action([d['rest']], ["rest"], 1, 10, (0, 0), 'rest')
        talkAlly = Action([d['talk_ally']], ["ally"], 1, 11, (0, 0), 'talk_ally')
        explore = Action(d['explore'], ['hunger', 'thirst', 'kill'], 1, 12, (0, 0), 'explore')
        me.FIGHT_MESSAGES = d['fight_messages']
        #create actions (will be overwritten in a minute)
        hunger = Goal("hunger", 2)
        thirst = Goal("thirst", 2)
        goalRest = Goal("rest", 0)
        kill = Goal("kill", 0)
        goalHide = Goal("hide", 0)
        getweapon = Goal("getweapon", 0)
        ally = Goal("ally", 0)

        goals = [hunger, thirst, goalRest, kill, goalHide, getweapon, ally]
        actions = [move_up, move_down, move_right, move_left, hunt, fight, scavenge, craft, hide, getwater, rest, talkAlly, explore]

        #create the goals here
        #not really needed right now

        init_locations = [(x, y) for x in range(15,30) for y in range(15,35)]
        districts = ['d' + str(x) for x in range(1, 4)]

        for d in districts:
            location = random.choice(init_locations)
            init_locations.remove(location)
            t1 = Tribute(goals, actions, *location, district=d, gender='male')
            me.tributes.append(t1)

            location = random.choice(init_locations)
            init_locations.remove(location)
            t2 = Tribute(goals, actions, *location, district=d, gender='female')
            me.tributes.append(t2)
            me.tributes_by_district.append((d, t1, t2))
        for i in range(len(me.tributes)):
            initTribute = me.create_goals(me.tributes[i])
            me.tributes[i].goals = initTribute
            #me.create_actions(me.tributes[i])

        mapToBeUsed = 'maps/allTerr.jpg'
        me.dims = (110, 70)
        me.map_dims = (50, 50)
        me.gameMap = readMap(mapToBeUsed)
        me.view = graphics.GameView(*me.dims)
        me.map = map.Map(mapToBeUsed)
        me.state = me.map.seed_game_state(me.tributes)  # game.GameState()
        me.is_looping = True
        me.curTrib = me.tributes[0]
        while me.is_looping and GameEngine.loop():
            pass

    @staticmethod
    def stop():
        GameEngine.is_looping = False

    @staticmethod
    def loop():
        """
        What to do on each loop iteration
        @return: None
        """
        me = GameEngine
        buttons = me.view.getButtons()
        names = me.view.getNames()
        for event in pygame.event.get():
            if event.type == KEYDOWN:
                if event.key == K_q:
                    me.PAUSED = not me.PAUSED
                if event.key == K_f:
                    me.curTrib.goals[0].modify_value(-10)
                if event.key == K_d:
                    me.curTrib.goals[1].modify_value(-10)
                if event.key == K_w:
                    me.curTrib.getWeapon()
                    me.curTrib.goals[5].modify_value(-10)

            if event.type == pygame.QUIT:
                pygame.quit()
                return False
            if event.type == pygame.MOUSEBUTTONDOWN or event.type == pygame.MOUSEBUTTONUP:
                pos = pygame.mouse.get_pos()
                x = pos[0] / 10
                y = pos[1] / 10
                for tribute in me.tributes:
                    if (x,y) == tribute.state:
                        me.curTrib = tribute
                    else:
                        namePos = 0
                        for button in buttons:
                            if button.collidepoint(pos):
                                for tribute in me.tributes:
                                    if tribute.first_name == names[namePos]:
                                        me.curTrib = tribute
                            namePos += 1

        if not me.PAUSED:
            #for tribute in me.tributes:
                #me.gameMap[tribute.old_state[0]][tribute.old_state[1]].tribute = None
            for x in range(50):
                for y in range(50):
                    me.gameMap[x][y].tribute = None

            for tribute in me.tributes:
                me.gameMap[tribute.state[0]][tribute.state[1]].tribute = tribute

            for tribute in me.tributes:
                tribute.act(me.gameMap, me.state)  # finds bestAction and does it.
                tribute.end_turn()
                death = tribute.checkDead()
                if death is not None:
                    print tribute.first_name, " ", tribute.last_name, " death by ", death
                    me.tributes.remove(tribute)
                    if me.curTrib == tribute and len(me.tributes) > 1:
                        me.curTrib = random.choice(me.tributes)
            if len(me.tributes) == 1:
                me.PAUSED = True;
            me.view.render(me.state, me.curTrib, me.tributes_by_district, me.tributes, 0)
            me.state.update()
        else:
            if len(me.tributes)==1:
                 me.view.render(me.state, me.curTrib, me.tributes_by_district, me.tributes, 1)
            else:
                me.view.render(me.state, me.curTrib, me.tributes_by_district, me.tributes, 0)
        return True

    @staticmethod
    def create_actions(tribute):
        #Move action
        move_up = Action([], '', 1, 0, (0, -1), 'move_up')
        move_down = Action([], '', 1, 1, (0, 1), 'move_down')
        move_right = Action([], '', 1, 2, (1, 0), 'move_right')
        move_left = Action([], '', 1, 3, (-1, 0), 'move_left')


        ################ INDEX LIST
        # 0-3 movement
        # 4 hunt
        # 5 fight
        # 6 scavenge
        # 7 craft
        # 8 hide
        # 9 water
        # 10 rest
        # 11 talk

        ############## ACTION ATTRIBUTES
        #1. How much I get back for doing the action in 2. EDIT THIS ONE
        #2. The action (lists, so it can affect more than one thing.)
        #3. Duration
        #4. Index
        #5. Movement Stuff (don't mess wid that)

        #as long as you hunt, you'll get at least 10 food points, maybe more depending on attributes & individual
        hungerStats = (tribute.attributes['endurance']/2)+tribute.attributes['hunting_skill']-(tribute.attributes['size']/5)
        foodEnergy = 10 + randint(0, hungerStats)
        foodRest = (tribute.attributes['stamina'] - 1)/100
        hunt = Action([foodEnergy,foodRest], ["hunger","rest"], 1, 4,(0,0),'hunt')

        #if you fight your "bloodlust" goes down. If you're friendly, killng someone drastically reduces your desire to kill
        #someone else. Unless you have a friendliness of 1
        killStats = (((1/tribute.attributes['bloodlust'])*3)+(tribute.attributes['friendliness']-1))
        if(int(tribute.district[1:]) == 1 or int(tribute.district[1:]) == 2 or int(tribute.district[1:])):
            killStats += 3
        else:
            killStats += 5
        fight = Action([killStats], ["kill"], 1, 5, (0,0),'fight')

        #If they find a weapon, they'll get back 15 "find weapon" points, which will generally cover
        #about 150 of wanting a weapon, and not having one. Same w/ crafting
        scavenge = Action([15], ["getweapon"], 1, 6, (0,0),'scavenge')
        craft = Action([15], ["getweapon"], 1, 7, (0,0),'craft')

        #If you're small and good at hiding, you get more points for it
        #You also recover some rest
        hideStats = ((3/(tribute.attributes['size']+tribute.attributes['strength']))+((tribute.attributes['camouflage_skill']-1)/4))
        hide = Action([hideStats, 3], ["hide", "rest"], 1, 8, (0,0),'hide')

        #As long as you drink, you'll get at least 10 drink points. Maybe more depending on attributes & individual
        thirstStats = (tribute.attributes['endurance'])-(tribute.attributes['size']/5)
        thirstEnergy = 10 + randint(0, thirstStats)
        getwater = Action([thirstEnergy], ["thirst"], 1, 9, (0,0),'get_water')

        #Resting will automatically recover 10 rest points. Maybe more depending on attributes & individual
        restStats = (tribute.attributes['stamina'] * 2)
        restEnergy = 10 + randint(0, restStats)
        rest = Action([restEnergy], ["rest"], 1, 10, (0,0), 'rest')

        #If you're friendly, talking to buddies recovers a bunch of talking-to-buddy points
        #And the smaller, weaker, and more terrible at fighting you are, the more recovery you get for making a buddy
        allyStats = (tribute.attributes['friendliness'] + (1/tribute.attributes['size']) + (1/tribute.attributes['strength']) + (0.5/tribute.attributes['fighting_skill']))
        talkAlly = Action([allyStats], ["ally"], 1, 11, (0,0),'talk_ally')

        explore = Action([3], ['multiple'], 1, 12, (0, 0), 'explore')

        newActions = [move_up, move_down, move_right, move_left, hunt,
                      fight, scavenge, craft, hide, getwater, rest, talkAlly, explore]

        tribute.actions = newActions

        return tribute

    @staticmethod
    def create_goals(tribute):
        #Base Hunger Thirst and Rest Values
        hunger = Goal("hunger", 2)
        thirst = Goal("thirst", 2)
        rest = Goal("rest", 5)

        #Start-of-Game goals based on attributes and individual
        district = int(tribute.district[1:])
        if district == 1 or district == 2 or district == 4:
            killBase = 5
        else:
            killBase = 1
        killLimit = killBase + ((tribute.attributes['bloodlust']-1)*2) + (11/tribute.attributes['friendliness']) + ((tribute.attributes['size']-1)/2) + ((tribute.attributes['fighting_skill']-1)/2) + ((tribute.attributes['strength']-1)/2)
        killStats = randint(killBase, killLimit) * 10

        if(tribute.attributes['size'] + tribute.attributes['strength']) < 4:
            hideBase = 10
        else:
            hideBase = 0
        hideLimit = hideBase + (10/tribute.attributes['size']) + (5/tribute.attributes['strength']) + (tribute.attributes['camouflage_skill'] * 2) - tribute.attributes['fighting_skill']
        hideStats = randint(hideBase, hideLimit)


        getWeaponStats = ((tribute.attributes['weapon_skill']-1)*2) - (tribute.attributes['crafting_skill']-1)
        allyStats = tribute.attributes['friendliness'] - tribute.attributes['bloodlust']


        kill = Goal("kill", killStats)
        hide = Goal("hide", hideStats)
        getweapon = Goal("getweapon", getWeaponStats)
        ally = Goal("ally", allyStats)
        fear = Goal("fear", 0)

        return [hunger, thirst, rest, kill, hide, getweapon, ally, fear]