xref: /phasta/phSolver/common/phasta.cc (revision 81372b97728c63721aa3da8fb2fcae64cd0c452b)
159599516SKenneth E. Jansen #include <stdio.h>
259599516SKenneth E. Jansen #include <stdlib.h>
359599516SKenneth E. Jansen #include <string.h>
406db1583SCameron Smith #include <cassert>
5ea8ce4a6SBenjamin Matthews 
6ea8ce4a6SBenjamin Matthews #define OMPI_SKIP_MPICXX 1
7ea8ce4a6SBenjamin Matthews #include <mpi.h>
8ea8ce4a6SBenjamin Matthews 
9513954efSKenneth E. Jansen #ifdef HAVE_PETSC
10513954efSKenneth E. Jansen #include <petscsys.h>
11513954efSKenneth E. Jansen #include <petscviewer.h>
12513954efSKenneth E. Jansen #endif
13ea8ce4a6SBenjamin 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 
30ea8ce4a6SBenjamin Matthews #include "common_c.h"
31ea8ce4a6SBenjamin Matthews #include "Input.h"
32*81372b97SCameron Smith #include "phiostats.h"
33ea8ce4a6SBenjamin Matthews #include "phstream.h"
34ea8ce4a6SBenjamin Matthews #include "streamio.h"
35ea8ce4a6SBenjamin Matthews 
36ea8ce4a6SBenjamin Matthews #include <FCMangle.h>
37ea8ce4a6SBenjamin Matthews #define input FortranCInterface_GLOBAL_(input,INPUT)
38ea8ce4a6SBenjamin Matthews #define proces FortranCInterface_GLOBAL_(proces,PROCES)
39ea8ce4a6SBenjamin Matthews #define timer FortranCInterface_GLOBAL_(timer,TIMER)
40ea8ce4a6SBenjamin Matthews 
4159599516SKenneth E. Jansen extern "C" char phasta_iotype[80];
4259599516SKenneth E. Jansen char phasta_iotype[80];
4359599516SKenneth E. Jansen 
4459599516SKenneth E. Jansen extern int SONFATH;
4559599516SKenneth E. Jansen extern "C" void proces();
4659599516SKenneth E. Jansen extern "C" void input();
479ae91bddSCameron Smith extern int input_fform(phSolver::Input&);
4859599516SKenneth E. Jansen extern void setIOparam(); // For SyncIO
491063a5c0SCameron Smith extern "C" void initPhastaCommonVars();
5059599516SKenneth E. Jansen 
5159599516SKenneth E. Jansen int myrank; /* made file global for ease in debugging */
5259599516SKenneth E. Jansen 
5359599516SKenneth E. Jansen void
5459599516SKenneth E. Jansen catchDebugger() {
5559599516SKenneth E. Jansen     while (1) {
5659599516SKenneth E. Jansen       int debuggerPresent=0;
5759599516SKenneth E. Jansen       int fakeSTOP = 1; // please stop HERE and assign as next line
5859599516SKenneth E. Jansen       // assign or set debuggerPresent=1
5959599516SKenneth E. Jansen       if(debuggerPresent) {
6059599516SKenneth E. Jansen         break;
6159599516SKenneth E. Jansen       }
6259599516SKenneth E. Jansen     }
6359599516SKenneth E. Jansen }
6459599516SKenneth E. Jansen 
6559599516SKenneth E. Jansen // some useful debugging functions
6659599516SKenneth E. Jansen 
6759599516SKenneth E. Jansen void
6859599516SKenneth E. Jansen pdarray( void* darray , int start, int end ) {
6959599516SKenneth E. Jansen     for( int i=start; i < end; i++ ){
7059599516SKenneth E. Jansen         cout << ((double*)darray)[i] << endl;
7159599516SKenneth E. Jansen     }
7259599516SKenneth E. Jansen }
7359599516SKenneth E. Jansen 
7459599516SKenneth E. Jansen void
7559599516SKenneth E. Jansen piarray( void* iarray , int start, int end ) {
7659599516SKenneth E. Jansen     for( int i=start; i < end; i++ ){
7759599516SKenneth E. Jansen         cout << ((int*)iarray)[i] << endl;
7859599516SKenneth E. Jansen     }
7959599516SKenneth E. Jansen }
8059599516SKenneth E. Jansen 
8106db1583SCameron Smith namespace {
826b966dd8SCameron Smith   int cdToParent() {
836b966dd8SCameron Smith     if( chdir("..") ) {
846b966dd8SCameron Smith       fprintf(stderr,"could not change to the parent directory\n");
856b966dd8SCameron Smith       return 1;
866b966dd8SCameron Smith     } else {
876b966dd8SCameron Smith       return 0;
886b966dd8SCameron Smith     }
896b966dd8SCameron Smith   }
9006db1583SCameron Smith   int run(phSolver::Input& ctrl) {
913969618bSCameron Smith     int size,ierr;
923969618bSCameron Smith     char inpfilename[100];
933969618bSCameron Smith     MPI_Comm_size (MPI_COMM_WORLD, &size);
943969618bSCameron Smith     MPI_Comm_rank (MPI_COMM_WORLD, &myrank);
953969618bSCameron Smith 
963969618bSCameron Smith     workfc.numpe = size;
973969618bSCameron Smith     workfc.myrank = myrank;
98ea868eb1SCameron Smith 
991063a5c0SCameron Smith     initPhastaCommonVars();
100ea868eb1SCameron Smith     /* Input data  */
1019ae91bddSCameron Smith     ierr = input_fform(ctrl);
102ea868eb1SCameron Smith     if(!ierr){
103ea868eb1SCameron Smith       sprintf(inpfilename,"%d-procs_case/",size);
104ea868eb1SCameron Smith       if( chdir( inpfilename ) ) {
105ea868eb1SCameron Smith         cerr << "could not change to the problem directory "
106ea868eb1SCameron Smith           << inpfilename << endl;
1074c223e05SCameron Smith         return -1;
108b13e4d76SCameron Smith       }
109ea868eb1SCameron Smith       MPI_Barrier(MPI_COMM_WORLD);
110*81372b97SCameron Smith       phastaio_initStats();
111ea868eb1SCameron Smith       input();
112ea868eb1SCameron Smith       /* now we can start the solver */
113ea868eb1SCameron Smith       proces();
114*81372b97SCameron Smith       phastaio_printStats();
115ea868eb1SCameron Smith     }
116ea868eb1SCameron Smith     else{
117ea868eb1SCameron Smith       printf("error during reading ascii input \n");
118ea868eb1SCameron Smith     }
119ea868eb1SCameron Smith     MPI_Barrier(MPI_COMM_WORLD);
120ea868eb1SCameron Smith     if ( myrank == 0 ) {
121ea868eb1SCameron Smith       printf("phasta.cc - last call before finalize!\n");
122ea868eb1SCameron Smith     }
1236b966dd8SCameron Smith     if( cdToParent() )
1244c223e05SCameron Smith       return -1;
1254c223e05SCameron Smith     return timdat.lstep;
126ea868eb1SCameron Smith   }
12706db1583SCameron Smith }
12806db1583SCameron Smith 
12906db1583SCameron Smith int phasta(phSolver::Input& ctrl) {
13006db1583SCameron Smith   outpar.input_mode = 0; //FIXME magic value for posix
13106db1583SCameron Smith   outpar.output_mode = 0; //FIXME magic value for posix
13206db1583SCameron Smith   return run(ctrl);
13306db1583SCameron Smith }
13406db1583SCameron Smith 
13506db1583SCameron Smith int phasta(phSolver::Input& ctrl, grstream grs) {
13606db1583SCameron Smith   assert(grs);
13706db1583SCameron Smith   outpar.input_mode = -1; //FIXME magic value for streams
13806db1583SCameron Smith   outpar.output_mode = 1; //FIXME magic value for syncio
13906db1583SCameron Smith   streamio_set_gr(grs);
14006db1583SCameron Smith   return run(ctrl);
14106db1583SCameron Smith }
14206db1583SCameron Smith 
14306db1583SCameron Smith int phasta(phSolver::Input& ctrl, RStream* rs) {
14406db1583SCameron Smith   fprintf(stderr, "HEY! if you see this email Cameron and tell him "
14506db1583SCameron Smith       "to implement %s(...) on line %d of %s "
14606db1583SCameron Smith       "... returning an error\n", __func__, __LINE__, __FILE__);
14706db1583SCameron Smith   return -1;
14806db1583SCameron Smith }
14906db1583SCameron Smith 
15006db1583SCameron Smith int phasta(phSolver::Input& ctrl, GRStream* grs, RStream* rs) {
15106db1583SCameron Smith   outpar.input_mode = -1; //FIXME magic value for streams
15206db1583SCameron Smith   outpar.output_mode = -1; //FIXME magic value for streams
15306db1583SCameron Smith   assert(grs);
15406db1583SCameron Smith   assert(rs);
15506db1583SCameron Smith   streamio_set_gr(grs);
15606db1583SCameron Smith   streamio_set_r(rs);
15706db1583SCameron Smith   return run(ctrl);
15806db1583SCameron Smith }
1596d494944SCameron Smith 
160163d1986SCameron Smith int phasta( int argc, char *argv[] ) {
16159599516SKenneth E. Jansen     int size,ierr;
16259599516SKenneth E. Jansen     char inpfilename[100];
16359599516SKenneth E. Jansen     char* pauseDebugger = getenv("catchDebugger");
16459599516SKenneth E. Jansen     MPI_Comm_size (MPI_COMM_WORLD, &size);
16559599516SKenneth E. Jansen     MPI_Comm_rank (MPI_COMM_WORLD, &myrank);
16659599516SKenneth E. Jansen 
167513954efSKenneth E. Jansen #ifdef HAVE_PETSC
168513954efSKenneth E. Jansen     PETSC_COMM_WORLD=MPI_COMM_WORLD;
169513954efSKenneth E. Jansen     PetscInitialize(&argc,&argv,PETSC_NULL,PETSC_NULL);
170513954efSKenneth E. Jansen     PetscInitializeFortran();
171513954efSKenneth E. Jansen     PetscPopSignalHandler(); //Let us segfault in peace ;-)
1726dec8975SKenneth E. Jansen // ok with Master    PetscOptionsView(NULL,PETSC_VIEWER_STDOUT_WORLD);
1736dec8975SKenneth E. Jansen // ok with 3.6x    PetscOptionsView(PETSC_VIEWER_STDOUT_WORLD);
174467b991bSKenneth E. Jansen     PetscOptionsView(PETSC_VIEWER_STDOUT_WORLD);
175129f10f8SBen Matthews     if(sizeof(PetscInt) != sizeof(long long int))
176129f10f8SBen Matthews     {
177129f10f8SBen Matthews       //PetscInt and gcorp_t (gen_ncorp.c)
178129f10f8SBen Matthews       //must be the same size. hard-coded for now
179129f10f8SBen Matthews       //FIXME
180129f10f8SBen Matthews 	    if(myrank == 0)
181129f10f8SBen Matthews 	    {
182129f10f8SBen Matthews 		    printf("WARNING: PETSc Index Size Mismatch\n");
183129f10f8SBen Matthews 		    printf("WARNING: Proceed at your own risk\n");
184129f10f8SBen Matthews 	    }
185129f10f8SBen Matthews     }
186513954efSKenneth E. Jansen     MPI_Barrier(MPI_COMM_WORLD);
187513954efSKenneth E. Jansen     if(myrank == 0)
188513954efSKenneth E. Jansen     {
189513954efSKenneth E. Jansen 	    printf("PETSc Initialized\n");
190513954efSKenneth E. Jansen 	    fflush(stdout);
191513954efSKenneth E. Jansen     }
192513954efSKenneth E. Jansen #endif
19359599516SKenneth E. Jansen     workfc.numpe = size;
19459599516SKenneth E. Jansen     workfc.myrank = myrank;
19559599516SKenneth E. Jansen 
19659599516SKenneth E. Jansen #if (defined WIN32)
19759599516SKenneth E. Jansen     if(argc > 2 ){
19859599516SKenneth E. Jansen       catchDebugger();
19959599516SKenneth E. Jansen     }
20059599516SKenneth E. Jansen #endif
20159599516SKenneth E. Jansen #if (1) // ALWAYS ( defined LAUNCH_GDB ) && !( defined WIN32 )
20259599516SKenneth E. Jansen 
20359599516SKenneth E. Jansen     if ( pauseDebugger ) {
20459599516SKenneth E. Jansen 
20559599516SKenneth E. Jansen         int parent_pid = getpid();
20659599516SKenneth E. Jansen         int gdb_child = fork();
20759599516SKenneth E. Jansen         cout << "gdb_child" << gdb_child << endl;
20859599516SKenneth E. Jansen 
20959599516SKenneth E. Jansen         if( gdb_child == 0 ) {
21059599516SKenneth E. Jansen 
21159599516SKenneth E. Jansen             cout << "Debugger Process initiating" << endl;
2121bfea438SCameron Smith             stringstream exec_string;
21359599516SKenneth E. Jansen 
21459599516SKenneth E. Jansen #if ( defined decalp )
21559599516SKenneth E. Jansen             exec_string <<"xterm -e idb "
21659599516SKenneth E. Jansen                         << " -pid "<< parent_pid <<" "<< argv[0] << endl;
21759599516SKenneth E. Jansen #endif
21859599516SKenneth E. Jansen #if ( defined LINUX )
21959599516SKenneth E. Jansen             exec_string <<"xterm -e gdb"
22059599516SKenneth E. Jansen                         << " -pid "<< parent_pid <<" "<< argv[0] << endl;
22159599516SKenneth E. Jansen #endif
22259599516SKenneth E. Jansen #if ( defined SUN4 )
22359599516SKenneth E. Jansen             exec_string <<"xterm -e dbx "
22459599516SKenneth E. Jansen                         << " - "<< parent_pid <<" "<< argv[0] << endl;
22559599516SKenneth E. Jansen #endif
22659599516SKenneth E. Jansen #if ( defined IRIX )
22759599516SKenneth E. Jansen             exec_string <<"xterm -e dbx "
22859599516SKenneth E. Jansen                         << " -p "<< parent_pid <<" "<< argv[0] << endl;
22959599516SKenneth E. Jansen #endif
2301bfea438SCameron Smith             string s = exec_string.str();
2311bfea438SCameron Smith             system( s.c_str() );
23259599516SKenneth E. Jansen             exit(0);
23359599516SKenneth E. Jansen         }
23459599516SKenneth E. Jansen         catchDebugger();
23559599516SKenneth E. Jansen     }
23659599516SKenneth E. Jansen 
23759599516SKenneth E. Jansen #endif
23859599516SKenneth E. Jansen 
23959599516SKenneth E. Jansen     /* Input data  */
24059599516SKenneth E. Jansen     if(argc > 1 ){
24159599516SKenneth E. Jansen         strcpy(inpfilename,argv[1]);
24259599516SKenneth E. Jansen     } else {
24359599516SKenneth E. Jansen         strcpy(inpfilename,"solver.inp");
24459599516SKenneth E. Jansen     }
2459ae91bddSCameron Smith     string defaultConf = ".";
2469ae91bddSCameron Smith     const char* path_to_config = getenv("PHASTA_CONFIG");
2479ae91bddSCameron Smith     if(path_to_config)
2489ae91bddSCameron Smith       defaultConf = path_to_config;
2499ae91bddSCameron Smith     defaultConf.append("/input.config");
2509ae91bddSCameron Smith     string userConf(inpfilename);
2519ae91bddSCameron Smith     phSolver::Input ctrl(userConf, defaultConf);
2521063a5c0SCameron Smith     initPhastaCommonVars();
2539ae91bddSCameron Smith     ierr = input_fform(ctrl);
25459599516SKenneth E. Jansen     if(!ierr){
25559599516SKenneth E. Jansen       sprintf(inpfilename,"%d-procs_case/",size);
25659599516SKenneth E. Jansen       if( chdir( inpfilename ) ) {
25759599516SKenneth E. Jansen         cerr << "could not change to the problem directory "
25859599516SKenneth E. Jansen           << inpfilename << endl;
2594c223e05SCameron Smith         return -1;
26059599516SKenneth E. Jansen       }
2612992f42eSCameron Smith       MPI_Barrier(MPI_COMM_WORLD);
26259599516SKenneth E. Jansen       setIOparam();
2639f4aafb6SCameron Smith       outpar.input_mode = outpar.nsynciofiles; //FIXME this is awful
2649f4aafb6SCameron Smith       outpar.output_mode = outpar.nsynciofiles; //FIXME this is awful
265*81372b97SCameron Smith       phastaio_initStats();
26659599516SKenneth E. Jansen       input();
26759599516SKenneth E. Jansen       /* now we can start the solver */
26859599516SKenneth E. Jansen       proces();
269*81372b97SCameron Smith       phastaio_printStats();
27059599516SKenneth E. Jansen     }
27159599516SKenneth E. Jansen     else{
27259599516SKenneth E. Jansen         printf("error during reading ascii input \n");
27359599516SKenneth E. Jansen     }
274513954efSKenneth E. Jansen #ifdef HAVE_PETSC
275513954efSKenneth E. Jansen     PetscFinalize();
276513954efSKenneth E. Jansen #endif
27759599516SKenneth E. Jansen     MPI_Barrier(MPI_COMM_WORLD);
27859599516SKenneth E. Jansen     if ( myrank == 0 ) {
27959599516SKenneth E. Jansen       printf("phasta.cc - last call before finalize!\n");
28059599516SKenneth E. Jansen     }
2816b966dd8SCameron Smith     if( cdToParent() )
2826b966dd8SCameron Smith       return -1;
2834c223e05SCameron Smith     return timdat.lstep;
28459599516SKenneth E. Jansen }
285