sc2ai 0.4.3 → 0.5.0

data/lib/sc2ai/protocol/score_pb.rb

@@ -17,23 +17,15 @@ module Api
       MELEE = 2
 
       def self.lookup(val)
-        if val == 0
-          :ENUM_SCORE_TYPE_UNSET
-        elsif val == 1
-          :CURRICULUM
-        elsif val == 2
-          :MELEE
-        end
+        return :ENUM_SCORE_TYPE_UNSET if val == 0
+        return :CURRICULUM if val == 1
+        return :MELEE if val == 2
       end
 
       def self.resolve(val)
-        if val == :ENUM_SCORE_TYPE_UNSET
-          0
-        elsif val == :CURRICULUM
-          1
-        elsif val == :MELEE
-          2
-        end
+        return 0 if val == :ENUM_SCORE_TYPE_UNSET
+        return 1 if val == :CURRICULUM
+        return 2 if val == :MELEE
       end
     end
     # required field readers
@@ -51,6 +43,7 @@ module Api
     # enum writers
     def score_type=(v)
       @score_type = Api::Score::ScoreType.resolve(v) || v
+      @_bitmask |= 0x0000000000000001
     end
 
     # BEGIN writers for optional fields
@@ -104,6 +97,10 @@ module Api
       self.class.encode(self)
     end
 
+    def has_score_type?
+      (@_bitmask & 0x0000000000000001) == 0x0000000000000001
+    end
+
     def has_score?
       (@_bitmask & 0x0000000000000002) == 0x0000000000000002
     end
@@ -181,7 +178,7 @@ module Api
 
           unknown_bytes = +"".b
           val = tag
-          while val != 0
+          loop do
             byte = val & 0x7F
 
             val >>= 7
@@ -192,6 +189,7 @@ module Api
 
             byte |= 0x80 if val != 0
             unknown_bytes << byte
+            break if val == 0
           end
 
           case wire_type
@@ -266,7 +264,7 @@ module Api
               end
 
             val = value
-            while val != 0
+            loop do
               byte = val & 0x7F
 
               val >>= 7
@@ -277,6 +275,7 @@ module Api
 
               byte |= 0x80 if val != 0
               unknown_bytes << byte
+              break if val == 0
             end
 
             unknown_bytes << buff.byteslice(index, value)
@@ -706,10 +705,10 @@ module Api
     end
     def _encode(buff)
       val = @score_type
-      if val != 0
+      if has_score_type?
         buff << 0x30
 
-        while val != 0
+        loop do
           byte = val & 0x7F
 
           val >>= 7
@@ -720,14 +719,15 @@ module Api
 
           byte |= 0x80 if val != 0
           buff << byte
+          break if val == 0
         end
       end
 
       val = @score
-      if val != 0
+      if has_score?
         buff << 0x38
 
-        while val != 0
+        loop do
           byte = val & 0x7F
 
           val >>= 7
@@ -738,6 +738,7 @@ module Api
 
           byte |= 0x80 if val != 0
           buff << byte
+          break if val == 0
         end
       end
 
@@ -794,11 +795,34 @@ module Api
 
     def to_h
       result = {}
-      result["score_type".to_sym] = @score_type
-      result["score".to_sym] = @score
-      result["score_details".to_sym] = @score_details.to_h
+
+      result[:"score_type"] = @score_type
+      result[:"score"] = @score
+      result[:"score_details"] = @score_details.to_h
+
       result
     end
+
+    def as_json(options = {})
+      result = {}
+
+      result["scoreType"] = @score_type if !options[:compact] || has_score_type?
+      result["score"] = @score if !options[:compact] || has_score?
+      result["scoreDetails"] = (
+        if @score_details.nil?
+          {}
+        else
+          @score_details.as_json(options)
+        end
+      ) if !options[:compact] || has_score_details?
+
+      result
+    end
+
+    def to_json(as_json_options = {})
+      require "json"
+      JSON.dump(as_json(as_json_options))
+    end
   end
   class CategoryScoreDetails
     def self.decode(buff)
@@ -990,7 +1014,7 @@ module Api
 
           unknown_bytes = +"".b
           val = tag
-          while val != 0
+          loop do
             byte = val & 0x7F
 
             val >>= 7
@@ -1001,6 +1025,7 @@ module Api
 
             byte |= 0x80 if val != 0
             unknown_bytes << byte
+            break if val == 0
           end
 
           case wire_type
@@ -1075,7 +1100,7 @@ module Api
               end
 
             val = value
-            while val != 0
+            loop do
               byte = val & 0x7F
 
               val >>= 7
@@ -1086,6 +1111,7 @@ module Api
 
               byte |= 0x80 if val != 0
               unknown_bytes << byte
+              break if val == 0
             end
 
             unknown_bytes << buff.byteslice(index, value)
@@ -1448,35 +1474,35 @@ module Api
     end
     def _encode(buff)
       val = @none
-      if val != 0
+      if has_none?
         buff << 0x0d
 
         [val].pack("e", buffer: buff)
       end
 
       val = @army
-      if val != 0
+      if has_army?
         buff << 0x15
 
         [val].pack("e", buffer: buff)
       end
 
       val = @economy
-      if val != 0
+      if has_economy?
         buff << 0x1d
 
         [val].pack("e", buffer: buff)
       end
 
       val = @technology
-      if val != 0
+      if has_technology?
         buff << 0x25
 
         [val].pack("e", buffer: buff)
       end
 
       val = @upgrade
-      if val != 0
+      if has_upgrade?
         buff << 0x2d
 
         [val].pack("e", buffer: buff)
@@ -1487,13 +1513,33 @@ module Api
 
     def to_h
       result = {}
-      result["none".to_sym] = @none
-      result["army".to_sym] = @army
-      result["economy".to_sym] = @economy
-      result["technology".to_sym] = @technology
-      result["upgrade".to_sym] = @upgrade
+
+      result[:"none"] = @none
+      result[:"army"] = @army
+      result[:"economy"] = @economy
+      result[:"technology"] = @technology
+      result[:"upgrade"] = @upgrade
+
+      result
+    end
+
+    def as_json(options = {})
+      result = {}
+
+      result["none"] = @none if !options[:compact] || has_none?
+      result["army"] = @army if !options[:compact] || has_army?
+      result["economy"] = @economy if !options[:compact] || has_economy?
+      result["technology"] = @technology if !options[:compact] ||
+        has_technology?
+      result["upgrade"] = @upgrade if !options[:compact] || has_upgrade?
+
       result
     end
+
+    def to_json(as_json_options = {})
+      require "json"
+      JSON.dump(as_json(as_json_options))
+    end
   end
   class VitalScoreDetails
     def self.decode(buff)
@@ -1641,7 +1687,7 @@ module Api
 
           unknown_bytes = +"".b
           val = tag
-          while val != 0
+          loop do
             byte = val & 0x7F
 
             val >>= 7
@@ -1652,6 +1698,7 @@ module Api
 
             byte |= 0x80 if val != 0
             unknown_bytes << byte
+            break if val == 0
           end
 
           case wire_type
@@ -1726,7 +1773,7 @@ module Api
               end
 
             val = value
-            while val != 0
+            loop do
               byte = val & 0x7F
 
               val >>= 7
@@ -1737,6 +1784,7 @@ module Api
 
               byte |= 0x80 if val != 0
               unknown_bytes << byte
+              break if val == 0
             end
 
             unknown_bytes << buff.byteslice(index, value)
@@ -1983,21 +2031,21 @@ module Api
     end
     def _encode(buff)
       val = @life
-      if val != 0
+      if has_life?
         buff << 0x0d
 
         [val].pack("e", buffer: buff)
       end
 
       val = @shields
-      if val != 0
+      if has_shields?
         buff << 0x15
 
         [val].pack("e", buffer: buff)
       end
 
       val = @energy
-      if val != 0
+      if has_energy?
         buff << 0x1d
 
         [val].pack("e", buffer: buff)
@@ -2008,11 +2056,28 @@ module Api
 
     def to_h
       result = {}
-      result["life".to_sym] = @life
-      result["shields".to_sym] = @shields
-      result["energy".to_sym] = @energy
+
+      result[:"life"] = @life
+      result[:"shields"] = @shields
+      result[:"energy"] = @energy
+
       result
     end
+
+    def as_json(options = {})
+      result = {}
+
+      result["life"] = @life if !options[:compact] || has_life?
+      result["shields"] = @shields if !options[:compact] || has_shields?
+      result["energy"] = @energy if !options[:compact] || has_energy?
+
+      result
+    end
+
+    def to_json(as_json_options = {})
+      require "json"
+      JSON.dump(as_json(as_json_options))
+    end
   end
   class ScoreDetails
     def self.decode(buff)
@@ -2664,7 +2729,7 @@ module Api
 
           unknown_bytes = +"".b
           val = tag
-          while val != 0
+          loop do
             byte = val & 0x7F
 
             val >>= 7
@@ -2675,6 +2740,7 @@ module Api
 
             byte |= 0x80 if val != 0
             unknown_bytes << byte
+            break if val == 0
           end
 
           case wire_type
@@ -2749,7 +2815,7 @@ module Api
               end
 
             val = value
-            while val != 0
+            loop do
               byte = val & 0x7F
 
               val >>= 7
@@ -2760,6 +2826,7 @@ module Api
 
               byte |= 0x80 if val != 0
               unknown_bytes << byte
+              break if val == 0
             end
 
             unknown_bytes << buff.byteslice(index, value)
@@ -5282,84 +5349,84 @@ module Api
     end
     def _encode(buff)
       val = @idle_production_time
-      if val != 0
+      if has_idle_production_time?
         buff << 0x0d
 
         [val].pack("e", buffer: buff)
       end
 
       val = @idle_worker_time
-      if val != 0
+      if has_idle_worker_time?
         buff << 0x15
 
         [val].pack("e", buffer: buff)
       end
 
       val = @total_value_units
-      if val != 0
+      if has_total_value_units?
         buff << 0x1d
 
         [val].pack("e", buffer: buff)
       end
 
       val = @total_value_structures
-      if val != 0
+      if has_total_value_structures?
         buff << 0x25
 
         [val].pack("e", buffer: buff)
       end
 
       val = @killed_value_units
-      if val != 0
+      if has_killed_value_units?
         buff << 0x2d
 
         [val].pack("e", buffer: buff)
       end
 
       val = @killed_value_structures
-      if val != 0
+      if has_killed_value_structures?
         buff << 0x35
 
         [val].pack("e", buffer: buff)
       end
 
       val = @collected_minerals
-      if val != 0
+      if has_collected_minerals?
         buff << 0x3d
 
         [val].pack("e", buffer: buff)
       end
 
       val = @collected_vespene
-      if val != 0
+      if has_collected_vespene?
         buff << 0x45
 
         [val].pack("e", buffer: buff)
       end
 
       val = @collection_rate_minerals
-      if val != 0
+      if has_collection_rate_minerals?
         buff << 0x4d
 
         [val].pack("e", buffer: buff)
       end
 
       val = @collection_rate_vespene
-      if val != 0
+      if has_collection_rate_vespene?
         buff << 0x55
 
         [val].pack("e", buffer: buff)
       end
 
       val = @spent_minerals
-      if val != 0
+      if has_spent_minerals?
         buff << 0x5d
 
         [val].pack("e", buffer: buff)
       end
 
       val = @spent_vespene
-      if val != 0
+      if has_spent_vespene?
         buff << 0x65
 
         [val].pack("e", buffer: buff)
@@ -6049,7 +6116,7 @@ module Api
       end
 
       val = @current_apm
-      if val != 0
+      if has_current_apm?
         buff << 0xdd
         buff << 0x01
 
@@ -6057,7 +6124,7 @@ module Api
       end
 
       val = @current_effective_apm
-      if val != 0
+      if has_current_effective_apm?
         buff << 0xe5
         buff << 0x01
 
@@ -6069,35 +6136,181 @@ module Api
 
     def to_h
       result = {}
-      result["idle_production_time".to_sym] = @idle_production_time
-      result["idle_worker_time".to_sym] = @idle_worker_time
-      result["total_value_units".to_sym] = @total_value_units
-      result["total_value_structures".to_sym] = @total_value_structures
-      result["killed_value_units".to_sym] = @killed_value_units
-      result["killed_value_structures".to_sym] = @killed_value_structures
-      result["collected_minerals".to_sym] = @collected_minerals
-      result["collected_vespene".to_sym] = @collected_vespene
-      result["collection_rate_minerals".to_sym] = @collection_rate_minerals
-      result["collection_rate_vespene".to_sym] = @collection_rate_vespene
-      result["spent_minerals".to_sym] = @spent_minerals
-      result["spent_vespene".to_sym] = @spent_vespene
-      result["food_used".to_sym] = @food_used.to_h
-      result["killed_minerals".to_sym] = @killed_minerals.to_h
-      result["killed_vespene".to_sym] = @killed_vespene.to_h
-      result["lost_minerals".to_sym] = @lost_minerals.to_h
-      result["lost_vespene".to_sym] = @lost_vespene.to_h
-      result["friendly_fire_minerals".to_sym] = @friendly_fire_minerals.to_h
-      result["friendly_fire_vespene".to_sym] = @friendly_fire_vespene.to_h
-      result["used_minerals".to_sym] = @used_minerals.to_h
-      result["used_vespene".to_sym] = @used_vespene.to_h
-      result["total_used_minerals".to_sym] = @total_used_minerals.to_h
-      result["total_used_vespene".to_sym] = @total_used_vespene.to_h
-      result["total_damage_dealt".to_sym] = @total_damage_dealt.to_h
-      result["total_damage_taken".to_sym] = @total_damage_taken.to_h
-      result["total_healed".to_sym] = @total_healed.to_h
-      result["current_apm".to_sym] = @current_apm
-      result["current_effective_apm".to_sym] = @current_effective_apm
+
+      result[:"idle_production_time"] = @idle_production_time
+      result[:"idle_worker_time"] = @idle_worker_time
+      result[:"total_value_units"] = @total_value_units
+      result[:"total_value_structures"] = @total_value_structures
+      result[:"killed_value_units"] = @killed_value_units
+      result[:"killed_value_structures"] = @killed_value_structures
+      result[:"collected_minerals"] = @collected_minerals
+      result[:"collected_vespene"] = @collected_vespene
+      result[:"collection_rate_minerals"] = @collection_rate_minerals
+      result[:"collection_rate_vespene"] = @collection_rate_vespene
+      result[:"spent_minerals"] = @spent_minerals
+      result[:"spent_vespene"] = @spent_vespene
+      result[:"food_used"] = @food_used.to_h
+      result[:"killed_minerals"] = @killed_minerals.to_h
+      result[:"killed_vespene"] = @killed_vespene.to_h
+      result[:"lost_minerals"] = @lost_minerals.to_h
+      result[:"lost_vespene"] = @lost_vespene.to_h
+      result[:"friendly_fire_minerals"] = @friendly_fire_minerals.to_h
+      result[:"friendly_fire_vespene"] = @friendly_fire_vespene.to_h
+      result[:"used_minerals"] = @used_minerals.to_h
+      result[:"used_vespene"] = @used_vespene.to_h
+      result[:"total_used_minerals"] = @total_used_minerals.to_h
+      result[:"total_used_vespene"] = @total_used_vespene.to_h
+      result[:"total_damage_dealt"] = @total_damage_dealt.to_h
+      result[:"total_damage_taken"] = @total_damage_taken.to_h
+      result[:"total_healed"] = @total_healed.to_h
+      result[:"current_apm"] = @current_apm
+      result[:"current_effective_apm"] = @current_effective_apm
+
+      result
+    end
+
+    def as_json(options = {})
+      result = {}
+
+      result["idleProductionTime"] = @idle_production_time if !options[
+        :compact
+      ] || has_idle_production_time?
+      result["idleWorkerTime"] = @idle_worker_time if !options[:compact] ||
+        has_idle_worker_time?
+      result["totalValueUnits"] = @total_value_units if !options[:compact] ||
+        has_total_value_units?
+      result["totalValueStructures"] = @total_value_structures if !options[
+        :compact
+      ] || has_total_value_structures?
+      result["killedValueUnits"] = @killed_value_units if !options[:compact] ||
+        has_killed_value_units?
+      result["killedValueStructures"] = @killed_value_structures if !options[
+        :compact
+      ] || has_killed_value_structures?
+      result["collectedMinerals"] = @collected_minerals if !options[:compact] ||
+        has_collected_minerals?
+      result["collectedVespene"] = @collected_vespene if !options[:compact] ||
+        has_collected_vespene?
+      result["collectionRateMinerals"] = @collection_rate_minerals if !options[
+        :compact
+      ] || has_collection_rate_minerals?
+      result["collectionRateVespene"] = @collection_rate_vespene if !options[
+        :compact
+      ] || has_collection_rate_vespene?
+      result["spentMinerals"] = @spent_minerals if !options[:compact] ||
+        has_spent_minerals?
+      result["spentVespene"] = @spent_vespene if !options[:compact] ||
+        has_spent_vespene?
+      result["foodUsed"] = (
+        if @food_used.nil?
+          {}
+        else
+          @food_used.as_json(options)
+        end
+      ) if !options[:compact] || has_food_used?
+      result["killedMinerals"] = (
+        if @killed_minerals.nil?
+          {}
+        else
+          @killed_minerals.as_json(options)
+        end
+      ) if !options[:compact] || has_killed_minerals?
+      result["killedVespene"] = (
+        if @killed_vespene.nil?
+          {}
+        else
+          @killed_vespene.as_json(options)
+        end
+      ) if !options[:compact] || has_killed_vespene?
+      result["lostMinerals"] = (
+        if @lost_minerals.nil?
+          {}
+        else
+          @lost_minerals.as_json(options)
+        end
+      ) if !options[:compact] || has_lost_minerals?
+      result["lostVespene"] = (
+        if @lost_vespene.nil?
+          {}
+        else
+          @lost_vespene.as_json(options)
+        end
+      ) if !options[:compact] || has_lost_vespene?
+      result["friendlyFireMinerals"] = (
+        if @friendly_fire_minerals.nil?
+          {}
+        else
+          @friendly_fire_minerals.as_json(options)
+        end
+      ) if !options[:compact] || has_friendly_fire_minerals?
+      result["friendlyFireVespene"] = (
+        if @friendly_fire_vespene.nil?
+          {}
+        else
+          @friendly_fire_vespene.as_json(options)
+        end
+      ) if !options[:compact] || has_friendly_fire_vespene?
+      result["usedMinerals"] = (
+        if @used_minerals.nil?
+          {}
+        else
+          @used_minerals.as_json(options)
+        end
+      ) if !options[:compact] || has_used_minerals?
+      result["usedVespene"] = (
+        if @used_vespene.nil?
+          {}
+        else
+          @used_vespene.as_json(options)
+        end
+      ) if !options[:compact] || has_used_vespene?
+      result["totalUsedMinerals"] = (
+        if @total_used_minerals.nil?
+          {}
+        else
+          @total_used_minerals.as_json(options)
+        end
+      ) if !options[:compact] || has_total_used_minerals?
+      result["totalUsedVespene"] = (
+        if @total_used_vespene.nil?
+          {}
+        else
+          @total_used_vespene.as_json(options)
+        end
+      ) if !options[:compact] || has_total_used_vespene?
+      result["totalDamageDealt"] = (
+        if @total_damage_dealt.nil?
+          {}
+        else
+          @total_damage_dealt.as_json(options)
+        end
+      ) if !options[:compact] || has_total_damage_dealt?
+      result["totalDamageTaken"] = (
+        if @total_damage_taken.nil?
+          {}
+        else
+          @total_damage_taken.as_json(options)
+        end
+      ) if !options[:compact] || has_total_damage_taken?
+      result["totalHealed"] = (
+        if @total_healed.nil?
+          {}
+        else
+          @total_healed.as_json(options)
+        end
+      ) if !options[:compact] || has_total_healed?
+      result["currentApm"] = @current_apm if !options[:compact] ||
+        has_current_apm?
+      result["currentEffectiveApm"] = @current_effective_apm if !options[
+        :compact
+      ] || has_current_effective_apm?
+
       result
     end
+
+    def to_json(as_json_options = {})
+      require "json"
+      JSON.dump(as_json(as_json_options))
+    end
   end
 end