From 25bf0872b6e3a28f93d222823b09e86549ea36d9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andr=C3=A9=20Nusser?= Date: Sun, 15 Jan 2017 21:21:31 +0100 Subject: Add new drumkit_creator functions. --- test/drumkit_creator.cc | 370 ++++++++++++++++++++++++++++++++++++++++++++++++ test/drumkit_creator.h | 101 +++++++++++++ 2 files changed, 471 insertions(+) create mode 100644 test/drumkit_creator.cc create mode 100644 test/drumkit_creator.h 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 + +#include +#include +#include +#include + +// FIXME: unix-centric paths, automatically clean up the drumkit directory + +namespace drumkit_creator +{ + +std::vector 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 wav_infos = { + WavInfo("1011.wav", 1, 0x1110), + WavInfo("2122.wav", 1, 0x2221) + }; + + std::vector audiofiles1(4, Audiofile{&wav_infos.front(), 1}); + std::vector audiofiles2(4, Audiofile{&wav_infos.back(), 1}); + + std::vector sample_data1{{"stroke1", std::move(audiofiles1)}}; + std::vector sample_data2{{"stroke1", std::move(audiofiles2)}}; + + std::vector 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 wav_infos = { + WavInfo("small_instr.wav", 549833) + }; + + std::vector audiofiles; + for (std::size_t i = 0; i < 13; ++i) { + audiofiles.push_back({&wav_infos.front(), i+1}); + } + + std::vector sample_data{{"stroke1", std::move(audiofiles)}}; + + std::vector 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 wav_infos = { + WavInfo("huge_instr.wav", 549833) + }; + + std::vector audiofiles; + for (std::size_t i = 0; i < 13; ++i) { + audiofiles.push_back({&wav_infos.front(), i+1}); + } + + std::vector sample_data; + for (std::size_t i = 0; i < 50; ++i) { + sample_data.push_back({"stroke" + std::to_string(i), audiofiles}); + } + + std::vector 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 = "\n" + "\n" + "\n" + "\n" + "\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 createData(const WavInfo& wav_info, std::size_t number_of_channels) +{ + std::vector 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::min(); + int upper_bound = std::numeric_limits::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 = "\n" + "\n"; + // FIXME sampleref + std::string postfix = "\n" + "\n" + "\n" + "\n" + "\n" + "\n"; + + std::string samples; + for (const auto& sample: data.sample_data) { + samples += "\n"; + + for (std::size_t i = 0; i < sample.audiofiles.size(); ++i) + { + const auto& audiofile = sample.audiofiles[i]; + samples += "filename + "\" filechannel=\"" + + std::to_string(audiofile.filechannel) + "\"/>\n"; + } + samples += "\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 = "\n" + "\n"; + std::string postfix = "\n"; + + // Channel string + std::string channels; + channels += "\n"; + for (std::size_t i = 0; i < data.number_of_channels; ++i) + { + channels += "\n"; + } + channels += "\n"; + + // Instrument string + std::string instruments; + instruments += "\n"; + for (const auto& instrument: data.instruments) + { + instruments += "\n"; + for (std::size_t i = 0; i < data.number_of_channels; ++i) + { + std::string i_str = std::to_string(i); + instruments += "\n"; + } + instruments += "\n"; + } + 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 diff --git a/test/drumkit_creator.h b/test/drumkit_creator.h new file mode 100644 index 0000000..93a6f36 --- /dev/null +++ b/test/drumkit_creator.h @@ -0,0 +1,101 @@ +/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ +/*************************************************************************** + * drumkit_creator.h + * + * 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. + */ +#pragma once + +#include +#include +#include + +namespace drumkit_creator +{ + +using Sample = uint16_t; + +//! If is_random is true then this overrules the sample member. If is_random +//! is false however, every sample is chosen as the one in the member variable. +struct WavInfo +{ + const std::string filename; + const std::size_t length; + + const bool is_random; + const Sample sample; + + WavInfo(const std::string& filename, std::size_t length) + : filename(filename), length(length), is_random(true), sample(0) {} + + WavInfo(const std::string& filename, std::size_t length, Sample sample) + : filename(filename), length(length), is_random(false), sample(sample) {} +}; + +struct Audiofile +{ + WavInfo* wav_info; + std::size_t filechannel; +}; + +struct SampleData +{ + std::string name; + // Vector of non-owning pointers and therefore it is raw. + std::vector audiofiles; +}; + +struct InstrumentData +{ + std::string name; + std::string filename; + std::vector sample_data; +}; + +struct DrumkitData +{ + std::string name; + std::size_t number_of_channels; + std::vector instruments; + std::vector wav_infos; +}; + +//! Creates a drumkit from data in the temporary directory of the OS and +//! returns its filename. +std::string create(const DrumkitData& data); + +//! Creates a single wav file +void createWav(const WavInfo& wav_info, std::size_t number_of_channels, const std::string& dir); + +//! Those functions create some special wav files, drumkits, and midimaps +std::string createStdKit(const std::string& name); +std::string createSmallKit(const std::string& name); +std::string createHugeKit(const std::string& name); + +std::string createSingleChannelWav(const std::string& name); +std::string createMultiChannelWav(const std::string& name); +std::string create0000Wav(const std::string& name); + +std::string createStdMidimap(const std::string& name); + +} // end drumkit_creator -- cgit v1.2.3