pokemongodb 1.3.0 → 1.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 0d9827fd8159f94b0f7b3d513414ed55920c9bc7
-  data.tar.gz: bc02fc64008488a605b5e6ad0880d0409e3ac86e
+  metadata.gz: 3de28ddbc8b64e47d7d5e8b06a7e92bac5913bf2
+  data.tar.gz: b456325269aeac29a42f2e90f193015464c43815
 SHA512:
-  metadata.gz: 59f00f5eee85dd3d76ff1b81b7214008c54a0c6a915491c193c7ccb1cd6590ebb48d7d3dcd9b44f2b57f58d25147eec4ba0f52c824f443d00bc0d82ca7b86178
-  data.tar.gz: 4d6b42e999753bcda35d146e7fc061ae62cff2958d25fdebd6b2adba82d08489889d4fd7c3ec150e05e610f1bc4aa22c8f79a8034888ea08b348d7e5c40070ff
+  metadata.gz: 48f359f71a4f0aad2ddb59edb4a5b30faf15f5d5b218411a8f0d085521613b0ecccf55f957b82c64cbb53399b56b16581e2f108dd56ed5f8f447b8a6d1f313b5
+  data.tar.gz: 4df8cce955437642b2ec81ad23f4d729a8c42987bb973f3cd86b803c4e34fb3717cab8eb26484c4cfe075d88d5ea3dee7f9ad92eaf2ca35b8d55c345f624820a

data/lib/pokemongodb.rb

@@ -22,5 +22,6 @@ end
 
 require 'pokemongodb/location'
 require 'pokemongodb/move'
+require 'pokemongodb/move_set'
 require 'pokemongodb/pokemon'
 require 'pokemongodb/type'

data/lib/pokemongodb/move_set.rb

@@ -0,0 +1,52 @@
+class Pokemongodb
+  class MoveSet < Pokemongodb
+    # Returns all available MoveSets for Pokemon
+    #
+    # Example:
+    #   >> Pokemongodb::MoveSet.from_pokemon(Pokemongodb::Pokemon::Bulbasaur)
+    #   => [#<Pokemongodb::MoveSet:0xXXXXXX @fast_move=Pokemongodb::Move::VineWhip, @charge_move=Pokemongodb::Move::SludgeBomb, @dps=18.76>, ...]
+    def self.from_pokemon(pokemon)
+      sets = []
+      pokemon.fast_moves.each do |fast_move|
+        pokemon.charge_moves.each do |charge_move|
+          sets << Pokemongodb::MoveSet.new(fast_move, charge_move, pokemon)
+        end
+      end
+      sets.sort_by { |set| set.dps }.reverse
+    end
+
+    def initialize(fast_move, charge_move, pokemon)
+      @fast_move = fast_move
+      @charge_move = charge_move
+      @pokemon = pokemon
+      @dps = dps
+    end
+
+    def dps
+      timer = 90.0
+      damage = 0
+      energy = 0.0
+
+      until timer <= 0
+        move = energy >= @charge_move.energy.abs ? @charge_move : @fast_move
+        timer -= move.cooldown
+        damage += calculated_power(move)
+        energy += calculated_energy(move)
+      end
+
+      return (damage / 90.0).round(2)
+    end
+
+    private
+    def calculated_power(move)
+      modifier = @pokemon.types.include?(move.type) ? 1.25 : 1
+      (move.power * modifier).round(2)
+    end
+
+    def calculated_energy(move)
+      return move.energy if move.energy <= 0
+      modifier = @pokemon.types.include?(move.type) ? 1.25 : 1
+      (move.energy * modifier).round(2)
+    end
+  end
+end

data/lib/pokemongodb/pokemon.rb

@@ -139,6 +139,15 @@ class Pokemongodb
       all.detect { |type| type.evolves_into == self }
     end
 
+    # Returns all available MoveSets for Pokemon
+    #
+    # Example:
+    #   >> Pokemongodb::Pokemon::Bulbasaur.move_sets
+    #   => [#<Pokemongodb::MoveSet:0xXXXXXX @fast_move=Pokemongodb::Move::VineWhip, @charge_move=Pokemongodb::Move::SludgeBomb, @dps=18.76>, ...]
+    def self.move_sets
+      Pokemongodb::MoveSet.from_pokemon(self)
+    end
+
     # Returns array of possible move types
     #
     # Example:

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pokemongodb
 version: !ruby/object:Gem::Version
-  version: 1.3.0
+  version: 1.4.0
 platform: ruby
 authors:
 - TeoUCSB82
@@ -69,6 +69,7 @@ files:
 - lib/pokemongodb/locations/wetland.rb
 - lib/pokemongodb/locations/woodland.rb
 - lib/pokemongodb/move.rb
+- lib/pokemongodb/move_set.rb
 - lib/pokemongodb/moves/acid.rb
 - lib/pokemongodb/moves/aerial_ace.rb
 - lib/pokemongodb/moves/air_cutter.rb