package d2scene

import (
	"image"
	"image/color"

	"github.com/OpenDiablo2/OpenDiablo2/d2core"

	"github.com/OpenDiablo2/D2Shared/d2common/d2resource"

	"github.com/OpenDiablo2/OpenDiablo2/d2render"

	"github.com/OpenDiablo2/D2Shared/d2common/d2enum"
	"github.com/OpenDiablo2/OpenDiablo2/d2corecommon/d2coreinterface"

	"github.com/OpenDiablo2/D2Shared/d2common"
	dh "github.com/OpenDiablo2/D2Shared/d2helper"
	"github.com/OpenDiablo2/OpenDiablo2/d2audio"
	"github.com/OpenDiablo2/OpenDiablo2/d2render/d2ui"
	"github.com/hajimehoshi/ebiten"
)

type HeroRenderInfo struct {
	Stance                   d2enum.HeroStance
	IdleSprite               *d2render.Sprite
	IdleSelectedSprite       *d2render.Sprite
	ForwardWalkSprite        *d2render.Sprite
	ForwardWalkSpriteOverlay *d2render.Sprite
	SelectedSprite           *d2render.Sprite
	SelectedSpriteOverlay    *d2render.Sprite
	BackWalkSprite           *d2render.Sprite
	BackWalkSpriteOverlay    *d2render.Sprite
	SelectionBounds          image.Rectangle
	SelectSfx                *d2audio.SoundEffect
	DeselectSfx              *d2audio.SoundEffect
}

type SelectHeroClass struct {
	uiManager          *d2ui.Manager
	soundManager       *d2audio.Manager
	sceneProvider      d2coreinterface.SceneProvider
	bgImage            *d2render.Sprite
	campfire           *d2render.Sprite
	headingLabel       d2ui.Label
	heroClassLabel     d2ui.Label
	heroDesc1Label     d2ui.Label
	heroDesc2Label     d2ui.Label
	heroDesc3Label     d2ui.Label
	heroNameTextbox    d2ui.TextBox
	heroNameLabel      d2ui.Label
	heroRenderInfo     map[d2enum.Hero]*HeroRenderInfo
	selectedHero       d2enum.Hero
	exitButton         d2ui.Button
	okButton           d2ui.Button
	expansionCheckbox  d2ui.Checkbox
	expansionCharLabel d2ui.Label
	hardcoreCheckbox   d2ui.Checkbox
	hardcoreCharLabel  d2ui.Label
}

func CreateSelectHeroClass(sceneProvider d2coreinterface.SceneProvider, uiManager *d2ui.Manager, soundManager *d2audio.Manager) *SelectHeroClass {
	result := &SelectHeroClass{
		uiManager:      uiManager,
		sceneProvider:  sceneProvider,
		soundManager:   soundManager,
		heroRenderInfo: make(map[d2enum.Hero]*HeroRenderInfo),
		selectedHero:   d2enum.HeroNone,
	}
	return result
}

func (v *SelectHeroClass) Load() []func() {
	v.soundManager.PlayBGM(d2resource.BGMTitle)
	return []func(){
		func() {
			v.bgImage, _ = d2render.LoadSprite(d2resource.CharacterSelectBackground, d2resource.PaletteFechar)
			v.bgImage.SetPosition(0, 0)
		},
		func() {
			v.headingLabel = d2ui.CreateLabel(d2resource.Font30, d2resource.PaletteUnits)
			fontWidth, _ := v.headingLabel.GetSize()
			v.headingLabel.SetPosition(400-int(fontWidth/2), 17)
			v.headingLabel.SetText("Select Hero Class")
			v.headingLabel.Alignment = d2ui.LabelAlignCenter
		},
		func() {
			v.heroClassLabel = d2ui.CreateLabel(d2resource.Font30, d2resource.PaletteUnits)
			v.heroClassLabel.Alignment = d2ui.LabelAlignCenter
			v.heroClassLabel.SetPosition(400, 65)
		},
		func() {
			v.heroDesc1Label = d2ui.CreateLabel(d2resource.Font16, d2resource.PaletteUnits)
			v.heroDesc1Label.Alignment = d2ui.LabelAlignCenter
			v.heroDesc1Label.SetPosition(400, 100)
		},
		func() {
			v.heroDesc2Label = d2ui.CreateLabel(d2resource.Font16, d2resource.PaletteUnits)
			v.heroDesc2Label.Alignment = d2ui.LabelAlignCenter
			v.heroDesc2Label.SetPosition(400, 115)
		},
		func() {
			v.heroDesc3Label = d2ui.CreateLabel(d2resource.Font16, d2resource.PaletteUnits)
			v.heroDesc3Label.Alignment = d2ui.LabelAlignCenter
			v.heroDesc3Label.SetPosition(400, 130)
		},
		func() {
			v.campfire, _ = d2render.LoadSprite(d2resource.CharacterSelectCampfire, d2resource.PaletteFechar)
			v.campfire.SetPosition(380, 335)
			v.campfire.PlayForward()
			v.campfire.SetBlend(true)
		},
		func() {
			v.exitButton = d2ui.CreateButton(d2ui.ButtonTypeMedium, d2common.TranslateString("#970"))
			v.exitButton.SetPosition(33, 537)
			v.exitButton.OnActivated(func() { v.onExitButtonClicked() })
			v.uiManager.AddWidget(&v.exitButton)
		},
		func() {
			v.okButton = d2ui.CreateButton(d2ui.ButtonTypeMedium, d2common.TranslateString("#971"))
			v.okButton.SetPosition(630, 537)
			v.okButton.OnActivated(func() { v.onOkButtonClicked() })
			v.okButton.SetVisible(false)
			v.okButton.SetEnabled(false)
			v.uiManager.AddWidget(&v.okButton)
		},
		func() {
			v.heroNameLabel = d2ui.CreateLabel(d2resource.Font16, d2resource.PaletteUnits)
			v.heroNameLabel.Alignment = d2ui.LabelAlignLeft
			v.heroNameLabel.Color = color.RGBA{216, 196, 128, 255}
			v.heroNameLabel.SetText(d2common.TranslateString("#1694"))
			v.heroNameLabel.SetPosition(321, 475)
		},
		func() {
			v.heroNameTextbox = d2ui.CreateTextbox()
			v.heroNameTextbox.SetPosition(318, 493)
			v.heroNameTextbox.SetVisible(false)
			v.uiManager.AddWidget(&v.heroNameTextbox)
		},
		func() {
			v.expansionCheckbox = d2ui.CreateCheckbox(true)
			v.expansionCheckbox.SetPosition(318, 526)
			v.expansionCheckbox.SetVisible(false)
			v.uiManager.AddWidget(&v.expansionCheckbox)
		},
		func() {
			v.expansionCharLabel = d2ui.CreateLabel(d2resource.Font16, d2resource.PaletteUnits)
			v.expansionCharLabel.Alignment = d2ui.LabelAlignLeft
			v.expansionCharLabel.Color = color.RGBA{216, 196, 128, 255}
			v.expansionCharLabel.SetText(d2common.TranslateString("#803"))
			v.expansionCharLabel.SetPosition(339, 526)
		},
		func() {
			v.hardcoreCheckbox = d2ui.CreateCheckbox(false)
			v.hardcoreCheckbox.SetPosition(318, 548)
			v.hardcoreCheckbox.SetVisible(false)
			v.uiManager.AddWidget(&v.hardcoreCheckbox)
		},
		func() {
			v.hardcoreCharLabel = d2ui.CreateLabel(d2resource.Font16, d2resource.PaletteUnits)
			v.hardcoreCharLabel.Alignment = d2ui.LabelAlignLeft
			v.hardcoreCharLabel.Color = color.RGBA{216, 196, 128, 255}
			v.hardcoreCharLabel.SetText(d2common.TranslateString("#1696"))
			v.hardcoreCharLabel.SetPosition(339, 548)
		},
		func() {
			v.heroRenderInfo[d2enum.HeroBarbarian] = &HeroRenderInfo{
				d2enum.HeroStanceIdle,
				d2render.MustLoadSprite(d2resource.CharacterSelectBarbarianUnselected, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelectBarbarianUnselectedH, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelectBarbarianForwardWalk, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelectBarbarianForwardWalkOverlay, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelectBarbarianSelected, d2resource.PaletteFechar),
				nil,
				d2render.MustLoadSprite(d2resource.CharacterSelectBarbarianBackWalk, d2resource.PaletteFechar),
				nil,
				image.Rectangle{Min: image.Point{364, 201}, Max: image.Point{90, 170}},
				v.soundManager.LoadSoundEffect(d2resource.SFXBarbarianSelect),
				v.soundManager.LoadSoundEffect(d2resource.SFXBarbarianDeselect),
			}
			v.heroRenderInfo[d2enum.HeroBarbarian].IdleSprite.SetPosition(400, 330)
			v.heroRenderInfo[d2enum.HeroBarbarian].IdleSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroBarbarian].IdleSelectedSprite.SetPosition(400, 330)
			v.heroRenderInfo[d2enum.HeroBarbarian].IdleSelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroBarbarian].ForwardWalkSprite.SetPosition(400, 330)
			v.heroRenderInfo[d2enum.HeroBarbarian].ForwardWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroBarbarian].ForwardWalkSprite.SetPlayLengthMs(2500)
			v.heroRenderInfo[d2enum.HeroBarbarian].ForwardWalkSprite.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroBarbarian].ForwardWalkSpriteOverlay.SetPosition(400, 330)
			v.heroRenderInfo[d2enum.HeroBarbarian].ForwardWalkSpriteOverlay.PlayForward()
			v.heroRenderInfo[d2enum.HeroBarbarian].ForwardWalkSpriteOverlay.SetPlayLengthMs(2500)
			v.heroRenderInfo[d2enum.HeroBarbarian].ForwardWalkSpriteOverlay.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroBarbarian].SelectedSprite.SetPosition(400, 330)
			v.heroRenderInfo[d2enum.HeroBarbarian].SelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroBarbarian].BackWalkSprite.SetPosition(400, 330)
			v.heroRenderInfo[d2enum.HeroBarbarian].BackWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroBarbarian].BackWalkSprite.SetPlayLengthMs(1000)
			v.heroRenderInfo[d2enum.HeroBarbarian].BackWalkSprite.SetPlayLoop(false)
		},
		func() {
			v.heroRenderInfo[d2enum.HeroSorceress] = &HeroRenderInfo{
				d2enum.HeroStanceIdle,
				d2render.MustLoadSprite(d2resource.CharacterSelecSorceressUnselected, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecSorceressUnselectedH, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecSorceressForwardWalk, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecSorceressForwardWalkOverlay, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecSorceressSelected, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecSorceressSelectedOverlay, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecSorceressBackWalk, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecSorceressBackWalkOverlay, d2resource.PaletteFechar),
				image.Rectangle{Min: image.Point{580, 240}, Max: image.Point{65, 160}},
				v.soundManager.LoadSoundEffect(d2resource.SFXSorceressSelect),
				v.soundManager.LoadSoundEffect(d2resource.SFXSorceressDeselect),
			}
			v.heroRenderInfo[d2enum.HeroSorceress].IdleSprite.SetPosition(626, 352)
			v.heroRenderInfo[d2enum.HeroSorceress].IdleSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroSorceress].IdleSprite.SetPlayLengthMs(2500)
			v.heroRenderInfo[d2enum.HeroSorceress].IdleSelectedSprite.SetPosition(626, 352)
			v.heroRenderInfo[d2enum.HeroSorceress].IdleSelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroSorceress].IdleSelectedSprite.SetPlayLengthMs(2500)
			v.heroRenderInfo[d2enum.HeroSorceress].ForwardWalkSprite.SetPosition(626, 352)
			v.heroRenderInfo[d2enum.HeroSorceress].ForwardWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroSorceress].ForwardWalkSprite.SetPlayLengthMs(2300)
			v.heroRenderInfo[d2enum.HeroSorceress].ForwardWalkSprite.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroSorceress].ForwardWalkSpriteOverlay.SetBlend(true)
			v.heroRenderInfo[d2enum.HeroSorceress].ForwardWalkSpriteOverlay.SetPosition(626, 352)
			v.heroRenderInfo[d2enum.HeroSorceress].ForwardWalkSpriteOverlay.PlayForward()
			v.heroRenderInfo[d2enum.HeroSorceress].ForwardWalkSpriteOverlay.SetPlayLengthMs(2300)
			v.heroRenderInfo[d2enum.HeroSorceress].ForwardWalkSpriteOverlay.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroSorceress].SelectedSprite.SetPosition(626, 352)
			v.heroRenderInfo[d2enum.HeroSorceress].SelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroSorceress].SelectedSprite.SetPlayLengthMs(450)
			v.heroRenderInfo[d2enum.HeroSorceress].SelectedSpriteOverlay.SetBlend(true)
			v.heroRenderInfo[d2enum.HeroSorceress].SelectedSpriteOverlay.SetPosition(626, 352)
			v.heroRenderInfo[d2enum.HeroSorceress].SelectedSpriteOverlay.PlayForward()
			v.heroRenderInfo[d2enum.HeroSorceress].SelectedSpriteOverlay.SetPlayLengthMs(450)
			v.heroRenderInfo[d2enum.HeroSorceress].BackWalkSprite.SetPosition(626, 352)
			v.heroRenderInfo[d2enum.HeroSorceress].BackWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroSorceress].BackWalkSprite.SetPlayLengthMs(1200)
			v.heroRenderInfo[d2enum.HeroSorceress].BackWalkSprite.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroSorceress].BackWalkSpriteOverlay.SetBlend(true)
			v.heroRenderInfo[d2enum.HeroSorceress].BackWalkSpriteOverlay.SetPosition(626, 352)
			v.heroRenderInfo[d2enum.HeroSorceress].BackWalkSpriteOverlay.PlayForward()
			v.heroRenderInfo[d2enum.HeroSorceress].BackWalkSpriteOverlay.SetPlayLengthMs(1200)
			v.heroRenderInfo[d2enum.HeroSorceress].BackWalkSpriteOverlay.SetPlayLoop(false)
		},
		func() {
			v.heroRenderInfo[d2enum.HeroNecromancer] = &HeroRenderInfo{
				d2enum.HeroStanceIdle,
				d2render.MustLoadSprite(d2resource.CharacterSelectNecromancerUnselected, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelectNecromancerUnselectedH, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecNecromancerForwardWalk, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecNecromancerForwardWalkOverlay, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecNecromancerSelected, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecNecromancerSelectedOverlay, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecNecromancerBackWalk, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecNecromancerBackWalkOverlay, d2resource.PaletteFechar),
				image.Rectangle{Min: image.Point{265, 220}, Max: image.Point{55, 175}},
				v.soundManager.LoadSoundEffect(d2resource.SFXNecromancerSelect),
				v.soundManager.LoadSoundEffect(d2resource.SFXNecromancerDeselect),
			}
			v.heroRenderInfo[d2enum.HeroNecromancer].IdleSprite.SetPosition(300, 335)
			v.heroRenderInfo[d2enum.HeroNecromancer].IdleSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroNecromancer].IdleSprite.SetPlayLengthMs(1200)
			v.heroRenderInfo[d2enum.HeroNecromancer].IdleSelectedSprite.SetPosition(300, 335)
			v.heroRenderInfo[d2enum.HeroNecromancer].IdleSelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroNecromancer].IdleSelectedSprite.SetPlayLengthMs(1200)
			v.heroRenderInfo[d2enum.HeroNecromancer].ForwardWalkSprite.SetPosition(300, 335)
			v.heroRenderInfo[d2enum.HeroNecromancer].ForwardWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroNecromancer].ForwardWalkSprite.SetPlayLengthMs(2000)
			v.heroRenderInfo[d2enum.HeroNecromancer].ForwardWalkSprite.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroNecromancer].ForwardWalkSpriteOverlay.SetBlend(true)
			v.heroRenderInfo[d2enum.HeroNecromancer].ForwardWalkSpriteOverlay.SetPosition(300, 335)
			v.heroRenderInfo[d2enum.HeroNecromancer].ForwardWalkSpriteOverlay.PlayForward()
			v.heroRenderInfo[d2enum.HeroNecromancer].ForwardWalkSpriteOverlay.SetPlayLengthMs(2000)
			v.heroRenderInfo[d2enum.HeroNecromancer].ForwardWalkSpriteOverlay.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroNecromancer].SelectedSprite.SetPosition(300, 335)
			v.heroRenderInfo[d2enum.HeroNecromancer].SelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroNecromancer].SelectedSpriteOverlay.SetBlend(true)
			v.heroRenderInfo[d2enum.HeroNecromancer].SelectedSpriteOverlay.SetPosition(300, 335)
			v.heroRenderInfo[d2enum.HeroNecromancer].SelectedSpriteOverlay.PlayForward()
			v.heroRenderInfo[d2enum.HeroNecromancer].BackWalkSprite.SetPosition(300, 335)
			v.heroRenderInfo[d2enum.HeroNecromancer].BackWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroNecromancer].BackWalkSprite.SetPlayLengthMs(1500)
			v.heroRenderInfo[d2enum.HeroNecromancer].BackWalkSprite.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroNecromancer].BackWalkSpriteOverlay.SetBlend(true)
			v.heroRenderInfo[d2enum.HeroNecromancer].BackWalkSpriteOverlay.SetPosition(300, 335)
			v.heroRenderInfo[d2enum.HeroNecromancer].BackWalkSpriteOverlay.PlayForward()
			v.heroRenderInfo[d2enum.HeroNecromancer].BackWalkSpriteOverlay.SetPlayLengthMs(1500)
			v.heroRenderInfo[d2enum.HeroNecromancer].BackWalkSpriteOverlay.SetPlayLoop(false)
		},
		func() {
			v.heroRenderInfo[d2enum.HeroPaladin] = &HeroRenderInfo{
				d2enum.HeroStanceIdle,
				d2render.MustLoadSprite(d2resource.CharacterSelectPaladinUnselected, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelectPaladinUnselectedH, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecPaladinForwardWalk, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecPaladinForwardWalkOverlay, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecPaladinSelected, d2resource.PaletteFechar),
				nil,
				d2render.MustLoadSprite(d2resource.CharacterSelecPaladinBackWalk, d2resource.PaletteFechar),
				nil,
				image.Rectangle{Min: image.Point{490, 210}, Max: image.Point{65, 180}},
				v.soundManager.LoadSoundEffect(d2resource.SFXPaladinSelect),
				v.soundManager.LoadSoundEffect(d2resource.SFXPaladinDeselect),
			}
			v.heroRenderInfo[d2enum.HeroPaladin].IdleSprite.SetPosition(521, 338)
			v.heroRenderInfo[d2enum.HeroPaladin].IdleSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroPaladin].IdleSprite.SetPlayLengthMs(2500)
			v.heroRenderInfo[d2enum.HeroPaladin].IdleSelectedSprite.SetPosition(521, 338)
			v.heroRenderInfo[d2enum.HeroPaladin].IdleSelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroPaladin].IdleSelectedSprite.SetPlayLengthMs(2500)
			v.heroRenderInfo[d2enum.HeroPaladin].ForwardWalkSprite.SetPosition(521, 338)
			v.heroRenderInfo[d2enum.HeroPaladin].ForwardWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroPaladin].ForwardWalkSprite.SetPlayLengthMs(3400)
			v.heroRenderInfo[d2enum.HeroPaladin].ForwardWalkSprite.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroPaladin].ForwardWalkSpriteOverlay.SetPosition(521, 338)
			v.heroRenderInfo[d2enum.HeroPaladin].ForwardWalkSpriteOverlay.PlayForward()
			v.heroRenderInfo[d2enum.HeroPaladin].ForwardWalkSpriteOverlay.SetPlayLengthMs(3400)
			v.heroRenderInfo[d2enum.HeroPaladin].ForwardWalkSpriteOverlay.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroPaladin].SelectedSprite.SetPosition(521, 338)
			v.heroRenderInfo[d2enum.HeroPaladin].SelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroPaladin].SelectedSprite.SetPlayLengthMs(650)
			v.heroRenderInfo[d2enum.HeroPaladin].BackWalkSprite.SetPosition(521, 338)
			v.heroRenderInfo[d2enum.HeroPaladin].BackWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroPaladin].BackWalkSprite.SetPlayLengthMs(1300)
			v.heroRenderInfo[d2enum.HeroPaladin].BackWalkSprite.SetPlayLoop(false)
		},
		func() {
			v.heroRenderInfo[d2enum.HeroAmazon] = &HeroRenderInfo{
				d2enum.HeroStanceIdle,
				d2render.MustLoadSprite(d2resource.CharacterSelectAmazonUnselected, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelectAmazonUnselectedH, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelecAmazonForwardWalk, d2resource.PaletteFechar),
				nil,
				d2render.MustLoadSprite(d2resource.CharacterSelecAmazonSelected, d2resource.PaletteFechar),
				nil,
				d2render.MustLoadSprite(d2resource.CharacterSelecAmazonBackWalk, d2resource.PaletteFechar),
				nil,
				image.Rectangle{Min: image.Point{70, 220}, Max: image.Point{55, 200}},
				v.soundManager.LoadSoundEffect(d2resource.SFXAmazonSelect),
				v.soundManager.LoadSoundEffect(d2resource.SFXAmazonDeselect),
			}
			v.heroRenderInfo[d2enum.HeroAmazon].IdleSprite.SetPosition(100, 339)
			v.heroRenderInfo[d2enum.HeroAmazon].IdleSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroAmazon].IdleSprite.SetPlayLengthMs(2500)
			v.heroRenderInfo[d2enum.HeroAmazon].IdleSelectedSprite.SetPosition(100, 339)
			v.heroRenderInfo[d2enum.HeroAmazon].IdleSelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroAmazon].IdleSelectedSprite.SetPlayLengthMs(2500)
			v.heroRenderInfo[d2enum.HeroAmazon].ForwardWalkSprite.SetPosition(100, 339)
			v.heroRenderInfo[d2enum.HeroAmazon].ForwardWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroAmazon].ForwardWalkSprite.SetPlayLengthMs(2200)
			v.heroRenderInfo[d2enum.HeroAmazon].ForwardWalkSprite.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroAmazon].SelectedSprite.SetPosition(100, 339)
			v.heroRenderInfo[d2enum.HeroAmazon].SelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroAmazon].SelectedSprite.SetPlayLengthMs(1350)
			v.heroRenderInfo[d2enum.HeroAmazon].BackWalkSprite.SetPosition(100, 339)
			v.heroRenderInfo[d2enum.HeroAmazon].BackWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroAmazon].BackWalkSprite.SetPlayLengthMs(1500)
			v.heroRenderInfo[d2enum.HeroAmazon].BackWalkSprite.SetPlayLoop(false)
		},
		func() {
			v.heroRenderInfo[d2enum.HeroAssassin] = &HeroRenderInfo{
				d2enum.HeroStanceIdle,
				d2render.MustLoadSprite(d2resource.CharacterSelectAssassinUnselected, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelectAssassinUnselectedH, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelectAssassinForwardWalk, d2resource.PaletteFechar),
				nil,
				d2render.MustLoadSprite(d2resource.CharacterSelectAssassinSelected, d2resource.PaletteFechar),
				nil,
				d2render.MustLoadSprite(d2resource.CharacterSelectAssassinBackWalk, d2resource.PaletteFechar),
				nil,
				image.Rectangle{Min: image.Point{175, 235}, Max: image.Point{50, 180}},
				v.soundManager.LoadSoundEffect(d2resource.SFXAssassinSelect),
				v.soundManager.LoadSoundEffect(d2resource.SFXAssassinDeselect),
			}
			v.heroRenderInfo[d2enum.HeroAssassin].IdleSprite.SetPosition(231, 365)
			v.heroRenderInfo[d2enum.HeroAssassin].IdleSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroAssassin].IdleSprite.SetPlayLengthMs(2500)
			v.heroRenderInfo[d2enum.HeroAssassin].IdleSelectedSprite.SetPosition(231, 365)
			v.heroRenderInfo[d2enum.HeroAssassin].IdleSelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroAssassin].IdleSelectedSprite.SetPlayLengthMs(2500)
			v.heroRenderInfo[d2enum.HeroAssassin].ForwardWalkSprite.SetPosition(231, 365)
			v.heroRenderInfo[d2enum.HeroAssassin].ForwardWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroAssassin].ForwardWalkSprite.SetPlayLengthMs(3800)
			v.heroRenderInfo[d2enum.HeroAssassin].ForwardWalkSprite.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroAssassin].SelectedSprite.SetPosition(231, 365)
			v.heroRenderInfo[d2enum.HeroAssassin].SelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroAssassin].SelectedSprite.SetPlayLengthMs(2500)
			v.heroRenderInfo[d2enum.HeroAssassin].BackWalkSprite.SetPosition(231, 365)
			v.heroRenderInfo[d2enum.HeroAssassin].BackWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroAssassin].BackWalkSprite.SetPlayLengthMs(1500)
			v.heroRenderInfo[d2enum.HeroAssassin].BackWalkSprite.SetPlayLoop(false)
		},
		func() {
			v.heroRenderInfo[d2enum.HeroDruid] = &HeroRenderInfo{
				d2enum.HeroStanceIdle,
				d2render.MustLoadSprite(d2resource.CharacterSelectDruidUnselected, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelectDruidUnselectedH, d2resource.PaletteFechar),
				d2render.MustLoadSprite(d2resource.CharacterSelectDruidForwardWalk, d2resource.PaletteFechar),
				nil,
				d2render.MustLoadSprite(d2resource.CharacterSelectDruidSelected, d2resource.PaletteFechar),
				nil,
				d2render.MustLoadSprite(d2resource.CharacterSelectDruidBackWalk, d2resource.PaletteFechar),
				nil,
				image.Rectangle{Min: image.Point{680, 220}, Max: image.Point{70, 195}},
				v.soundManager.LoadSoundEffect(d2resource.SFXDruidSelect),
				v.soundManager.LoadSoundEffect(d2resource.SFXDruidDeselect),
			}
			v.heroRenderInfo[d2enum.HeroDruid].IdleSprite.SetPosition(720, 370)
			v.heroRenderInfo[d2enum.HeroDruid].IdleSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroDruid].IdleSprite.SetPlayLengthMs(1500)
			v.heroRenderInfo[d2enum.HeroDruid].IdleSelectedSprite.SetPosition(720, 370)
			v.heroRenderInfo[d2enum.HeroDruid].IdleSelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroDruid].IdleSelectedSprite.SetPlayLengthMs(1500)
			v.heroRenderInfo[d2enum.HeroDruid].ForwardWalkSprite.SetPosition(720, 370)
			v.heroRenderInfo[d2enum.HeroDruid].ForwardWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroDruid].ForwardWalkSprite.SetPlayLengthMs(4800)
			v.heroRenderInfo[d2enum.HeroDruid].ForwardWalkSprite.SetPlayLoop(false)
			v.heroRenderInfo[d2enum.HeroDruid].SelectedSprite.SetPosition(720, 370)
			v.heroRenderInfo[d2enum.HeroDruid].SelectedSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroDruid].SelectedSprite.SetPlayLengthMs(1500)
			v.heroRenderInfo[d2enum.HeroDruid].BackWalkSprite.SetPosition(720, 370)
			v.heroRenderInfo[d2enum.HeroDruid].BackWalkSprite.PlayForward()
			v.heroRenderInfo[d2enum.HeroDruid].BackWalkSprite.SetPlayLengthMs(1500)
			v.heroRenderInfo[d2enum.HeroDruid].BackWalkSprite.SetPlayLoop(false)
		},
	}
}

func (v *SelectHeroClass) Unload() {
	for i := range v.heroRenderInfo {
		v.heroRenderInfo[i].SelectSfx.Stop()
		v.heroRenderInfo[i].DeselectSfx.Stop()
	}
	v.heroRenderInfo = nil
}

func (v SelectHeroClass) onExitButtonClicked() {
	v.sceneProvider.SetNextScene(CreateCharacterSelect(v.sceneProvider, v.uiManager, v.soundManager))
}

func (v SelectHeroClass) onOkButtonClicked() {
	gameState := d2core.CreateGameState(v.heroNameTextbox.GetText(), v.selectedHero, v.hardcoreCheckbox.GetCheckState())
	v.sceneProvider.SetNextScene(CreateGame(v.sceneProvider, v.uiManager, v.soundManager, gameState))
}

func (v *SelectHeroClass) Render(screen *ebiten.Image) {
	v.bgImage.RenderSegmented(screen, 4, 3, 0)
	v.headingLabel.Render(screen)
	if v.selectedHero != d2enum.HeroNone {
		v.heroClassLabel.Render(screen)
		v.heroDesc1Label.Render(screen)
		v.heroDesc2Label.Render(screen)
		v.heroDesc3Label.Render(screen)
	}
	for heroClass, heroInfo := range v.heroRenderInfo {
		if heroInfo.Stance == d2enum.HeroStanceIdle || heroInfo.Stance == d2enum.HeroStanceIdleSelected {
			v.renderHero(screen, heroClass)
		}
	}
	for heroClass, heroInfo := range v.heroRenderInfo {
		if heroInfo.Stance != d2enum.HeroStanceIdle && heroInfo.Stance != d2enum.HeroStanceIdleSelected {
			v.renderHero(screen, heroClass)
		}
	}
	v.campfire.Render(screen)
	if v.heroNameTextbox.GetVisible() {
		v.heroNameLabel.Render(screen)
		v.expansionCharLabel.Render(screen)
		v.hardcoreCharLabel.Render(screen)
	}
}

func (v *SelectHeroClass) Update(tickTime float64) {
	canSelect := true
	for _, info := range v.heroRenderInfo {
		if info.Stance != d2enum.HeroStanceIdle && info.Stance != d2enum.HeroStanceIdleSelected && info.Stance != d2enum.HeroStanceSelected {
			canSelect = false
			break
		}
	}
	allIdle := true
	for heroType, data := range v.heroRenderInfo {
		if allIdle && data.Stance != d2enum.HeroStanceIdle {
			allIdle = false
		}
		v.updateHeroSelectionHover(heroType, canSelect)
	}
	if v.selectedHero != d2enum.HeroNone && allIdle {
		v.selectedHero = d2enum.HeroNone
	}
	v.heroNameTextbox.Update()
	v.okButton.SetEnabled(len(v.heroNameTextbox.GetText()) >= 2)
}

func (v *SelectHeroClass) updateHeroSelectionHover(hero d2enum.Hero, canSelect bool) {
	renderInfo := v.heroRenderInfo[hero]
	switch renderInfo.Stance {
	case d2enum.HeroStanceApproaching:
		if renderInfo.ForwardWalkSprite.IsOnLastFrame() {
			renderInfo.Stance = d2enum.HeroStanceSelected
			setSpriteToFirstFrame(renderInfo.SelectedSprite)
			setSpriteToFirstFrame(renderInfo.SelectedSpriteOverlay)
		}
		return
	case d2enum.HeroStanceRetreating:
		if renderInfo.BackWalkSprite.IsOnLastFrame() {
			renderInfo.Stance = d2enum.HeroStanceIdle
			setSpriteToFirstFrame(renderInfo.IdleSprite)
		}
		return
	}
	if !canSelect {
		return
	}
	if renderInfo.Stance == d2enum.HeroStanceSelected {
		return
	}
	mouseX := v.uiManager.CursorX
	mouseY := v.uiManager.CursorY
	b := renderInfo.SelectionBounds
	mouseHover := (mouseX >= b.Min.X) && (mouseX <= b.Min.X+b.Max.X) && (mouseY >= b.Min.Y) && (mouseY <= b.Min.Y+b.Max.Y)
	if mouseHover && v.uiManager.CursorButtonPressed(d2ui.CursorButtonLeft) {
		v.heroNameTextbox.SetVisible(true)
		v.okButton.SetVisible(true)
		v.expansionCheckbox.SetVisible(true)
		v.hardcoreCheckbox.SetVisible(true)
		renderInfo.Stance = d2enum.HeroStanceApproaching
		setSpriteToFirstFrame(renderInfo.ForwardWalkSprite)
		setSpriteToFirstFrame(renderInfo.ForwardWalkSpriteOverlay)
		for _, heroInfo := range v.heroRenderInfo {
			if heroInfo.Stance != d2enum.HeroStanceSelected {
				continue
			}
			heroInfo.SelectSfx.Stop()
			heroInfo.DeselectSfx.Play()
			heroInfo.Stance = d2enum.HeroStanceRetreating
			setSpriteToFirstFrame(heroInfo.BackWalkSprite)
			setSpriteToFirstFrame(heroInfo.BackWalkSpriteOverlay)
		}
		v.selectedHero = hero
		v.updateHeroText()
		renderInfo.SelectSfx.Play()

		return
	}

	if mouseHover && renderInfo.Stance != d2enum.HeroStanceIdleSelected {
		renderInfo.IdleSelectedSprite.SetCurrentFrame(renderInfo.IdleSprite.GetCurrentFrame())
		renderInfo.Stance = d2enum.HeroStanceIdleSelected
	} else if !mouseHover && renderInfo.Stance != d2enum.HeroStanceIdle {
		renderInfo.IdleSprite.SetCurrentFrame(renderInfo.IdleSelectedSprite.GetCurrentFrame())
		renderInfo.Stance = d2enum.HeroStanceIdle
	}

	if v.selectedHero == d2enum.HeroNone && mouseHover {
		v.selectedHero = hero
		v.updateHeroText()
	}

}

func (v *SelectHeroClass) renderHero(screen *ebiten.Image, hero d2enum.Hero) {
	renderInfo := v.heroRenderInfo[hero]
	switch renderInfo.Stance {
	case d2enum.HeroStanceIdle:
		drawSprite(renderInfo.IdleSprite, screen)
	case d2enum.HeroStanceIdleSelected:
		drawSprite(renderInfo.IdleSelectedSprite, screen)
	case d2enum.HeroStanceApproaching:
		drawSprite(renderInfo.ForwardWalkSprite, screen)
		drawSprite(renderInfo.ForwardWalkSpriteOverlay, screen)
	case d2enum.HeroStanceSelected:
		drawSprite(renderInfo.SelectedSprite, screen)
		drawSprite(renderInfo.SelectedSpriteOverlay, screen)
	case d2enum.HeroStanceRetreating:
		drawSprite(renderInfo.BackWalkSprite, screen)
		drawSprite(renderInfo.BackWalkSpriteOverlay, screen)
	}
}

func (v *SelectHeroClass) updateHeroText() {
	switch v.selectedHero {
	case d2enum.HeroNone:
		return
	case d2enum.HeroBarbarian:
		v.heroClassLabel.SetText(d2common.TranslateString("partycharbar"))
		v.setDescLabels("#1709")
	case d2enum.HeroNecromancer:
		v.heroClassLabel.SetText(d2common.TranslateString("partycharnec"))
		v.setDescLabels("#1704")
	case d2enum.HeroPaladin:
		v.heroClassLabel.SetText(d2common.TranslateString("partycharpal"))
		v.setDescLabels("#1711")
	case d2enum.HeroAssassin:
		v.heroClassLabel.SetText(d2common.TranslateString("partycharass"))
		v.setDescLabels("#305")
	case d2enum.HeroSorceress:
		v.heroClassLabel.SetText(d2common.TranslateString("partycharsor"))
		v.setDescLabels("#1710")
	case d2enum.HeroAmazon:
		v.heroClassLabel.SetText(d2common.TranslateString("partycharama"))
		v.setDescLabels("#1698")
	case d2enum.HeroDruid:
		v.heroClassLabel.SetText(d2common.TranslateString("partychardru"))
		v.setDescLabels("#304")
	}
	/*
	   if (selectedHero == null)
	                   return;

	               switch (selectedHero.Value)
	               {

	               }

	               heroClassLabel.Location = new Point(400 - (heroClassLabel.TextArea.Width / 2), 65);
	               heroDesc1Label.Location = new Point(400 - (heroDesc1Label.TextArea.Width / 2), 100);
	               heroDesc2Label.Location = new Point(400 - (heroDesc2Label.TextArea.Width / 2), 115);
	               heroDesc3Label.Location = new Point(400 - (heroDesc3Label.TextArea.Width / 2), 130);
	*/
}

func (v *SelectHeroClass) setDescLabels(descKey string) {
	heroDesc := d2common.TranslateString(descKey)
	parts := dh.SplitIntoLinesWithMaxWidth(heroDesc, 37)
	if len(parts) > 1 {
		v.heroDesc1Label.SetText(parts[0])
	} else {
		v.heroDesc1Label.SetText("")
	}
	if len(parts) > 1 {
		v.heroDesc2Label.SetText(parts[1])
	} else {
		v.heroDesc2Label.SetText("")
	}
	if len(parts) > 2 {
		v.heroDesc3Label.SetText(parts[2])
	} else {
		v.heroDesc3Label.SetText("")
	}
}

func setSpriteToFirstFrame(sprite *d2render.Sprite) {
	if sprite != nil {
		sprite.Rewind()
	}
}

func drawSprite(sprite *d2render.Sprite, target *ebiten.Image) {
	if sprite != nil {
		sprite.Render(target)
	}
}