106 lines
4.5 KiB
C#
106 lines
4.5 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Reflection;
|
|
using Gamecraft.Wires;
|
|
using GamecraftModdingAPI.Engines;
|
|
using HarmonyLib;
|
|
using RobocraftX.Blocks;
|
|
using RobocraftX.Character;
|
|
using RobocraftX.Common;
|
|
using RobocraftX.Common.Players;
|
|
using Svelto.DataStructures;
|
|
using Svelto.ECS;
|
|
|
|
namespace GamecraftModdingAPI.Blocks
|
|
{
|
|
public class BlockCloneEngine : IApiEngine
|
|
{
|
|
private static Type copyEngineType =
|
|
AccessTools.TypeByName("Gamecraft.GUI.Tweaks.Engines.CopyTweaksOnPickEngine");
|
|
private static Type copyWireEngineType =
|
|
AccessTools.TypeByName("Gamecraft.Wires.WireConnectionCopyOnPickEngine");
|
|
private static Type createWireEngineType =
|
|
AccessTools.TypeByName("RobocraftX.GUI.Wires.WireConnectionCreateOnPlaceEngine");
|
|
|
|
private MethodBase copyFromBlock = AccessTools.Method(copyEngineType, "CopyTweaksFromBlock");
|
|
private MethodBase copyToBlock = AccessTools.Method(copyEngineType, "ApplyTweaksToPlacedBlock");
|
|
private MethodBase copyWireFromBlock = AccessTools.Method(copyWireEngineType, "CopyWireInputsAndOutputs");
|
|
private MethodBase copyWireToBlock = AccessTools.Method(createWireEngineType, "PlaceWiresOnPlaceNewCube");
|
|
|
|
public void Ready()
|
|
{
|
|
}
|
|
|
|
public EntitiesDB entitiesDB { get; set; }
|
|
|
|
public void Dispose()
|
|
{
|
|
}
|
|
|
|
public void CopyBlockStats(EGID sourceID, EGID targetID)
|
|
{
|
|
var allCharacters = (LocalFasterReadOnlyList<ExclusiveGroupStruct>) CharacterExclusiveGroups.AllCharacters;
|
|
foreach (var ((pickedBlockColl, count), _) in entitiesDB.QueryEntities<PickedBlockExtraDataStruct>(allCharacters))
|
|
{
|
|
for (int i = 0; i < count; ++i)
|
|
{
|
|
ref PickedBlockExtraDataStruct pickedBlock = ref pickedBlockColl[i];
|
|
var oldStruct = pickedBlock;
|
|
pickedBlock.pickedBlockEntityID = sourceID;
|
|
pickedBlock.placedBlockEntityID = targetID;
|
|
pickedBlock.placedBlockTweaksMustCopy = true;
|
|
if (entitiesDB.Exists<DBEntityStruct>(pickedBlock.pickedBlockEntityID)
|
|
&& entitiesDB.Exists<DBEntityStruct>(pickedBlock.placedBlockEntityID))
|
|
{
|
|
copyFromBlock.Invoke(Patch.copyEngine, new object[] {pickedBlock.ID, pickedBlock});
|
|
|
|
uint playerID = Player.LocalPlayer.Id;
|
|
var parameters = new object[] {playerID, pickedBlock};
|
|
copyWireFromBlock.Invoke(Patch.copyWireEngine, parameters);
|
|
pickedBlock = (PickedBlockExtraDataStruct) parameters[1]; //ref arg
|
|
|
|
copyToBlock.Invoke(Patch.copyEngine, new object[] {pickedBlock.ID, pickedBlock});
|
|
|
|
ExclusiveGroupStruct group = WiresExclusiveGroups.WIRES_COPY_GROUP + playerID;
|
|
copyWireToBlock.Invoke(Patch.createWireEngine, new object[] {group, pickedBlock.ID});
|
|
|
|
pickedBlock.placedBlockTweaksMustCopy = false;
|
|
}
|
|
|
|
pickedBlock = oldStruct; //Make sure to not interfere with the game - Although that might not be the case with the wire copying
|
|
}
|
|
}
|
|
}
|
|
|
|
[HarmonyPatch]
|
|
private static class Patch
|
|
{
|
|
public static object copyEngine;
|
|
public static object copyWireEngine;
|
|
public static object createWireEngine;
|
|
|
|
public static void Postfix(object __instance)
|
|
{
|
|
if (__instance.GetType() == copyEngineType)
|
|
copyEngine = __instance;
|
|
else if (__instance.GetType() == copyWireEngineType)
|
|
copyWireEngine = __instance;
|
|
else if (__instance.GetType() == createWireEngineType)
|
|
createWireEngine = __instance;
|
|
}
|
|
|
|
public static IEnumerable<MethodBase> TargetMethods()
|
|
{
|
|
return new[]
|
|
{
|
|
AccessTools.GetDeclaredConstructors(copyEngineType)[0],
|
|
AccessTools.GetDeclaredConstructors(copyWireEngineType)[0],
|
|
AccessTools.GetDeclaredConstructors(createWireEngineType)[0]
|
|
};
|
|
}
|
|
}
|
|
|
|
public string Name { get; } = "GamecraftModdingAPIBlockCloneGameEngine";
|
|
public bool isRemovable { get; } = false;
|
|
}
|
|
} |