testCode/edu/berkeley/fleet/util/Mask.java

   1 package edu.berkeley.fleet.util;
   2 import edu.berkeley.fleet.api.*;
   3 import edu.berkeley.fleet.*;
   4 import java.io.*;
   5
   6 public class Mask {
   7
   8     private final String str;
   9
  10     public final long mask;
  11     public final long val;
  12     public final long valmask;
  13     public final int valmaskmin;
  14     public final int valmaskmax;
  15     public final int valmaskwidth;
  16     public final int allmax;
  17     public final int width;
  18
  19     public String verilog(String var) {
  20         // FIXME: throw an exception if this is called when no 1-bits or 0-bits were specified (ie only v-bits)
  21
  22         StringBuffer sb = new StringBuffer();
  23         sb.append("{");
  24         boolean first = true;
  25         int count = 0;
  26         for(int i=63; i>=0; i--) {
  27             if ((mask & (1L << i)) == 0) continue;
  28             if (!first) sb.append(",");
  29             first = false;
  30             sb.append(var+"["+i+"]");
  31             count++;
  32         }
  33         sb.append("}=="+count+"'b");
  34         for(int i=63; i>=0; i--) {
  35             if ((mask & (1L << i)) == 0) continue;
  36             sb.append( (val & (1L << i))==0 ? "0" : "1" );
  37         }
  38         return "("+sb.toString()+")";
  39         //return "(("+var+" & "+allmax+"'b"+Long.toString(mask,2)+")=="+allmax+"'b"+Long.toString(val,2)+")";
  40     }
  41
  42     public long getval(long in) {
  43         return (in & valmask) >>> valmaskmin;
  44     }
  45     public BitVector getvalAsBitVector(long in) {
  46         // FIXME
  47         return new BitVector(getWidth()).set(getval(in));
  48     }
  49     public BitVector getvalAsBitVector(BitVector in) {
  50         // FIXME
  51         return new BitVector(getWidth()).set(getval(in.toLong()));
  52     }
  53     public long getval(BitVector targ) {
  54         long in = 0;
  55         for(int i=0; i<targ.length(); i++)
  56             if (targ.get(i))
  57                 in |= (1L << i);
  58         return (in & valmask) >>> valmaskmin;
  59     }
  60     public long setval(long in, BitVector targ) {
  61         long ret = in;
  62         if (targ.length() != (1+valmaskmax-valmaskmin))
  63             throw new RuntimeException("size mismatch trying to do "+this+".setval("+targ+")");
  64         for(int i=valmaskmin; i<=valmaskmax; i++)
  65             if (targ.get(i-valmaskmin))
  66                 ret |= (1L << i);
  67             else
  68                 ret &= ~(1L << i);
  69         return ret;
  70     }
  71     public long setval(long in, long targ) {
  72         if (((targ << valmaskmin) & ~valmask) != 0) throw new RuntimeException("setval() with argument bigger than mask field");
  73         return (in & ~valmask) | ((targ << valmaskmin) & valmask);
  74     }
  75     public long set(long in) {
  76         return (in & ~mask) | val;
  77     }
  78     public boolean get(long in) {
  79         return (in & mask) == val;
  80     }
  81
  82     public String toString() {
  83         return str;
  84     }
  85
  86     public int getWidth() {
  87         int ret = 0;
  88         long m = 1;
  89         for(int i=0; i<64; i++) {
  90             if ((valmask & m)!=0) ret++;
  91             m = m << 1;
  92         }
  93         return ret;
  94     }
  95
  96     public Mask(String s) {
  97         this.str = s;
  98         long mask = 0;
  99         long valmask = 0;
 100         long val = 0;
 101         int valmaskmin = Integer.MAX_VALUE;
 102         int valmaskmax = 0;
 103         int allmax = 0;
 104         this.width = s.length();
 105         for(int i=0; i<s.length(); i++) {
 106             char c = s.charAt(s.length()-1-i);
 107             switch(c) {
 108                 case '.': break;
 109                 case '0': mask |= (1L<<i); val |= (0L<<i); break;
 110                 case '1': mask |= (1L<<i); val |= (1L<<i); break;
 111                 case 'v': valmask |= (1L<<i); valmaskmin = Math.min(valmaskmin,i); valmaskmax = Math.max(valmaskmax,i); break;
 112                 default: throw new Error(""+c);
 113             }
 114             if (c!='.') allmax = Math.max(allmax,i);
 115         }
 116         this.mask = mask;
 117         this.val = val;
 118         this.valmask = valmask;
 119         this.valmaskmin = valmaskmin;
 120         this.valmaskmax = valmaskmax;
 121         this.allmax = allmax+1;
 122         this.valmaskwidth = 1 + valmaskmax - valmaskmin;
 123     }
 124
 125     public static long signExtend(long input, int wordWidth) {
 126         if ((input & (1L << (wordWidth-1)))!=0)
 127             input |= (-1L) << wordWidth;
 128         return input;
 129     }
 130
 131 }