feat: slightly better serializer for unit actions and state, added attack action too

megamek/src/megamek/ai/dataset/TsvSerde.java → megamek/src/megamek/ai/dataset/TabSeparatedValueSerializer.java

@@ -1,8 +1,9 @@
 /*
- * Copyright (C) 2017-2025 The MegaMek Team. All Rights Reserved.
+ * Copyright (C) 2025 The MegaMek Team. All Rights Reserved.
  *
  * This file is part of MegaMek.
  *
+ *
  * MegaMek is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License (GPL),
  * version 3 or (at your option) any later version,
@@ -27,24 +28,28 @@
  */
 package megamek.ai.dataset;
 
+import java.text.DecimalFormat;
+import java.text.DecimalFormatSymbols;
+import java.util.Locale;
+
 /**
- * <p>Abstract class to serialize/deserialize objects to/from TSV format.</p>
- * <p>It does not have a fromTsv function because I could not find a way to make a good API for it.</p>
- * @param <T> type of object to serialize/deserialize
+ * Abstract class to serialize objects to TSV format.
  * @author Luana Coppio
  */
-public abstract class TsvSerde<T> {
+public abstract class TabSeparatedValueSerializer<T> {
+
+    protected final DecimalFormat LOG_DECIMAL = new DecimalFormat("0.00", new DecimalFormatSymbols(Locale.US));
 
     /**
      * Serializes an object to TSV format.
      * @param obj object to serialize
-     * @return the object serialized in TSV format
+     * @return the serialized string formatted as a tab separated value entry
      */
-    public abstract String toTsv(T obj);
+    public abstract String serialize(T obj);
 
     /**
      * Returns the header line for the TSV format.
-     * @return the header line
+     * @return the header line for the tab separated value entry
      */
     public abstract String getHeaderLine();
 }

megamek/src/megamek/ai/dataset/UnitAction.java

@@ -1,8 +1,9 @@
 /*
- * Copyright (C) 2017-2025 The MegaMek Team. All Rights Reserved.
+ * Copyright (C) 2025 The MegaMek Team. All Rights Reserved.
  *
  * This file is part of MegaMek.
  *
+ *
  * MegaMek is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License (GPL),
  * version 3 or (at your option) any later version,
@@ -27,80 +28,86 @@
  */
 package megamek.ai.dataset;
 
+import java.util.List;
+import java.util.stream.Stream;
+
+import megamek.ai.utility.EntityFeatureUtils;
 import megamek.client.ui.SharedUtility;
-import megamek.common.Coords;
+import megamek.common.Compute;
 import megamek.common.Entity;
+import megamek.common.IAero;
 import megamek.common.MovePath;
 import megamek.common.MoveStep;
-
-import java.util.List;
+import megamek.common.UnitRole;
 
 /**
- * Represents a unit action.
- * @param id unit id
- * @param facing facing direction
- * @param fromX x coordinate of the starting position
- * @param fromY y coordinate of the starting position
- * @param toX x coordinate of the final position
- * @param toY y coordinate of the final position
- * @param hexesMoved number of hexes moved
- * @param distance distance moved
- * @param mpUsed movement points used
- * @param maxMp maximum movement points
- * @param mpP movement points percentage
- * @param heatP heat percentage
- * @param armorP armor percentage
- * @param internalP internal percentage
- * @param jumping jumping
- * @param prone prone
- * @param legal is move legal
+ * UnitAction is a record that represents the action of a unit in the game.
  * @author Luana Coppio
  */
-public record UnitAction(int id, int teamId, int playerId, String chassis, String model, int facing, int fromX, int fromY, int toX, int toY, int hexesMoved, int distance, int mpUsed,
-                         int maxMp, double mpP, double heatP, double armorP, double internalP, boolean jumping, boolean prone,
-                         boolean legal, double chanceOfFailure, List<MovePath.MoveStepType> steps, boolean bot) {
+public record UnitAction(int id, int teamId, int playerId, String chassis, String model, int facing,
+      int fromX, int fromY, int toX, int toY, int hexesMoved, int distance,
+      int mpUsed, int maxMp, double mpP, double heatP, double armorP, double internalP,
+      boolean jumping, boolean prone, boolean legal, double chanceOfFailure,
+      List<MovePath.MoveStepType> steps, boolean bot,
+      boolean hasEcm, int armor, int internal, int bv, int maxRange, int totalDamage,
+      float armorFrontP, float armorLeftP, float armorRightP, float armorBackP, UnitRole role,
+      List<Integer> weaponDamageFacingShortMediumLongRange) {
 
-    public static  UnitAction fromMovePath(MovePath movePath) {
+    /**
+     * Creates a UnitAction from a MovePath.
+     * @param movePath The MovePath to convert
+     * @return The UnitAction
+     */
+    public static UnitAction fromMovePath(MovePath movePath) {
         Entity entity = movePath.getEntity();
-        double chanceOfFailure = SharedUtility.getPSRList(movePath).stream().map(psr -> psr.getValue() / 36d).reduce(0.0, (a, b) -> a * b);
-        var steps = movePath.getStepVector().stream().map(MoveStep::getType).toList();
+
         return new UnitAction(
-            entity.getId(),
-            entity.getOwner() != null ? entity.getOwner().getTeam() : -1,
-            entity.getOwner() != null ? entity.getOwner().getId() : -1,
-            entity.getChassis(),
-            entity.getModel(),
-            movePath.getFinalFacing(),
-            movePath.getStartCoords() != null ? movePath.getStartCoords().getX() : -1,
-            movePath.getStartCoords() != null ? movePath.getStartCoords().getY() : -1,
-            movePath.getFinalCoords() != null ? movePath.getFinalCoords().getX() : -1,
-            movePath.getFinalCoords() != null ? movePath.getFinalCoords().getY() : -1,
-            movePath.getHexesMoved(),
-            movePath.getDistanceTravelled(),
-            movePath.getMpUsed(),
-            movePath.getMaxMP(),
-            movePath.getMaxMP() > 0 ? (double) movePath.getMpUsed() / movePath.getMaxMP() : 0.0,
-            entity.getHeatCapacity() > 0 ? entity.getHeat() / (double) entity.getHeatCapacity() : 0.0,
-            entity.getArmorRemainingPercent(),
-            entity.getInternalRemainingPercent(),
-            movePath.isJumping(),
-            movePath.getFinalProne(),
-            movePath.isMoveLegal(),
-            chanceOfFailure,
-            steps,
-            entity.getOwner().isBot()
+              entity.getId(),
+              entity.getOwner() != null ? entity.getOwner().getTeam() : -1,
+              entity.getOwner() != null ? entity.getOwner().getId() : -1,
+              entity.getChassis(),
+              entity.getModel(),
+              movePath.getFinalFacing(),
+              movePath.getStartCoords() != null ? movePath.getStartCoords().getX() : -1,
+              movePath.getStartCoords() != null ? movePath.getStartCoords().getY() : -1,
+              movePath.getFinalCoords() != null ? movePath.getFinalCoords().getX() : -1,
+              movePath.getFinalCoords() != null ? movePath.getFinalCoords().getY() : -1,
+              movePath.getHexesMoved(),
+              movePath.getDistanceTravelled(),
+              movePath.getMpUsed(),
+              movePath.getMaxMP(),
+              movePath.getMaxMP() > 0 ? (double) movePath.getMpUsed() / movePath.getMaxMP() : 0.0,
+              entity.getHeatCapacity() > 0 ? entity.getHeat() / (double) entity.getHeatCapacity() : 0.0,
+              entity.getArmorRemainingPercent(),
+              entity.getInternalRemainingPercent(),
+              movePath.isJumping(),
+              movePath.getFinalProne(),
+              movePath.isMoveLegal(),
+              SharedUtility.getPSRList(movePath).stream()
+                    .map(psr -> psr.getValue() / 36d)
+                    .reduce(1.0, (a, b) -> a * b),
+              movePath.getStepVector().stream().map(MoveStep::getType).toList(),
+              entity.getOwner().isBot(),
+              entity.hasActiveECM(),
+              entity.getTotalArmor(),
+              entity instanceof IAero aero ? aero.getSI() : entity.getTotalInternal(),
+              entity.getInitialBV(),
+              entity.getMaxWeaponRange(),
+              Compute.computeTotalDamage(entity.getWeaponList()),
+              EntityFeatureUtils.getTargetFrontHealthStats(entity),
+              EntityFeatureUtils.getTargetLeftSideHealthStats(entity),
+              EntityFeatureUtils.getTargetRightSideHealthStats(entity),
+              EntityFeatureUtils.getTargetBackHealthStats(entity),
+              entity.getRole(),
+              entity.getWeaponList().stream().flatMap(weapon -> {
+                  int damage = Compute.computeTotalDamage(weapon);
+                  int facing = weapon.isRearMounted() ? -entity.getWeaponArc(weapon.getLocation()) :
+                                     entity.getWeaponArc(weapon.getLocation());
+                  int shortRange = weapon.getType().getShortRange();
+                  int mediumRange = weapon.getType().getMediumRange();
+                  int longRange = weapon.getType().getLongRange();
+                  return Stream.of(damage, facing, shortRange, mediumRange, longRange);
+              }).toList()
         );
     }
-
-    public Coords currentPosition() {
-        return new Coords(fromX, fromY);
-    }
-
-    public boolean isHuman() {
-        return !bot;
-    }
-
-    public Coords finalPosition() {
-        return new Coords(toX, toY);
-    }
 }