120 lines
5.1 KiB
C#
120 lines
5.1 KiB
C#
using System.Collections.Generic;
|
|
using System.IO;
|
|
using CounterDrone.Core;
|
|
using CounterDrone.Core.Algorithms;
|
|
using CounterDrone.Core.Models;
|
|
using CounterDrone.Core.Repository;
|
|
using CounterDrone.Core.Services;
|
|
using UnityEngine;
|
|
using UVector3 = UnityEngine.Vector3;
|
|
|
|
namespace CounterDrone.Unity
|
|
{
|
|
/// <summary>一键启动：创建想定 → 推荐方案 → 运行仿真</summary>
|
|
public class SimulationBootstrap : MonoBehaviour
|
|
{
|
|
[Header("Simulation Settings")]
|
|
[SerializeField] private float _droneSpeed = 200f;
|
|
[SerializeField] private float _routeLength = 20000f;
|
|
[SerializeField] private float _timeScale = 4f;
|
|
|
|
public SimulationRunner Runner { get; private set; }
|
|
|
|
void Start()
|
|
{
|
|
var paths = new UnityPathProvider();
|
|
|
|
// 清理上次残留数据
|
|
if (System.IO.File.Exists(paths.GetMainDbPath()))
|
|
System.IO.File.Delete(paths.GetMainDbPath());
|
|
if (System.IO.Directory.Exists(paths.GetFramesDir()))
|
|
System.IO.Directory.Delete(paths.GetFramesDir(), true);
|
|
|
|
var dbm = new DatabaseManager(paths);
|
|
var db = dbm.OpenMainDb();
|
|
|
|
// 弹药规格
|
|
db.Insert(new AmmunitionSpec
|
|
{
|
|
AerosolType = (int)AerosolType.InertGas, Name = "惰性气体弹",
|
|
InitialRadius = 50, CoreDensity = 1.0, DispersionRateBase = 5,
|
|
MaxRadius = 500, MaxDuration = 120, EffectiveConcentration = 0.05,
|
|
});
|
|
|
|
// 创建想定
|
|
var scenario = new ScenarioService(
|
|
new SimTaskRepository(db), new CombatSceneRepository(db),
|
|
new ControlZoneRepository(db), new TargetConfigRepository(db),
|
|
new EquipmentDeploymentRepository(db), new CloudDispersalRepository(db),
|
|
new RoutePlanRepository(db), new WaypointRepository(db));
|
|
|
|
var task = scenario.CreateTask("Unity Demo", "");
|
|
var taskId = task.Id;
|
|
|
|
scenario.SaveScene(taskId, new CombatScene
|
|
{
|
|
SceneType = (int)SceneType.Plain, WindSpeed = 3,
|
|
WindDirection = (int)WindDirection.W, Temperature = 22,
|
|
});
|
|
scenario.SaveTarget(taskId, new TargetConfig
|
|
{
|
|
TargetType = (int)TargetType.Piston, PowerType = (int)PowerType.Piston,
|
|
Quantity = 1, TypicalSpeed = _droneSpeed, TypicalAltitude = 500,
|
|
});
|
|
scenario.SaveRoute(taskId, new RoutePlan { FormationMode = (int)FormationMode.Single },
|
|
new List<Waypoint>
|
|
{
|
|
new Waypoint { PosX = 0, PosY = 500, PosZ = 0, Speed = _droneSpeed },
|
|
new Waypoint { PosX = _routeLength, PosY = 500, PosZ = 0, Speed = _droneSpeed },
|
|
});
|
|
|
|
// 推荐方案
|
|
var detail = scenario.GetTaskDetail(taskId);
|
|
var ammoCatalog = db.Table<AmmunitionSpec>().ToList();
|
|
var rec = new DefaultDefenseAdvisor(ammoCatalog).Recommend(new ThreatProfile
|
|
{
|
|
Environment = detail.Scene, Targets = detail.Targets,
|
|
Route = detail.Route, Waypoints = detail.Waypoints,
|
|
});
|
|
|
|
scenario.SaveCloudDispersal(taskId, rec.Best.RecommendedCloud);
|
|
|
|
var equips = new List<EquipmentDeployment>();
|
|
foreach (var p in rec.Best.Platforms)
|
|
equips.Add(new EquipmentDeployment
|
|
{
|
|
EquipmentRole = (int)EquipmentRole.LaunchPlatform,
|
|
PlatformType = (int)p.Type, Quantity = 1,
|
|
PositionX = p.Position.X, PositionY = p.Position.Y, PositionZ = p.Position.Z,
|
|
AerosolType = (int)rec.Best.RecommendedAerosolType,
|
|
MunitionCount = 1, Cooldown = p.Cooldown, MuzzleVelocity = p.MuzzleVelocity,
|
|
});
|
|
scenario.SaveDeployment(taskId, equips);
|
|
|
|
db.Close();
|
|
|
|
Debug.Log($"推荐方案: {rec.Best.RecommendedAerosolType}, {rec.Best.Platforms.Count}门炮, 拦截概率 {rec.Best.InterceptProbability:P0}");
|
|
|
|
// 运行仿真
|
|
Runner = GetComponent<SimulationRunner>();
|
|
if (Runner == null) Runner = gameObject.AddComponent<SimulationRunner>();
|
|
else Runner.enabled = true;
|
|
Runner.LoadAndStart(taskId);
|
|
|
|
// 设置倍速
|
|
if (Runner.Engine != null)
|
|
{
|
|
Runner.Engine.TimeScale = _timeScale;
|
|
Debug.Log($"Engine state: {Runner.Engine.State}, drones: {Runner.Engine.Drones.Count}");
|
|
}
|
|
|
|
// 移动 Game 视图相机
|
|
var cam = Camera.main;
|
|
if (cam == null) { var cgo = new GameObject("MainCamera"); cgo.AddComponent<Camera>(); cam = cgo.GetComponent<Camera>(); cam.tag = "MainCamera"; }
|
|
cam.transform.position = new UVector3(_routeLength / 2f, 2000, -3000);
|
|
cam.transform.LookAt(new UVector3(_routeLength / 2f, 500, 0));
|
|
Debug.Log("Camera positioned. In Scene view, double-click 'Simulation' in Hierarchy to focus.");
|
|
}
|
|
}
|
|
}
|