src/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     public String verilogVal(String var) {
  42         //return "(("+var+" & "+allmax+"'b"+Long.toString(valmask,2)+") >> "+valmaskmin+")";
  43         return ""+var+"["+valmaskmax+":"+valmaskmin+"]";
  44     }
  45
  46     public long getval(long in) {
  47         return (in & valmask) >>> valmaskmin;
  48     }
  49     public long getval(BitVector targ) {
  50         long in = 0;
  51         for(int i=0; i<targ.length(); i++)
  52             if (targ.get(i))
  53                 in |= (1L << i);
  54         return (in & valmask) >>> valmaskmin;
  55     }
  56     public long setval(long in, BitVector targ) {
  57         long ret = 0;
  58         for(int i=0; i<targ.length(); i++)
  59             if (targ.get(i))
  60                 ret |= (1L << i);
  61         if (targ.get(targ.length()-1))
  62             ret |= (-1L << targ.length());
  63         return setval(in, ret);
  64     }
  65     public long setval(long in, long targ) {
  66         if (((targ << valmaskmin) & ~valmask) != 0) throw new RuntimeException("setval() with argument bigger than mask field");
  67         return (in & ~valmask) | ((targ << valmaskmin) & valmask);
  68     }
  69     public long set(long in) {
  70         return (in & ~mask) | val;
  71     }
  72     public boolean get(long in) {
  73         return (in & mask) == val;
  74     }
  75
  76     public String toString() {
  77         return str;
  78     }
  79
  80     public int getWidth() {
  81         int ret = 0;
  82         long m = 1;
  83         for(int i=0; i<64; i++) {
  84             if ((valmask & m)!=0) ret++;
  85             m = m << 1;
  86         }
  87         return ret;
  88     }
  89
  90     public Mask(String s) {
  91         this.str = s;
  92         long mask = 0;
  93         long valmask = 0;
  94         long val = 0;
  95         int valmaskmin = Integer.MAX_VALUE;
  96         int valmaskmax = 0;
  97         int allmax = 0;
  98         this.width = s.length();
  99         for(int i=0; i<s.length(); i++) {
 100             char c = s.charAt(s.length()-1-i);
 101             switch(c) {
 102                 case '.': break;
 103                 case '0': mask |= (1L<<i); val |= (0L<<i); break;
 104                 case '1': mask |= (1L<<i); val |= (1L<<i); break;
 105                 case 'v': valmask |= (1L<<i); valmaskmin = Math.min(valmaskmin,i); valmaskmax = Math.max(valmaskmax,i); break;
 106                 default: throw new Error(""+c);
 107             }
 108             if (c!='.') allmax = Math.max(allmax,i);
 109         }
 110         this.mask = mask;
 111         this.val = val;
 112         this.valmask = valmask;
 113         this.valmaskmin = valmaskmin;
 114         this.valmaskmax = valmaskmax;
 115         this.allmax = allmax+1;
 116         this.valmaskwidth = 1 + valmaskmax - valmaskmin;
 117     }
 118
 119     public static long signExtend(long input, int wordWidth) {
 120         if ((input & (1L << (wordWidth-1)))!=0)
 121             input |= (-1L) << wordWidth;
 122         return input;
 123     }
 124
 125 }