mirror of
https://github.com/hajimehoshi/ebiten.git
synced 2024-12-27 04:08:53 +01:00
214 lines
4.5 KiB
Go
214 lines
4.5 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"
|
||
|
"image"
|
||
|
"io/ioutil"
|
||
|
"os"
|
||
|
"path/filepath"
|
||
|
"time"
|
||
|
|
||
|
"github.com/hajimehoshi/ebiten/internal/png"
|
||
|
"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 {
|
||
|
dump := func() (string, error) {
|
||
|
f, err := ioutil.TempFile("", "")
|
||
|
if err != nil {
|
||
|
return "", err
|
||
|
}
|
||
|
defer f.Close()
|
||
|
|
||
|
if err := png.Encode(f, screen); err != nil {
|
||
|
return "", err
|
||
|
}
|
||
|
return f.Name(), nil
|
||
|
}
|
||
|
|
||
|
name, err := dump()
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
newname, err := availableFilename("screenshot_", ".png")
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
if err := os.Rename(name, 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
|
||
|
}
|
||
|
|
||
|
dump := func(img image.Image, index int) error {
|
||
|
filename := filepath.Join(dir, fmt.Sprintf("%d.png", index))
|
||
|
f, err := os.Create(filename)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
defer f.Close()
|
||
|
|
||
|
if err := png.Encode(f, img); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
for i, img := range shareable.Images() {
|
||
|
if err := dump(img, i); 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
|
||
|
}
|