drumgizmo/output/wavfile.cc


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

/* -*- 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;
}