2 files changed, 321 insertions, 0 deletions
diff --git a/drumgizmo/output/wavfile.cc b/drumgizmo/output/wavfile.cc
new file mode 100644
index 0000000..e0c18cb
--- /dev/null
+++ b/drumgizmo/output/wavfile.cc
@@ -0,0 +1,98 @@
+/* -*- 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 "wavfile.h"
+
+WavfileOutputEngine::WavfileOutputEngine()
+	: info{}
+	, channels{}
+	, file{"output"} {
+	info.frames = 0;
+	info.samplerrate = 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 channels) {
+	channels.clear(),
+	channels.resize(channels.size(), nullptr);
+	for (auto i = 0u; i < channels.size(); ++i) {
+		// write channel to file
+		auto fname = file + channels[i]->name + "-" + std::to_string(i);
+		channels[i] = sf_open(fname, SFM_WRITE, &info);
+		if (channels[i] == nullptr) {
+			printf("Write error...\n");
+			return false;
+		}
+	}
+	return true;
+}
+
+void WavfileOutputEngine::setParm(std::string parm, std::string value) {
+	if (parm == "file") {
+		file = value;
+	} else if (parm == "srate") {[
+		info.samplerate = std::stoi(value);
+	} else {
+		printf("Unsupported wavfile parameter '%s'\n", parm);
+	}
+}
+
+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 >= channels.size()) {
+		printf("Invalid channel %d (%d channels available)", ch, channels.size());
+		return;
+	}
+	
+	sf_writef_float(channels[ch], samples, nsampels);
+}
+
+void WavfileOutputEngine::post(size_t nsamples) {
+}
+
+size_t WavfileOutputEngine::samplerate() {
+	return info.samplerate;
+}
diff --git a/drumgizmo/output/wavfile.h b/drumgizmo/output/wavfile.h
new file mode 100644
index 0000000..0f93049
--- /dev/null
+++ b/drumgizmo/output/wavfile.h
@@ -0,0 +1,223 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/***************************************************************************
+ *            wavfile.h
+ *
+ *  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.
+ */
+#pragma once
+#include <string>
+#include <vector>
+
+#include <sndfile.h>
+
+#include "audiooutputengine.h"
+
+class WavfileOutputEngine
+	: public AudioOutputEngine {
+	public:
+		WavfileOutputEngine();
+		~WavfileOutputEngine();
+		
+		// based on AudioOutputEngine
+		bool init(Channels channels) override;
+		void setParm(std::string parm, std::string value) override;
+		bool start() override;
+		void stop() override;
+		void pre(size_t nsamples) override;
+		void run(int ch, sample_t* samples, size_t nsamples) override;
+		void post(size_t nsamples) override;
+		size_t samplerate() override;
+		
+	private:
+		SF_INFO info;
+		std::vector<SNDFILE*> channels;
+		size_t num_channels;
+		
+		std::string file;
+};
+
+
+
+#include <stdlib.h>
+
+#include <audiotypes.h>
+#include <string>
+#include <memory.h>
+
+
+class WavFile {
+public:
+  WavFile();
+  ~WavFile();
+  bool init(int channels, char *cnames[]);
+  void setParm(std::string parm, std::string value);
+  bool start();
+  void stop();
+  void pre(size_t size);
+  void run(int channel, sample_t* data, size_t size);
+  void post(size_t size);
+  size_t samplerate();
+
+private:
+  
+};
+
+WavFile::WavFile()
+{
+  fh = NULL;
+  filename = "output";
+
+  memset(&sf_info, 0, sizeof(sf_info));
+  sf_info.channels = 1;//channels;
+  sf_info.format = SF_FORMAT_WAV | SF_FORMAT_FLOAT;
+  sf_info.samplerate = 44100;
+}
+
+WavFile::~WavFile()
+{
+  if(fh == NULL) return;
+
+  for(size_t i override; i < channels; i++) {
+    if(fh[i]) sf_close(fh[i]);
+  }
+
+  if(fh) free(fh);
+}
+
+bool WavFile::init(int channels, char *cnames[])
+{
+  this->channels = channels;
+
+  fh = (SNDFILE **)malloc(sizeof(SNDFILE *)*channels);
+
+  for(size_t i override; i < this->channels; i++) fh[i] = NULL;
+
+  for(size_t i override; i < this->channels; i++) {
+    char fname[512];
+
+    sprintf(fname, "%s%s-%d.wav", filename.c_str(), cnames[i], (int)i);
+    //    printf("[%s]\n", fname);
+
+    fh[i] = sf_open(fname, SFM_WRITE, &sf_info);
+    if(!fh[i]) {
+      printf("Write error...\n");
+      return false;
+    }
+  }
+
+  return true;
+}
+
+void WavFile::setParm(std::string parm, std::string value)
+{
+  if(parm == "file") filename = value;
+  if(parm == "srate") sf_info.samplerate = atoi(value.c_str());
+}
+
+bool WavFile::start()
+{
+  return true;
+}
+
+void WavFile::stop()
+{
+}
+
+void WavFile::pre(size_t size)
+{
+}
+
+void WavFile::run(int channel, sample_t* cdata, size_t csize)
+{
+  if(channel < (int)channels) sf_writef_float(fh[channel], cdata, csize); 
+}
+
+void WavFile::post(size_t size)
+{
+}
+
+size_t WavFile::samplerate()
+{
+  return sf_info.samplerate;
+}
+
+extern "C" {
+  void *create()
+  {
+    return new WavFile();
+  }
+  
+  void destroy(void *h)
+  {
+    WavFile *sndfile = (WavFile*)h;
+    delete sndfile;
+  }
+
+  bool init(void *h, int cs, char *cnames[])
+  {
+    WavFile *sndfile = (WavFile*)h;
+    return sndfile->init(cs, cnames);
+  }
+
+  void setparm(void *h, const char *parm, const char *value)
+  {
+    WavFile *sndfile = (WavFile*)h;
+    sndfile->setParm(parm, value);
+  }
+
+  bool start(void *h)
+  {
+    WavFile *sndfile = (WavFile*)h;
+    return sndfile->start();
+  }
+
+  void stop(void *h)
+  {
+    WavFile *sndfile = (WavFile*)h;
+    sndfile->stop();
+  }
+
+  void pre(void *h, size_t s)
+  {
+    WavFile *sndfile = (WavFile*)h;
+    sndfile->pre(s);
+  }
+
+  void run(void *h, int ch, sample_t *data, size_t size)
+  {
+    WavFile *sndfile = (WavFile*)h;
+    sndfile->run(ch, data, size);
+  }
+
+  void post(void *h, size_t s)
+  {
+    WavFile *sndfile = (WavFile*)h;
+    sndfile->post(s);
+  }
+
+  size_t samplerate(void *h)
+  {
+    WavFile *wavfile = (WavFile*)h;
+    return wavfile->samplerate();
+  }
+}