WIP

1 files changed, 430 insertions, 0 deletions

diff --git a/src/canvastoolranges.cc b/src/canvastoolranges.cc
new file mode 100644
index 0000000..9a76163
--- /dev/null
+++ b/src/canvastoolranges.cc
@@ -0,0 +1,430 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/***************************************************************************
+ *            canvastoolranges.cc
+ *
+ *  Thu Jul 28 20:16:59 CEST 2011
+ *  Copyright 2011 Bent Bisballe Nyeng
+ *  deva@aasimon.org
+ ****************************************************************************/
+
+/*
+ *  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 "canvastoolranges.h"
+
+#include "project.h"
+
+#include <math.h>
+#include <stdio.h>
+#include <iostream>
+
+#define mapX(x) canvas->mapX(x)
+#define mapY(x) canvas->mapY(x)
+#define unmapX(x) canvas->unmapX(x)
+#define unmapY(x) canvas->unmapY(x)
+
+CanvasToolRanges::CanvasToolRanges(Instrument& instrument,
+                                   Canvas* c, Ranges& s,
+                                   Ranges& p)
+	: instrument(instrument)
+	, ranges(s)
+	, ranges_preview(p)
+{
+	hold = 100;
+
+	canvas = c;
+
+	range_is_moving_left = false;
+	range_is_moving_right = false;
+
+	colSelBg = QColor(255, 0, 0, 60);
+	colSel = QColor(255, 0, 0, 160);
+	colActiveSelBg = QColor(255, 255, 0, 60);
+	colActiveSel = QColor(255, 255, 0, 160);
+	colPreviewSelBg = QColor(0, 0, 255, 60);
+	colPreviewSel = QColor(0, 0, 255, 160);
+}
+
+bool CanvasToolRanges::mouseMoveEvent(QMouseEvent* event)
+{
+	sel_id_t active_range = ranges.active();
+	Range act_sel = ranges.get(active_range);
+
+	if(range_is_moving_left)
+	{
+		float val = unmapX(event->x());
+		if(val > act_sel.to) val = act_sel.to - 1;
+		act_sel.from = std::max(0.0f, val);
+		ranges.update(active_range, act_sel);
+		canvas->update();
+		return true;
+	}
+
+	if(range_is_moving_right)
+	{
+		float val = unmapX(event->x());
+		if(val < act_sel.from) val = act_sel.from + 1;
+		act_sel.to = std::min(val, (float)canvas->size);
+		ranges.update(active_range, act_sel);
+		canvas->update();
+		return true;
+	}
+
+	if(event->button() != Qt::LeftButton)
+	{
+		// Check if a range is being dragged.
+		QVector<sel_id_t> ids = ranges.ids();
+		QVector<sel_id_t>::iterator i = ids.begin();
+		while(i != ids.end())
+		{
+			Range sel = ranges.get(*i);
+			if(abs(event->x() - mapX(sel.from)) < 2
+			   || abs(event->x() - mapX(sel.to)) < 2)
+			{
+				canvas->setCursor(Qt::SplitHCursor);
+				return true;
+			}
+			i++;
+		}
+	}
+
+	return false;
+}
+
+bool CanvasToolRanges::mousePressEvent(QMouseEvent* event)
+{
+	if(event->button() == Qt::LeftButton)
+	{
+		// Check if a range is being dragged.
+		QVector<sel_id_t> ids = ranges.ids();
+		QVector<sel_id_t>::iterator i = ids.begin();
+		while(i != ids.end())
+		{
+			Range sel = ranges.get(*i);
+			if(abs(event->x() - mapX(sel.from)) < 2)
+			{
+				ranges.setActive(*i);
+				range_is_moving_left = true;
+				return true;
+			}
+
+			if(abs(event->x() - mapX(sel.to)) < 2)
+			{
+				ranges.setActive(*i);
+				range_is_moving_right = true;
+				return true;
+			}
+
+			i++;
+		}
+
+		// Check if a range is being selected.
+		i = ids.begin();
+		while(i != ids.end())
+		{
+			Range sel = ranges.get(*i);
+			if(event->x() > mapX(sel.from) &&
+			   event->x() < mapX(sel.to))
+			{
+				ranges.setActive(*i);
+				canvas->update();
+				return true;
+			}
+
+			i++;
+		}
+
+		// Make new range
+		int from = unmapX(event->x());
+		Range new_range{.from = from, .to = from};
+		sel_id_t id = ranges.add(new_range);
+		ranges.setActive(id);
+		range_is_moving_right = true;
+		canvas->update();
+		return true;
+	}
+
+	return false;
+}
+
+bool CanvasToolRanges::mouseReleaseEvent(QMouseEvent* event)
+{
+	if(event->button() == Qt::LeftButton)
+	{
+		if(range_is_moving_left || range_is_moving_right)
+		{
+			range_is_moving_left = false;
+			range_is_moving_right = false;
+			canvas->setCursor(Qt::ArrowCursor);
+			canvas->update();
+			return true;
+		}
+	}
+
+	return false;
+}
+
+void CanvasToolRanges::paintEvent(QPaintEvent* event, QPainter& painter)
+{
+	int pos = unmapX(event->rect().x());
+	int width = unmapX(event->rect().width());
+
+	{
+		QVector<sel_id_t> ids = ranges.ids();
+		QVector<sel_id_t>::iterator i = ids.begin();
+		while(i != ids.end())
+		{
+			Range sel = ranges.get(*i);
+			int from = sel.from;
+			int to = sel.to;
+			int fadein = sel.fadein;
+			int fadeout = sel.fadeout;
+			if(from > pos + width || to + width < pos)
+			{ i++; continue; }
+			if(ranges.active() == *i)
+			{
+				painter.setBrush(colActiveSelBg);
+				painter.setPen(colActiveSel);
+			}
+			else
+			{
+				painter.setBrush(colSelBg);
+				painter.setPen(colSel);
+			}
+			painter.drawRect(mapX(from), mapY(-1.0),
+			                 mapX(to) - mapX(from), mapY(1.0) - mapY(-1.0));
+			painter.drawLine(mapX(from), mapY(0.0), mapX(from + fadein), mapY(-1.0));
+			painter.drawLine(mapX(from), mapY(0.0), mapX(from + fadein), mapY(1.0));
+			painter.drawLine(mapX(to - fadeout), mapY(-1.0), mapX(to), mapY(0.0));
+			painter.drawLine(mapX(to - fadeout), mapY(1.0), mapX(to), mapY(0.0));
+			i++;
+		}
+	}
+
+	if(show_preview)
+	{
+		QVector<sel_id_t> ids = ranges_preview.ids();
+		QVector<sel_id_t>::iterator i = ids.begin();
+		while(i != ids.end())
+		{
+			Range sel = ranges_preview.get(*i);
+			int from = sel.from;
+			int to = sel.to;
+			int fadein = sel.fadein;
+			int fadeout = sel.fadeout;
+			if(from > pos + width || to + width < pos)
+			{ i++; continue; }
+			painter.setBrush(colPreviewSelBg);
+			painter.setPen(colPreviewSel);
+
+			painter.drawRect(mapX(from), mapY(-1.0),
+			                 mapX(to) - mapX(from), mapY(1.0) - mapY(-1.0));
+			painter.drawLine(mapX(from), mapY(0.0), mapX(from + fadein), mapY(-1.0));
+			painter.drawLine(mapX(from), mapY(0.0), mapX(from + fadein), mapY(1.0));
+			painter.drawLine(mapX(to - fadeout), mapY(-1.0), mapX(to), mapY(0.0));
+			painter.drawLine(mapX(to - fadeout), mapY(1.0), mapX(to), mapY(0.0));
+			i++;
+		}
+	}
+}
+
+void CanvasToolRanges::keyReleaseEvent(QKeyEvent* event)
+{
+	if(ranges.active() != SEL_NONE && event->key() == Qt::Key_Delete)
+	{
+		ranges.remove(ranges.active());
+		canvas->update();
+	}
+}
+
+void CanvasToolRanges::holdChanged(int h)
+{
+	hold = h;
+}
+
+void CanvasToolRanges::noiseFloorChanged(int t)
+{
+	double div = 666.0 / 0.00003;
+	noise_floor = (double)t/div;
+}
+
+void CanvasToolRanges::fadeoutChanged(int t)
+{
+	fadeout = (double)t/1000.0;
+}
+
+void CanvasToolRanges::autoCreateRanges()
+{
+	doAutoCreateRanges(false);
+}
+
+void CanvasToolRanges::autoCreateRangesPreview()
+{
+	doAutoCreateRanges(true);
+}
+
+void CanvasToolRanges::setPositionData(const float* positionData1,
+                                       size_t positionSize1,
+                                       const float* positionData2,
+                                       size_t positionSize2)
+{
+	std::cout << __PRETTY_FUNCTION__ << std::endl;
+	std::cout << positionSize1 << " " << positionSize2 << std::endl;
+	this->positionData1 = positionData1;
+	this->positionSize1 = positionSize1;
+
+	this->positionData2 = positionData2;
+	this->positionSize2 = positionSize2;
+}
+
+void CanvasToolRanges::doAutoCreateRanges(bool preview)
+{
+	std::cout << __PRETTY_FUNCTION__ << std::endl;
+
+	// TODO: Get data from master file instead of canvas
+	float* data = canvas->data;
+	size_t size = canvas->size;
+
+	if(preview) ranges_preview.clear();
+
+	for(size_t i = 0; i < size; i++)
+	{
+		if(fabs(data[i]) > fabs(instrument.getThreshold()))
+		{
+			int from = i;
+
+			if(data[from] > 0.0)
+			{
+				while(/*data[from] > data[from-1] &&*/ // Falling
+				      data[from-1] > 0.0 // Not crossing zero
+					)
+				{
+					from--;
+				}
+			}
+			else if(data[from] < 0.0)
+			{
+				while(/*data[from] < data[from-1] &&*/ // Rising
+				      data[from-1] < 0.0 // Not crossing zero
+					)
+				{
+					from--;
+				}
+			}
+
+			int to = i;
+			float runavg = fabs(data[from]);
+			while((runavg > noise_floor || to < from + hold) &&
+			      to < (int)size)
+			{
+				double p = 0.9;
+				runavg = runavg * p + fabs(data[to]) * (1 - p);
+				to++;
+			}
+
+			Range range{
+				.from = from,
+				.to = to,
+				.fadein = 2,
+				.fadeout = ((to - from) / 3) * fadeout
+				};
+
+			/////////////////////////////////////////////////////////////////////////
+			if(positionData1 && positionData2)
+			{
+				//TODO: set window in ms instead of samples
+				int pos_window_size{1000}; // +/- 500 from detected,
+				double threshold = 0.1; // 1/3 is about right
+
+				float norm{0};
+				for(int idx = range.from - pos_window_size / 2;
+				    idx < range.from + pos_window_size / 2;
+				    ++idx)
+				{
+					norm = std::max(std::abs(positionData1[idx]), norm);
+				}
+
+				int pos1{range.from - pos_window_size / 2};
+				for(int idx = range.from - pos_window_size / 2;
+				    idx < range.from + pos_window_size / 2;
+				    ++idx)
+				{
+					if(std::abs(positionData1[idx]) / norm > threshold)
+					{
+						pos1 = idx;
+						range.pos1 = pos1;
+						break;
+					}
+				}
+
+				norm = 0;
+				for(int idx = range.from - pos_window_size / 2;
+				    idx < range.from + pos_window_size / 2;
+				    ++idx)
+				{
+					norm = std::max(std::abs(positionData1[idx]), norm);
+				}
+
+				int pos2{range.from + pos_window_size / 2};
+				for(int idx = range.from - pos_window_size / 2;
+				    idx < range.from + pos_window_size / 2;
+				    ++idx)
+				{
+					if(std::abs(positionData2[idx]) / norm > threshold)
+					{
+						pos2 = idx;
+						range.pos2 = pos2;
+						break;
+					}
+				}
+
+				constexpr double speed_of_sound{343100}; // mm/s
+				constexpr double samplerate{48000}; // 48kHz - TODO get from audio
+
+				range.position = std::abs(pos1 - pos2) * speed_of_sound / samplerate / 2.0;
+			}
+			/////////////////////////////////////////////////////////////////////////
+
+			if(preview)
+			{
+				ranges_preview.add(range);
+			}
+			else
+			{
+				ranges.add(range);
+			}
+
+			i = to + 1;
+		}
+	}
+	canvas->update();
+}
+
+void CanvasToolRanges::clearRanges()
+{
+	ranges.clear();
+	range_is_moving_left = false;
+	range_is_moving_right = false;
+	canvas->setCursor(Qt::ArrowCursor);
+	canvas->update();
+}
+
+void CanvasToolRanges::setShowPreview(bool s)
+{
+	show_preview = s;
+	canvas->update();
+}