X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/2533feb9860c10978740930dde3af6eb3e926d40..d74bb413a4b25a580d7cfc96065b638292a2e679:/src/bindings/lua/simgrid_lua.c

diff --git a/src/bindings/lua/simgrid_lua.c b/src/bindings/lua/simgrid_lua.c
index da274aea16..7e442ff138 100644
--- a/src/bindings/lua/simgrid_lua.c
+++ b/src/bindings/lua/simgrid_lua.c
@@ -15,6 +15,7 @@
 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(lua,bindings,"Lua Bindings");
 
 #define TASK_MODULE_NAME "simgrid.Task"
+#define HOST_MODULE_NAME "simgrid.Host"
 
 /* ********************************************************************************* */
 /*                            helper functions                                       */
@@ -25,12 +26,9 @@ static void stackDump (const char *msg, lua_State *L) {
   char *p=buff;
   int i;
   int top = lua_gettop(L);
-//  void *ptr;
+
   fflush(stdout);
   p+=sprintf(p,"STACK(top=%d): ",top);
-/*  lua_getglobal(L,"vardump");
-  xbt_assert0(lua_isfunction(L,-1), "Vardump not found");
-*/
 
   for (i = 1; i <= top; i++) {  /* repeat for each level */
 
@@ -84,39 +82,36 @@ static m_task_t checkTask (lua_State *L,int index) {
 }
 
 /** @brief leaves a new userdata on top of the stack, sets its metatable, and sets the Task pointer inside the userdata */
-static m_task_t *pushTask (lua_State *L,m_task_t tk) {
-
+// NOT USED
+/*static m_task_t *pushTask (lua_State *L,m_task_t tk) {
   m_task_t *pi = NULL;
   pi = (m_task_t*)lua_newuserdata(L,sizeof(m_task_t));
   *pi=tk;
-
   DEBUG1("push lua task with Name : %s \n",MSG_task_get_name(*pi));
   luaL_getmetatable(L,TASK_MODULE_NAME);
   lua_setmetatable(L,-2);
   return pi;
-
-}
+}*/
 
 /* ********************************************************************************* */
 /*                           wrapper functions                                       */
 /* ********************************************************************************* */
 
+/**
+ *  Task
+ */
+
 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);
   INFO0("Creating task");
-
   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);
@@ -132,13 +127,13 @@ static int Task_get_name(lua_State *L) {
 }
 
 static int Task_computation_duration(lua_State *L){
-  m_task_t tk = checkTask(L,1);
+  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);
+  m_task_t tk = checkTask(L,-1);
   int res = MSG_task_execute(tk);
   lua_pushnumber(L,res);
   return 1;
@@ -151,13 +146,13 @@ static int Task_destroy(lua_State *L) {
   return 1;
 }
 static int Task_send(lua_State *L)  {
-  stackDump("send ",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_has_data(tk)) // Don't mess up with my stack: the receiver didn't copy the data yet
+  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) {
@@ -176,6 +171,7 @@ static int Task_send(lua_State *L)  {
     }
   return 0;
 }
+
 static int Task_recv(lua_State *L)  {
   m_task_t tk = NULL;
   const char *mailbox = luaL_checkstring(L,-1);
@@ -214,9 +210,8 @@ static const luaL_reg Task_methods[] = {
     {0,0}
 };
 static int Task_gc(lua_State *L) {
-  m_task_t tk=checkTask(L,1);
+  m_task_t tk=checkTask(L,-1);
   if (tk) MSG_task_destroy(tk);
-  //printf("GoodBye Task(%p)\n",lua_touserdata(L,1));
   return 0;
 }
 
@@ -225,13 +220,89 @@ static int Task_tostring(lua_State *L) {
   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 const luaL_reg Host_methods[] = {
+    {"getByName",   Host_get_by_name},
+    {"name",  		Host_get_name},
+    {"number",    	Host_number},
+    {0,0}
+};
+
+static int Host_gc(lua_State *L)
+{
+  m_host_t ht = checkHost(L,-1);
+  if (ht) ht = NULL;
+  return 0;
+}
+
+static int Host_tostring(lua_State *L)
+{
+  lua_pushfstring(L,"Host :%p",lua_touserdata(L,1));
+  return 1;
+}
+
+static const luaL_reg Host_meta[] = {
+    {"__gc",  Host_gc},
+    {"__tostring",  Host_tostring},
+    {0,0}
+};
+
 /*
  * Environment related
  */
@@ -239,7 +310,6 @@ extern lua_State *simgrid_lua_state;
 
 static int run_lua_code(int argc,char **argv) {
   DEBUG1("Run lua code %s",argv[0]);
-//  fprintf(stderr,"Run lua code %s\n", (argv ? argv[0] : "(null)"));
   lua_State *L = lua_newthread(simgrid_lua_state);
   int ref = luaL_ref(simgrid_lua_state, LUA_REGISTRYINDEX); // protect the thread from being garbage collected
   int res = 1;
@@ -362,7 +432,7 @@ int luaopen_simgrid(lua_State* L) {
 
   /* register the core C functions to lua */
   luaL_register(L, "simgrid", simgrid_funcs);
-  /* register the tasks to lua */
+  /* register the task methods to lua */
   luaL_openlib(L,TASK_MODULE_NAME,Task_methods,0); //create methods table,add it to the globals
   luaL_newmetatable(L,TASK_MODULE_NAME); //create metatable for Task,add it to the Lua registry
   luaL_openlib(L,0,Task_meta,0);// fill metatable
@@ -374,6 +444,18 @@ int luaopen_simgrid(lua_State* L) {
   lua_rawset(L,-3); //hide metatable:metatable.__metatable = methods
   lua_pop(L,1);   //drop metatable
 
+  /* register the hosts methods to lua*/
+  luaL_openlib(L,HOST_MODULE_NAME,Host_methods,0);
+  luaL_newmetatable(L,HOST_MODULE_NAME);
+  luaL_openlib(L,0,Host_meta,0);
+  lua_pushliteral(L,"__index");
+  lua_pushvalue(L,-3);
+  lua_rawset(L,-3);
+  lua_pushliteral(L,"__metatable");
+  lua_pushvalue(L,-3);
+  lua_rawset(L,-3);
+  lua_pop(L,1);
+
   /* Keep the context mechanism informed of our lua world today */
   simgrid_lua_state = L;