frontend.cpp

// by wuwbobo2021 <https://github.com/wuwbobo2021>, <wuwbobo@outlook.com>
// Licensed under LGPL version 2.1.

#include <simple-cairo-plot/frontend.h>

#include <chrono>
#include <iostream>

#include <glibmm/stringutils.h>

using namespace SimpleCairoPlot;

Frontend::Frontend() {}

Frontend::Frontend(std::vector<VariableAccessPtr>& ptrs, unsigned int buf_size)
{
	this->init(ptrs, buf_size);
}

Frontend::~Frontend()
{
	this->close();
}

void Frontend::init(std::vector<VariableAccessPtr>& ptrs, unsigned int buf_size)
{
	if (ptrs.size() == 0 || buf_size < 2)
		throw std::runtime_error("Frontend::init(): invalid parameter.");
	
	this->ptrs = ptrs; this->buf_size = buf_size;
}

#ifndef _WIN32
void Frontend::open()
{
	if (this->thread_gtk || this->window) return;
	this->thread_gtk = new std::thread(&Frontend::app_run, this);
	
	while (! this->window)
		std::this_thread::sleep_for(std::chrono::milliseconds(10));
	std::this_thread::sleep_for(std::chrono::milliseconds(10));
}
#endif

Recorder& Frontend::recorder() const
{
	unsigned int wait_ms = 0;
	while (!this->window && wait_ms < 5000) {
		std::this_thread::sleep_for(std::chrono::milliseconds(10)); wait_ms += 10;
	}
	if (! this->window)
		throw std::runtime_error("Frontend::recorder(): frontend is not opened.");
	return *this->rec;
}

void Frontend::run()
{
	if (this->thread_gtk) {
		this->thread_gtk->join();
		delete this->thread_gtk; this->thread_gtk = NULL;
	} else
		this->app_run();
}

void Frontend::close()
{
	if (! this->window) return;
	
	if (this->rec) this->rec->stop();
	this->dispatcher_gtk->connect(sigc::mem_fun(*this, &Frontend::close_window));
	this->dispatcher_gtk->emit(); //eventually deletes itself
	
	if (this->thread_gtk) {
		this->thread_gtk->join();
		delete this->thread_gtk; this->thread_gtk = NULL;
	} else {
		while (this->window)
			std::this_thread::sleep_for(std::chrono::milliseconds(1));
	}
}

/*------------------------------ private functions ------------------------------*/

void Frontend::app_run()
{
	Glib::RefPtr<Gtk::Application> app = Gtk::Application::create(this->app_name);
	this->dispatcher_gtk = new Glib::Dispatcher;
	this->create_window();
	this->create_file_dialog();
	
	app->run(*this->window);
	
	this->window = NULL; //the window is already destructed when the thread exits Application::run()
	delete this->file_dialog; delete this->dispatcher_gtk;
} // Unsolved problem on Windows when running in a new thread created by open(): Segmentation fault received here.

void Frontend::create_window()
{
	this->rec = Gtk::manage(new Recorder(this->ptrs, this->buf_size));
	this->rec->signal_full().connect(sigc::mem_fun(*this, &Frontend::on_buffers_full));
	
	this->button_start_stop = Gtk::manage(new Gtk::Button("Start"));
	this->button_start_stop->signal_clicked().connect(
		sigc::mem_fun(*this, &Frontend::on_button_start_stop_clicked));
	
	Gtk::Button* button_open = Gtk::manage(new Gtk::Button("Open")),
	           * button_save = Gtk::manage(new Gtk::Button("Save"));
	button_open->signal_clicked().connect(sigc::mem_fun(*this, &Frontend::on_button_open_clicked));
	button_save->signal_clicked().connect(sigc::mem_fun(*this, &Frontend::on_button_save_clicked));

	Gtk::Box* box = Gtk::manage(new Gtk::Box(Gtk::ORIENTATION_HORIZONTAL)),
	        * bar = Gtk::manage(new Gtk::Box(Gtk::ORIENTATION_VERTICAL));
	
	bar->set_border_width(5); bar->set_spacing(5);
	bar->pack_start(*button_start_stop, Gtk::PACK_SHRINK);
	bar->pack_start(*button_open, Gtk::PACK_SHRINK);
	bar->pack_start(*button_save, Gtk::PACK_SHRINK);
	
	box->set_border_width(5);
	box->pack_start(*this->rec, Gtk::PACK_EXPAND_WIDGET);
	box->pack_start(*bar, Gtk::PACK_SHRINK);

	this->window = new Gtk::Window();
	this->window->set_title(this->title);
	this->window->set_default_size(640, 400);
	this->window->add(*box);
	this->window->show_all_children();
}

void Frontend::create_file_dialog()
{
	Glib::RefPtr<Gtk::FileFilter> filter = Gtk::FileFilter::create();
	filter->set_name("CSV files (.csv)");
	filter->add_pattern("*.csv"); filter->add_mime_type("text/csv");
	this->file_dialog = new Gtk::FileChooserDialog("Open .csv file", Gtk::FILE_CHOOSER_ACTION_OPEN);
	this->file_dialog->set_select_multiple(false);
	this->file_dialog->add_filter(filter);
	this->file_dialog->add_button("_Cancel", Gtk::RESPONSE_CANCEL);
	this->file_dialog->add_button("_OK", Gtk::RESPONSE_OK);
	this->file_dialog->set_transient_for(*this->window);
	this->file_dialog->set_modal(true);
}

void Frontend::on_buffers_full()
{
	this->window->set_title(this->title + " - Full...");
}

void Frontend::on_button_start_stop_clicked()
{
	if (! this->rec->is_recording()) {
		this->rec->start();
		this->button_start_stop->set_label("Stop ");
		this->window->set_title(this->title + " - Recording...");
	} else {
		this->rec->stop();
		this->button_start_stop->set_label("Start");
		this->window->set_title(this->title);
	}
}

void Frontend::on_button_open_clicked()
{
	if (this->rec->is_recording()) this->on_button_start_stop_clicked();
	
	this->file_dialog->set_title("Open .csv file");
	this->file_dialog->set_action(Gtk::FILE_CHOOSER_ACTION_OPEN);
	Gtk::ResponseType resp = (Gtk::ResponseType) this->file_dialog->run();
	this->file_dialog->close();
	if (resp != Gtk::RESPONSE_OK) return;
	
	std::string path = this->file_dialog->get_file()->get_path();
	this->rec->open_csv(path);
}

void Frontend::on_button_save_clicked()
{
	using Glib::str_has_suffix;
	#ifdef _WIN32
		const std::string Slash = "\\";
	#else
		const std::string Slash = "/";
	#endif
	
	if (this->rec->is_recording()) this->on_button_start_stop_clicked();
	
	this->file_dialog->set_title("Save as .csv file");
	this->file_dialog->set_action(Gtk::FILE_CHOOSER_ACTION_SAVE);
	Gtk::ResponseType resp = (Gtk::ResponseType) this->file_dialog->run();
	this->file_dialog->close();
	if (resp != Gtk::RESPONSE_OK) return;
	
	std::string path = this->file_dialog->get_current_folder();
	if (! str_has_suffix(path, Slash)) path += Slash;
	path += this->file_dialog->get_current_name();
	if (! str_has_suffix(path, ".csv")) path += ".csv"; 
	
	bool suc = this->rec->save_csv(path);
	if (! suc) this->window->set_title(this->title + " - Failed to save as file");
}

// used by dispatcher_gtk
void Frontend::close_window()
{
	this->window->close();	
}