OpenDiablo2/d2core/d2gui/sprite.go

package d2gui

import (
	"log"

	"github.com/OpenDiablo2/OpenDiablo2/d2common/d2enum"
	"github.com/OpenDiablo2/OpenDiablo2/d2common/d2interface"
	"github.com/OpenDiablo2/OpenDiablo2/d2core/d2asset"
)

// AnimationDirection is a the animation play direction
type AnimationDirection int

// AnimationDirection types
const (
	DirectionForward AnimationDirection = iota
	DirectionBackward
)

// static check that Sprite implements widget
var _ widget = &Sprite{}

// Sprite is an image
type Sprite struct {
	widgetBase

	segmentsX   int
	segmentsY   int
	frameOffset int

	animation d2interface.Animation
}

// AnimatedSprite is a sprite that has animation
type AnimatedSprite struct {
	*Sprite
}

func createSprite(imagePath, palettePath string, assetManager *d2asset.AssetManager) (*Sprite, error) {
	animation, err := assetManager.LoadAnimation(imagePath, palettePath)
	if err != nil {
		return nil, err
	}

	sprite := &Sprite{}
	sprite.animation = animation
	sprite.SetVisible(true)

	return sprite, nil
}

func createAnimatedSprite(
	imagePath, palettePath string,
	direction AnimationDirection,
	assetManager *d2asset.AssetManager,
) (*AnimatedSprite, error) {
	animation, err := assetManager.LoadAnimation(imagePath, palettePath)
	if err != nil {
		return nil, err
	}

	sprite := &AnimatedSprite{Sprite: &Sprite{animation: animation}}

	if direction == DirectionForward {
		sprite.animation.PlayForward()
	} else {
		sprite.animation.PlayBackward()
	}

	sprite.SetVisible(true)

	return sprite, nil
}

func (s *AnimatedSprite) render(target d2interface.Surface) {
	_, frameHeight := s.animation.GetCurrentFrameSize()

	target.PushTranslation(s.x, s.y-frameHeight)
	defer target.Pop()

	s.animation.Render(target)
}

// SetSegmented sets the segment properties of the sprite
func (s *Sprite) SetSegmented(segmentsX, segmentsY, frameOffset int) {
	s.segmentsX = segmentsX
	s.segmentsY = segmentsY
	s.frameOffset = frameOffset
}

func (s *Sprite) render(target d2interface.Surface) {
	err := renderSegmented(s.animation, s.segmentsX, s.segmentsY, s.frameOffset, target)
	if err != nil {
		log.Println(err)
	}
}

func (s *Sprite) advance(elapsed float64) error {
	return s.animation.Advance(elapsed)
}

func (s *Sprite) getSize() (width, height int) {
	return s.animation.GetCurrentFrameSize()
}

// SetEffect sets the draw effect for the sprite
func (s *Sprite) SetEffect(e d2enum.DrawEffect) {
	s.animation.SetEffect(e)
}