13
0
livetrax/luasession/luasession.cc

658 lines
14 KiB
C++
Raw Normal View History

2016-01-30 11:26:14 -05:00
#include <assert.h>
#include <getopt.h>
#include <stdint.h>
2016-01-30 11:26:14 -05:00
#include <cstdio>
#include <iostream>
#include <list>
2020-03-26 13:49:39 -04:00
#include <string>
2016-01-30 11:26:14 -05:00
#include <vector>
#ifdef PLATFORM_WINDOWS
# include <io.h>
# include <windows.h>
#else
# include <unistd.h>
#endif
2016-01-30 11:26:14 -05:00
#include <glibmm.h>
#include "pbd/basename.h"
2016-01-30 11:26:14 -05:00
#include "pbd/debug.h"
#include "pbd/error.h"
2020-03-26 13:49:39 -04:00
#include "pbd/event_loop.h"
2016-01-30 11:26:14 -05:00
#include "pbd/failed_constructor.h"
#include "pbd/pthread_utils.h"
#include "pbd/reallocpool.h"
#include "pbd/receiver.h"
#include "pbd/transmitter.h"
#include "pbd/win_console.h"
2016-01-30 11:26:14 -05:00
#include "ardour/ardour.h"
#include "ardour/audioengine.h"
#include "ardour/filename_extensions.h"
#include "ardour/filesystem_paths.h"
#include "ardour/luabindings.h"
#include "ardour/session.h"
#include "ardour/types.h"
#include "ardour/vst_types.h"
#include <readline/history.h>
2020-03-26 13:49:39 -04:00
#include <readline/readline.h>
2016-01-30 11:26:14 -05:00
#include "LuaBridge/LuaBridge.h"
2020-03-26 13:49:39 -04:00
#include "lua/luastate.h"
2016-01-30 11:26:14 -05:00
using namespace std;
using namespace ARDOUR;
using namespace PBD;
2020-03-26 13:49:39 -04:00
static const char* localedir = LOCALEDIR;
2016-01-30 11:26:14 -05:00
static PBD::ScopedConnectionList engine_connections;
static PBD::ScopedConnectionList session_connections;
2020-03-26 13:49:39 -04:00
static Session* session = NULL;
static LuaState* lua;
static bool keep_running = true;
static bool terminate_when_halted = false;
/* extern VST functions */
int vstfx_init (void*) { return 0; }
void vstfx_exit () {}
void vstfx_destroy_editor (VSTState*) {}
2016-01-30 11:26:14 -05:00
class LuaReceiver : public Receiver
{
2020-03-26 13:49:39 -04:00
protected:
void receive (Transmitter::Channel chn, const char* str)
{
const char* prefix = "";
switch (chn) {
case Transmitter::Debug:
/* ignore */
return;
2020-03-26 13:49:39 -04:00
case Transmitter::Info:
/* ignore */
return;
case Transmitter::Warning:
prefix = "[WARNING]: ";
break;
case Transmitter::Error:
prefix = "[ERROR]: ";
break;
2020-03-26 13:49:39 -04:00
case Transmitter::Fatal:
prefix = "[FATAL]: ";
break;
case Transmitter::Throw:
/* this isn't supposed to happen */
abort ();
}
2016-01-30 11:26:14 -05:00
2020-03-26 13:49:39 -04:00
/* note: iostreams are already thread-safe: no external
* lock required. */
2020-03-26 13:49:39 -04:00
std::cout << prefix << str << std::endl;
2016-01-30 11:26:14 -05:00
2020-03-26 13:49:39 -04:00
if (chn == Transmitter::Fatal) {
console_madness_end ();
2020-03-26 13:49:39 -04:00
::exit (9);
2016-01-30 11:26:14 -05:00
}
2020-03-26 13:49:39 -04:00
}
2016-01-30 11:26:14 -05:00
};
class MyEventLoop : public sigc::trackable, public EventLoop
{
2020-03-26 13:49:39 -04:00
public:
MyEventLoop (std::string const& name)
: EventLoop (name)
{
run_loop_thread = Glib::Threads::Thread::self ();
}
2016-01-30 11:26:14 -05:00
2020-03-26 13:49:39 -04:00
void call_slot (InvalidationRecord* ir, const boost::function<void()>& f)
{
if (Glib::Threads::Thread::self () == run_loop_thread) {
cout << string_compose ("%1/%2 direct dispatch of call slot via functor @ %3, invalidation %4\n", event_loop_name (), pthread_name (), &f, ir);
f ();
} else {
cout << string_compose ("%1/%2 queue call-slot using functor @ %3, invalidation %4\n", event_loop_name (), pthread_name (), &f, ir);
assert (!ir);
f (); // XXX TODO, queue and process during run ()
2016-01-30 11:26:14 -05:00
}
2020-03-26 13:49:39 -04:00
}
2016-01-30 11:26:14 -05:00
2020-03-26 13:49:39 -04:00
void run ()
{
; // TODO process Events, if any
}
2016-01-30 11:26:14 -05:00
2020-03-26 13:49:39 -04:00
Glib::Threads::Mutex& slot_invalidation_mutex ()
{
return request_buffer_map_lock;
}
2016-01-30 11:26:14 -05:00
2020-03-26 13:49:39 -04:00
private:
Glib::Threads::Thread* run_loop_thread;
Glib::Threads::Mutex request_buffer_map_lock;
2016-01-30 11:26:14 -05:00
};
2020-03-26 13:49:39 -04:00
static MyEventLoop* event_loop = NULL;
/* ****************************************************************************/
/* internal helper fn and callbacks */
2020-03-26 13:49:39 -04:00
static void
init ()
2016-01-30 11:26:14 -05:00
{
if (!ARDOUR::init (true, localedir)) {
cerr << "Ardour failed to initialize\n" << endl;
console_madness_end ();
2016-01-30 11:26:14 -05:00
::exit (EXIT_FAILURE);
}
assert (!event_loop);
event_loop = new MyEventLoop ("lua");
EventLoop::set_event_loop_for_thread (event_loop);
SessionEvent::create_per_thread_pool ("lua", 4096);
2016-01-30 11:26:14 -05:00
static LuaReceiver lua_receiver;
lua_receiver.listen_to (warning);
2016-01-30 11:26:14 -05:00
lua_receiver.listen_to (error);
lua_receiver.listen_to (fatal);
}
2020-03-26 13:49:39 -04:00
static void
set_session (ARDOUR::Session* s)
2016-01-30 11:26:14 -05:00
{
session = s;
2016-01-30 11:26:14 -05:00
assert (lua);
lua_State* L = lua->getState ();
LuaBindings::set_session (L, session);
2016-01-30 11:26:14 -05:00
lua->collect_garbage (); // drop references
}
2020-03-26 13:49:39 -04:00
static void
unset_session ()
2016-01-30 11:26:14 -05:00
{
session_connections.drop_connections ();
set_session (NULL);
}
2020-03-26 13:49:39 -04:00
static int
prepare_engine ()
{
AudioEngine* engine = AudioEngine::instance ();
if (!engine->current_backend ()) {
if (!engine->set_backend ("None (Dummy)", "Unit-Test", "")) {
std::cerr << "Cannot create Audio/MIDI engine\n";
return -1;
}
}
if (!engine->current_backend ()) {
std::cerr << "Cannot create Audio/MIDI engine\n";
return -1;
}
if (engine->running ()) {
engine->stop ();
}
return 0;
}
2020-03-26 13:49:39 -04:00
static int
start_engine (uint32_t rate)
{
AudioEngine* engine = AudioEngine::instance ();
if (engine->set_sample_rate (rate)) {
std::cerr << "Cannot set session's samplerate.\n";
return -1;
}
if (engine->start () != 0) {
std::cerr << "Cannot start Audio/MIDI engine\n";
return -1;
}
return 0;
}
2020-03-26 13:49:39 -04:00
static Session*
_create_session (string dir, string state, uint32_t rate) // throws
{
if (prepare_engine ()) {
return 0;
}
std::string s = Glib::build_filename (dir, state + statefile_suffix);
if (Glib::file_test (dir, Glib::FILE_TEST_EXISTS)) {
std::cerr << "Session already exists: " << s << "\n";
return 0;
}
if (start_engine (rate)) {
return 0;
}
// TODO add option/bindings for this
BusProfile bus_profile;
bus_profile.master_out_channels = 2;
2020-03-26 13:49:39 -04:00
AudioEngine* engine = AudioEngine::instance ();
Session* session = new Session (*engine, dir, state, &bus_profile);
return session;
}
2020-03-26 13:49:39 -04:00
static Session*
_load_session (string const& dir, string state) // throws
2016-01-30 11:26:14 -05:00
{
if (prepare_engine ()) {
2016-01-30 11:26:14 -05:00
return 0;
}
if (state.empty ()) {
state = Session::get_snapshot_from_instant (dir);
}
if (state.empty ()) {
state = PBD::basename_nosuffix (dir);
}
2020-03-26 13:49:39 -04:00
float sr;
2016-01-30 11:26:14 -05:00
SampleFormat sf;
2020-03-26 13:49:39 -04:00
std::string v;
std::string s = Glib::build_filename (dir, state + statefile_suffix);
2016-01-30 11:26:14 -05:00
if (!Glib::file_test (dir, Glib::FILE_TEST_EXISTS)) {
std::cerr << "Cannot find session: " << s << "\n";
return 0;
}
if (Session::get_info_from_path (s, sr, sf, v) != 0) {
2016-01-30 11:26:14 -05:00
std::cerr << "Cannot get samplerate from session.\n";
return 0;
}
if (start_engine (sr)) {
2016-01-30 11:26:14 -05:00
return 0;
}
2020-03-26 13:49:39 -04:00
AudioEngine* engine = AudioEngine::instance ();
Session* session = new Session (*engine, dir, state);
2016-01-30 11:26:14 -05:00
return session;
}
/* ****************************************************************************/
/* lua bound functions */
2020-03-26 13:49:39 -04:00
static Session*
create_session (string dir, string state, uint32_t rate)
{
Session* s = 0;
if (session) {
2020-03-26 13:49:39 -04:00
cerr << "Session already open"
<< "\n";
return 0;
}
try {
s = _create_session (dir, state, rate);
} catch (failed_constructor& e) {
cerr << "failed_constructor: " << e.what () << "\n";
return 0;
} catch (AudioEngine::PortRegistrationFailure& e) {
cerr << "PortRegistrationFailure: " << e.what () << "\n";
return 0;
} catch (exception& e) {
cerr << "exception: " << e.what () << "\n";
return 0;
} catch (...) {
cerr << "unknown exception.\n";
return 0;
}
Glib::usleep (1000000); // allow signal propagation, callback/thread-pool setup
if (!s) {
return 0;
}
set_session (s);
s->DropReferences.connect_same_thread (session_connections, &unset_session);
return s;
}
2020-03-26 13:49:39 -04:00
static Session*
load_session (string dir, string state)
2016-01-30 11:26:14 -05:00
{
Session* s = 0;
if (session) {
2020-03-26 13:49:39 -04:00
cerr << "Session already open"
<< "\n";
2016-01-30 11:26:14 -05:00
return 0;
}
try {
s = _load_session (dir, state);
} catch (failed_constructor& e) {
cerr << "failed_constructor: " << e.what () << "\n";
return 0;
} catch (AudioEngine::PortRegistrationFailure& e) {
cerr << "PortRegistrationFailure: " << e.what () << "\n";
return 0;
} catch (exception& e) {
cerr << "exception: " << e.what () << "\n";
return 0;
} catch (...) {
cerr << "unknown exception.\n";
return 0;
}
Glib::usleep (1000000); // allow signal propagation, callback/thread-pool setup
if (!s) {
return 0;
}
2016-01-30 11:26:14 -05:00
set_session (s);
s->DropReferences.connect_same_thread (session_connections, &unset_session);
return s;
}
2020-03-26 13:49:39 -04:00
static int
set_debug_options (const char* opts)
{
return PBD::parse_debug_options (opts);
}
2020-03-26 13:49:39 -04:00
static void
close_session ()
2016-01-30 11:26:14 -05:00
{
delete session;
assert (!session);
2016-01-30 11:26:14 -05:00
}
2020-03-26 13:49:39 -04:00
static int
close_session_lua (lua_State* L)
2016-01-30 11:26:14 -05:00
{
if (!session) {
2020-03-26 13:49:39 -04:00
cerr << "No open session"
<< "\n";
2016-01-30 11:26:14 -05:00
return 0;
}
close_session ();
return 0;
}
2020-03-26 13:49:39 -04:00
static void
delay (float d)
{
2016-01-30 11:26:14 -05:00
if (d > 0) {
Glib::usleep (d * 1000000);
}
}
2020-03-26 13:49:39 -04:00
static int
do_quit (lua_State* L)
{
keep_running = false;
return 0;
}
static void
engine_halted (const char* err)
{
if (terminate_when_halted) {
cerr << "Engine halted: " << err << "\n";
console_madness_end ();
::exit (EXIT_FAILURE);
}
}
/* ****************************************************************************/
2020-03-26 13:49:39 -04:00
static void
my_lua_print (std::string s)
{
std::cout << s << "\n";
}
2020-03-26 13:49:39 -04:00
static void
setup_lua ()
2016-01-30 11:26:14 -05:00
{
assert (!lua);
lua = new LuaState ();
lua->Print.connect (&my_lua_print);
lua_State* L = lua->getState ();
LuaBindings::stddef (L);
LuaBindings::common (L);
LuaBindings::session (L);
2016-03-19 11:55:51 -04:00
LuaBindings::osc (L);
2016-01-30 11:26:14 -05:00
luabridge::getGlobalNamespace (L)
2020-03-26 13:49:39 -04:00
.beginNamespace ("_G")
.addFunction ("create_session", &create_session)
.addFunction ("load_session", &load_session)
.addFunction ("close_session", &close_session)
.addFunction ("sleep", &delay)
.addFunction ("quit", &do_quit)
.addFunction ("set_debug_options", &set_debug_options)
.endNamespace ();
2016-01-30 11:26:14 -05:00
// add a Session::close() method
2016-01-30 11:26:14 -05:00
luabridge::getGlobalNamespace (L)
2020-03-26 13:49:39 -04:00
.beginNamespace ("ARDOUR")
.beginClass<Session> ("Session")
.addExtCFunction ("close", &close_session_lua)
.endClass ()
.endNamespace ();
2016-01-30 11:26:14 -05:00
// push instance to global namespace (C++ lifetime)
2020-03-26 13:49:39 -04:00
luabridge::push<AudioEngine*> (L, AudioEngine::create ());
2016-01-30 11:26:14 -05:00
lua_setglobal (L, "AudioEngine");
AudioEngine::instance ()->stop ();
AudioEngine::instance()->Halted.connect_same_thread (engine_connections, boost::bind (&engine_halted, _1));
2016-01-30 11:26:14 -05:00
}
static int
2020-03-26 13:49:39 -04:00
incomplete (lua_State* L, int status)
{
if (status == LUA_ERRSYNTAX) {
2020-03-26 13:49:39 -04:00
size_t lmsg;
const char* msg = lua_tolstring (L, -1, &lmsg);
if (lmsg >= 5 && strcmp (msg + lmsg - 5, "<eof>") == 0) {
lua_pop (L, 1);
return 1;
}
}
return 0;
}
static void
interactive_interpreter ()
2016-01-30 11:26:14 -05:00
{
using_history ();
2020-03-26 13:49:39 -04:00
std::string histfile = Glib::build_filename (user_config_directory (), "/luahist");
2016-01-30 11:26:14 -05:00
rl_bind_key ('\t', rl_insert); // disable completion
2020-03-26 13:49:39 -04:00
read_history (histfile.c_str ());
2016-01-30 11:26:14 -05:00
2020-03-26 13:49:39 -04:00
char* line = NULL;
while (keep_running && (line = readline ("> "))) {
2020-03-26 13:49:39 -04:00
event_loop->run ();
2016-01-30 11:26:14 -05:00
if (!strcmp (line, "quit")) {
2020-03-26 13:49:39 -04:00
free (line);
line = NULL;
2016-01-30 11:26:14 -05:00
break;
}
2016-01-30 11:26:14 -05:00
if (strlen (line) == 0) {
2020-03-26 13:49:39 -04:00
free (line);
line = NULL;
2016-01-30 11:26:14 -05:00
continue;
}
do {
2020-03-26 13:49:39 -04:00
LuaState lt;
lua_State* L = lt.getState ();
int status = luaL_loadbuffer (L, line, strlen (line), "=stdin");
if (!incomplete (L, status)) {
break;
}
2020-03-26 13:49:39 -04:00
char* l2 = readline (">> ");
if (!l2) {
break;
}
if (strlen (l2) == 0) {
continue;
}
2020-03-26 13:49:39 -04:00
line = (char*)realloc ((void*)line, (strlen (line) + strlen (l2) + 2) * sizeof (char));
strcat (line, "\n");
strcat (line, l2);
free (l2);
} while (1);
if (lua->do_command (line)) {
/* error */
2020-03-26 13:49:39 -04:00
free (line);
line = NULL;
continue;
2016-01-30 11:26:14 -05:00
}
add_history (line);
2020-03-26 13:49:39 -04:00
event_loop->run ();
free (line);
line = NULL;
2016-01-30 11:26:14 -05:00
}
2016-04-24 08:41:07 -04:00
free (line);
2016-01-30 11:26:14 -05:00
printf ("\n");
2020-03-26 13:49:39 -04:00
write_history (histfile.c_str ());
}
static bool
is_tty ()
{
#ifdef PLATFORM_WINDOWS
return _isatty (_fileno (stdin));
#else
return isatty (0);
#endif
}
2020-03-26 13:49:39 -04:00
static void
usage ()
{
printf ("ardour-lua - interactive Ardour Lua interpreter.\n\n");
printf ("Usage: ardour-lua [ OPTIONS ] [ file [args] ]\n\n");
/* 1 2 3 4 5 6 7 8
*2345678901234567890123456789012345678901234567890123456789012345678901234567890*/
printf ("Options:\n\
-h, --help display this help and exit\n\
-i, --interactive enter interactive mode after executing 'script',\n\
force the interpreter to run interactively\n\
-X, --exit-when-halted terminate when the audio-engine halts\n\
unexpectedly (disconnect, or too many xruns)\n\
-V, --version print version information and exit\n\
\n");
printf ("\n\
Ardour at your finger tips...\n\
\n");
printf ("Report bugs to <http://tracker.ardour.org/>\n"
"Website: <http://ardour.org/>\n");
console_madness_end ();
::exit (EXIT_SUCCESS);
}
2020-03-26 13:49:39 -04:00
int
main (int argc, char** argv)
{
const char* optstring = "hiVX";
const struct option longopts[] = {
{ "help", 0, 0, 'h' },
{ "interactive", 0, 0, 'i' },
{ "version", 0, 0, 'V' },
{ "exit-when-halted", 0, 0, 'X' },
};
bool interactive = false;
console_madness_begin ();
int c = 0;
while (EOF != (c = getopt_long (argc, argv,
2020-03-26 13:49:39 -04:00
optstring, longopts, (int*)0))) {
switch (c) {
case 'h':
usage ();
break;
case 'i':
interactive = true;
break;
case 'V':
printf ("ardour-lua version %s\n\n", VERSIONSTRING);
printf ("Copyright (C) GPL 2015-2020 Robin Gareus <robin@gareus.org>\n");
console_madness_end ();
exit (EXIT_SUCCESS);
break;
case 'X':
terminate_when_halted = true;
break;
default:
cerr << "Error: unrecognized option. See --help for usage information.\n";
console_madness_end ();
::exit (EXIT_FAILURE);
break;
}
}
init ();
setup_lua ();
{
/* push arguments to script, use scoped LuaRef */
lua_State* L = lua->getState ();
luabridge::LuaRef arg (luabridge::newTable (L));
for (int i = 1; i < argc - optind; ++i) {
arg[i] = std::string (argv[i + optind]);
}
luabridge::push (L, arg);
lua_setglobal (L, "arg");
}
int res = 0;
if (argc > optind && 0 != strcmp (argv[optind], "-")) {
res = lua->do_file (argv[optind]);
if (!interactive) {
keep_running = false;
}
}
if (!keep_running) {
/* continue to exit */
} else if (is_tty () || interactive) {
interactive_interpreter ();
} else {
res = luaL_dofile (lua->getState (), NULL);
}
2016-01-30 11:26:14 -05:00
if (session) {
2016-01-30 11:26:14 -05:00
close_session ();
}
engine_connections.drop_connections ();
delete lua;
lua = NULL;
AudioEngine::instance ()->stop ();
AudioEngine::destroy ();
ARDOUR::cleanup ();
delete event_loop;
pthread_cancel_all ();
console_madness_end ();
return res;
2016-01-30 11:26:14 -05:00
}