ebiten/imagedumper_desktop.go
2019-07-20 01:46:33 +09:00

176 lines
3.8 KiB
Go

// Copyright 2018 The Ebiten Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// +build !android
// +build !js
// +build !ios
package ebiten
import (
"fmt"
"os"
"time"
"github.com/hajimehoshi/ebiten/internal/shareable"
)
// availableFilename returns a filename that is valid as a new file or directory.
func availableFilename(prefix, postfix string) (string, error) {
const datetimeFormat = "20060102030405"
now := time.Now()
name := fmt.Sprintf("%s%s%s", prefix, now.Format(datetimeFormat), postfix)
for i := 1; ; i++ {
if _, err := os.Stat(name); err != nil {
if os.IsNotExist(err) {
break
}
if !os.IsNotExist(err) {
return "", err
}
}
name = fmt.Sprintf("%s%s_%d%s", prefix, now.Format(datetimeFormat), i, postfix)
}
return name, nil
}
func takeScreenshot(screen *Image) error {
newname, err := availableFilename("screenshot_", ".png")
if err != nil {
return err
}
if err := screen.mipmap.orig.Dump(newname); err != nil {
return err
}
if _, err := fmt.Fprintf(os.Stderr, "Saved screenshot: %s\n", newname); err != nil {
return err
}
return nil
}
func dumpInternalImages() error {
dir, err := availableFilename("internalimages_", "")
if err != nil {
return err
}
if err := os.Mkdir(dir, 0755); err != nil {
return err
}
if err := shareable.DumpImages(dir); err != nil {
return err
}
if _, err := fmt.Fprintf(os.Stderr, "Dumped the internal images at: %s\n", dir); err != nil {
return err
}
return nil
}
type imageDumper struct {
f func(screen *Image) error
keyState map[Key]int
hasScreenshotKey bool
screenshotKey Key
toTakeScreenshot bool
hasDumpInternalImagesKey bool
dumpInternalImagesKey Key
toDumpInternalImages bool
}
func (i *imageDumper) update(screen *Image) error {
const (
envScreenshotKey = "EBITEN_SCREENSHOT_KEY"
envInternalImagesKey = "EBITEN_INTERNAL_IMAGES_KEY"
)
if err := i.f(screen); err != nil {
return err
}
// If keyState is nil, all values are not initialized.
if i.keyState == nil {
i.keyState = map[Key]int{}
if keyname := os.Getenv(envScreenshotKey); keyname != "" {
if key, ok := keyNameToKey(keyname); ok {
i.hasScreenshotKey = true
i.screenshotKey = key
}
}
if keyname := os.Getenv(envInternalImagesKey); keyname != "" {
if isDebug() {
if key, ok := keyNameToKey(keyname); ok {
i.hasDumpInternalImagesKey = true
i.dumpInternalImagesKey = key
}
} else {
fmt.Fprintf(os.Stderr, "%s is disabled. Specify a build tag 'ebitendebug' to enable it.\n", envInternalImagesKey)
}
}
}
keys := map[Key]struct{}{}
if i.hasScreenshotKey {
keys[i.screenshotKey] = struct{}{}
}
if i.hasDumpInternalImagesKey {
keys[i.dumpInternalImagesKey] = struct{}{}
}
for key := range keys {
if IsKeyPressed(key) {
i.keyState[key]++
if i.keyState[key] == 1 {
if i.hasScreenshotKey && key == i.screenshotKey {
i.toTakeScreenshot = true
}
if i.hasDumpInternalImagesKey && key == i.dumpInternalImagesKey {
i.toDumpInternalImages = true
}
}
} else {
i.keyState[key] = 0
}
}
if IsDrawingSkipped() {
return nil
}
if i.toTakeScreenshot {
i.toTakeScreenshot = false
if err := takeScreenshot(screen); err != nil {
return err
}
}
if i.toDumpInternalImages {
i.toDumpInternalImages = false
if err := dumpInternalImages(); err != nil {
return err
}
}
return nil
}