org.eclipse.stem.ui.ge/src/org/eclipse/stem/ui/ge/kml/KmlDisplay.java - stem/org.eclipse.stem - Git at Google

 package org.eclipse.stem.ui.ge.kml;

 import java.awt.Rectangle;
 import java.util.Iterator;
 import java.util.Map;

 import org.eclipse.core.runtime.Assert;
 import org.eclipse.stem.ui.ge.Aspect;
 import org.eclipse.stem.ui.ge.GEData;
 import org.eclipse.stem.ui.ge.GELog;
 import org.eclipse.stem.ui.ge.views.GEPreferencePage;
 import org.w3c.dom.Document;
 import org.w3c.dom.Element;

 /*******************************************************************************
  * Copyright (c) 2006 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
  *******************************************************************************/

 /**
  * Standard DISPLAY OF SEIR aspects.
  *
  */
 public class KmlDisplay implements IKmlDisplay   {

 	/**
      * initialization
      *
      * currently we have nothing to do
      *
      *
      */
     public void init() {

     }
     /**
      * Create the KML control file to display all of the
      * polygon (county) areas
      *
      * We will have N PlaceMarks, each of which will
      * be associated with a differnt Style (for color)
      * The polygons will be placed in a placemark based
      * on the color selection.
      *
      * @param controlFileName File where KML will be written
      * @param geData Map of GEData objects
      * @param aspect  which type of map (population,susceptiple...
      * @param cycle current number of cycle being displayed
      */

 	public void generatePolygons(String controlFileName,
 			Map<String,GEData> geData, Aspect aspect) {
         int written = 0;
 		//GELog.debug(this,"aspect="+aspect+" number="+geData.size());

 		KmlDoc kml = new KmlDoc();
 		Document doc = kml.getDocument();
 		Element root = kml.setRoot(null);
 		Element document = kml.appendElement(root,"Document");
 		Style[] styles = setupStyles(kml,document,aspect);
 		int max = aspect.getRange().length;

 		//Setup the placemarks for polygons with same attribute
 		PlaceMark[] placeMark = new PlaceMark[max];
 		Element[] multiGeometry = new Element[max];
 		for (int j=0;j<max;j++) {
 			placeMark[j] = new PlaceMark(doc);
 			document.appendChild(placeMark[j].getElement());
 			placeMark[j].setDescription("STEM map");

 			placeMark[j].setName("STEM"+j);
 			placeMark[j].setStyleUrl(styles[j]);
 			multiGeometry[j] = doc.createElement("MultiGeometry");
 			placeMark[j].appendChild(multiGeometry[j]);
 		}

 		// now we generate a polygon for each county
 //		int debugcounter =0;
 		// Get latest BBox info from GoogleEarth
 		Rectangle bboxView = BBoxInfo.getBBox();
 		Iterator it = geData.entrySet().iterator();
 		boolean bboxFilter = GEPreferencePage.isBBOX();
 		while(it.hasNext()) {
 			int type = 0;
 			Map.Entry e = (Map.Entry)it.next();
 			GEData data = (GEData)e.getValue();

 			String county = data.getName();
 			// check if we are filtering out some areas
 			if (bboxFilter && filter(data,bboxView))  {
 				type = 0;
 			} else {
 				type = selectByValue(aspect,data);
 			}

 //			 DEBUG temp
 //			String locid = loc.getId().toString();
 //			if (county.startsWith("Middlesex")) {
 //				GELog.debug(this,"generatePolygons: "+i+": "+county+":"+locid+
 //						": "+type+": "+coord);
 //			}
 			if (type > 0) {
 				//GELog.debug(this,county);
 				int n = data.getNumAreas();
 				for (int i = 0; i < n; i++) {

 					String coord = data.getCoordinatesAsString(i);
 					//if (i > 0)
 					//   GELog.debug(this,i+": "+county+" "+coord);
 					Polygon polygon = new Polygon(doc);
 					kml.appendText(polygon.getElement(), "name", county);
 					polygon.setCoordinates(coord);
 					multiGeometry[type].appendChild(polygon.getElement());
 				}
 				written++;
 			}

 		}

 		GELog.debug(this,"write "+controlFileName+
 				" with "+written+" entries.");
 		kml.writeFile(controlFileName);
 		//GELog.debug(this,"end write of "+controlFileName);
 		return;
 	}

 	/**
 	 * check if this entry should be filtered out
 	 *
 	 * @param data  Instance that contains info for this admin area
 	 * @param bboxView Screen boundary box from GoogleEarth
 	 * @return true if it should be filtered out.
 	 */
 	private boolean filter(GEData data,Rectangle bboxView) {
 		boolean filtered = false;  // default - not filtered
 		if (bboxView != null ) {
 			Rectangle bbox = data.getBBox();
 			filtered = ! BBoxInfo.testContainment(bbox,bboxView);
 			return filtered;
 		}
 //				if (county.indexOf(" MA ")>=0  ||
 //				    county.indexOf(" MT ")>=0    ) {
 //			        filtered = ! testContainment(bbox,bboxView);
 //			    	GELog.debug(this," bbox:    "+bbox);
 //			    	GELog.debug(this," bboxView:"+bboxView);
 //				}
 //		if (GEPreferencePage.isTest()) {
 //			String county = data.getName();
 //			if (county.indexOf(" US-")>= 0 ) {
 //				filtered = false;
 //
 //			} else if (county.indexOf(" MEX")>= 0 ) {
 //				filtered = false;
 //			} else {
 //				filtered = true;
 //			}
 //
 //			//GELog.debug(this,county+" "+filtered);
 //		}
 		return filtered;
 	}


 //	<Style id="Style1">
 //	  <LineStyle>
 //	     <width>0.3</width>
 //	  </LineStyle>
 //	  <PolyStyle>
 //	    <outline>0</outline>
 //	    <fill>1</fill>
 //	    <color>800000f0</color>
 //	  </PolyStyle>
 //	</Style>
   /**
    * Build the Style statements that will be refereded to
    * by the polygons listed later.
    *
    * @param kml
    * @param document
    * @param aspect
    * @return Style[] array of KML Style instances
    */
   private Style[] setupStyles(KmlDoc kml,Element document, Aspect aspect) {
 	Document doc = kml.getDocument();
 	int max = aspect.getRange().length;
 	// define the Style sections
 	Style[] styles = new Style[max];
 	for (int k=0;k<max;k++) {
 		styles[k] = new Style(doc,"Style"+k);
 		styles[k].setLineStyle(0.3);
 		document.appendChild(styles[k].getElement());
 		/* set the PolyStyle value with color in hex format:
 		 *          oobbggbb
 		 *            00 Opacity (00 to ff)
 		 *            bb Blue (00 to ff)
 		 *            gg green (00 to ff)
 		 *            rr red  (00 to ff)
 		 */
 		int oo = 0;
 		int blue = 0;
 		int red = 0;
 		int green = 0;
 		/**
 		 * Use our preference to outline the borders
 		 */
 		boolean outline = GEPreferencePage.isShowBorder();

 		if (k==0) {
 			// Style0 does not show the polygon
 			styles[0].setPolyStyle("0", outline);
 		} else {
 			blue = aspect.getBlue(k);
 			red = aspect.getRed(k);
 			green = aspect.getGreen(k);
 			oo = aspect.getOpacity(k);

 			String hex = StemKml.getHex(oo)+
 			             StemKml.getHex(blue)+
 			             StemKml.getHex(green)+
 			             StemKml.getHex(red);
 			styles[k].setPolyStyle(hex,outline);
 		}
 	}
 	return styles;
   }


   /**
    * Select the color to be used by the GoogleEarth Polygon
    * based on the relative value for the property
    *
    * @param aspect  enum value specifing which type of
    *                display is shown (population, susceptible,
    *                exposed,infectious,recovered)
    * @param geData
    *
    * @return  value 0 to MAX that determines which grouping
    *   of polygons this area will be displayed with.
    */

   public int selectByValue(Aspect aspect, GEData geData) {

   	int type = 0;

   	try {
   		String code = aspect.getCode();
   		Assert.isNotNull(code, "Aspect code is null");
   		if (code == null)
   			return 0;
   		Map<String,Double> map = geData.getValueMap();
   		Assert.isNotNull(map, "Aspect map is null");
   		double value = 0.0;
   		if (map != null) {
   			Double d = map.get(code);
   			if (d != null)
   			    value = d;
   		} else {
   			return 0;
   		}

   		double[] range = aspect.getRange();
   		for (int i=0; i<range.length; i++) {
   			if (value <= range[i]) {
   				type = i;
   				break;
   			}
   		}


   	} catch (RuntimeException e1) {

   		GELog.error("selectByValue",e1);
   	}
   	return type;
   }

 }
	package org.eclipse.stem.ui.ge.kml;

	import java.awt.Rectangle;
	import java.util.Iterator;
	import java.util.Map;

	import org.eclipse.core.runtime.Assert;
	import org.eclipse.stem.ui.ge.Aspect;
	import org.eclipse.stem.ui.ge.GEData;
	import org.eclipse.stem.ui.ge.GELog;
	import org.eclipse.stem.ui.ge.views.GEPreferencePage;
	import org.w3c.dom.Document;
	import org.w3c.dom.Element;

	/*******************************************************************************
	* Copyright (c) 2006 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
	*******************************************************************************/

	/**
	* Standard DISPLAY OF SEIR aspects.
	*
	*/
	public class KmlDisplay implements IKmlDisplay {

	/**
	* initialization
	*
	* currently we have nothing to do
	*
	*
	*/
	public void init() {

	}
	/**
	* Create the KML control file to display all of the
	* polygon (county) areas
	*
	* We will have N PlaceMarks, each of which will
	* be associated with a differnt Style (for color)
	* The polygons will be placed in a placemark based
	* on the color selection.
	*
	* @param controlFileName File where KML will be written
	* @param geData Map of GEData objects
	* @param aspect which type of map (population,susceptiple...
	* @param cycle current number of cycle being displayed
	*/

	public void generatePolygons(String controlFileName,
	Map<String,GEData> geData, Aspect aspect) {
	int written = 0;
	//GELog.debug(this,"aspect="+aspect+" number="+geData.size());

	KmlDoc kml = new KmlDoc();
	Document doc = kml.getDocument();
	Element root = kml.setRoot(null);
	Element document = kml.appendElement(root,"Document");
	Style[] styles = setupStyles(kml,document,aspect);
	int max = aspect.getRange().length;

	//Setup the placemarks for polygons with same attribute
	PlaceMark[] placeMark = new PlaceMark[max];
	Element[] multiGeometry = new Element[max];
	for (int j=0;j<max;j++) {
	placeMark[j] = new PlaceMark(doc);
	document.appendChild(placeMark[j].getElement());
	placeMark[j].setDescription("STEM map");

	placeMark[j].setName("STEM"+j);
	placeMark[j].setStyleUrl(styles[j]);
	multiGeometry[j] = doc.createElement("MultiGeometry");
	placeMark[j].appendChild(multiGeometry[j]);
	}

	// now we generate a polygon for each county
	// int debugcounter =0;
	// Get latest BBox info from GoogleEarth
	Rectangle bboxView = BBoxInfo.getBBox();
	Iterator it = geData.entrySet().iterator();
	boolean bboxFilter = GEPreferencePage.isBBOX();
	while(it.hasNext()) {
	int type = 0;
	Map.Entry e = (Map.Entry)it.next();
	GEData data = (GEData)e.getValue();

	String county = data.getName();
	// check if we are filtering out some areas
	if (bboxFilter && filter(data,bboxView)) {
	type = 0;
	} else {
	type = selectByValue(aspect,data);
	}

	// DEBUG temp
	// String locid = loc.getId().toString();
	// if (county.startsWith("Middlesex")) {
	// GELog.debug(this,"generatePolygons: "+i+": "+county+":"+locid+
	// ": "+type+": "+coord);
	// }
	if (type > 0) {
	//GELog.debug(this,county);
	int n = data.getNumAreas();
	for (int i = 0; i < n; i++) {

	String coord = data.getCoordinatesAsString(i);
	//if (i > 0)
	// GELog.debug(this,i+": "+county+" "+coord);
	Polygon polygon = new Polygon(doc);
	kml.appendText(polygon.getElement(), "name", county);
	polygon.setCoordinates(coord);
	multiGeometry[type].appendChild(polygon.getElement());
	}
	written++;
	}

	}

	GELog.debug(this,"write "+controlFileName+
	" with "+written+" entries.");
	kml.writeFile(controlFileName);
	//GELog.debug(this,"end write of "+controlFileName);
	return;
	}

	/**
	* check if this entry should be filtered out
	*
	* @param data Instance that contains info for this admin area
	* @param bboxView Screen boundary box from GoogleEarth
	* @return true if it should be filtered out.
	*/
	private boolean filter(GEData data,Rectangle bboxView) {
	boolean filtered = false; // default - not filtered
	if (bboxView != null ) {
	Rectangle bbox = data.getBBox();
	filtered = ! BBoxInfo.testContainment(bbox,bboxView);
	return filtered;
	}
	// if (county.indexOf(" MA ")>=0 \|\|
	// county.indexOf(" MT ")>=0 ) {
	// filtered = ! testContainment(bbox,bboxView);
	// GELog.debug(this," bbox: "+bbox);
	// GELog.debug(this," bboxView:"+bboxView);
	// }
	// if (GEPreferencePage.isTest()) {
	// String county = data.getName();
	// if (county.indexOf(" US-")>= 0 ) {
	// filtered = false;
	//
	// } else if (county.indexOf(" MEX")>= 0 ) {
	// filtered = false;
	// } else {
	// filtered = true;
	// }
	//
	// //GELog.debug(this,county+" "+filtered);
	// }
	return filtered;
	}


	// <Style id="Style1">
	// <LineStyle>
	// <width>0.3</width>
	// </LineStyle>
	// <PolyStyle>
	// <outline>0</outline>
	// <fill>1</fill>
	// <color>800000f0</color>
	// </PolyStyle>
	// </Style>
	/**
	* Build the Style statements that will be refereded to
	* by the polygons listed later.
	*
	* @param kml
	* @param document
	* @param aspect
	* @return Style[] array of KML Style instances
	*/
	private Style[] setupStyles(KmlDoc kml,Element document, Aspect aspect) {
	Document doc = kml.getDocument();
	int max = aspect.getRange().length;
	// define the Style sections
	Style[] styles = new Style[max];
	for (int k=0;k<max;k++) {
	styles[k] = new Style(doc,"Style"+k);
	styles[k].setLineStyle(0.3);
	document.appendChild(styles[k].getElement());
	/* set the PolyStyle value with color in hex format:
	* oobbggbb
	* 00 Opacity (00 to ff)
	* bb Blue (00 to ff)
	* gg green (00 to ff)
	* rr red (00 to ff)
	*/
	int oo = 0;
	int blue = 0;
	int red = 0;
	int green = 0;
	/**
	* Use our preference to outline the borders
	*/
	boolean outline = GEPreferencePage.isShowBorder();

	if (k==0) {
	// Style0 does not show the polygon
	styles[0].setPolyStyle("0", outline);
	} else {
	blue = aspect.getBlue(k);
	red = aspect.getRed(k);
	green = aspect.getGreen(k);
	oo = aspect.getOpacity(k);

	String hex = StemKml.getHex(oo)+
	StemKml.getHex(blue)+
	StemKml.getHex(green)+
	StemKml.getHex(red);
	styles[k].setPolyStyle(hex,outline);
	}
	}
	return styles;
	}


	/**
	* Select the color to be used by the GoogleEarth Polygon
	* based on the relative value for the property
	*
	* @param aspect enum value specifing which type of
	* display is shown (population, susceptible,
	* exposed,infectious,recovered)
	* @param geData
	*
	* @return value 0 to MAX that determines which grouping
	* of polygons this area will be displayed with.
	*/

	public int selectByValue(Aspect aspect, GEData geData) {

	int type = 0;

	try {
	String code = aspect.getCode();
	Assert.isNotNull(code, "Aspect code is null");
	if (code == null)
	return 0;
	Map<String,Double> map = geData.getValueMap();
	Assert.isNotNull(map, "Aspect map is null");
	double value = 0.0;
	if (map != null) {
	Double d = map.get(code);
	if (d != null)
	value = d;
	} else {
	return 0;
	}

	double[] range = aspect.getRange();
	for (int i=0; i<range.length; i++) {
	if (value <= range[i]) {
	type = i;
	break;
	}
	}



	} catch (RuntimeException e1) {

	GELog.error("selectByValue",e1);
	}
	return type;
	}

	}