mirror of
https://github.com/Dadido3/noita-mapcap.git
synced 2024-10-16 08:17:32 +00:00
David Vogel
b1a10870c1
- Add compatibility for newest Noita beta - Modify STREAMING_CHUNK_TARGET, GRID_MAX_UPDATES_PER_FRAME and GRID_MIN_UPDATES_PER_FRAME magic numbers for a more robust capturing process - Add LimitGroup to util.go - Add webp-level command line flag to define the webp compression level - Rework progress bar to make it work in DZI export mode - Refactor image exporter functions - Use LimitGroup to make DZI export multithreaded - Add BlendMethodFast which doesn't mix tile pixels - Up Go version to 1.22 - Use Dadido3/go-libwebp for WebP encoding
68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
// Copyright (c) 2023-2024 David Vogel
|
|
//
|
|
// This software is released under the MIT License.
|
|
// https://opensource.org/licenses/MIT
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"image"
|
|
"image/png"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/cheggaaa/pb/v3"
|
|
)
|
|
|
|
func exportPNGStitchedImage(stitchedImage *StitchedImage, outputPath string, bar *pb.ProgressBar) error {
|
|
log.Printf("Creating output file %q.", outputPath)
|
|
|
|
// If there is a progress bar, start a goroutine that regularly updates it.
|
|
// We will base the progress on the number of pixels read from the stitched image.
|
|
if bar != nil {
|
|
_, max := stitchedImage.Progress()
|
|
bar.SetRefreshRate(250 * time.Millisecond).SetTotal(int64(max)).Start()
|
|
|
|
done := make(chan struct{})
|
|
defer func() {
|
|
done <- struct{}{}
|
|
bar.SetCurrent(bar.Total()).Finish()
|
|
}()
|
|
|
|
go func() {
|
|
ticker := time.NewTicker(250 * time.Millisecond)
|
|
for {
|
|
select {
|
|
case <-done:
|
|
return
|
|
case <-ticker.C:
|
|
value, max := stitchedImage.Progress()
|
|
bar.SetCurrent(int64(value)).SetTotal(int64(max))
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
return exportPNG(stitchedImage, outputPath)
|
|
}
|
|
|
|
func exportPNG(img image.Image, outputPath string) error {
|
|
f, err := os.Create(outputPath)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create file: %w", err)
|
|
}
|
|
defer f.Close()
|
|
|
|
encoder := png.Encoder{
|
|
CompressionLevel: png.DefaultCompression,
|
|
}
|
|
|
|
if err := encoder.Encode(f, img); err != nil {
|
|
return fmt.Errorf("failed to encode image %q: %w", outputPath, err)
|
|
}
|
|
|
|
return nil
|
|
}
|