TinkerKnobs.C

Go to the documentation of this file.

// TinkerKnobs.C
// 10/14/97 WAH: Added knobs::Parse.

#include "TinkerKnobs.H"

#define TINKERKNOBS_DEBUG
//#define TINKERKNOBS_MAIN

#include <stdio.h>
#include <stdlib.h>
#include <ctype.h>
#include <string.h>

#define DEFAULT_KNOBTABLE_SIZE (64)

int
knobs::Load (char *knobfile)
{
  if (knobfile == NULL)
    return 0;

  char commandline[200];

  sprintf (commandline, "m4 %s", knobfile);

  FILE *fp = popen (commandline, "r");
  if (fp == NULL)
    return 0;

  curfname = knobfile;
  int numread = Load (fp);
  pclose (fp);
  curfname = NULL;
  return numread;
} // end knobs::Load (char *)

int
knobs::Load (FILE *kfp)
{
  char line[200], param_name[140], param_type[10], param_string[200];
  knobnamelist *knltail;
  int numknobs = 0, numread, linenum = 0;

  for (knltail = names; knltail != NULL && knltail->next != NULL;
       knltail = knltail->next) /* do nothing */;

  while ((fgets (line, 199, kfp)) != NULL)
    {
      linenum++;
      if (line[0] == '#' || isspace(line[0]))
        continue;

      numread = sscanf (line, "%s%s%s", param_name, param_type,
                        param_string);
      if (numread != 3)
        {
          fprintf (stderr, "Error reading knob file %s, line %d.\n",
                   curfname, linenum);
          return numknobs;
        } // end if

      if (strcmp (param_type, "integer") == 0)
        {
          int d = atoi (param_string);
          t->Set (param_name, d);
        } // end if
      else if (strcmp (param_type, "boolean") == 0)
        {
          char d; int i;

          for (i = 0; i < strlen (param_string); i++)
            param_string[i] = toupper (param_string[i]);
          if (strcmp (param_string, "TRUE") == 0 ||
              strcmp (param_string, "T") == 0 ||
              strcmp (param_string, "YES") == 0 ||
              strcmp (param_string, "Y") == 0)
            d = KNOB_TRUE;
          else if (strcmp (param_string, "FALSE") == 0 ||
                   strcmp (param_string, "F") == 0 ||
                   strcmp (param_string, "NO") == 0 ||
                   strcmp (param_string, "N") == 0)
            d = KNOB_FALSE;
          else
            {
              fprintf (stderr, "Misspelled boolean in knob file %s, "
                       "knob %s. Setting to false.\n", curfname,
                       param_name);
              d = KNOB_FALSE;
            } // end else
          t->Set (param_name, (char) d);
        } // end else if
      else if (strcmp (param_type, "string") == 0)
        {
          char *d = new char [strlen (param_string) + 1];
          strcpy (d, param_string);
          t->Set (param_name, d);
        } // end else if
      else if (strcmp (param_type, "float") == 0)
        {
          double d = atof (param_string);
          t->Set (param_name, d);
        } // end else if
      else
        {
          fprintf (stderr, "Unrecognized knob type %s for knob %s in knob "
                   "file %s.\n", param_type, param_name, curfname);
          continue;
        } // end else

      numknobs++;
      if (names != NULL)
        {
          knobnamelist *knl;

          for (knl = names; knl != NULL; knl = knl->next)
            if (strcmp (knl->name, param_name) == 0)
              break;
          if (knl == NULL)
            {
              knltail->next = new knobnamelist (param_name);
              knltail = knltail->next;
            } // end if
        } // end if
      else
        names = knltail = new knobnamelist (param_name);

    } // end while

  fclose (kfp);
  return numknobs;
} // end knobs::Load (FILE *)

int
knobs::Save (char *knobfile = NULL)
{
  if (knobfile == NULL)
    knobfile = customknobfile;

  if (strcmp (knobfile, getenv ("TINKERKNOBS")) == 0)
    {
      fprintf (stderr, "Not allowed to overwrite common knob file.\n");
      return 0;
    } // end if

  FILE *fp = fopen (knobfile, "w");
  if (fp == NULL)
    return 0;

  curfname = knobfile;
  int numsaved = Save (fp);
  fclose (fp);
  curfname = NULL;
  return numsaved;
} // end knobs::Save (char *)

int
knobs::Save (FILE *kfp)
{
  char line[200], param_name[140], param_type[10], param_string[100];
  int numknobs = 0, putstatus, linenum = 1;
  knobnamelist *knl;

  putstatus = fputs ("# Generated knob file\n", kfp);
  if (putstatus == EOF)
    {
      fprintf (stderr, "Error writing first line of %s.\n", curfname);
      return 0;
    } // end if

  for (knl = names; knl != NULL; knl = knl->next)
    {
      strcpy (param_name, knl->name);
      switch (t->TypeOf (param_name))
        {
        case THT_CHAR:
          strcpy (param_type, "boolean");
          if (t->LookupChar (param_name) == KNOB_TRUE)
            strcpy (param_string, "true");
          else
            strcpy (param_string, "false");
          break;
        case THT_INT:
          strcpy (param_type, "integer");
          sprintf (param_string, "%d", t->LookupInt (param_name));
          break;
        case THT_DOUBLE:
          strcpy (param_type, "float");
          sprintf (param_string, "%lf", t->LookupDouble (param_name));
          break;
        case THT_POINTER:
          strcpy (param_type, "string");
          strcpy (param_string, (char *) t->LookupPointer (param_name));
          break;
        default:
          fprintf (stderr, "Corrupted hashtable!\n");
          exit(1);
        } // end switch

      sprintf (line, "%s\t%s\t%s\n", param_name, param_type, param_string);
      putstatus = fputs (line, kfp);
      if (putstatus == EOF)
        {
          fprintf (stderr, "Error writing knob %s to knob file %s.\n",
                   param_name, curfname);
          return numknobs;
        } // end if

      numknobs++; linenum++;
    } // end for

  return numknobs;
} // end knobs::Save (FILE *)

int
knobs::Parse (int argc, char *argv[])
{
  char *shortarg, *knob_name, *knob_type, *knob_value, *tmparg;
  int len, copyamount, numknobs = 0;
  knobnamelist *knltail;

  for (knltail = names; knltail != NULL && knltail->next != NULL;
       knltail = knltail->next) /* do nothing */;

  // Process all arguments
  for (int i = 1; i < argc; i++ )
    {
      //      shortarg = &(argv[i][2]);

      // Parse the current argument. Arguments are of the form:
      // <knob-panel>:<knob-name>     <knob-type>       <knob-value>

      if (strncmp (argv[i], "-K", 2) != 0) continue;  // not knob argument
      len = strlen (argv[i]);
      //      fprintf (stderr, "Knob arg is %s.\n", argv[i]);
      if ( len < 3 )
        {
          fprintf (stderr, "Knobs::Parse: Knob string %s too short.\n",
                   argv[i]);
          exit (-1);
        } // end if
      // If you find a quote, check for a match. If not, assume they were
      // stripped by the shell.
      shortarg = new char [len];
      if (argv[i][2] == '\'' || argv[i][2] == '\"')
        {
          // Make sure that the last char is also a quote.
          if (argv[i][len-1] != argv[i][2])
            {
              fprintf (stderr, "knobs::Parse: Badly formed knob string:\n"
                       "%s\n", argv[i]);
              exit (-1);
            } // end if

          strncpy (shortarg, &(argv[i][3]), len - 3);
          shortarg[len - 3] = '\0';
        } // end if
      else
        {
          strncpy (shortarg, &(argv[i][2]), len - 2);
          shortarg[len - 2] = '\0';
        } // end else
      //      fprintf (stderr, "Short arg is %s.\n", shortarg);
            
      if ((knob_name = strtok (shortarg, " ")) == NULL)
        {
          fprintf (stderr, "knobs::Parse: Invalid knob string:\n"
                   "%s\n", shortarg);
          exit (-1);
        } // end if
      if ((knob_type = strtok ( NULL, " ")) == NULL)
        {
          fprintf (stderr, "knobs::Parse: Invalid knob string:\n"
                   "%s\n", shortarg);
          exit (-1);
        } // end if
      if ((knob_value = strtok (NULL, " ")) == NULL)
        {
          fprintf (stderr, "knobs::Parse: Invalid knob string:\n"
                   "%s\n", shortarg);
          exit (-1);
        } // end if

      // derr( "   Knob name: " << knob_name << ", knob type: " << knob_type
      //            << ", knob value: " << knob_value << "\n" );
      if (strcmp (knob_type, "boolean") == 0)
        {
          for (int j = 0; j < strlen (knob_value); j++)
            knob_value[j] = toupper (knob_value[j]);
          if (strcmp (knob_value, "YES") == 0 ||
              strcmp (knob_value, "Y") == 0 ||
              strcmp (knob_value, "T") == 0 ||
              strcmp (knob_value, "TRUE") == 0)
            t->Set (knob_name, KNOB_TRUE);
          else if (strcmp (knob_value, "NO") == 0 ||
              strcmp (knob_value, "N") == 0 ||
              strcmp (knob_value, "F") == 0 ||
              strcmp (knob_value, "FALSE") == 0)
            t->Set (knob_name, KNOB_FALSE);
          else
            {
              fprintf (stderr, "knobs::Parse: Invalid boolean value %s.\n",
                       knob_value);
              exit (-1);
            } // end else
        } // end if
      else if (strcmp (knob_type, "integer") == 0)
        {
          int rc;

          rc = atoi (knob_value);
          t->Set (knob_name, rc);
        } // end else if
      else if (strcmp (knob_type, "float") == 0)
        {
          double rc;

          rc = atof (knob_value);
          t->Set (knob_name, rc);
        } // end else if
      else if (strcmp (knob_type, "string") == 0)
        {
          char *tempstring = new char [strlen (knob_value) + 1];
          strcpy (tempstring, knob_value);
          t->Set (knob_name, tempstring);
          //      delete[] tempstring;
        } // end else
      else
        {
          fprintf (stderr, "knobs::Parse: Invalid knob type %s.\n", knob_type);
          exit (-1);
        } // end else

      numknobs++;
      if (names != NULL)
        {
          knobnamelist *knl;

          for (knl = names; knl != NULL; knl = knl->next)
            if (strcmp (knl->name, knob_name) == 0)
              break;
          if (knl == NULL)
            {
              knltail->next = new knobnamelist (knob_name);
              knltail = knltail->next;
            } // end if
        } // end if
      else
        names = knltail = new knobnamelist (knob_name);
    } // end for

  return numknobs;
} // end knobs::Parse

#ifdef TINKERKNOBS_MAIN

int main()
{
  knobs *k;
  char *string;

  k = new knobs();
  k->SetDefaultPanel ("bill");
  k->Read ("favorite-Transformer", string);
  printf ("%s is cool.\n", string);
  k->Read ("favorite-Transformer", string);
  printf ("%s is fast.\n", string);
  k->Save ("MrKnobby.txt");
  delete k;
  return 0;
} // end main (test)

#endif

---