mirror of https://bitbucket.org/ausocean/av.git
76 lines
2.3 KiB
Go
76 lines
2.3 KiB
Go
/*
|
|
LICENSE
|
|
Copyright (C) 2019 the Australian Ocean Lab (AusOcean)
|
|
|
|
This is free software: you can redistribute it and/or modify them
|
|
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.
|
|
|
|
It 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
|
|
in gpl.txt. If not, see http://www.gnu.org/licenses.
|
|
*/
|
|
|
|
package revid
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
|
|
"bitbucket.org/ausocean/av/codec/codecutil"
|
|
"bitbucket.org/ausocean/av/container/mts"
|
|
"bitbucket.org/ausocean/av/input/audio"
|
|
"bitbucket.org/ausocean/utils/logger"
|
|
)
|
|
|
|
// startAudioDevice is used to start capturing audio from an audio device and processing it.
|
|
// It returns a function that can be used to stop the device and any errors that occur.
|
|
func (r *Revid) startAudioDevice() (func() error, error) {
|
|
// Create audio device.
|
|
ac := &audio.Config{
|
|
SampleRate: r.config.SampleRate,
|
|
Channels: r.config.Channels,
|
|
RecPeriod: r.config.RecPeriod,
|
|
BitDepth: r.config.BitDepth,
|
|
Codec: r.config.InputCodec,
|
|
}
|
|
mts.Meta.Add("sampleRate", strconv.Itoa(r.config.SampleRate))
|
|
mts.Meta.Add("channels", strconv.Itoa(r.config.Channels))
|
|
mts.Meta.Add("period", fmt.Sprintf("%.6f", r.config.RecPeriod))
|
|
mts.Meta.Add("bitDepth", strconv.Itoa(r.config.BitDepth))
|
|
switch r.config.InputCodec {
|
|
case codecutil.PCM:
|
|
mts.Meta.Add("codec", "pcm")
|
|
case codecutil.ADPCM:
|
|
mts.Meta.Add("codec", "adpcm")
|
|
default:
|
|
r.config.Logger.Log(logger.Fatal, pkg+"no audio codec set in config")
|
|
}
|
|
|
|
ai, err := audio.NewDevice(ac, r.config.Logger)
|
|
if err != nil {
|
|
r.config.Logger.Log(logger.Fatal, pkg+"failed to create audio device", "error", err.Error())
|
|
}
|
|
|
|
// Start audio device
|
|
err = ai.Start()
|
|
if err != nil {
|
|
r.config.Logger.Log(logger.Fatal, pkg+"failed to start audio device", "error", err.Error())
|
|
}
|
|
|
|
// Process output from audio device.
|
|
r.config.ChunkSize = ai.ChunkSize()
|
|
r.wg.Add(1)
|
|
go r.processFrom(ai, time.Duration(float64(time.Second)/r.config.WriteRate))
|
|
return func() error {
|
|
ai.Stop()
|
|
return nil
|
|
}, nil
|
|
}
|