xref: /phasta/phSolver/common/phasta.cc (revision 9d99a6ba0e084ba8b386a590177ffa2eb15369d9)
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 
30*9d99a6baSCameron Smith #include "phasta_version.h"
31ea8ce4a6SBenjamin Matthews #include "common_c.h"
32ea8ce4a6SBenjamin Matthews #include "Input.h"
3381372b97SCameron Smith #include "phiostats.h"
34ea8ce4a6SBenjamin Matthews #include "phstream.h"
35ea8ce4a6SBenjamin Matthews #include "streamio.h"
36ea8ce4a6SBenjamin Matthews 
37ea8ce4a6SBenjamin Matthews #include <FCMangle.h>
38ea8ce4a6SBenjamin Matthews #define input FortranCInterface_GLOBAL_(input,INPUT)
39ea8ce4a6SBenjamin Matthews #define proces FortranCInterface_GLOBAL_(proces,PROCES)
40ea8ce4a6SBenjamin Matthews #define timer FortranCInterface_GLOBAL_(timer,TIMER)
41ea8ce4a6SBenjamin Matthews 
4259599516SKenneth E. Jansen extern "C" char phasta_iotype[80];
4359599516SKenneth E. Jansen char phasta_iotype[80];
4459599516SKenneth E. Jansen 
4559599516SKenneth E. Jansen extern int SONFATH;
4659599516SKenneth E. Jansen extern "C" void proces();
4759599516SKenneth E. Jansen extern "C" void input();
489ae91bddSCameron Smith extern int input_fform(phSolver::Input&);
4959599516SKenneth E. Jansen extern void setIOparam(); // For SyncIO
501063a5c0SCameron Smith extern "C" void initPhastaCommonVars();
5159599516SKenneth E. Jansen 
5259599516SKenneth E. Jansen int myrank; /* made file global for ease in debugging */
5359599516SKenneth E. Jansen 
5459599516SKenneth E. Jansen void
5559599516SKenneth E. Jansen catchDebugger() {
5659599516SKenneth E. Jansen     while (1) {
5759599516SKenneth E. Jansen       int debuggerPresent=0;
5859599516SKenneth E. Jansen       int fakeSTOP = 1; // please stop HERE and assign as next line
5959599516SKenneth E. Jansen       // assign or set debuggerPresent=1
6059599516SKenneth E. Jansen       if(debuggerPresent) {
6159599516SKenneth E. Jansen         break;
6259599516SKenneth E. Jansen       }
6359599516SKenneth E. Jansen     }
6459599516SKenneth E. Jansen }
6559599516SKenneth E. Jansen 
6659599516SKenneth E. Jansen // some useful debugging functions
6759599516SKenneth E. Jansen 
6859599516SKenneth E. Jansen void
6959599516SKenneth E. Jansen pdarray( void* darray , int start, int end ) {
7059599516SKenneth E. Jansen     for( int i=start; i < end; i++ ){
7159599516SKenneth E. Jansen         cout << ((double*)darray)[i] << endl;
7259599516SKenneth E. Jansen     }
7359599516SKenneth E. Jansen }
7459599516SKenneth E. Jansen 
7559599516SKenneth E. Jansen void
7659599516SKenneth E. Jansen piarray( void* iarray , int start, int end ) {
7759599516SKenneth E. Jansen     for( int i=start; i < end; i++ ){
7859599516SKenneth E. Jansen         cout << ((int*)iarray)[i] << endl;
7959599516SKenneth E. Jansen     }
8059599516SKenneth E. Jansen }
8159599516SKenneth E. Jansen 
8206db1583SCameron Smith namespace {
836b966dd8SCameron Smith   int cdToParent() {
846b966dd8SCameron Smith     if( chdir("..") ) {
856b966dd8SCameron Smith       fprintf(stderr,"could not change to the parent directory\n");
866b966dd8SCameron Smith       return 1;
876b966dd8SCameron Smith     } else {
886b966dd8SCameron Smith       return 0;
896b966dd8SCameron Smith     }
906b966dd8SCameron Smith   }
9106db1583SCameron Smith   int run(phSolver::Input& ctrl) {
923969618bSCameron Smith     int size,ierr;
933969618bSCameron Smith     char inpfilename[100];
943969618bSCameron Smith     MPI_Comm_size (MPI_COMM_WORLD, &size);
953969618bSCameron Smith     MPI_Comm_rank (MPI_COMM_WORLD, &myrank);
963969618bSCameron Smith 
97*9d99a6baSCameron Smith     if(!myrank)
98*9d99a6baSCameron Smith       printf("PHASTA Git hash %s\n", phasta_version());
99*9d99a6baSCameron Smith 
1003969618bSCameron Smith     workfc.numpe = size;
1013969618bSCameron Smith     workfc.myrank = myrank;
102ea868eb1SCameron Smith 
1031063a5c0SCameron Smith     initPhastaCommonVars();
104ea868eb1SCameron Smith     /* Input data  */
1059ae91bddSCameron Smith     ierr = input_fform(ctrl);
106ea868eb1SCameron Smith     if(!ierr){
107ea868eb1SCameron Smith       sprintf(inpfilename,"%d-procs_case/",size);
108ea868eb1SCameron Smith       if( chdir( inpfilename ) ) {
109ea868eb1SCameron Smith         cerr << "could not change to the problem directory "
110ea868eb1SCameron Smith           << inpfilename << endl;
1114c223e05SCameron Smith         return -1;
112b13e4d76SCameron Smith       }
113ea868eb1SCameron Smith       MPI_Barrier(MPI_COMM_WORLD);
11481372b97SCameron Smith       phastaio_initStats();
115ea868eb1SCameron Smith       input();
116ea868eb1SCameron Smith       /* now we can start the solver */
117ea868eb1SCameron Smith       proces();
11881372b97SCameron Smith       phastaio_printStats();
119ea868eb1SCameron Smith     }
120ea868eb1SCameron Smith     else{
121ea868eb1SCameron Smith       printf("error during reading ascii input \n");
122ea868eb1SCameron Smith     }
123ea868eb1SCameron Smith     MPI_Barrier(MPI_COMM_WORLD);
124ea868eb1SCameron Smith     if ( myrank == 0 ) {
125ea868eb1SCameron Smith       printf("phasta.cc - last call before finalize!\n");
126ea868eb1SCameron Smith     }
1276b966dd8SCameron Smith     if( cdToParent() )
1284c223e05SCameron Smith       return -1;
1294c223e05SCameron Smith     return timdat.lstep;
130ea868eb1SCameron Smith   }
13106db1583SCameron Smith }
13206db1583SCameron Smith 
13306db1583SCameron Smith int phasta(phSolver::Input& ctrl) {
13406db1583SCameron Smith   outpar.input_mode = 0; //FIXME magic value for posix
13506db1583SCameron Smith   outpar.output_mode = 0; //FIXME magic value for posix
13606db1583SCameron Smith   return run(ctrl);
13706db1583SCameron Smith }
13806db1583SCameron Smith 
13906db1583SCameron Smith int phasta(phSolver::Input& ctrl, grstream grs) {
14006db1583SCameron Smith   assert(grs);
14106db1583SCameron Smith   outpar.input_mode = -1; //FIXME magic value for streams
14206db1583SCameron Smith   outpar.output_mode = 1; //FIXME magic value for syncio
14306db1583SCameron Smith   streamio_set_gr(grs);
14406db1583SCameron Smith   return run(ctrl);
14506db1583SCameron Smith }
14606db1583SCameron Smith 
14706db1583SCameron Smith int phasta(phSolver::Input& ctrl, RStream* rs) {
14806db1583SCameron Smith   fprintf(stderr, "HEY! if you see this email Cameron and tell him "
14906db1583SCameron Smith       "to implement %s(...) on line %d of %s "
15006db1583SCameron Smith       "... returning an error\n", __func__, __LINE__, __FILE__);
15106db1583SCameron Smith   return -1;
15206db1583SCameron Smith }
15306db1583SCameron Smith 
15406db1583SCameron Smith int phasta(phSolver::Input& ctrl, GRStream* grs, RStream* rs) {
15506db1583SCameron Smith   outpar.input_mode = -1; //FIXME magic value for streams
15606db1583SCameron Smith   outpar.output_mode = -1; //FIXME magic value for streams
15706db1583SCameron Smith   assert(grs);
15806db1583SCameron Smith   assert(rs);
15906db1583SCameron Smith   streamio_set_gr(grs);
16006db1583SCameron Smith   streamio_set_r(rs);
16106db1583SCameron Smith   return run(ctrl);
16206db1583SCameron Smith }
1636d494944SCameron Smith 
164163d1986SCameron Smith int phasta( int argc, char *argv[] ) {
16559599516SKenneth E. Jansen     int size,ierr;
16659599516SKenneth E. Jansen     char inpfilename[100];
16759599516SKenneth E. Jansen     char* pauseDebugger = getenv("catchDebugger");
16859599516SKenneth E. Jansen     MPI_Comm_size (MPI_COMM_WORLD, &size);
16959599516SKenneth E. Jansen     MPI_Comm_rank (MPI_COMM_WORLD, &myrank);
170*9d99a6baSCameron Smith     if(!myrank)
171*9d99a6baSCameron Smith       printf("PHASTA Git hash %s\n", phasta_version());
17259599516SKenneth E. Jansen 
173513954efSKenneth E. Jansen #ifdef HAVE_PETSC
174513954efSKenneth E. Jansen     PETSC_COMM_WORLD=MPI_COMM_WORLD;
175513954efSKenneth E. Jansen     PetscInitialize(&argc,&argv,PETSC_NULL,PETSC_NULL);
176513954efSKenneth E. Jansen     PetscInitializeFortran();
177513954efSKenneth E. Jansen     PetscPopSignalHandler(); //Let us segfault in peace ;-)
1786dec8975SKenneth E. Jansen // ok with Master    PetscOptionsView(NULL,PETSC_VIEWER_STDOUT_WORLD);
1796dec8975SKenneth E. Jansen // ok with 3.6x    PetscOptionsView(PETSC_VIEWER_STDOUT_WORLD);
180467b991bSKenneth E. Jansen     PetscOptionsView(PETSC_VIEWER_STDOUT_WORLD);
181129f10f8SBen Matthews     if(sizeof(PetscInt) != sizeof(long long int))
182129f10f8SBen Matthews     {
183129f10f8SBen Matthews       //PetscInt and gcorp_t (gen_ncorp.c)
184129f10f8SBen Matthews       //must be the same size. hard-coded for now
185129f10f8SBen Matthews       //FIXME
186129f10f8SBen Matthews 	    if(myrank == 0)
187129f10f8SBen Matthews 	    {
188129f10f8SBen Matthews 		    printf("WARNING: PETSc Index Size Mismatch\n");
189129f10f8SBen Matthews 		    printf("WARNING: Proceed at your own risk\n");
190129f10f8SBen Matthews 	    }
191129f10f8SBen Matthews     }
192513954efSKenneth E. Jansen     MPI_Barrier(MPI_COMM_WORLD);
193513954efSKenneth E. Jansen     if(myrank == 0)
194513954efSKenneth E. Jansen     {
195513954efSKenneth E. Jansen 	    printf("PETSc Initialized\n");
196513954efSKenneth E. Jansen 	    fflush(stdout);
197513954efSKenneth E. Jansen     }
198513954efSKenneth E. Jansen #endif
19959599516SKenneth E. Jansen     workfc.numpe = size;
20059599516SKenneth E. Jansen     workfc.myrank = myrank;
20159599516SKenneth E. Jansen 
20259599516SKenneth E. Jansen #if (defined WIN32)
20359599516SKenneth E. Jansen     if(argc > 2 ){
20459599516SKenneth E. Jansen       catchDebugger();
20559599516SKenneth E. Jansen     }
20659599516SKenneth E. Jansen #endif
20759599516SKenneth E. Jansen #if (1) // ALWAYS ( defined LAUNCH_GDB ) && !( defined WIN32 )
20859599516SKenneth E. Jansen 
20959599516SKenneth E. Jansen     if ( pauseDebugger ) {
21059599516SKenneth E. Jansen 
21159599516SKenneth E. Jansen         int parent_pid = getpid();
21259599516SKenneth E. Jansen         int gdb_child = fork();
21359599516SKenneth E. Jansen         cout << "gdb_child" << gdb_child << endl;
21459599516SKenneth E. Jansen 
21559599516SKenneth E. Jansen         if( gdb_child == 0 ) {
21659599516SKenneth E. Jansen 
21759599516SKenneth E. Jansen             cout << "Debugger Process initiating" << endl;
2181bfea438SCameron Smith             stringstream exec_string;
21959599516SKenneth E. Jansen 
22059599516SKenneth E. Jansen #if ( defined decalp )
22159599516SKenneth E. Jansen             exec_string <<"xterm -e idb "
22259599516SKenneth E. Jansen                         << " -pid "<< parent_pid <<" "<< argv[0] << endl;
22359599516SKenneth E. Jansen #endif
22459599516SKenneth E. Jansen #if ( defined LINUX )
22559599516SKenneth E. Jansen             exec_string <<"xterm -e gdb"
22659599516SKenneth E. Jansen                         << " -pid "<< parent_pid <<" "<< argv[0] << endl;
22759599516SKenneth E. Jansen #endif
22859599516SKenneth E. Jansen #if ( defined SUN4 )
22959599516SKenneth E. Jansen             exec_string <<"xterm -e dbx "
23059599516SKenneth E. Jansen                         << " - "<< parent_pid <<" "<< argv[0] << endl;
23159599516SKenneth E. Jansen #endif
23259599516SKenneth E. Jansen #if ( defined IRIX )
23359599516SKenneth E. Jansen             exec_string <<"xterm -e dbx "
23459599516SKenneth E. Jansen                         << " -p "<< parent_pid <<" "<< argv[0] << endl;
23559599516SKenneth E. Jansen #endif
2361bfea438SCameron Smith             string s = exec_string.str();
2371bfea438SCameron Smith             system( s.c_str() );
23859599516SKenneth E. Jansen             exit(0);
23959599516SKenneth E. Jansen         }
24059599516SKenneth E. Jansen         catchDebugger();
24159599516SKenneth E. Jansen     }
24259599516SKenneth E. Jansen 
24359599516SKenneth E. Jansen #endif
24459599516SKenneth E. Jansen 
24559599516SKenneth E. Jansen     /* Input data  */
24659599516SKenneth E. Jansen     if(argc > 1 ){
24759599516SKenneth E. Jansen         strcpy(inpfilename,argv[1]);
24859599516SKenneth E. Jansen     } else {
24959599516SKenneth E. Jansen         strcpy(inpfilename,"solver.inp");
25059599516SKenneth E. Jansen     }
2519ae91bddSCameron Smith     string defaultConf = ".";
2529ae91bddSCameron Smith     const char* path_to_config = getenv("PHASTA_CONFIG");
2539ae91bddSCameron Smith     if(path_to_config)
2549ae91bddSCameron Smith       defaultConf = path_to_config;
2559ae91bddSCameron Smith     defaultConf.append("/input.config");
2569ae91bddSCameron Smith     string userConf(inpfilename);
2579ae91bddSCameron Smith     phSolver::Input ctrl(userConf, defaultConf);
2581063a5c0SCameron Smith     initPhastaCommonVars();
2599ae91bddSCameron Smith     ierr = input_fform(ctrl);
26059599516SKenneth E. Jansen     if(!ierr){
26159599516SKenneth E. Jansen       sprintf(inpfilename,"%d-procs_case/",size);
26259599516SKenneth E. Jansen       if( chdir( inpfilename ) ) {
26359599516SKenneth E. Jansen         cerr << "could not change to the problem directory "
26459599516SKenneth E. Jansen           << inpfilename << endl;
2654c223e05SCameron Smith         return -1;
26659599516SKenneth E. Jansen       }
2672992f42eSCameron Smith       MPI_Barrier(MPI_COMM_WORLD);
26859599516SKenneth E. Jansen       setIOparam();
2699f4aafb6SCameron Smith       outpar.input_mode = outpar.nsynciofiles; //FIXME this is awful
2709f4aafb6SCameron Smith       outpar.output_mode = outpar.nsynciofiles; //FIXME this is awful
27181372b97SCameron Smith       phastaio_initStats();
27259599516SKenneth E. Jansen       input();
27359599516SKenneth E. Jansen       /* now we can start the solver */
27459599516SKenneth E. Jansen       proces();
27581372b97SCameron Smith       phastaio_printStats();
27659599516SKenneth E. Jansen     }
27759599516SKenneth E. Jansen     else{
27859599516SKenneth E. Jansen         printf("error during reading ascii input \n");
27959599516SKenneth E. Jansen     }
280513954efSKenneth E. Jansen #ifdef HAVE_PETSC
281513954efSKenneth E. Jansen     PetscFinalize();
282513954efSKenneth E. Jansen #endif
28359599516SKenneth E. Jansen     MPI_Barrier(MPI_COMM_WORLD);
28459599516SKenneth E. Jansen     if ( myrank == 0 ) {
28559599516SKenneth E. Jansen       printf("phasta.cc - last call before finalize!\n");
28659599516SKenneth E. Jansen     }
2876b966dd8SCameron Smith     if( cdToParent() )
2886b966dd8SCameron Smith       return -1;
2894c223e05SCameron Smith     return timdat.lstep;
29059599516SKenneth E. Jansen }
291