summaryrefslogtreecommitdiff
path: root/test/drumkit_creator.cc
diff options
context:
space:
mode:
Diffstat (limited to 'test/drumkit_creator.cc')
-rw-r--r--test/drumkit_creator.cc370
1 files changed, 370 insertions, 0 deletions
diff --git a/test/drumkit_creator.cc b/test/drumkit_creator.cc
new file mode 100644
index 0000000..f773c6f
--- /dev/null
+++ b/test/drumkit_creator.cc
@@ -0,0 +1,370 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/***************************************************************************
+ * drumkit_creator.cc
+ *
+ * Thu Jan 12 18:51:34 CET 2017
+ * Copyright 2017 André Nusser
+ * andre.nusser@googlemail.com
+ ****************************************************************************/
+
+/*
+ * 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 "drumkit_creator.h"
+
+#include "../src/random.h"
+
+#include <sndfile.h>
+
+#include <iostream>
+#include <fstream>
+#include <cstdlib>
+#include <algorithm>
+
+// FIXME: unix-centric paths, automatically clean up the drumkit directory
+
+namespace drumkit_creator
+{
+
+std::vector<Sample> createData(const WavInfo& wav_info,
+ std::size_t number_of_channels);
+std::string createTemporaryDirectory(const std::string& name);
+void createInstrument(const InstrumentData& data, std::size_t number_of_channels,
+ const std::string& dir);
+std::string createDrumkitFile(const DrumkitData& data, const std::string& dir);
+
+bool is_valid(const DrumkitData& data)
+{
+ // TODO Check the consistency of the data.
+ return true;
+}
+
+std::string create(const DrumkitData& data)
+{
+ std::string drumkit_filename;
+
+ if (is_valid(data))
+ {
+ auto dir = createTemporaryDirectory("drumkit");
+
+ for (const auto& wav_info: data.wav_infos) {
+ createWav(wav_info, data.number_of_channels, dir);
+ }
+
+ for (const auto& instrument: data.instruments) {
+ createInstrument(instrument, data.number_of_channels, dir);
+ }
+
+ drumkit_filename = createDrumkitFile(data, dir);
+ }
+ else
+ {
+ // TODO report error
+ std::cout << "ERROR: create" << std::endl;
+ }
+
+ return drumkit_filename;
+}
+
+void createWav(const WavInfo& wav_info, std::size_t number_of_channels, const std::string& dir)
+{
+ SF_INFO sfinfo;
+ sfinfo.samplerate = 44100;
+ sfinfo.channels = number_of_channels;
+ sfinfo.format = SF_FORMAT_WAV | SF_FORMAT_PCM_16;
+
+ std::string filename = dir + "/" + wav_info.filename;
+ auto sndfile = sf_open(filename.c_str(), SFM_WRITE, &sfinfo);
+ if (!sndfile) {
+ std::cout << "ERROR: Wav file couldn't be created: "
+ << sf_strerror(sndfile) << std::endl;
+ return;
+ }
+
+ auto data_vec = createData(wav_info, number_of_channels);
+
+ auto written_count = sf_write_raw(sndfile, data_vec.data(), 2*data_vec.size());
+ std::cout << "Written: " << written_count << std::endl;
+
+ sf_write_sync(sndfile);
+ sf_close(sndfile);
+}
+
+std::string createStdKit(const std::string& name)
+{
+ std::vector<WavInfo> wav_infos = {
+ WavInfo("1011.wav", 1, 0x1110),
+ WavInfo("2122.wav", 1, 0x2221)
+ };
+
+ std::vector<Audiofile> audiofiles1(4, Audiofile{&wav_infos.front(), 1});
+ std::vector<Audiofile> audiofiles2(4, Audiofile{&wav_infos.back(), 1});
+
+ std::vector<SampleData> sample_data1{{"stroke1", std::move(audiofiles1)}};
+ std::vector<SampleData> sample_data2{{"stroke1", std::move(audiofiles2)}};
+
+ std::vector<InstrumentData> instruments = {
+ InstrumentData{"instr1", "instr1.xml", std::move(sample_data1)},
+ InstrumentData{"instr2", "instr2.xml", std::move(sample_data2)}
+ };
+
+ auto kit_data = DrumkitData{name, 4, instruments, wav_infos};
+
+ return create(kit_data);
+}
+
+std::string createSmallKit(const std::string& name)
+{
+ std::vector<WavInfo> wav_infos = {
+ WavInfo("small_instr.wav", 549833)
+ };
+
+ std::vector<Audiofile> audiofiles;
+ for (std::size_t i = 0; i < 13; ++i) {
+ audiofiles.push_back({&wav_infos.front(), i+1});
+ }
+
+ std::vector<SampleData> sample_data{{"stroke1", std::move(audiofiles)}};
+
+ std::vector<InstrumentData> instruments = {
+ InstrumentData{"small_instr", "small_instr.xml", std::move(sample_data)}
+ };
+
+ auto kit_data = DrumkitData{name, 13, instruments, wav_infos};
+
+ return create(kit_data);
+}
+
+std::string createHugeKit(const std::string& name)
+{
+ std::vector<WavInfo> wav_infos = {
+ WavInfo("huge_instr.wav", 549833)
+ };
+
+ std::vector<Audiofile> audiofiles;
+ for (std::size_t i = 0; i < 13; ++i) {
+ audiofiles.push_back({&wav_infos.front(), i+1});
+ }
+
+ std::vector<SampleData> sample_data;
+ for (std::size_t i = 0; i < 50; ++i) {
+ sample_data.push_back({"stroke" + std::to_string(i), audiofiles});
+ }
+
+ std::vector<InstrumentData> instruments;
+ for (std::size_t i = 0; i < 50; ++i) {
+ instruments.push_back({"huge_instr" + std::to_string(i),
+ "huge_instr.xml",
+ sample_data});
+ }
+
+ auto kit_data = DrumkitData{name, 13, instruments, wav_infos};
+
+ return create(kit_data);
+}
+
+std::string createSingleChannelWav(const std::string& name)
+{
+ auto dir = createTemporaryDirectory("wavfiles");
+
+ auto wav_info = WavInfo(name, 173516);
+ createWav(wav_info,
+ 1,
+ dir);
+
+ return dir + "/" + name;
+}
+
+std::string createMultiChannelWav(const std::string& name)
+{
+ auto dir = createTemporaryDirectory("wavfiles");
+
+ auto wav_info = WavInfo(name, 549833);
+ createWav(wav_info,
+ 13,
+ dir);
+
+ return dir + "/" + name;
+}
+
+std::string create0000Wav(const std::string& name)
+{
+ auto dir = createTemporaryDirectory("wavfiles");
+
+ auto wav_info = WavInfo(name, 1, 0x0000);
+ createWav(wav_info,
+ 1,
+ dir);
+
+ return dir + "/" + name;
+}
+
+std::string createStdMidimap(const std::string& name)
+{
+ auto dir = createTemporaryDirectory("midimap");
+
+ std::string content = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
+ "<midimap>\n"
+ "<map note=\"1\" instr=\"instr1\"/>\n"
+ "<map note=\"2\" instr=\"instr2\"/>\n"
+ "</midimap>\n";
+
+ std::string filename = dir + "/" + name + ".xml";
+ std::ofstream file;
+ file.open(filename);
+ if (file.is_open())
+ {
+ file << content;
+ }
+ else
+ {
+ // TODO print error
+ std::cout << "ERROR: instrument" << std::endl;
+ }
+ file.close();
+
+ return filename;
+}
+
+//
+// Helper functions
+//
+
+std::string createTemporaryDirectory(const std::string& name)
+{
+ // FIXME Use cross-platform temporary directory
+ std::string dir_template = "/tmp/drumgizmo_" + name + "XXXXXX";
+ const auto dir_name = mkdtemp(&dir_template[0]);
+
+ if (dir_name) {
+ return std::string(dir_name);
+ }
+
+ return "";
+}
+
+std::vector<Sample> createData(const WavInfo& wav_info, std::size_t number_of_channels)
+{
+ std::vector<Sample> data_vec(number_of_channels * wav_info.length, wav_info.sample);
+ if (wav_info.is_random)
+ {
+ Random rand(42); // Fix the seed to make it reproducable.
+ int lower_bound = std::numeric_limits<Sample>::min();
+ int upper_bound = std::numeric_limits<Sample>::max();
+
+ std::generate(data_vec.begin(),
+ data_vec.end(),
+ [&](){ return Sample(rand.intInRange(lower_bound, upper_bound)); });
+ }
+
+ return data_vec;
+}
+
+void createInstrument(const InstrumentData& data, std::size_t number_of_channels,
+ const std::string& dir)
+{
+ std::string prefix = "<?xml version='1.0' encoding='UTF-8'?>\n"
+ "<instrument name=\"" + data.name + "\">\n";
+ // FIXME sampleref
+ std::string postfix = "<velocities>\n"
+ "<velocity lower=\"0\" upper=\"1\">\n"
+ "<sampleref probability=\"1\" name=\"stroke1\"/>\n"
+ "</velocity>\n"
+ "</velocities>\n"
+ "</instrument>\n";
+
+ std::string samples;
+ for (const auto& sample: data.sample_data) {
+ samples += "<sample name=\"" + sample.name + "\">\n";
+
+ for (std::size_t i = 0; i < sample.audiofiles.size(); ++i)
+ {
+ const auto& audiofile = sample.audiofiles[i];
+ samples += "<audiofile channel=\"ch" + std::to_string(i) + "\" file=\""
+ + audiofile.wav_info->filename + "\" filechannel=\""
+ + std::to_string(audiofile.filechannel) + "\"/>\n";
+ }
+ samples += "</sample>\n";
+ }
+
+ // Write to file
+ std::string filename = dir + "/" + data.filename;
+ std::ofstream file;
+ file.open(filename);
+ if (file.is_open())
+ {
+ file << prefix + samples + postfix;
+ }
+ else
+ {
+ // TODO print error
+ std::cout << "ERROR: instrument" << std::endl;
+ }
+ file.close();
+}
+
+std::string createDrumkitFile(const DrumkitData& data, const std::string& dir)
+{
+ // Pre- and postfix string
+ std::string prefix = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
+ "<drumkit name=\"" + data.name + "\" description=\"An drumkit generated"
+ " by the drumkit_generator\">\n";
+ std::string postfix = "</drumkit>\n";
+
+ // Channel string
+ std::string channels;
+ channels += "<channels>\n";
+ for (std::size_t i = 0; i < data.number_of_channels; ++i)
+ {
+ channels += "<channel name=\"ch" + std::to_string(i) + "\"/>\n";
+ }
+ channels += "</channels>\n";
+
+ // Instrument string
+ std::string instruments;
+ instruments += "<instruments>\n";
+ for (const auto& instrument: data.instruments)
+ {
+ instruments += "<instrument name=\"" + instrument.name + "\" file=\"" + instrument.filename + "\">\n";
+ for (std::size_t i = 0; i < data.number_of_channels; ++i)
+ {
+ std::string i_str = std::to_string(i);
+ instruments += "<channelmap in=\"ch" + i_str + "\" out=\"ch" + i_str + "\"/>\n";
+ }
+ instruments += "</instrument>\n";
+ }
+ instruments += "</instruments>\n";
+
+ // Write everything to a drumkit file
+ std::string filename = dir + "/" + data.name + ".xml";
+ std::ofstream file;
+ file.open(filename);
+ if (file.is_open())
+ {
+ file << prefix + channels + instruments + postfix;
+ }
+ else
+ {
+ // TODO print error
+ std::cout << "ERROR: drumkit" << std::endl;
+ }
+ file.close();
+
+ return filename;
+}
+
+} // end drumkit_creator