ProductionRule.java

package games.strategy.engine.data;

import java.util.Map.Entry;

import games.strategy.triplea.Constants;
import games.strategy.util.IntegerMap;

public class ProductionRule extends DefaultNamed {
  private static final long serialVersionUID = -6598296283127741307L;
  private IntegerMap<Resource> m_cost = new IntegerMap<>();
  private IntegerMap<NamedAttachable> m_results = new IntegerMap<>();

  /** Creates new ProductionRule. */
  public ProductionRule(final String name, final GameData data) {
    super(name, data);
  }

  /** Creates new ProductionRule. */
  public ProductionRule(final String name, final GameData data, final IntegerMap<NamedAttachable> results,
      final IntegerMap<Resource> costs) {
    super(name, data);
    m_results = results;
    m_cost = costs;
  }

  protected void addCost(final Resource resource, final int quantity) {
    m_cost.put(resource, quantity);
  }

  /**
   * Benefits must be a resource or a unit.
   */
  protected void addResult(final NamedAttachable obj, final int quantity) {
    if (!(obj instanceof UnitType) && !(obj instanceof Resource)) {
      throw new IllegalArgumentException("results must be units or resources, not:" + obj.getClass().getName());
    }
    m_results.put(obj, quantity);
  }

  public IntegerMap<Resource> getCosts() {
    return m_cost.copy();
  }

  public IntegerMap<NamedAttachable> getResults() {
    return m_results;
  }

  @Override
  public String toString() {
    return "ProductionRule:" + getName();
  }

  public String toStringCosts() {
    final StringBuilder sb = new StringBuilder();
    final Resource pus;
    getData().acquireReadLock();
    try {
      pus = getData().getResourceList().getResource(Constants.PUS);
    } finally {
      getData().releaseReadLock();
    }
    if (m_cost.getInt(pus) != 0) {
      sb.append("; ");
      sb.append(m_cost.getInt(pus));
      sb.append(" ").append(pus.getName());
    }
    for (final Entry<Resource, Integer> entry : m_cost.entrySet()) {
      final Resource r = entry.getKey();
      if (r.equals(pus)) {
        continue;
      }
      final int c = entry.getValue();
      sb.append("; ");
      sb.append(c);
      sb.append(" ").append(r.getName());
    }
    return sb.toString().replaceFirst("; ", "");
  }
}