milkbucket/cmd/window.go

/*
Copyright © 2024 filifa

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package cmd

/*
#cgo LDFLAGS: -lprojectM-4
#include "projectM-4/projectM.h"
#include <stdlib.h>
*/
import "C"

import (
	"container/ring"
	"encoding/binary"
	"errors"
	"io"
	"os"
	"unsafe"

	"github.com/veandco/go-sdl2/sdl"
)

// bufSize is the size of the buffer to store data from stdin. NOTE: this might
// need tuning.
const bufSize uint = 2048

// milkDropWindow represents the window projectm will render visualizations in.
type milkDropWindow struct {
	window  *sdl.Window
	context sdl.GLContext
	handle  C.projectm_handle
	preset  *ring.Ring
}

/*
 * setupPresets initializes a ring to store the passed presets in.
 */
func (m *milkDropWindow) setupPresets(presets []string) {
	m.preset = ring.New(len(presets))
	for _, preset := range presets {
		m.preset.Value = preset
		m.preset = m.preset.Next()
	}
}

/*
 * prevPreset sets the milkDropWindow's preset to the one after the current
 * preset.
 */
func (m *milkDropWindow) nextPreset(smooth bool) {
	m.preset = m.preset.Next()
	m.loadPreset(smooth)
}

/*
 * prevPreset sets the milkDropWindow's preset to the one before the current
 * preset.
 */
func (m *milkDropWindow) prevPreset(smooth bool) {
	m.preset = m.preset.Prev()
	m.loadPreset(smooth)
}

/*
 * newMilkDropWindow returns a new milkDropWindow with the given width and
 * height, and sets presets available to the window.
 */
func newMilkDropWindow(width, height int32, presets []string) (*milkDropWindow, error) {
	var m milkDropWindow
	var err error

	m.setupPresets(presets)

	m.window, err = sdl.CreateWindow("milkbucket", sdl.WINDOWPOS_UNDEFINED, sdl.WINDOWPOS_UNDEFINED, width, height, sdl.WINDOW_OPENGL|sdl.WINDOW_RESIZABLE)
	if err != nil {
		return &m, err
	}

	m.context, err = m.window.GLCreateContext()
	if err != nil {
		return &m, err
	}

	m.handle = C.projectm_create()
	if m.handle == nil {
		return &m, errors.New("error creating projectM instance")
	}

	C.projectm_set_window_size(m.handle, C.ulong(width), C.ulong(height))

	return &m, nil
}

/*
 * destroy handles cleanup of the milkDropWindow.
 */
func (m *milkDropWindow) destroy() {
	C.projectm_destroy(m.handle)
	sdl.GLDeleteContext(m.context)
	m.window.Destroy()
}

/*
 * loadPreset loads the preset file currently pointed to by m.preset. smooth
 * determines whether there is a smooth transition between the current preset
 * and the new preset.
 */
func (m *milkDropWindow) loadPreset(smooth bool) {
	preset := m.preset.Value.(string)
	cPreset := C.CString(preset)
	defer C.free(unsafe.Pointer(cPreset))
	C.projectm_load_preset_file(m.handle, cPreset, C.bool(smooth))
}

/*
 * render reads data from stdin and passes the data to projectm for rendering.
 * It returns a bool for if the program should keep running and an error.
 */
func (m *milkDropWindow) render() (bool, error) {
	var audioData [bufSize]int16

	err := binary.Read(os.Stdin, binary.LittleEndian, &audioData)
	if err == io.ErrUnexpectedEOF {
		return false, nil
	} else if err != nil {
		return true, err
	}

	C.projectm_pcm_add_int16(m.handle, (*C.int16_t)(&audioData[0]), C.uint(bufSize), C.PROJECTM_STEREO)

	C.projectm_opengl_render_frame(m.handle)
	m.window.GLSwap()

	return true, nil
}

/*
 * resize informs projectm of the new window size.
 */
func (m *milkDropWindow) resize(w, h int32) {
	C.projectm_set_window_size(m.handle, C.ulong(w), C.ulong(h))
}