Receiver.java

package org.visico.utilitydss.server.processsim;


import java.io.File;
import java.util.Calendar;
import java.util.TreeMap;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerConfigurationException;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;

import org.visico.utilitydss.shared.Section;
import org.w3c.dom.Attr;
import org.w3c.dom.Document;
import org.w3c.dom.Element;

import desmoj.core.report.Message;
import desmoj.core.report.MessageReceiver;
import desmoj.core.report.Reporter;

import java.util.Calendar;
import java.util.Date;

import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.ChartUtilities;
import org.jfree.chart.JFreeChart;
import org.jfree.data.category.IntervalCategoryDataset;
import org.jfree.data.gantt.Task;
import org.jfree.data.gantt.TaskSeries;
import org.jfree.data.gantt.TaskSeriesCollection;
import org.jfree.data.time.SimpleTimePeriod;
import org.jfree.ui.ApplicationFrame;
import org.jfree.ui.RefineryUtilities;


public class Receiver implements MessageReceiver
{

	//private DocumentBuilderFactory docFactory;
	private Element tasks;
	private TreeMap<String,Element> sectiontasks;
	private Document doc;
	private TaskSeries s1;
	private Schedule schedule; 
		
	static private int id_counter = 0;
		
	Calendar project_start;
		
	public Receiver() 
	{
		project_start = Calendar.getInstance();
		s1 = new TaskSeries("Scheduled");
		schedule = new Schedule();
		buildBaseXML();
		
	}
	
	private void buildBaseXML() 
	{
		// fstream = new FileWriter("out.txt");
		DocumentBuilderFactory docFactory = DocumentBuilderFactory.newInstance();
		DocumentBuilder docBuilder;
		try 
		{
			docBuilder = docFactory.newDocumentBuilder();
		
		
			// root elements
			doc = docBuilder.newDocument();
			Element rootElement = doc.createElement("project");
			doc.appendChild(rootElement);
			
			//
			 
			// tasks elements
			tasks = doc.createElement("tasks");
			rootElement.appendChild(tasks);
				
			sectiontasks = new TreeMap<String,Element>();
			
			Element taskdisplay = doc.createElement("taskdisplaycolumns");
			rootElement.appendChild(taskdisplay);
			
			Element displayName = doc.createElement("displaycolumn");
			Attr id = doc.createAttribute("property-id");
			id.setValue("tpd3");
			displayName.setAttributeNode(id);
			Attr order = doc.createAttribute("order");
			order.setValue("0");
			displayName.setAttributeNode(order);
			Attr width = doc.createAttribute("width");
			width.setValue("242");
			displayName.setAttributeNode(width);
			Attr visible = doc.createAttribute("visible");
			visible.setValue("true");
			displayName.setAttributeNode(visible);
			taskdisplay.appendChild(displayName);
		} 
		catch (ParserConfigurationException e) 
		{
			e.printStackTrace();
		}
	}
	
	@Override
	public void receive(Message m) 
	{
		
		
		if (m instanceof ActivityMessage)
		{
			// only AM's that match with the selected detail level in myModel are allowed to pass
			if(((ActivityMessage) m).getDetaillevel() == UtilitySimulation.getActivityMsg() || (((ActivityMessage) m).getDetaillevel()-3) == UtilitySimulation.getActivityMsgConnection() 
					|| (((ActivityMessage) m).getDetaillevel()-6) == UtilitySimulation.getActivityMsgPut()  || ((ActivityMessage) m).getDetaillevel() == 0) 
			{
				
				// this function collects the ActivityMessage based messages sent from 
				// the life-cycle of each section and creates a CPM schedule. 
				ActivityMessage am = (ActivityMessage)m;

				Location l = schedule.addNew(am.getSection().getName());
				Calendar end = am.start();
				end.add(Calendar.DAY_OF_MONTH, (int) am.duration());
				
				WorkItem t = new WorkItem(am.work(), am.start().getTimeInMillis(), am.end().getTimeInMillis());
				l.addWorkItem(t);
							
				
				// add XML entry
				// TODO: move to schedule class
				
				Element secTask = sectiontasks.get(am.getSection().getName());
				
				Element work = doc.createElement("task");
				
				Attr id = doc.createAttribute("id");
				id.setValue(Integer.toString(id_counter));
				work.setAttributeNode(id);
				
				Attr name = doc.createAttribute("name");
				name.setValue(am.work());
				work.setAttributeNode(name);
				
				Calendar start = am.start();
				Attr start_att = doc.createAttribute("start");
				start_att.setValue((start.get(Calendar.MONTH) + 1) + "-"
				        + start.get(Calendar.DATE) + "-" + start.get(Calendar.YEAR));
				work.setAttributeNode(start_att);
				
				Attr duration = doc.createAttribute("duration");
				duration.setValue(Double.toString(am.duration()));
				work.setAttributeNode(duration);
				
				secTask.appendChild(work);
				
				id_counter++;
				
				// and add to the chart
				// TODO: move to schedule class
				// Calendar end = am.start();
				// end.add(Calendar.DAY_OF_MONTH, (int) am.duration());
				
				// SimpleTimePeriod p = new SimpleTimePeriod(am.start().getTimeInMillis(), (long) (am.start().getTimeInMillis() + am.duration() * 8.64e7)); - replaced by line below - 
				SimpleTimePeriod p = new SimpleTimePeriod(am.start().getTimeInMillis(), am.end().getTimeInMillis() );
				s1.add(new Task(am.getSection().getName() + " " + am.work(), p));
				}
			
			else{
				//System.out.println("booo" + ((ActivityMessage) m).getDetaillevel() + UtilitySimulation.getActivityMsg() + ((ActivityMessage) m).work());		
				}
				
		
		}
		
	}

	@Override
	public void receive(Reporter arg0) {
		// TODO Auto-generated method stub
		
	}
	
	public void writeXML()
	{
		try 
		{
			TransformerFactory transformerFactory = TransformerFactory.newInstance();
			Transformer transformer;
			
			transformer = transformerFactory.newTransformer();
			
			DOMSource source = new DOMSource(doc);
			StreamResult result = new StreamResult(new File("schedule.xml"));
			transformer.transform(source, result);
		}
		catch (TransformerConfigurationException e) 
		{
				// TODO Auto-generated catch block
				e.printStackTrace();
		} catch (TransformerException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
	
	
	
	public void createSectionElement(ParentProcess section2)
	{
		Element section = doc.createElement("task");
		
		Attr id = doc.createAttribute("id");
		id.setValue(Integer.toString(id_counter+1));
		section.setAttributeNode(id);
		
		Attr name = doc.createAttribute("name");
		name.setValue(section2.getName());
		section.setAttributeNode(name);
		
		tasks.appendChild(section);
		sectiontasks.put(section2.getName(), section);
		id_counter++;
	}
	
	public void exportGantt()
	{
		schedule.createGanttJPGSectionMain();	
	}
	
	private static Date date(final int day, final int month, final int year) {

        final Calendar calendar = Calendar.getInstance();
        calendar.set(year, month, day);
        final Date result = calendar.getTime();
        return result;

    }
	
	 private JFreeChart createChart(final IntervalCategoryDataset dataset) {
	        final JFreeChart chart = ChartFactory.createGanttChart(
	            "Gantt Chart Demo",  // chart title
	            "Task",              // domain axis label
	            "Date",              // range axis label
	            dataset,             // data
	            true,                // include legend
	            true,                // tooltips
	            false                // urls
	        );    
//	        chart.getCategoryPlot().getDomainAxis().setMaxCategoryLabelWidthRatio(10.0f);
	        return chart;    
	    }

	 
}