Add Saved selection for modules and Transfer Ownership Module

addons/common/XEH_postInit.sqf

@@ -318,6 +318,31 @@ if (isServer) then {
         _object enableSimulationGlobal _enable;
     }] call CBA_fnc_addEventHandler;
 
+    [QGVAR(transferOwnership), {
+        params ["_entities", "_target"];
+        if (!(_entities isEqualType [])) then {
+            _entities = [_entities];
+        };
+        private _clientID = 0;
+        if (_target isEqualType 0) then {
+            _clientID = _target;
+        };
+        if (_target isEqualType objNull) then {
+            _clientID = owner _target;
+        };
+        {
+            if (_x isEqualType grpNull) then {
+                _x setGroupOwner _clientID;
+            } else {
+                if (group _x == grpNull) then {
+                    group _x setGroupOwner _clientID;
+                } else {
+                    _x setOwner _clientID;
+                };
+            };
+        } forEach _entities;
+    }] call CBA_fnc_addEventHandler;
+
     [QGVAR(setFriend), {
         params ["_side1", "_side2", "_value"];
         _side1 setFriend [_side2, _value];

addons/dialog/functions/fnc_close.sqf

@@ -38,5 +38,7 @@ if (_confirmed) then {
     [_values, _args] call _onCancel;
 };
 
+[QGVAR(close), [_display, _confirmed]] call CBA_fnc_localEvent;
+
 // Close dialog, returning false to not override engine driven IDC_OK and IDC_CANCEL
 false

addons/editor/XEH_PREP.hpp

@@ -1,6 +1,7 @@
 PREP(addGroupIcons);
 PREP(declutterEmptyTree);
 PREP(fixSideButtons);
+PREP(getSelection);
 PREP(handleKeyDown);
 PREP(handleLoad);
 PREP(handleModeButtons);
@@ -9,6 +10,8 @@ PREP(handleSearchButton);
 PREP(handleSearchClick);
 PREP(handleSearchKeyDown);
 PREP(handleSearchKeyUp);
+PREP(handleSelectionChanged);
 PREP(handleSideButtons);
 PREP(handleTreeButtons);
 PREP(handleUnload);
+PREP(toggleSelectionPreview);

addons/editor/XEH_preInit.sqf

@@ -9,13 +9,33 @@ PREP_RECOMPILE_END;
 #include "initSettings.sqf"
 #include "initKeybinds.sqf"
 
+GVAR(lastSelection) = [];
+GVAR(savedSelection) = [];
+GVAR(lastModuleIcon) = "";
+GVAR(savedModuleIcon) = "";
+GVAR(colour) = ["IGUI", "TEXT_RGB"] call BIS_fnc_displayColorGet;
 GVAR(clipboard) = [];
 GVAR(includeCrew) = true;
 
 ["ModuleCurator_F", "Init", {
     params ["_logic"];
 
     _logic addEventHandler ["CuratorObjectPlaced", {call FUNC(handleObjectPlaced)}];
+    _logic addEventHandler ["CuratorGroupSelectionChanged", {call FUNC(handleSelectionChanged)}];
+    _logic addEventHandler ["CuratorMarkerSelectionChanged", {call FUNC(handleSelectionChanged)}];
+    _logic addEventHandler ["CuratorObjectSelectionChanged", {call FUNC(handleSelectionChanged)}];
+    _logic addEventHandler ["CuratorWaypointSelectionChanged", {call FUNC(handleSelectionChanged)}];
 }, true, [], true] call CBA_fnc_addClassEventHandler;
 
+[QGVAR(ModuleSelChanged), {
+    params ["_moduleName"];
+
+    GVAR(savedModuleIcon) = GVAR(lastModuleIcon);
+    GVAR(lastModuleIcon) = getText (configFile >> "CfgVehicles" >> _moduleName >> "icon");
+
+    // Toggle selection preview
+    private _hasSelectionPreview = (getNumber (configFile >> "CfgVehicles" >> _moduleName >> QGVAR(hasSelectionPreview)) isEqualTo 1);
+    [_hasSelectionPreview] call FUNC(toggleSelectionPreview);
+}] call CBA_fnc_addEventHandler;
+
 ADDON = true;

addons/editor/functions/fnc_getSelection.sqf

@@ -0,0 +1,54 @@
+#include "script_component.hpp"
+/*
+ * Author: Ampersand, Kex
+ * Retrieves and marks the entities a module is applied to.
+ * This function is meant to be used inside module functions only.
+ *
+ * Arguments:
+ * None
+ *
+ * Return Value:
+ * 0: List of selected objects <ARRAY>
+ * 1: List of selected groups <ARRAY>
+ * 2: List of selected waypoints <ARRAY>
+ * 3: List of selected markers <ARRAY>
+ *
+ * Example:
+ * (call zen_editor_fnc_getSelection) params ["_objects", "_groups"];
+ *
+ * Public: No
+ */
+
+BIS_fnc_curatorObjectPlaced_mouseOver params [["_entityType", ""], ["_entity", nil]];
+
+// Retrieve selection
+GVAR(lastSelection) = switch (_entityType) do {
+    case "OBJECT": {
+        [[_entity], [], [], []];
+    };
+    case "GROUP": {
+        [[], [_entity], [], []];
+    };
+    case "ARRAY": {
+        [[], [], [_entity], []];
+    };
+    case "STRING": {
+        [[], [], [], [_entity]];
+    };
+    default {
+        GVAR(savedSelection);
+    };
+};
+
+// Turn on selection preview
+GVAR(lastModuleIcon) = GVAR(savedModuleIcon);
+[true] call FUNC(toggleSelectionPreview);
+
+// Turn off selection preview when dialog is colosed
+GVAR(endSelectionHandle) = [QEGVAR(dialog,close), {
+    [false] call FUNC(toggleSelectionPreview);
+    [QEGVAR(dialog,close), GVAR(endSelectionHandle)] call CBA_fnc_removeEventHandler;
+    GVAR(endSelectionHandle) = nil;
+}] call CBA_fnc_addEventHandler;
+
+GVAR(lastSelection)

addons/editor/functions/fnc_handleLoad.sqf

@@ -98,15 +98,30 @@ _display displayAddEventHandler ["KeyDown", {call FUNC(handleKeyDown)}];
     IDC_RSCDISPLAYCURATOR_MAINMAP
 ];
 
+// Add tree selection changed events
+private _ctrlTreeModule = _display displayCtrl IDC_RSCDISPLAYCURATOR_CREATE_MODULES;
+_ctrlTreeModule ctrlAddEventHandler ["TreeSelChanged", {
+    params ["_ctrlTreeModule", "_selectedPath"];
+
+    [QGVAR(ModuleSelChanged), _ctrlTreeModule tvData _selectedPath] call CBA_fnc_localEvent;
+}];
+
 private _ctrlTreeRecent = _display displayCtrl IDC_RSCDISPLAYCURATOR_CREATE_RECENT;
 _ctrlTreeRecent ctrlAddEventHandler ["TreeSelChanged", {
     params ["_ctrlTreeRecent", "_selectedPath"];
+    private _recentTreeData = _ctrlTreeRecent tvData _selectedPath;
 
     // Store data of selected item to allow for deleting the of crew of objects placed through the recent tree
-    // tvCurSel is unavailable once the selected item has been placed, the empty path check ensures that the
+    // tvCurSel is unavailable once the selected item has been placed, the empty string check ensures that the
     // data is not cleared since this event occurs before the object placed event
-    if !(_selectedPath isEqualTo []) then {
-        GVAR(recentTreeData) = _ctrlTreeRecent tvData _selectedPath;
+    if !(_recentTreeData isEqualTo "") then {
+        GVAR(recentTreeData) = _recentTreeData;
+    };
+
+    if (_recentTreeData isKindOf "Module_F") then {
+        [QGVAR(ModuleSelChanged), _recentTreeData] call CBA_fnc_localEvent;
+    } else {
+        [QGVAR(ModuleSelChanged), ""] call CBA_fnc_localEvent;
     };
 }];
 

addons/editor/functions/fnc_handleSelectionChanged.sqf

@@ -0,0 +1,23 @@
+#include "script_component.hpp"
+/*
+ * Author: Ampersand
+ * Saves curator selected entities for use by modules.
+ *
+ * Arguments:
+ * 0: Curator <OBJECT>
+ * 0: Entity: group, marker, object, or waypoint <GROUP, STRING, OBJECT, ARRAY>
+ *
+ * Return Value:
+ * Handled <BOOL>
+ *
+ * Example:
+ * [CONTROL] call zen_editor_fnc_handleSelectionChanged
+ *
+ * Public: No
+ */
+
+params ["_curator", "_entity"];
+GVAR(savedSelection) = GVAR(lastSelection);
+GVAR(lastSelection) = curatorSelected;
+
+false

addons/editor/functions/fnc_toggleSelectionPreview.sqf

@@ -0,0 +1,52 @@
+#include "script_component.hpp"
+/*
+ * Author: Ampersand, Kex
+ * Toggle preview of the current selection
+ *
+ * Arguments:
+ * 0: Turn on preview <BOOL>
+ *
+ * Return Value:
+ * None
+ *
+ * Example:
+ * [true] call zen_editor_fnc_toggleSelectionPreview;
+ *
+ * Public: No
+ */
+
+params [["_turnOn", true, [true]]];
+
+if (_turnOn) then {
+    if (isNil QGVAR(selectionIconHandler)) then {
+        GVAR(selectionIconHandler) = [{
+            // Show preview if mouse is not over an entity
+            if (curatorMouseOver isEqualTo [""]) then {
+                if (isNil QGVAR(drawSelectioIcons)) then {
+                    GVAR(drawSelectioIcons) = addMissionEventHandler ["Draw3D", {
+                        {
+                            drawIcon3D [
+                                GVAR(lastModuleIcon),
+                                GVAR(colour), getPos _x, 1, 1, 0
+                            ];
+                        } forEach (GVAR(lastSelection) select 0);
+                    }];
+                };
+            } else {
+                if !(isNil QGVAR(drawSelectioIcons)) then {
+                    removeMissionEventHandler ["Draw3D", GVAR(drawSelectioIcons)];
+                    GVAR(drawSelectioIcons) = nil
+                };
+            };
+        }, 0] call CBA_fnc_addPerFrameHandler;
+    };
+} else {
+    if !(isNil QGVAR(selectionIconHandler)) then {
+        [GVAR(selectionIconHandler)] call CBA_fnc_removePerFrameHandler;
+        GVAR(selectionIconHandler) = nil;
+    };
+    if !(isNil QGVAR(drawSelectioIcons)) then {
+        removeMissionEventHandler ["Draw3D", GVAR(drawSelectioIcons)];
+        GVAR(drawSelectioIcons) = nil;
+    };
+};

addons/modules/CfgVehicles.hpp

@@ -422,6 +422,15 @@ class CfgVehicles {
         function = QFUNC(moduleToggleLamps);
         icon = QPATHTOF(ui\street_lamp_ca.paa);
     };
+    class GVAR(moduleTransferOwnership): GVAR(moduleBase) {
+        curatorCanAttach = 1;
+        category = QGVAR(DevTools);
+        displayName = CSTRING(ModuleTransferOwnership);
+        function = QFUNC(moduleTransferOwnership);
+        icon = "\a3\ui_f\data\Map\VehicleIcons\iconVirtual_ca.paa";
+        portrait = "\a3\ui_f\data\Map\VehicleIcons\iconVirtual_ca.paa";
+        EGVAR(editor,hasSelectionPreview) = 1;
+    };
     class GVAR(moduleTurretOptics): GVAR(moduleBase) {
         curatorCanAttach = 1;
         category = QGVAR(Equipment);

addons/modules/XEH_PREP.hpp

@@ -80,6 +80,7 @@ PREP(moduleTeleportPlayers);
 PREP(moduleToggleFlashlights);
 PREP(moduleToggleIRLasers);
 PREP(moduleToggleLamps);
+PREP(moduleTransferOwnership);
 PREP(moduleTurretOptics);
 PREP(moduleUnGarrison);
 PREP(moduleVisibility);

addons/modules/config.cpp

@@ -65,6 +65,7 @@ class CfgPatches {
             QGVAR(moduleToggleFlashlights),
             QGVAR(moduleToggleIRLasers),
             QGVAR(moduleToggleLamps),
+            QGVAR(ModuleTransferOwnership),
             QGVAR(moduleTurretOptics),
             QGVAR(moduleUnGarrison),
             QGVAR(moduleVisibility),