samer@0: /*
samer@0:  *	VVector.java
samer@0:  *
samer@0:  *	Copyright (c) 2000, Samer Abdallah, King's College London.
samer@0:  *	All rights reserved.
samer@0:  *
samer@0:  *	This software is provided AS iS and WITHOUT ANY WARRANTY;
samer@0:  *	without even the implied warranty of MERCHANTABILITY or
samer@0:  *	FITNESS FOR A PARTICULAR PURPOSE.
samer@0:  */
samer@0: 
samer@0: package samer.maths;
samer@0: import  samer.core.*;
samer@0: import  samer.core.util.*;
samer@0: import  samer.tools.*;
samer@0: import  java.awt.*;
samer@0: import  java.awt.event.*;
samer@0: import  java.util.*;
samer@0: import  java.io.*;
samer@0: /**
samer@0: 	A Viewable Vec. The default viewer displays only
samer@0: 	the name and size of the vector, but the attached
samer@0: 	Agent allows creation of other types of viewer.
samer@0:   */
samer@0: 
samer@0: public class VVector extends Viewable implements Vec, Agent, Saveable
samer@0: {
samer@0: 	Vec	vec;
samer@0: 
samer@0: 	protected VVector() {}
samer@0: 
samer@0: 	public VVector(Node node, Vec v) {
samer@0: 		super(node); vec=v; setAgent(this);
samer@0: 		Shell.registerViewable(this);
samer@0: 	}
samer@0: 
samer@0: 	public VVector(Node node, double [] x) { this(node,new Vec.ForArray(x)); }
samer@0: 	public VVector(String nm, double [] x) { this(new Node(nm), x); }
samer@0: 	public VVector(String nm, int n) { this(new Node(nm), n); }
samer@0: 	public VVector(Node node, int n) {
samer@0: 		this(node,new Vec.ForArray(new double[n]));
samer@0: 	}
samer@0: 
samer@0: 	public void addSaver() { addAgent(new Saver(this,getNode())); }
samer@0: 
samer@0: 	// Vec implementation
samer@0: 	public Iterator	iterator()	{ return vec.iterator(); }
samer@0: 	public int			size()		{ return vec.size(); }
samer@0: 	public Mat			mat()			{ return vec.mat(); }
samer@0: 	public double[]	array()		{ return vec.array(); }
samer@0: 
samer@0: 	public Vec    getVec() { return vec; }
samer@0: 	public Viewer getViewer()
samer@0: 	{
samer@0: 		BaseViewer		vwr = new BaseViewer(this);
samer@0: 		StringBuffer	buf=new StringBuffer();
samer@0: 
samer@0: 		buf.append(getLabel())
samer@0: 			.append(" (").append(vec.size())
samer@0: 			.append(")");
samer@0: 		vwr.setText(buf.toString());
samer@0: 
samer@0: 		return vwr;
samer@0: 	}
samer@0: 
samer@0: 	public void copyFrom(double[] y) { Mathx.copy(y,array()); changed(); }
samer@0: 	public void copyTo(double[] y) { Mathx.copy(array(),y); }
samer@0: 	// Saveable implementation
samer@0: 	public void save(OutputStream out) throws Exception {
samer@0: 		new ObjectOutputStream(out).writeObject(array());
samer@0: 	}
samer@0: 
samer@0: 	public void load(InputStream in) throws Exception {
samer@0: 		Object o=new ObjectInputStream(in).readObject();
samer@0: 		Mathx.copy((double [])o,array()); 	changed();
samer@0: 	}
samer@0: 
samer@0: 		/** Write the given Vec to a PrintWriter as one line of text,
samer@0: 		consisting of space delimited numbers. */
samer@0: 
samer@0: 	public void write(Writer w) { print(new PrintWriter(w)); }
samer@0: 	public void print(PrintWriter prn) {
samer@0: 		Vec.Iterator i=vec.iterator();
samer@0: 		prn.print(i.next());
samer@0: 		while (i.more()) {
samer@0: 			prn.print(' ');
samer@0: 			prn.print(i.next());
samer@0: 		}
samer@0: 		prn.println();
samer@0: 		prn.flush();
samer@0: 	}
samer@0: 
samer@0: 	/** Reads a line of numbers from the reader and stores in the Vec */
samer@0: 	public void read(Reader input) throws IOException
samer@0: 	{
samer@0: 		StreamTokenizer tok= new StreamTokenizer(new BufferedReader(input));
samer@0: 
samer@0: 		/*
samer@0: 		a)	words are made from only +,-,.,e,E, or digits
samer@0: 		 b) if other tokens may exist in a line, these are ignored
samer@0: 		 c) if there is too much stuff on one line, the rest is discarded.
samer@0: 		 d) if there aren't enough numbers, throw an exception.
samer@0: 		 */
samer@0: 
samer@0: 		tok.resetSyntax();
samer@0: 		tok.wordChars('+','+');
samer@0: 		tok.wordChars('-','-');
samer@0: 		tok.wordChars('0','9');
samer@0: 		tok.wordChars('.','.');
samer@0: 		tok.wordChars('e','e');
samer@0: 		tok.wordChars('E','E');
samer@0: 		tok.commentChar('#');
samer@0: 		tok.commentChar(';');
samer@0: 		tok.commentChar('%');
samer@0: 		tok.whitespaceChars(0, ' ');
samer@0: 		tok.eolIsSignificant(true);
samer@0: 
samer@0:       // Ignore initial empty lines
samer@0: 		while (tok.nextToken() == StreamTokenizer.TT_EOL);
samer@0: 		if (tok.ttype == StreamTokenizer.TT_EOF)
samer@0: 			throw new EOFException("EOF on vector read.");
samer@0: 
samer@0: 		double [] x=vec.array();
samer@0: 		int			i=0;
samer@0: 
samer@0: 		// get first x.length words and convert to numbers
samer@0: 		do {
samer@0: 			if (tok.ttype==StreamTokenizer.TT_WORD)
samer@0: 				x[i++]=Double.parseDouble(tok.sval);
samer@0: 		} while (tok.nextToken()!=StreamTokenizer.TT_EOL && i<x.length);
samer@0: 		// !! what if EOF?
samer@0: 		if (i<(x.length-1)) throw new IOException("not enough numbers");
samer@0: 
samer@0: 		// now mop up rest of line
samer@0: 		while (tok.ttype!=StreamTokenizer.TT_EOL) tok.nextToken();
samer@0: 		changed();
samer@0: 	}
samer@0: 
samer@0: 	public void getCommands(Registry r) {
samer@0: 		r.add("zeros").add("negate").add("set").group();
samer@0: 		r.add("plotter")
samer@0: 		 .add("image")
samer@0: 		 .add("reshaped image")
samer@0: 		 .add("trace")
samer@0: 		 .add("image editor")
samer@0: 		 .add("plotter editor")
samer@0: 		 .add("multitrace");
samer@0: 		r.group();
samer@0: 		r.add("publish");
samer@0: 
samer@0: 	}
samer@0: 
samer@0: 	private void expose(Viewer v) { Shell.expose(v,"window"); }
samer@0: 
samer@0: 	public void execute(String cmd, Environment env) throws Exception
samer@0: 	{
samer@0: 		if (cmd.equals("zeros")) {
samer@0: 
samer@0: 			double [] x = vec.array();
samer@0: 			if (x!=null) Mathx.zero(x);
samer@0: 			else {	// try using Mat interface instead
samer@0: 				Mat mat = vec.mat();
samer@0: 				if (mat!=null) for (int i=0; i<mat.width(); i++) mat.set(1,i,0);
samer@0: 				else throw new Exception("Vector inaccessible: "+getLabel());
samer@0: 			}
samer@0: 			changed();
samer@0: 
samer@0: 		} else if (cmd.equals("set")) {
samer@0: 
samer@0: 			double y=X._double(env.datum(),1);
samer@0: 			double [] x = vec.array();
samer@0: 			if (x!=null) Mathx.setAll(x,y);
samer@0: 			else {	// try using Mat interface instead
samer@0: 				Mat mat = vec.mat();
samer@0: 				if (mat!=null) for (int i=0; i<mat.width(); i++) mat.set(1,i,y);
samer@0: 				else throw new Exception("Vector inaccessible: "+getLabel());
samer@0: 			}
samer@0: 			changed();
samer@0: 
samer@0: 		} else if (cmd.equals("negate")) {
samer@0: 
samer@0: 			double [] x = vec.array();
samer@0: 			if (x!=null) Mathx.negate(x);
samer@0: 			else {	// try using Mat interface instead
samer@0: 				Mat mat = vec.mat();
samer@0: 				if (mat!=null) for (int i=0; i<mat.width(); i++) mat.set(1,i,-mat.get(1,i));
samer@0: 				else throw new Exception("Vector inaccessible: "+getLabel());
samer@0: 			}
samer@0: 			changed();
samer@0: 
samer@0: 		} else if (cmd.equals("plotter")) {
samer@0: 
samer@0: 			pushNode("plotter");
samer@0: 			expose(new VectorPlotter(this,getVec()));
samer@0: 			Shell.pop();
samer@0: 
samer@0: 		} else if (cmd.equals("plotter editor")) {
samer@0: 
samer@0: 			pushNode("editor.plotter");
samer@0: 			expose(new VectorEditor(this,getVec()));
samer@0: 			Shell.pop();
samer@0: 
samer@0: 		} else if (cmd.equals("image")) {
samer@0: 
samer@0: 			Node n=pushNode("image");
samer@0: 			expose(createVectorImage(n,vec.size(),1));
samer@0: 			Shell.pop();
samer@0: 
samer@0: 		} else if (cmd.equals("reshaped image")) {
samer@0: 
samer@0: 			Node n=pushNode("reshaped.image");
samer@0: 			int w=Shell.getInt("width",1);
samer@0: 			int h=Shell.getInt("height",vec.size());
samer@0: 			expose(createVectorImage( n, w, h));
samer@0: 			Shell.pop();
samer@0: 
samer@0: 		} else if (cmd.equals("image editor")) {
samer@0: 
samer@0: 			Node n=pushNode("editor.image");
samer@0: 
samer@0: 			final Shell.Window win = Shell.getWindow("window");
samer@0: 			final ImageViewer vwr = createVectorImage(n,vec.size(),1);
samer@0: 			final MatEditor editor = new MatEditor(mat(),vwr,this,vwr);
samer@0: 
samer@0: 			win.container().setLayout( new BorderLayout(6,6));
samer@0: 			win.container().add( vwr.getComponent(), "Center");
samer@0: 			// win.container().add( editor.setto.getViewer().getComponent(),"South");
samer@0: 			Shell.registerViewable(editor.setto);
samer@0: 			win.addWindowListener( new WindowAdapter() {
samer@0: 				public void windowClosing(WindowEvent e) {
samer@0: 					editor.detach();
samer@0: 					vwr.detach();
samer@0: 					win.dispose();
samer@0: 				}
samer@0: 			} );
samer@0: 			win.expose();
samer@0: 
samer@0: 			// vwr.exposeCommands(this);
samer@0: 
samer@0: 			Shell.pop();
samer@0: 
samer@0: 		} else if (cmd.equals("trace")) {
samer@0: 
samer@0: 			Node node=pushNode("trace");
samer@0: 			ImageSourceBase imgsrc=createImageSource(1,vec.size());
samer@0: 			expose(new ImageTrace(imgsrc,this));
samer@0: 			Shell.pop();
samer@0: 
samer@0: 		} else if (cmd.equals("multitrace")) {
samer@0: 
samer@0: 			Node node=pushNode("trace");
samer@0: 			expose(new VectorTraceObs(this,node));
samer@0: 			Shell.pop();
samer@0: 		} else if (cmd.equals("publish")) {
samer@0: 			Shell.put(X.string(env.datum(),node.fullName()),this);
samer@0: 		}
samer@0: 	}
samer@0: 
samer@0: 
samer@0: 	private Node pushNode(String nm) {
samer@0: 		Node n=new Node(nm,getNode());
samer@0: 		Shell.push(n);
samer@0: 		return n;
samer@0: 	}
samer@0: 
samer@0: 	private ImageSourceBase createImageSource(int w, int h)
samer@0: 	{
samer@0: 		// this chooses an ImageSource depending on
samer@0: 		// the capabilities of the Vec
samer@0: 		double [] array=array();
samer@0: 
samer@0: 		if (array!=null) {
samer@0: 			if (w>1) return new ArrayImageSource(array,w,h);
samer@0: 			else     return new ArrayVImageSource(array);
samer@0: 		} else {
samer@0: 			return new IteratorImageSource(getVec(),w,h);
samer@0: 		}
samer@0: 	}
samer@0: 
samer@0: 	private ImageViewer createVectorImage( final Node n, int w, int h)
samer@0: 	{
samer@0: 
samer@0: 		/*
samer@0: 			The VImageViewer handles displaying the image of
samer@0: 			a supplied ImageSource, but this object handles
samer@0: 			scaling of the colour map to match the range of
samer@0: 			values in a Vec, accessed by iterator rather than
samer@0: 			array.
samer@0: 		  */
samer@0: 		return new ImageViewer( createImageSource(w,h), this) {
samer@0: 			public void scale()
samer@0: 			{
samer@0: 				Vec.Iterator	i=vec.iterator();
samer@0: 				double			max, min, x;
samer@0: 
samer@0: 				min=max=i.next();
samer@0: 				while (i.more()) {
samer@0: 					x=i.next();
samer@0: 					if (x>max) max=x;
samer@0: 					if (x<min) min=x;
samer@0: 				}
samer@0: 
samer@0: 				getVMap().setDomain(min,max);
samer@0: 				getVMap().changed(this);
samer@0: 			}
samer@0: 		};
samer@0: 	}
samer@0: }
samer@0: 
samer@0: 
samer@0: class VectorTraceObs extends VectorTrace implements Observer {
samer@0: 	Observable obs;
samer@0: 	Node		node;
samer@0: 
samer@0: 	VectorTraceObs(VVector v, Node n) {
samer@0: 		super(v);
samer@0: 		obs=v;
samer@0: 		node=n;
samer@0: 	}
samer@0: 
samer@0: 	public void attach() {	super.attach(); obs.addObserver(this); }
samer@0: 	public void detach() {	obs.deleteObserver(this); super.detach(); }
samer@0: 	public void update(Observable o, Object arg)
samer@0: 	{
samer@0: 		if (o==obs) {
samer@0: 			if (arg==Viewable.DISPOSING) {
samer@0: 				Shell.releaseViewer(this);
samer@0: 			} else {
samer@0: 				try { run(); }
samer@0: 				catch (Exception ex) {}
samer@0: 			}
samer@0: 		} else super.update(o,arg);
samer@0: 	}
samer@0: }