source: TOOLS/cmd_launch.c @ 2133

Last change on this file since 2133 was 2133, checked in by acosce, 11 years ago

New tools allows packing simulation already pack (so pack with a bigger frequency)

File size: 2.1 KB
Line 
1#include <mpi.h>
2#include <stdio.h>
3#include <stdlib.h>
4#include <string.h>
5
6#define CMD_MAXLEN 4096
7#define CMD_MASTER 0
8#define TAG_AVAIL 1
9#define TAG_CMD   2
10
11void read_and_exec(char *filename);
12void read_and_send(int slaves, char *filename);
13void recv_and_exec(int rank);
14void read_next_command(FILE *fl, char *cmd);
15void exec_command(char *cmd, int rank);
16
17int main(int argc,char **argv)
18{
19        FILE *fl;
20        int rank,size;
21
22        MPI_Init(&argc,&argv);
23        MPI_Comm_rank(MPI_COMM_WORLD,&rank);
24        MPI_Comm_size(MPI_COMM_WORLD,&size); 
25        if (rank == CMD_MASTER)
26                printf("master is %d , nb slaves: %d\n",
27                       CMD_MASTER,size-1);     
28
29        if (size == 1)
30                read_and_exec(argv[1]);
31        else if(size != 1 && rank == CMD_MASTER)
32                read_and_send(size-1,argv[1]);
33        else
34                recv_and_exec(rank);
35        MPI_Finalize();
36}
37
38void read_and_exec(char *filename)
39{
40        FILE *fl;
41        char cmd[CMD_MAXLEN];
42        double t;
43
44        fl = fopen(filename,"r");
45        do {
46          read_next_command(fl,cmd);
47                if(!strlen(cmd))
48                        break;
49                exec_command(cmd,CMD_MASTER);
50        } while (strlen(cmd));
51        fclose(fl);
52}
53
54void read_and_send(int slaves,char *filename)
55{
56        int i, num;
57        MPI_Status status;
58        FILE *fl;
59        char cmd[CMD_MAXLEN];
60
61        fl = fopen(filename,"r");
62        i=0;
63        do {
64                read_next_command(fl,cmd);
65                MPI_Recv(&num,1,MPI_INT,MPI_ANY_SOURCE,
66                         TAG_AVAIL,MPI_COMM_WORLD,&status);
67                MPI_Send(cmd,CMD_MAXLEN,MPI_CHAR,num,
68                         TAG_CMD,MPI_COMM_WORLD);
69                if (!strlen(cmd)) /* terminate command sent */
70                        i++;
71        } while (i < slaves);
72        fclose(fl);
73}
74
75void recv_and_exec(int rank)
76{
77        char cmd[CMD_MAXLEN]; 
78        MPI_Status status; 
79
80        do {
81                MPI_Send(&rank,1,MPI_INT,CMD_MASTER,
82                         TAG_AVAIL,MPI_COMM_WORLD); 
83                MPI_Recv(cmd,CMD_MAXLEN,MPI_CHAR,CMD_MASTER,
84                         TAG_CMD,MPI_COMM_WORLD,&status); 
85                if(!strlen(cmd)) /* terminate command */
86                        break;
87                exec_command(cmd,rank);
88        } while (strlen(cmd));
89}
90
91void read_next_command(FILE *fl,char *cmd)
92{
93        if(!fgets(cmd,CMD_MAXLEN,fl)) cmd[0]='\0';
94        if(cmd[strlen(cmd)-1]=='\n') cmd[strlen(cmd)-1]='\0';
95}
96
97void exec_command(char *cmd, int rank)
98{
99        double t;
100        int r;
101
102        t = MPI_Wtime(); 
103        r=system(cmd);
104        t = MPI_Wtime() - t; 
105        fprintf(stderr,"#executed by process\t%d in %gs\twith status\t%d : %s\n",
106                rank,t,r,cmd);
107}
Note: See TracBrowser for help on using the repository browser.