ComputeThread.hh

Go to the documentation of this file.

#pragma once

#include <websocketpp/config/asio_no_tls_client.hpp>
#include <websocketpp/client.hpp>
#include <websocketpp/common/thread.hpp>
#include <websocketpp/common/functional.hpp>
#include <thread>
#include <glibmm/spawn.h>

typedef websocketpp::client<websocketpp::config::asio_client> WSClient;
typedef websocketpp::config::asio_client::message_type::ptr message_ptr;
typedef websocketpp::lib::lock_guard<websocketpp::lib::mutex> scoped_lock;

#include "DataCell.hh"

namespace cadabra {
    
    class GUIBase;
    class DocumentThread;

    
    class ComputeThread {
        public:
            
            ComputeThread();
            ComputeThread(const ComputeThread& )=delete; // You cannot copy this object
            ~ComputeThread();

            void set_master(GUIBase *, DocumentThread *);
            

            void run(); 
            

            void execute_cell(DTree::iterator);


            void stop();


            void restart_kernel();

            // Determine if there are still cells running on the server.
            // FIXME: this does not guarantee thread-safety but at the moment
            // is only used for updating status bars etc.
            // FIXME: can be moved to DocumentThread.

            int number_of_cells_executing(void) const;


            void terminate();
            
        private:
            GUIBase        *gui;
            DocumentThread *docthread;

            // For debugging purposes, we keep record of the gui thread id,
            // so that we can flag when code runs on the wrong thread.
            // Gets initialised in the ComputeThread constructor.
            std::thread::id  gui_thread_id; 

            // Keeping track of cells which are running on the server, in
            // a form which allows us to look them up quickly based only
            // on the id (which is all that the server knows about). 

            // FIXME: moving this away into documentthread, so that we only need to refer to id's.
            std::map<DataCell::id_t, DTree::iterator> running_cells;

            // WebSocket++ things.
            WSClient wsclient;
            bool     connection_is_open, restarting_kernel;
            WSClient::connection_ptr    connection;
            websocketpp::connection_hdl our_connection_hdl;
            void init();
            void try_connect();
            void try_spawn_server();
            void on_open(websocketpp::connection_hdl hdl);
            void on_fail(websocketpp::connection_hdl hdl);
            void on_close(websocketpp::connection_hdl hdl);
            void on_message(websocketpp::connection_hdl hdl, message_ptr msg);

            void cell_finished_running(DataCell::id_t);
            
            void all_cells_nonrunning();


            // Self-started server
            Glib::Pid       server_pid;
            int             server_stdout, server_stderr;
            unsigned short  port;
    };

}