kmeansingleclusterscatter.java

一个用于排队系统仿真的开源软件,有非常形象的图象仿真过程!

package jmt.engine.jwat.workloadAnalysis.chart;

import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Point;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.awt.event.MouseMotionListener;
import java.awt.geom.AffineTransform;
import java.awt.image.BufferedImage;
import java.io.File;
import java.lang.reflect.InvocationTargetException;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.util.Date;

import javax.swing.AbstractAction;
import javax.swing.JComponent;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JMenu;
import javax.swing.JMenuItem;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JPopupMenu;

import jmt.engine.jwat.TimeConsumingWorker;
import jmt.engine.jwat.VariableNumber;
import jmt.engine.jwat.workloadAnalysis.clustering.kMean.KMean;
import jmt.engine.jwat.workloadAnalysis.utils.ChangeVariableListener;
import jmt.engine.jwat.workloadAnalysis.utils.JavaWatColor;
import jmt.engine.jwat.workloadAnalysis.utils.ModelWorkloadAnalysis;
import jmt.gui.jwat.input.ProgressMonitorShow;

import org.freehep.graphics2d.VectorGraphics;
import org.freehep.util.export.ExportDialog;

public class KMeanSingleClusterScatter extends JPanel {
	private int HEIGHT_TEXT_X = 30;
	private int WIDTH_TEXT_Y = 30;
	private int HEIGHT_VALUES_X = 90;
	private int WIDTH_VALUES_Y = 90;
	private int TOP_SPACE = 50;
	private int RIGHT_SPACE = 20;
	private int HEIGHT = 500;
	private int WIDTH = 500;
	private int pointSize = 1;
	
	private Point UpperLeftGraph = new Point(WIDTH_TEXT_Y+WIDTH_VALUES_Y,TOP_SPACE);
	private Point UpperRightGraph = new Point(WIDTH_TEXT_Y+WIDTH_VALUES_Y + WIDTH,TOP_SPACE);
	private Point LowerLeftGraph = new Point(WIDTH_TEXT_Y+WIDTH_VALUES_Y,HEIGHT+TOP_SPACE);
	private Point LowerRightGraph = new Point(WIDTH_TEXT_Y+WIDTH_VALUES_Y+WIDTH,HEIGHT+TOP_SPACE);
	private Point YText = new Point(WIDTH_TEXT_Y / 2,(TOP_SPACE + HEIGHT + HEIGHT_TEXT_X + HEIGHT_VALUES_X) / 2);
	private Point XText = new Point((WIDTH_TEXT_Y + WIDTH_VALUES_Y + WIDTH + RIGHT_SPACE) / 2,TOP_SPACE + HEIGHT + HEIGHT_VALUES_X + HEIGHT_TEXT_X / 2 );
	
	private BufferedImage graph;			//Immagine grafico
	private boolean first = true;			//Indica che devono essere ridisegnati i punti nella buffered image
	private boolean zoomming = false;		//Indica che si sta operando una selezione di zoom In
	private int xStart,yStart,xEnd,yEnd;	//Posizione di inizio e fine della selezione di zoom
	private ModelWorkloadAnalysis model;	//Modello contenente i valori da plottare
	
	//private Vector currClustering; 
	private int curClust = 1;
	
	private int xVar;						//Variabile da plottare in x
	private int yVar;						//Variabile da plottare in y
	private double xMin;					//Valori limite correnti del grafico
	private double xMax;
	private double yMin;
	private double yMax;
	private DecimalFormat valueFormat;		//Formato visualizzazione dei valori sugli assi
	private GlassPanel p;
	private JFrame parent;
	private KMeanSingleClusterScatterPopupMenu popup = new KMeanSingleClusterScatterPopupMenu();
	//private int clustering;
	private short[][] c1;
	private int match;
	
	public KMeanSingleClusterScatter(int x,int y,ModelWorkloadAnalysis model,JFrame f,int clustering,int clust,int match){ //Passare il numero del clustering
		super();
		parent = f;
		graph = new BufferedImage(WIDTH+1,HEIGHT+1,BufferedImage.TYPE_INT_RGB);
		valueFormat = new DecimalFormat("####.##E0");
		parent.setResizable(false);
		curClust = clust;
		//this.clustering = clustering;
		this.match = match;
		//currClustering = (Vector)((KMean)(model.getListOfClustering().get(clustering))).getClusteringAssignment().get(curClust);
		c1 = ((KMean)(model.getListOfClustering().get(clustering))).getAsseg();
		xVar = x;
		yVar = y;
		this.model = model;
		this.model.addOnChangeVariableValue(new ChangeVariableListener(){
			public void onChangeVariableValues() {
				first = true;
				xMin = KMeanSingleClusterScatter.this.model.getMatrix().getVariables()[xVar].getUniStats().getMinValue();
				xMax = KMeanSingleClusterScatter.this.model.getMatrix().getVariables()[xVar].getUniStats().getMaxValue();
				yMin = KMeanSingleClusterScatter.this.model.getMatrix().getVariables()[yVar].getUniStats().getMinValue();
				yMax = KMeanSingleClusterScatter.this.model.getMatrix().getVariables()[yVar].getUniStats().getMaxValue();
				repaint();
			}
		});
		
		xMin = this.model.getMatrix().getVariables()[xVar].getUniStats().getMinValue();
		xMax = this.model.getMatrix().getVariables()[xVar].getUniStats().getMaxValue();
		yMin = this.model.getMatrix().getVariables()[yVar].getUniStats().getMinValue();
		yMax = this.model.getMatrix().getVariables()[yVar].getUniStats().getMaxValue();
		
		p = new GlassPanel();
		f.setGlassPane(p);
		p.setVisible(true);
		addMouseListener(p);
		addMouseMotionListener(p);
		
		//Aggiunta menuItem del popup
		for(int i = 1 ; i <= clust+1; i++){
			JMenuItem m = new JMenuItem("Cluster " + (i));
			m.addActionListener(new KMScatterMenuChoose(i-1));
			popup.clusters.add(m);		
		}
	}
	
	private class KMScatterMenuChoose implements ActionListener{
		private int clustChoose;
		public KMScatterMenuChoose(int clust){
			clustChoose = clust;
		}
		public void actionPerformed(ActionEvent e) {
			//curClust = clustChoose;
			match = clustChoose;
			//currClustering = (Vector)((KMean)(model.getListOfClustering().get(clustering))).getClusteringAssignment().get(curClust);
			//c1 = ((KMean)(model.getListOfClustering().get(clustering))).getAsseg();
			first = true;
			KMeanSingleClusterScatter.this.repaint();
		}
	}
	
	public void paintComponent(Graphics g){
		if (g == null) return;

        VectorGraphics vg = VectorGraphics.create(g);
        super.paintComponent(vg);

		// Pulizia piano lavoro
        g.setColor(Color.WHITE);
		g.fillRect(0,0,640,670);
		// Sfondo bianco del grafico
		//g.setColor(Color.WHITE);
		g.fillRect(UpperLeftGraph.x,UpperLeftGraph.y,WIDTH,HEIGHT);
		// Sfondo grigio per label informativa
		g.setColor(Color.LIGHT_GRAY);
		g.fillRect(0,0,UpperRightGraph.x + RIGHT_SPACE,UpperLeftGraph.y - 10);
		g.setColor(Color.BLACK);
		g.drawRect(0,0,UpperRightGraph.x + RIGHT_SPACE,UpperLeftGraph.y - 10);
		Font fn=new Font("Arial",Font.ITALIC,16);
		g.setFont(fn);
		g.drawString("Single left click and drag to zoom, single right click to choose scatter options",50,20);
		// Bordi neri del grafico
		g.setColor(Color.BLACK);
		g.drawLine(UpperLeftGraph.x-1,UpperLeftGraph.y-1,UpperRightGraph.x,UpperRightGraph.y-1);	//UP
		g.drawLine(UpperLeftGraph.x-1,UpperLeftGraph.y-1,LowerLeftGraph.x-1,LowerLeftGraph.y);		//LEFT
		g.drawLine(LowerLeftGraph.x,LowerLeftGraph.y,LowerRightGraph.x,LowerRightGraph.y);			//BOTTOM
		g.drawLine(LowerRightGraph.x,LowerRightGraph.y,UpperRightGraph.x,UpperRightGraph.y);		//RIGHT
		
		// Testo variabile in x
		fn=new Font("Arial",Font.PLAIN,15);
		g.setFont(fn);
		g.drawString(model.getMatrix().getVariables()[xVar].getName(),XText.x,XText.y);
		// Elenco etichette valori in Y
		fn=new Font("Arial",Font.PLAIN,9);
		g.setFont(fn);
		for(int i = 0; i <= 9;i++){
			if(model.getMatrix().getVariables()[yVar].getType() == VariableNumber.DATE){
				SimpleDateFormat f = new SimpleDateFormat("dd.MM.yy HH:mm:ss");
				g.drawString(f.format(new Date((long)(yMax - (yMax - yMin)/9 * i))),WIDTH_TEXT_Y,UpperLeftGraph.y + (i * (HEIGHT / 9)));
			}else{
				g.drawString(valueFormat.format(yMax - (yMax - yMin)/9 * i),WIDTH_TEXT_Y,UpperLeftGraph.y + (i * (HEIGHT / 9)));
			}
		}
		// Testo variabile in Y
		fn=new Font("Arial",Font.PLAIN,17);
		fn=fn.deriveFont(AffineTransform.getRotateInstance(3.14/2 + 3.14));
		g.setFont(fn);
		g.drawString(model.getMatrix().getVariables()[yVar].getName(),YText.x,YText.y);
		// Testo etichette valori in x
		fn=new Font("Arial",Font.PLAIN,9);
		fn=fn.deriveFont(AffineTransform.getRotateInstance(3.14/2 + 3.14));
		g.setFont(fn);
		for(int i = 0; i <= 9;i++){
			if(model.getMatrix().getVariables()[xVar].getType() == VariableNumber.DATE){
				SimpleDateFormat f = new SimpleDateFormat("dd.MM.yy HH:mm:ss");
				g.drawString(f.format(new Date((long)(xMin + (xMax - xMin)/9 * i))),LowerLeftGraph.x + (i * (WIDTH / 9)),LowerLeftGraph.y + HEIGHT_VALUES_X);
			}else{
				g.drawString(valueFormat.format(xMin + (xMax - xMin)/9 * i),LowerLeftGraph.x + (i * (WIDTH / 9)),LowerLeftGraph.y + HEIGHT_VALUES_X);
			}
		}
		// Disegno punti
		if(first){
			first = false;
			new TimeConsumingWorker(new ProgressMonitorShow(this,"Plotting observations ...",100)){
				public Object construct() {
					Graphics2D g1 = (Graphics2D)graph.getGraphics();
					
					g1.setColor(Color.BLACK);
					g1.drawLine(0,500,500,500);
					g1.drawLine(500,500,500,0);
					
					g1.setColor(Color.WHITE);
					g1.fillRect(0,0,WIDTH,HEIGHT);
					g1.setColor(Color.LIGHT_GRAY);
					g1.setStroke(new BasicStroke(1,BasicStroke.CAP_BUTT,BasicStroke.JOIN_BEVEL,0,new float[]{1.5F},0));
					for(int i = 1;i < 9; i++){
						g1.drawLine(1,1+ (i * (HEIGHT / 9)),499,1 + (i * (HEIGHT / 9)));
						g1.drawLine(1+(i * (WIDTH / 9)),1,1+(i * (WIDTH / 9)),499);
					}
					g1.setColor(Color.RED);
					VariableNumber x = model.getMatrix().getVariables()[xVar];
					x.setRangeIntervallo(x.getIndexMin(xMin),x.getIndexMax(xMax),yMin,yMax,yVar);
					
					try {
						initShow(x.Size());
					} catch (InterruptedException e1) { e1.printStackTrace(); } 
					  catch (InvocationTargetException e1) { e1.printStackTrace(); }
					
					int xPos = 0;
					g1.setColor(JavaWatColor.getColor(match));
					for(int i = 0;i < x.Size(); i++){
						try{
							xPos = x.getNextInt();
							if(i%getStep()==0) updateInfos(i,"Plotting obs " + i,false);
							if(c1[curClust][x.getObsID(xPos)-1] == match){
								g1.fillOval((int)((x.getValue(xPos) - xMin) / (xMax - xMin) * (WIDTH-1)),(HEIGHT-1) - (int)((x.getValue(xPos,yVar) - yMin) / (yMax - yMin) * (HEIGHT-1)
									),pointSize,pointSize);
							}
						}catch(Exception e){
							break;
						}
					}
					updateInfos(x.Size(),"End",true);
					return null;
				}

				public void finished() {
					KMeanSingleClusterScatter.this.repaint();					
				}}.start();
		}
		g.drawImage(graph,UpperLeftGraph.x,UpperLeftGraph.y,null);
		
	}
	
	/**
     * Shows a screenshot dialog used to select screenshot format
     */
    public void showScreenShotDialog() {
        ExportDialog export = new ExportDialog("JWAT - version ???");
        export.showExportDialog(this, "Export as image...", this, "graph");
    }
    
	protected class GlassPanel extends JComponent implements MouseListener,MouseMotionListener{
		
		public GlassPanel(){
		}

		/**
	     * Overrides paintComponent method to add support for screenshots
	     */