Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
GRAS is not the only one using libdata. AMOK does too, and other could. So, it is...
[simgrid.git] / src / gras / Virtu / sg_process.c
1 /* $Id$ */
2
3 /* process_sg - GRAS process handling on simulator                          */
4
5 /* Copyright (c) 2003, 2004 Martin Quinson. All rights reserved.            */
6
7 /* This program is free software; you can redistribute it and/or modify it
8  * under the terms of the license (GNU LGPL) which comes with this package. */
9
10 #include "xbt/ex.h"
11 #include "gras_modinter.h" /* module initialization interface */
12 #include "gras/Virtu/virtu_sg.h"
13 #include "gras/Msg/msg_interface.h" /* For some checks at simulation end */
14 #include "gras/Transport/transport_interface.h" /* For some checks at simulation end */
15
16 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(gras_virtu_process);
17
18 void
19 gras_process_init() {
20   gras_hostdata_t *hd=(gras_hostdata_t *)MSG_host_get_data(MSG_host_self());
21   gras_procdata_t *pd=xbt_new0(gras_procdata_t,1);
22   gras_trp_procdata_t trp_pd;
23   gras_sg_portrec_t prmeas,pr;
24   int i;
25   
26   if (MSG_process_set_data(MSG_process_self(),(void*)pd) != MSG_OK)
27     THROW0(system_error,0,"Error in MSG_process_set_data()");
28    
29   gras_procdata_init();
30
31   if (!hd) {
32     /* First process on this host */
33     hd=xbt_new(gras_hostdata_t,1);
34     hd->refcount = 1;
35     hd->ports = xbt_dynar_new(sizeof(gras_sg_portrec_t),NULL);
36
37     memset(hd->proc, 0, sizeof(hd->proc[0]) * XBT_MAX_CHANNEL); 
38
39     if (MSG_host_set_data(MSG_host_self(),(void*)hd) != MSG_OK)
40       THROW0(system_error,0,"Error in MSG_host_set_data()");
41   } else {
42     hd->refcount++;
43   }
44   
45   /* take a free channel for this process */
46   trp_pd = (gras_trp_procdata_t)gras_libdata_by_name("gras_trp");
47   for (i=0; i<XBT_MAX_CHANNEL && hd->proc[i]; i++);
48   if (i == XBT_MAX_CHANNEL) 
49     THROW2(system_error,0,
50            "Can't add a new process on %s, because all channels are already in use. Please increase MAX CHANNEL (which is %d for now) and recompile GRAS.",
51             MSG_host_get_name(MSG_host_self()),XBT_MAX_CHANNEL);
52
53   trp_pd->chan = i;
54   hd->proc[ i ] = MSG_process_self_PID();
55
56   /* regiter it to the ports structure */
57   pr.port = -1;
58   pr.tochan = i;
59   pr.meas = 0;
60   xbt_dynar_push(hd->ports,&pr);
61
62   /* take a free meas channel for this process */
63   for (i=0; i<XBT_MAX_CHANNEL && hd->proc[i]; i++);
64   if (i == XBT_MAX_CHANNEL) {
65     THROW2(system_error,0,
66            "Can't add a new process on %s, because all channels are already in use. Please increase MAX CHANNEL (which is %d for now) and recompile GRAS.",
67             MSG_host_get_name(MSG_host_self()),XBT_MAX_CHANNEL);
68   }
69   trp_pd->measChan = i;
70
71   hd->proc[ i ] = MSG_process_self_PID();
72
73   /* register it to the ports structure */
74   prmeas.port = -1;
75   prmeas.tochan = i;
76   prmeas.meas = 1;
77   xbt_dynar_push(hd->ports,&prmeas);
78
79   VERB2("Creating process '%s' (%d)",
80            MSG_process_get_name(MSG_process_self()),
81            MSG_process_self_PID());
82 }
83
84 void
85 gras_process_exit() {
86   gras_hostdata_t *hd=(gras_hostdata_t *)MSG_host_get_data(MSG_host_self());
87   gras_procdata_t *pd=(gras_procdata_t*)MSG_process_get_data(MSG_process_self());
88
89   gras_msg_procdata_t msg_pd=(gras_msg_procdata_t)gras_libdata_by_name("gras_msg");
90   gras_trp_procdata_t trp_pd=(gras_trp_procdata_t)gras_libdata_by_name("gras_trp");
91   int myPID=MSG_process_self_PID();
92   int cpt;
93   gras_sg_portrec_t pr;
94
95   xbt_assert0(hd,"Run gras_process_init (ie, gras_init)!!");
96
97   VERB2("GRAS: Finalizing process '%s' (%d)",
98         MSG_process_get_name(MSG_process_self()),MSG_process_self_PID());
99
100   if (xbt_dynar_length(msg_pd->msg_queue))
101     WARN1("process %d terminated, but some messages are still queued",
102           MSG_process_self_PID());
103
104   for (cpt=0; cpt< XBT_MAX_CHANNEL; cpt++)
105     if (myPID == hd->proc[cpt])
106       hd->proc[cpt] = 0;
107
108   xbt_dynar_foreach(hd->ports, cpt, pr) {
109     if (pr.port == trp_pd->chan || pr.port == trp_pd->measChan) {
110       xbt_dynar_cursor_rm(hd->ports, &cpt);
111     }
112   }
113
114   if ( ! --(hd->refcount)) {
115     xbt_dynar_free(&hd->ports);
116     free(hd);
117   }
118   gras_procdata_exit();
119   free(pd);
120 }
121
122 /* **************************************************************************
123  * Process data
124  * **************************************************************************/
125
126 gras_procdata_t *gras_procdata_get(void) {
127   gras_procdata_t *pd=
128     (gras_procdata_t *)MSG_process_get_data(MSG_process_self());
129
130   xbt_assert0(pd,"Run gras_process_init! (ie, gras_init)");
131
132   return pd;
133 }
134
135 const char* xbt_procname(void) {
136   const char *res = NULL;
137   m_process_t process = MSG_process_self();
138   if ((process != NULL) && (process->simdata))
139     res = MSG_process_get_name(process);
140   if (res) 
141     return res;
142   else
143     return "";
144 }
145
146 long int gras_os_getpid(void) {
147   m_process_t process = MSG_process_self();
148   if ((process != NULL) && (process->simdata))
149     return (long int)MSG_process_get_PID(MSG_process_self());
150   else
151     return (long int)0;
152 }