/* -*- 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"

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();
}

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, false);
}

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);
		redraw();

		shows_overlay = false;
	}
}

void DrumkitTab::triggerAudition(int x, int y, bool show_hit)
{
	// FIXME: introduce matrix/grid class and then make notation more beautiful.
	auto pos = (x - drumkit_image_x) + (y - drumkit_image_y)*map_image->width();
	auto index = pos_to_colour_index[pos];
	if (index == -1)
	{
		return;
	}
	auto const& colour = colours[index];
	auto const& instrument = to_instrument_name[index];

	++settings.audition_counter;
	settings.audition_instrument = instrument;
	settings.audition_velocity = current_velocity;

	if (show_hit)
	{
		Painter painter(*this);
		painter.drawRestrictedImage(drumkit_image_x, drumkit_image_y, colour, *map_image);
		shows_instrument_overlay = true;
	}

	current_instrument = instrument;
	updateInstrumentLabel();

	redraw();
}

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

void DrumkitTab::updateInstrumentLabel()
{
	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(height*width, -1);
	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() == 1.)
			{
				continue;
			}

			auto it = std::find(colours.begin(), colours.end(), colour);
			if (it == colours.end())
			{
				colours.push_back(colour);
			}

			int index = std::distance(colours.begin(), it);
			pos_to_colour_index[x + y*width] = index;
		}
	}

	// 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::