models/populations/org.eclipse.stem.populationmodels/src/org/eclipse/stem/populationmodels/standard/impl/StandardPopulationModelLabelImpl.java

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

/*******************************************************************************
 * Copyright (c) 2009 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.Calendar;

import org.eclipse.emf.ecore.EClass;
import org.eclipse.stem.core.graph.Exchange;
import org.eclipse.stem.core.graph.ExchangeType;
import org.eclipse.stem.core.graph.GraphFactory;
import org.eclipse.stem.core.graph.IntegrationLabelValue;
import org.eclipse.stem.core.model.STEMTime;
import org.eclipse.stem.definitions.labels.PopulationLabel;
import org.eclipse.stem.populationmodels.standard.DemographicPopulationModel;
import org.eclipse.stem.populationmodels.standard.PopulationGroup;
import org.eclipse.stem.populationmodels.standard.PopulationModel;
import org.eclipse.stem.populationmodels.standard.StandardFactory;
import org.eclipse.stem.populationmodels.standard.StandardPackage;
import org.eclipse.stem.populationmodels.standard.StandardPopulationModel;
import org.eclipse.stem.populationmodels.standard.StandardPopulationModelLabel;
import org.eclipse.stem.populationmodels.standard.StandardPopulationModelLabelValue;

/**
 * <!-- begin-user-doc -->
 * An implementation of the model object '<em><b>Population Model Label</b></em>'.
 * <!-- end-user-doc -->
 * <p>
 * </p>
 *
 * @generated
 */
public class StandardPopulationModelLabelImpl extends PopulationModelLabelImpl implements StandardPopulationModelLabel {
	/**
	 * <!-- begin-user-doc -->
	 * <!-- end-user-doc -->
	 * @generated NOT
	 */
	protected StandardPopulationModelLabelImpl() {
		super();
		setCurrentValue(StandardFactory.eINSTANCE.createStandardPopulationModelLabelValue());
		setNextValue(StandardFactory.eINSTANCE.createStandardPopulationModelLabelValue());
		setTempValue(StandardFactory.eINSTANCE.createStandardPopulationModelLabelValue());
		setProbeValue(StandardFactory.eINSTANCE.createStandardPopulationModelLabelValue());
		setErrorScale(StandardFactory.eINSTANCE.createStandardPopulationModelLabelValue());
		setDeltaValue(StandardFactory.eINSTANCE.createStandardPopulationModelLabelValue());
	}
	
	/**
	 * Return the identifier for the label, i.e. the population identifier
	 * 
	 * @return String The identifier
	 */
	
	public String getIdentifier() {
		return this.getPopulationIdentifier();
	}

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

	/**
	 * Reset the dynamic label. This sets the {@link LabelValue}s back to their
	 * initial states.
	 * 
	 * @model volatile="true" transient="true" changeable="false"
	 */
	@Override
	public void reset() {
		super.reset();
		((StandardPopulationModelLabelValue)this.getCurrentValue()).reset();
		((StandardPopulationModelLabelValue)this.getNextValue()).reset();
		this.getDeltaValue().reset();
		this.getTempValue().reset();
		this.getProbeValue().reset();
		this.getErrorScale().reset();
		
		// Get the fraction of the population group if DemographicPopulationModel is used
		double fraction = 1.0;
		double initialRescaling = ((PopulationModel)getDecorator()).getInitialRescalingFactor();
		if (getDecorator() instanceof DemographicPopulationModel) {
			DemographicPopulationModel model = (DemographicPopulationModel) getDecorator();
			
			for (PopulationGroup group : model.getPopulationGroups()) {
				if (group.getIdentifier() == getPopulationIdentifier()) {
					fraction = group.getFraction();
				}
			}
		}
		
		double originalCount = this.getPopulationLabel().getCurrentPopulationValue().getCount() * fraction * initialRescaling;
		((StandardPopulationModelLabelValue)this.getCurrentValue()).setCount(originalCount);
		
		// Set the next value too. This is to avoid switching between zero values and non-zero values
		// when the graph updates to the next value for nodes that are not being updated by this instance
		((IntegrationLabelValue)getNextValue()).set((IntegrationLabelValue)getCurrentValue());
		
	}
	
	/**
	 * Reset to the correct population value for the
	 * passed in time
	 * 
	 * @param time Reference time
	 */
	/*public void reset(STEMTime time) {
		// Only rewind population if it's a standard population model
		if(this.getDecorator() instanceof StandardPopulationModel) {
			StandardPopulationModel spm = (StandardPopulationModel)this.getDecorator();
			PopulationLabel plabel = this.getPopulationLabel();
			int year = plabel.getValidYear();
			Calendar c = Calendar.getInstance();
			c.setTime(time.getTime());
			int startYear = c.get(Calendar.YEAR);
			double growth = spm.getBirthRate() - spm.getDeathRate();
			double dailyGrowth = (spm.getTimePeriod() / 86400000.0)*growth;
			double diff = ((double)startYear - (double)year)*365.25; // approx
			double currentPopulation = ((StandardPopulationModelLabelValue)this.getCurrentValue()).getCount();
			dailyGrowth = 1.0 + dailyGrowth;
			double newPopulation = currentPopulation * Math.pow(dailyGrowth, diff);
			((StandardPopulationModelLabelValue)this.getCurrentValue()).setCount(newPopulation);	
			// Store the additions/substractions in the delta value so that disease models can
			// adjust their counts
			double popdiff = newPopulation - currentPopulation;
			if(popdiff > 0.0) {
				Exchange exchange = GraphFactory.eINSTANCE.createExchange();
				
				exchange.setType(ExchangeType.BIRTHS_AND_DEATHS);
				exchange.setCount(popdiff);
				((IntegrationLabelValue)this.getDeltaValue()).getArrivals().add(exchange);
				((StandardPopulationModelLabelValue)this.getDeltaValue()).setBirths(popdiff);
			}
			else {
				Exchange exchange = GraphFactory.eINSTANCE.createExchange();
				
				exchange.setType(ExchangeType.BIRTHS_AND_DEATHS);
				exchange.setCount(-popdiff);
				((IntegrationLabelValue)this.getDeltaValue()).getDepartures().add(exchange);
				((StandardPopulationModelLabelValue)this.getDeltaValue()).setDeaths(-popdiff);
			}

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

		StringBuffer result = new StringBuffer(
				getPopulationIdentifier() == null ? "null" : getPopulationIdentifier());
		result.append(" [");
		result.append(Math.floor(((StandardPopulationModelLabelValue)getCurrentValue()).getCount()));
		result.append("]");
		return result.toString();
	}
} //StandardPopulationModelLabelImpl