- }
- p += sprintf(p, " "); /* put a separator */
- }
- INFO2("%s%s", msg, buff);
-}
-
-/** @brief ensures that a userdata on the stack is a task and returns the pointer inside the userdata */
-static m_task_t checkTask(lua_State * L, int index)
-{
- m_task_t *pi, tk;
- luaL_checktype(L, index, LUA_TTABLE);
- lua_getfield(L, index, "__simgrid_task");
- pi = (m_task_t *) luaL_checkudata(L, -1, TASK_MODULE_NAME);
- if (pi == NULL)
- luaL_typerror(L, index, TASK_MODULE_NAME);
- tk = *pi;
- if (!tk)
- luaL_error(L, "null Task");
- lua_pop(L, 1);
- return tk;
-}
-
-/* ********************************************************************************* */
-/* wrapper functions */
-/* ********************************************************************************* */
-
-/**
- * A task is either something to compute somewhere, or something to exchange between two hosts (or both).
- * It is defined by a computing amount and a message size.
- *
- */
-
-/* * * *
- * * Constructors * *
- * * * */
-/**
- * Construct an new task with the specified processing amount and amount
- * of data needed.
- *
- * @param name Task's name
- *
- * @param computeDuration A value of the processing amount (in flop) needed to process the task.
- * If 0, then it cannot be executed with the execute() method.
- * This value has to be >= 0.
- *
- * @param messageSize A value of amount of data (in bytes) needed to transfert this task.
- * If 0, then it cannot be transfered with the get() and put() methods.
- * This value has to be >= 0.
- */
-static int Task_new(lua_State * L)
-{
- const char *name = luaL_checkstring(L, 1);
- int comp_size = luaL_checkint(L, 2);
- int msg_size = luaL_checkint(L, 3);
- m_task_t msg_task = MSG_task_create(name, comp_size, msg_size, NULL);
- lua_newtable(L); /* create a table, put the userdata on top of it */
- m_task_t *lua_task = (m_task_t *) lua_newuserdata(L, sizeof(m_task_t));
- *lua_task = msg_task;
- luaL_getmetatable(L, TASK_MODULE_NAME);
- lua_setmetatable(L, -2);
- lua_setfield(L, -2, "__simgrid_task"); /* put the userdata as field of the table */
- /* remove the args from the stack */
- lua_remove(L, 1);
- lua_remove(L, 1);
- lua_remove(L, 1);
- return 1;
-}
-
-static int Task_get_name(lua_State * L)
-{
- m_task_t tk = checkTask(L, -1);
- lua_pushstring(L, MSG_task_get_name(tk));
- return 1;
-}
-
-static int Task_computation_duration(lua_State * L)
-{
- m_task_t tk = checkTask(L, -1);
- lua_pushnumber(L, MSG_task_get_compute_duration(tk));
- return 1;
-}
-
-static int Task_execute(lua_State * L)
-{
- m_task_t tk = checkTask(L, -1);
- int res = MSG_task_execute(tk);
- lua_pushnumber(L, res);
- return 1;
-}
-
-static int Task_destroy(lua_State * L)
-{
- m_task_t tk = checkTask(L, -1);
- int res = MSG_task_destroy(tk);
- lua_pushnumber(L, res);
- return 1;
-}
-
-static int Task_send(lua_State * L)
-{
- //stackDump("send ",L);
- m_task_t tk = checkTask(L, -2);
- const char *mailbox = luaL_checkstring(L, -1);
- lua_pop(L, 1); // remove the string so that the task is on top of it
- MSG_task_set_data(tk, L); // Copy my stack into the task, so that the receiver can copy the lua task directly
- MSG_error_t res = MSG_task_send(tk, mailbox);
- while (MSG_task_get_data(tk) != NULL) // Don't mess up with my stack: the receiver didn't copy the data yet
- MSG_process_sleep(0); // yield
-
- if (res != MSG_OK)
- switch (res) {
- case MSG_TIMEOUT:
- ERROR0("MSG_task_send failed : Timeout");
- break;
- case MSG_TRANSFER_FAILURE:
- ERROR0("MSG_task_send failed : Transfer Failure");
- break;
- case MSG_HOST_FAILURE:
- ERROR0("MSG_task_send failed : Host Failure ");
- break;
- default:
- ERROR0
- ("MSG_task_send failed : Unexpected error , please report this bug");
- break;
- }
- return 0;
-}
-
-static int Task_recv(lua_State * L)
-{
- m_task_t tk = NULL;
- const char *mailbox = luaL_checkstring(L, -1);
- MSG_error_t res = MSG_task_receive(&tk, mailbox);
-
- lua_State *sender_stack = MSG_task_get_data(tk);
- lua_xmove(sender_stack, L, 1); // copy the data directly from sender's stack
- MSG_task_set_data(tk, NULL);
-
- if (res != MSG_OK)
- switch (res) {
- case MSG_TIMEOUT:
- ERROR0("MSG_task_receive failed : Timeout");
- break;
- case MSG_TRANSFER_FAILURE:
- ERROR0("MSG_task_receive failed : Transfer Failure");
- break;
- case MSG_HOST_FAILURE:
- ERROR0("MSG_task_receive failed : Host Failure ");
- break;
- default:
- ERROR0
- ("MSG_task_receive failed : Unexpected error , please report this bug");
- break;
- }
-
- return 1;
-}
-
-static const luaL_reg Task_methods[] = {
- {"new", Task_new},
- {"name", Task_get_name},
- {"computation_duration", Task_computation_duration},
- {"execute", Task_execute},
- {"destroy", Task_destroy},
- {"send", Task_send},
- {"recv", Task_recv},
- {0, 0}
-};
-
-static int Task_gc(lua_State * L)
-{
- m_task_t tk = checkTask(L, -1);
- if (tk)
- MSG_task_destroy(tk);
- return 0;
-}
-
-static int Task_tostring(lua_State * L)
-{
- lua_pushfstring(L, "Task :%p", lua_touserdata(L, 1));
- return 1;
-}
-
-static const luaL_reg Task_meta[] = {
- {"__gc", Task_gc},
- {"__tostring", Task_tostring},
- {0, 0}
-};
-
-/**
- * Host
- */
-static m_host_t checkHost(lua_State * L, int index)
-{
- m_host_t *pi, ht;
- luaL_checktype(L, index, LUA_TTABLE);
- lua_getfield(L, index, "__simgrid_host");
- pi = (m_host_t *) luaL_checkudata(L, -1, HOST_MODULE_NAME);
- if (pi == NULL)
- luaL_typerror(L, index, HOST_MODULE_NAME);
- ht = *pi;
- if (!ht)
- luaL_error(L, "null Host");
- lua_pop(L, 1);
- return ht;
-}
-
-static int Host_get_by_name(lua_State * L)
-{
- const char *name = luaL_checkstring(L, 1);
- DEBUG0("Getting Host from name...");
- m_host_t msg_host = MSG_get_host_by_name(name);
- if (!msg_host) {
- luaL_error(L, "null Host : MSG_get_host_by_name failled");
- }
- lua_newtable(L); /* create a table, put the userdata on top of it */
- m_host_t *lua_host = (m_host_t *) lua_newuserdata(L, sizeof(m_host_t));
- *lua_host = msg_host;
- luaL_getmetatable(L, HOST_MODULE_NAME);
- lua_setmetatable(L, -2);
- lua_setfield(L, -2, "__simgrid_host"); /* put the userdata as field of the table */
- /* remove the args from the stack */
- lua_remove(L, 1);
- return 1;
-}
-
-
-static int Host_get_name(lua_State * L)
-{
- m_host_t ht = checkHost(L, -1);
- lua_pushstring(L, MSG_host_get_name(ht));
- return 1;
-}
-
-static int Host_number(lua_State * L)
-{
- lua_pushnumber(L, MSG_get_host_number());
- return 1;
-}
-
-static int Host_at(lua_State * L)
-{
- int index = luaL_checkinteger(L, 1);
- m_host_t host = MSG_get_host_table()[index - 1]; // lua indexing start by 1 (lua[1] <=> C[0])
- lua_newtable(L); /* create a table, put the userdata on top of it */
- m_host_t *lua_host = (m_host_t *) lua_newuserdata(L, sizeof(m_host_t));
- *lua_host = host;
- luaL_getmetatable(L, HOST_MODULE_NAME);
- lua_setmetatable(L, -2);
- lua_setfield(L, -2, "__simgrid_host"); /* put the userdata as field of the table */
- return 1;