update scripts
This commit is contained in:
@ -3,7 +3,6 @@ using UnityEngine;
|
||||
using Unity.MLAgents;
|
||||
using Unity.MLAgents.Sensors;
|
||||
using Unity.MLAgents.Actuators;
|
||||
using System.Collections.Generic;
|
||||
|
||||
[RequireComponent(typeof(MovementController))]
|
||||
public class NPC : Agent, ICharacter
|
||||
@ -11,6 +10,7 @@ public class NPC : Agent, ICharacter
|
||||
[HideInInspector]
|
||||
public Character AgentCharacter;
|
||||
public CharacterCondition Condition;
|
||||
private FlagZone flagZone;
|
||||
|
||||
public NPC_BaseState NPC_State { get; private set; }
|
||||
|
||||
@ -47,29 +47,38 @@ public class NPC : Agent, ICharacter
|
||||
public override void OnEpisodeBegin()
|
||||
{
|
||||
NPC_State = DirectState;
|
||||
flagZone = GameObject.FindObjectOfType<FlagZone>();
|
||||
}
|
||||
|
||||
public override void CollectObservations(VectorSensor sensor)
|
||||
{
|
||||
var candidates = moveController.getPointsCandidate();
|
||||
|
||||
sensor.AddObservation(Condition.HealthPoints);
|
||||
sensor.AddObservation(Condition.ArmourPoints);
|
||||
sensor.AddObservation(Condition.Ammunition);
|
||||
sensor.AddObservation((int)NPC_State.State);
|
||||
|
||||
var candidates = moveController.getPointsCandidate();
|
||||
sensor.AddObservation((!flagZone.isNotOccup).ToInt());
|
||||
sensor.AddObservation(AgentCharacter.LastTimeHit);
|
||||
sensor.AddObservation(Condition.GetHealthPointsInQuantile());
|
||||
sensor.AddObservation(candidates.Count);
|
||||
sensor.AddObservation(GameManager.IsEnemyNearby(gameObject.transform.position, AgentCharacter.Team));
|
||||
|
||||
foreach (var point in candidates)
|
||||
{
|
||||
Debug.Log((float)moveController.CurrentNavPoint.PointId);
|
||||
|
||||
bufferSensor.AppendObservation(new float[] {
|
||||
//1 position in navpointId
|
||||
(float)moveController.currentPosition.PointId,
|
||||
(float)moveController.CurrentNavPoint.PointId,
|
||||
//2 distance to flag
|
||||
moveController.currentPosition.FlagDistance,
|
||||
moveController.FlagDistance,
|
||||
//3 death count in point
|
||||
moveController.currentPosition.DeathAttr,
|
||||
moveController.CurrentNavPoint.DeathAttr,
|
||||
//4 flagEnemyDistance
|
||||
GameManager.IsCloserToFlagFromNextNavPoint(point, transform.position)==true?1:0,
|
||||
GameManager.IsCloserToFlagFromNextNavPoint(point, transform.position).ToInt(),
|
||||
//5 EnemyVsNavPointDistance
|
||||
GameManager.IsCloserToEnemyThanToNextNavPoint(point,transform.position, AgentCharacter.Team)==true?1:0
|
||||
GameManager.IsCloserToEnemyThanToNextNavPoint(point,transform.position, AgentCharacter.Team).ToInt()
|
||||
});
|
||||
|
||||
}
|
||||
@ -103,7 +112,7 @@ public class NPC : Agent, ICharacter
|
||||
if (Condition.HealthPoints < 0)
|
||||
{
|
||||
OnKilledEvent?.Invoke(this);
|
||||
moveController.currentPosition.DeathAttr += 1;
|
||||
moveController.CurrentNavPoint.DeathAttr += 1;
|
||||
}
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user