source: src/main/java/negotiator/onetomany/DemandPlan.java@ 222

package negotiator.onetomany;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Set;

/**
 * @author Faria Nassiri-Mofakham
 *
 */
public class DemandPlan 
{
        private HashMap<Product, Integer> demandPlan;


        public DemandPlan() 
        {
                this.demandPlan = new HashMap<Product, Integer>();
        }

        public HashMap<Product, Integer> getDemandPlan() 
        {
                return demandPlan;
        }

//      public void addProduct(Product p, Integer i) 
//      {
//              demandPlan.add(p,i);
//      }

        @Override
        public String toString() 
        {
                return demandPlan.toString();
        }
        
        public void setQuantity(Product p, Integer n)
        {
                demandPlan.put(p,n);
        }
        
        public Portfolio getPortfolio()
        {
                return (Portfolio) demandPlan.keySet();
        }
        
        public Integer getQuantity(Product p)
        {
                if (this.demandPlan.containsKey(p))
                        return demandPlan.get(p);
                else
                        return 0;
        }
        
        //Generates a random number between 1 to 5
//      public Integer intQuantity()
        public int intQuantity()

        {
//              double q = Math.random();
//              q *=5;
//              q++;
//                              return Integer((int)q);
                //return (int)q;
                return (int)Math.random()*5+1;
        }
        
        public Set<Product> getKeys()
        {
                return demandPlan.keySet();
        }
        
        public Set<Integer> getValues()
        {
                return (Set<Integer>) demandPlan.values(); 
        }
}