models/populations/org.eclipse.stem.populationmodels/src/org/eclipse/stem/populationmodels/standard/impl/StandardPopulationInitializerImpl.java - stem/org.eclipse.stem - Git at Google

 package org.eclipse.stem.populationmodels.standard.impl;

 /*******************************************************************************
  * Copyright (c) 2010 IBM Corporation and others.
  * All rights reserved. This program and the accompanying materials
  * are made available under the terms of the Eclipse Public License v1.0
  * which accompanies this distribution, and is available at
  * http://www.eclipse.org/legal/epl-v10.html
  *
  * Contributors:
  *     IBM Corporation - initial API and implementation
  *******************************************************************************/

 import java.util.ArrayList;

 import org.eclipse.emf.common.notify.Notification;
 import org.eclipse.emf.common.util.URI;
 import org.eclipse.emf.ecore.EClass;
 import org.eclipse.emf.ecore.impl.ENotificationImpl;
 import org.eclipse.osgi.util.NLS;
 import org.eclipse.stem.core.Utility;
 import org.eclipse.stem.core.graph.Edge;
 import org.eclipse.stem.core.graph.Graph;
 import org.eclipse.stem.core.graph.Node;
 import org.eclipse.stem.core.graph.NodeLabel;
 import org.eclipse.stem.core.model.Decorator;
 import org.eclipse.stem.core.model.Model;
 import org.eclipse.stem.core.model.STEMTime;
 import org.eclipse.stem.core.scenario.ScenarioInitializationException;
 import org.eclipse.stem.definitions.labels.AreaLabel;
 import org.eclipse.stem.definitions.labels.LabelsFactory;
 import org.eclipse.stem.definitions.labels.PopulationLabel;
 import org.eclipse.stem.definitions.labels.PopulationLabelValue;
 import org.eclipse.stem.definitions.labels.RelativePhysicalRelationshipLabel;
 import org.eclipse.stem.populationmodels.Activator;
 import org.eclipse.stem.populationmodels.standard.StandardPackage;
 import org.eclipse.stem.populationmodels.standard.StandardPopulationInitializer;

 /**
  * <!-- begin-user-doc -->
  * An implementation of the model object '<em><b>Population Initializer</b></em>'.
  * <!-- end-user-doc -->
  * <p>
  * The following features are implemented:
  * <ul>
  *   <li>{@link org.eclipse.stem.populationmodels.standard.impl.StandardPopulationInitializerImpl#getIndividuals <em>Individuals</em>}</li>
  *   <li>{@link org.eclipse.stem.populationmodels.standard.impl.StandardPopulationInitializerImpl#isUseDensity <em>Use Density</em>}</li>
  *   <li>{@link org.eclipse.stem.populationmodels.standard.impl.StandardPopulationInitializerImpl#getTargetISOKey <em>Target ISO Key</em>}</li>
  *   <li>{@link org.eclipse.stem.populationmodels.standard.impl.StandardPopulationInitializerImpl#getTargetURI <em>Target URI</em>}</li>
  * </ul>
  * </p>
  *
  * @generated
  */
 public class StandardPopulationInitializerImpl extends PopulationInitializerImpl implements StandardPopulationInitializer {
 	/**
 	 * The default value of the '{@link #getIndividuals() <em>Individuals</em>}' attribute.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @see #getIndividuals()
 	 * @generated
 	 * @ordered
 	 */
 	protected static final double INDIVIDUALS_EDEFAULT = 1.0;

 	/**
 	 * The cached value of the '{@link #getIndividuals() <em>Individuals</em>}' attribute.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @see #getIndividuals()
 	 * @generated
 	 * @ordered
 	 */
 	protected double individuals = INDIVIDUALS_EDEFAULT;

 	/**
 	 * The default value of the '{@link #isUseDensity() <em>Use Density</em>}' attribute.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @see #isUseDensity()
 	 * @generated
 	 * @ordered
 	 */
 	protected static final boolean USE_DENSITY_EDEFAULT = false;

 	/**
 	 * The cached value of the '{@link #isUseDensity() <em>Use Density</em>}' attribute.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @see #isUseDensity()
 	 * @generated
 	 * @ordered
 	 */
 	protected boolean useDensity = USE_DENSITY_EDEFAULT;

 	/**
 	 * The default value of the '{@link #getTargetISOKey() <em>Target ISO Key</em>}' attribute.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @see #getTargetISOKey()
 	 * @generated
 	 * @ordered
 	 */
 	protected static final String TARGET_ISO_KEY_EDEFAULT = null;

 	/**
 	 * The cached value of the '{@link #getTargetISOKey() <em>Target ISO Key</em>}' attribute.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @see #getTargetISOKey()
 	 * @generated
 	 * @ordered
 	 */
 	protected String targetISOKey = TARGET_ISO_KEY_EDEFAULT;

 	/**
 	 * The default value of the '{@link #getTargetURI() <em>Target URI</em>}' attribute.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @see #getTargetURI()
 	 * @generated
 	 * @ordered
 	 */
 	protected static final URI TARGET_URI_EDEFAULT = null;

 	/**
 	 * The cached value of the '{@link #getTargetURI() <em>Target URI</em>}' attribute.
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @see #getTargetURI()
 	 * @generated
 	 * @ordered
 	 */
 	protected URI targetURI = TARGET_URI_EDEFAULT;

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated NOT
 	 */
 	public StandardPopulationInitializerImpl() {
 		super();
 		// Populate the population labels

 	}

 	@Override
 	public void prepare(Model model, STEMTime time) {
 		// Nothing to do
 	}

 	@Override
 	public void decorateGraph(STEMTime time)  throws ScenarioInitializationException {
 		if(this.isGraphDecorated()) return;
 		checkPopulationModels();
 		ArrayList<Node>nodes = new ArrayList<Node>();
 		ArrayList<Node>negativeNodes = new ArrayList<Node>();

 		this.getNodes(this.getGraph(), this.getTargetISOKey(), nodes, negativeNodes);

 		if(nodes.size() == 0)
 			throw new ScenarioInitializationException(NLS.bind(Messages.StandardPopulationInitializerImpl_ERR_POP_INIT_NODE_NOT_FOUND, new Object[] {this.getURI().toString(), this.getTargetISOKey()}), this, new Exception());

 		for(Node n:nodes) {
 			// Check for existing label
 			PopulationLabel existingLabel = null;
 			for(NodeLabel lab:n.getLabels()) {
 				if(lab instanceof PopulationLabel
 						&& ((PopulationLabel)lab).getPopulationIdentifier().equals(this.getPopulationIdentifier())) {
 					existingLabel = (PopulationLabel)lab;
 				}
 			}
 			if(existingLabel != null) {
 					// Initialize with new value
 					initializeLabel((PopulationLabel)existingLabel, time, false);
 			} else {
 				// Create a new label
 				PopulationLabel newLabel = LabelsFactory.eINSTANCE.createPopulationLabel();
 				newLabel.setPopulationIdentifier(this.getPopulationIdentifier());
 				newLabel.setURIOfIdentifiableToBeLabeled(n.getURI());
 				newLabel.setNode(n);
 				Graph g = this.getGraph();
 				URI newURI = createPopulationLabelURI(n, time);
 				g.getNodeLabels().put(newURI, newLabel);
 				initializeLabel((PopulationLabel)newLabel, time, false);
 			}
 		}

 		// Now do the rest of the nodes in the graph

 		for(Node n:negativeNodes) {
 			// Check for existing label
 			PopulationLabel existingLabel = null;
 			for(NodeLabel lab:n.getLabels()) {
 				if(lab instanceof PopulationLabel
 						&& ((PopulationLabel)lab).getPopulationIdentifier().equals(this.getPopulationIdentifier())) {
 					existingLabel = (PopulationLabel)lab;
 				}
 			}
 			if(existingLabel == null) {
 				// Create a new label. We don't override the existing label
 				PopulationLabel newLabel = LabelsFactory.eINSTANCE.createPopulationLabel();
 				newLabel.setPopulationIdentifier(this.getPopulationIdentifier());
 				newLabel.setURIOfIdentifiableToBeLabeled(n.getURI());
 				newLabel.setNode(n);
 				Graph g = this.getGraph();
 				URI newURI = createPopulationLabelURI(n, time);
 				g.getNodeLabels().put(newURI, newLabel);
 				initializeLabel((PopulationLabel)newLabel, time, true);
 			}
 		}
 		return;
 	}

 	protected void getNodes(Graph g, String key, ArrayList<Node>list, ArrayList<Node>negList) {

 		// First find other population initializers in the graph that has the same
 		// population identifier but a lower level key. Any node in the substree
 		// of that lower level key should not be touched.

 		ArrayList<StandardPopulationInitializer>lowerLevelInitializers = new ArrayList<StandardPopulationInitializer>();

 		for(Decorator d:g.getDecorators()) {
 			if(d instanceof StandardPopulationInitializer &&
 					((StandardPopulationInitializer)d).getPopulationIdentifier().equals(this.getPopulationIdentifier()) &&
 					Utility.keyLevel(((StandardPopulationInitializer)d).getTargetISOKey()) > Utility.keyLevel(key))
 				lowerLevelInitializers.add((StandardPopulationInitializer)d);
 		}

 		for(Node n:g.getNodes().values()) {
 			boolean foundSubInitializer = false;
 			for(StandardPopulationInitializer spi:lowerLevelInitializers)
 				if(isSelfOrHasParent(n, spi.getTargetISOKey()))
 					{foundSubInitializer = true;break;}
 			if(foundSubInitializer) continue;

 			if((key == null || key.trim().equals("")) && isLeaf(n)) list.add(n); //$NON-NLS-1$
 			else if(n.getURI().lastSegment().equals(key) && isLeaf(n)) list.add(n);
 			// Check if any of the parents is the key
 			else if(hasParent(n, key) && isLeaf(n)) list.add(n);
 			else if(isLeaf(n)) negList.add(n);
 		}
 	}

 	public boolean isLeaf(Node n) {
 		for(Edge e:n.getEdges())
 			if(e.getLabel() instanceof RelativePhysicalRelationshipLabel &&
 					e.getA().equals(n))
 				return false;
 		return true;
 	}

 	public boolean hasParent(Node n, String key) {
 		for(Edge e:n.getEdges()) {
 			// Make sure it's a physical containment edge
 			boolean phys = e.getLabel() instanceof RelativePhysicalRelationshipLabel;
 			if(!phys)
 				continue;
 			if(e.getA().getURI().lastSegment().equals(key)) return true;
 			else if(Utility.keyLevel(e.getA().getURI().lastSegment()) < Utility.keyLevel(n.getURI().lastSegment()))
 				return hasParent(e.getA(), key);
 		}
 		return false;
 	}

 	public boolean isSelfOrHasParent(Node n, String key) {
 		if(n.getURI().lastSegment().equals(key)) return true;
 		return hasParent(n, key);
 	}


 	/**
 	 *
 	 */
 	@SuppressWarnings("nls")
 	@Override
 	protected void initializeLabel(PopulationLabel lab, STEMTime time, boolean zeroValue) throws ScenarioInitializationException {
 		if(this.isUseDensity()) {
 			Node n = lab.getNode();
 			if(n == null) {
 				throw new ScenarioInitializationException(NLS.bind(Messages.StandardPopulationInitializerImpl_CANNOT_INIT_POPULATION, new Object[] {getPopulationIdentifier(), lab.toString()}), this, new Exception());
 			}
 			double area = 0.0;
 			for(NodeLabel l:n.getLabels()) {
 				if(l instanceof AreaLabel) {
 					area = ((AreaLabel)l).getCurrentAreaValue().getArea();
 					break;
 				}
 			}
 			if(area == 0.0) {
 				 if(!n.getURI().toString().contains("/transport/pipe")) // No area for transport system nodes is expected. //$NON-NLS-1$
 					 	throw new ScenarioInitializationException(NLS.bind(Messages.StandardPopulationInitializerImpl_UNABLE_TO_FIND_AREA, new Object[] {n.getURI().toString(), this.getURI().toString()}),this, new Exception());
 				 PopulationLabelValue plv = lab.getCurrentPopulationValue();
 				 if(!zeroValue) plv.setCount(this.getIndividuals());
 				 else plv.setCount(0.0);
 			} else {
 				 PopulationLabelValue plv = lab.getCurrentPopulationValue();
 				 if(!zeroValue) plv.setCount(this.getIndividuals()*area);
 				 else plv.setCount(0.0);
 			}
 		} else {
 			 PopulationLabelValue plv = lab.getCurrentPopulationValue();
 			 if(!zeroValue) plv.setCount(this.getIndividuals());
 			 else plv.setCount(0.0);
 		}
 		// Set the valid year to the start year of the sequencer

 		lab.setValidYear(getYear(time));
 	}


 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	@Override
 	protected EClass eStaticClass() {
 		return StandardPackage.Literals.STANDARD_POPULATION_INITIALIZER;
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	public double getIndividuals() {
 		return individuals;
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	public void setIndividuals(double newIndividuals) {
 		double oldIndividuals = individuals;
 		individuals = newIndividuals;
 		if (eNotificationRequired())
 			eNotify(new ENotificationImpl(this, Notification.SET, StandardPackage.STANDARD_POPULATION_INITIALIZER__INDIVIDUALS, oldIndividuals, individuals));
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	public boolean isUseDensity() {
 		return useDensity;
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	public void setUseDensity(boolean newUseDensity) {
 		boolean oldUseDensity = useDensity;
 		useDensity = newUseDensity;
 		if (eNotificationRequired())
 			eNotify(new ENotificationImpl(this, Notification.SET, StandardPackage.STANDARD_POPULATION_INITIALIZER__USE_DENSITY, oldUseDensity, useDensity));
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	public String getTargetISOKey() {
 		return targetISOKey;
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	public void setTargetISOKey(String newTargetISOKey) {
 		String oldTargetISOKey = targetISOKey;
 		targetISOKey = newTargetISOKey;
 		if (eNotificationRequired())
 			eNotify(new ENotificationImpl(this, Notification.SET, StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_ISO_KEY, oldTargetISOKey, targetISOKey));
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	public URI getTargetURI() {
 		return targetURI;
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	public void setTargetURI(URI newTargetURI) {
 		URI oldTargetURI = targetURI;
 		targetURI = newTargetURI;
 		if (eNotificationRequired())
 			eNotify(new ENotificationImpl(this, Notification.SET, StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_URI, oldTargetURI, targetURI));
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	@Override
 	public Object eGet(int featureID, boolean resolve, boolean coreType) {
 		switch (featureID) {
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__INDIVIDUALS:
 				return getIndividuals();
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__USE_DENSITY:
 				return isUseDensity();
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_ISO_KEY:
 				return getTargetISOKey();
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_URI:
 				return getTargetURI();
 		}
 		return super.eGet(featureID, resolve, coreType);
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	@Override
 	public void eSet(int featureID, Object newValue) {
 		switch (featureID) {
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__INDIVIDUALS:
 				setIndividuals((Double)newValue);
 				return;
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__USE_DENSITY:
 				setUseDensity((Boolean)newValue);
 				return;
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_ISO_KEY:
 				setTargetISOKey((String)newValue);
 				return;
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_URI:
 				setTargetURI((URI)newValue);
 				return;
 		}
 		super.eSet(featureID, newValue);
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	@Override
 	public void eUnset(int featureID) {
 		switch (featureID) {
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__INDIVIDUALS:
 				setIndividuals(INDIVIDUALS_EDEFAULT);
 				return;
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__USE_DENSITY:
 				setUseDensity(USE_DENSITY_EDEFAULT);
 				return;
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_ISO_KEY:
 				setTargetISOKey(TARGET_ISO_KEY_EDEFAULT);
 				return;
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_URI:
 				setTargetURI(TARGET_URI_EDEFAULT);
 				return;
 		}
 		super.eUnset(featureID);
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	@Override
 	public boolean eIsSet(int featureID) {
 		switch (featureID) {
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__INDIVIDUALS:
 				return individuals != INDIVIDUALS_EDEFAULT;
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__USE_DENSITY:
 				return useDensity != USE_DENSITY_EDEFAULT;
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_ISO_KEY:
 				return TARGET_ISO_KEY_EDEFAULT == null ? targetISOKey != null : !TARGET_ISO_KEY_EDEFAULT.equals(targetISOKey);
 			case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_URI:
 				return TARGET_URI_EDEFAULT == null ? targetURI != null : !TARGET_URI_EDEFAULT.equals(targetURI);
 		}
 		return super.eIsSet(featureID);
 	}

 	/**
 	 * <!-- begin-user-doc -->
 	 * <!-- end-user-doc -->
 	 * @generated
 	 */
 	@Override
 	public String toString() {
 		if (eIsProxy()) return super.toString();

 		StringBuffer result = new StringBuffer(super.toString());
 		result.append(" (individuals: "); //$NON-NLS-1$
 		result.append(individuals);
 		result.append(", useDensity: "); //$NON-NLS-1$
 		result.append(useDensity);
 		result.append(", targetISOKey: "); //$NON-NLS-1$
 		result.append(targetISOKey);
 		result.append(", targetURI: "); //$NON-NLS-1$
 		result.append(targetURI);
 		result.append(')');
 		return result.toString();
 	}

 } //StandardPopulationInitializerImpl
	package org.eclipse.stem.populationmodels.standard.impl;

	/*******************************************************************************
	* Copyright (c) 2010 IBM Corporation and others.
	* All rights reserved. This program and the accompanying materials
	* are made available under the terms of the Eclipse Public License v1.0
	* which accompanies this distribution, and is available at
	* http://www.eclipse.org/legal/epl-v10.html
	*
	* Contributors:
	* IBM Corporation - initial API and implementation
	*******************************************************************************/

	import java.util.ArrayList;

	import org.eclipse.emf.common.notify.Notification;
	import org.eclipse.emf.common.util.URI;
	import org.eclipse.emf.ecore.EClass;
	import org.eclipse.emf.ecore.impl.ENotificationImpl;
	import org.eclipse.osgi.util.NLS;
	import org.eclipse.stem.core.Utility;
	import org.eclipse.stem.core.graph.Edge;
	import org.eclipse.stem.core.graph.Graph;
	import org.eclipse.stem.core.graph.Node;
	import org.eclipse.stem.core.graph.NodeLabel;
	import org.eclipse.stem.core.model.Decorator;
	import org.eclipse.stem.core.model.Model;
	import org.eclipse.stem.core.model.STEMTime;
	import org.eclipse.stem.core.scenario.ScenarioInitializationException;
	import org.eclipse.stem.definitions.labels.AreaLabel;
	import org.eclipse.stem.definitions.labels.LabelsFactory;
	import org.eclipse.stem.definitions.labels.PopulationLabel;
	import org.eclipse.stem.definitions.labels.PopulationLabelValue;
	import org.eclipse.stem.definitions.labels.RelativePhysicalRelationshipLabel;
	import org.eclipse.stem.populationmodels.Activator;
	import org.eclipse.stem.populationmodels.standard.StandardPackage;
	import org.eclipse.stem.populationmodels.standard.StandardPopulationInitializer;

	/**
	* <!-- begin-user-doc -->
	* An implementation of the model object '<em><b>Population Initializer</b></em>'.
	* <!-- end-user-doc -->
	* <p>
	* The following features are implemented:
	* <ul>
	* <li>{@link org.eclipse.stem.populationmodels.standard.impl.StandardPopulationInitializerImpl#getIndividuals <em>Individuals</em>}</li>
	* <li>{@link org.eclipse.stem.populationmodels.standard.impl.StandardPopulationInitializerImpl#isUseDensity <em>Use Density</em>}</li>
	* <li>{@link org.eclipse.stem.populationmodels.standard.impl.StandardPopulationInitializerImpl#getTargetISOKey <em>Target ISO Key</em>}</li>
	* <li>{@link org.eclipse.stem.populationmodels.standard.impl.StandardPopulationInitializerImpl#getTargetURI <em>Target URI</em>}</li>
	* </ul>
	* </p>
	*
	* @generated
	*/
	public class StandardPopulationInitializerImpl extends PopulationInitializerImpl implements StandardPopulationInitializer {
	/**
	* The default value of the '{@link #getIndividuals() <em>Individuals</em>}' attribute.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @see #getIndividuals()
	* @generated
	* @ordered
	*/
	protected static final double INDIVIDUALS_EDEFAULT = 1.0;

	/**
	* The cached value of the '{@link #getIndividuals() <em>Individuals</em>}' attribute.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @see #getIndividuals()
	* @generated
	* @ordered
	*/
	protected double individuals = INDIVIDUALS_EDEFAULT;

	/**
	* The default value of the '{@link #isUseDensity() <em>Use Density</em>}' attribute.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @see #isUseDensity()
	* @generated
	* @ordered
	*/
	protected static final boolean USE_DENSITY_EDEFAULT = false;

	/**
	* The cached value of the '{@link #isUseDensity() <em>Use Density</em>}' attribute.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @see #isUseDensity()
	* @generated
	* @ordered
	*/
	protected boolean useDensity = USE_DENSITY_EDEFAULT;

	/**
	* The default value of the '{@link #getTargetISOKey() <em>Target ISO Key</em>}' attribute.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @see #getTargetISOKey()
	* @generated
	* @ordered
	*/
	protected static final String TARGET_ISO_KEY_EDEFAULT = null;

	/**
	* The cached value of the '{@link #getTargetISOKey() <em>Target ISO Key</em>}' attribute.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @see #getTargetISOKey()
	* @generated
	* @ordered
	*/
	protected String targetISOKey = TARGET_ISO_KEY_EDEFAULT;

	/**
	* The default value of the '{@link #getTargetURI() <em>Target URI</em>}' attribute.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @see #getTargetURI()
	* @generated
	* @ordered
	*/
	protected static final URI TARGET_URI_EDEFAULT = null;

	/**
	* The cached value of the '{@link #getTargetURI() <em>Target URI</em>}' attribute.
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @see #getTargetURI()
	* @generated
	* @ordered
	*/
	protected URI targetURI = TARGET_URI_EDEFAULT;

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated NOT
	*/
	public StandardPopulationInitializerImpl() {
	super();
	// Populate the population labels

	}

	@Override
	public void prepare(Model model, STEMTime time) {
	// Nothing to do
	}

	@Override
	public void decorateGraph(STEMTime time) throws ScenarioInitializationException {
	if(this.isGraphDecorated()) return;
	checkPopulationModels();
	ArrayList<Node>nodes = new ArrayList<Node>();
	ArrayList<Node>negativeNodes = new ArrayList<Node>();

	this.getNodes(this.getGraph(), this.getTargetISOKey(), nodes, negativeNodes);

	if(nodes.size() == 0)
	throw new ScenarioInitializationException(NLS.bind(Messages.StandardPopulationInitializerImpl_ERR_POP_INIT_NODE_NOT_FOUND, new Object[] {this.getURI().toString(), this.getTargetISOKey()}), this, new Exception());

	for(Node n:nodes) {
	// Check for existing label
	PopulationLabel existingLabel = null;
	for(NodeLabel lab:n.getLabels()) {
	if(lab instanceof PopulationLabel
	&& ((PopulationLabel)lab).getPopulationIdentifier().equals(this.getPopulationIdentifier())) {
	existingLabel = (PopulationLabel)lab;
	}
	}
	if(existingLabel != null) {
	// Initialize with new value
	initializeLabel((PopulationLabel)existingLabel, time, false);
	} else {
	// Create a new label
	PopulationLabel newLabel = LabelsFactory.eINSTANCE.createPopulationLabel();
	newLabel.setPopulationIdentifier(this.getPopulationIdentifier());
	newLabel.setURIOfIdentifiableToBeLabeled(n.getURI());
	newLabel.setNode(n);
	Graph g = this.getGraph();
	URI newURI = createPopulationLabelURI(n, time);
	g.getNodeLabels().put(newURI, newLabel);
	initializeLabel((PopulationLabel)newLabel, time, false);
	}
	}

	// Now do the rest of the nodes in the graph

	for(Node n:negativeNodes) {
	// Check for existing label
	PopulationLabel existingLabel = null;
	for(NodeLabel lab:n.getLabels()) {
	if(lab instanceof PopulationLabel
	&& ((PopulationLabel)lab).getPopulationIdentifier().equals(this.getPopulationIdentifier())) {
	existingLabel = (PopulationLabel)lab;
	}
	}
	if(existingLabel == null) {
	// Create a new label. We don't override the existing label
	PopulationLabel newLabel = LabelsFactory.eINSTANCE.createPopulationLabel();
	newLabel.setPopulationIdentifier(this.getPopulationIdentifier());
	newLabel.setURIOfIdentifiableToBeLabeled(n.getURI());
	newLabel.setNode(n);
	Graph g = this.getGraph();
	URI newURI = createPopulationLabelURI(n, time);
	g.getNodeLabels().put(newURI, newLabel);
	initializeLabel((PopulationLabel)newLabel, time, true);
	}
	}
	return;
	}

	protected void getNodes(Graph g, String key, ArrayList<Node>list, ArrayList<Node>negList) {

	// First find other population initializers in the graph that has the same
	// population identifier but a lower level key. Any node in the substree
	// of that lower level key should not be touched.

	ArrayList<StandardPopulationInitializer>lowerLevelInitializers = new ArrayList<StandardPopulationInitializer>();

	for(Decorator d:g.getDecorators()) {
	if(d instanceof StandardPopulationInitializer &&
	((StandardPopulationInitializer)d).getPopulationIdentifier().equals(this.getPopulationIdentifier()) &&
	Utility.keyLevel(((StandardPopulationInitializer)d).getTargetISOKey()) > Utility.keyLevel(key))
	lowerLevelInitializers.add((StandardPopulationInitializer)d);
	}

	for(Node n:g.getNodes().values()) {
	boolean foundSubInitializer = false;
	for(StandardPopulationInitializer spi:lowerLevelInitializers)
	if(isSelfOrHasParent(n, spi.getTargetISOKey()))
	{foundSubInitializer = true;break;}
	if(foundSubInitializer) continue;

	if((key == null \|\| key.trim().equals("")) && isLeaf(n)) list.add(n); //$NON-NLS-1$
	else if(n.getURI().lastSegment().equals(key) && isLeaf(n)) list.add(n);
	// Check if any of the parents is the key
	else if(hasParent(n, key) && isLeaf(n)) list.add(n);
	else if(isLeaf(n)) negList.add(n);
	}
	}

	public boolean isLeaf(Node n) {
	for(Edge e:n.getEdges())
	if(e.getLabel() instanceof RelativePhysicalRelationshipLabel &&
	e.getA().equals(n))
	return false;
	return true;
	}

	public boolean hasParent(Node n, String key) {
	for(Edge e:n.getEdges()) {
	// Make sure it's a physical containment edge
	boolean phys = e.getLabel() instanceof RelativePhysicalRelationshipLabel;
	if(!phys)
	continue;
	if(e.getA().getURI().lastSegment().equals(key)) return true;
	else if(Utility.keyLevel(e.getA().getURI().lastSegment()) < Utility.keyLevel(n.getURI().lastSegment()))
	return hasParent(e.getA(), key);
	}
	return false;
	}

	public boolean isSelfOrHasParent(Node n, String key) {
	if(n.getURI().lastSegment().equals(key)) return true;
	return hasParent(n, key);
	}


	/**
	*
	*/
	@SuppressWarnings("nls")
	@Override
	protected void initializeLabel(PopulationLabel lab, STEMTime time, boolean zeroValue) throws ScenarioInitializationException {
	if(this.isUseDensity()) {
	Node n = lab.getNode();
	if(n == null) {
	throw new ScenarioInitializationException(NLS.bind(Messages.StandardPopulationInitializerImpl_CANNOT_INIT_POPULATION, new Object[] {getPopulationIdentifier(), lab.toString()}), this, new Exception());
	}
	double area = 0.0;
	for(NodeLabel l:n.getLabels()) {
	if(l instanceof AreaLabel) {
	area = ((AreaLabel)l).getCurrentAreaValue().getArea();
	break;
	}
	}
	if(area == 0.0) {
	if(!n.getURI().toString().contains("/transport/pipe")) // No area for transport system nodes is expected. //$NON-NLS-1$
	throw new ScenarioInitializationException(NLS.bind(Messages.StandardPopulationInitializerImpl_UNABLE_TO_FIND_AREA, new Object[] {n.getURI().toString(), this.getURI().toString()}),this, new Exception());
	PopulationLabelValue plv = lab.getCurrentPopulationValue();
	if(!zeroValue) plv.setCount(this.getIndividuals());
	else plv.setCount(0.0);
	} else {
	PopulationLabelValue plv = lab.getCurrentPopulationValue();
	if(!zeroValue) plv.setCount(this.getIndividuals()*area);
	else plv.setCount(0.0);
	}
	} else {
	PopulationLabelValue plv = lab.getCurrentPopulationValue();
	if(!zeroValue) plv.setCount(this.getIndividuals());
	else plv.setCount(0.0);
	}
	// Set the valid year to the start year of the sequencer

	lab.setValidYear(getYear(time));
	}


	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	@Override
	protected EClass eStaticClass() {
	return StandardPackage.Literals.STANDARD_POPULATION_INITIALIZER;
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	public double getIndividuals() {
	return individuals;
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	public void setIndividuals(double newIndividuals) {
	double oldIndividuals = individuals;
	individuals = newIndividuals;
	if (eNotificationRequired())
	eNotify(new ENotificationImpl(this, Notification.SET, StandardPackage.STANDARD_POPULATION_INITIALIZER__INDIVIDUALS, oldIndividuals, individuals));
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	public boolean isUseDensity() {
	return useDensity;
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	public void setUseDensity(boolean newUseDensity) {
	boolean oldUseDensity = useDensity;
	useDensity = newUseDensity;
	if (eNotificationRequired())
	eNotify(new ENotificationImpl(this, Notification.SET, StandardPackage.STANDARD_POPULATION_INITIALIZER__USE_DENSITY, oldUseDensity, useDensity));
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	public String getTargetISOKey() {
	return targetISOKey;
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	public void setTargetISOKey(String newTargetISOKey) {
	String oldTargetISOKey = targetISOKey;
	targetISOKey = newTargetISOKey;
	if (eNotificationRequired())
	eNotify(new ENotificationImpl(this, Notification.SET, StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_ISO_KEY, oldTargetISOKey, targetISOKey));
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	public URI getTargetURI() {
	return targetURI;
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	public void setTargetURI(URI newTargetURI) {
	URI oldTargetURI = targetURI;
	targetURI = newTargetURI;
	if (eNotificationRequired())
	eNotify(new ENotificationImpl(this, Notification.SET, StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_URI, oldTargetURI, targetURI));
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	@Override
	public Object eGet(int featureID, boolean resolve, boolean coreType) {
	switch (featureID) {
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__INDIVIDUALS:
	return getIndividuals();
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__USE_DENSITY:
	return isUseDensity();
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_ISO_KEY:
	return getTargetISOKey();
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_URI:
	return getTargetURI();
	}
	return super.eGet(featureID, resolve, coreType);
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	@Override
	public void eSet(int featureID, Object newValue) {
	switch (featureID) {
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__INDIVIDUALS:
	setIndividuals((Double)newValue);
	return;
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__USE_DENSITY:
	setUseDensity((Boolean)newValue);
	return;
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_ISO_KEY:
	setTargetISOKey((String)newValue);
	return;
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_URI:
	setTargetURI((URI)newValue);
	return;
	}
	super.eSet(featureID, newValue);
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	@Override
	public void eUnset(int featureID) {
	switch (featureID) {
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__INDIVIDUALS:
	setIndividuals(INDIVIDUALS_EDEFAULT);
	return;
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__USE_DENSITY:
	setUseDensity(USE_DENSITY_EDEFAULT);
	return;
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_ISO_KEY:
	setTargetISOKey(TARGET_ISO_KEY_EDEFAULT);
	return;
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_URI:
	setTargetURI(TARGET_URI_EDEFAULT);
	return;
	}
	super.eUnset(featureID);
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	@Override
	public boolean eIsSet(int featureID) {
	switch (featureID) {
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__INDIVIDUALS:
	return individuals != INDIVIDUALS_EDEFAULT;
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__USE_DENSITY:
	return useDensity != USE_DENSITY_EDEFAULT;
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_ISO_KEY:
	return TARGET_ISO_KEY_EDEFAULT == null ? targetISOKey != null : !TARGET_ISO_KEY_EDEFAULT.equals(targetISOKey);
	case StandardPackage.STANDARD_POPULATION_INITIALIZER__TARGET_URI:
	return TARGET_URI_EDEFAULT == null ? targetURI != null : !TARGET_URI_EDEFAULT.equals(targetURI);
	}
	return super.eIsSet(featureID);
	}

	/**
	* <!-- begin-user-doc -->
	* <!-- end-user-doc -->
	* @generated
	*/
	@Override
	public String toString() {
	if (eIsProxy()) return super.toString();

	StringBuffer result = new StringBuffer(super.toString());
	result.append(" (individuals: "); //$NON-NLS-1$
	result.append(individuals);
	result.append(", useDensity: "); //$NON-NLS-1$
	result.append(useDensity);
	result.append(", targetISOKey: "); //$NON-NLS-1$
	result.append(targetISOKey);
	result.append(", targetURI: "); //$NON-NLS-1$
	result.append(targetURI);
	result.append(')');
	return result.toString();
	}

	} //StandardPopulationInitializerImpl