fixed balers stats

Royal-Modding · Mar 15, 2021 · 8e72ee1 · 8e72ee1
1 parent 44a64c9
commit 8e72ee1
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/src/extendedSpecializations/BaleWrapper.lua b/src/extendedSpecializations/BaleWrapper.lua
@@ -38,9 +38,9 @@ function ExtendedBaleWrapper:onLoadStats()
     end
 end
 
-function ExtendedBaleWrapper:pickupWrapperBale(superFunc, ...)
-    superFunc(self, ...)
-    if self.isServer then
+function ExtendedBaleWrapper:pickupWrapperBale(superFunc, bale, baleType, ...)
+    superFunc(self, bale, baleType, ...)
+    if self.isServer and baleType ~= nil and bale.i3dFilename ~= baleType.wrapperBaleFilename then
         local spec = self:getAdvancedStatsSpecTable(ExtendedBaleWrapper.SPEC_TABLE_NAME)
         self:updateStat(spec.advancedStatistics["WrappedBales"], 1)
     end

diff --git a/src/extendedSpecializations/Baler.lua b/src/extendedSpecializations/Baler.lua
@@ -49,7 +49,7 @@ function ExtendedBaler:dropBale(superFunc, ...)
 end
 
 function ExtendedBaler:onFillUnitFillLevelChanged(fillUnitIndex, fillLevelDelta, fillTypeIndex, toolType, fillPositionData, appliedDelta)
-    if self.isServer and appliedDelta > 0 then
+    if self.isServer and appliedDelta > 0 and fillUnitIndex == 1 then
         local spec = self:getAdvancedStatsSpecTable(ExtendedBaler.SPEC_TABLE_NAME)
         self:updateStat(spec.advancedStatistics["LoadedLiters"], appliedDelta)
     end

diff --git a/src/extendedSpecializations/Tedder.lua b/src/extendedSpecializations/Tedder.lua
@@ -34,7 +34,7 @@ function ExtendedTedder:onLoadStats()
             spec.advancedStatisticsPrefix,
             {
                 {"WorkedLitres", AdvancedStats.UNITS.LITRE, true},
-                {"WorkedHectares", AdvancedStats.UNITS.HECTARE}
+                {"WorkedHectares", AdvancedStats.UNITS.HECTARE, true}
             }
         )
     end

diff --git a/src/extendedSpecializations/Windrower.lua b/src/extendedSpecializations/Windrower.lua
@@ -33,7 +33,7 @@ function ExtendedWindrower:onLoadStats()
             spec.advancedStatisticsPrefix,
             {
                 {"WorkedLitres", AdvancedStats.UNITS.LITRE, true},
-                {"WorkedHectares", AdvancedStats.UNITS.HECTARE}
+                {"WorkedHectares", AdvancedStats.UNITS.HECTARE, true}
             }
         )
     end