132 lines
3.4 KiB
Go
132 lines
3.4 KiB
Go
package nori
|
|
|
|
import (
|
|
"fmt"
|
|
"git.tuxpa.in/a/zlog/log"
|
|
"github.com/disintegration/imaging"
|
|
"github.com/phrozen/blend"
|
|
"gitlab.com/gfxlabs/gfximg/apng"
|
|
"image"
|
|
"image/color"
|
|
"image/draw"
|
|
)
|
|
|
|
func copyImage(img image.Image) image.Image {
|
|
copied := image.NewNRGBA(img.Bounds())
|
|
draw.Draw(copied, img.Bounds(), img, img.Bounds().Min, draw.Over)
|
|
return copied
|
|
}
|
|
|
|
type RenderFunc = func(img image.Image, delay int)
|
|
|
|
func RenderAnimation(n *Nori, num int, f RenderFunc) error {
|
|
g := n.Gawi
|
|
if len(n.Animations[num].Frames) == 0 {
|
|
return fmt.Errorf("no frames found for animation")
|
|
}
|
|
images := make([]image.Image, 0, len(n.Animations[num].Frames))
|
|
durations := make([]uint32, 0, len(n.Animations[num].Frames))
|
|
for _, frame := range n.Animations[num].Frames {
|
|
planes := frame.Planes
|
|
var canvasRect image.Rectangle
|
|
for i, plane := range planes {
|
|
if int(plane.BitmapId) >= len(g.Images) {
|
|
log.Printf("could not find bitmap %d, only have %d", plane.BitmapId, len(g.Images))
|
|
}
|
|
bitmap := g.Images[plane.BitmapId]
|
|
pt := image.Pt(int(plane.PlaneX), int(plane.PlaneY)) // where to put the point
|
|
rc := bitmap.Img.Bounds().Add(pt) // translate rectangle to put in the global canvas
|
|
if i == 0 {
|
|
canvasRect = rc
|
|
} else {
|
|
canvasRect = canvasRect.Union(rc)
|
|
}
|
|
}
|
|
img := image.NewNRGBA(canvasRect)
|
|
for _, plane := range planes {
|
|
bitmap := g.Images[plane.BitmapId]
|
|
drawBitmap := bitmap.Img
|
|
transparent := false
|
|
//flipx
|
|
if plane.RenderFlag&1 != 0 {
|
|
if bitmap.Img == drawBitmap {
|
|
drawBitmap = copyImage(bitmap.Img)
|
|
}
|
|
drawBitmap = imaging.FlipH(drawBitmap)
|
|
}
|
|
//flipy
|
|
if plane.RenderFlag&2 != 0 {
|
|
if bitmap.Img == drawBitmap {
|
|
drawBitmap = copyImage(bitmap.Img)
|
|
}
|
|
drawBitmap = imaging.FlipV(drawBitmap)
|
|
}
|
|
// is transparent
|
|
if plane.RenderFlag&0x20 != 0 {
|
|
transparent = true
|
|
}
|
|
_ = transparent
|
|
pt := image.Pt(int(plane.PlaneX), int(plane.PlaneY)) // where to put the point
|
|
rc := drawBitmap.Bounds().Add(pt)
|
|
src := drawBitmap
|
|
switch plane.Blend {
|
|
case BlendMode_InvertMul, BlendMode_InvertMul5:
|
|
img = imaging.Invert(img)
|
|
fallthrough
|
|
case BlendMode_Alpha:
|
|
case BlendMode_Mul, BlendMode_Mul7:
|
|
blend.BlendImage(img, drawBitmap, blend.Multiply)
|
|
case BlendMode_Add, BlendMode_Add8:
|
|
blend.BlendImage(img, drawBitmap, blend.Add)
|
|
case BlendMode_None:
|
|
default:
|
|
return fmt.Errorf("unknown blend mode: %d", plane.Blend)
|
|
}
|
|
|
|
mask := (image.Image)(nil)
|
|
if transparent {
|
|
mask = &image.Uniform{C: color.RGBA{A: 160}}
|
|
}
|
|
|
|
draw.DrawMask(
|
|
img,
|
|
rc,
|
|
src,
|
|
src.Bounds().Min,
|
|
mask,
|
|
src.Bounds().Min,
|
|
draw.Over,
|
|
)
|
|
}
|
|
images = append(images, img)
|
|
durations = append(durations, frame.Duration)
|
|
}
|
|
or := images[0].Bounds()
|
|
for _, realFrame := range images {
|
|
or = realFrame.Bounds().Union(or)
|
|
}
|
|
if or.Size().X == 0 || or.Size().Y == 0 {
|
|
return fmt.Errorf("no frames found for animation")
|
|
}
|
|
for i, realFrame := range images {
|
|
//log.Info().Int("frame", num).Interface("rect", or).Interface("frame", realFrame.Bounds()).Msg("")
|
|
src := realFrame
|
|
big := image.NewNRGBA64(or)
|
|
draw.Draw(
|
|
big,
|
|
src.Bounds(),
|
|
src,
|
|
src.Bounds().Min,
|
|
draw.Src,
|
|
)
|
|
fr := apng.Frame{
|
|
Image: big,
|
|
DelayDenominator: 1000,
|
|
DelayNumerator: uint16(durations[i]),
|
|
}
|
|
f(big, int(fr.DelayNumerator))
|
|
}
|
|
return nil
|
|
|
|
}
|