mirror of
https://github.com/ishanjitchattaraj/simple-sober-external.git
synced 2026-10-27 12:44:36 +00:00
116 lines
11 KiB
C#
116 lines
11 KiB
C#
using System.Numerics;
|
|
using System.Text.Json;
|
|
using System.Runtime.InteropServices;
|
|
using SoberEsp;
|
|
public static class Program
|
|
{
|
|
[STAThread]public static int Main(string[] args)
|
|
{
|
|
string? Option(string name){int i=Array.IndexOf(args,name);return i>=0&&i+1<args.Length?args[i+1]:null;}
|
|
string home=Path.GetFullPath(Option("--home")??Environment.GetEnvironmentVariable("SOBER_ESP_HOME")??AppContext.BaseDirectory);Directory.CreateDirectory(home);string socket=Option("--socket")??Path.Combine(home,"memory.sock");
|
|
using var cancel=new CancellationTokenSource();Console.CancelKeyPress+=(_,e)=>{e.Cancel=true;cancel.Cancel();};
|
|
using var terminate=PosixSignalRegistration.Create(PosixSignal.SIGTERM,context=>{context.Cancel=true;cancel.Cancel();});
|
|
using var hangup=PosixSignalRegistration.Create(PosixSignal.SIGHUP,context=>{context.Cancel=true;cancel.Cancel();});
|
|
try
|
|
{
|
|
if(args.Contains("--self-test")){SelfTest();return 0;}
|
|
if(args.Contains("--bridge"))
|
|
{
|
|
if(Option("--watch-pid") is {} watch)
|
|
{
|
|
int watched=int.Parse(watch);
|
|
_=Task.Run(async()=>{try{using var process=System.Diagnostics.Process.GetProcessById(watched);await process.WaitForExitAsync(cancel.Token);}catch(ArgumentException){}catch(OperationCanceledException){}finally{cancel.Cancel();}});
|
|
}
|
|
new BridgeServer(socket,args.Contains("--allow-camera-writes")).Run(cancel.Token);return 0;
|
|
}
|
|
OpenTK.Windowing.GraphicsLibraryFramework.GLFW.InitHint(OpenTK.Windowing.GraphicsLibraryFramework.InitHintPlatform.Platform,OpenTK.Windowing.GraphicsLibraryFramework.Platform.X11);
|
|
var bridge=new MemoryBridge(socket);var tracker=new Tracker(bridge,home);
|
|
if(args.Contains("--render-test")){using var test=new Overlay(tracker,home,true);test.Run();return 0;}
|
|
var worker=Task.Run(()=>tracker.Run(cancel.Token));
|
|
if(args.Contains("--snapshot"))
|
|
{
|
|
var end=DateTime.UtcNow.AddSeconds(30);while(DateTime.UtcNow<end&&tracker.Latest.Status!="Live"&&!cancel.IsCancellationRequested)Thread.Sleep(50);var state=tracker.Latest;if(state.Status!="Live")throw new IOException(state.Status);Console.WriteLine(JsonSerializer.Serialize(state,new JsonSerializerOptions{WriteIndented=true,IncludeFields=true}));cancel.Cancel();return 0;
|
|
}
|
|
using var instance=new FileStream(Path.Combine(home,"overlay.lock"),FileMode.OpenOrCreate,FileAccess.ReadWrite,FileShare.None);File.WriteAllText(Path.Combine(home,"overlay.pid"),Environment.ProcessId.ToString());
|
|
try{using var overlay=new Overlay(tracker,home);using var registration=cancel.Token.Register(()=>overlay.Close());overlay.Run();}finally{cancel.Cancel();File.Delete(Path.Combine(home,"overlay.pid"));}
|
|
return 0;
|
|
}
|
|
catch(Exception e){Console.Error.WriteLine(e);return 1;}
|
|
}
|
|
static void SelfTest()
|
|
{
|
|
float[] identity=[1,0,0,0,1,0,0,0,1,0,0,0];var center=Tracker.Project(new(0,0,-10),identity,MathF.PI/2,1000,1000);if(center is null||Vector3.Distance(center.Value,new(500,500,10))>.01)throw new Exception("Center projection failed");
|
|
var axes=Tracker.Project(new(1,1,-10),identity,MathF.PI/2,1000,1000);if(axes is null||Vector3.Distance(axes.Value,new(550,450,10))>.01)throw new Exception("Axes projection failed");
|
|
if(Tracker.Project(new(0,0,10),identity,1,100,100)!=null||Tracker.Project(new(float.NaN,0,-10),identity,1,100,100)!=null)throw new Exception("Clipping failed");
|
|
float[] rotated=[0,0,1,0,1,0,-1,0,0,20,3,4];var result=Tracker.Project(new(10,3,4),rotated,MathF.PI/2,1000,1000);if(result is null||Vector3.Distance(result.Value,new(500,500,10))>.01)throw new Exception("Rotated projection failed");
|
|
var maps=BridgeServer.ReadMaps(Environment.ProcessId);if(maps.Count==0||maps.Any(m=>m.Start>=m.End))throw new Exception("Memory-map parsing failed");if(Tracker.Project(new(0,0,-10),identity,float.NaN,100,100)!=null)throw new Exception("NaN FOV accepted");
|
|
var settings=new Settings();var entity=new Entity("test",new(490,400,20,100),10,default,new(510,500));
|
|
if(AimMath.Select([entity],Vector2.One,new(500,500),settings)!=entity)throw new Exception("Aim selection failed");
|
|
settings.MaxDistance=5;if(AimMath.Select([entity],Vector2.One,new(500,500),settings)!=null)throw new Exception("Aim distance filter failed");
|
|
settings.MaxDistance=100;settings.FovRadius=5;if(AimMath.Select([entity],Vector2.One,new(500,500),settings)!=null)throw new Exception("Aim FOV filter failed");
|
|
var nearMouse=new Entity("same name",new(490,400,20,100),100,default,new(510,500),1,80);
|
|
var nearPlayer=new Entity("same name",new(540,400,20,100),200,default,new(550,500),2,10);
|
|
var ranking=new Settings();var crosshair=new Vector2(500,500);
|
|
if(AimMath.Select([nearMouse,nearPlayer],Vector2.One,crosshair,ranking)?.Id!=1)throw new Exception("Mouse priority failed");
|
|
var left=nearMouse with{AimPoint=new(470,500),Id=3};
|
|
var right=nearPlayer with{AimPoint=new(510,500),Id=4};
|
|
if(AimMath.Select([left,right],Vector2.One,crosshair,ranking)?.Id!=4)throw new Exception("Crosshair selection biased left");
|
|
if(AimMath.Select([left with{AimPoint=new(490,500)},right with{AimPoint=new(530,500)}],Vector2.One,crosshair,ranking)?.Id!=3)throw new Exception("Crosshair selection biased right");
|
|
if(AimMath.Select([left,right],new(2,1.5f),new(1000,750),ranking)?.Id!=4)throw new Exception("Scaled crosshair selection failed");
|
|
ranking.TargetPriority=TargetPriority.ClosestToPlayer;
|
|
if(AimMath.Select([nearMouse,nearPlayer],Vector2.One,crosshair,ranking)?.Id!=2)throw new Exception("Player priority used camera distance");
|
|
if(AimMath.Select([nearPlayer with{PlayerDistance=null}],Vector2.One,crosshair,ranking)!=null)throw new Exception("Missing local position must not select a player");
|
|
ranking.FovRadius=20;if(AimMath.Select([nearMouse,nearPlayer],Vector2.One,crosshair,ranking)?.Id!=1)throw new Exception("Player priority ignored FOV");
|
|
ranking.FovRadius=140;ranking.TargetPriority=TargetPriority.ClosestToMouse;ranking.AimLock=true;
|
|
var selection=new AimSelection();
|
|
if(selection.Update([nearMouse,nearPlayer],Vector2.One,crosshair,ranking,true)?.Id!=1)throw new Exception("Initial lock failed");
|
|
var moved=nearMouse with{AimPoint=new(560,500)};
|
|
var challenger=nearPlayer with{AimPoint=new(501,500)};
|
|
if(selection.Update([moved,challenger],Vector2.One,crosshair,ranking,true)!=moved)throw new Exception("Lock hopped targets or kept stale coordinates");
|
|
if(selection.Update([challenger],Vector2.One,crosshair,ranking,true)!=null)throw new Exception("Missing locked target caused a hop");
|
|
if(selection.Update([moved with{AimPoint=new(900,500)},challenger],Vector2.One,crosshair,ranking,true)!=null)throw new Exception("Lock ignored FOV");
|
|
if(selection.Update([moved with{Distance=9999},challenger],Vector2.One,crosshair,ranking,true)!=null)throw new Exception("Lock ignored range");
|
|
selection.Update([moved,challenger],Vector2.One,crosshair,ranking,false);
|
|
if(selection.Update([moved,challenger],Vector2.One,crosshair,ranking,true)?.Id!=2)throw new Exception("Release did not unlock");
|
|
ranking.AimLock=false;
|
|
if(selection.Update([nearMouse,nearPlayer],Vector2.One,crosshair,ranking,true)?.Id!=1)throw new Exception("Disabling lock did not resume ranking");
|
|
// Default/missing player distance must remain serializable in --snapshot output.
|
|
_=JsonSerializer.Serialize(nearMouse with{PlayerDistance=null});
|
|
var step=AimMath.Motion(new(100,0),1,8,1d/60);if(step.X<=0||step.X>=100||step.Y!=0)throw new Exception("Aim smoothing failed");
|
|
if(AimMath.Motion(new(10000,0),3,1,1).Length()>40.01f)throw new Exception("Aim motion cap failed");
|
|
var full=AimMath.Motion(new(10,0),1,8,1d/30);
|
|
var half=AimMath.Motion(new(10,0),1,8,1d/60);
|
|
var halfAgain=AimMath.Motion(new Vector2(10,0)-half,1,8,1d/60);
|
|
if(Vector2.Distance(full,half+halfAgain)>.001f)throw new Exception("Aim frame-rate independence failed");
|
|
CameraTests();ExplorerTests.Run();CharacterTrackingTests.Run();
|
|
string settingsPath=Path.Combine(Path.GetTempPath(),"sober-settings-"+Guid.NewGuid()+".json");
|
|
try{var saved=new Settings{FovRadius=250,Smoothness=12,Names=false,Sensitivity=20,AimLock=true,TargetPriority=TargetPriority.ClosestToPlayer,AimMethod=AimMethod.Camera};saved.Save(settingsPath);var loaded=Settings.Load(settingsPath);if(loaded.FovRadius!=250||loaded.Smoothness!=12||loaded.Names||loaded.Sensitivity!=20||!loaded.AimLock||loaded.TargetPriority!=TargetPriority.ClosestToPlayer||loaded.AimMethod!=AimMethod.Camera)throw new Exception("Settings round-trip failed");File.WriteAllText(settingsPath,"invalid");if(Settings.Load(settingsPath).AimEnabled)throw new Exception("Invalid settings fallback failed");}finally{File.Delete(settingsPath);}
|
|
Console.WriteLine("Self-tests passed: fast movement, close/ragdoll projection, head failures, part replacement, nested characters, explorer hierarchy/pagination/cancellation, camera rotation/projection/smoothing, centered/scaled crosshair selection, target priorities, stable aim lock, release/reacquisition, settings persistence/fallback, frame-rate independence, aim selection, distance/FOV filters, smoothing, motion cap, NaN FOV, projection, axes, rotation, clipping and /proc map parsing.");
|
|
}
|
|
static void CameraTests()
|
|
{
|
|
float[] original=[1,0,0,0,1,0,0,0,1,20,3,4];
|
|
foreach(var target in new Vector3[]{new(30,5,-6),new(10,1,-6),new(20,13,4),new(20,-7,4),new(20,3,14)})
|
|
{
|
|
var rotation=CameraAim.Rotation(original,target,1);var aimed=(float[])original.Clone();Array.Copy(rotation,aimed,rotation.Length);
|
|
var projected=Tracker.Project(target,aimed,MathF.PI/2,1000,800);
|
|
if(projected is null||Vector2.Distance(new(projected.Value.X,projected.Value.Y),new(500,400))>.01f)throw new Exception("Camera rotation did not center target");
|
|
if(aimed[9]!=20||aimed[10]!=3||aimed[11]!=4)throw new Exception("Camera aim changed position");
|
|
// Also validate this result as the next input; covers handedness and pole rotations.
|
|
_=CameraAim.Rotation(aimed,new(30,5,-6),.1f);
|
|
}
|
|
var unchanged=CameraAim.Rotation(original,new(30,3,-6),0);
|
|
if(Enumerable.Range(0,9).Any(i=>Math.Abs(unchanged[i]-original[i])>.0001f))throw new Exception("Zero camera response changed orientation");
|
|
var half=CameraAim.Rotation(original,new(30,3,-6),.5f);var halfway=(float[])original.Clone();Array.Copy(half,halfway,9);
|
|
var before=Tracker.Project(new(30,3,-6),original,MathF.PI/2,1000,800)!.Value;
|
|
var after=Tracker.Project(new(30,3,-6),halfway,MathF.PI/2,1000,800)!.Value;
|
|
if(!(after.X>500&&after.X<before.X))throw new Exception("Camera smoothing did not approach the target");
|
|
try{CameraAim.Rotation(original,new(float.NaN,0,0),1);throw new Exception("Camera accepted NaN target");}catch(IOException){}
|
|
try{CameraAim.Rotation(original,new(20,3,4),1);throw new Exception("Camera accepted zero target direction");}catch(IOException){}
|
|
var invalid=(float[])original.Clone();invalid[0]=2;
|
|
try{CameraAim.Rotation(invalid,new(30,3,-6),1);throw new Exception("Camera accepted invalid basis");}catch(IOException){}
|
|
if(half.Length*sizeof(float)!=Offsets.Camera.RotationBytes)throw new Exception("Camera write size mismatch");
|
|
}
|
|
|
|
}
|