StateTools.java

/*******************************************************************************
 * Copyright 2015 Maximilian Stark | Dakror <mail@dakror.de>
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/


package de.dakror.vloxlands.ai.state;

import com.badlogic.gdx.math.Vector3;
import com.badlogic.gdx.utils.Timer;
import com.badlogic.gdx.utils.Timer.Task;

import de.dakror.vloxlands.ai.job.EnterStructureJob;
import de.dakror.vloxlands.ai.job.PickupJob;
import de.dakror.vloxlands.ai.path.AStar;
import de.dakror.vloxlands.ai.path.Path;
import de.dakror.vloxlands.game.Game;
import de.dakror.vloxlands.game.entity.creature.Human;
import de.dakror.vloxlands.game.entity.structure.NodeType;
import de.dakror.vloxlands.game.entity.structure.Structure;
import de.dakror.vloxlands.game.entity.structure.Warehouse;
import de.dakror.vloxlands.game.item.ItemStack;
import de.dakror.vloxlands.game.query.PathBundle;
import de.dakror.vloxlands.game.query.Query;
import de.dakror.vloxlands.util.event.Callback;

/**
 * @author Dakror
 */
public class StateTools {
	public static boolean equipTool(Human human, Class<?> tool, boolean queue, Vector3 pathStart) {
		if (tool == null && human.getTool().isNull()) return false;
		
		if (tool == null && !human.getTool().isNull()) {
			PathBundle pb = Game.world.query(new Query(human).searchClass(Warehouse.class).structure(true).capacityForTransported(true).transport(human.getTool()).node(NodeType.deposit));
			if (pb != null) {
				PickupJob pj = new PickupJob(human, pb.structure, new ItemStack(), true, false);
				if (!queue) human.setJob(pb.path, pj);
				else human.queueJob(pb.path, pj);
				
				if (pb.path.getLast() != null) pathStart.set(pb.path.getLast());
				
				return true;
			}
		} else {
			if (human.getTool().isNull() || !(human.getTool().getItem().getClass().isAssignableFrom(tool))) {
				PathBundle pb = Game.world.query(new Query(human).searchClass(Warehouse.class).structure(true).tool(tool).node(NodeType.pickup));
				if (pb != null) {
					PickupJob pj = new PickupJob(human, pb.structure, new ItemStack(pb.structure.getInventory().getAnyItemForToolType(tool), 1), true, false);
					if (!queue) human.setJob(pb.path, pj);
					else human.queueJob(pb.path, pj);
					
					if (pb.path.getLast() != null) pathStart.set(pb.path.getLast());
					
					return true;
				}
			}
		}
		
		return false;
	}
	
	public static void initForWorkplace(final Human human) {
		Vector3 pathStart = human.getVoxelBelow();
		boolean queue = StateTools.equipTool(human, human.getWorkPlace().getWorkerTool(), false, pathStart);
		
		Structure workPlace = human.getWorkPlace();
		EnterStructureJob esj = new EnterStructureJob(human, workPlace, false);
		esj.setEndEvent(new Callback() {
			@Override
			public void trigger() {
				Timer.schedule(new Task() {
					@Override
					public void run() {
						human.changeState(human.getWorkPlace().getWorkerState());
					}
				}, 1);
			}
		});
		if (queue) {
			Path p = AStar.findPath(pathStart, workPlace.getStructureNode(pathStart, NodeType.entry).pos.cpy().add(workPlace.getVoxelPos()), human, NodeType.entry.useGhostTarget);
			human.queueJob(p, esj);
		} else human.setJob(null, esj);
	}
	
	public static Path getHomePath(Human human, Vector3 start, NodeType nodeType) {
		return AStar.findPath(start, human.getWorkPlace().getStructureNode(human.getVoxelBelow(), nodeType).pos.cpy().add(human.getWorkPlace().getVoxelPos()), human, nodeType.useGhostTarget);
	}
	
	public static Path getHomePath(Human human, NodeType nodeType) {
		return getHomePath(human, human.getVoxelBelow(), nodeType);
	}
	
	public static boolean isWorkingTime() {
		return Game.time > 0;
	}
}