Renderer/src/main/java/electrosphere/entity/state/idle/ServerIdleTree.java

package electrosphere.entity.state.idle;

import electrosphere.entity.state.BehaviorTree;
import electrosphere.entity.state.attack.ServerAttackTree;
import electrosphere.entity.state.attack.ServerAttackTree.AttackTreeState;
import electrosphere.entity.state.idle.IdleTree.IdleTreeState;
import electrosphere.entity.state.movement.AirplaneMovementTree;
import electrosphere.entity.state.movement.ServerGroundMovementTree;
import electrosphere.engine.Globals;
import electrosphere.entity.Entity;
import electrosphere.entity.EntityDataStrings;
import electrosphere.entity.EntityUtils;
import electrosphere.entity.state.movement.ServerGroundMovementTree.MovementTreeState;
import electrosphere.entity.types.creature.CreatureUtils;
import electrosphere.net.parser.net.message.EntityMessage;
import electrosphere.net.parser.net.message.SynchronizationMessage;
import electrosphere.net.synchronization.BehaviorTreeIdEnums;
import electrosphere.net.synchronization.annotation.SyncedField;
import electrosphere.net.synchronization.annotation.SynchronizedBehaviorTree;
import electrosphere.renderer.actor.Actor;
import electrosphere.renderer.anim.Animation;
import electrosphere.server.datacell.utils.DataCellSearchUtils;
import electrosphere.server.datacell.utils.ServerBehaviorTreeUtils;
import electrosphere.server.poseactor.PoseActor;

import java.util.concurrent.CopyOnWriteArrayList;


@SynchronizedBehaviorTree(name = "serverIdle", isServer = true, correspondingTree = "idle")
/**
 * Tree for playing an idle animation when an entity isn't doing anything
 */
public class ServerIdleTree implements BehaviorTree {

    @SyncedField
    private IdleTreeState state;

    Entity parent;

    CopyOnWriteArrayList<EntityMessage> networkMessageQueue = new CopyOnWriteArrayList<EntityMessage>();

    public ServerIdleTree(Entity e){
        state = IdleTreeState.IDLE;
        parent = e;
    }

    /**
     * <p> Automatically generated </p>
     * <p>
     * Gets state.
     * </p>
     */
    public IdleTreeState getState(){
        return state;
    }

    public void start(){
        //TODO: check if can start moving
        state = IdleTreeState.IDLE;
    }

    public void interrupt(){
        state = IdleTreeState.NOT_IDLE;
    }

    public void stop(){
        state = IdleTreeState.NOT_IDLE;
    }

    public void simulate(float deltaTime){
        PoseActor entityActor = EntityUtils.getPoseActor(parent);

        boolean movementTreeIsIdle = movementTreeIsIdle();

        boolean hasAttackTree = parent.containsKey(EntityDataStrings.SERVER_ATTACK_TREE);
        ServerAttackTree attackTree = null;
        if(hasAttackTree){
            attackTree = CreatureUtils.serverGetAttackTree(parent);
        }

        boolean isIdle;

        //state machine
        switch(state){
            case IDLE:
                if(entityActor != null){
                    if(
                        (!entityActor.isPlayingAnimation() || !entityActor.isPlayingAnimation(Animation.ANIMATION_IDLE_1))
                    ){
                        entityActor.playAnimation(Animation.ANIMATION_IDLE_1,3);
                        entityActor.incrementAnimationTime(0.0001);
                    }
                }
                isIdle = true;
                if(!movementTreeIsIdle){
                    isIdle = false;
                }
                if(hasAttackTree){
                    if(attackTree.getState() != AttackTreeState.IDLE){
                        isIdle = false;
                    }
                }
                if(!isIdle){
                    this.setState(IdleTreeState.NOT_IDLE);
                }
                break;
            case NOT_IDLE:
                isIdle = true;
                if(!movementTreeIsIdle){
                    isIdle = false;
                }
                if(hasAttackTree){
                    if(attackTree.getState() != AttackTreeState.IDLE){
                        isIdle = false;
                    }
                }
                if(isIdle){
                    this.setState(IdleTreeState.IDLE);
                }
                break;
        }
    }

    boolean movementTreeIsIdle(){
        boolean rVal = false;
        boolean hasMovementTree = parent.containsKey(EntityDataStrings.SERVER_MOVEMENT_BT);
        if(hasMovementTree){
            BehaviorTree movementTree = CreatureUtils.serverGetEntityMovementTree(parent);
            if(movementTree instanceof ServerGroundMovementTree){
                if(((ServerGroundMovementTree)movementTree).getState() == MovementTreeState.IDLE){
                    rVal = true;
                }
            } else if(movementTree instanceof AirplaneMovementTree){
                rVal = false;
            }
        }
        return rVal;
    }

    public void addNetworkMessage(EntityMessage networkMessage) {
        networkMessageQueue.add(networkMessage);
    }

    /**
     * <p> (initially) Automatically generated </p>
     * <p> More parameters can be safely added to this method</p>
     * <p>
     * Attaches this tree to the entity.
     * </p>
     * @param entity The entity to attach to
     * @param tree The behavior tree to attach
     */
    public static ServerIdleTree attachTree(Entity parent){
        ServerIdleTree rVal = new ServerIdleTree(parent);
        //put manual code here (setting params, etc)
    
    
        //!!WARNING!! from here below should not be touched
        //This was generated automatically to properly alert various systems that the btree exists and should be tracked
        ServerBehaviorTreeUtils.attachBTreeToEntity(parent, rVal);
        parent.putData(EntityDataStrings.TREE_SERVERIDLE, rVal);
        Globals.entityValueTrackingService.attachTreeToEntity(parent, BehaviorTreeIdEnums.BTREE_SERVERIDLE_ID);
        return rVal;
    }

    /**
     * <p> Automatically generated </p>
     * <p>
     * Detatches this tree from the entity.
     * </p>
     * @param entity The entity to detach to
     * @param tree The behavior tree to detach
     */
    public static void detachTree(Entity entity, BehaviorTree tree){
        Globals.entityValueTrackingService.detatchTreeFromEntity(entity, BehaviorTreeIdEnums.BTREE_SERVERIDLE_ID);
    }

    /**
     * <p> Automatically generated </p>
     * <p>
     * Sets state and handles the synchronization logic for it.
     * </p>
     * @param state The value to set state to.
     */
    public void setState(IdleTreeState state){
        this.state = state;
        int value = IdleTree.getIdleTreeStateEnumAsShort(state);
        DataCellSearchUtils.getEntityDataCell(parent).broadcastNetworkMessage(SynchronizationMessage.constructUpdateClientStateMessage(parent.getId(), 3, 3, value));
    }
    
    /**
     * <p>
     * Gets the ServerIdleTree of the entity
     * </p>
     * @param entity the entity
     * @return The ServerIdleTree
     */
    public static ServerIdleTree getServerIdleTree(Entity entity){
        return (ServerIdleTree)entity.getData(EntityDataStrings.TREE_SERVERIDLE);
    }
}