summaryrefslogtreecommitdiff
path: root/autoart/autoaudio.go
blob: 05a91d48930903e9c1672506f046c0f9d0da8098 (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
/*
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])
}