1 package com.billkuker.rocketry.motorsim.io;
\r
4 import java.io.FileReader;
\r
5 import java.io.FileWriter;
\r
6 import java.io.IOException;
\r
7 import java.io.ObjectInputStream;
\r
8 import java.io.ObjectOutputStream;
\r
9 import java.io.StringReader;
\r
10 import java.io.StringWriter;
\r
11 import java.net.URI;
\r
12 import java.net.URISyntaxException;
\r
14 import org.jscience.physics.amount.Amount;
\r
16 import com.billkuker.rocketry.motorsim.Fuel;
\r
17 import com.billkuker.rocketry.motorsim.Motor;
\r
18 import com.billkuker.rocketry.motorsim.fuel.FuelResolver;
\r
19 import com.billkuker.rocketry.motorsim.fuel.FuelResolver.FuelNotFound;
\r
20 import com.thoughtworks.xstream.XStream;
\r
21 import com.thoughtworks.xstream.converters.ConversionException;
\r
22 import com.thoughtworks.xstream.converters.Converter;
\r
23 import com.thoughtworks.xstream.converters.MarshallingContext;
\r
24 import com.thoughtworks.xstream.converters.UnmarshallingContext;
\r
25 import com.thoughtworks.xstream.converters.javabean.JavaBeanConverter;
\r
26 import com.thoughtworks.xstream.io.HierarchicalStreamReader;
\r
27 import com.thoughtworks.xstream.io.HierarchicalStreamWriter;
\r
29 public class MotorIO {
\r
31 static class FuelConverter implements Converter{
\r
34 public boolean canConvert(Class c) {
\r
35 return Fuel.class.isAssignableFrom(c);
\r
39 public void marshal(Object o, HierarchicalStreamWriter w,
\r
40 MarshallingContext ctx) {
\r
42 w.setValue(f.getURI().toString());*/
\r
43 ctx.convertAnother(((Fuel)o).getURI());
\r
47 public Object unmarshal(HierarchicalStreamReader r,
\r
48 UnmarshallingContext ctx) {
\r
49 Object o = ctx.currentObject();
\r
51 Class<?> c = ctx.getRequiredType();
\r
53 uri = ctx.convertAnother(o, URI.class);
\r
54 } catch ( ConversionException e ){
\r
55 e.printStackTrace();
\r
57 if ( uri != null && uri instanceof URI ){
\r
59 return FuelResolver.getFuel((URI)uri);
\r
60 } catch (FuelNotFound e) {
\r
61 // TODO Auto-generated catch block
\r
62 e.printStackTrace();
\r
67 return FuelResolver.getFuel(new URI("motorsim:" + c.getSimpleName()));
\r
68 } catch (FuelNotFound e) {
\r
69 // TODO Auto-generated catch block
\r
70 e.printStackTrace();
\r
71 } catch (URISyntaxException e) {
\r
72 // TODO Auto-generated catch block
\r
73 e.printStackTrace();
\r
80 static class AmountConverter implements Converter{
\r
82 @SuppressWarnings("unchecked")
\r
83 public void marshal(Object o, HierarchicalStreamWriter w, MarshallingContext c) {
\r
84 Amount a = (Amount)o;
\r
86 //Leave off the fractional part if it is not relevant so we get exact values back
\r
88 text = a.getExactValue() + " " + a.getUnit();
\r
90 text = a.doubleValue(a.getUnit()) + " " + a.getUnit();
\r
94 public Object unmarshal(HierarchicalStreamReader r, UnmarshallingContext c) {
\r
95 String text = r.getValue();
\r
96 return Amount.valueOf(text);
\r
99 public boolean canConvert(Class c) {
\r
100 return c.equals(Amount.class);
\r
107 @SuppressWarnings("deprecation")
\r
108 private static XStream getXStream(){
\r
109 XStream xstream = new XStream();
\r
110 xstream.setMode(XStream.XPATH_ABSOLUTE_REFERENCES);
\r
111 xstream.registerConverter(new AmountConverter());
\r
112 xstream.registerConverter(new FuelConverter());
\r
113 xstream.registerConverter(new JavaBeanConverter(xstream.getClassMapper(), "class"), -20);
\r
117 public static void writeMotor(Motor m, File f) throws IOException{
\r
118 FileWriter fout = new FileWriter(f);
\r
119 ObjectOutputStream out = getXStream().createObjectOutputStream(fout);
\r
120 out.writeObject(m);
\r
125 public static Motor readMotor(File f) throws IOException{
\r
126 FileReader fin = new FileReader(f);
\r
127 ObjectInputStream in = getXStream().createObjectInputStream(fin);
\r
130 m = (Motor)in.readObject();
\r
131 } catch (ClassNotFoundException e) {
\r
132 throw new IOException("Class not found", e);
\r