OpenDiablo2/d2core/d2gui/manager.go

package d2gui

import (
	"image/color"
	"math"

	"github.com/OpenDiablo2/OpenDiablo2/d2common/d2resource"
	"github.com/OpenDiablo2/OpenDiablo2/d2core/d2asset"
	"github.com/OpenDiablo2/OpenDiablo2/d2core/d2input"
	"github.com/OpenDiablo2/OpenDiablo2/d2core/d2render"
)

type manager struct {
	Layout

	cursorAnim    *d2asset.Animation
	cursorX       int
	cursorY       int
	cursorVisible bool

	loadingAnim *d2asset.Animation
	loading     bool
}

func createGuiManager() (*manager, error) {
	cursorAnim, err := d2asset.LoadAnimation(d2resource.CursorDefault, d2resource.PaletteUnits)
	if err != nil {
		return nil, err
	}

	loadingAnim, err := d2asset.LoadAnimation(d2resource.LoadingScreen, d2resource.PaletteLoading)
	if err != nil {
		return nil, err
	}

	manager := &manager{
		cursorAnim:    cursorAnim,
		loadingAnim:   loadingAnim,
		cursorVisible: true,
	}

	if err := d2input.BindHandler(manager); err != nil {
		return nil, err
	}

	return manager, nil
}

func (m *manager) OnMouseButtonDown(event d2input.MouseEvent) bool {
	return m.Layout.onMouseButtonDown(event)
}

func (m *manager) OnMouseButtonUp(event d2input.MouseEvent) bool {
	return m.Layout.onMouseButtonUp(event)
}

func (m *manager) OnMouseMove(event d2input.MouseMoveEvent) bool {
	m.cursorX = event.X
	m.cursorY = event.Y

	return m.Layout.onMouseMove(event)
}

func (m *manager) render(target d2render.Surface) error {
	if m.loading {
		if err := m.renderLoadScreen(target); err != nil {
			return err
		}
	} else {
		if err := m.Layout.render(target); err != nil {
			return err
		}
	}

	if m.cursorVisible {
		if err := m.renderCursor(target); err != nil {
			return err
		}
	}

	return nil
}

func (m *manager) renderLoadScreen(target d2render.Surface) error {
	target.Clear(color.Black)

	screenWidth, screenHeight := target.GetSize()
	animWidth, animHeight := m.loadingAnim.GetCurrentFrameSize()
	target.PushTranslation(screenWidth/2-animWidth/2, screenHeight/2+animHeight/2)
	target.PushTranslation(0, -animHeight)
	defer target.PopN(2)

	return m.loadingAnim.Render(target)
}

func (m *manager) renderCursor(target d2render.Surface) error {
	_, height := m.cursorAnim.GetCurrentFrameSize()
	target.PushTranslation(m.cursorX, m.cursorY)
	target.PushTranslation(0, -height)
	defer target.PopN(2)

	return m.cursorAnim.Render(target)
}

func (m *manager) advance(elapsed float64) error {
	if !m.loading {
		if err := m.Layout.advance(elapsed); err != nil {
			return err
		}
	}

	return nil
}

func (m *manager) showLoadScreen(progress float64) {
	progress = math.Min(progress, 1.0)
	progress = math.Max(progress, 0.0)

	animation := m.loadingAnim
	frameCount := animation.GetFrameCount()
	animation.SetCurrentFrame(int(float64(frameCount-1.0) * progress))

	m.loading = true
}

func (m *manager) hideLoadScreen() {
	m.loading = false
}

func (m *manager) showCursor() {
	m.cursorVisible = true
}

func (m *manager) hideCursor() {
	m.cursorVisible = false
}

func (m *manager) clear() {
	m.Layout.clear()
	m.hideLoadScreen()
}