xref: /phasta/phSolver/common/phasta.cc (revision ea8ce4a6ab68fa92d4dce7e8ab959398ce88b9bb)
159599516SKenneth E. Jansen #include <stdio.h>
259599516SKenneth E. Jansen #include <stdlib.h>
359599516SKenneth E. Jansen #include <string.h>
406db1583SCameron Smith #include <cassert>
5*ea8ce4a6SBenjamin Matthews 
6*ea8ce4a6SBenjamin Matthews #define OMPI_SKIP_MPICXX 1
7*ea8ce4a6SBenjamin Matthews #include <mpi.h>
8*ea8ce4a6SBenjamin Matthews 
9513954efSKenneth E. Jansen #ifdef HAVE_PETSC
10513954efSKenneth E. Jansen #include <petscsys.h>
11513954efSKenneth E. Jansen #include <petscviewer.h>
12513954efSKenneth E. Jansen #endif
13*ea8ce4a6SBenjamin Matthews 
1459599516SKenneth E. Jansen #include <sys/types.h>
1559599516SKenneth E. Jansen #include <sys/stat.h>
1659599516SKenneth E. Jansen 
1759599516SKenneth E. Jansen #if !(defined IOSTREAMH)
1859599516SKenneth E. Jansen #include <iostream>
191bfea438SCameron Smith #include <sstream>
2059599516SKenneth E. Jansen using namespace std;
2159599516SKenneth E. Jansen #endif
2259599516SKenneth E. Jansen 
2359599516SKenneth E. Jansen #ifdef intel
2459599516SKenneth E. Jansen #include <direct.h>
2559599516SKenneth E. Jansen #define chdir _chdir
2659599516SKenneth E. Jansen #else
2759599516SKenneth E. Jansen #include <unistd.h>
2859599516SKenneth E. Jansen #endif
2959599516SKenneth E. Jansen 
30*ea8ce4a6SBenjamin Matthews #include "common_c.h"
31*ea8ce4a6SBenjamin Matthews #include "Input.h"
32*ea8ce4a6SBenjamin Matthews #include "phstream.h"
33*ea8ce4a6SBenjamin Matthews #include "streamio.h"
34*ea8ce4a6SBenjamin Matthews 
35*ea8ce4a6SBenjamin Matthews #include <FCMangle.h>
36*ea8ce4a6SBenjamin Matthews #define input FortranCInterface_GLOBAL_(input,INPUT)
37*ea8ce4a6SBenjamin Matthews #define proces FortranCInterface_GLOBAL_(proces,PROCES)
38*ea8ce4a6SBenjamin Matthews #define timer FortranCInterface_GLOBAL_(timer,TIMER)
39*ea8ce4a6SBenjamin Matthews 
4059599516SKenneth E. Jansen extern "C" char phasta_iotype[80];
4159599516SKenneth E. Jansen char phasta_iotype[80];
4259599516SKenneth E. Jansen 
4359599516SKenneth E. Jansen extern int SONFATH;
4459599516SKenneth E. Jansen extern "C" void proces();
4559599516SKenneth E. Jansen extern "C" void input();
469ae91bddSCameron Smith extern int input_fform(phSolver::Input&);
4759599516SKenneth E. Jansen extern void setIOparam(); // For SyncIO
481063a5c0SCameron Smith extern "C" void initPhastaCommonVars();
4959599516SKenneth E. Jansen 
5059599516SKenneth E. Jansen int myrank; /* made file global for ease in debugging */
5159599516SKenneth E. Jansen 
5259599516SKenneth E. Jansen void
5359599516SKenneth E. Jansen catchDebugger() {
5459599516SKenneth E. Jansen     while (1) {
5559599516SKenneth E. Jansen       int debuggerPresent=0;
5659599516SKenneth E. Jansen       int fakeSTOP = 1; // please stop HERE and assign as next line
5759599516SKenneth E. Jansen       // assign or set debuggerPresent=1
5859599516SKenneth E. Jansen       if(debuggerPresent) {
5959599516SKenneth E. Jansen         break;
6059599516SKenneth E. Jansen       }
6159599516SKenneth E. Jansen     }
6259599516SKenneth E. Jansen }
6359599516SKenneth E. Jansen 
6459599516SKenneth E. Jansen // some useful debugging functions
6559599516SKenneth E. Jansen 
6659599516SKenneth E. Jansen void
6759599516SKenneth E. Jansen pdarray( void* darray , int start, int end ) {
6859599516SKenneth E. Jansen     for( int i=start; i < end; i++ ){
6959599516SKenneth E. Jansen         cout << ((double*)darray)[i] << endl;
7059599516SKenneth E. Jansen     }
7159599516SKenneth E. Jansen }
7259599516SKenneth E. Jansen 
7359599516SKenneth E. Jansen void
7459599516SKenneth E. Jansen piarray( void* iarray , int start, int end ) {
7559599516SKenneth E. Jansen     for( int i=start; i < end; i++ ){
7659599516SKenneth E. Jansen         cout << ((int*)iarray)[i] << endl;
7759599516SKenneth E. Jansen     }
7859599516SKenneth E. Jansen }
7959599516SKenneth E. Jansen 
8006db1583SCameron Smith namespace {
816b966dd8SCameron Smith   int cdToParent() {
826b966dd8SCameron Smith     if( chdir("..") ) {
836b966dd8SCameron Smith       fprintf(stderr,"could not change to the parent directory\n");
846b966dd8SCameron Smith       return 1;
856b966dd8SCameron Smith     } else {
866b966dd8SCameron Smith       return 0;
876b966dd8SCameron Smith     }
886b966dd8SCameron Smith   }
8906db1583SCameron Smith   int run(phSolver::Input& ctrl) {
903969618bSCameron Smith     int size,ierr;
913969618bSCameron Smith     char inpfilename[100];
923969618bSCameron Smith     MPI_Comm_size (MPI_COMM_WORLD, &size);
933969618bSCameron Smith     MPI_Comm_rank (MPI_COMM_WORLD, &myrank);
943969618bSCameron Smith 
953969618bSCameron Smith     workfc.numpe = size;
963969618bSCameron Smith     workfc.myrank = myrank;
97ea868eb1SCameron Smith 
981063a5c0SCameron Smith     initPhastaCommonVars();
99ea868eb1SCameron Smith     /* Input data  */
1009ae91bddSCameron Smith     ierr = input_fform(ctrl);
101ea868eb1SCameron Smith     if(!ierr){
102ea868eb1SCameron Smith       sprintf(inpfilename,"%d-procs_case/",size);
103ea868eb1SCameron Smith       if( chdir( inpfilename ) ) {
104ea868eb1SCameron Smith         cerr << "could not change to the problem directory "
105ea868eb1SCameron Smith           << inpfilename << endl;
1064c223e05SCameron Smith         return -1;
107b13e4d76SCameron Smith       }
108ea868eb1SCameron Smith       MPI_Barrier(MPI_COMM_WORLD);
109ea868eb1SCameron Smith       input();
110ea868eb1SCameron Smith       /* now we can start the solver */
111ea868eb1SCameron Smith       proces();
112ea868eb1SCameron Smith     }
113ea868eb1SCameron Smith     else{
114ea868eb1SCameron Smith       printf("error during reading ascii input \n");
115ea868eb1SCameron Smith     }
116ea868eb1SCameron Smith     MPI_Barrier(MPI_COMM_WORLD);
117ea868eb1SCameron Smith     if ( myrank == 0 ) {
118ea868eb1SCameron Smith       printf("phasta.cc - last call before finalize!\n");
119ea868eb1SCameron Smith     }
1206b966dd8SCameron Smith     if( cdToParent() )
1214c223e05SCameron Smith       return -1;
1224c223e05SCameron Smith     return timdat.lstep;
123ea868eb1SCameron Smith   }
12406db1583SCameron Smith }
12506db1583SCameron Smith 
12606db1583SCameron Smith int phasta(phSolver::Input& ctrl) {
12706db1583SCameron Smith   outpar.input_mode = 0; //FIXME magic value for posix
12806db1583SCameron Smith   outpar.output_mode = 0; //FIXME magic value for posix
12906db1583SCameron Smith   return run(ctrl);
13006db1583SCameron Smith }
13106db1583SCameron Smith 
13206db1583SCameron Smith int phasta(phSolver::Input& ctrl, grstream grs) {
13306db1583SCameron Smith   assert(grs);
13406db1583SCameron Smith   outpar.input_mode = -1; //FIXME magic value for streams
13506db1583SCameron Smith   outpar.output_mode = 1; //FIXME magic value for syncio
13606db1583SCameron Smith   streamio_set_gr(grs);
13706db1583SCameron Smith   return run(ctrl);
13806db1583SCameron Smith }
13906db1583SCameron Smith 
14006db1583SCameron Smith int phasta(phSolver::Input& ctrl, RStream* rs) {
14106db1583SCameron Smith   fprintf(stderr, "HEY! if you see this email Cameron and tell him "
14206db1583SCameron Smith       "to implement %s(...) on line %d of %s "
14306db1583SCameron Smith       "... returning an error\n", __func__, __LINE__, __FILE__);
14406db1583SCameron Smith   return -1;
14506db1583SCameron Smith }
14606db1583SCameron Smith 
14706db1583SCameron Smith int phasta(phSolver::Input& ctrl, GRStream* grs, RStream* rs) {
14806db1583SCameron Smith   outpar.input_mode = -1; //FIXME magic value for streams
14906db1583SCameron Smith   outpar.output_mode = -1; //FIXME magic value for streams
15006db1583SCameron Smith   assert(grs);
15106db1583SCameron Smith   assert(rs);
15206db1583SCameron Smith   streamio_set_gr(grs);
15306db1583SCameron Smith   streamio_set_r(rs);
15406db1583SCameron Smith   return run(ctrl);
15506db1583SCameron Smith }
1566d494944SCameron Smith 
157163d1986SCameron Smith int phasta( int argc, char *argv[] ) {
15859599516SKenneth E. Jansen     int size,ierr;
15959599516SKenneth E. Jansen     char inpfilename[100];
16059599516SKenneth E. Jansen     char* pauseDebugger = getenv("catchDebugger");
16159599516SKenneth E. Jansen     MPI_Comm_size (MPI_COMM_WORLD, &size);
16259599516SKenneth E. Jansen     MPI_Comm_rank (MPI_COMM_WORLD, &myrank);
16359599516SKenneth E. Jansen 
164513954efSKenneth E. Jansen #ifdef HAVE_PETSC
165513954efSKenneth E. Jansen     PETSC_COMM_WORLD=MPI_COMM_WORLD;
166513954efSKenneth E. Jansen     PetscInitialize(&argc,&argv,PETSC_NULL,PETSC_NULL);
167513954efSKenneth E. Jansen     PetscInitializeFortran();
168513954efSKenneth E. Jansen     PetscPopSignalHandler(); //Let us segfault in peace ;-)
169175e1b6bSKenneth E. Jansen     PetscOptionsView(NULL,PETSC_VIEWER_STDOUT_WORLD);
1706dec8975SKenneth E. Jansen // ok with Master    PetscOptionsView(NULL,PETSC_VIEWER_STDOUT_WORLD);
1716dec8975SKenneth E. Jansen // ok with 3.6x    PetscOptionsView(PETSC_VIEWER_STDOUT_WORLD);
172129f10f8SBen Matthews     if(sizeof(PetscInt) != sizeof(long long int))
173129f10f8SBen Matthews     {
174129f10f8SBen Matthews       //PetscInt and gcorp_t (gen_ncorp.c)
175129f10f8SBen Matthews       //must be the same size. hard-coded for now
176129f10f8SBen Matthews       //FIXME
177129f10f8SBen Matthews 	    if(myrank == 0)
178129f10f8SBen Matthews 	    {
179129f10f8SBen Matthews 		    printf("WARNING: PETSc Index Size Mismatch\n");
180129f10f8SBen Matthews 		    printf("WARNING: Proceed at your own risk\n");
181129f10f8SBen Matthews 	    }
182129f10f8SBen Matthews     }
183513954efSKenneth E. Jansen     MPI_Barrier(MPI_COMM_WORLD);
184513954efSKenneth E. Jansen     if(myrank == 0)
185513954efSKenneth E. Jansen     {
186513954efSKenneth E. Jansen 	    printf("PETSc Initialized\n");
187513954efSKenneth E. Jansen 	    fflush(stdout);
188513954efSKenneth E. Jansen     }
189513954efSKenneth E. Jansen #endif
19059599516SKenneth E. Jansen     workfc.numpe = size;
19159599516SKenneth E. Jansen     workfc.myrank = myrank;
19259599516SKenneth E. Jansen 
19359599516SKenneth E. Jansen #if (defined WIN32)
19459599516SKenneth E. Jansen     if(argc > 2 ){
19559599516SKenneth E. Jansen       catchDebugger();
19659599516SKenneth E. Jansen     }
19759599516SKenneth E. Jansen #endif
19859599516SKenneth E. Jansen #if (1) // ALWAYS ( defined LAUNCH_GDB ) && !( defined WIN32 )
19959599516SKenneth E. Jansen 
20059599516SKenneth E. Jansen     if ( pauseDebugger ) {
20159599516SKenneth E. Jansen 
20259599516SKenneth E. Jansen         int parent_pid = getpid();
20359599516SKenneth E. Jansen         int gdb_child = fork();
20459599516SKenneth E. Jansen         cout << "gdb_child" << gdb_child << endl;
20559599516SKenneth E. Jansen 
20659599516SKenneth E. Jansen         if( gdb_child == 0 ) {
20759599516SKenneth E. Jansen 
20859599516SKenneth E. Jansen             cout << "Debugger Process initiating" << endl;
2091bfea438SCameron Smith             stringstream exec_string;
21059599516SKenneth E. Jansen 
21159599516SKenneth E. Jansen #if ( defined decalp )
21259599516SKenneth E. Jansen             exec_string <<"xterm -e idb "
21359599516SKenneth E. Jansen                         << " -pid "<< parent_pid <<" "<< argv[0] << endl;
21459599516SKenneth E. Jansen #endif
21559599516SKenneth E. Jansen #if ( defined LINUX )
21659599516SKenneth E. Jansen             exec_string <<"xterm -e gdb"
21759599516SKenneth E. Jansen                         << " -pid "<< parent_pid <<" "<< argv[0] << endl;
21859599516SKenneth E. Jansen #endif
21959599516SKenneth E. Jansen #if ( defined SUN4 )
22059599516SKenneth E. Jansen             exec_string <<"xterm -e dbx "
22159599516SKenneth E. Jansen                         << " - "<< parent_pid <<" "<< argv[0] << endl;
22259599516SKenneth E. Jansen #endif
22359599516SKenneth E. Jansen #if ( defined IRIX )
22459599516SKenneth E. Jansen             exec_string <<"xterm -e dbx "
22559599516SKenneth E. Jansen                         << " -p "<< parent_pid <<" "<< argv[0] << endl;
22659599516SKenneth E. Jansen #endif
2271bfea438SCameron Smith             string s = exec_string.str();
2281bfea438SCameron Smith             system( s.c_str() );
22959599516SKenneth E. Jansen             exit(0);
23059599516SKenneth E. Jansen         }
23159599516SKenneth E. Jansen         catchDebugger();
23259599516SKenneth E. Jansen     }
23359599516SKenneth E. Jansen 
23459599516SKenneth E. Jansen #endif
23559599516SKenneth E. Jansen 
23659599516SKenneth E. Jansen     /* Input data  */
23759599516SKenneth E. Jansen     if(argc > 1 ){
23859599516SKenneth E. Jansen         strcpy(inpfilename,argv[1]);
23959599516SKenneth E. Jansen     } else {
24059599516SKenneth E. Jansen         strcpy(inpfilename,"solver.inp");
24159599516SKenneth E. Jansen     }
2429ae91bddSCameron Smith     string defaultConf = ".";
2439ae91bddSCameron Smith     const char* path_to_config = getenv("PHASTA_CONFIG");
2449ae91bddSCameron Smith     if(path_to_config)
2459ae91bddSCameron Smith       defaultConf = path_to_config;
2469ae91bddSCameron Smith     defaultConf.append("/input.config");
2479ae91bddSCameron Smith     string userConf(inpfilename);
2489ae91bddSCameron Smith     phSolver::Input ctrl(userConf, defaultConf);
2491063a5c0SCameron Smith     initPhastaCommonVars();
2509ae91bddSCameron Smith     ierr = input_fform(ctrl);
25159599516SKenneth E. Jansen     if(!ierr){
25259599516SKenneth E. Jansen       sprintf(inpfilename,"%d-procs_case/",size);
25359599516SKenneth E. Jansen       if( chdir( inpfilename ) ) {
25459599516SKenneth E. Jansen         cerr << "could not change to the problem directory "
25559599516SKenneth E. Jansen           << inpfilename << endl;
2564c223e05SCameron Smith         return -1;
25759599516SKenneth E. Jansen       }
2582992f42eSCameron Smith       MPI_Barrier(MPI_COMM_WORLD);
25959599516SKenneth E. Jansen       setIOparam();
2609f4aafb6SCameron Smith       outpar.input_mode = outpar.nsynciofiles; //FIXME this is awful
2619f4aafb6SCameron Smith       outpar.output_mode = outpar.nsynciofiles; //FIXME this is awful
26259599516SKenneth E. Jansen       input();
26359599516SKenneth E. Jansen       /* now we can start the solver */
26459599516SKenneth E. Jansen       proces();
26559599516SKenneth E. Jansen     }
26659599516SKenneth E. Jansen     else{
26759599516SKenneth E. Jansen         printf("error during reading ascii input \n");
26859599516SKenneth E. Jansen     }
269513954efSKenneth E. Jansen #ifdef HAVE_PETSC
270513954efSKenneth E. Jansen     PetscFinalize();
271513954efSKenneth E. Jansen #endif
27259599516SKenneth E. Jansen     MPI_Barrier(MPI_COMM_WORLD);
27359599516SKenneth E. Jansen     if ( myrank == 0 ) {
27459599516SKenneth E. Jansen       printf("phasta.cc - last call before finalize!\n");
27559599516SKenneth E. Jansen     }
2766b966dd8SCameron Smith     if( cdToParent() )
2776b966dd8SCameron Smith       return -1;
2784c223e05SCameron Smith     return timdat.lstep;
27959599516SKenneth E. Jansen }
280