Polaris: switches_util.cc Source File

switches_util.cc Go to the documentation of this file. /// #include <stdlib.h> #include <fstream.h> #include <strstream.h> #include <limits.h> #include "../Collection/Database.h" #include "../Collection/KeyIterator.h" #include "../String.h" #include "../IntElem.h" #include "../macros.h" #include "../p-assert.h" #include "switches_util.h" Database<String, IntElem> switches; char polarisOutputFileName[PATH_MAX]; extern "C" { int getopt(int argc, char *const *argv, const char *optstring); } int p_getsubopt( char **optionp, const char *const *tokens, char **valuep ) { char *token; char *eq = 0; char *ch = *optionp; token = *valuep = 0; if (ch == '\0') return -1; /// ... find the end of the option and the equal sign if possible. for ( ; *ch && *ch != ','; ++ch) { if (*ch == '=') eq = ch; } token = *optionp; if (eq != 0) { /// ... if we have an '=' sign then record the argument and /// ... null terminate the token. *valuep = eq + 1; *eq = '\0'; } if (*ch == '\0') *optionp = ch; else { *ch = '\0'; *optionp = ch + 1; } for (char **t = CASTAWAY(char **) tokens; *t != 0; ++t) { if (strcmp( *t, token ) == 0) return ( (CASTAWAY(char **) t) - (CASTAWAY(char **) tokens) ); } return -1; } int switch_value(char const *switch_name) { IntElem *val = switches.find_ref(switch_name); if (!val) { cerr << "WARNING: switch " << switch_name << " does not exist\n"; return 0; } return val->value(); } void define_switch(char const *switch_name, int switch_value) { String swn = switch_name; switches.ins( swn, new IntElem(switch_value)); } void adjust_switch_value(char const *switch_name, int new_value) { switches.find_ref(switch_name)->value(new_value); } /// parse_switches - parse the polaris command line /// /// supported options: /// -s <switch-name>=<value> Assign the given switch the given value /// -f <switch-file-name> Use the given switch file for switch values /// -v Report the current Polaris version /// -i <intrinsic-def-filename> Use the given file to define intrinsics /// (over-rides the use of the default dummy_pgm.f) int parse_switches(int argc, char *argv[], char const *version, char **intrin_file_name) { *intrin_file_name = 0; polarisOutputFileName[0] = '\0'; /// ... read the POLARIS_SWITCHES database. read_switches(); /// ... Do the command line parsing. int c; extern char *optarg; extern int optind; optind = 1; int cmd_error = 0; typedef char *char_p; char **tokens = new char_p[ switches.entries()+2 ]; int position = 0; for (KeyIterator<String, IntElem> sw_iter = switches; sw_iter.valid(); ++sw_iter) { tokens[ position++ ] = CASTAWAY(char *) (const char *) sw_iter.current_key(); } tokens[ position ] = 0; while ((c = getopt(argc, argv, "f:i:s:vo:")) != -1) { switch (c) { case 'f': { char *file_name = optarg; read_switch_file( file_name ); } break; case 'i': { *intrin_file_name = optarg; } break; case 'o': strcpy(polarisOutputFileName, optarg); break; case 's': { char *value, *options = optarg; while (*options != '\0') { char *last_option = options; int index = p_getsubopt(&options, tokens, &value); if (index == -1) { cout << "unknown switch " << last_option << endl; cmd_error++; } else { int new_value = 1; if (value) new_value = atoi(value); adjust_switch_value(tokens[index], new_value); } } } break; case 'v': { cout << "Polaris version " << version << endl; } break; case '?': cmd_error++; } if (cmd_error) { cout << "Usage: " << argv[0] << endl; cout << " driver [-v] [-f switchfile] [-s switch1=value1,switch2=value2,...] source" << endl; exit(1); } } delete [] tokens; return optind; } int read_switch_file( const char * file_name ) { String sw_file = file_name; ifstream inp(sw_file); if (inp.fail()) { cout << "could not open switch file: " << sw_file << "\n"; return -1; } char *tokens[500], buffer[500]; int default_values[500]; for (int i = 0; !inp.eof(); i++) { inp.getline(buffer, sizeof(buffer)); if ((buffer[0] == '#') || (strlen(buffer) == 0)) { /// ... Ignore comment and empty lines } else { istrstream str(buffer); tokens[i] = new char[1024]; str >> tokens[i] >> default_values[i]; if (strlen(tokens[i]) > 0) { /// ... Ignore whitespace lines define_switch(tokens[i], default_values[i]); } delete [] tokens[i]; } } inp.close(); return 0; } int read_switches() { String sw_file; char *env_var = "POLARIS_SWITCHES"; char *ptr; if ((ptr = getenv( env_var )) != 0) sw_file = ptr; else { ptr = getenv( "POLARIS_ROOT" ); p_assert(ptr, "Must 'setenv POLARIS_ROOT <Polaris root directory>'"); sw_file = ptr; sw_file += "/switches"; } return read_switch_file( sw_file ); } void dump_switches( ostream &o ) { const String *id; int column = 0; for (id = switches.first_ref(); id; id = switches.successor_ref(*id)) { IntElem *val = switches.find_ref( *id ); if ((column++ % 3) == 0) { o << endl; o << "* "; } o << " " <<(const char *)*id << " "<<val->value(); /// o << " " << form("%16s", (const char *)*id) /// << form("%3d", val->value()); } o << endl; }

© 1995-2005 University of Illinois, Urbana-Champaign. All rights reserved.  Fri Mar 25 23:06:13 2005