source: ai2020/group7/src/main/java/geniusweb/exampleparties/nataliaparty/DoubleMap.java

package geniusweb.exampleparties.nataliaparty;

import java.util.HashMap;
import java.util.Map;

import geniusweb.issuevalue.Value;

public class DoubleMap {

        private Map<String, Map<Value, Double>> map;

        public DoubleMap() {
                this.map = new HashMap<>();
        }

        public void put(String issue, Value value, double x) {
                Map<Value, Double> values;
                if (map.containsKey(issue)) {
                        values = map.get(issue);
                } else {
                        values = new HashMap<>();
                        map.put(issue, values);
                }
                values.put(value, x);
        }

        public double get(String issue, Value value) {
                Map<Value, Double> values;
                if (map.containsKey(issue)) {
                        values = map.get(issue);
                } else {
                        return 0;
                }
                if (values.containsKey(value)) {
                        return values.get(value);
                } else {
                        return 0;
                }
        }

        public void add(String issue, Value value, double x) {
                Map<Value, Double> values;
                if (map.containsKey(issue)) {
                        values = map.get(issue);
                } else {
                        values = new HashMap<>();
                        map.put(issue, values);
                }
                if (values.containsKey(value)) {
                        values.replace(value, values.get(value) + x);
                } else {
                        values.put(value, x);
                }
        }
}