LightSourceIconOverlay.java

/*
 * Copyright (c) 2014 tabletoptool.com team.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/gpl.html
 * 
 * Contributors:
 *     rptools.com team - initial implementation
 *     tabletoptool.com team - further development
 */
package com.t3.client.ui.zone;

import java.awt.Graphics2D;
import java.awt.geom.Area;

import com.t3.client.AppStyle;
import com.t3.model.AttachedLightSource;
import com.t3.model.LightSource;
import com.t3.model.Token;

public class LightSourceIconOverlay implements ZoneOverlay {

	@Override
	public void paintOverlay(ZoneRenderer renderer, Graphics2D g) {

		for (Token token : renderer.getZone().getAllTokens()) {

			if (token.hasLightSources()) {
				boolean foundNormalLight = false;
				for (AttachedLightSource attachedLightSource : token.getLightSources()) {
					LightSource lightSource = attachedLightSource.getLightSource();
					if (lightSource != null && lightSource.getType() == LightSource.Type.NORMAL) {
						foundNormalLight = true;
						break;
					}
				}
				if (!foundNormalLight) {
					continue;
				}
				
				Area area = renderer.getTokenBounds(token);
				if (area == null) {
					continue;
				}

				int x = area.getBounds().x + (area.getBounds().width - AppStyle.lightSourceIcon.getWidth())/2;
				int y = area.getBounds().y + (area.getBounds().height - AppStyle.lightSourceIcon.getHeight())/2;
				g.drawImage(AppStyle.lightSourceIcon, x, y, null);
			}
		}
	}
}