Example #1
0
void AbstractionModule::finalizeSignatures() {
  NodeManager* nm = NodeManager::currentNM();
  Debug("bv-abstraction") << "AbstractionModule::finalizeSignatures num signatures = " << d_signatures.size() <<"\n";
  TNodeSet new_signatures;

  // "unify" signatures
  for (SignatureMap::const_iterator ss = d_signatures.begin(); ss != d_signatures.end(); ++ss) {
    for (SignatureMap::const_iterator tt = ss; tt != d_signatures.end(); ++tt) {
      TNode t = getGeneralization(tt->first);
      TNode s = getGeneralization(ss->first);
      
      if (t != s) {
        int status = comparePatterns(s, t);
        Assert (status); 
        if (status < 0)
          continue;
        if (status == 1) {
          storeGeneralization(t, s); 
        } else {
          storeGeneralization(s, t); 
        }
      }
    }
  }
  // keep only most general signatures
  for (SignatureMap::iterator it = d_signatures.begin(); it != d_signatures.end(); ) {
    TNode sig = it->first; 
    TNode gen = getGeneralization(sig);
    if (sig != gen) {
      Assert (d_signatures.find(gen) != d_signatures.end()); 
      // update the count
      d_signatures[gen]+= d_signatures[sig];
      d_signatures.erase(it++); 
    } else {
      ++it;
    }
  }


  // remove signatures that are not frequent enough
  for (SignatureMap::iterator it = d_signatures.begin(); it != d_signatures.end(); ) {
    if (it->second <= 7) {
      d_signatures.erase(it++); 
    } else {
      ++it;
    }
  }
  
  for (SignatureMap::const_iterator it = d_signatures.begin(); it != d_signatures.end(); ++it) {
    TNode signature = it->first;
    // we already processed this signature
    Assert (d_signatureToFunc.find(signature) == d_signatureToFunc.end());

    Debug("bv-abstraction") << "Processing signature " << signature << " count " << it->second << "\n";
    std::vector<TypeNode> arg_types;
    TNodeSet seen;
    collectArgumentTypes(signature, arg_types, seen);
    Assert (signature.getType().isBoolean());
    // make function return a bitvector of size 1
    //Node bv_function = utils::mkNode(kind::ITE, signature, utils::mkConst(1, 1u), utils::mkConst(1, 0u)); 
    TypeNode range = NodeManager::currentNM()->mkBitVectorType(1);
    
    TypeNode abs_type = nm->mkFunctionType(arg_types, range);
    Node abs_func = nm->mkSkolem("abs_$$", abs_type, "abstraction function for bv theory");
    Debug("bv-abstraction") << " abstracted by function " << abs_func << "\n";

    // NOTE: signature expression type is BOOLEAN
    d_signatureToFunc[signature] = abs_func;
    d_funcToSignature[abs_func] = signature; 
  }

  d_statistics.d_numFunctionsAbstracted.setData(d_signatureToFunc.size());
  
  Debug("bv-abstraction") << "AbstractionModule::finalizeSignatures abstracted " << d_signatureToFunc.size() << " signatures. \n";
}