Merge pull request #23 from MorpheusXAUT/ccams-squawk-assignment

CCAMS squawk assignment

Author: MorpheusXAUT

DelHel/CDelHel.cpp

@@ -33,8 +33,12 @@ CDelHel::CDelHel() : EuroScopePlugIn::CPlugIn(
 	this->logMinMaxRFL = false;
 	this->checkMinMaxRFL = false;
 	this->flashOnMessage = false;
+	this->topSkyAvailable = false;
+	this->ccamsAvailable = false;
+	this->preferTopSkySquawkAssignment = false;
 
 	this->LoadSettings();
+	this->CheckLoadedPlugins();
 
 	this->ReadAirportConfig();
 	this->ReadRoutingConfig();
@@ -55,7 +59,7 @@ bool CDelHel::OnCompileCommand(const char* sCommandLine)
 	if (starts_with(args[0], ".delhel")) {
 		if (args.size() == 1) {
 			std::ostringstream msg;
-			msg << "Version " << PLUGIN_VERSION << " loaded. Available commands: auto, debug, nap, reload, reset, update, rflblw, logminmaxrfl, minmaxrfl, flash";
+			msg << "Version " << PLUGIN_VERSION << " loaded. Available commands: auto, debug, nap, reload, reset, update, rflblw, logminmaxrfl, minmaxrfl, flash, ccams";
 
 			this->LogMessage(msg.str());
 
@@ -190,6 +194,21 @@ bool CDelHel::OnCompileCommand(const char* sCommandLine)
 
 			this->SaveSettings();
 
+			return true;
+		}
+		else if (args[1] == "prefertopsky") {
+			if (this->preferTopSkySquawkAssignment) {
+				this->LogMessage("No longer preferring TopSky squawk assignment, will use CCAMS if loaded", "Config");
+			}
+			else {
+				this->LogMessage("Preferring TopSky squawk assignment, even if CCAMS is loaded", "Config");
+			}
+
+			this->preferTopSkySquawkAssignment = !this->preferTopSkySquawkAssignment;
+
+			this->SaveSettings();
+			this->CheckLoadedPlugins();
+
 			return true;
 		}
 	}
@@ -295,7 +314,7 @@ void CDelHel::LoadSettings()
 	if (settings) {
 		std::vector<std::string> splitSettings = split(settings, SETTINGS_DELIMITER);
 
-		if (splitSettings.size() < 7) {
+		if (splitSettings.size() < 8) {
 			this->LogMessage("Invalid saved settings found, reverting to default.");
 
 			this->SaveSettings();
@@ -310,6 +329,7 @@ void CDelHel::LoadSettings()
 		std::istringstream(splitSettings[4]) >> this->logMinMaxRFL;
 		std::istringstream(splitSettings[5]) >> this->checkMinMaxRFL;
 		std::istringstream(splitSettings[6]) >> this->flashOnMessage;
+		std::istringstream(splitSettings[7]) >> this->preferTopSkySquawkAssignment;
 
 		this->LogDebugMessage("Successfully loaded settings.");
 	}
@@ -327,7 +347,8 @@ void CDelHel::SaveSettings()
 		<< this->warnRFLBelowCFL << SETTINGS_DELIMITER
 		<< this->logMinMaxRFL << SETTINGS_DELIMITER
 		<< this->checkMinMaxRFL << SETTINGS_DELIMITER
-		<< this->flashOnMessage;
+		<< this->flashOnMessage << SETTINGS_DELIMITER
+		<< this->preferTopSkySquawkAssignment;
 
 	this->SaveDataToSettings(PLUGIN_NAME, "DelHel settings", ss.str().c_str());
 }
@@ -557,8 +578,15 @@ validation CDelHel::ProcessFlightPlan(EuroScopePlugIn::CFlightPlan& fp, bool nap
 				return res;
 			}
 
-			if (!cad.SetSquawk(VFR_SQUAWK)) {
-				this->LogDebugMessage("Failed to set VFR squawk", cs);
+			if (this->radarScreen != nullptr && this->ccamsAvailable && !this->preferTopSkySquawkAssignment) {
+				this->radarScreen->StartTagFunction(cs.c_str(), nullptr, 0, cs.c_str(), CCAMS_PLUGIN_NAME, CCAMS_TAG_FUNC_ASSIGN_SQUAWK_VFR, POINT(), RECT());
+				this->LogDebugMessage("Triggered automatic VFR squawk assignment via CCAMS", cs);
+			}
+			else {
+				// TopSky doesn't have a dedicated VFR squawk assignment function. Force hardcoded VFR squawk assignment
+				if (!cad.SetSquawk(VFR_SQUAWK)) {
+					this->LogDebugMessage("Failed to set VFR squawk", cs);
+				}
 			}
 
 			this->LogDebugMessage("Skipping processing of VFR flightplan route", cs);
@@ -734,11 +762,24 @@ validation CDelHel::ProcessFlightPlan(EuroScopePlugIn::CFlightPlan& fp, bool nap
 		}
 
 		if (this->radarScreen == nullptr) {
-			this->LogDebugMessage("Radar screen not initialised, cannot trigger automatic squawk assignment via TopSky", cs);
+			this->LogDebugMessage("Radar screen not initialised, cannot trigger automatic squawk assignment via TopSky or CCAMS", cs);
 		}
 		else {
-			this->radarScreen->StartTagFunction(cs.c_str(), nullptr, 0, cs.c_str(), TOPSKY_PLUGIN_NAME, TOPSKY_TAG_FUNC_ASSIGN_SQUAWK, POINT(), RECT());
-			this->LogDebugMessage("Triggered automatic squawk assignment via TopSky", cs);
+			if (this->preferTopSkySquawkAssignment && this->topSkyAvailable) {
+				this->radarScreen->StartTagFunction(cs.c_str(), nullptr, 0, cs.c_str(), TOPSKY_PLUGIN_NAME, TOPSKY_TAG_FUNC_ASSIGN_SQUAWK, POINT(), RECT());
+				this->LogDebugMessage("Triggered automatic squawk assignment via TopSky", cs);
+			}
+			else if (this->ccamsAvailable) {
+				this->radarScreen->StartTagFunction(cs.c_str(), nullptr, 0, cs.c_str(), CCAMS_PLUGIN_NAME, CCAMS_TAG_FUNC_ASSIGN_SQUAWK_AUTO, POINT(), RECT());
+				this->LogDebugMessage("Triggered automatic squawk assignment via CCAMS", cs);
+			}
+			else if (this->topSkyAvailable) {
+				this->radarScreen->StartTagFunction(cs.c_str(), nullptr, 0, cs.c_str(), TOPSKY_PLUGIN_NAME, TOPSKY_TAG_FUNC_ASSIGN_SQUAWK, POINT(), RECT());
+				this->LogDebugMessage("Triggered automatic squawk assignment via TopSky", cs);
+			}
+			else {
+				this->LogDebugMessage("Neither TopSky nor CCAMS are loaded, cannot trigger automatic squawk assignment", cs);
+			}
 		}
 
 		this->LogDebugMessage("Successfully processed flightplan", cs);
@@ -1030,6 +1071,47 @@ void CDelHel::CheckForUpdate()
 	this->latestVersion = std::future<std::string>();
 }
 
+void CDelHel::CheckLoadedPlugins()
+{
+	this->topSkyAvailable = false;
+	this->ccamsAvailable = false;
+
+	HMODULE hMods[1024];
+	HANDLE hProcess;
+	DWORD cbNeeded;
+	unsigned int i;
+
+	hProcess = OpenProcess(PROCESS_QUERY_INFORMATION | PROCESS_VM_READ, FALSE, GetCurrentProcessId());
+	if (hProcess == NULL) {
+		this->LogDebugMessage("Failed to check loaded plugins");
+		return;
+	}
+
+	if (EnumProcessModules(hProcess, hMods, sizeof(hMods), &cbNeeded)) {
+		for (i = 0; i < (cbNeeded / sizeof(HMODULE)); i++) {
+			TCHAR szModName[MAX_PATH];
+			if (GetModuleFileNameEx(hProcess, hMods[i], szModName, sizeof(szModName) / sizeof(TCHAR))) {
+				std::string moduleName = szModName;
+				size_t pos = moduleName.find_last_of("\\");
+				if (pos != std::string::npos) {
+					moduleName = moduleName.substr(pos + 1);
+				}
+
+				if (moduleName == TOPSKY_DLL_NAME) {
+					this->topSkyAvailable = true;
+					this->LogDebugMessage("Found TopSky plugin", "Config");
+				}
+				else if (moduleName == CCAMS_DLL_NAME) {
+					this->ccamsAvailable = true;
+					this->LogDebugMessage("Found CCAMS plugin", "Config");
+				}
+			}
+		}
+	}
+
+	CloseHandle(hProcess);
+}
+
 void __declspec (dllexport) EuroScopePlugInInit(EuroScopePlugIn::CPlugIn** ppPlugInInstance)
 {
 	*ppPlugInInstance = pPlugin = new CDelHel();

DelHel/CDelHel.h

@@ -8,6 +8,7 @@
 #include <fstream>
 #include <filesystem>
 #include <algorithm>
+#include <psapi.h>
 
 #include "EuroScope/EuroScopePlugIn.h"
 #include "semver/semver.hpp"
@@ -47,6 +48,9 @@ class CDelHel : public EuroScopePlugIn::CPlugIn
 	bool logMinMaxRFL;
 	bool checkMinMaxRFL;
 	bool flashOnMessage;
+	bool topSkyAvailable;
+	bool ccamsAvailable;
+	bool preferTopSkySquawkAssignment;
 	std::future<std::string> latestVersion;
 	std::map<std::string, airport> airports;
 	std::vector<std::string> processed;
@@ -69,5 +73,6 @@ class CDelHel : public EuroScopePlugIn::CPlugIn
 	void LogDebugMessage(std::string message, std::string type);
 
 	void CheckForUpdate();
+	void CheckLoadedPlugins();
 };
 

DelHel/DelHel.rc

@@ -3,9 +3,9 @@
 #include <regex>
 
 #define PLUGIN_NAME    "DelHel"
-#define PLUGIN_VERSION "0.1.0"
+#define PLUGIN_VERSION "0.2.0"
 #define PLUGIN_AUTHOR  "Nick Mueller"
-#define PLUGIN_LICENSE "(c) 2020-2022, MIT License"
+#define PLUGIN_LICENSE "(c) 2020-2023, MIT License"
 #define PLUGIN_LATEST_VERSION_URL "https://raw.githubusercontent.com/MorpheusXAUT/DelHel/master/version.txt"
 #define PLUGIN_LATEST_DOWNLOAD_URL "https://github.com/MorpheusXAUT/DelHel/releases/latest"
 
@@ -19,8 +19,14 @@ const int TAG_FUNC_PROCESS_FP_NAP = 102;
 const int TAG_FUNC_PROCESS_FP_NON_NAP = 103;
 
 constexpr auto TOPSKY_PLUGIN_NAME = "TopSky plugin";
+constexpr auto TOPSKY_DLL_NAME = "TopSky.dll";
 const int TOPSKY_TAG_FUNC_ASSIGN_SQUAWK = 667;
 
+constexpr auto CCAMS_PLUGIN_NAME = "CCAMS";
+constexpr auto CCAMS_DLL_NAME = "CCAMS.dll";
+const int CCAMS_TAG_FUNC_ASSIGN_SQUAWK_AUTO = 871;
+const int CCAMS_TAG_FUNC_ASSIGN_SQUAWK_VFR = 873;
+
 const COLORREF TAG_COLOR_NONE = 0;
 const COLORREF TAG_COLOR_RED = RGB(200, 0, 0);
 const COLORREF TAG_COLOR_ORANGE = RGB(255, 165, 0);