edits

[satune.git] / src / ASTAnalyses / Encoding / encodinggraph.cc

diff --git a/src/ASTAnalyses/Encoding/encodinggraph.cc b/src/ASTAnalyses/Encoding/encodinggraph.cc
index 08db96d7b55cc47e02c7be0e394ae1135270ce5a..3bcf68ea6f823542d2ce810bc763944af560c349 100644 (file)
--- a/src/ASTAnalyses/Encoding/encodinggraph.cc
+++ b/src/ASTAnalyses/Encoding/encodinggraph.cc
@@ -62,7 +62,26 @@ void EncodingGraph::encode() {
                case ELEMFUNCRETURN: {
                        ElementEncoding *encoding=getElementEncoding(e);
                        if (encoding->getElementEncodingType() == ELEM_UNASSIGNED) {
                case ELEMFUNCRETURN: {
                        ElementEncoding *encoding=getElementEncoding(e);
                        if (encoding->getElementEncodingType() == ELEM_UNASSIGNED) {

-                               //Do assignment...
+                               EncodingNode *n = getNode(e);
+                               ASSERT(n != NULL);
+                               ElementEncodingType encodetype=n->getEncoding();
+                               encoding->setElementEncodingType(encodetype);
+                               if (encodetype == UNARY || encodetype == ONEHOT) {
+                                       encoding->encodingArrayInitialization();
+                               } else if (encodetype == BINARYINDEX) {
+                                       EncodingSubGraph * subgraph = graphMap.get(n);
+                                       uint encodingSize = subgraph->getEncodingSize(n);
+                                       uint paddedSize = encoding->getSizeEncodingArray(encodingSize);
+                                       encoding->allocInUseArrayElement(paddedSize);
+                                       encoding->allocEncodingArrayElement(paddedSize);
+                                       Set * s=e->getRange();
+                                       for(uint i=0;i<s->getSize();i++) {
+                                               uint64_t value=s->getElement(i);
+                                               uint encodingIndex=subgraph->getEncoding(n, value);
+                                               encoding->setInUseElement(encodingIndex);
+                                               encoding->encodingArray[encodingIndex] = value;
+                                       }
+                               }

                        }
                        break;
                }
                        }
                        break;
                }


@@ -292,6 +311,14 @@ EncodingNode * EncodingGraph::createNode(Element *e) {
        return n;
 }
 
        return n;
 }
 

+EncodingNode * EncodingGraph::getNode(Element *e) {
+       if (e->type == ELEMCONST)
+               return NULL;
+       Set *s = e->getRange();
+       EncodingNode *n = encodingMap.get(s);
+       return n;
+}
+

 void EncodingNode::addElement(Element *e) {
        elements.add(e);
 }
 void EncodingNode::addElement(Element *e) {
        elements.add(e);
 }