summaryrefslogtreecommitdiff
path: root/autoart/autoaudio.go
blob: cc5965ece9850764bd174c461672359b836eb435 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
/*
Copyright (C) 2019 Leo Tenenbaum

This file is part of AutoArt.

AutoArt 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.

AutoArt 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 AutoArt.  If not, see <https://www.gnu.org/licenses/>.
*/

package autoart

import (
	"github.com/pommicket/autoart/autoutils"
	"io"
)

func GenerateAudio(output io.Writer, duration float64, sampleRate int32,
	functionLength int, rectifier int) error {
	samples := int64(duration * float64(sampleRate))
	err := autoutils.WriteAudioHeader(output, samples, 1, sampleRate)
	if err != nil {
		return err
	}

	vars := make([]float64, 1)
	const sampleBufferSize = 4096
	sampleBuffer := make([]uint8, sampleBufferSize)
	sampleBufferIndex := 0

	var function autoutils.Function
	function.Generate(1, functionLength)

	for s := int64(0); s < samples; s++ {
		t := float64(s) / float64(sampleRate)
		vars[0] = t
		value := rectify(function.Evaluate(vars), rectifier)
		sampleBuffer[sampleBufferIndex] = uint8(255 * value)
		sampleBufferIndex++
		if sampleBufferIndex == sampleBufferSize {
			err = autoutils.WriteAudioSamples(output, sampleBuffer)
			if err != nil {
				return err
			}
			sampleBufferIndex = 0
		}
	}
	return autoutils.WriteAudioSamples(output, sampleBuffer[:sampleBufferIndex])
}