105 lines
3.5 KiB
C#
105 lines
3.5 KiB
C#
#nullable enable
|
|
using System.Collections;
|
|
using System.Collections.Generic;
|
|
using UnityEngine;
|
|
using UnityEngine.Serialization;
|
|
|
|
public class AIEntity : Entity
|
|
{
|
|
[FormerlySerializedAs("stats")] [SerializeField] AIStats AIStats = null!;
|
|
public string? ennemyName {get; protected set; }
|
|
|
|
abstract class BaseStateAI : BaseState{
|
|
protected AIEntity entity;
|
|
public BaseStateAI(AIEntity entity){
|
|
this.entity = entity;
|
|
}
|
|
}
|
|
|
|
class SeekState : BaseStateAI{
|
|
public SeekState(AIEntity entity) : base(entity){
|
|
|
|
}
|
|
|
|
public override BaseState? UpdateState(){
|
|
Entity targetEntity = entity.GetTarget().GetComponent<Entity>();
|
|
if(targetEntity != null){
|
|
if(targetEntity.IsAlive()){//target is alive, keep chasing it
|
|
return null;
|
|
}else{//target is dead, go to findTargetState
|
|
return new FindTargetState(entity);;
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public override BaseState? FixedUpdateState(){
|
|
Transform target = entity.GetTarget();
|
|
entity.direction = Vector3.RotateTowards(entity.direction, (target.position - entity.transform.position), entity.rotSpeed*Time.fixedDeltaTime, 0.0f);
|
|
if(!entity.IsInAttackRange()){
|
|
entity.rb.MovePosition(entity.transform.position + entity.direction * entity.movementSpeed * Time.fixedDeltaTime);
|
|
}else{
|
|
return new AttackState(entity);
|
|
}
|
|
return null;
|
|
}
|
|
}
|
|
|
|
class FindTargetState : BaseStateAI{
|
|
float closeEnough;
|
|
public FindTargetState(AIEntity entity) : base(entity){
|
|
}
|
|
|
|
public override BaseState? UpdateState(){
|
|
Entity[] entities = FindObjectsOfType<Entity>();
|
|
float lastDist = -1;
|
|
Entity chosenEntity = null!;
|
|
foreach (Entity other in entities){// Find the closest entity
|
|
if(other.name == entity.ennemyName){
|
|
float distance = Vector2.Distance(other.transform.position, entity.transform.position);
|
|
if(distance < lastDist){
|
|
lastDist = distance;
|
|
chosenEntity = other;
|
|
if(lastDist <= entity.AIStats.closeEnough)break;
|
|
}
|
|
}
|
|
}
|
|
if(chosenEntity != null){
|
|
entity.SetTarget(chosenEntity.transform);
|
|
}
|
|
return new SeekState(entity);
|
|
}
|
|
|
|
}
|
|
|
|
class AttackState : BaseStateAI{
|
|
public AttackState(AIEntity entity) : base(entity){
|
|
}
|
|
|
|
public override BaseState? UpdateState(){
|
|
if(entity.IsInAttackRange()){
|
|
if(entity.attackTimer >= entity.attackCooldown){
|
|
entity.attackTimer = 0;
|
|
return Attack();
|
|
}else{
|
|
entity.attackTimer += Time.deltaTime;
|
|
}
|
|
|
|
}
|
|
return null;
|
|
}
|
|
|
|
private BaseState? Attack(){
|
|
Entity targetEntity = entity.GetTarget().GetComponent<Entity>();
|
|
if(targetEntity != null){
|
|
targetEntity.TakeDamage(entity.attackDmg);
|
|
bool isTargetAlive = targetEntity.IsAlive();
|
|
if(!isTargetAlive){
|
|
return new FindTargetState(entity);
|
|
}
|
|
}
|
|
return null;
|
|
}
|
|
}
|
|
}
|