Heatmaps/InfluenceComponent.cpp at main · DRossen/Heatmaps · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#include "stdafx.h"
#include "InfluenceComponent.h"
#include "InterestCurves.h"
#include <Engine/Source/ComponentSystem/GameObject.h>
#include <Engine/Source/AI/HeatmapSystem/HeatmapManager.h>
#include <Project/Source/Enemies/EnemyComponent.h>

namespace AI
{
	InfluenceComponent::InfluenceComponent(KE::GameObject& aGo) : KE::Component(aGo),
		myPosition(aGo.myTransform.GetPositionRef())
	{

	}

	InfluenceComponent::~InfluenceComponent()
	{

	}

	void InfluenceComponent::Awake()
	{

	}

	void InfluenceComponent::SetData(void* aDataObject)
	{
		InfluenceComponentData& data = *(InfluenceComponentData*)(aDataObject);

		myTeam = data.team;
		myImprints = data.influenceTypes;
		data.heatmapManager->Register(*this);
	}

	void InfluenceComponent::AssignVelocityPtr(const Vector3f* aVelocity)
	{
		myVelocity = aVelocity;
		isDynamic = true;
	}

	const InfluenceData* InfluenceComponent::GetTemplate(HeatType aType) const
	{
		for (auto& data : myImprints)
		{
			if(data.type != aType) continue;

			return &data;
		}
		return nullptr;
	}

	const int InfluenceComponent::GetID() const
	{
		return myGameObject.myID;
	}
}