path: root/plugingui/drumkittab.cc

/* -*- Mode: c++ -*- */
/***************************************************************************
 *            drumkittab.cc
 *
 *  Fri Jun  8 21:50:03 CEST 2018
 *  Copyright 2018 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 "drumkittab.h"

#include <algorithm>
#include <iomanip>
#include <sstream>

#include "cpp11fix.h" // required for c++11
#include "painter.h"
#include "settings.h"

// TODO:
// make consistent with clicks etc

namespace GUI
{

DrumkitTab::DrumkitTab(Widget* parent,
                       Settings& settings/*,
                       SettingsNotifier& settings_notifier,
                       Config& config*/)
	: Widget(parent)
	, settings(settings)
	/*, settings_notifier(settings_notifier)
	, config(config)*/
{
	init("/home/chaot/Programming/drumgizmo/crocellkit01.png", "/home/chaot/Programming/drumgizmo/crocellkit01_map.png");

	velocity_label.move(10, height()-velocity_label.height()-5);
	updateVelocityLabel();
	velocity_label.resizeToText();

	instrument_name_label.move(velocity_label.width()+30, height()-instrument_name_label.height()-5);
	updateInstrumentLabel(-1);

	pos_to_colour_index.setDefaultValue(-1);
}

void DrumkitTab::resize(std::size_t width, std::size_t height)
{
	Widget::resize(width, height);

	if (drumkit_image) {
		Painter painter(*this);
		painter.clear();

		drumkit_image_x = (this->width()-drumkit_image->width())/2;
		drumkit_image_y = (this->height()-drumkit_image->height())/2;
		painter.drawImage(drumkit_image_x, drumkit_image_y, *drumkit_image);
	}

	velocity_label.move(10, height-velocity_label.height()-5);
	instrument_name_label.move(velocity_label.width()+30, height-instrument_name_label.height()-5);
}

void DrumkitTab::buttonEvent(ButtonEvent* buttonEvent)
{
	if (map_image) {
		if (buttonEvent->button == MouseButton::right) {
			if (buttonEvent->direction == GUI::Direction::down) {
				Painter painter(*this);
				painter.drawImage(drumkit_image_x, drumkit_image_y, *map_image);
				redraw();

				shows_overlay = true;
				return;
			}
			if (buttonEvent->direction == GUI::Direction::up) {
				Painter painter(*this);
				painter.clear();
				painter.drawImage(drumkit_image_x, drumkit_image_y, *drumkit_image);
				redraw();

				shows_overlay = false;
				return;
			}
		}
	}

	if (buttonEvent->button == MouseButton::left)
	{
	    if (buttonEvent->direction == GUI::Direction::down)
		{
			triggerAudition(buttonEvent->x, buttonEvent->y);
		}
	    if (buttonEvent->direction == GUI::Direction::up)
		{
			if (shows_instrument_overlay) {
				Painter painter(*this);
				painter.clear();
				painter.drawImage(drumkit_image_x, drumkit_image_y, *drumkit_image);
				if (shows_overlay) {
					painter.drawImage(drumkit_image_x, drumkit_image_y, *map_image);
				}
				redraw();
			}
			shows_instrument_overlay = false;
		}
	}
}

void DrumkitTab::scrollEvent(ScrollEvent* scrollEvent)
{
	current_velocity -=  0.01*scrollEvent->delta;
	current_velocity = std::max(std::min(current_velocity, 1.0f), 0.0f);
	updateVelocityLabel();
	velocity_label.resizeToText();

	triggerAudition(scrollEvent->x, scrollEvent->y);
}

void DrumkitTab::mouseMoveEvent(MouseMoveEvent* mouseMoveEvent)
{
	// change to image coordinates
	auto const x = mouseMoveEvent->x - drumkit_image_x;
	auto const y = mouseMoveEvent->y - drumkit_image_y;

	auto index = pos_to_colour_index(x, y);

	highlightInstrument(index);
	updateInstrumentLabel(index);
	redraw();
}

void DrumkitTab::mouseLeaveEvent()
{
	if (map_image && (shows_overlay || shows_instrument_overlay)) {
		Painter painter(*this);
		painter.clear();
		painter.drawImage(drumkit_image_x, drumkit_image_y, *drumkit_image);

		shows_overlay = false;
		redraw();
	}
}

void DrumkitTab::triggerAudition(int x, int y)
{
	// change to image coordinates
	x -= drumkit_image_x;
	y -= drumkit_image_y;

	auto index = pos_to_colour_index(x, y);
	if (index == -1) { return; }

	++settings.audition_counter;
	settings.audition_instrument = to_instrument_name[index];
	settings.audition_velocity = current_velocity;

	redraw();
}

void DrumkitTab::highlightInstrument(int index)
{
	if (index == current_index) { return; }
	current_index = index;

	Painter painter(*this);
	painter.clear();
	painter.drawImage(drumkit_image_x, drumkit_image_y, *drumkit_image);
	shows_overlay = false;

	if (index != -1) {
		auto const& colour = colours[index];
		auto const& positions = colour_index_to_positions[index];
		painter.draw(positions.begin(), positions.end(), drumkit_image_x, drumkit_image_y, colour);

		shows_instrument_overlay = true;
	}
}

void DrumkitTab::updateVelocityLabel()
{
	std::stringstream stream;
	stream << std::fixed << std::setprecision(2) << current_velocity;
	velocity_label.setText("Velocity: " + stream.str());
}

void DrumkitTab::updateInstrumentLabel(int index)
{
	current_instrument = (index == -1 ? "" : to_instrument_name[index]);
	instrument_name_label.setText("Instrument: " + current_instrument);
	instrument_name_label.resizeToText();
}

void DrumkitTab::init(std::string const& image_file, std::string const& map_file)
{
	drumkit_image = std::make_unique<Image>(image_file);
	map_image = std::make_unique<Image>(map_file);

	// collect all colours and build lookup table
	auto const height = map_image->height();
	auto const width = map_image->width();

	colours.clear();
	pos_to_colour_index.assign(width, height, -1);
	colour_index_to_positions.clear();
	to_instrument_name.clear();

	for (std::size_t y = 0; y < map_image->height(); ++y)
	{
		for (std::size_t x = 0; x < map_image->width(); ++x)
		{
			auto colour = map_image->getPixel(x, y);

			if (colour.alpha() == 0.) { continue; }

			auto it = std::find(colours.begin(), colours.end(), colour);
			int index = std::distance(colours.begin(), it);

			if (it == colours.end())
			{
				// XXX: avoids low alpha values due to feathering of edges
				colours.emplace_back(colour.red(), colour.green(), colour.blue(), 0.7);
				colour_index_to_positions.emplace_back();
			}

			pos_to_colour_index(x, y) = index;
			colour_index_to_positions[index].emplace_back(x, y);
		}
	}

	// set instrument names
	to_instrument_name.resize(colours.size());
	for (std::size_t i = 0; i < colours.size(); ++i)
	{
		for (auto const& pair: colour_instrument_pairs)
		{
			if (pair.colour == colours[i]) {
				to_instrument_name[i] = pair.instrument;
			}
		}
	}
}

} // GUI::