(ns com.aurellem.gb.types

   (:use (com.aurellem.gb gb-driver util constants))

   (:import [com.aurellem.gb.gb_driver SaveState]))

 

 (def type-code->type-name

   {0x00 :normal

    0x01 :fighting

    0x02 :flying

    0x03 :poision

    0x04 :ground

    0x05 :rock

    0x07 :bug

    0x08 :ghost

    0x14 :fire

    0x15 :water

    0x16 :grass

    0x17 :electric

    0x18 :psychic

    0x19 :ice

    0x1A :dragon})

 

 (def type-name->type-code

   (zipmap (vals type-code->type-name)

           (keys type-code->type-name)))

 

 (def pokemon-1-type-start-address 0xD16F)

 

 (defn pokemon-type-start-address [poke-num]

   (+ pokemon-1-type-start-address

      (* pokemon-record-width poke-num)))

 

 (defn give-type

   ([^SaveState state poke-num types]

      (assert (<= 0 poke-num 5))

      (let [types*

            (if (= (count types) 1)

              [(first types) (first types)]

              types)]

        (set-memory-range

         state

         (pokemon-type-start-address poke-num)

         (map type-name->type-code types))))

   ([poke-num types]

      (give-type @current-state poke-num types)))

 

 (defn read-type

   ([^SaveState state poke-num]

      (assert (<= 0 poke-num 5))

      (let [types-start (pokemon-type-start-address poke-num)

            [type-1 type-2]

            (subvec (vec (memory state))

                    types-start (+ 2 types-start))]

        (if (= type-1 type-2)

          [(type-code->type-name type-1)]

          (mapv type-code->type-name [type-1 type-2]))))

   ([poke-num]

      (read-type @current-state poke-num)))

 

 

 (def pokemon->type

   {:rhydon          

    :kangaskhan      

    :nidoran-male    

    :clefairy        

    :spearow         

    :voltorb         

    :nidoking        

    :slowbro         

    :ivysaur         

    :exeggutor       

    :lickitung       

    :exeggcute       

    :grimer          

    :gengar          

    :nidoran-female  

    :nidoqueen       

    :cubone          

    :rhyhorn         

    :lapras          

    :arcanine        

    :mew             

    :gyarados        

    :shellder        

    :tentacool       

    :gastly          

    :scyther         

    :staryu          

    :blastoise       

    :pinsir          

    :tangela         

    :growlithe       

    :onix            

    :fearow          

    :pidgey          

    :slowpoke        

    :kadabra         

    :graveler        

    :chansey         

    :machoke         

    :mr-mime         

    :hitmonlee       

    :hitmonchan      

    :arbok           

    :parasect        

    :psyduck         

    :drowzee         

    :golem           

    :magmar          

    :electabuzz      

    :magneton        

    :koffing         

    :mankey          

    :seel            

    :diglett         

    :tauros          

    :farfetch'd      

    :venonat         

    :dragonite       

    :doduo           

    :poliwag         

    :jynx            

    :moltres         

    :articuno        

    :zapdos          

    :ditto           

    :meowth          

    :krabby          

    :vulpix          

    :ninetails       

    :pikachu         

    :riachu          

    :dratini         

    :dragonair       

    :kabuto          

    :kabutops        

    :horsea          

    :sedra           

    :sandshrew       

    :sandslash       

    :omanyte         

    :omastar         

    :jigglypuff      

    :wigglytuff      

    :eevee           

    :flareon         

    :jolteon         

    :vaporeon        

    :machop          

    :zubat           

    :ekans           

    :paras           

    :poliwhirl       

    :poliwrath       

    :weedle          

    :kakuna          

    :beedrill        

    :dodrio          

    :primeape        

    :dugtrio         

    :venomoth        

    :dewgong         

    :caterpie        

    :metapod         

    :butterfree      

    :machamp         

    :golduck         

    :hypno           

    :golbat          

    :mewtwo          

    :snorlax         

    :magikarp        

    :muk             

    :kingler         

    :cloyster        

    :electrode       

    :clefable        

    :wheezing        

    :persian         

    :marowak         

    :haunter         

    :abra            

    :alakazam        

    :pidgeotto       

    :pidgeot         

    :starmie         

    :bulbasaur       

    :venusaur        

    :tentacruel      

    :goldeen         

    :seaking         

    :ponyta          

    :rapidash        

    :rattata         

    :raticate        

    :nidorino        

    :nidorina        

    :geodude         

    :porygon         

    :aerodactyl      

    :magnemite       

    :charmander      

    :squirtle        

    :charmeleon      

    :wartortle       

    :charizard       

    :oddish          

    :gloom           

    :vileplume       

    :bellsprout      

    :weepenbell      

    :victreebel