Repair/RepairCompiler/MCC/CLI.java

   1 package MCC;
   2
   3 import java.util.*;
   4 import MCC.IR.DebugItem;
   5
   6 /**
   7  * A generic command-line interface for 6.035 compilers.  This class
   8  * provides command-line parsing for student projects.  It recognizes
   9  * the required <tt>-target</tt>, <tt>-debug</tt>, <tt>-opt</tt>, and
  10  * <tt>-o</tt> switches, and generates a name for input and output
  11  * files.
  12  *
  13  * @author  le01, 6.035 Staff (<tt>6.035-staff@mit.edu</tt>)
  14  * @version <tt>$Id: CLI.java,v 1.14 2005/02/20 20:31:28 bdemsky Exp $</tt>
  15  */
  16 public class CLI {
  17     /**
  18      * Array indicating which optimizations should be performed.  If
  19      * a particular element is true, it indicates that the optimization
  20      * named in the optnames[] parameter to parse with the same index
  21      * should be performed.
  22      */
  23     public boolean opts[];
  24
  25     /**
  26      * Vector of String containing the command-line arguments which could
  27      * not otherwise be parsed.
  28      */
  29     public Vector extras;
  30
  31     /**
  32      * Vector of String containing the optimizations which could not be
  33      * parsed.  It is okay to complain about anything in this list, even
  34      * without the <tt>-debug</tt> flag.
  35      */
  36     public Vector extraopts;
  37
  38     /**
  39      * Name of the file to put the output in.
  40      */
  41     public String outfile;
  42
  43     /**
  44      * Name of the file to get input from.  This is null if the user didn't
  45      * provide a file name.
  46      */
  47     public String infile;
  48
  49     /**
  50      * The debug flag.  This is true if <tt>-debug</tt> was passed on
  51      * the command line, requesting debugging output.
  52      */
  53     public boolean debug;
  54
  55     /**
  56      * Verbose output
  57      */
  58     public int verbose;
  59
  60     /**
  61      * Public constructor.  Sets up default values for all of the
  62      * result fields.  Specifically, sets the input and output files
  63      * to null, the target to DEFAULT, and the extras and extraopts
  64      * arrays to new empty Vectors.
  65      */
  66     public CLI() {
  67         outfile = null;
  68         infile = null;
  69         extras = new Vector();
  70         extraopts = new Vector();
  71         verbose = 0;
  72     }
  73
  74     /**
  75      * Parse the command-line arguments.  Sets all of the result fields
  76      * accordingly. <BR>
  77      * <ul>
  78      * <li><TT>-target <I>target</I></TT> sets the CLI.target field based
  79      * on the <I>target</I> specified. </li>
  80      * <li><TT>scan</TT> or <TT>scanner</TT> specifies CLI.SCAN</li>
  81      * <li><TT>parse</TT> specifies CLI.PARSE</li>
  82      * <li><TT>inter</TT> specifies CLI.INTER</li>
  83      * <li><TT>lowir</TT> specifies CLI.LOWIR</li>
  84      * <TT>assembly</TT> or <TT>codegen</TT> specifies CLI.ASSEMBLY</li>
  85      * </ul>
  86      * The boolean array opts[] indicates which, if any, of the
  87      * optimizations in optnames[] should be performed; these arrays
  88      * are in the same order.
  89      *
  90      * @param args Array of arguments passed in to the program's Main
  91      *   function.
  92      * @param optnames Ordered array of recognized optimization names.  */
  93     public void parse(String args[]) {
  94
  95         String optnames[] = {};
  96         int context = 0;
  97         String ext = ".out";
  98
  99         opts = new boolean[optnames.length];
 100
 101         if (args.length==0) {
 102             System.out.println("-debugcompiler -- print out debug messages");
 103             System.out.println("-depth depthnum constraintnum -- generate dependency graph from constraintnum with depth of depthnum");
 104             System.out.println("-depthconj depthnum constraintnum conjunctionnum -- generate dependency graph from constraintnum with depth of depthnum");
 105             System.out.println("-instrument -- generate instrumentation code");
 106             System.out.println("-aggressivesearch");
 107             System.out.println("-prunequantifiernodes");
 108             System.out.println("-cplusplus");
 109             System.out.println("-time");
 110             System.out.println("-omitcomp");
 111             System.out.println("-mergenodes");
 112             System.exit(-1);
 113         }
 114
 115         for (int i = 0; i < args.length; i++) {
 116             if (args[i].equals("-debugcompiler")) {
 117                 context = 0;
 118                 debug = true;
 119             } else if (args[i].equals("-checkonly")) {
 120                 Compiler.REPAIR=false;
 121             } else if (args[i].equals("-omitcomp")) {
 122                 Compiler.OMITCOMP=true;
 123             } else if (args[i].equals("-mergenodes")) {
 124                 Compiler.MERGENODES=true;
 125             } else if (args[i].equals("-depth")) {
 126                 Compiler.debuggraphs.add(new DebugItem(Integer.parseInt(args[i+1]),Integer.parseInt(args[i+2])));
 127                 i+=2;
 128             } else if (args[i].equals("-depthconj")) {
 129                 Compiler.debuggraphs.add(new DebugItem(Integer.parseInt(args[i+1]),Integer.parseInt(args[i+2]),Integer.parseInt(args[i+3])));
 130                 i+=3;
 131             } else if (args[i].equals("-debug")) {
 132                 Compiler.GENERATEDEBUGHOOKS=true;
 133             } else if (args[i].equals("-time")) {
 134                 Compiler.TIME=true;
 135             } else if (args[i].equals("-instrument")) {
 136                 Compiler.GENERATEINSTRUMENT=true;
 137             } else if (args[i].equals("-aggressivesearch")) {
 138                 Compiler.AGGRESSIVESEARCH=true;
 139             } else if (args[i].equals("-prunequantifiernodes")) {
 140                 Compiler.PRUNEQUANTIFIERS=true;
 141             } else if (args[i].equals("-cplusplus")) {
 142                 Compiler.ALLOCATECPLUSPLUS=true;
 143             } else if (args[i].equals("-verbose") || args[i].equals("-v")) {
 144                 context = 0;
 145                 verbose++;
 146             } else if (args[i].equals("-opt"))
 147                 context = 1;
 148             else if (args[i].equals("-o"))
 149                 context = 2;
 150             else if (context == 1) {
 151                 boolean hit = false;
 152                 for (int j = 0; j < optnames.length; j++) {
 153                     if (args[i].equals("all") ||
 154                         (args[i].equals(optnames[j]))) {
 155                         hit = true;
 156                         opts[j] = true;
 157                     }
 158                     if (args[i].equals("-" + optnames[j])) {
 159                         hit = true;
 160                         opts[j] = false;
 161                     }
 162                 }
 163                 if (!hit)
 164                     extraopts.addElement(args[i]);
 165             } else if (context == 2) {
 166                 outfile = args[i];
 167                 context = 0;
 168             } else {
 169                 boolean hit = false;
 170                 for (int j = 0; j < optnames.length; j++) {
 171                     if (args[i].equals("-" + optnames[j])) {
 172                         hit = true;
 173                         opts[j] = true;
 174                     }
 175                 }
 176                 if (!hit) {
 177                     extras.addElement(args[i]);
 178                 }
 179             }
 180         }
 181
 182         // grab infile and lose extra args
 183         int i = 0;
 184         while (infile == null && i < extras.size()) {
 185             String fn = (String) extras.elementAt(i);
 186
 187             if (fn.charAt(0) != '-')
 188             {
 189                 infile = fn;
 190                 extras.removeElementAt(i);
 191             }
 192             i++;
 193         }
 194     }
 195 }