OptionalExtraTerritoryBordersDrawable.java

package games.strategy.triplea.ui.screen.drawable;

import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Polygon;
import java.awt.Rectangle;
import java.awt.geom.AffineTransform;
import java.util.Iterator;
import java.util.List;

import games.strategy.engine.data.GameData;
import games.strategy.engine.data.Territory;
import games.strategy.triplea.ui.mapdata.MapData;

public class OptionalExtraTerritoryBordersDrawable implements IDrawable {
  private final String m_territoryName;
  private final OptionalExtraBorderLevel m_level;

  public OptionalExtraTerritoryBordersDrawable(final String territoryName, final OptionalExtraBorderLevel level) {
    m_territoryName = territoryName;
    m_level = level;
  }

  @Override
  public void draw(final Rectangle bounds, final GameData data, final Graphics2D graphics, final MapData mapData,
      final AffineTransform unscaled, final AffineTransform scaled) {
    final Territory territory = data.getMap().getTerritory(m_territoryName);
    final List<Polygon> polys = mapData.getPolygons(territory);
    final Iterator<Polygon> iter2 = polys.iterator();
    while (iter2.hasNext()) {
      Polygon polygon = iter2.next();
      // if we dont have to draw, dont
      if (!polygon.intersects(bounds) && !polygon.contains(bounds)) {
        continue;
      }
      // use a copy since we will move the polygon
      polygon = new Polygon(polygon.xpoints, polygon.ypoints, polygon.npoints);
      polygon.translate(-bounds.x, -bounds.y);
      graphics.setColor(Color.BLACK);
      graphics.drawPolygon(polygon);
    }
  }

  @Override
  public int getLevel() {
    if (m_level == OptionalExtraBorderLevel.HIGH) {
      return OPTIONAL_EXTRA_TERRITORY_BORDERS_HIGH_LEVEL;
    }
    return OPTIONAL_EXTRA_TERRITORY_BORDERS_MEDIUM_LEVEL;
  }
}