summaryrefslogtreecommitdiff
path: root/drumgizmo/output/wavfile.cc
blob: 678bbe37d8e46ab8d25121ad77395b5c366b91b4 (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
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/***************************************************************************
 *            wavfile.cc
 *
 *  Mi 20. Jan 16:57:16 CET 2016
 *  Copyright 2016 Christian Glöckner
 *  cgloeckner@freenet.de
 ****************************************************************************/

/*
 *  This file is part of DrumGizmo.
 *
 *  DrumGizmo 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 2 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 General Public License for more details.
 *
 *  You should have received a copy of the GNU 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 <iostream>

#include "wavfile.h"

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

WavfileOutputEngine::~WavfileOutputEngine()
{
	for (auto &ptr : channels)
	{
		if (ptr != nullptr)
		{
			sf_close(ptr);
		}
	}
}

bool WavfileOutputEngine::init(Channels data)
{
	channels.clear(), channels.resize(data.size()); // value-initialized with null
	for (auto i = 0u; i < data.size(); ++i)
	{
		// write channel to file
		auto fname = file + data[i].name + "-" + std::to_string(i) + ".wav";
		channels[i] = sf_open(fname.c_str(), SFM_WRITE, &info);
		if (channels[i] == nullptr)
		{
			std::cerr << "[WaffileOutputEngine] Failed to initialize "
			          << "channel #" << i << "\n";
			return false;
		}
	}
	return true;
}

void WavfileOutputEngine::setParm(std::string parm, 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 (static_cast<unsigned int>(ch) >= channels.size())
	{
		std::cerr << "[WavfileOutputEngine] cannot access channel #" << ch << " ("
		          << channels.size() << " channels available)\n";
		return;
	}

	sf_writef_float(channels[ch], samples, nsamples);
}

void WavfileOutputEngine::post(size_t nsamples) {}

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