package org.simantics.district.network.profile;

import java.awt.Color;

import org.simantics.Simantics;
import org.simantics.db.ReadGraph;
import org.simantics.db.Resource;
import org.simantics.db.common.procedure.adapter.TransientCacheAsyncListener;
import org.simantics.db.exception.DatabaseException;
import org.simantics.scenegraph.INode;
import org.simantics.scenegraph.g2d.nodes.SingleElementNode;
import org.simantics.scenegraph.profile.EvaluationContext;
import org.simantics.scenegraph.profile.common.ProfileVariables;

/**
 * @author Tuukka Lehtonen
 * @since 1.35.0
 */
public class DNElementColorStyle extends ThrottledStyleBase<Color> {

	private static final boolean DEBUG = false;

	@Override
	public Color calculateThrottledStyle(ReadGraph graph, Resource runtimeDiagram, Resource entry, Resource groupItem) throws DatabaseException {
		DiagramSettings ds = graph.syncRequest(new DiagramSettingsRequest(runtimeDiagram), TransientCacheAsyncListener.instance());
		// Prevent PendingVariableExceptions from coming through
		boolean wasSynchronous = graph.setSynchronous(true);
		try {
			if (ds.elementColoringFunction.isPresent()) {
				if (DEBUG)
					System.out.print("elementColoringFunction: " + ds.elementColoringFunction + "(" + groupItem + "): ");
				Double t = Simantics.applySCLRead(graph, ds.elementColoringFunction.get(), groupItem);
				if (DEBUG)
					System.out.print(t);
				if (t != null) {
					Color result = ds.coloringGradient.get(t);
					//System.out.println("color(" + t + "): " + result);
					return result;
				}
			}
		}
		finally {
			graph.setSynchronous(wasSynchronous);
		}
		return null;
	}

	@Override
	public void applyThrottledStyleForNode(EvaluationContext observer, INode node, Color color) {
		SingleElementNode n = (SingleElementNode) node;
		for (INode nn : n.getNodes())
			ProfileVariables.claimNodeProperty(nn, "dynamicColor", color, observer);
	}

	@Override
	protected void cleanupStyleForNode(EvaluationContext evaluationContext, INode node) {
		SingleElementNode n = (SingleElementNode) node;
		for (INode nn : n.getNodes())
			ProfileVariables.claimNodeProperty(nn, "dynamicColor", null, evaluationContext);
	}

}