Resume command

Not functional yet: need to deal with memory of argv
This commit is contained in:
Ian Jauslin 2023-11-03 12:08:25 -04:00
parent 3efdbf4451
commit 3d94694017
2 changed files with 169 additions and 25 deletions

View File

@ -18,6 +18,7 @@ limitations under the License.
#define COMMAND_UK 1 #define COMMAND_UK 1
#define COMMAND_ENSTROPHY 2 #define COMMAND_ENSTROPHY 2
#define COMMAND_QUIET 3 #define COMMAND_QUIET 3
#define COMMAND_RESUME 4
#define DRIVING_ZERO 1 #define DRIVING_ZERO 1
#define DRIVING_TEST 2 #define DRIVING_TEST 2

View File

@ -20,6 +20,7 @@ limitations under the License.
#include <string.h> #include <string.h>
#include <stdlib.h> #include <stdlib.h>
#include <errno.h> #include <errno.h>
#include <wordexp.h>
#include "constants.cpp" #include "constants.cpp"
#include "driving.h" #include "driving.h"
@ -61,9 +62,12 @@ int print_usage();
int print_params(nstrophy_parameters parameters, char* initfile_str, char* drivingfile_str, FILE* file); int print_params(nstrophy_parameters parameters, char* initfile_str, char* drivingfile_str, FILE* file);
// read command line arguments // read command line arguments
int read_args(int argc, const char* argv[], char** params, unsigned int* command, unsigned int* nthreads, char** savefile_str, char** utfile_str); int read_args(int argc, const char* argv[], char** params, unsigned int* command, unsigned int* nthreads, char** savefile_str, char** utfile_str, char** resumefile_str);
int set_default_params(nstrophy_parameters* parameters);
int read_params(char* param_str, nstrophy_parameters* parameters, char** initfile_str, char** drivingfile_str); int read_params(char* param_str, nstrophy_parameters* parameters, char** initfile_str, char** drivingfile_str);
int set_parameter(char* lhs, char* rhs, nstrophy_parameters* parameters, bool* setN1, bool* setN2, char** initfile_str, char** drivingfile_str); int set_parameter(char* lhs, char* rhs, nstrophy_parameters* parameters, bool* setN1, bool* setN2, char** initfile_str, char** drivingfile_str);
// read args from file
int args_from_file(char** params, unsigned int* command, unsigned int* nthreads, char** savefile_str, char** utfile_str, char* file_str);
// set driving force // set driving force
_Complex double* set_driving(nstrophy_parameters parameters); _Complex double* set_driving(nstrophy_parameters parameters);
@ -77,9 +81,17 @@ void sig_handler (int signo);
volatile bool g_abort = false; volatile bool g_abort = false;
// signal handler // signal handler
void sig_handler (int signo){ void sig_handler (int signo){
if (signo == SIGINT || signo == SIGTERM){ if (signo == SIGINT){
fprintf(stderr,"received signal SIGINT, interrupting computation\n");
g_abort = true; g_abort = true;
} }
else if (signo == SIGTERM){
fprintf(stderr,"received signal SIGTERM, interrupting computation\n");
g_abort = true;
}
else{
fprintf(stderr,"received signal %d\n",signo);
}
} }
@ -98,23 +110,50 @@ int main (
char* utfile_str=NULL; char* utfile_str=NULL;
char* initfile_str=NULL; char* initfile_str=NULL;
char* drivingfile_str=NULL; char* drivingfile_str=NULL;
char* resumefile_str=NULL;
FILE* savefile=NULL; FILE* savefile=NULL;
FILE* utfile=NULL; FILE* utfile=NULL;
command=0; command=0;
// read command line arguments // read command line arguments
ret=read_args(argc, argv, &param_str, &command, &nthreads, &savefile_str, &utfile_str); ret=read_args(argc, argv, &param_str, &command, &nthreads, &savefile_str, &utfile_str, &resumefile_str);
if(ret<0){ if(ret<0){
return(-1); return(-1);
} }
// set default params
set_default_params(&parameters);
// read params // read params
ret=read_params(param_str, &parameters, &initfile_str, &drivingfile_str); ret=read_params(param_str, &parameters, &initfile_str, &drivingfile_str);
if(ret<0){ if(ret<0){
return(-1); return(-1);
} }
// if command is 'resume', then read args from file
if(command==COMMAND_RESUME){
ret=args_from_file(&param_str, &command, &nthreads, &savefile_str, &utfile_str, resumefile_str);
if(ret<0){
return(-1);
}
// read params
ret=read_params(param_str, &parameters, &initfile_str, &drivingfile_str);
if(ret<0){
return(-1);
}
// reread arguments (to allow overrides from the command line)
ret=read_args(argc, argv, &param_str, &command, &nthreads, &savefile_str, &utfile_str, &resumefile_str);
if(ret<0){
return(-1);
}
// reread params
ret=read_params(param_str, &parameters, &initfile_str, &drivingfile_str);
if(ret<0){
return(-1);
}
}
// open initfile // open initfile
if(initfile_str!=NULL){ if(initfile_str!=NULL){
parameters.initfile=fopen(initfile_str,"r"); parameters.initfile=fopen(initfile_str,"r");
@ -213,7 +252,7 @@ int main (
// usage message // usage message
int print_usage(){ int print_usage(){
fprintf(stderr, "usage:\n nstrophy [-t nthreads] [-p parameters] [-s savefile] [-u u_outfile] <command>\n\n"); fprintf(stderr, "usage:\n nstrophy [-t nthreads] [-p parameters] [-s savefile] [-u u_outfile] <command>\n where <command> is one of\n enstrophy\n uk\n quiet\n resume <resume_file>\n\n");
return(0); return(0);
} }
@ -319,6 +358,7 @@ int print_params(
#define CP_FLAG_NTHREADS 2 #define CP_FLAG_NTHREADS 2
#define CP_FLAG_SAVEFILE 3 #define CP_FLAG_SAVEFILE 3
#define CP_FLAG_UTFILE 4 #define CP_FLAG_UTFILE 4
#define CP_FLAG_RESUME 5
int read_args( int read_args(
int argc, int argc,
const char* argv[], const char* argv[],
@ -326,7 +366,8 @@ int read_args(
unsigned int* command, unsigned int* command,
unsigned int* nthreads, unsigned int* nthreads,
char** savefile_str, char** savefile_str,
char** utfile_str char** utfile_str,
char** resumefile_str
){ ){
int i; int i;
int ret; int ret;
@ -385,6 +426,11 @@ int read_args(
*utfile_str=(char*)argv[i]; *utfile_str=(char*)argv[i];
flag=0; flag=0;
} }
// resume file
else if(flag==CP_FLAG_RESUME){
*resumefile_str=(char*)argv[i];
flag=0;
}
// computation to run // computation to run
else{ else{
if(strcmp(argv[i], "uk")==0){ if(strcmp(argv[i], "uk")==0){
@ -396,37 +442,31 @@ int read_args(
else if(strcmp(argv[i], "quiet")==0){ else if(strcmp(argv[i], "quiet")==0){
*command=COMMAND_QUIET; *command=COMMAND_QUIET;
} }
else if(strcmp(argv[i], "resume")==0){
*command=COMMAND_RESUME;
flag=CP_FLAG_RESUME;
}
else{ else{
fprintf(stderr, "error: unrecognized command: '%s'\n",argv[i]); fprintf(stderr, "error: unrecognized command: '%s'\n",argv[i]);
return(-1); return(-1);
} }
flag=0;
} }
} }
// check that if the command is 'resume', then resumefile has been set
if(*command==COMMAND_RESUME && *resumefile_str==NULL){
fprintf(stderr, "error: 'resume' command used, but no resume file\n");
print_usage();
return(-1);
}
return(0); return(0);
} }
// read parameters string // set default parameters
int read_params( int set_default_params(
char* param_str, nstrophy_parameters* parameters
nstrophy_parameters* parameters,
char** initfile_str,
char** drivingfile_str
){ ){
int ret;
// pointer in params
char* ptr;
// buffer and associated pointer
char *buffer_lhs, *lhs_ptr;
char *buffer_rhs, *rhs_ptr;
// whether N was set explicitly
bool setN1=false;
bool setN2=false;
// whether lhs (false is rhs)
bool lhs=true;
// defaults
parameters->init_en=1.54511597324389e+02; parameters->init_en=1.54511597324389e+02;
parameters->irreversible=true; parameters->irreversible=true;
parameters->K1=16; parameters->K1=16;
@ -451,6 +491,28 @@ int read_params(
parameters->algorithm=ALGORITHM_RK4; parameters->algorithm=ALGORITHM_RK4;
parameters->keep_en_cst=false; parameters->keep_en_cst=false;
return(0);
}
// read parameters string
int read_params(
char* param_str,
nstrophy_parameters* parameters,
char** initfile_str,
char** drivingfile_str
){
int ret;
// pointer in params
char* ptr;
// buffer and associated pointer
char *buffer_lhs, *lhs_ptr;
char *buffer_rhs, *rhs_ptr;
// whether N was set explicitly
bool setN1=false;
bool setN2=false;
// whether lhs (false is rhs)
bool lhs=true;
if (param_str!=NULL){ if (param_str!=NULL){
// init // init
buffer_lhs=calloc(sizeof(char),strlen(param_str)); buffer_lhs=calloc(sizeof(char),strlen(param_str));
@ -775,6 +837,87 @@ int set_parameter(
return(0); return(0);
} }
// read args from file
int args_from_file(
char** params,
unsigned int* command,
unsigned int* nthreads,
char** savefile_str,
char** utfile_str,
char* file_str
){
char* line;
unsigned int len=256;
unsigned int pos=0;
char* line_realloc;
char c;
// open file
FILE* file=fopen(file_str,"r");
if(file==NULL){
fprintf(stderr,"Error opening file '%s' for reading: %s\n", file_str, strerror(errno));
return(-1);
}
// allocate line buffer
line=calloc(sizeof(char), len);
while(1){
c=fgetc(file);
// end of file
if (feof(file)){
break;
}
// newline: read line and reset buffer
if(c=='\n' || c=='\r'){
// read entry
// ignore lines with fewer than three characters
if(pos>=3){
// find line starting with "#!"
if(line[0]=='#' && line[1]=='!'){
wordexp_t pwordexp;
wordexp(line+2,&pwordexp,0);
// read arguments
read_args(pwordexp.we_wordc, (const char **)pwordexp.we_wordv, params, command, nthreads, savefile_str, utfile_str, NULL);
wordfree(&pwordexp);
// exit
break;
}
}
// reset buffer
pos=0;
line[pos]='\0';
}
// add to buffer
else {
// check that there is room in buffer
if(pos==len){
// too short: reallocate
line_realloc=calloc(sizeof(char), 2*len);
for(pos=0;pos<len;pos++){
line_realloc[pos]=line[pos];
}
free(line);
line=line_realloc;
len=2*len;
}
// add c to line
line[pos]=c;
pos++;
line[pos]='\0';
}
}
free(line);
// close file
fclose(file);
return(0);
}
// set driving force // set driving force
_Complex double* set_driving( _Complex double* set_driving(
nstrophy_parameters parameters nstrophy_parameters parameters