infinitefusion-e18/Data/Scripts/051_AddOns/UI_Pokedex_SpritesPage.rb

class PokemonPokedexInfo_Scene
  #todo add indicator to show which one is the main sprite -
  # also maybe add an indicator in main list for when a sprite has available alts

  Y_POSITION_SMALL = 40 #90
  Y_POSITION_BIG = 60
  X_POSITION_PREVIOUS = -30 #20
  X_POSITION_SELECTED = 105
  X_POSITION_NEXT = 340 #380

  Y_POSITION_BG_SMALL = 70
  Y_POSITION_BG_BIG = 93
  X_POSITION_BG_PREVIOUS = -1
  X_POSITION_BG_SELECTED = 145
  X_POSITION_BG_NEXT = 363

  def drawPageForms()
    #@selected_index=0
    @sprites["background"].setBitmap(_INTL("Graphics/Pictures/Pokedex/bg_forms"))
    overlay = @sprites["overlay"].bitmap
    base = Color.new(88, 88, 80)
    shadow = Color.new(168, 184, 184)

    #alts_list= pbGetAvailableAlts
    @selected_index = 0 if !@selected_index
    update_displayed
  end

  def init_selected_bg
    @sprites["bgSelected_previous"] = IconSprite.new(0, 0, @viewport)
    @sprites["bgSelected_previous"].x = X_POSITION_BG_PREVIOUS
    @sprites["bgSelected_previous"].y = Y_POSITION_BG_SMALL
    @sprites["bgSelected_previous"].setBitmap(_INTL("Graphics/Pictures/Pokedex/bg_forms_selected_small"))
    @sprites["bgSelected_previous"].visible = false

    @sprites["bgSelected_center"] = IconSprite.new(0, 0, @viewport)
    @sprites["bgSelected_center"].x = X_POSITION_BG_SELECTED
    @sprites["bgSelected_center"].y = Y_POSITION_BG_BIG
    @sprites["bgSelected_center"].setBitmap(_INTL("Graphics/Pictures/Pokedex/bg_forms_selected_large"))
    @sprites["bgSelected_center"].visible = false

    @sprites["bgSelected_next"] = IconSprite.new(0, 0, @viewport)
    @sprites["bgSelected_next"].x = X_POSITION_BG_NEXT
    @sprites["bgSelected_next"].y = Y_POSITION_BG_SMALL
    @sprites["bgSelected_next"].setBitmap(_INTL("Graphics/Pictures/Pokedex/bg_forms_selected_small"))
    @sprites["bgSelected_next"].visible = false

    @creditsOverlay = BitmapSprite.new(Graphics.width, Graphics.height, @viewport).bitmap

  end

  def initializeSpritesPage(altsList)
    @forms_list = list_pokemon_forms()
    @formIndex = 0

    init_selected_bg
    @speciesData = getSpecies(@species)

    @selected_index = 0
    set_displayed_to_current_alt(altsList)


    @sprites["selectedSprite"] = IconSprite.new(0, 0, @viewport)
    @sprites["selectedSprite"].x = X_POSITION_SELECTED
    @sprites["selectedSprite"].y = Y_POSITION_BIG
    @sprites["selectedSprite"].z = 999999
    @sprites["selectedSprite"].visible = false
    @sprites["selectedSprite"].zoom_x = 1
    @sprites["selectedSprite"].zoom_y = 1

    @sprites["previousSprite"] = IconSprite.new(0, 0, @viewport)
    @sprites["previousSprite"].x = X_POSITION_PREVIOUS
    @sprites["previousSprite"].y = Y_POSITION_SMALL
    @sprites["previousSprite"].visible = false
    @sprites["previousSprite"].zoom_x = Settings::FRONTSPRITE_SCALE #/2
    @sprites["previousSprite"].zoom_y = Settings::FRONTSPRITE_SCALE #/2

    @sprites["nextSprite"] = IconSprite.new(0, 0, @viewport)
    @sprites["nextSprite"].x = X_POSITION_NEXT
    @sprites["nextSprite"].y = Y_POSITION_SMALL
    @sprites["nextSprite"].visible = false
    @sprites["nextSprite"].zoom_x = Settings::FRONTSPRITE_SCALE #/2
    @sprites["nextSprite"].zoom_y = Settings::FRONTSPRITE_SCALE #/2

    @sprites["selectedSprite"].z = 9999999
    @sprites["previousSprite"].z = 9999999
    @sprites["nextSprite"].z = 9999999

    @sprites["selectedSprite"].setBitmap(altsList[@selected_index])

    if altsList.size >= 2
      @sprites["nextSprite"].setBitmap(altsList[@selected_index + 1])
      @sprites["nextSprite"].visible = true
    end

    if altsList.size >= 3
      @sprites["previousSprite"].setBitmap(altsList[-1])
      @sprites["previousSprite"].visible = true
    end

  end

  def get_currently_selected_sprite()
    species_id = getDexNumberForSpecies(@species).to_s
    $PokemonGlobal.alt_sprite_substitutions = {} if !$PokemonGlobal.alt_sprite_substitutions
    return $PokemonGlobal.alt_sprite_substitutions[species_id]
  end

  def set_displayed_to_current_alt(altsList)
    species_id = getDexNumberForSpecies(@species).to_s
    $PokemonGlobal.alt_sprite_substitutions = {} if !$PokemonGlobal.alt_sprite_substitutions
    return if !$PokemonGlobal.alt_sprite_substitutions[species_id]

    current_sprite =$PokemonGlobal.alt_sprite_substitutions[species_id]
    index = @selected_index
    for alt in altsList
      if alt == current_sprite
        @selected_index = index
        return
      end
      index +=1
    end
  end


  def pbGetAvailableForms(species=nil)
    chosen_species = species != nil ? species : @species
    dex_num = getDexNumberForSpecies(chosen_species)
    if dex_num <= NB_POKEMON
      download_all_unfused_alt_sprites(dex_num)
    else
      body_id = getBodyID(chosen_species)
      head_id = getHeadID(chosen_species, body_id)
      download_custom_sprite(head_id, body_id)
      download_autogen_sprite(head_id, body_id)
      download_all_alt_sprites(head_id, body_id)
    end
    return PokedexUtils.new.pbGetAvailableAlts(chosen_species, @formIndex)
  end

  def hide_all_selected_windows
    @sprites["bgSelected_previous"].visible = false if @sprites["bgSelected_previous"]
    @sprites["bgSelected_center"].visible = false if @sprites["bgSelected_center"]
    @sprites["bgSelected_next"].visible = false if @sprites["bgSelected_next"]
  end

  def update_selected
    hide_all_selected_windows
    previous_index = @selected_index == 0 ? @available.size - 1 : @selected_index - 1
    next_index = @selected_index == @available.size - 1 ? 0 : @selected_index + 1

    @sprites["bgSelected_previous"].visible = true if is_main_sprite(previous_index) && @available.size > 2
    @sprites["bgSelected_center"].visible = true if is_main_sprite(@selected_index)
    @sprites["bgSelected_next"].visible = true if is_main_sprite(next_index) && @available.size > 1
  end

  def isBaseSpritePath(path)
    filename = File.basename(path).downcase
    return filename.match?(/\A\d+\.png\Z/)
  end

  def update_displayed
    @sprites["selectedSprite"].setBitmap(@available[@selected_index])
    nextIndex = @selected_index + 1
    previousIndex = @selected_index - 1
    if nextIndex > @available.size - 1
      nextIndex = 0
    end
    if previousIndex < 0
      previousIndex = @available.size - 1
    end
    @sprites["previousSprite"].visible = @available.size > 2
    @sprites["nextSprite"].visible = @available.size > 1

    @sprites["previousSprite"].setBitmap(@available[previousIndex]) if previousIndex != nextIndex

    @sprites["selectedSprite"].setBitmap(@available[@selected_index])
    @sprites["nextSprite"].setBitmap(@available[nextIndex])

    selected_bitmap = @sprites["selectedSprite"].getBitmap
    sprite_path = selected_bitmap.path
    isBaseSprite = isBaseSpritePath(@available[@selected_index])
    is_generated = sprite_path.start_with?(Settings::BATTLERS_FOLDER) && !isBaseSprite
    echoln is_generated
    showSpriteCredits(selected_bitmap.filename, is_generated)
    update_selected
  end

  def showSpriteCredits(filename, generated_sprite = false)
    @creditsOverlay.dispose

    x = Graphics.width / 2 - 75
    y = Graphics.height - 60
    spritename = File.basename(filename, '.*')

    if !generated_sprite
      echoln spritename
      discord_name = getSpriteCredits(spritename)
      discord_name = "Unknown artist" if !discord_name
    else
      #todo give credits to Japeal - need to differenciate unfused sprites
      discord_name = "" #"Japeal\n(Generated)"
    end

    author_name = File.basename(discord_name, '#*')

    label_base_color = Color.new(248, 248, 248)
    label_shadow_color = Color.new(104, 104, 104)
    @creditsOverlay = BitmapSprite.new(Graphics.width, Graphics.height, @viewport).bitmap
    textpos = [[author_name, x, y, 0, label_base_color, label_shadow_color]]
    pbDrawTextPositions(@creditsOverlay, textpos)
  end

  def list_pokemon_forms
    dexNum = dexNum(@species)
    if dexNum > NB_POKEMON
      body_id = getBodyID(dexNum)
    else
      if @species.is_a?(Symbol)
        body_id = get_body_number_from_symbol(@species)
      else
        body_id = dexNum
      end
    end
    forms_list = []
    found_last_form = false
    form_index = 0
    while !found_last_form
      form_index += 1
      form_path = Settings::BATTLERS_FOLDER + body_id.to_s + "_" + form_index.to_s
      if File.directory?(form_path)
        forms_list << form_index
      else
        found_last_form = true
      end
    end
    return forms_list
  end



  def pbChooseAlt(brief=false)
    loop do
      @sprites["rightarrow"].visible = true
      @sprites["leftarrow"].visible = true
      if @forms_list.length >= 1
        @sprites["uparrow"].visible = true
        @sprites["downarrow"].visible = true
      end
      multiple_forms = @forms_list.length > 0
      Graphics.update
      Input.update
      pbUpdate
      if Input.trigger?(Input::LEFT)
        pbPlayCursorSE
        @selected_index -= 1 #(index+@available.length-1)%@available.length
        if @selected_index < 0
          @selected_index = @available.size - 1
        end
        update_displayed
      elsif Input.trigger?(Input::RIGHT)
        pbPlayCursorSE
        @selected_index += 1
        if @selected_index > @available.size - 1
          @selected_index = 0
        end
        update_displayed
      elsif Input.trigger?(Input::UP) && multiple_forms
        pbPlayCursorSE
        @formIndex += 1
        if @formIndex > @forms_list.length
          @formIndex = 0
        end
        @available = pbGetAvailableForms()
        @selected_index = 0
        update_displayed
      elsif Input.trigger?(Input::DOWN) && multiple_forms
        pbPlayCursorSE
        @formIndex -= 1
        if @formIndex < 0
          @formIndex = @forms_list.length
        end
        @available = pbGetAvailableForms()
        @selected_index = 0
        update_displayed
      elsif Input.trigger?(Input::BACK)
        pbPlayCancelSE
        break
      elsif Input.trigger?(Input::USE)
        pbPlayDecisionSE
        if select_sprite(brief)
          @endscene = true
          break
        end
      end
    end
    @sprites["uparrow"].visible = false
    @sprites["downarrow"].visible = false
  end

  # def is_main_sprite(index = nil)
  #   return false if !@available
  #   if index == nil
  #     index = @selected_index
  #   end
  #   return true if @available.size <= 1
  #   if @speciesData.always_use_generated
  #     selected_sprite = @available[index]
  #     return selected_sprite.start_with?(Settings::BATTLERS_FOLDER)
  #   end
  #   return index == 0
  # end

  def is_main_sprite(index = nil)
    if !index
      index = @selected_index
    end
    selected_sprite = @available[index]
    species_id = getDexNumberForSpecies(@species).to_s
    $PokemonGlobal.alt_sprite_substitutions = {} if !$PokemonGlobal.alt_sprite_substitutions
    if $PokemonGlobal.alt_sprite_substitutions[species_id]
      return $PokemonGlobal.alt_sprite_substitutions[species_id] == selected_sprite
    end
    is_generated = !selected_sprite.include?(Settings::CUSTOM_BATTLERS_FOLDER_INDEXED)
    if is_generated
      return !checkIfCustomSpriteExistsByPath(selected_sprite)
    end
    return !sprite_is_alt(selected_sprite)
  end

  def sprite_is_alt(sprite_path)
    spritename = File.basename(sprite_path, '.*')
    return spritename.match?(/[a-zA-Z]/)
  end

  def select_sprite(brief=false)
    if @available.length > 1
      if is_main_sprite
        if brief
          pbMessage("This sprite will remain the displayed sprite")
          return true
        else
          pbMessage("This sprite is already the displayed sprite")
        end
      else
        if @forms_list.length > 0
          message = _INTL('Would you like to use this sprite instead of the current sprite for form {1}?', @formIndex)
        else
          message = 'Would you like to use this sprite instead of the current sprite?'
        end
        if pbConfirmMessage(_INTL(message))
          swap_main_sprite()
          return true
        end
      end
    else
      pbMessage("This is the only sprite available for this Pokémon!")
    end
    return false
  end

  def swap_main_sprite
    old_main_sprite = @available[0]
    new_main_sprite = @available[@selected_index]
    species_number = dexNum(@species)
    set_alt_sprite_substitution(species_number, new_main_sprite, @formIndex)
  end

  # def swap_main_sprite
  #   begin
  #     old_main_sprite = @available[0]
  #     new_main_sprite = @available[@selected_index]
  #
  #     if main_sprite_is_non_custom()
  #       @speciesData.set_always_use_generated_sprite(false)
  #       return
  #       # new_name_without_ext = File.basename(old_main_sprite, ".png")
  #       # new_name_without_letter=new_name_without_ext.chop
  #       # File.rename(new_main_sprite, Settings::CUSTOM_BATTLERS_FOLDER+new_name_without_letter + ".png")
  #     end
  #
  #     if new_main_sprite.start_with?(Settings::BATTLERS_FOLDER)
  #       @speciesData.set_always_use_generated_sprite(true)
  #       return
  #       # new_name_without_ext = File.basename(old_main_sprite, ".png")
  #       # File.rename(old_main_sprite, Settings::CUSTOM_BATTLERS_FOLDER+new_name_without_ext+"x" + ".png")x
  #       # return
  #     end
  #     File.rename(new_main_sprite, new_main_sprite + "temp")
  #     File.rename(old_main_sprite, new_main_sprite)
  #     File.rename(new_main_sprite + "temp", old_main_sprite)
  #   rescue
  #     pbMessage("There was an error while swapping the sprites. Please save and restart the game as soon as possible.")
  #   end
  # end

  # def main_sprite_is_non_custom()
  #   speciesData = getSpecies(@species)
  #   return speciesData.always_use_generated || @available.size <= 1
  # end
end

class PokemonGlobalMetadata
  attr_accessor :alt_sprite_substitutions
end

def set_alt_sprite_substitution(original_sprite_name, selected_alt, formIndex = 0)
  if !$PokemonGlobal.alt_sprite_substitutions
    $PokemonGlobal.alt_sprite_substitutions = {}
  end
  if formIndex
    form_suffix = formIndex != 0 ? "_" + formIndex.to_s : ""
  else
    form_suffix = ""
  end
  $PokemonGlobal.alt_sprite_substitutions[original_sprite_name.to_s + form_suffix] = selected_alt
end