src/jalview/bin/argparser/BootstrapArgs.java

   1 package jalview.bin.argparser;
   2
   3 import java.util.ArrayList;
   4 import java.util.Arrays;
   5 import java.util.HashMap;
   6 import java.util.List;
   7 import java.util.Map;
   8
   9 import jalview.bin.argparser.Arg.Opt;
  10 import jalview.util.FileUtils;
  11
  12 public class BootstrapArgs
  13 {
  14   // only need one
  15   private Map<Arg, List<String>> bootstrapArgMap = new HashMap<>();
  16
  17   public static BootstrapArgs getBootstrapArgs(String[] args)
  18   {
  19     List<String> argList = new ArrayList<>(Arrays.asList(args));
  20     return new BootstrapArgs(argList);
  21   }
  22
  23   private BootstrapArgs(List<String> args)
  24   {
  25     parse(args);
  26   }
  27
  28   private void parse(List<String> args)
  29   {
  30     if (args == null)
  31       return;
  32     for (int i = 0; i < args.size(); i++)
  33     {
  34       String arg = args.get(i);
  35       String argName = null;
  36       String val = null;
  37       if (arg.startsWith(ArgParser.DOUBLEDASH))
  38       {
  39         // remove "--"
  40         arg = arg.substring(ArgParser.DOUBLEDASH.length());
  41         int equalPos = arg.indexOf(ArgParser.EQUALS);
  42         if (equalPos > -1
  43                 && ArgParser.argMap.containsKey(arg.substring(0, equalPos)))
  44         {
  45           argName = arg.substring(0, equalPos);
  46           val = arg.substring(equalPos + 1);
  47         }
  48         // check for boolean prepended by "no"
  49         else if (arg.startsWith(ArgParser.NEGATESTRING)
  50                 && ArgParser.argMap.containsKey(
  51                         arg.substring(ArgParser.NEGATESTRING.length())))
  52         {
  53           argName = arg.substring(ArgParser.NEGATESTRING.length());
  54           val = "false";
  55         }
  56         else if (ArgParser.argMap.containsKey(arg))
  57         {
  58           argName = arg;
  59           val = "true";
  60         }
  61
  62         Arg a = ArgParser.argMap.get(argName);
  63
  64         if (a == null || !a.hasOption(Opt.BOOTSTRAP))
  65         {
  66           // not a valid bootstrap arg
  67           continue;
  68         }
  69
  70         if (a.hasOption(Opt.STRING))
  71         {
  72           if (equalPos == -1)
  73           {
  74             addAll(a, ArgParser.getShellGlobbedFilenameValues(a, args,
  75                     i + 1));
  76           }
  77           else
  78           {
  79             if (a.hasOption(Opt.GLOB))
  80               addAll(a, FileUtils.getFilenamesFromGlob(val));
  81             else
  82               add(a, val);
  83           }
  84         }
  85         else
  86         {
  87           add(a, val);
  88         }
  89       }
  90     }
  91   }
  92
  93   public boolean contains(Arg a)
  94   {
  95     return bootstrapArgMap.containsKey(a);
  96   }
  97
  98   public List<String> getList(Arg a)
  99   {
 100     return bootstrapArgMap.get(a);
 101   }
 102
 103   private List<String> getOrCreateList(Arg a)
 104   {
 105     List<String> l = getList(a);
 106     if (l == null)
 107     {
 108       l = new ArrayList<>();
 109       putList(a, l);
 110     }
 111     return l;
 112   }
 113
 114   private void putList(Arg a, List<String> l)
 115   {
 116     bootstrapArgMap.put(a, l);
 117   }
 118
 119   /*
 120    * Creates a new list if not used before,
 121    * adds the value unless the existing list is non-empty
 122    * and the arg is not MULTI (so first expressed value is
 123    * retained).
 124    */
 125   private void add(Arg a, String s)
 126   {
 127     List<String> l = getOrCreateList(a);
 128     if (a.hasOption(Opt.MULTI) || l.size() == 0)
 129     {
 130       l.add(s);
 131     }
 132   }
 133
 134   private void addAll(Arg a, List<String> al)
 135   {
 136     List<String> l = getOrCreateList(a);
 137     if (a.hasOption(Opt.MULTI))
 138     {
 139       l.addAll(al);
 140     }
 141   }
 142
 143   /*
 144    * Retrieves the first value even if MULTI.
 145    * A convenience for non-MULTI args.
 146    */
 147   public String get(Arg a)
 148   {
 149     if (!bootstrapArgMap.containsKey(a))
 150       return null;
 151     List<String> aL = bootstrapArgMap.get(a);
 152     return (aL == null || aL.size() == 0) ? null : aL.get(0);
 153   }
 154
 155   public boolean getBoolean(Arg a, boolean d)
 156   {
 157     if (!bootstrapArgMap.containsKey(a))
 158       return d;
 159     return Boolean.parseBoolean(get(a));
 160   }
 161
 162   public boolean getBoolean(Arg a)
 163   {
 164     if (!(a.hasOption(Opt.BOOLEAN) || a.hasOption(Opt.UNARY)))
 165     {
 166       return false;
 167     }
 168     if (bootstrapArgMap.containsKey(a))
 169       return Boolean.parseBoolean(get(a));
 170     else
 171       return a.getDefaultBoolValue();
 172   }
 173 }