PlanetFleet/factory.cpp

#include "factory.h"
#include "player.h"
#include "gameObjectFactory.h"
#include "activeGameState.h"

#include <stateManager.h>
#include <solUtil.h>

#include <vector>

namespace battleship{
	using namespace vb01;
	using namespace std;
	using namespace gameBase;

	Factory::Factory(Player *player, int id, Vector3 pos, Quaternion rot, int buildStatus, Unit::State state) : Structure(player, id, pos, rot, buildStatus, state){}

	void Factory::update(){
		Structure::update();

		if(!unitQueue.empty())
			train();
	}

	int Factory::getNumQueueUnitsById(int unitId){
		int numUnits = 0;

		for(int qu : unitQueue)
			if(qu == unitId)
				numUnits++;

		return numUnits;
	}

	void Factory::train(){
		bool training = !unitQueue.empty();
		buildStatusForeground->setVisible(training);
		buildStatusBackground->setVisible(training);

		if(training){
			ActiveGameState *activeState = (ActiveGameState*)GameManager::getSingleton()->getStateManager()->getAppStateByType(AppStateType::ACTIVE_STATE);
			Player *mainPlayer = (activeState ? activeState->getPlayer() : nullptr);

			vector<Player*> selectingPlayers = getSelectingPlayers();
			bool mainPlayerSelecting = (activeState && find(selectingPlayers.begin(), selectingPlayers.end(), mainPlayer) != selectingPlayers.end());

			Unit::displayUnitStats(buildStatusForeground, buildStatusBackground, trainingStatus, 100, mainPlayer == player && mainPlayerSelecting, Vector2(0, -10));
			sol::table targTable = generateView()["units"][unitQueue[0] + 1];
			int costRate = (int)targTable["cost"] / 100, trainRate = (int)targTable["buildTime"] / 100;

			if(player->getRefineds() >= costRate && getTime() - lastTrainTime > trainRate){
				trainingStatus++;
				player->setRefineds(player->getRefineds() - costRate);
				lastTrainTime = getTime();
			}

			if(trainingStatus >= 100){
				Unit *unit = GameObjectFactory::createUnit(player, unitQueue[0], pos, rot);
				player->addUnit(unit);
				unit->setOrder(Order(Order::TYPE::MOVE, vector<Order::Target>{Order::Target(nullptr, pos + 30 * dirVec)}, Vector3::VEC_ZERO));

				unitQueue.erase(unitQueue.begin());
				trainingStatus = 0;

				player->incVehiclesBuilt();
			}
		}
	}
}