summaryrefslogtreecommitdiff
path: root/test/dgreftest/wavfileoutputengine.cc
blob: 6f971add97f7bbbd2645ba1813cee0e056b49501 (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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/***************************************************************************
 *            wavfileoutputengine.cc
 *
 *  Sat May 14 13:26:51 CEST 2016
 *  Copyright 2016 Bent Bisballe Nyeng
 *  deva@aasimon.org
 ****************************************************************************/

/*
 *  This file is part of DrumGizmo.
 *
 *  DrumGizmo is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation; either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  DrumGizmo 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 Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with DrumGizmo; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA.
 */
#include "wavfileoutputengine.h"

#include <iostream>

WavfileOutputEngine::WavfileOutputEngine()
	: AudioOutputEngine{}
	, info{}
	, file{"output"}
{
	info.samplerate = 44100;
	info.channels = 1;
	info.format = SF_FORMAT_WAV | SF_FORMAT_FLOAT;
}

WavfileOutputEngine::~WavfileOutputEngine()
{
	sf_close(handle);
}

bool WavfileOutputEngine::init(const Channels& data)
{
	info.channels = data.size();

	handle = sf_open(file.c_str(), SFM_WRITE, &info);
	if(handle == nullptr)
	{
		std::cerr << "[WavfileOutputEngine] Failed to open "
		          << file << " for writing.\n";
		return false;
	}

	return true;
}

void WavfileOutputEngine::setParm(const std::string& parm, const std::string& value)
{
	if(parm == "file")
	{
		// apply output filename prefix
		file = value;
	}
	else if(parm == "srate")
	{
		// try to apply samplerate
		try
		{
			info.samplerate = std::stoi(value);
		}
		catch(...)
		{
			std::cerr << "[WavfileOutputEngine] Invalid samplerate " << value
			          << "\n";
		}
	}
	else
	{
		std::cerr << "[WavfileOutputEngine] Unsupported parameter '" << parm
		          << "'\n";
	}
}

bool WavfileOutputEngine::start()
{
	return true;
}

void WavfileOutputEngine::stop()
{
}

void WavfileOutputEngine::pre(size_t nsamples)
{
}

void WavfileOutputEngine::run(int ch, sample_t* samples, size_t nsamples)
{
	if(ch >= info.channels)
	{
		std::cerr << "[WavfileOutputEngine] cannot access channel #" << ch
		          << " (" << info.channels << " channels available)\n";
		return;
	}

	for(std::size_t i = 0; i < nsamples; ++i)
	{
		buffer[i * info.channels + ch] = samples[i];
	}
}

void WavfileOutputEngine::post(size_t nsamples)
{

	sf_writef_float(handle, buffer, nsamples);
}

size_t WavfileOutputEngine::getSamplerate() const
{
	return info.samplerate;
}