-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add
-fire option to fps for a fire effect and i key to toggle i…
…nfo text (#69) * Add `-fire` option to `fps` for a fire effect * make linters happy * ansi 256 color support * Better yet fewer colors out of the 216 (12 colors) * Single rand now * Fixing instant fps calc in -fire mode was missing... the fire render * make linters happy * Show palette at start * slowdown the decay/off mode * Take a tiny bit more vertical space with the fire height * extend red/orange zone and shrink white zone in truecolor palette * Added "i" key to toggle info/text display * linter
- Loading branch information
Showing
3 changed files
with
211 additions
and
28 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,152 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "math/rand/v2" | ||
|
|
||
| "fortio.org/safecast" | ||
| "fortio.org/terminal/ansipixels" | ||
| ) | ||
|
|
||
| type FireState struct { | ||
| h int | ||
| w int | ||
| buffer []byte | ||
| on bool | ||
| } | ||
|
|
||
| var fire *FireState | ||
|
|
||
| var ( | ||
| v2colTrueColor [256]string | ||
| v2col256 [256]string | ||
| ) | ||
|
|
||
| func init() { | ||
| for i := range 256 { | ||
| r := min(255, 3*i) | ||
| g := min(255, max(0, (i-84)*2)) | ||
| b := min(255, max(0, (i-208)*5)) | ||
| v2colTrueColor[i] = fmt.Sprintf("\033[38;2;%d;%d;%dm█", r, g, b) | ||
| } | ||
| // 0 1 2 3 4 5 6 7 8 9 10 11 | ||
| for i, color := range []int{16, 52, 88, 124, 166, 202, 208, 214, 220, 226, 228, 231} { | ||
| v2col256[i] = fmt.Sprintf("\033[38;5;%dm█", color) | ||
| } | ||
| } | ||
|
|
||
| func InitFire(ap *ansipixels.AnsiPixels) *FireState { | ||
| f := &FireState{h: ap.H - 2*ap.Margin, w: ap.W - 2*ap.Margin} | ||
| f.buffer = make([]byte, f.h*f.w) | ||
| return f | ||
| } | ||
|
|
||
| func ToggleFire() { | ||
| if fire == nil { | ||
| return | ||
| } | ||
| if fire.on { | ||
| fire.Off() | ||
| } else { | ||
| fire.Start() | ||
| } | ||
| } | ||
|
|
||
| func (f *FireState) At(x, y int) byte { | ||
| return f.buffer[y*f.w+x] | ||
| } | ||
|
|
||
| func (f *FireState) Set(x, y int, v byte) { | ||
| f.buffer[y*f.w+x] = v | ||
| } | ||
|
|
||
| func (f *FireState) Start() { | ||
| for x := range f.w { | ||
| f.Set(x, f.h-1, 255) | ||
| } | ||
| f.on = true | ||
| } | ||
|
|
||
| // Turn off the fire at the bottom. | ||
| func (f *FireState) Off() { | ||
| for x := range f.w { | ||
| f.Set(x, f.h-1, 1) | ||
| } | ||
| f.on = false | ||
| } | ||
|
|
||
| func (f *FireState) Update() { | ||
| for y := f.h - 2; y >= 0; y-- { | ||
| for x := range f.w { | ||
| r := rand.Float32() //nolint:gosec // this _is_ randv2! | ||
| dx := safecast.MustTruncate[int](3*r - 1.5) // -1, 0, 1 | ||
| v := f.At((x+dx+f.w)%f.w, y+1) | ||
| pv := f.At(x, y) | ||
| if pv > v { // slow-ish decay when "off" | ||
| delta := max(1, byte(r*float32(pv-v))) | ||
| v = max(1, pv-delta) | ||
| f.Set(x, y, v) | ||
| continue | ||
| } | ||
| newV := byte(max(0, float32(v)-r*3.2*255./(float32(f.h-1)))) | ||
| if newV == 0 && pv != 0 { | ||
| newV = 1 | ||
| } | ||
| f.Set(x, y, newV) | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func (f *FireState) Render(ap *ansipixels.AnsiPixels) { | ||
| for y := range f.h { | ||
| first := true | ||
| prevX := -999 | ||
| prevColor := "" | ||
| for x := range f.w { | ||
| v := f.At(x, y) | ||
| if v == 0 { | ||
| continue | ||
| } | ||
| switch { | ||
| case first: | ||
| ap.MoveCursor(x+ap.Margin, y+ap.Margin) | ||
| first = false | ||
| case x != prevX+1: | ||
| ap.MoveHorizontally(x + ap.Margin) | ||
| } | ||
| prevX = x | ||
| var newColor string | ||
| if ap.TrueColor { | ||
| newColor = v2colTrueColor[v] | ||
| } else { | ||
| newColor = v2col256[3*int(v)/64] | ||
| } | ||
| if newColor != prevColor { | ||
| ap.WriteString(newColor) | ||
| prevColor = newColor | ||
| } else { | ||
| ap.WriteRune(ansipixels.FullPixel) | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func AnimateFire(ap *ansipixels.AnsiPixels, frame int64) { | ||
| if frame == 0 { | ||
| fire = InitFire(ap) | ||
| fire.Start() | ||
| } | ||
| fire.Update() | ||
| fire.Render(ap) | ||
| } | ||
|
|
||
| func ShowPalette(ap *ansipixels.AnsiPixels) { | ||
| f := InitFire(ap) | ||
| // Show/debug the palette: | ||
| for x := range f.w { | ||
| v := safecast.MustConvert[byte]((255 * (x + 1)) / f.w) | ||
| f.Set(x, f.h-3, v) | ||
| f.Set(x, f.h-2, v) | ||
| } | ||
| f.Render(ap) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters