ebiten/examples/raycasting/main.go

316 lines
8.0 KiB
Go
Raw Normal View History

2019-01-12 15:46:03 +01:00
// Copyright 2019 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.
package main
import (
"bytes"
"errors"
"fmt"
"image"
"image/color"
_ "image/png"
"log"
"math"
"sort"
2020-10-03 19:35:13 +02:00
"github.com/hajimehoshi/ebiten/v2"
"github.com/hajimehoshi/ebiten/v2/ebitenutil"
"github.com/hajimehoshi/ebiten/v2/examples/resources/images"
"github.com/hajimehoshi/ebiten/v2/inpututil"
"github.com/hajimehoshi/ebiten/v2/vector"
2019-01-12 15:46:03 +01:00
)
const (
screenWidth = 240
screenHeight = 240
2019-02-17 17:26:31 +01:00
padding = 20
2019-01-12 15:46:03 +01:00
)
var (
bgImage *ebiten.Image
shadowImage = ebiten.NewImage(screenWidth, screenHeight)
triangleImage = ebiten.NewImage(screenWidth, screenHeight)
2019-01-12 15:46:03 +01:00
)
func init() {
// Decode an image from the image file's byte slice.
2019-01-12 15:46:03 +01:00
img, _, err := image.Decode(bytes.NewReader(images.Tile_png))
if err != nil {
log.Fatal(err)
}
bgImage = ebiten.NewImageFromImage(img)
2019-01-12 15:46:03 +01:00
triangleImage.Fill(color.White)
}
2019-02-17 17:26:31 +01:00
type line struct {
2019-01-12 15:46:03 +01:00
X1, Y1, X2, Y2 float64
}
2019-02-17 17:26:31 +01:00
func (l *line) angle() float64 {
2019-01-12 15:54:15 +01:00
return math.Atan2(l.Y2-l.Y1, l.X2-l.X1)
}
2019-02-17 17:26:31 +01:00
type object struct {
walls []line
}
func (o object) points() [][2]float64 {
// Get one of the endpoints for all segments,
// + the startpoint of the first one, for non-closed paths
var points [][2]float64
for _, wall := range o.walls {
points = append(points, [2]float64{wall.X2, wall.Y2})
}
p := [2]float64{o.walls[0].X1, o.walls[0].Y1}
if p[0] != points[len(points)-1][0] && p[1] != points[len(points)-1][1] {
points = append(points, [2]float64{o.walls[0].X1, o.walls[0].Y1})
}
2019-02-17 17:26:31 +01:00
return points
}
func newRay(x, y, length, angle float64) line {
return line{
2019-01-12 15:46:03 +01:00
X1: x,
Y1: y,
X2: x + length*math.Cos(angle),
Y2: y + length*math.Sin(angle),
}
}
// intersection calculates the intersection of given two lines.
2019-02-17 17:26:31 +01:00
func intersection(l1, l2 line) (float64, float64, bool) {
2019-01-12 15:54:15 +01:00
// https://en.wikipedia.org/wiki/Line%E2%80%93line_intersection#Given_two_points_on_each_line
2019-01-12 15:46:03 +01:00
denom := (l1.X1-l1.X2)*(l2.Y1-l2.Y2) - (l1.Y1-l1.Y2)*(l2.X1-l2.X2)
tNum := (l1.X1-l2.X1)*(l2.Y1-l2.Y2) - (l1.Y1-l2.Y1)*(l2.X1-l2.X2)
uNum := -((l1.X1-l1.X2)*(l1.Y1-l2.Y1) - (l1.Y1-l1.Y2)*(l1.X1-l2.X1))
if denom == 0 {
2019-02-17 17:26:31 +01:00
return 0, 0, false
2019-01-12 15:46:03 +01:00
}
t := tNum / denom
if t > 1 || t < 0 {
2019-02-17 17:26:31 +01:00
return 0, 0, false
2019-01-12 15:46:03 +01:00
}
u := uNum / denom
if u > 1 || u < 0 {
2019-02-17 17:26:31 +01:00
return 0, 0, false
2019-01-12 15:46:03 +01:00
}
x := l1.X1 + t*(l1.X2-l1.X1)
y := l1.Y1 + t*(l1.Y2-l1.Y1)
2019-02-17 17:26:31 +01:00
return x, y, true
2019-01-12 15:46:03 +01:00
}
2019-02-17 17:26:31 +01:00
// rayCasting returns a slice of line originating from point cx, cy and intersecting with objects
func rayCasting(cx, cy float64, objects []object) []line {
2019-01-12 15:46:03 +01:00
const rayLength = 1000 // something large enough to reach all objects
2019-02-17 17:26:31 +01:00
var rays []line
for _, obj := range objects {
2019-01-12 15:46:03 +01:00
// Cast two rays per point
2019-02-17 17:26:31 +01:00
for _, p := range obj.points() {
l := line{cx, cy, p[0], p[1]}
2019-01-12 15:54:15 +01:00
angle := l.angle()
2019-01-12 15:46:03 +01:00
for _, offset := range []float64{-0.005, 0.005} {
points := [][2]float64{}
ray := newRay(cx, cy, rayLength, angle+offset)
// Unpack all objects
for _, o := range objects {
2019-02-17 17:26:31 +01:00
for _, wall := range o.walls {
if px, py, ok := intersection(ray, wall); ok {
2019-01-12 15:46:03 +01:00
points = append(points, [2]float64{px, py})
}
}
}
// Find the point closest to start of ray
min := math.Inf(1)
2019-02-17 17:26:31 +01:00
minI := -1
2019-01-12 15:46:03 +01:00
for i, p := range points {
d2 := (cx-p[0])*(cx-p[0]) + (cy-p[1])*(cy-p[1])
if d2 < min {
min = d2
minI = i
}
}
2019-02-17 17:26:31 +01:00
rays = append(rays, line{cx, cy, points[minI][0], points[minI][1]})
2019-01-12 15:46:03 +01:00
}
}
}
// Sort rays based on angle, otherwise light triangles will not come out right
sort.Slice(rays, func(i int, j int) bool {
2019-01-12 15:54:15 +01:00
return rays[i].angle() < rays[j].angle()
2019-01-12 15:46:03 +01:00
})
return rays
}
func (g *Game) handleMovement() {
if ebiten.IsKeyPressed(ebiten.KeyD) || ebiten.IsKeyPressed(ebiten.KeyArrowRight) {
g.px += 4
2019-01-12 15:46:03 +01:00
}
if ebiten.IsKeyPressed(ebiten.KeyS) || ebiten.IsKeyPressed(ebiten.KeyArrowDown) {
g.py += 4
2019-01-12 15:46:03 +01:00
}
if ebiten.IsKeyPressed(ebiten.KeyA) || ebiten.IsKeyPressed(ebiten.KeyArrowLeft) {
g.px -= 4
2019-01-12 15:46:03 +01:00
}
if ebiten.IsKeyPressed(ebiten.KeyW) || ebiten.IsKeyPressed(ebiten.KeyArrowUp) {
g.py -= 4
2019-01-12 15:46:03 +01:00
}
// +1/-1 is to stop player before it reaches the border
if g.px >= screenWidth-padding {
g.px = screenWidth - padding - 1
2019-01-12 15:46:03 +01:00
}
if g.px <= padding {
g.px = padding + 1
2019-01-12 15:46:03 +01:00
}
if g.py >= screenHeight-padding {
g.py = screenHeight - padding - 1
2019-01-12 15:46:03 +01:00
}
if g.py <= padding {
g.py = padding + 1
2019-01-12 15:46:03 +01:00
}
}
2019-02-17 17:26:31 +01:00
func rayVertices(x1, y1, x2, y2, x3, y3 float64) []ebiten.Vertex {
return []ebiten.Vertex{
{DstX: float32(x1), DstY: float32(y1), SrcX: 0, SrcY: 0, ColorR: 1, ColorG: 1, ColorB: 1, ColorA: 1},
{DstX: float32(x2), DstY: float32(y2), SrcX: 0, SrcY: 0, ColorR: 1, ColorG: 1, ColorB: 1, ColorA: 1},
{DstX: float32(x3), DstY: float32(y3), SrcX: 0, SrcY: 0, ColorR: 1, ColorG: 1, ColorB: 1, ColorA: 1},
2019-02-17 17:26:31 +01:00
}
}
type Game struct {
showRays bool
px, py int
objects []object
}
func (g *Game) Update() error {
2019-01-12 15:46:03 +01:00
if inpututil.IsKeyJustPressed(ebiten.KeyEscape) {
return errors.New("game ended by player")
}
if inpututil.IsKeyJustPressed(ebiten.KeyR) {
g.showRays = !g.showRays
2019-01-12 15:46:03 +01:00
}
g.handleMovement()
2019-01-12 15:46:03 +01:00
return nil
}
2019-01-12 15:46:03 +01:00
func (g *Game) Draw(screen *ebiten.Image) {
2019-01-12 15:46:03 +01:00
// Reset the shadowImage
shadowImage.Fill(color.Black)
rays := rayCasting(float64(g.px), float64(g.py), g.objects)
2019-01-12 15:46:03 +01:00
// Subtract ray triangles from shadow
opt := &ebiten.DrawTrianglesOptions{}
opt.Address = ebiten.AddressRepeat
opt.Blend = ebiten.BlendSourceOut
2019-01-12 15:46:03 +01:00
for i, line := range rays {
nextLine := rays[(i+1)%len(rays)]
// Draw triangle of area between rays
v := rayVertices(float64(g.px), float64(g.py), nextLine.X2, nextLine.Y2, line.X2, line.Y2)
2019-01-12 15:46:03 +01:00
shadowImage.DrawTriangles(v, []uint16{0, 1, 2}, triangleImage, opt)
}
// Draw background
2019-02-17 17:26:31 +01:00
screen.DrawImage(bgImage, nil)
2019-01-12 15:46:03 +01:00
if g.showRays {
2019-01-12 15:46:03 +01:00
// Draw rays
for _, r := range rays {
vector.StrokeLine(screen, float32(r.X1), float32(r.Y1), float32(r.X2), float32(r.Y2), 1, color.RGBA{255, 255, 0, 150})
2019-01-12 15:46:03 +01:00
}
}
// Draw shadow
op := &ebiten.DrawImageOptions{}
op.ColorScale.ScaleAlpha(0.7)
2019-01-12 15:46:03 +01:00
screen.DrawImage(shadowImage, op)
// Draw walls
for _, obj := range g.objects {
2019-02-17 17:26:31 +01:00
for _, w := range obj.walls {
vector.StrokeLine(screen, float32(w.X1), float32(w.Y1), float32(w.X2), float32(w.Y2), 1, color.RGBA{255, 0, 0, 255})
2019-01-12 15:46:03 +01:00
}
}
// Draw player as a rect
vector.DrawFilledRect(screen, float32(g.px)-2, float32(g.py)-2, 4, 4, color.Black)
vector.DrawFilledRect(screen, float32(g.px)-1, float32(g.py)-1, 2, 2, color.RGBA{255, 100, 100, 255})
2019-01-12 15:46:03 +01:00
if g.showRays {
2019-01-12 15:46:03 +01:00
ebitenutil.DebugPrintAt(screen, "R: hide rays", padding, 0)
} else {
ebitenutil.DebugPrintAt(screen, "R: show rays", padding, 0)
}
ebitenutil.DebugPrintAt(screen, "WASD: move", 160, 0)
ebitenutil.DebugPrintAt(screen, fmt.Sprintf("TPS: %0.2f", ebiten.ActualTPS()), 51, 51)
ebitenutil.DebugPrintAt(screen, fmt.Sprintf("Rays: 2*%d", len(rays)/2), padding, 222)
2019-01-12 15:46:03 +01:00
}
func (g *Game) Layout(outsideWidth, outsideHeight int) (int, int) {
return screenWidth, screenHeight
}
2019-01-12 15:46:03 +01:00
2019-02-17 17:26:31 +01:00
func rect(x, y, w, h float64) []line {
return []line{
{x, y, x, y + h},
{x, y + h, x + w, y + h},
{x + w, y + h, x + w, y},
{x + w, y, x, y},
}
}
2019-01-12 15:46:03 +01:00
func main() {
g := &Game{
px: screenWidth / 2,
py: screenHeight / 2,
}
2019-01-12 15:46:03 +01:00
// Add outer walls
g.objects = append(g.objects, object{rect(padding, padding, screenWidth-2*padding, screenHeight-2*padding)})
2019-01-12 15:46:03 +01:00
// Angled wall
g.objects = append(g.objects, object{[]line{{50, 110, 100, 150}}})
2019-01-12 15:46:03 +01:00
// Rectangles
g.objects = append(g.objects, object{rect(45, 50, 70, 20)})
g.objects = append(g.objects, object{rect(150, 50, 30, 60)})
2019-01-12 15:46:03 +01:00
ebiten.SetWindowSize(screenWidth*2, screenHeight*2)
ebiten.SetWindowTitle("Ray casting and shadows (Ebitengine Demo)")
if err := ebiten.RunGame(g); err != nil {
2019-01-12 15:46:03 +01:00
log.Fatal(err)
}
}