Skip to content

Commit

Permalink
- Aimmy 1.3.0 Update -
Browse files Browse the repository at this point in the history 
ChangeLog:
 - Adjusted AI to load via CPU if DirectML failed
 - Added configs to the UI (You will find a new configs folder in bin)
 - Fixed configs not changing FOV or confidence correctly
 - Fixed broken filewatcher (due to Nori's push :3)
 - Added KalmanFilter Predictions on aiming (toggle)
 - Added OnnxV8 Model check on load
 - More error checking / catching
 - Fixed a few small bugs
 - Cleaned up some code here and there
  • Loading branch information
Babyhamsta authored and Babyhamsta committed Nov 27, 2023
1 parent 0eb3672 commit d7f2788
Show file tree
Hide file tree
Showing 7 changed files with 162 additions and 64 deletions.
43 changes: 33 additions & 10 deletions AimmyWPF/AIModel.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
using System.Windows.Forms;
using System;
using System.Runtime.InteropServices;
using System.Diagnostics;

namespace AimmyAimbot
{
Expand All @@ -35,24 +36,46 @@ public class AIModel

public AIModel(string modelPath)
{
try
_modeloptions = new RunOptions();

var sessionOptions = new SessionOptions
{
_modeloptions = new RunOptions();
EnableCpuMemArena = true,
EnableMemoryPattern = true,
GraphOptimizationLevel = GraphOptimizationLevel.ORT_ENABLE_ALL,
ExecutionMode = ExecutionMode.ORT_PARALLEL
};

var sessionOptions = new SessionOptions();
sessionOptions.EnableCpuMemArena = true;
sessionOptions.EnableMemoryPattern = true;
sessionOptions.GraphOptimizationLevel = GraphOptimizationLevel.ORT_ENABLE_ALL;
sessionOptions.ExecutionMode = ExecutionMode.ORT_PARALLEL;
try
{
sessionOptions.AppendExecutionProvider_DML();

_onnxModel = new InferenceSession(modelPath, sessionOptions);
_outputNames = _onnxModel.OutputMetadata.Keys.ToList();
}
catch (Exception ex)
{
MessageBox.Show($"There was an error starting the OnnxModel: {ex}");
System.Windows.Application.Current.Shutdown();
MessageBox.Show($"There was an error starting the OnnxModel via DirectML: {ex}\n\nProgram will attempt to use CPU only, performance may be poor.", "Model Error");
try
{
sessionOptions.AppendExecutionProvider_CPU();
_onnxModel = new InferenceSession(modelPath, sessionOptions);
_outputNames = _onnxModel.OutputMetadata.Keys.ToList();
}
catch (Exception innerEx)
{
MessageBox.Show($"There was an error starting the model via CPU: {innerEx}", "Model Error");
System.Windows.Application.Current.Shutdown();
}
}

// Checking output shape
foreach (var output in _onnxModel.OutputMetadata)
{
var shape = _onnxModel.OutputMetadata[output.Key].Dimensions;
if (shape.Length != 3 || shape[0] != 1 || shape[1] != 5 || shape[2] != 8400)
{
MessageBox.Show($"Output shape {string.Join("x", shape)} does not match the expected shape of 1x5x8400.\n\nThis model will not work with Aimmy, please use an ONNX V8 model.", "Model Error");
}
}
}

Expand Down
4 changes: 3 additions & 1 deletion AimmyWPF/AimmyWPF.csproj
View file
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,8 @@
</PropertyGroup>

<ItemGroup>
<PackageReference Include="Accord.Math" Version="3.8.0" />
<PackageReference Include="Accord.Statistics" Version="3.8.0" />
<PackageReference Include="Costura.Fody" Version="5.7.0">
<PrivateAssets>all</PrivateAssets>
<IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
Expand Down Expand Up @@ -65,5 +67,5 @@
<LastGenOutput>Settings.Designer.cs</LastGenOutput>
</None>
</ItemGroup>

</Project>
1 change: 1 addition & 0 deletions AimmyWPF/Class/Bools.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ namespace AimmyWPF.Class
class Bools
{
public static bool AIAimAligner = false;
public static bool AIPredictions = false;
public static bool ThirdPersonAim = false;
public static bool Triggerbot = false;
public static bool CollectDataWhilePlaying = false;
Expand Down
117 changes: 72 additions & 45 deletions AimmyWPF/MainWindow.xaml.cs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,19 +16,25 @@
using AimmyAimbot;
using Class;
using Newtonsoft.Json;
using System.Diagnostics;

namespace AimmyWPF
{
public partial class MainWindow : Window
{
private PredictionManager predictionManager;
private OverlayWindow FOVOverlay;
private FileSystemWatcher fileWatcher;
private FileSystemWatcher ConfigfileWatcher;

private string lastLoadedModel = "N/A";
private string lastLoadedConfig = "N/A";

private readonly BrushConverter brushcolor = new BrushConverter();

private int TimeSinceLastClick = 0;
private DateTime LastClickTime = DateTime.MinValue;

private const uint MOUSEEVENTF_LEFTDOWN = 0x0002;
private const uint MOUSEEVENTF_LEFTUP = 0x0004;
private const uint MOUSEEVENTF_MOVE = 0x0001; // Movement flag
Expand Down Expand Up @@ -63,6 +69,7 @@ private enum MenuPosition
private Dictionary<string, bool> toggleState = new Dictionary<string, bool>
{
{ "AimbotToggle", false },
{ "PredictionToggle", false },
{ "AimViewToggle", false },
{ "TriggerBot", false },
{ "CollectData", false },
Expand Down Expand Up @@ -95,20 +102,20 @@ public MainWindow()
RequirementsManager RM = new RequirementsManager();
if (!RM.IsVCRedistInstalled())
{
System.Windows.MessageBox.Show("Visual C++ Redistributables x64 are not installed on this device, please install them before using Aimmy to avoid issues.");
System.Windows.MessageBox.Show("Visual C++ Redistributables x64 are not installed on this device, please install them before using Aimmy to avoid issues.", "Load Error");
System.Diagnostics.Process.Start("https://aka.ms/vs/17/release/vc_redist.x64.exe");
}

// Check for required folders
string baseDir = AppDomain.CurrentDomain.BaseDirectory;
string[] dirs = { "bin", "bin/models", "bin/images" };
string[] dirs = { "bin", "bin/models", "bin/images", "bin/configs" };

foreach (string dir in dirs)
{
string fullPath = Path.Combine(baseDir, dir);
if (!Directory.Exists(fullPath))
{
System.Windows.MessageBox.Show($"The '{dir}' folder does not exist, please ensure the folder is in the same directory as the exe.");
System.Windows.MessageBox.Show($"The '{dir}' folder does not exist, please ensure the folder is in the same directory as the exe.", "Load Error");
System.Windows.Application.Current.Shutdown();
}
}
Expand Down Expand Up @@ -145,16 +152,22 @@ public MainWindow()
InitializeFileWatcher();
InitializeConfigWatcher();

// Load PredictionManager
predictionManager = new PredictionManager();
predictionManager.InitializeKalmanFilter();

// Load all models into listbox
LoadModelsIntoListBox();
LoadConfigsIntoListBox();
//InitializeModel();

SelectorListBox.SelectionChanged += new SelectionChangedEventHandler(SelectorListBox_SelectionChanged);
ConfigSelectorListBox.SelectionChanged += new SelectionChangedEventHandler(ConfigSelectorListBox_SelectionChanged);

// Create FOV Overlay
FOVOverlay = new OverlayWindow();
FOVOverlay.Hide();
FOVOverlay.FovSize = (int)aimmySettings["FOV_Size"];

// Start the loop that runs the model
Task.Run(() => StartModelCaptureLoop());
}
Expand Down Expand Up @@ -246,7 +259,6 @@ public async Task ModelCapture(bool TriggerOnly = false)

if (!TriggerOnly)
{
// Scale the coordinates from the 640x640 image to the actual screen size
float scaleX = (float)ScreenWidth / 640f;
float scaleY = (float)ScreenHeight / 640f;

Expand All @@ -255,9 +267,19 @@ public async Task ModelCapture(bool TriggerOnly = false)
int detectedX = (int)((closestPrediction.Rectangle.X * scaleX) + XOffset);
int detectedY = (int)((closestPrediction.Rectangle.Y * scaleY) + YOffset);

MoveCrosshair(detectedX, detectedY);
// Handle Prediction
if (toggleState["PredictionToggle"])
{
predictionManager.UpdateKalmanFilter(detectedX, detectedY);
var predictedPosition = predictionManager.GetEstimatedPosition();
MoveCrosshair(predictedPosition.X, predictedPosition.Y);
}
else
{
MoveCrosshair(detectedX, detectedY);
}
}
else
else
{
Task.Run(() => DoTriggerClick());
}
Expand Down Expand Up @@ -324,7 +346,7 @@ private void SetToggleState(AToggle toggle)
// Stop them from turning on anything until model has been selected.
if ((toggle.Reader.Name == "AimbotToggle" || toggle.Reader.Name == "TriggerBot" || toggle.Reader.Name == "CollectData") && lastLoadedModel == "N/A")
{
System.Windows.MessageBox.Show("Please select a model in the Model Selector before toggling.");
System.Windows.MessageBox.Show("Please select a model in the Model Selector before toggling.", "Toggle Error");
return;
}

Expand Down Expand Up @@ -458,11 +480,6 @@ void LoadAimMenu()
SetupToggle(Enable_AIAimAligner, state => Bools.AIAimAligner = state, Bools.AIAimAligner);
AimScroller.Children.Add(Enable_AIAimAligner);

/*AToggle ThirdPersonAim = new AToggle("Enable Third Person Aim");
ThirdPersonAim.Reader.Name = "AimViewToggle";
SetupToggle(ThirdPersonAim, state => Bools.ThirdPersonAim = state, Bools.ThirdPersonAim);
AimScroller.Children.Add(ThirdPersonAim);*/

AKeyChanger Change_KeyPress = new AKeyChanger("Change Keybind", "Right");
Change_KeyPress.Reader.Click += (s, x) =>
{
Expand All @@ -477,6 +494,12 @@ void LoadAimMenu()

AimScroller.Children.Add(Change_KeyPress);

AToggle Enable_AIPredictions = new AToggle(this, "Enable Predictions",
"This will use a KalmanFilter algorithm to predict aim patterns for better tracing of enemies.");
Enable_AIPredictions.Reader.Name = "PredictionToggle";
SetupToggle(Enable_AIPredictions, state => Bools.AIPredictions = state, Bools.AIPredictions);
AimScroller.Children.Add(Enable_AIPredictions);

AToggle Show_FOV = new AToggle(this, "Show FOV",
"This will show a circle around your screen that show what the AI is considering on the screen at a given moment.");
Show_FOV.Reader.Name = "ShowFOV";
Expand Down Expand Up @@ -626,7 +649,6 @@ private void LoadModelsIntoListBox()
SelectorListBox.Items.Add(fileName);
}

// Preselect the first file in the ListBox
if (SelectorListBox.Items.Count > 0)
{
if (!SelectorListBox.Items.Contains(lastLoadedModel) && lastLoadedModel != "N/A")
Expand All @@ -640,40 +662,45 @@ private void LoadModelsIntoListBox()
}
SelectedModelNotifier.Content = "Loaded Model: " + lastLoadedModel;
}
else
{
System.Windows.MessageBox.Show("No models found, please put a .onnx model in bin/models.");
System.Windows.Application.Current.Shutdown();
}
}

private void SelectorListBox_SelectionChanged(object sender, SelectionChangedEventArgs e)
{
InitializeModel();
}

// nori's config stuff, might need optimization later

#region N4ri's Config Code (god please fix this)

void LoadConfig(string path)
{
if (ConfigSelectorListBox.SelectedItem != null)
if (ConfigSelectorListBox.SelectedItem != null && lastLoadedModel != "N/A")
{
dynamic AimmyJSON = JsonConvert.DeserializeObject(File.ReadAllText(path));

System.Windows.MessageBox.Show("The creator of this model suggests you use this model:" +
"\n" +
AimmyJSON.Suggested_Model, "Suggested Model - Aimmy");

aimmySettings["FOV_Size"] = AimmyJSON.FOV_Size;
aimmySettings["FOV_Size"] = (int)AimmyJSON.FOV_Size;
FOVOverlay.FovSize = (int)aimmySettings["FOV_Size"];
_onnxModel.FovSize = (int)aimmySettings["FOV_Size"];

aimmySettings["Mouse_Sens"] = AimmyJSON.Mouse_Sensitivity;
aimmySettings["Y_Offset"] = AimmyJSON.Y_Offset;
aimmySettings["X_Offset"] = AimmyJSON.X_Offset;
aimmySettings["Trigger_Delay"] = AimmyJSON.Auto_Trigger_Delay;

aimmySettings["AI_Min_Conf"] = AimmyJSON.AI_Minimum_Confidence;
_onnxModel.ConfidenceThreshold = (float)(aimmySettings["AI_Min_Conf"] / 100.0f);

lastLoadedConfig = ConfigSelectorListBox.SelectedItem.ToString();

// Reload the UI
ReloadMenu();
}
else
{
ConfigSelectorListBox.SelectedItem = null;
System.Windows.MessageBox.Show("Please select a model in the Model Selector before loading a config.", "Config Error");
}
}

void ReloadMenu()
Expand All @@ -697,13 +724,13 @@ private void ConfigWatcher_Reload(object sender, FileSystemEventArgs e)

private void InitializeConfigWatcher()
{
fileWatcher = new FileSystemWatcher();
fileWatcher.Path = "bin/configs";
fileWatcher.Filter = "*.json";
fileWatcher.EnableRaisingEvents = true;
fileWatcher.Created += ConfigWatcher_Reload;
fileWatcher.Deleted += ConfigWatcher_Reload;
fileWatcher.Renamed += ConfigWatcher_Reload;
ConfigfileWatcher = new FileSystemWatcher();
ConfigfileWatcher.Path = "bin/configs";
ConfigfileWatcher.Filter = "*.json";
ConfigfileWatcher.EnableRaisingEvents = true;
ConfigfileWatcher.Created += ConfigWatcher_Reload;
ConfigfileWatcher.Deleted += ConfigWatcher_Reload;
ConfigfileWatcher.Renamed += ConfigWatcher_Reload;
}

private void LoadConfigsIntoListBox()
Expand All @@ -717,42 +744,42 @@ private void LoadConfigsIntoListBox()
ConfigSelectorListBox.Items.Add(fileName);
}

// Preselect the first file in the ListBox
if (ConfigSelectorListBox.Items.Count > 0)
{
if (!ConfigSelectorListBox.Items.Contains(lastLoadedModel) && lastLoadedModel != "N/A")
if (!ConfigSelectorListBox.Items.Contains(lastLoadedConfig) && lastLoadedConfig != "N/A")
{
ConfigSelectorListBox.SelectedIndex = 0;
lastLoadedModel = ConfigSelectorListBox.Items[0].ToString();
lastLoadedConfig = ConfigSelectorListBox.Items[0].ToString();
}
else
{
SelectorListBox.SelectedItem = lastLoadedModel;
ConfigSelectorListBox.SelectedItem = lastLoadedConfig;
}
SelectedModelNotifier.Content = "Loaded Model: " + lastLoadedModel;
}
else
{
System.Windows.MessageBox.Show("No configs found, please put a .json file in bin/configs.");
System.Windows.Application.Current.Shutdown();
SelectedConfigNotifier.Content = "Loaded Config: " + lastLoadedConfig;
}
}

private void ConfigSelectorListBox_SelectionChanged(object sender, SelectionChangedEventArgs e)
{
LoadConfig($"bin/configs/{ConfigSelectorListBox.SelectedItem.ToString()}");
if (ConfigSelectorListBox.SelectedItem != null)
{
LoadConfig($"bin/configs/{ConfigSelectorListBox.SelectedItem.ToString()}");
}
}
#endregion

void LoadModelStoreMenu()
{
if (AvailableModels.Count > 0)
{
foreach (var entries in AvailableModels)
{
ModelStoreScroller.Children.Add(new ADownloadGateway(entries));
}
}
else
{
LackOfModelsText.Visibility = Visibility.Visible;
}
}

void LoadSettingsMenu()
Expand Down Expand Up @@ -786,7 +813,7 @@ void LoadSettingsMenu()
// Prevent double messageboxes..
if (AIMinimumConfidence.Slider.Value != aimmySettings["AI_Min_Conf"])
{
System.Windows.MessageBox.Show("Unable to set confidence, please select a model and try again.");
System.Windows.MessageBox.Show("Unable to set confidence, please select a model and try again.", "Slider Error");
AIMinimumConfidence.Slider.Value = aimmySettings["AI_Min_Conf"];
}
}
Expand Down
Loading

0 comments on commit d7f2788

Please sign in to comment.