source: fedd/abac-src/rbtm/util/ResultEvidenceMap.java @ c9b3f49

package edu.stanford.peer.rbtm.util;

import java.util.*;

import edu.stanford.peer.rbtm.RBTMConstants;
//import edu.stanford.peer.rbtm.util.*;

/** Stores results and their evidences. */
public class ResultEvidenceMap implements RBTMConstants
{
        static final public int DEFAULT_CAPACITY = 23;

    //private int       _type = TRACK_NONE;
    private int         _type = TRACK_ALL;
    private HashMap _map;

    public ResultEvidenceMap(int trackType, int capacity) {
        _type = trackType; 
        _map = new HashMap(capacity);
    }

    public ResultEvidenceMap() { 
                this(TRACK_ALL, DEFAULT_CAPACITY);
    }

    public ResultEvidenceMap(int trackType) { 
            this(trackType, DEFAULT_CAPACITY);
        }

    /**
     * Retrieve a Result-to-Evidence mapping. If constructed with trackType
     * TRACK_NONE, the evidence will be null.
     * @return the evidence, if any, bound to the result.
     */
    public Object getResultEvidence(Object result) {
        return _map.get(result);
    }

        /**
         * Add a Result-to-Evidence mapping.  
         * @return true if this is a new Result, false if it is not
         */
    public boolean putResultEvidence(Object result, Object evidence) {
        if (! _map.containsKey(result)) {  // New result
                        switch (_type) {
                        case TRACK_NONE:
                    _map.put(result, null);
                                break;
                        case TRACK_ANY:
                                _map.put(result, evidence);
                                break;
                        case TRACK_ALL:
                                HashSet eviSet = new HashSet(7);
                                eviSet.add(evidence);
                                System.out.println("###Adding " + result + ","
                                                   + evidence);
                                _map.put(result, eviSet);
                                break;
                        }
            return true;
        } else {  // Result already exist
            if (_type == TRACK_ALL) {
                ((Set)_map.get(result)).add(evidence);
            }
            return false;
        }
    }
    
    public boolean containsResult(Object result) {
        return _map.containsKey(result);
    }

        public Set resultSet() {
                return _map.keySet();
        }

        public String toString() {
            return  "["+ _type + "]" + _map.toString();
            //return _map.toString();
        }
}