Merge pull request #72 from Antosik/feat/neutral-token

feat: add neutral token drops data
odota · Aug 25, 2024 · 9a39bf2 · 9a39bf2
2 parents 385b818 + cac8c41
commit 9a39bf2
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 2 deletions.
diff --git a/processors/parseSchema.mjs b/processors/parseSchema.mjs
@@ -66,5 +66,6 @@ export default {
     randomed: false,
     repicked: false,
     pred_vict: false,
+    neutral_tokens_log: [],
   })),
 };
diff --git a/processors/populate.mjs b/processors/populate.mjs
@@ -66,7 +66,8 @@ function populate(e, container, meta) {
         } else if (
           e.type === 'purchase_log' ||
           e.type === 'kills_log' ||
-          e.type === 'runes_log'
+          e.type === 'runes_log' ||
+          e.type === 'neutral_tokens_log'
         ) {
           arrEntry = {
             time: e.time,

diff --git a/processors/processExpand.mjs b/processors/processExpand.mjs
@@ -630,6 +630,12 @@ function processExpand(entries, meta) {
     cosmetics(e) {
       expand(e);
     },
+    neutral_token(e) {
+      expand({
+        ...e,
+        type: 'neutral_tokens_log',
+      });
+    },
   };
   for (let i = 0; i < entries.length; i += 1) {
     const e = entries[i];

diff --git a/src/main/java/opendota/Parse.java b/src/main/java/opendota/Parse.java
@@ -275,6 +275,9 @@ public Integer getPlayerSlotFromEntity(Context ctx, Entity e) {
         if (slot == null) {
             slot = getEntityProperty(e, "m_iPlayerID", null);
         }
+        if (slot == null) {
+            slot = getEntityProperty(e, "m_iPlayerOwnerID", null);
+        }
         if (slot != null) {
             slot /= 2;
         }
@@ -473,7 +476,9 @@ public void onCombatLogEntry(Context ctx, CombatLogEntry cle) {
 
     @OnEntityEntered
     public void onEntityEntered(Context ctx, Entity e) {
-        if (e.getDtClass().getDtName().equals("CDOTAWearableItem")) {
+        String entityName = e.getDtClass().getDtName();
+
+        if (entityName.equals("CDOTAWearableItem")) {
             Integer accountId = getEntityProperty(e, "m_iAccountID", null);
             Integer itemDefinitionIndex = getEntityProperty(e, "m_iItemDefinitionIndex", null);
             // System.err.format("%s,%s\n", accountId, itemDefinitionIndex);
@@ -483,6 +488,12 @@ public void onEntityEntered(Context ctx, Entity e) {
                 Integer playerSlot = steamid_to_playerslot.get(accountId64);
                 cosmeticsMap.put(itemDefinitionIndex, playerSlot);
             }
+        } else if (entityName.startsWith("CDOTA_Item_Tier") && entityName.endsWith("Token")) {
+            Entry entry = new Entry(time);
+            entry.type = "neutral_token";
+            entry.slot = getPlayerSlotFromEntity(ctx, e);
+            entry.key = entityName.substring("CDOTA_Item_".length()); // Tier1Token
+            output(entry);
         }
     }