ebiten/examples/mandelbrot/main.go
2022-08-29 11:17:17 +09:00

110 lines
2.4 KiB
Go

// Copyright 2017 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.
//go:build example
// +build example
package main
import (
"log"
"math"
"github.com/hajimehoshi/ebiten/v2"
)
const (
screenWidth = 640
screenHeight = 640
maxIt = 128
)
var (
palette [maxIt]byte
)
func init() {
for i := range palette {
palette[i] = byte(math.Sqrt(float64(i)/float64(len(palette))) * 0x80)
}
}
func color(it int) (r, g, b byte) {
if it == maxIt {
return 0xff, 0xff, 0xff
}
c := palette[it]
return c, c, c
}
type Game struct {
offscreen *ebiten.Image
offscreenPix []byte
}
func NewGame() *Game {
g := &Game{
offscreen: ebiten.NewImage(screenWidth, screenHeight),
offscreenPix: make([]byte, screenWidth*screenHeight*4),
}
// Now it is not feasible to call updateOffscreen every frame due to performance.
g.updateOffscreen(-0.75, 0.25, 2)
return g
}
func (gm *Game) updateOffscreen(centerX, centerY, size float64) {
for j := 0; j < screenHeight; j++ {
for i := 0; i < screenHeight; i++ {
x := float64(i)*size/screenWidth - size/2 + centerX
y := (screenHeight-float64(j))*size/screenHeight - size/2 + centerY
c := complex(x, y)
z := complex(0, 0)
it := 0
for ; it < maxIt; it++ {
z = z*z + c
if real(z)*real(z)+imag(z)*imag(z) > 4 {
break
}
}
r, g, b := color(it)
p := 4 * (i + j*screenWidth)
gm.offscreenPix[p] = r
gm.offscreenPix[p+1] = g
gm.offscreenPix[p+2] = b
gm.offscreenPix[p+3] = 0xff
}
}
gm.offscreen.WritePixels(gm.offscreenPix)
}
func (g *Game) Update() error {
return nil
}
func (g *Game) Draw(screen *ebiten.Image) {
screen.DrawImage(g.offscreen, nil)
}
func (g *Game) Layout(outsideWidth, outsideHeight int) (int, int) {
return screenWidth, screenHeight
}
func main() {
ebiten.SetWindowSize(screenWidth, screenHeight)
ebiten.SetWindowTitle("Mandelbrot (Ebitengine Demo)")
if err := ebiten.RunGame(NewGame()); err != nil {
log.Fatal(err)
}
}