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             {
  81               addAll(a, FileUtils.getFilenamesFromGlob(val));
  82             }
  83             else
  84             {
  85               add(a, val);
  86             }
  87           }
  88         }
  89         else
  90         {
  91           add(a, val);
  92         }
  93       }
  94     }
  95   }
  96
  97   public boolean contains(Arg a)
  98   {
  99     return bootstrapArgMap.containsKey(a);
 100   }
 101
 102   public List<String> getList(Arg a)
 103   {
 104     return bootstrapArgMap.get(a);
 105   }
 106
 107   private List<String> getOrCreateList(Arg a)
 108   {
 109     List<String> l = getList(a);
 110     if (l == null)
 111     {
 112       l = new ArrayList<>();
 113       putList(a, l);
 114     }
 115     return l;
 116   }
 117
 118   private void putList(Arg a, List<String> l)
 119   {
 120     bootstrapArgMap.put(a, l);
 121   }
 122
 123   /*
 124    * Creates a new list if not used before,
 125    * adds the value unless the existing list is non-empty
 126    * and the arg is not MULTI (so first expressed value is
 127    * retained).
 128    */
 129   private void add(Arg a, String s)
 130   {
 131     List<String> l = getOrCreateList(a);
 132     if (a.hasOption(Opt.MULTI) || l.size() == 0)
 133     {
 134       l.add(s);
 135     }
 136   }
 137
 138   private void addAll(Arg a, List<String> al)
 139   {
 140     List<String> l = getOrCreateList(a);
 141     if (a.hasOption(Opt.MULTI))
 142     {
 143       l.addAll(al);
 144     }
 145   }
 146
 147   /*
 148    * Retrieves the first value even if MULTI.
 149    * A convenience for non-MULTI args.
 150    */
 151   public String get(Arg a)
 152   {
 153     if (!bootstrapArgMap.containsKey(a))
 154       return null;
 155     List<String> aL = bootstrapArgMap.get(a);
 156     return (aL == null || aL.size() == 0) ? null : aL.get(0);
 157   }
 158
 159   public boolean getBoolean(Arg a, boolean d)
 160   {
 161     if (!bootstrapArgMap.containsKey(a))
 162       return d;
 163     return Boolean.parseBoolean(get(a));
 164   }
 165
 166   public boolean getBoolean(Arg a)
 167   {
 168     if (!(a.hasOption(Opt.BOOLEAN) || a.hasOption(Opt.UNARY)))
 169     {
 170       return false;
 171     }
 172     if (bootstrapArgMap.containsKey(a))
 173       return Boolean.parseBoolean(get(a));
 174     else
 175       return a.getDefaultBoolValue();
 176   }
 177 }