X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=src%2FAlloyEnc%2Fsignatureenc.cc;h=6971a5edcda4978f700531735ba3d81e934d5a35;hb=3896ad686a910868d7bf2988cd83a4fe3da700b2;hp=f0ed880fb33a6c8b8836dea2b37f5f81a24016b2;hpb=c8012458e3e2777c72ed1d9b217dfa3d69422a4d;p=satune.git

diff --git a/src/AlloyEnc/signatureenc.cc b/src/AlloyEnc/signatureenc.cc
index f0ed880..6971a5e 100644
--- a/src/AlloyEnc/signatureenc.cc
+++ b/src/AlloyEnc/signatureenc.cc
@@ -31,6 +31,17 @@ void SignatureEnc::updateMaxValue(Set *set){
 	}
 }
 
+BooleanSig *SignatureEnc::getBooleanSignature(Boolean *bvar){
+	BooleanSig *bsig = (BooleanSig *)encoded.get((void *)bvar);
+	if(bsig == NULL){
+		bsig = new BooleanSig(signatures.getSize());
+		encoded.put(bvar, bsig);
+		signatures.push(bsig);
+		alloyEncoder->writeToFile(bsig->getSignature());
+	}
+	return bsig;
+}
+
 ElementSig *SignatureEnc::getElementSignature(Element *element){
 	ElementSig *esig = (ElementSig *)encoded.get((void *)element);
 	if(esig == NULL){
@@ -44,8 +55,6 @@ ElementSig *SignatureEnc::getElementSignature(Element *element){
 			updateMaxValue(set);
 		}
 		esig = new ElementSig(signatures.getSize(), ssig);
-		element->print();
-		model_print(" = Element%u\n", signatures.getSize());
 		encoded.put(element, esig);
 		signatures.push(esig);
 		alloyEncoder->writeToFile(esig->getSignature());