Aimbot-ParallelEnv/Assets/Script/GameScript/RaySensors.cs

183 lines
7.2 KiB
C#
Raw Normal View History

2022-10-25 19:07:39 +00:00
using System;
using System.Collections.Generic;
using UnityEngine;
/*该scrip用于创建复数条ray于视角内并探测被ray射到的物体*/
public class RaySensors : MonoBehaviour
{
[SerializeField] private Camera agentCam;
[SerializeField] private Camera TPSCam;
[SerializeField] private Material lineMeterial;
[SerializeField] private GameObject rayInfoPrefab;
[SerializeField] private GameObject agentCanvas;
2022-10-25 19:07:39 +00:00
[SerializeField, Range(0, 500)] public float viewDistance = 100; // how long the ray can detect
2022-10-25 19:07:39 +00:00
//[SerializeField, Range(0, 1)] public float totalRange = 1f; // Total view range Max = 1
[SerializeField, Range(0, 1)] public float focusRange = 0.15f; // center focus range
public int halfOuterRayNum = 7; // >=2
2022-10-25 19:07:39 +00:00
public int focusRayNum = 5; // >= 1 and must be odd num!
[Header("InGameLineSetting")]
public bool showInGameRay = true;
2022-10-25 19:07:39 +00:00
public bool showDebugRay = true;
public bool showInGameRayInfo = true;
public float lineWidth = 0.05f;
[Header("RayCastResult")]
public float[] rayTagResult;
public float[] rayTagResultOneHot;
2022-10-25 19:07:39 +00:00
public float[] rayDisResult;
[System.NonSerialized] public int totalRayNum;
private string myTag = "";
private GameObject[] linesOBJ;
private GameObject[] rayInfoOBJ;
private LineRenderer[] lineRenderers;
private RayInfoUI[] rayInfoUIs;
public List<GameObject> inViewEnemies = new List<GameObject>();
private List<string> tags = new List<string> { "Wall", "Enemy" };
private Onehot oneHotTags = new Onehot();
2022-10-25 19:07:39 +00:00
private void Start()
{
myTag = gameObject.tag;
2022-10-25 19:07:39 +00:00
totalRayNum = halfOuterRayNum * 2 + focusRayNum;
rayTagResult = new float[totalRayNum];
rayDisResult = new float[totalRayNum];
rayTagResultOneHot = new float[totalRayNum*ObjectTags.Tags.Count];
2022-10-25 19:07:39 +00:00
linesOBJ = new GameObject[totalRayNum];
lineRenderers = new LineRenderer[totalRayNum];
rayInfoOBJ = new GameObject[totalRayNum];
rayInfoUIs = new RayInfoUI[totalRayNum];
oneHotTags.Initialize(tags);
for (int i = 0; i < totalRayNum; i++)
2022-10-25 19:07:39 +00:00
{
linesOBJ[i] = new GameObject();
linesOBJ[i].name = "rayCastLine-" + Convert.ToString(i);
linesOBJ[i].transform.parent = agentCam.transform;
linesOBJ[i].AddComponent<LineRenderer>();
lineRenderers[i] = linesOBJ[i].GetComponent<LineRenderer>();
lineRenderers[i].material = lineMeterial;
rayInfoOBJ[i] = (GameObject)Instantiate(rayInfoPrefab);
rayInfoOBJ[i].transform.SetParent(agentCanvas.transform, false);
2022-10-25 19:07:39 +00:00
rayInfoOBJ[i].name = "rayInfo-" + Convert.ToString(i);
rayInfoUIs[i] = rayInfoOBJ[i].GetComponent<RayInfoUI>();
2022-10-25 19:07:39 +00:00
}
}
private void SingleRaycastUpdate(Ray ray,int rayIndex)
2022-10-25 19:07:39 +00:00
{
// get Raycast hit infomation and return Tag and distance
RaycastHit nowHit;
Color rayColor = Color.gray;
2022-10-25 19:07:39 +00:00
float lineLength = viewDistance;
string rayInfoText = "";
Vector3 rayInfoPosition;
if (Physics.Raycast(ray, out nowHit, viewDistance)) // 若在viewDistance范围内有碰撞
2022-10-25 19:07:39 +00:00
{
// Tag
string thisTag = nowHit.collider.tag;
// result update
rayTagResult[rayIndex] = ObjectTags.TagToInt(thisTag);
rayDisResult[rayIndex] = nowHit.distance;
int oneHotIndex = rayIndex * ObjectTags.Tags.Count + ObjectTags.TagToInt(thisTag);
rayTagResultOneHot[oneHotIndex] = 1f;
if (rayTagResult[rayIndex] == ObjectTags.TagToInt("Enemy"))
2022-10-25 19:07:39 +00:00
{
inViewEnemies.Add(nowHit.transform.gameObject);
2022-10-25 19:07:39 +00:00
}
// ingame info update
lineLength = nowHit.distance;
rayInfoText = thisTag + "\n" + Convert.ToString(nowHit.distance);
rayColor = ObjectTags.TagToCololr(thisTag);
2022-10-25 19:07:39 +00:00
}
else // 若在viewDistance范围无碰撞
{
// Result update
// rayTagResultOneHot keep zero
rayTagResult[rayIndex] = -1;
rayDisResult[rayIndex] = -1;
// Info
rayInfoText = "Empty";
2022-10-25 19:07:39 +00:00
}
// draw Info In Game
2022-10-25 19:07:39 +00:00
rayInfoPosition = ray.origin + (ray.direction * lineLength);
if (showInGameRay)
{
DrawLine(ray, lineLength, lineRenderers[rayIndex], rayColor);
2022-10-25 19:07:39 +00:00
}
else
{
TurnOffLine(lineRenderers[rayIndex], rayColor);
2022-10-25 19:07:39 +00:00
}
// drawRay in game
if (showInGameRayInfo) rayInfoUIs[rayIndex].UpdateInfo(rayInfoText, rayInfoPosition, rayColor, TPSCam);
2022-10-25 19:07:39 +00:00
// Show log
if (showDebugRay) Debug.DrawRay(ray.origin, ray.direction * viewDistance, rayColor); // drawRay in debug
// Debug.Log(ray.origin + ray.direction);
// Debug.Log(thisRayTagResult);
// Debug.Log(tagToInt(nowHit.collider.thisTag));
2022-10-25 19:07:39 +00:00
}
private void DrawLine(Ray ray, float lineLength, LineRenderer lineRenderer, Color lineColor)
2022-10-25 19:07:39 +00:00
{
lineRenderer.startColor = lineColor;
lineRenderer.endColor = lineColor;
lineRenderer.startWidth = lineWidth;
lineRenderer.endWidth = lineWidth;
lineRenderer.SetPosition(0, ray.origin);
lineRenderer.SetPosition(1, ray.origin + (ray.direction * lineLength));
2022-10-25 19:07:39 +00:00
}
private void TurnOffLine(LineRenderer lineRenderer, Color lineColor)
2022-10-25 19:07:39 +00:00
{
lineRenderer.startColor = lineColor;
lineRenderer.endColor = lineColor;
lineRenderer.startWidth = 0f;
lineRenderer.endWidth = 0f;
lineRenderer.SetPosition(0, new Vector3(0, 0, 0));
lineRenderer.SetPosition(1, new Vector3(0, 0, 0));
2022-10-25 19:07:39 +00:00
}
public void UpdateRayInfo()
2022-10-25 19:07:39 +00:00
{
float focusLEdge = agentCam.pixelWidth * (1 - focusRange) / 2;
float focusREdge = agentCam.pixelWidth * (1 + focusRange) / 2;
float camPixelHeight = agentCam.pixelHeight;
inViewEnemies.Clear();
ClearResult();
2022-10-25 19:07:39 +00:00
for (int i = 0; i < halfOuterRayNum; i++) // create left outside rays; 0 ~ focusLeftEdge
{
Vector3 point = new Vector3(i * focusLEdge / (halfOuterRayNum - 1), camPixelHeight / 2, 0);
Ray nowRay = agentCam.ScreenPointToRay(point);
SingleRaycastUpdate(nowRay,i);
2022-10-25 19:07:39 +00:00
}
for (int i = 0; i < halfOuterRayNum; i++) // create right outside rays; focusRightEdge ~ MaxPixelHeight
{
Vector3 point = new Vector3(focusREdge + (i * focusLEdge / (halfOuterRayNum - 1)), camPixelHeight / 2, 0);
Ray nowRay = agentCam.ScreenPointToRay(point);
SingleRaycastUpdate(nowRay, halfOuterRayNum + i);
2022-10-25 19:07:39 +00:00
}
for (int i = 0; i < focusRayNum; i++) // create center focus rays; focusLeftEdge ~ focusLeftEdge
{
Vector3 point = new Vector3(focusLEdge + ((i + 1) * (focusREdge - focusLEdge) / (focusRayNum + 1)), camPixelHeight / 2, 0);
Ray nowRay = agentCam.ScreenPointToRay(point);
SingleRaycastUpdate(nowRay, halfOuterRayNum * 2 + i);
2022-10-25 19:07:39 +00:00
}
}
private void ClearResult()
{
rayTagResult = new float[totalRayNum];
rayDisResult = new float[totalRayNum];
rayTagResultOneHot = new float[totalRayNum * ObjectTags.Tags.Count];
}
}