int main(int argc, char** argv) {
const int buf_len = 512;
int rc, docheck, print_flag;
char fname[buf_len], outname[buf_len], statefname[buf_len];
FILE *F, *FS;
mytm TM;
// These are permanent stores and we need to worry about memory management
Subcatchment *SUB = NULL;
NameStore *NODE_NAMES;
SMap HASH;
if ( argc < 3 ) {
printf("Usage: %s <netlistfile> <state file name>\n", argv[0]);
printf(" %s -checkonly <netlistfile>\n", argv[0]);
return (-1);
}
docheck = 0;
if ( argc == 3 && strcmp(argv[1],"-checkonly")==0 ) {
docheck = 1;
strncpy(fname, argv[2], buf_len);
} else {
docheck = 0;
strncpy(fname, argv[1], buf_len);
strncpy(statefname, argv[2], buf_len);
}
if ( (F=fopen(fname, "r")) == NULL ) {
printf("Bummer: unable to open netlist file %s\n", argv[1]);
return (-1);
}
PrintHeader(stdout);
/* allocate the subcatchment and the topograph */
SUB = new Subcatchment(0);
NODE_NAMES = new NameStore();
// default option values
OPT.Alpha() = 0.8;
OPT.Beta() = 1.03;
OPT.HT() = 1e-3;
OPT.MinDT() = 0.001;
OPT.SSTol() = 5e-8;
OPT.Tol() = 1e-6;
OPT.EpsilonA() = 1e-4;
OPT.DebugLevel() = 0;
rc = read_spt_from_file(F, SUB, NODE_NAMES, &HASH, stdout); // also modifies OPT and
// STAT in the global
// namespace
// save the netlist file
STAT.SetInFile( get_basename( fname ) );
if (rc < 0 ) goto out; // there is error!
// do a topo check
rc = SUB->TopologyCheck();
if (rc < 0 ) {
SUB->TopoPrintErrMsg(stdout, "== tchk ==", NODE_NAMES->Store() );
goto out;
}
// quit if docheck == 1
if ( OPT.CheckOnly() == 1 || docheck==1 ) {
printf("The connectivity of the given netlist appears to be correct\n");
goto out;
}
// make solver
SUB->MakeSolver( SUB->GetNumNodes() );
// SUB->InitSolutions();
FS = fopen( statefname, "r");
rc = ERROR;
if ( FS ) {
const int perf_check=0; // force to ignore the check. Could be dangerous!
rc = SUB->LoadSteadyStateFromFile(FS, perf_check);
SUB->CopyXpToXtm1();
SUB->CopyXpToX();
if ( rc == OK ) printf("[II]: Loaded states from \"%s\"\n", statefname );
fclose(FS);
}
// steady solve, two phases
if ( rc!=OK) { // either no state file, or state file is not good
rc = SUB->SteadySolve(600, 600, 2*OPT.Tol());
if (rc<0) {
fprintf(stdout,"[EE]: first phase of steady-state solve failed\n");
goto out;
}
rc = SUB->SteadySolve(25.0, 45, 45, OPT.Tol() );
if (rc<0) {
fprintf(stdout,"[EE]: second phase of steady-state solve failed\n");
goto out;
}
}
// store back if needed
if ( rc > 0 && STAT.SSFile() ) {
FS=fopen(STAT.SSFile(),"w");
SUB->SaveSteadyStateToFile(FS);
fclose(FS);
fprintf(stdout, "[II]: Saved steady state to file \"%s\"\n", STAT.SSFile());
}
// run unsteady
print_flag = 0;
if (OPT.PrintQ()==1) print_flag |= PRT_Q;
if (OPT.PrintA()==1) print_flag |= PRT_A;
if (OPT.PrintZ()==1) print_flag |= PRT_Z;
if (OPT.PrintD()==1) print_flag |= PRT_D;
sprintf(outname,"%s.output.dat", fname);
if ( print_flag ) {
FS = fopen(outname,"w");
if ( FS == NULL ) {
fprintf(stdout,"[II]: Unable to open output file %s. Request ignored\n", outname);
} else {
fprintf(stdout,"[II]: Unsteady results will be stored in \"%s\"\n", outname);
}
if (OPT.PrintXY()==1) print_flag |= PRT_XY; // XY printing is only turned on when
// others are enabled
} else {
fprintf(stdout,"[II]: No printing request made. Nothing will be stored\n");
FS = NULL;
}
TM.start();
rc = SUB->UnsteadySolve(OPT.StopTime(), 1, 25, OPT.Tol(), NULL, FS, print_flag, OPT.PrintStart(), NODE_NAMES->Store() );
TM.stop();
if ( FS ) fclose(FS);
fprintf(stdout, "[II]: Simulation of the %d-min event took %.3f seconds.\n", (int)(OPT.StopTime()/60.0), TM.read() );
// store the states
FS=fopen(statefname, "w");
if (!FS) {
printf("Bummer: unable to open file \"%s\" to write state. Check permissions!\n", statefname);
} else {
SUB->SaveSteadyStateToFile(FS);
fprintf(stdout,"[II]: Saved final states to file \"%s\"\n", statefname);
}
if (FS) fclose(FS);
out:
if (F) fclose(F);
if ( SUB ) delete SUB;
if ( NODE_NAMES) delete NODE_NAMES;
return 0;
}
