tools/gras/stub_generator.c

   1 /*      $Id$     */
   2
   3 /* gras_stub_generator - creates the main() to use a GRAS program           */
   4
   5 /* Copyright (c) 2003,2004,2005 Martin Quinson, Arnaud Legrand.
   6    All rights reserved.                  */
   7
   8 /* This program is free software; you can redistribute it and/or modify it
   9  * under the terms of the license (GNU LGPL) which comes with this package. */
  10
  11 #include"xbt/sysdep.h"
  12 #include"xbt/dict.h"
  13 #include"xbt/error.h"
  14 #include "surf/surf_parse.h"
  15 #include "surf/surf.h"
  16
  17 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(stubgen,gras,"Stub generator");
  18
  19 const char *WARN = "/***********\n * DO NOT EDIT! THIS FILE WERE AUTOMATICALLY GENERATED FROM %s BY gras_stub_generator\n ***********/\n";
  20 const char *SIM_FILENAME = "_%s_simulator.c";
  21 const char *SIM_BINARYNAME = "%s_simulator";
  22 const char *SIM_FILENAME_LDADD = "%s_simulator_LDADD";
  23 const char *SIM_FILENAME_SOURCES = "%s_simulator_SOURCES";
  24 const char *RL_FILENAME = "_%s_%s.c";
  25 const char *RL_BINARYNAME = "%s_%s";
  26 const char *RL_FILENAME_LDADD = "%s_%s_LDADD";
  27 const char *RL_FILENAME_SOURCES = "%s_%s_SOURCES";
  28 const char *MAKEFILE_FILENAME = "%s.Makefile.am";
  29
  30 char *warning = NULL;
  31
  32 /**********************************************/
  33 /**** Generate the file for the simulator *****/
  34 /**********************************************/
  35
  36 const char *SIM_PREEMBULE =
  37 "#include <stdlib.h>\n"
  38 "#include <stdio.h>\n"
  39 "#include \"msg/msg.h\"\n"
  40 "#include <gras.h>\n"
  41 "\n"
  42 "char *gras_log=NULL;\n";
  43
  44
  45 const char *SIM_LAUNCH_FUNC =
  46 "int launch_%s(int argc, char **argv) {\n"
  47 "  char **myargv=argv;\n"
  48 "  int myargc=argc;\n"
  49 "  int i;\n"
  50 "  int retcode;\n"
  51 "    \n"
  52 "  if (gras_log) {\n"
  53 "    myargv=malloc((argc+1) * sizeof(char**));\n"
  54 "    for (i=0; i<argc; i++)\n"
  55 "      myargv[i] = argv[i];\n"
  56 "    myargv[myargc++] = gras_log;\n"
  57 "  }\n"
  58 "  retcode = %s(myargc,myargv);\n"
  59 "  if (myargv != argv)\n"
  60 "    free(myargv);\n"
  61 "  return retcode;\n"
  62 "}\n";
  63
  64 const char* SIM_MAIN_PREEMBULE =
  65 "int main (int argc,char *argv[]) {\n"
  66 "  int i,j;\n"
  67 "\n"
  68 "  /*  Simulation setup */\n"
  69 "  MSG_global_init_args(&argc,argv);\n"
  70 "  if (argc != 3) {\n"
  71 "    fprintf(stderr, \"Usage: %%s platform_file application_description.txt [--gras-log=...]\\n\",argv[0]);\n"
  72 "    exit(1);\n"
  73 "  }\n"
  74 "\n"
  75 "  MSG_set_channel_number(10); // GRAS_MAX_CHANNEL hardcoded since Alvin killed its definition\n"
  76 "  MSG_create_environment(argv[1]);\n"
  77 "\n"
  78 "  /*  Application deployment */\n";
  79
  80 const char *SIM_MAIN_POSTEMBULE = "\n"
  81 "\n"
  82 "  MSG_launch_application(argv[2]);\n"
  83 "\n"
  84 "  /*  Run the simulation */\n"
  85 "  MSG_main();\n"
  86 "\n"
  87 "  /* cleanup the place */\n"
  88 "  MSG_clean();\n"
  89 "  if (gras_log)\n"
  90 "    free(gras_log);\n"
  91 "  return 0;\n"
  92 "}\n";
  93
  94 /**********************************************/
  95 /**** Generate the file for the real life *****/
  96 /**********************************************/
  97
  98 const char *RL_CODE =
  99 "#include <stdio.h>\n"
 100 "#include <signal.h>\n"
 101 "#include <gras.h>\n"
 102 "\n"
 103 "/* user code */\n"
 104 "int %s(int argc, char *argv[]);\n"
 105 "\n"
 106 "int main(int argc, char *argv[]){\n"
 107 "  int errcode;\n"
 108 "\n"
 109 "  errcode=%s(argc,argv);\n"
 110 " \n"
 111 "  return errcode;\n"
 112 "}\n";
 113
 114 /**********************************************/
 115 /********* Parse XML deployment file **********/
 116 /**********************************************/
 117 xbt_dict_t process_function_set = NULL;
 118
 119 static void parse_process_init(void)
 120 {
 121   void *p = (void *) 1231;
 122   xbt_dict_set(process_function_set, A_process_function, p, NULL);
 123 }
 124
 125 static void generate_sim(char *project)
 126 {
 127   xbt_dict_cursor_t cursor=NULL;
 128   char *key = NULL;
 129   void *data = NULL;
 130   char *filename = NULL;
 131   FILE *OUT = NULL;
 132   filename = xbt_new(char,strlen(project) + strlen(SIM_FILENAME));
 133   sprintf(filename,SIM_FILENAME,project);
 134
 135   OUT=fopen(filename,"w");
 136   xbt_assert1(OUT, "Unable to open %s for writing",filename);
 137
 138   fprintf(OUT, "%s\n",warning);
 139   fprintf(OUT, SIM_PREEMBULE);
 140   xbt_dict_foreach(process_function_set,cursor,key,data) {
 141     fprintf(OUT,"int %s(int argc,char *argv[]);\n",key);
 142   }
 143   fprintf(OUT,"\n");
 144   xbt_dict_foreach(process_function_set,cursor,key,data) {
 145     fprintf(OUT,"int launch_%s(int argc,char *argv[]);\n",key);
 146   }
 147   fprintf(OUT, "\n%s\n",warning);
 148   xbt_dict_foreach(process_function_set,cursor,key,data) {
 149     fprintf(OUT,SIM_LAUNCH_FUNC,key,key);
 150   }
 151   fprintf(OUT, "\n%s\n",warning);
 152
 153   fprintf(OUT, SIM_MAIN_PREEMBULE);
 154   xbt_dict_foreach(process_function_set,cursor,key,data) {
 155     fprintf(OUT,"  MSG_function_register(\"%s\", launch_%s);\n",key,key);
 156   }
 157   fprintf(OUT, SIM_MAIN_POSTEMBULE);
 158   fclose(OUT);
 159   xbt_free(filename);
 160 }
 161
 162 static void generate_rl(char *project)
 163 {
 164   xbt_dict_cursor_t cursor=NULL;
 165   char *key = NULL;
 166   void *data = NULL;
 167   char *filename = NULL;
 168   FILE *OUT = NULL;
 169
 170   xbt_dict_foreach(process_function_set,cursor,key,data) {
 171     filename = xbt_new(char,strlen(project) + strlen(RL_FILENAME) + strlen(key));
 172     sprintf(filename,RL_FILENAME,project,key);
 173
 174     OUT=fopen(filename,"w");
 175     xbt_assert1(OUT, "Unable to open %s for writing",filename);
 176
 177     fprintf(OUT, "\n%s\n",warning);
 178     fprintf(OUT, RL_CODE, key,key);
 179     fprintf(OUT, "\n%s\n",warning);
 180     fclose(OUT);
 181     xbt_free(filename);
 182   }
 183 }
 184
 185 static void generate_makefile(char *project, char *deployment)
 186 {
 187   xbt_dict_cursor_t cursor=NULL;
 188   char *key = NULL;
 189   void *data = NULL;
 190   char *filename = NULL;
 191   FILE *OUT = NULL;
 192
 193   filename = xbt_new(char,strlen(project) + strlen(MAKEFILE_FILENAME));
 194   sprintf(filename,MAKEFILE_FILENAME, project);
 195
 196   OUT=fopen(filename,"w");
 197   xbt_assert1(OUT, "Unable to open %s for writing",filename);
 198
 199   fprintf(OUT, "# AUTOMAKE variable definition\n");
 200   fprintf(OUT, "INCLUDES= @CFLAGS_SimGrid@\n\n");
 201   fprintf(OUT, "PROGRAMS=");
 202   fprintf(OUT, SIM_BINARYNAME,project);
 203
 204   xbt_dict_foreach(process_function_set,cursor,key,data) {
 205     fprintf(OUT, " ");
 206     fprintf(OUT, RL_BINARYNAME, project, key);
 207   }
 208
 209   fprintf(OUT, "\n\n");
 210   fprintf(OUT, SIM_FILENAME_SOURCES,project);
 211   fprintf(OUT, "=\t");
 212   fprintf(OUT, SIM_FILENAME,project);
 213   fprintf(OUT, " %s.c\n", project);
 214   fprintf(OUT, SIM_FILENAME_LDADD, project);
 215   fprintf(OUT, "=\tpath/to/libsimgrid.a\n\n");
 216
 217   xbt_dict_foreach(process_function_set,cursor,key,data) {
 218     fprintf(OUT, RL_FILENAME_SOURCES, project,key);
 219     fprintf(OUT, "=\t");
 220     fprintf(OUT, RL_FILENAME, project,key);
 221     fprintf(OUT, " %s.c\n", project);
 222     fprintf(OUT, RL_FILENAME_LDADD, project, key);
 223     fprintf(OUT, "=\tpath/to/libgras.a\n\n");
 224   }
 225
 226   fprintf(OUT, "\n# cleanup temps\n");
 227   fprintf(OUT, "CLEANFILES= ");
 228   fprintf(OUT, SIM_FILENAME, project);
 229
 230   xbt_dict_foreach(process_function_set,cursor,key,data) {
 231     fprintf(OUT, " ");
 232     fprintf(OUT, RL_FILENAME, project,key);
 233   }
 234   fprintf(OUT, "\n");
 235
 236   fprintf(OUT, "\n# generate temps\n");
 237   fprintf(OUT, "\n# A rule to generate the source file each time the deployment file changes\n");
 238
 239   xbt_dict_foreach(process_function_set,cursor,key,data) {
 240     fprintf(OUT, RL_FILENAME, project,key);
 241     fprintf(OUT, " ");
 242   }
 243   fprintf(OUT, SIM_FILENAME, project);
 244   fprintf(OUT, ": %s\n", deployment);
 245   fprintf(OUT, "\tstub_generator %s %s >/dev/null\n", project, filename);
 246 }
 247
 248 int main(int argc, char *argv[])
 249 {
 250   char *project_name = NULL;
 251   char *deployment_file = NULL;
 252
 253   surf_init(&argc, argv);
 254
 255   xbt_assert1((argc ==3),"Usage: %s project_name deployment_file\n",argv[0]);
 256
 257   project_name = argv[1];
 258   deployment_file = argv[2];
 259
 260   process_function_set = xbt_dict_new();
 261
 262   STag_process_fun = parse_process_init;
 263   surf_parse_open(deployment_file);
 264   if(surf_parse()) xbt_assert1(0,"Parse error in %s",deployment_file);
 265   surf_parse_close();
 266
 267   warning = xbt_new(char,strlen(WARN)+strlen(deployment_file)+10);
 268   sprintf(warning,WARN,deployment_file);
 269
 270   generate_sim(project_name);
 271   generate_rl(project_name);
 272   generate_makefile(project_name, deployment_file);
 273
 274   xbt_free(warning);
 275   return 0;
 276 }