git.gag.com Git - debian/openrocket/blob - core/src/net/sf/openrocket/file/openrocket/importt/OpenRocketLoader.java

   1 package net.sf.openrocket.file.openrocket.importt;
   2
   3 import java.io.IOException;
   4 import java.io.InputStream;
   5 import java.lang.reflect.Constructor;
   6 import java.lang.reflect.InvocationTargetException;
   7 import java.util.ArrayList;
   8 import java.util.HashMap;
   9 import java.util.List;
  10 import java.util.Locale;
  11 import java.util.regex.Matcher;
  12 import java.util.regex.Pattern;
  13
  14 import net.sf.openrocket.aerodynamics.Warning;
  15 import net.sf.openrocket.aerodynamics.WarningSet;
  16 import net.sf.openrocket.database.Databases;
  17 import net.sf.openrocket.document.OpenRocketDocument;
  18 import net.sf.openrocket.document.Simulation;
  19 import net.sf.openrocket.document.Simulation.Status;
  20 import net.sf.openrocket.document.StorageOptions;
  21 import net.sf.openrocket.file.AbstractRocketLoader;
  22 import net.sf.openrocket.file.MotorFinder;
  23 import net.sf.openrocket.file.RocketLoadException;
  24 import net.sf.openrocket.file.simplesax.AbstractElementHandler;
  25 import net.sf.openrocket.file.simplesax.ElementHandler;
  26 import net.sf.openrocket.file.simplesax.PlainTextHandler;
  27 import net.sf.openrocket.file.simplesax.SimpleSAX;
  28 import net.sf.openrocket.logging.LogHelper;
  29 import net.sf.openrocket.material.Material;
  30 import net.sf.openrocket.motor.Motor;
  31 import net.sf.openrocket.preset.ComponentPreset;
  32 import net.sf.openrocket.rocketcomponent.BodyComponent;
  33 import net.sf.openrocket.rocketcomponent.BodyTube;
  34 import net.sf.openrocket.rocketcomponent.Bulkhead;
  35 import net.sf.openrocket.rocketcomponent.CenteringRing;
  36 import net.sf.openrocket.rocketcomponent.ClusterConfiguration;
  37 import net.sf.openrocket.rocketcomponent.Clusterable;
  38 import net.sf.openrocket.rocketcomponent.EllipticalFinSet;
  39 import net.sf.openrocket.rocketcomponent.EngineBlock;
  40 import net.sf.openrocket.rocketcomponent.ExternalComponent;
  41 import net.sf.openrocket.rocketcomponent.ExternalComponent.Finish;
  42 import net.sf.openrocket.rocketcomponent.FinSet;
  43 import net.sf.openrocket.rocketcomponent.FinSet.TabRelativePosition;
  44 import net.sf.openrocket.rocketcomponent.FreeformFinSet;
  45 import net.sf.openrocket.rocketcomponent.IllegalFinPointException;
  46 import net.sf.openrocket.rocketcomponent.InnerTube;
  47 import net.sf.openrocket.rocketcomponent.InternalComponent;
  48 import net.sf.openrocket.rocketcomponent.LaunchLug;
  49 import net.sf.openrocket.rocketcomponent.MassComponent;
  50 import net.sf.openrocket.rocketcomponent.MassObject;
  51 import net.sf.openrocket.rocketcomponent.MotorMount;
  52 import net.sf.openrocket.rocketcomponent.NoseCone;
  53 import net.sf.openrocket.rocketcomponent.Parachute;
  54 import net.sf.openrocket.rocketcomponent.RadiusRingComponent;
  55 import net.sf.openrocket.rocketcomponent.RecoveryDevice;
  56 import net.sf.openrocket.rocketcomponent.ReferenceType;
  57 import net.sf.openrocket.rocketcomponent.RingComponent;
  58 import net.sf.openrocket.rocketcomponent.Rocket;
  59 import net.sf.openrocket.rocketcomponent.RocketComponent;
  60 import net.sf.openrocket.rocketcomponent.RocketComponent.Position;
  61 import net.sf.openrocket.rocketcomponent.ShockCord;
  62 import net.sf.openrocket.rocketcomponent.Stage;
  63 import net.sf.openrocket.rocketcomponent.Streamer;
  64 import net.sf.openrocket.rocketcomponent.StructuralComponent;
  65 import net.sf.openrocket.rocketcomponent.SymmetricComponent;
  66 import net.sf.openrocket.rocketcomponent.ThicknessRingComponent;
  67 import net.sf.openrocket.rocketcomponent.Transition;
  68 import net.sf.openrocket.rocketcomponent.TrapezoidFinSet;
  69 import net.sf.openrocket.rocketcomponent.TubeCoupler;
  70 import net.sf.openrocket.simulation.CustomExpression;
  71 import net.sf.openrocket.simulation.FlightData;
  72 import net.sf.openrocket.simulation.FlightDataBranch;
  73 import net.sf.openrocket.simulation.FlightDataType;
  74 import net.sf.openrocket.simulation.FlightEvent;
  75 import net.sf.openrocket.simulation.FlightEvent.Type;
  76 import net.sf.openrocket.simulation.SimulationOptions;
  77 import net.sf.openrocket.startup.Application;
  78 import net.sf.openrocket.unit.UnitGroup;
  79 import net.sf.openrocket.util.BugException;
  80 import net.sf.openrocket.util.Color;
  81 import net.sf.openrocket.util.Coordinate;
  82 import net.sf.openrocket.util.GeodeticComputationStrategy;
  83 import net.sf.openrocket.util.LineStyle;
  84 import net.sf.openrocket.util.Reflection;
  85
  86 import org.xml.sax.InputSource;
  87 import org.xml.sax.SAXException;
  88
  89
  90 /**
  91  * Class that loads a rocket definition from an OpenRocket rocket file.
  92  * <p>
  93  * This class uses SAX to read the XML file format.  The
  94  * #loadFromStream(InputStream) method simply sets the system up and
  95  * starts the parsing, while the actual logic is in the private inner class
  96  * <code>OpenRocketHandler</code>.
  97  *
  98  * @author Sampo Niskanen <sampo.niskanen@iki.fi>
  99  */
 100 public class OpenRocketLoader extends AbstractRocketLoader {
 101         private static final LogHelper log = Application.getLogger();
 102
 103
 104         @Override
 105         public OpenRocketDocument loadFromStream(InputStream source, MotorFinder motorFinder) throws RocketLoadException,
 106         IOException {
 107                 log.info("Loading .ork file");
 108                 DocumentLoadingContext context = new DocumentLoadingContext();
 109                 context.setMotorFinder(motorFinder);
 110
 111                 InputSource xmlSource = new InputSource(source);
 112                 OpenRocketHandler handler = new OpenRocketHandler(context);
 113
 114
 115                 try {
 116                         SimpleSAX.readXML(xmlSource, handler, warnings);
 117                 } catch (SAXException e) {
 118                         log.warn("Malformed XML in input");
 119                         throw new RocketLoadException("Malformed XML in input.", e);
 120                 }
 121
 122
 123                 OpenRocketDocument doc = handler.getDocument();
 124                 doc.getDefaultConfiguration().setAllStages();
 125
 126                 // Deduce suitable time skip
 127                 double timeSkip = StorageOptions.SIMULATION_DATA_NONE;
 128                 for (Simulation s : doc.getSimulations()) {
 129                         if (s.getStatus() == Simulation.Status.EXTERNAL ||
 130                                         s.getStatus() == Simulation.Status.NOT_SIMULATED)
 131                                 continue;
 132                         if (s.getSimulatedData() == null)
 133                                 continue;
 134                         if (s.getSimulatedData().getBranchCount() == 0)
 135                                 continue;
 136                         FlightDataBranch branch = s.getSimulatedData().getBranch(0);
 137                         if (branch == null)
 138                                 continue;
 139                         List<Double> list = branch.get(FlightDataType.TYPE_TIME);
 140                         if (list == null)
 141                                 continue;
 142
 143                         double previousTime = Double.NaN;
 144                         for (double time : list) {
 145                                 if (time - previousTime < timeSkip)
 146                                         timeSkip = time - previousTime;
 147                                 previousTime = time;
 148                         }
 149                 }
 150                 // Round value
 151                 timeSkip = Math.rint(timeSkip * 100) / 100;
 152
 153                 doc.getDefaultStorageOptions().setSimulationTimeSkip(timeSkip);
 154                 doc.getDefaultStorageOptions().setCompressionEnabled(false); // Set by caller if compressed
 155                 doc.getDefaultStorageOptions().setExplicitlySet(false);
 156
 157                 doc.clearUndo();
 158                 log.info("Loading done");
 159                 return doc;
 160         }
 161
 162 }
 163
 164
 165
 166 class DocumentConfig {
 167
 168         /* Remember to update OpenRocketSaver as well! */
 169         public static final String[] SUPPORTED_VERSIONS = { "1.0", "1.1", "1.2", "1.3", "1.4", "1.5" };
 170
 171         /**
 172          * Divisor used in converting an integer version to the point-represented version.
 173          * The integer version divided by this value is the major version and the remainder is
 174          * the minor version.  For example 101 corresponds to file version "1.1".
 175          */
 176         public static final int FILE_VERSION_DIVISOR = 100;
 177
 178
 179         ////////  Component constructors
 180         static final HashMap<String, Constructor<? extends RocketComponent>> constructors = new HashMap<String, Constructor<? extends RocketComponent>>();
 181         static {
 182                 try {
 183                         // External components
 184                         constructors.put("bodytube", BodyTube.class.getConstructor(new Class<?>[0]));
 185                         constructors.put("transition", Transition.class.getConstructor(new Class<?>[0]));
 186                         constructors.put("nosecone", NoseCone.class.getConstructor(new Class<?>[0]));
 187                         constructors.put("trapezoidfinset", TrapezoidFinSet.class.getConstructor(new Class<?>[0]));
 188                         constructors.put("ellipticalfinset", EllipticalFinSet.class.getConstructor(new Class<?>[0]));
 189                         constructors.put("freeformfinset", FreeformFinSet.class.getConstructor(new Class<?>[0]));
 190                         constructors.put("launchlug", LaunchLug.class.getConstructor(new Class<?>[0]));
 191
 192                         // Internal components
 193                         constructors.put("engineblock", EngineBlock.class.getConstructor(new Class<?>[0]));
 194                         constructors.put("innertube", InnerTube.class.getConstructor(new Class<?>[0]));
 195                         constructors.put("tubecoupler", TubeCoupler.class.getConstructor(new Class<?>[0]));
 196                         constructors.put("bulkhead", Bulkhead.class.getConstructor(new Class<?>[0]));
 197                         constructors.put("centeringring", CenteringRing.class.getConstructor(new Class<?>[0]));
 198
 199                         constructors.put("masscomponent", MassComponent.class.getConstructor(new Class<?>[0]));
 200                         constructors.put("shockcord", ShockCord.class.getConstructor(new Class<?>[0]));
 201                         constructors.put("parachute", Parachute.class.getConstructor(new Class<?>[0]));
 202                         constructors.put("streamer", Streamer.class.getConstructor(new Class<?>[0]));
 203
 204                         // Other
 205                         constructors.put("stage", Stage.class.getConstructor(new Class<?>[0]));
 206
 207                 } catch (NoSuchMethodException e) {
 208                         throw new BugException(
 209                                         "Error in constructing the 'constructors' HashMap.");
 210                 }
 211         }
 212
 213
 214         ////////  Parameter setters
 215         /*
 216          * The keys are of the form Class:param, where Class is the class name and param
 217          * the element name.  Setters are searched for in descending class order.
 218          * A setter of null means setting the parameter is not allowed.
 219          */
 220         static final HashMap<String, Setter> setters = new HashMap<String, Setter>();
 221         static {
 222                 // RocketComponent
 223                 setters.put("RocketComponent:name", new StringSetter(
 224                                 Reflection.findMethod(RocketComponent.class, "setName", String.class)));
 225                 setters.put("RocketComponent:color", new ColorSetter(
 226                                 Reflection.findMethod(RocketComponent.class, "setColor", Color.class)));
 227                 setters.put("RocketComponent:linestyle", new EnumSetter<LineStyle>(
 228                                 Reflection.findMethod(RocketComponent.class, "setLineStyle", LineStyle.class),
 229                                 LineStyle.class));
 230                 setters.put("RocketComponent:position", new PositionSetter());
 231                 setters.put("RocketComponent:overridemass", new OverrideSetter(
 232                                 Reflection.findMethod(RocketComponent.class, "setOverrideMass", double.class),
 233                                 Reflection.findMethod(RocketComponent.class, "setMassOverridden", boolean.class)));
 234                 setters.put("RocketComponent:overridecg", new OverrideSetter(
 235                                 Reflection.findMethod(RocketComponent.class, "setOverrideCGX", double.class),
 236                                 Reflection.findMethod(RocketComponent.class, "setCGOverridden", boolean.class)));
 237                 setters.put("RocketComponent:overridesubcomponents", new BooleanSetter(
 238                                 Reflection.findMethod(RocketComponent.class, "setOverrideSubcomponents", boolean.class)));
 239                 setters.put("RocketComponent:comment", new StringSetter(
 240                                 Reflection.findMethod(RocketComponent.class, "setComment", String.class)));
 241                 setters.put("RocketComponent:preset", new ComponentPresetSetter(
 242                                 Reflection.findMethod(RocketComponent.class, "loadPreset", ComponentPreset.class)));
 243
 244                 // ExternalComponent
 245                 setters.put("ExternalComponent:finish", new EnumSetter<Finish>(
 246                                 Reflection.findMethod(ExternalComponent.class, "setFinish", Finish.class),
 247                                 Finish.class));
 248                 setters.put("ExternalComponent:material", new MaterialSetter(
 249                                 Reflection.findMethod(ExternalComponent.class, "setMaterial", Material.class),
 250                                 Material.Type.BULK));
 251
 252                 // BodyComponent
 253                 setters.put("BodyComponent:length", new DoubleSetter(
 254                                 Reflection.findMethod(BodyComponent.class, "setLength", double.class)));
 255
 256                 // SymmetricComponent
 257                 setters.put("SymmetricComponent:thickness", new DoubleSetter(
 258                                 Reflection.findMethod(SymmetricComponent.class, "setThickness", double.class),
 259                                 "filled",
 260                                 Reflection.findMethod(SymmetricComponent.class, "setFilled", boolean.class)));
 261
 262                 // BodyTube
 263                 setters.put("BodyTube:radius", new DoubleSetter(
 264                                 Reflection.findMethod(BodyTube.class, "setOuterRadius", double.class),
 265                                 "auto",
 266                                 Reflection.findMethod(BodyTube.class, "setOuterRadiusAutomatic", boolean.class)));
 267
 268                 // Transition
 269                 setters.put("Transition:shape", new EnumSetter<Transition.Shape>(
 270                                 Reflection.findMethod(Transition.class, "setType", Transition.Shape.class),
 271                                 Transition.Shape.class));
 272                 setters.put("Transition:shapeclipped", new BooleanSetter(
 273                                 Reflection.findMethod(Transition.class, "setClipped", boolean.class)));
 274                 setters.put("Transition:shapeparameter", new DoubleSetter(
 275                                 Reflection.findMethod(Transition.class, "setShapeParameter", double.class)));
 276
 277                 setters.put("Transition:foreradius", new DoubleSetter(
 278                                 Reflection.findMethod(Transition.class, "setForeRadius", double.class),
 279                                 "auto",
 280                                 Reflection.findMethod(Transition.class, "setForeRadiusAutomatic", boolean.class)));
 281                 setters.put("Transition:aftradius", new DoubleSetter(
 282                                 Reflection.findMethod(Transition.class, "setAftRadius", double.class),
 283                                 "auto",
 284                                 Reflection.findMethod(Transition.class, "setAftRadiusAutomatic", boolean.class)));
 285
 286                 setters.put("Transition:foreshoulderradius", new DoubleSetter(
 287                                 Reflection.findMethod(Transition.class, "setForeShoulderRadius", double.class)));
 288                 setters.put("Transition:foreshoulderlength", new DoubleSetter(
 289                                 Reflection.findMethod(Transition.class, "setForeShoulderLength", double.class)));
 290                 setters.put("Transition:foreshoulderthickness", new DoubleSetter(
 291                                 Reflection.findMethod(Transition.class, "setForeShoulderThickness", double.class)));
 292                 setters.put("Transition:foreshouldercapped", new BooleanSetter(
 293                                 Reflection.findMethod(Transition.class, "setForeShoulderCapped", boolean.class)));
 294
 295                 setters.put("Transition:aftshoulderradius", new DoubleSetter(
 296                                 Reflection.findMethod(Transition.class, "setAftShoulderRadius", double.class)));
 297                 setters.put("Transition:aftshoulderlength", new DoubleSetter(
 298                                 Reflection.findMethod(Transition.class, "setAftShoulderLength", double.class)));
 299                 setters.put("Transition:aftshoulderthickness", new DoubleSetter(
 300                                 Reflection.findMethod(Transition.class, "setAftShoulderThickness", double.class)));
 301                 setters.put("Transition:aftshouldercapped", new BooleanSetter(
 302                                 Reflection.findMethod(Transition.class, "setAftShoulderCapped", boolean.class)));
 303
 304                 // NoseCone - disable disallowed elements
 305                 setters.put("NoseCone:foreradius", null);
 306                 setters.put("NoseCone:foreshoulderradius", null);
 307                 setters.put("NoseCone:foreshoulderlength", null);
 308                 setters.put("NoseCone:foreshoulderthickness", null);
 309                 setters.put("NoseCone:foreshouldercapped", null);
 310
 311                 // FinSet
 312                 setters.put("FinSet:fincount", new IntSetter(
 313                                 Reflection.findMethod(FinSet.class, "setFinCount", int.class)));
 314                 setters.put("FinSet:rotation", new DoubleSetter(
 315                                 Reflection.findMethod(FinSet.class, "setBaseRotation", double.class), Math.PI / 180.0));
 316                 setters.put("FinSet:thickness", new DoubleSetter(
 317                                 Reflection.findMethod(FinSet.class, "setThickness", double.class)));
 318                 setters.put("FinSet:crosssection", new EnumSetter<FinSet.CrossSection>(
 319                                 Reflection.findMethod(FinSet.class, "setCrossSection", FinSet.CrossSection.class),
 320                                 FinSet.CrossSection.class));
 321                 setters.put("FinSet:cant", new DoubleSetter(
 322                                 Reflection.findMethod(FinSet.class, "setCantAngle", double.class), Math.PI / 180.0));
 323                 setters.put("FinSet:tabheight", new DoubleSetter(
 324                                 Reflection.findMethod(FinSet.class, "setTabHeight", double.class)));
 325                 setters.put("FinSet:tablength", new DoubleSetter(
 326                                 Reflection.findMethod(FinSet.class, "setTabLength", double.class)));
 327                 setters.put("FinSet:tabposition", new FinTabPositionSetter());
 328
 329                 // TrapezoidFinSet
 330                 setters.put("TrapezoidFinSet:rootchord", new DoubleSetter(
 331                                 Reflection.findMethod(TrapezoidFinSet.class, "setRootChord", double.class)));
 332                 setters.put("TrapezoidFinSet:tipchord", new DoubleSetter(
 333                                 Reflection.findMethod(TrapezoidFinSet.class, "setTipChord", double.class)));
 334                 setters.put("TrapezoidFinSet:sweeplength", new DoubleSetter(
 335                                 Reflection.findMethod(TrapezoidFinSet.class, "setSweep", double.class)));
 336                 setters.put("TrapezoidFinSet:height", new DoubleSetter(
 337                                 Reflection.findMethod(TrapezoidFinSet.class, "setHeight", double.class)));
 338
 339                 // EllipticalFinSet
 340                 setters.put("EllipticalFinSet:rootchord", new DoubleSetter(
 341                                 Reflection.findMethod(EllipticalFinSet.class, "setLength", double.class)));
 342                 setters.put("EllipticalFinSet:height", new DoubleSetter(
 343                                 Reflection.findMethod(EllipticalFinSet.class, "setHeight", double.class)));
 344
 345                 // FreeformFinSet points handled as a special handler
 346
 347                 // LaunchLug
 348                 setters.put("LaunchLug:radius", new DoubleSetter(
 349                                 Reflection.findMethod(LaunchLug.class, "setOuterRadius", double.class)));
 350                 setters.put("LaunchLug:length", new DoubleSetter(
 351                                 Reflection.findMethod(LaunchLug.class, "setLength", double.class)));
 352                 setters.put("LaunchLug:thickness", new DoubleSetter(
 353                                 Reflection.findMethod(LaunchLug.class, "setThickness", double.class)));
 354                 setters.put("LaunchLug:radialdirection", new DoubleSetter(
 355                                 Reflection.findMethod(LaunchLug.class, "setRadialDirection", double.class),
 356                                 Math.PI / 180.0));
 357
 358                 // InternalComponent - nothing
 359
 360                 // StructuralComponent
 361                 setters.put("StructuralComponent:material", new MaterialSetter(
 362                                 Reflection.findMethod(StructuralComponent.class, "setMaterial", Material.class),
 363                                 Material.Type.BULK));
 364
 365                 // RingComponent
 366                 setters.put("RingComponent:length", new DoubleSetter(
 367                                 Reflection.findMethod(RingComponent.class, "setLength", double.class)));
 368                 setters.put("RingComponent:radialposition", new DoubleSetter(
 369                                 Reflection.findMethod(RingComponent.class, "setRadialPosition", double.class)));
 370                 setters.put("RingComponent:radialdirection", new DoubleSetter(
 371                                 Reflection.findMethod(RingComponent.class, "setRadialDirection", double.class),
 372                                 Math.PI / 180.0));
 373
 374                 // ThicknessRingComponent - radius on separate components due to differing automatics
 375                 setters.put("ThicknessRingComponent:thickness", new DoubleSetter(
 376                                 Reflection.findMethod(ThicknessRingComponent.class, "setThickness", double.class)));
 377
 378                 // EngineBlock
 379                 setters.put("EngineBlock:outerradius", new DoubleSetter(
 380                                 Reflection.findMethod(EngineBlock.class, "setOuterRadius", double.class),
 381                                 "auto",
 382                                 Reflection.findMethod(EngineBlock.class, "setOuterRadiusAutomatic", boolean.class)));
 383
 384                 // TubeCoupler
 385                 setters.put("TubeCoupler:outerradius", new DoubleSetter(
 386                                 Reflection.findMethod(TubeCoupler.class, "setOuterRadius", double.class),
 387                                 "auto",
 388                                 Reflection.findMethod(TubeCoupler.class, "setOuterRadiusAutomatic", boolean.class)));
 389
 390                 // InnerTube
 391                 setters.put("InnerTube:outerradius", new DoubleSetter(
 392                                 Reflection.findMethod(InnerTube.class, "setOuterRadius", double.class)));
 393                 setters.put("InnerTube:clusterconfiguration", new ClusterConfigurationSetter());
 394                 setters.put("InnerTube:clusterscale", new DoubleSetter(
 395                                 Reflection.findMethod(InnerTube.class, "setClusterScale", double.class)));
 396                 setters.put("InnerTube:clusterrotation", new DoubleSetter(
 397                                 Reflection.findMethod(InnerTube.class, "setClusterRotation", double.class),
 398                                 Math.PI / 180.0));
 399
 400                 // RadiusRingComponent
 401
 402                 // Bulkhead
 403                 setters.put("RadiusRingComponent:innerradius", new DoubleSetter(
 404                                 Reflection.findMethod(RadiusRingComponent.class, "setInnerRadius", double.class)));
 405                 setters.put("Bulkhead:outerradius", new DoubleSetter(
 406                                 Reflection.findMethod(Bulkhead.class, "setOuterRadius", double.class),
 407                                 "auto",
 408                                 Reflection.findMethod(Bulkhead.class, "setOuterRadiusAutomatic", boolean.class)));
 409
 410                 // CenteringRing
 411                 setters.put("CenteringRing:innerradius", new DoubleSetter(
 412                                 Reflection.findMethod(CenteringRing.class, "setInnerRadius", double.class),
 413                                 "auto",
 414                                 Reflection.findMethod(CenteringRing.class, "setInnerRadiusAutomatic", boolean.class)));
 415                 setters.put("CenteringRing:outerradius", new DoubleSetter(
 416                                 Reflection.findMethod(CenteringRing.class, "setOuterRadius", double.class),
 417                                 "auto",
 418                                 Reflection.findMethod(CenteringRing.class, "setOuterRadiusAutomatic", boolean.class)));
 419
 420
 421                 // MassObject
 422                 setters.put("MassObject:packedlength", new DoubleSetter(
 423                                 Reflection.findMethod(MassObject.class, "setLength", double.class)));
 424                 setters.put("MassObject:packedradius", new DoubleSetter(
 425                                 Reflection.findMethod(MassObject.class, "setRadius", double.class)));
 426                 setters.put("MassObject:radialposition", new DoubleSetter(
 427                                 Reflection.findMethod(MassObject.class, "setRadialPosition", double.class)));
 428                 setters.put("MassObject:radialdirection", new DoubleSetter(
 429                                 Reflection.findMethod(MassObject.class, "setRadialDirection", double.class),
 430                                 Math.PI / 180.0));
 431
 432                 // MassComponent
 433                 setters.put("MassComponent:mass", new DoubleSetter(
 434                                 Reflection.findMethod(MassComponent.class, "setComponentMass", double.class)));
 435
 436                 // ShockCord
 437                 setters.put("ShockCord:cordlength", new DoubleSetter(
 438                                 Reflection.findMethod(ShockCord.class, "setCordLength", double.class)));
 439                 setters.put("ShockCord:material", new MaterialSetter(
 440                                 Reflection.findMethod(ShockCord.class, "setMaterial", Material.class),
 441                                 Material.Type.LINE));
 442
 443                 // RecoveryDevice
 444                 setters.put("RecoveryDevice:cd", new DoubleSetter(
 445                                 Reflection.findMethod(RecoveryDevice.class, "setCD", double.class),
 446                                 "auto",
 447                                 Reflection.findMethod(RecoveryDevice.class, "setCDAutomatic", boolean.class)));
 448                 setters.put("RecoveryDevice:deployevent", new EnumSetter<RecoveryDevice.DeployEvent>(
 449                                 Reflection.findMethod(RecoveryDevice.class, "setDeployEvent", RecoveryDevice.DeployEvent.class),
 450                                 RecoveryDevice.DeployEvent.class));
 451                 setters.put("RecoveryDevice:deployaltitude", new DoubleSetter(
 452                                 Reflection.findMethod(RecoveryDevice.class, "setDeployAltitude", double.class)));
 453                 setters.put("RecoveryDevice:deploydelay", new DoubleSetter(
 454                                 Reflection.findMethod(RecoveryDevice.class, "setDeployDelay", double.class)));
 455                 setters.put("RecoveryDevice:material", new MaterialSetter(
 456                                 Reflection.findMethod(RecoveryDevice.class, "setMaterial", Material.class),
 457                                 Material.Type.SURFACE));
 458
 459                 // Parachute
 460                 setters.put("Parachute:diameter", new DoubleSetter(
 461                                 Reflection.findMethod(Parachute.class, "setDiameter", double.class)));
 462                 setters.put("Parachute:linecount", new IntSetter(
 463                                 Reflection.findMethod(Parachute.class, "setLineCount", int.class)));
 464                 setters.put("Parachute:linelength", new DoubleSetter(
 465                                 Reflection.findMethod(Parachute.class, "setLineLength", double.class)));
 466                 setters.put("Parachute:linematerial", new MaterialSetter(
 467                                 Reflection.findMethod(Parachute.class, "setLineMaterial", Material.class),
 468                                 Material.Type.LINE));
 469
 470                 // Streamer
 471                 setters.put("Streamer:striplength", new DoubleSetter(
 472                                 Reflection.findMethod(Streamer.class, "setStripLength", double.class)));
 473                 setters.put("Streamer:stripwidth", new DoubleSetter(
 474                                 Reflection.findMethod(Streamer.class, "setStripWidth", double.class)));
 475
 476                 // Rocket
 477                 // <motorconfiguration> handled by separate handler
 478                 setters.put("Rocket:referencetype", new EnumSetter<ReferenceType>(
 479                                 Reflection.findMethod(Rocket.class, "setReferenceType", ReferenceType.class),
 480                                 ReferenceType.class));
 481                 setters.put("Rocket:customreference", new DoubleSetter(
 482                                 Reflection.findMethod(Rocket.class, "setCustomReferenceLength", double.class)));
 483                 setters.put("Rocket:designer", new StringSetter(
 484                                 Reflection.findMethod(Rocket.class, "setDesigner", String.class)));
 485                 setters.put("Rocket:revision", new StringSetter(
 486                                 Reflection.findMethod(Rocket.class, "setRevision", String.class)));
 487
 488                 // Stage
 489                 setters.put("Stage:separationevent", new EnumSetter<Stage.SeparationEvent>(
 490                                 Reflection.findMethod(Stage.class, "setSeparationEvent", Stage.SeparationEvent.class),
 491                                 Stage.SeparationEvent.class));
 492                 setters.put("Stage:separationdelay", new DoubleSetter(
 493                                 Reflection.findMethod(Stage.class, "setSeparationDelay", double.class)));
 494
 495         }
 496
 497
 498         /**
 499          * Search for a enum value that has the corresponding name as an XML value.  The current
 500          * conversion from enum name to XML value is to lowercase the name and strip out all
 501          * underscore characters.  This method returns a match to these criteria, or <code>null</code>
 502          * if no such enum exists.
 503          *
 504          * @param <T>                   then enum type.
 505          * @param name                  the XML value, null ok.
 506          * @param enumClass             the class of the enum.
 507          * @return                              the found enum value, or <code>null</code>.
 508          */
 509         public static <T extends Enum<T>> Enum<T> findEnum(String name,
 510                         Class<? extends Enum<T>> enumClass) {
 511
 512                 if (name == null)
 513                         return null;
 514                 name = name.trim();
 515                 for (Enum<T> e : enumClass.getEnumConstants()) {
 516                         if (e.name().toLowerCase(Locale.ENGLISH).replace("_", "").equals(name)) {
 517                                 return e;
 518                         }
 519                 }
 520                 return null;
 521         }
 522
 523
 524         /**
 525          * Convert a string to a double including formatting specifications of the OpenRocket
 526          * file format.  This accepts all formatting that is valid for
 527          * <code>Double.parseDouble(s)</code> and a few others as well ("Inf", "-Inf").
 528          *
 529          * @param s             the string to parse.
 530          * @return              the numerical value.
 531          * @throws NumberFormatException        the the string cannot be parsed.
 532          */
 533         public static double stringToDouble(String s) throws NumberFormatException {
 534                 if (s == null)
 535                         throw new NumberFormatException("null string");
 536                 if (s.equalsIgnoreCase("NaN"))
 537                         return Double.NaN;
 538                 if (s.equalsIgnoreCase("Inf"))
 539                         return Double.POSITIVE_INFINITY;
 540                 if (s.equalsIgnoreCase("-Inf"))
 541                         return Double.NEGATIVE_INFINITY;
 542                 return Double.parseDouble(s);
 543         }
 544 }
 545
 546
 547
 548
 549
 550 /**
 551  * The starting point of the handlers.  Accepts a single <openrocket> element and hands
 552  * the contents to be read by a OpenRocketContentsHandler.
 553  */
 554 class OpenRocketHandler extends AbstractElementHandler {
 555         private final DocumentLoadingContext context;
 556         private OpenRocketContentHandler handler = null;
 557
 558         public OpenRocketHandler(DocumentLoadingContext context) {
 559                 this.context = context;
 560         }
 561
 562         /**
 563          * Return the OpenRocketDocument read from the file, or <code>null</code> if a document
 564          * has not been read yet.
 565          *
 566          * @return      the document read, or null.
 567          */
 568         public OpenRocketDocument getDocument() {
 569                 return handler.getDocument();
 570         }
 571
 572         @Override
 573         public ElementHandler openElement(String element, HashMap<String, String> attributes,
 574                         WarningSet warnings) {
 575
 576                 // Check for unknown elements
 577                 if (!element.equals("openrocket")) {
 578                         warnings.add(Warning.fromString("Unknown element " + element + ", ignoring."));
 579                         return null;
 580                 }
 581
 582                 // Check for first call
 583                 if (handler != null) {
 584                         warnings.add(Warning.fromString("Multiple document elements found, ignoring later "
 585                                         + "ones."));
 586                         return null;
 587                 }
 588
 589                 // Check version number
 590                 String version = null;
 591                 String creator = attributes.remove("creator");
 592                 String docVersion = attributes.remove("version");
 593                 for (String v : DocumentConfig.SUPPORTED_VERSIONS) {
 594                         if (v.equals(docVersion)) {
 595                                 version = v;
 596                                 break;
 597                         }
 598                 }
 599                 if (version == null) {
 600                         String str = "Unsupported document version";
 601                         if (docVersion != null)
 602                                 str += " " + docVersion;
 603                         if (creator != null && !creator.trim().equals(""))
 604                                 str += " (written using '" + creator.trim() + "')";
 605                         str += ", attempting to read file anyway.";
 606                         warnings.add(str);
 607                 }
 608
 609                 context.setFileVersion(parseVersion(docVersion));
 610
 611                 handler = new OpenRocketContentHandler(context);
 612                 return handler;
 613         }
 614
 615
 616         private int parseVersion(String docVersion) {
 617                 if (docVersion == null)
 618                         return 0;
 619
 620                 Matcher m = Pattern.compile("^([0-9]+)\\.([0-9]+)$").matcher(docVersion);
 621                 if (m.matches()) {
 622                         int major = Integer.parseInt(m.group(1));
 623                         int minor = Integer.parseInt(m.group(2));
 624                         return major * DocumentConfig.FILE_VERSION_DIVISOR + minor;
 625                 } else {
 626                         return 0;
 627                 }
 628         }
 629
 630         @Override
 631         public void closeElement(String element, HashMap<String, String> attributes,
 632                         String content, WarningSet warnings) throws SAXException {
 633                 attributes.remove("version");
 634                 attributes.remove("creator");
 635                 super.closeElement(element, attributes, content, warnings);
 636         }
 637
 638
 639 }
 640
 641
 642 /**
 643  * Handles the content of the <openrocket> tag.
 644  */
 645 class OpenRocketContentHandler extends AbstractElementHandler {
 646         private final DocumentLoadingContext context;
 647         private final OpenRocketDocument doc;
 648         private final Rocket rocket;
 649
 650         private boolean rocketDefined = false;
 651         private boolean simulationsDefined = false;
 652
 653         public OpenRocketContentHandler(DocumentLoadingContext context) {
 654                 this.context = context;
 655                 this.rocket = new Rocket();
 656                 this.doc = new OpenRocketDocument(rocket);
 657         }
 658
 659
 660         public OpenRocketDocument getDocument() {
 661                 if (!rocketDefined)
 662                         return null;
 663                 return doc;
 664         }
 665
 666         @Override
 667         public ElementHandler openElement(String element, HashMap<String, String> attributes,
 668                         WarningSet warnings) {
 669
 670                 if (element.equals("rocket")) {
 671                         if (rocketDefined) {
 672                                 warnings.add(Warning
 673                                                 .fromString("Multiple rocket designs within one document, "
 674                                                                 + "ignoring later ones."));
 675                                 return null;
 676                         }
 677                         rocketDefined = true;
 678                         return new ComponentParameterHandler(rocket, context);
 679                 }
 680
 681                 if (element.equals("simulations")) {
 682                         if (simulationsDefined) {
 683                                 warnings.add(Warning
 684                                                 .fromString("Multiple simulation definitions within one document, "
 685                                                                 + "ignoring later ones."));
 686                                 return null;
 687                         }
 688                         simulationsDefined = true;
 689                         return new SimulationsHandler(doc, context);
 690                 }
 691
 692                 warnings.add(Warning.fromString("Unknown element " + element + ", ignoring."));
 693
 694                 return null;
 695         }
 696 }
 697
 698
 699
 700
 701 /**
 702  * A handler that creates components from the corresponding elements.  The control of the
 703  * contents is passed on to ComponentParameterHandler.
 704  */
 705 class ComponentHandler extends AbstractElementHandler {
 706         private final DocumentLoadingContext context;
 707         private final RocketComponent parent;
 708
 709         public ComponentHandler(RocketComponent parent, DocumentLoadingContext context) {
 710                 this.parent = parent;
 711                 this.context = context;
 712         }
 713
 714         @Override
 715         public ElementHandler openElement(String element, HashMap<String, String> attributes,
 716                         WarningSet warnings) {
 717
 718                 // Attempt to construct new component
 719                 Constructor<? extends RocketComponent> constructor = DocumentConfig.constructors
 720                                 .get(element);
 721                 if (constructor == null) {
 722                         warnings.add(Warning.fromString("Unknown element " + element + ", ignoring."));
 723                         return null;
 724                 }
 725
 726                 RocketComponent c;
 727                 try {
 728                         c = constructor.newInstance();
 729                 } catch (InstantiationException e) {
 730                         throw new BugException("Error constructing component.", e);
 731                 } catch (IllegalAccessException e) {
 732                         throw new BugException("Error constructing component.", e);
 733                 } catch (InvocationTargetException e) {
 734                         throw Reflection.handleWrappedException(e);
 735                 }
 736
 737                 parent.addChild(c);
 738
 739                 return new ComponentParameterHandler(c, context);
 740         }
 741 }
 742
 743
 744 /**
 745  * A handler that populates the parameters of a previously constructed rocket component.
 746  * This uses the setters, or delegates the handling to another handler for specific
 747  * elements.
 748  */
 749 class ComponentParameterHandler extends AbstractElementHandler {
 750         private final DocumentLoadingContext context;
 751         private final RocketComponent component;
 752
 753         public ComponentParameterHandler(RocketComponent c, DocumentLoadingContext context) {
 754                 this.component = c;
 755                 this.context = context;
 756         }
 757
 758         @Override
 759         public ElementHandler openElement(String element, HashMap<String, String> attributes,
 760                         WarningSet warnings) {
 761
 762                 // Check for specific elements that contain other elements
 763                 if (element.equals("subcomponents")) {
 764                         return new ComponentHandler(component, context);
 765                 }
 766                 if (element.equals("motormount")) {
 767                         if (!(component instanceof MotorMount)) {
 768                                 warnings.add(Warning.fromString("Illegal component defined as motor mount."));
 769                                 return null;
 770                         }
 771                         return new MotorMountHandler((MotorMount) component, context);
 772                 }
 773                 if (element.equals("finpoints")) {
 774                         if (!(component instanceof FreeformFinSet)) {
 775                                 warnings.add(Warning.fromString("Illegal component defined for fin points."));
 776                                 return null;
 777                         }
 778                         return new FinSetPointHandler((FreeformFinSet) component, context);
 779                 }
 780                 if (element.equals("motorconfiguration")) {
 781                         if (!(component instanceof Rocket)) {
 782                                 warnings.add(Warning.fromString("Illegal component defined for motor configuration."));
 783                                 return null;
 784                         }
 785                         return new MotorConfigurationHandler((Rocket) component, context);
 786                 }
 787
 788
 789                 return PlainTextHandler.INSTANCE;
 790         }
 791
 792         @Override
 793         public void closeElement(String element, HashMap<String, String> attributes,
 794                         String content, WarningSet warnings) {
 795
 796                 if (element.equals("subcomponents") || element.equals("motormount") ||
 797                                 element.equals("finpoints") || element.equals("motorconfiguration")) {
 798                         return;
 799                 }
 800
 801                 // Search for the correct setter class
 802
 803                 Class<?> c;
 804                 for (c = component.getClass(); c != null; c = c.getSuperclass()) {
 805                         String setterKey = c.getSimpleName() + ":" + element;
 806                         Setter s = DocumentConfig.setters.get(setterKey);
 807                         if (s != null) {
 808                                 // Setter found
 809                                 s.set(component, content, attributes, warnings);
 810                                 break;
 811                         }
 812                         if (DocumentConfig.setters.containsKey(setterKey)) {
 813                                 // Key exists but is null -> invalid parameter
 814                                 c = null;
 815                                 break;
 816                         }
 817                 }
 818                 if (c == null) {
 819                         warnings.add(Warning.fromString("Unknown parameter type '" + element + "' for "
 820                                         + component.getComponentName() + ", ignoring."));
 821                 }
 822         }
 823 }
 824
 825
 826 /**
 827  * A handler that reads the <point> specifications within the freeformfinset's
 828  * <finpoints> elements.
 829  */
 830 class FinSetPointHandler extends AbstractElementHandler {
 831         @SuppressWarnings("unused")
 832         private final DocumentLoadingContext context;
 833         private final FreeformFinSet finset;
 834         private final ArrayList<Coordinate> coordinates = new ArrayList<Coordinate>();
 835
 836         public FinSetPointHandler(FreeformFinSet finset, DocumentLoadingContext context) {
 837                 this.finset = finset;
 838                 this.context = context;
 839         }
 840
 841         @Override
 842         public ElementHandler openElement(String element, HashMap<String, String> attributes,
 843                         WarningSet warnings) {
 844                 return PlainTextHandler.INSTANCE;
 845         }
 846
 847
 848         @Override
 849         public void closeElement(String element, HashMap<String, String> attributes,
 850                         String content, WarningSet warnings) throws SAXException {
 851
 852                 String strx = attributes.remove("x");
 853                 String stry = attributes.remove("y");
 854                 if (strx == null || stry == null) {
 855                         warnings.add(Warning.fromString("Illegal fin points specification, ignoring."));
 856                         return;
 857                 }
 858                 try {
 859                         double x = Double.parseDouble(strx);
 860                         double y = Double.parseDouble(stry);
 861                         coordinates.add(new Coordinate(x, y));
 862                 } catch (NumberFormatException e) {
 863                         warnings.add(Warning.fromString("Illegal fin points specification, ignoring."));
 864                         return;
 865                 }
 866
 867                 super.closeElement(element, attributes, content, warnings);
 868         }
 869
 870         @Override
 871         public void endHandler(String element, HashMap<String, String> attributes,
 872                         String content, WarningSet warnings) {
 873                 try {
 874                         finset.setPoints(coordinates.toArray(new Coordinate[0]));
 875                 } catch (IllegalFinPointException e) {
 876                         warnings.add(Warning.fromString("Freeform fin set point definitions illegal, ignoring."));
 877                 }
 878         }
 879 }
 880
 881
 882 class MotorMountHandler extends AbstractElementHandler {
 883         private final DocumentLoadingContext context;
 884         private final MotorMount mount;
 885         private MotorHandler motorHandler;
 886
 887         public MotorMountHandler(MotorMount mount, DocumentLoadingContext context) {
 888                 this.mount = mount;
 889                 this.context = context;
 890                 mount.setMotorMount(true);
 891         }
 892
 893         @Override
 894         public ElementHandler openElement(String element, HashMap<String, String> attributes,
 895                         WarningSet warnings) {
 896
 897                 if (element.equals("motor")) {
 898                         motorHandler = new MotorHandler(context);
 899                         return motorHandler;
 900                 }
 901
 902                 if (element.equals("ignitionevent") ||
 903                                 element.equals("ignitiondelay") ||
 904                                 element.equals("overhang")) {
 905                         return PlainTextHandler.INSTANCE;
 906                 }
 907
 908                 warnings.add(Warning.fromString("Unknown element '" + element + "' encountered, ignoring."));
 909                 return null;
 910         }
 911
 912
 913
 914         @Override
 915         public void closeElement(String element, HashMap<String, String> attributes,
 916                         String content, WarningSet warnings) throws SAXException {
 917
 918                 if (element.equals("motor")) {
 919                         String id = attributes.get("configid");
 920                         if (id == null || id.equals("")) {
 921                                 warnings.add(Warning.fromString("Illegal motor specification, ignoring."));
 922                                 return;
 923                         }
 924
 925                         Motor motor = motorHandler.getMotor(warnings);
 926                         mount.setMotor(id, motor);
 927                         mount.setMotorDelay(id, motorHandler.getDelay(warnings));
 928                         return;
 929                 }
 930
 931                 if (element.equals("ignitionevent")) {
 932                         MotorMount.IgnitionEvent event = null;
 933                         for (MotorMount.IgnitionEvent e : MotorMount.IgnitionEvent.values()) {
 934                                 if (e.name().toLowerCase(Locale.ENGLISH).replaceAll("_", "").equals(content)) {
 935                                         event = e;
 936                                         break;
 937                                 }
 938                         }
 939                         if (event == null) {
 940                                 warnings.add(Warning.fromString("Unknown ignition event type '" + content + "', ignoring."));
 941                                 return;
 942                         }
 943                         mount.setIgnitionEvent(event);
 944                         return;
 945                 }
 946
 947                 if (element.equals("ignitiondelay")) {
 948                         double d;
 949                         try {
 950                                 d = Double.parseDouble(content);
 951                         } catch (NumberFormatException nfe) {
 952                                 warnings.add(Warning.fromString("Illegal ignition delay specified, ignoring."));
 953                                 return;
 954                         }
 955                         mount.setIgnitionDelay(d);
 956                         return;
 957                 }
 958
 959                 if (element.equals("overhang")) {
 960                         double d;
 961                         try {
 962                                 d = Double.parseDouble(content);
 963                         } catch (NumberFormatException nfe) {
 964                                 warnings.add(Warning.fromString("Illegal overhang specified, ignoring."));
 965                                 return;
 966                         }
 967                         mount.setMotorOverhang(d);
 968                         return;
 969                 }
 970
 971                 super.closeElement(element, attributes, content, warnings);
 972         }
 973 }
 974
 975
 976
 977
 978 class MotorConfigurationHandler extends AbstractElementHandler {
 979         @SuppressWarnings("unused")
 980         private final DocumentLoadingContext context;
 981         private final Rocket rocket;
 982         private String name = null;
 983         private boolean inNameElement = false;
 984
 985         public MotorConfigurationHandler(Rocket rocket, DocumentLoadingContext context) {
 986                 this.rocket = rocket;
 987                 this.context = context;
 988         }
 989
 990         @Override
 991         public ElementHandler openElement(String element, HashMap<String, String> attributes,
 992                         WarningSet warnings) {
 993
 994                 if (inNameElement || !element.equals("name")) {
 995                         warnings.add(Warning.FILE_INVALID_PARAMETER);
 996                         return null;
 997                 }
 998                 inNameElement = true;
 999
1000                 return PlainTextHandler.INSTANCE;
1001         }
1002
1003         @Override
1004         public void closeElement(String element, HashMap<String, String> attributes,
1005                         String content, WarningSet warnings) {
1006                 name = content;
1007         }
1008
1009         @Override
1010         public void endHandler(String element, HashMap<String, String> attributes,
1011                         String content, WarningSet warnings) throws SAXException {
1012
1013                 String configid = attributes.remove("configid");
1014                 if (configid == null || configid.equals("")) {
1015                         warnings.add(Warning.FILE_INVALID_PARAMETER);
1016                         return;
1017                 }
1018
1019                 if (!rocket.addMotorConfigurationID(configid)) {
1020                         warnings.add("Duplicate motor configuration ID used.");
1021                         return;
1022                 }
1023
1024                 if (name != null && name.trim().length() > 0) {
1025                         rocket.setMotorConfigurationName(configid, name);
1026                 }
1027
1028                 if ("true".equals(attributes.remove("default"))) {
1029                         rocket.getDefaultConfiguration().setMotorConfigurationID(configid);
1030                 }
1031
1032                 super.closeElement(element, attributes, content, warnings);
1033         }
1034 }
1035
1036
1037 class MotorHandler extends AbstractElementHandler {
1038         /** File version where latest digest format was introduced */
1039         private static final int MOTOR_DIGEST_VERSION = 104;
1040
1041         private final DocumentLoadingContext context;
1042         private Motor.Type type = null;
1043         private String manufacturer = null;
1044         private String designation = null;
1045         private String digest = null;
1046         private double diameter = Double.NaN;
1047         private double length = Double.NaN;
1048         private double delay = Double.NaN;
1049
1050         public MotorHandler(DocumentLoadingContext context) {
1051                 this.context = context;
1052         }
1053
1054
1055         @Override
1056         public ElementHandler openElement(String element, HashMap<String, String> attributes,
1057                         WarningSet warnings) {
1058                 return PlainTextHandler.INSTANCE;
1059         }
1060
1061
1062         /**
1063          * Return the motor to use, or null.
1064          */
1065         public Motor getMotor(WarningSet warnings) {
1066                 return context.getMotorFinder().findMotor(type, manufacturer, designation, diameter, length, digest, warnings);
1067         }
1068
1069         /**
1070          * Return the delay to use for the motor.
1071          */
1072         public double getDelay(WarningSet warnings) {
1073                 if (Double.isNaN(delay)) {
1074                         warnings.add(Warning.fromString("Motor delay not specified, assuming no ejection charge."));
1075                         return Motor.PLUGGED;
1076                 }
1077                 return delay;
1078         }
1079
1080
1081         @Override
1082         public void closeElement(String element, HashMap<String, String> attributes,
1083                         String content, WarningSet warnings) throws SAXException {
1084
1085                 content = content.trim();
1086
1087                 if (element.equals("type")) {
1088
1089                         // Motor type
1090                         type = null;
1091                         for (Motor.Type t : Motor.Type.values()) {
1092                                 if (t.name().toLowerCase(Locale.ENGLISH).equals(content.trim())) {
1093                                         type = t;
1094                                         break;
1095                                 }
1096                         }
1097                         if (type == null) {
1098                                 warnings.add(Warning.fromString("Unknown motor type '" + content + "', ignoring."));
1099                         }
1100
1101                 } else if (element.equals("manufacturer")) {
1102
1103                         // Manufacturer
1104                         manufacturer = content.trim();
1105
1106                 } else if (element.equals("designation")) {
1107
1108                         // Designation
1109                         designation = content.trim();
1110
1111                 } else if (element.equals("digest")) {
1112
1113                         // Digest is used only for file versions saved using the same digest algorithm
1114                         if (context.getFileVersion() >= MOTOR_DIGEST_VERSION) {
1115                                 digest = content.trim();
1116                         }
1117
1118                 } else if (element.equals("diameter")) {
1119
1120                         // Diameter
1121                         diameter = Double.NaN;
1122                         try {
1123                                 diameter = Double.parseDouble(content.trim());
1124                         } catch (NumberFormatException e) {
1125                                 // Ignore
1126                         }
1127                         if (Double.isNaN(diameter)) {
1128                                 warnings.add(Warning.fromString("Illegal motor diameter specified, ignoring."));
1129                         }
1130
1131                 } else if (element.equals("length")) {
1132
1133                         // Length
1134                         length = Double.NaN;
1135                         try {
1136                                 length = Double.parseDouble(content.trim());
1137                         } catch (NumberFormatException ignore) {
1138                         }
1139
1140                         if (Double.isNaN(length)) {
1141                                 warnings.add(Warning.fromString("Illegal motor diameter specified, ignoring."));
1142                         }
1143
1144                 } else if (element.equals("delay")) {
1145
1146                         // Delay
1147                         delay = Double.NaN;
1148                         if (content.equals("none")) {
1149                                 delay = Motor.PLUGGED;
1150                         } else {
1151                                 try {
1152                                         delay = Double.parseDouble(content.trim());
1153                                 } catch (NumberFormatException ignore) {
1154                                 }
1155
1156                                 if (Double.isNaN(delay)) {
1157                                         warnings.add(Warning.fromString("Illegal motor delay specified, ignoring."));
1158                                 }
1159
1160                         }
1161
1162                 } else {
1163                         super.closeElement(element, attributes, content, warnings);
1164                 }
1165         }
1166
1167 }
1168
1169
1170
1171 class SimulationsHandler extends AbstractElementHandler {
1172         private final DocumentLoadingContext context;
1173         private final OpenRocketDocument doc;
1174         private SingleSimulationHandler handler;
1175
1176         public SimulationsHandler(OpenRocketDocument doc, DocumentLoadingContext context) {
1177                 this.doc = doc;
1178                 this.context = context;
1179         }
1180
1181         @Override
1182         public ElementHandler openElement(String element, HashMap<String, String> attributes,
1183                         WarningSet warnings) {
1184
1185                 if (!element.equals("simulation")) {
1186                         warnings.add("Unknown element '" + element + "', ignoring.");
1187                         return null;
1188                 }
1189
1190                 handler = new SingleSimulationHandler(doc, context);
1191                 return handler;
1192         }
1193
1194         @Override
1195         public void closeElement(String element, HashMap<String, String> attributes,
1196                         String content, WarningSet warnings) throws SAXException {
1197                 attributes.remove("status");
1198                 super.closeElement(element, attributes, content, warnings);
1199         }
1200
1201
1202 }
1203
1204 class SingleSimulationHandler extends AbstractElementHandler {
1205         private static final LogHelper log = Application.getLogger();
1206
1207         private final DocumentLoadingContext context;
1208
1209         private final OpenRocketDocument doc;
1210
1211         private String name;
1212
1213         private SimulationConditionsHandler conditionHandler;
1214         private FlightDataHandler dataHandler;
1215         private CustomExpressionsHandler customExpressionsHandler;
1216
1217         private ArrayList<CustomExpression> customExpressions = new ArrayList<CustomExpression>();
1218         private final List<String> listeners = new ArrayList<String>();
1219
1220         public SingleSimulationHandler(OpenRocketDocument doc, DocumentLoadingContext context) {
1221                 this.doc = doc;
1222                 this.context = context;
1223         }
1224
1225         public void setCustomExpressions(ArrayList<CustomExpression> expressions){
1226                 this.customExpressions = expressions;
1227         }
1228
1229         @Override
1230         public ElementHandler openElement(String element, HashMap<String, String> attributes,
1231                         WarningSet warnings) {
1232
1233                 if (element.equals("name") || element.equals("simulator") ||
1234                                 element.equals("calculator") || element.equals("listener")) {
1235                         return PlainTextHandler.INSTANCE;
1236                 } else if (element.equals("customexpressions")) {
1237                         customExpressionsHandler = new CustomExpressionsHandler(this, context);
1238                         return customExpressionsHandler;
1239                 } else if (element.equals("conditions")) {
1240                         conditionHandler = new SimulationConditionsHandler(doc.getRocket(), context);
1241                         return conditionHandler;
1242                 } else if (element.equals("flightdata")) {
1243                         dataHandler = new FlightDataHandler(context);
1244                         return dataHandler;
1245                 } else {
1246                         warnings.add("Unknown element '" + element + "', ignoring.");
1247                         return null;
1248                 }
1249         }
1250
1251         @Override
1252         public void closeElement(String element, HashMap<String, String> attributes,
1253                         String content, WarningSet warnings) {
1254
1255                 if (element.equals("name")) {
1256                         name = content;
1257                 } else if (element.equals("simulator")) {
1258                         if (!content.trim().equals("RK4Simulator")) {
1259                                 warnings.add("Unknown simulator '" + content.trim() + "' specified, ignoring.");
1260                         }
1261                 } else if (element.equals("calculator")) {
1262                         if (!content.trim().equals("BarrowmanCalculator")) {
1263                                 warnings.add("Unknown calculator '" + content.trim() + "' specified, ignoring.");
1264                         }
1265                 } else if (element.equals("listener") && content.trim().length() > 0) {
1266                         listeners.add(content.trim());
1267                 }
1268
1269         }
1270
1271         @Override
1272         public void endHandler(String element, HashMap<String, String> attributes,
1273                         String content, WarningSet warnings) {
1274
1275                 String s = attributes.get("status");
1276                 Simulation.Status status = (Status) DocumentConfig.findEnum(s, Simulation.Status.class);
1277                 if (status == null) {
1278                         warnings.add("Simulation status unknown, assuming outdated.");
1279                         status = Simulation.Status.OUTDATED;
1280                 }
1281
1282                 SimulationOptions conditions;
1283                 if (conditionHandler != null) {
1284                         conditions = conditionHandler.getConditions();
1285                 } else {
1286                         warnings.add("Simulation conditions not defined, using defaults.");
1287                         conditions = new SimulationOptions(doc.getRocket());
1288                 }
1289
1290                 if (name == null)
1291                         name = "Simulation";
1292
1293                 FlightData data;
1294                 if (dataHandler == null)
1295                         data = null;
1296                 else
1297                         data = dataHandler.getFlightData();
1298
1299                 Simulation simulation = new Simulation(doc.getRocket(), status, name,
1300                                 conditions, listeners, data);
1301
1302                 // Note : arraylist implementation in simulation different from standard one
1303                 for (CustomExpression exp : customExpressions){
1304                         exp.setSimulation(simulation);
1305                         if (exp.checkAll())
1306                                 simulation.addCustomExpression(exp);
1307                 }
1308
1309                 doc.addSimulation(simulation);
1310         }
1311 }
1312
1313 class CustomExpressionsHandler extends AbstractElementHandler {
1314         private final DocumentLoadingContext context;
1315         private final SingleSimulationHandler simHandler;
1316         public CustomExpression currentExpression = new CustomExpression();
1317         private final ArrayList<CustomExpression> customExpressions = new ArrayList<CustomExpression>();
1318
1319
1320         public CustomExpressionsHandler(SingleSimulationHandler simHandler, DocumentLoadingContext context) {
1321                 this.context = context;
1322                 this.simHandler = simHandler;
1323         }
1324
1325         @Override
1326         public ElementHandler openElement(String element,
1327                         HashMap<String, String> attributes, WarningSet warnings)
1328                         throws SAXException {
1329
1330                 if (element.equals("expression")){
1331                         currentExpression = new CustomExpression();
1332                 }
1333
1334                 return this;
1335         }
1336
1337         @Override
1338         public void closeElement(String element, HashMap<String, String> attributes,
1339                 String content, WarningSet warnings) {
1340
1341                 if (element.equals("expression"))
1342                         customExpressions.add(currentExpression);
1343
1344                 if (element.equals("name"))
1345                         currentExpression.setName(content);
1346
1347                 else if (element.equals("symbol"))
1348                         currentExpression.setSymbol(content);
1349
1350                 else if (element.equals("unit"))
1351                         currentExpression.setUnit(content);
1352
1353                 else if (element.equals("expressionstring"))
1354                         currentExpression.setExpression(content);
1355
1356         }
1357
1358         @Override
1359         public void endHandler(String element, HashMap<String, String> attributes,
1360                         String content, WarningSet warnings) {
1361                 simHandler.setCustomExpressions(customExpressions);
1362         }
1363 }
1364
1365 class SimulationConditionsHandler extends AbstractElementHandler {
1366         private final DocumentLoadingContext context;
1367         private SimulationOptions conditions;
1368         private AtmosphereHandler atmosphereHandler;
1369
1370         public SimulationConditionsHandler(Rocket rocket, DocumentLoadingContext context) {
1371                 this.context = context;
1372                 conditions = new SimulationOptions(rocket);
1373                 // Set up default loading settings (which may differ from the new defaults)
1374                 conditions.setGeodeticComputation(GeodeticComputationStrategy.FLAT);
1375         }
1376
1377         public SimulationOptions getConditions() {
1378                 return conditions;
1379         }
1380
1381         @Override
1382         public ElementHandler openElement(String element, HashMap<String, String> attributes,
1383                         WarningSet warnings) {
1384                 if (element.equals("atmosphere")) {
1385                         atmosphereHandler = new AtmosphereHandler(attributes.get("model"), context);
1386                         return atmosphereHandler;
1387                 }
1388                 return PlainTextHandler.INSTANCE;
1389         }
1390
1391         @Override
1392         public void closeElement(String element, HashMap<String, String> attributes,
1393                         String content, WarningSet warnings) {
1394
1395                 double d = Double.NaN;
1396                 try {
1397                         d = Double.parseDouble(content);
1398                 } catch (NumberFormatException ignore) {
1399                 }
1400
1401
1402                 if (element.equals("configid")) {
1403                         if (content.equals("")) {
1404                                 conditions.setMotorConfigurationID(null);
1405                         } else {
1406                                 conditions.setMotorConfigurationID(content);
1407                         }
1408                 } else if (element.equals("launchrodlength")) {
1409                         if (Double.isNaN(d)) {
1410                                 warnings.add("Illegal launch rod length defined, ignoring.");
1411                         } else {
1412                                 conditions.setLaunchRodLength(d);
1413                         }
1414                 } else if (element.equals("launchrodangle")) {
1415                         if (Double.isNaN(d)) {
1416                                 warnings.add("Illegal launch rod angle defined, ignoring.");
1417                         } else {
1418                                 conditions.setLaunchRodAngle(d * Math.PI / 180);
1419                         }
1420                 } else if (element.equals("launchroddirection")) {
1421                         if (Double.isNaN(d)) {
1422                                 warnings.add("Illegal launch rod direction defined, ignoring.");
1423                         } else {
1424                                 conditions.setLaunchRodDirection(d * Math.PI / 180);
1425                         }
1426                 } else if (element.equals("windaverage")) {
1427                         if (Double.isNaN(d)) {
1428                                 warnings.add("Illegal average windspeed defined, ignoring.");
1429                         } else {
1430                                 conditions.setWindSpeedAverage(d);
1431                         }
1432                 } else if (element.equals("windturbulence")) {
1433                         if (Double.isNaN(d)) {
1434                                 warnings.add("Illegal wind turbulence intensity defined, ignoring.");
1435                         } else {
1436                                 conditions.setWindTurbulenceIntensity(d);
1437                         }
1438                 } else if (element.equals("launchaltitude")) {
1439                         if (Double.isNaN(d)) {
1440                                 warnings.add("Illegal launch altitude defined, ignoring.");
1441                         } else {
1442                                 conditions.setLaunchAltitude(d);
1443                         }
1444                 } else if (element.equals("launchlatitude")) {
1445                         if (Double.isNaN(d)) {
1446                                 warnings.add("Illegal launch latitude defined, ignoring.");
1447                         } else {
1448                                 conditions.setLaunchLatitude(d);
1449                         }
1450                 } else if (element.equals("launchlongitude")) {
1451                         if (Double.isNaN(d)) {
1452                                 warnings.add("Illegal launch longitude.");
1453                         } else {
1454                                 conditions.setLaunchLongitude(d);
1455                         }
1456                 } else if (element.equals("geodeticmethod")) {
1457                         GeodeticComputationStrategy gcs =
1458                                         (GeodeticComputationStrategy) DocumentConfig.findEnum(content, GeodeticComputationStrategy.class);
1459                         if (gcs != null) {
1460                                 conditions.setGeodeticComputation(gcs);
1461                         } else {
1462                                 warnings.add("Unknown geodetic computation method '" + content + "'");
1463                         }
1464                 } else if (element.equals("atmosphere")) {
1465                         atmosphereHandler.storeSettings(conditions, warnings);
1466                 } else if (element.equals("timestep")) {
1467                         if (Double.isNaN(d)) {
1468                                 warnings.add("Illegal time step defined, ignoring.");
1469                         } else {
1470                                 conditions.setTimeStep(d);
1471                         }
1472                 }
1473         }
1474 }
1475
1476
1477 class AtmosphereHandler extends AbstractElementHandler {
1478         @SuppressWarnings("unused")
1479         private final DocumentLoadingContext context;
1480         private final String model;
1481         private double temperature = Double.NaN;
1482         private double pressure = Double.NaN;
1483
1484         public AtmosphereHandler(String model, DocumentLoadingContext context) {
1485                 this.model = model;
1486                 this.context = context;
1487         }
1488
1489         @Override
1490         public ElementHandler openElement(String element, HashMap<String, String> attributes,
1491                         WarningSet warnings) {
1492                 return PlainTextHandler.INSTANCE;
1493         }
1494
1495         @Override
1496         public void closeElement(String element, HashMap<String, String> attributes,
1497                         String content, WarningSet warnings) throws SAXException {
1498
1499                 double d = Double.NaN;
1500                 try {
1501                         d = Double.parseDouble(content);
1502                 } catch (NumberFormatException ignore) {
1503                 }
1504
1505                 if (element.equals("basetemperature")) {
1506                         if (Double.isNaN(d)) {
1507                                 warnings.add("Illegal base temperature specified, ignoring.");
1508                         }
1509                         temperature = d;
1510                 } else if (element.equals("basepressure")) {
1511                         if (Double.isNaN(d)) {
1512                                 warnings.add("Illegal base pressure specified, ignoring.");
1513                         }
1514                         pressure = d;
1515                 } else {
1516                         super.closeElement(element, attributes, content, warnings);
1517                 }
1518         }
1519
1520
1521         public void storeSettings(SimulationOptions cond, WarningSet warnings) {
1522                 if (!Double.isNaN(pressure)) {
1523                         cond.setLaunchPressure(pressure);
1524                 }
1525                 if (!Double.isNaN(temperature)) {
1526                         cond.setLaunchTemperature(temperature);
1527                 }
1528
1529                 if ("isa".equals(model)) {
1530                         cond.setISAAtmosphere(true);
1531                 } else if ("extendedisa".equals(model)) {
1532                         cond.setISAAtmosphere(false);
1533                 } else {
1534                         cond.setISAAtmosphere(true);
1535                         warnings.add("Unknown atmospheric model, using ISA.");
1536                 }
1537         }
1538
1539 }
1540
1541
1542 class FlightDataHandler extends AbstractElementHandler {
1543         private final DocumentLoadingContext context;
1544
1545         private FlightDataBranchHandler dataHandler;
1546         private WarningSet warningSet = new WarningSet();
1547         private List<FlightDataBranch> branches = new ArrayList<FlightDataBranch>();
1548
1549         private FlightData data;
1550
1551
1552         public FlightDataHandler(DocumentLoadingContext context) {
1553                 this.context = context;
1554         }
1555
1556         public FlightData getFlightData() {
1557                 return data;
1558         }
1559
1560         @Override
1561         public ElementHandler openElement(String element, HashMap<String, String> attributes,
1562                         WarningSet warnings) {
1563
1564                 if (element.equals("warning")) {
1565                         return PlainTextHandler.INSTANCE;
1566                 }
1567                 if (element.equals("databranch")) {
1568                         if (attributes.get("name") == null || attributes.get("types") == null) {
1569                                 warnings.add("Illegal flight data definition, ignoring.");
1570                                 return null;
1571                         }
1572                         dataHandler = new FlightDataBranchHandler(attributes.get("name"),
1573                                         attributes.get("types"), context);
1574                         return dataHandler;
1575                 }
1576
1577                 warnings.add("Unknown element '" + element + "' encountered, ignoring.");
1578                 return null;
1579         }
1580
1581
1582         @Override
1583         public void closeElement(String element, HashMap<String, String> attributes,
1584                         String content, WarningSet warnings) {
1585
1586                 if (element.equals("databranch")) {
1587                         FlightDataBranch branch = dataHandler.getBranch();
1588                         if (branch.getLength() > 0) {
1589                                 branches.add(branch);
1590                         }
1591                 } else if (element.equals("warning")) {
1592                         warningSet.add(Warning.fromString(content));
1593                 }
1594         }
1595
1596
1597         @Override
1598         public void endHandler(String element, HashMap<String, String> attributes,
1599                         String content, WarningSet warnings) {
1600
1601                 if (branches.size() > 0) {
1602                         data = new FlightData(branches.toArray(new FlightDataBranch[0]));
1603                 } else {
1604                         double maxAltitude = Double.NaN;
1605                         double maxVelocity = Double.NaN;
1606                         double maxAcceleration = Double.NaN;
1607                         double maxMach = Double.NaN;
1608                         double timeToApogee = Double.NaN;
1609                         double flightTime = Double.NaN;
1610                         double groundHitVelocity = Double.NaN;
1611                         double launchRodVelocity = Double.NaN;
1612                         double deploymentVelocity = Double.NaN;
1613
1614                         try {
1615                                 maxAltitude = DocumentConfig.stringToDouble(attributes.get("maxaltitude"));
1616                         } catch (NumberFormatException ignore) {
1617                         }
1618                         try {
1619                                 maxVelocity = DocumentConfig.stringToDouble(attributes.get("maxvelocity"));
1620                         } catch (NumberFormatException ignore) {
1621                         }
1622                         try {
1623                                 maxAcceleration = DocumentConfig.stringToDouble(attributes.get("maxacceleration"));
1624                         } catch (NumberFormatException ignore) {
1625                         }
1626                         try {
1627                                 maxMach = DocumentConfig.stringToDouble(attributes.get("maxmach"));
1628                         } catch (NumberFormatException ignore) {
1629                         }
1630                         try {
1631                                 timeToApogee = DocumentConfig.stringToDouble(attributes.get("timetoapogee"));
1632                         } catch (NumberFormatException ignore) {
1633                         }
1634                         try {
1635                                 flightTime = DocumentConfig.stringToDouble(attributes.get("flighttime"));
1636                         } catch (NumberFormatException ignore) {
1637                         }
1638                         try {
1639                                 groundHitVelocity =
1640                                                 DocumentConfig.stringToDouble(attributes.get("groundhitvelocity"));
1641                         } catch (NumberFormatException ignore) {
1642                         }
1643                         try {
1644                                 launchRodVelocity = DocumentConfig.stringToDouble(attributes.get("launchrodvelocity"));
1645                         } catch (NumberFormatException ignore) {
1646                         }
1647                         try {
1648                                 deploymentVelocity = DocumentConfig.stringToDouble(attributes.get("deploymentvelocity"));
1649                         } catch (NumberFormatException ignore) {
1650                         }
1651
1652                         data = new FlightData(maxAltitude, maxVelocity, maxAcceleration, maxMach,
1653                                         timeToApogee, flightTime, groundHitVelocity, launchRodVelocity, deploymentVelocity);
1654                 }
1655
1656                 data.getWarningSet().addAll(warningSet);
1657                 data.immute();
1658         }
1659
1660
1661 }
1662
1663
1664 class FlightDataBranchHandler extends AbstractElementHandler {
1665         @SuppressWarnings("unused")
1666         private final DocumentLoadingContext context;
1667         private final FlightDataType[] types;
1668         private final FlightDataBranch branch;
1669
1670         public FlightDataBranchHandler(String name, String typeList, DocumentLoadingContext context) {
1671                 this.context = context;
1672                 String[] split = typeList.split(",");
1673                 types = new FlightDataType[split.length];
1674                 for (int i = 0; i < split.length; i++) {
1675                         types[i] = FlightDataType.getType(split[i], "None ("+split[i]+")", UnitGroup.UNITS_NONE);
1676                         // TODO: HIGH: Deal with symbols
1677                 }
1678
1679                 // TODO: LOW: May throw an IllegalArgumentException
1680                 branch = new FlightDataBranch(name, types);
1681         }
1682
1683         public FlightDataBranch getBranch() {
1684                 branch.immute();
1685                 return branch;
1686         }
1687
1688         @Override
1689         public ElementHandler openElement(String element, HashMap<String, String> attributes,
1690                         WarningSet warnings) {
1691
1692                 if (element.equals("datapoint"))
1693                         return PlainTextHandler.INSTANCE;
1694                 if (element.equals("event"))
1695                         return PlainTextHandler.INSTANCE;
1696
1697                 warnings.add("Unknown element '" + element + "' encountered, ignoring.");
1698                 return null;
1699         }
1700
1701
1702         @Override
1703         public void closeElement(String element, HashMap<String, String> attributes,
1704                         String content, WarningSet warnings) {
1705
1706                 if (element.equals("event")) {
1707                         double time;
1708                         FlightEvent.Type type;
1709
1710                         try {
1711                                 time = DocumentConfig.stringToDouble(attributes.get("time"));
1712                         } catch (NumberFormatException e) {
1713                                 warnings.add("Illegal event specification, ignoring.");
1714                                 return;
1715                         }
1716
1717                         type = (Type) DocumentConfig.findEnum(attributes.get("type"), FlightEvent.Type.class);
1718                         if (type == null) {
1719                                 warnings.add("Illegal event specification, ignoring.");
1720                                 return;
1721                         }
1722
1723                         branch.addEvent(new FlightEvent(type, time));
1724                         return;
1725                 }
1726
1727                 if (!element.equals("datapoint")) {
1728                         warnings.add("Unknown element '" + element + "' encountered, ignoring.");
1729                         return;
1730                 }
1731
1732                 // element == "datapoint"
1733
1734
1735                 // Check line format
1736                 String[] split = content.split(",");
1737                 if (split.length != types.length) {
1738                         warnings.add("Data point did not contain correct amount of values, ignoring point.");
1739                         return;
1740                 }
1741
1742                 // Parse the doubles
1743                 double[] values = new double[split.length];
1744                 for (int i = 0; i < values.length; i++) {
1745                         try {
1746                                 values[i] = DocumentConfig.stringToDouble(split[i]);
1747                         } catch (NumberFormatException e) {
1748                                 warnings.add("Data point format error, ignoring point.");
1749                                 return;
1750                         }
1751                 }
1752
1753                 // Add point to branch
1754                 branch.addPoint();
1755                 for (int i = 0; i < types.length; i++) {
1756                         branch.setValue(types[i], values[i]);
1757                 }
1758         }
1759 }
1760
1761
1762
1763
1764
1765 /////////////////    Setters implementation
1766
1767
1768 ////  Interface
1769 interface Setter {
1770         /**
1771          * Set the specified value to the given component.
1772          *
1773          * @param component             the component to which to set.
1774          * @param value                 the value within the element.
1775          * @param attributes    attributes for the element.
1776          * @param warnings              the warning set to use.
1777          */
1778         public void set(RocketComponent component, String value,
1779                         HashMap<String, String> attributes, WarningSet warnings);
1780 }
1781
1782
1783 ////  StringSetter - sets the value to the contained String
1784 class StringSetter implements Setter {
1785         private final Reflection.Method setMethod;
1786
1787         public StringSetter(Reflection.Method set) {
1788                 setMethod = set;
1789         }
1790
1791         @Override
1792         public void set(RocketComponent c, String s, HashMap<String, String> attributes,
1793                         WarningSet warnings) {
1794                 setMethod.invoke(c, s);
1795         }
1796 }
1797
1798 ////  IntSetter - set an integer value
1799 class IntSetter implements Setter {
1800         private final Reflection.Method setMethod;
1801
1802         public IntSetter(Reflection.Method set) {
1803                 setMethod = set;
1804         }
1805
1806         @Override
1807         public void set(RocketComponent c, String s, HashMap<String, String> attributes,
1808                         WarningSet warnings) {
1809                 try {
1810                         int n = Integer.parseInt(s);
1811                         setMethod.invoke(c, n);
1812                 } catch (NumberFormatException e) {
1813                         warnings.add(Warning.FILE_INVALID_PARAMETER);
1814                 }
1815         }
1816 }
1817
1818
1819 //// BooleanSetter - set a boolean value
1820 class BooleanSetter implements Setter {
1821         private final Reflection.Method setMethod;
1822
1823         public BooleanSetter(Reflection.Method set) {
1824                 setMethod = set;
1825         }
1826
1827         @Override
1828         public void set(RocketComponent c, String s, HashMap<String, String> attributes,
1829                         WarningSet warnings) {
1830
1831                 s = s.trim();
1832                 if (s.equalsIgnoreCase("true")) {
1833                         setMethod.invoke(c, true);
1834                 } else if (s.equalsIgnoreCase("false")) {
1835                         setMethod.invoke(c, false);
1836                 } else {
1837                         warnings.add(Warning.FILE_INVALID_PARAMETER);
1838                 }
1839         }
1840 }
1841
1842
1843
1844 ////  DoubleSetter - sets a double value or (alternatively) if a specific string is encountered
1845 ////  calls a setXXX(boolean) method.
1846 class DoubleSetter implements Setter {
1847         private final Reflection.Method setMethod;
1848         private final String specialString;
1849         private final Reflection.Method specialMethod;
1850         private final double multiplier;
1851
1852         /**
1853          * Set only the double value.
1854          * @param set   set method for the double value.
1855          */
1856         public DoubleSetter(Reflection.Method set) {
1857                 this.setMethod = set;
1858                 this.specialString = null;
1859                 this.specialMethod = null;
1860                 this.multiplier = 1.0;
1861         }
1862
1863         /**
1864          * Multiply with the given multiplier and set the double value.
1865          * @param set   set method for the double value.
1866          * @param mul   multiplier.
1867          */
1868         public DoubleSetter(Reflection.Method set, double mul) {
1869                 this.setMethod = set;
1870                 this.specialString = null;
1871                 this.specialMethod = null;
1872                 this.multiplier = mul;
1873         }
1874
1875         /**
1876          * Set the double value, or if the value equals the special string, use the
1877          * special setter and set it to true.
1878          *
1879          * @param set                   double setter.
1880          * @param special               special string
1881          * @param specialMethod boolean setter.
1882          */
1883         public DoubleSetter(Reflection.Method set, String special,
1884                         Reflection.Method specialMethod) {
1885                 this.setMethod = set;
1886                 this.specialString = special;
1887                 this.specialMethod = specialMethod;
1888                 this.multiplier = 1.0;
1889         }
1890
1891
1892         @Override
1893         public void set(RocketComponent c, String s, HashMap<String, String> attributes,
1894                         WarningSet warnings) {
1895
1896                 s = s.trim();
1897
1898                 // Check for special case
1899                 if (specialMethod != null && s.equalsIgnoreCase(specialString)) {
1900                         specialMethod.invoke(c, true);
1901                         return;
1902                 }
1903
1904                 // Normal case
1905                 try {
1906                         double d = Double.parseDouble(s);
1907                         setMethod.invoke(c, d * multiplier);
1908                 } catch (NumberFormatException e) {
1909                         warnings.add(Warning.FILE_INVALID_PARAMETER);
1910                 }
1911         }
1912 }
1913
1914
1915 class OverrideSetter implements Setter {
1916         private final Reflection.Method setMethod;
1917         private final Reflection.Method enabledMethod;
1918
1919         public OverrideSetter(Reflection.Method set, Reflection.Method enabledMethod) {
1920                 this.setMethod = set;
1921                 this.enabledMethod = enabledMethod;
1922         }
1923
1924         @Override
1925         public void set(RocketComponent c, String s, HashMap<String, String> attributes,
1926                         WarningSet warnings) {
1927
1928                 try {
1929                         double d = Double.parseDouble(s);
1930                         setMethod.invoke(c, d);
1931                         enabledMethod.invoke(c, true);
1932                 } catch (NumberFormatException e) {
1933                         warnings.add(Warning.FILE_INVALID_PARAMETER);
1934                 }
1935         }
1936 }
1937
1938 ////  EnumSetter  -  sets a generic enum type
1939 class EnumSetter<T extends Enum<T>> implements Setter {
1940         private final Reflection.Method setter;
1941         private final Class<T> enumClass;
1942
1943         public EnumSetter(Reflection.Method set, Class<T> enumClass) {
1944                 this.setter = set;
1945                 this.enumClass = enumClass;
1946         }
1947
1948         @Override
1949         public void set(RocketComponent c, String name, HashMap<String, String> attributes,
1950                         WarningSet warnings) {
1951
1952                 Enum<?> setEnum = DocumentConfig.findEnum(name, enumClass);
1953                 if (setEnum == null) {
1954                         warnings.add(Warning.FILE_INVALID_PARAMETER);
1955                         return;
1956                 }
1957
1958                 setter.invoke(c, setEnum);
1959         }
1960 }
1961
1962
1963 ////  ColorSetter  -  sets a Color value
1964 class ColorSetter implements Setter {
1965         private final Reflection.Method setMethod;
1966
1967         public ColorSetter(Reflection.Method set) {
1968                 setMethod = set;
1969         }
1970
1971         @Override
1972         public void set(RocketComponent c, String s, HashMap<String, String> attributes,
1973                         WarningSet warnings) {
1974
1975                 String red = attributes.get("red");
1976                 String green = attributes.get("green");
1977                 String blue = attributes.get("blue");
1978
1979                 if (red == null || green == null || blue == null) {
1980                         warnings.add(Warning.FILE_INVALID_PARAMETER);
1981                         return;
1982                 }
1983
1984                 int r, g, b;
1985                 try {
1986                         r = Integer.parseInt(red);
1987                         g = Integer.parseInt(green);
1988                         b = Integer.parseInt(blue);
1989                 } catch (NumberFormatException e) {
1990                         warnings.add(Warning.FILE_INVALID_PARAMETER);
1991                         return;
1992                 }
1993
1994                 if (r < 0 || g < 0 || b < 0 || r > 255 || g > 255 || b > 255) {
1995                         warnings.add(Warning.FILE_INVALID_PARAMETER);
1996                         return;
1997                 }
1998
1999                 Color color = new Color(r, g, b);
2000                 setMethod.invoke(c, color);
2001
2002                 if (!s.trim().equals("")) {
2003                         warnings.add(Warning.FILE_INVALID_PARAMETER);
2004                 }
2005         }
2006 }
2007
2008 ////ComponentPresetSetter  -  sets a ComponentPreset value
2009 class ComponentPresetSetter implements Setter {
2010         private final Reflection.Method setMethod;
2011
2012         public ComponentPresetSetter(Reflection.Method set) {
2013                 this.setMethod = set;
2014         }
2015
2016         @Override
2017         public void set(RocketComponent c, String name, HashMap<String, String> attributes,
2018                         WarningSet warnings) {
2019 // FIXME - probably need more data in the warning messages - like what component preset...
2020                 String manufacturerName = attributes.get("manufacturer");
2021                 if ( manufacturerName == null ) {
2022                         warnings.add(Warning.fromString("Invalid ComponentPreset, no manufacturer specified.  Ignored"));
2023                         return;
2024                 }
2025
2026                 String productNo = attributes.get("partno");
2027                 if ( productNo == null ) {
2028                         warnings.add(Warning.fromString("Invalid ComponentPreset, no partno specified.  Ignored"));
2029                         return;
2030                 }
2031
2032                 String digest = attributes.get("digest");
2033                 if ( digest == null ) {
2034                         warnings.add(Warning.fromString("Invalid ComponentPreset, no digest specified."));
2035                 }
2036
2037                 String type = attributes.get("type");
2038                 if ( type == null ) {
2039                         warnings.add(Warning.fromString("Invalid ComponentPreset, no type specified."));
2040                 }
2041
2042                 List<ComponentPreset> presets = Application.getComponentPresetDao().find( manufacturerName, productNo );
2043
2044                 ComponentPreset matchingPreset = null;
2045
2046                 for( ComponentPreset preset: presets ) {
2047                         if ( digest != null && preset.getDigest().equals(digest) ) {
2048                                 // Found one with matching digest.  Take it.
2049                                 matchingPreset = preset;
2050                                 break;
2051                         }
2052                         if ( type != null && preset.getType().name().equals(type) && matchingPreset != null) {
2053                                 // Found the first one with matching type.
2054                                 matchingPreset = preset;
2055                         }
2056                 }
2057
2058                 // Was any found?
2059                 if ( matchingPreset == null ) {
2060                         warnings.add(Warning.fromString("No matching ComponentPreset found"));
2061                         return;
2062                 }
2063
2064                 if ( digest != null && !matchingPreset.getDigest().equals(digest) ) {
2065                         warnings.add(Warning.fromString("ComponentPreset has wrong digest"));
2066                 }
2067
2068                 setMethod.invoke(c, matchingPreset);
2069         }
2070 }
2071
2072
2073 ////MaterialSetter  -  sets a Material value
2074 class MaterialSetter implements Setter {
2075         private final Reflection.Method setMethod;
2076         private final Material.Type type;
2077
2078         public MaterialSetter(Reflection.Method set, Material.Type type) {
2079                 this.setMethod = set;
2080                 this.type = type;
2081         }
2082
2083         @Override
2084         public void set(RocketComponent c, String name, HashMap<String, String> attributes,
2085                         WarningSet warnings) {
2086
2087                 Material mat;
2088
2089                 // Check name != ""
2090                 name = name.trim();
2091                 if (name.equals("")) {
2092                         warnings.add(Warning.fromString("Illegal material specification, ignoring."));
2093                         return;
2094                 }
2095
2096                 // Parse density
2097                 double density;
2098                 String str;
2099                 str = attributes.remove("density");
2100                 if (str == null) {
2101                         warnings.add(Warning.fromString("Illegal material specification, ignoring."));
2102                         return;
2103                 }
2104                 try {
2105                         density = Double.parseDouble(str);
2106                 } catch (NumberFormatException e) {
2107                         warnings.add(Warning.fromString("Illegal material specification, ignoring."));
2108                         return;
2109                 }
2110
2111                 // Parse thickness
2112                 //              double thickness = 0;
2113                 //              str = attributes.remove("thickness");
2114                 //              try {
2115                 //                      if (str != null)
2116                 //                              thickness = Double.parseDouble(str);
2117                 //              } catch (NumberFormatException e){
2118                 //                      warnings.add(Warning.fromString("Illegal material specification, ignoring."));
2119                 //                      return;
2120                 //              }
2121
2122                 // Check type if specified
2123                 str = attributes.remove("type");
2124                 if (str != null && !type.name().toLowerCase(Locale.ENGLISH).equals(str)) {
2125                         warnings.add(Warning.fromString("Illegal material type specified, ignoring."));
2126                         return;
2127                 }
2128
2129                 mat = Databases.findMaterial(type, name, density, false);
2130
2131                 setMethod.invoke(c, mat);
2132         }
2133 }
2134
2135
2136
2137
2138 class PositionSetter implements Setter {
2139
2140         @Override
2141         public void set(RocketComponent c, String value, HashMap<String, String> attributes,
2142                         WarningSet warnings) {
2143
2144                 RocketComponent.Position type = (Position) DocumentConfig.findEnum(attributes.get("type"),
2145                                 RocketComponent.Position.class);
2146                 if (type == null) {
2147                         warnings.add(Warning.FILE_INVALID_PARAMETER);
2148                         return;
2149                 }
2150
2151                 double pos;
2152                 try {
2153                         pos = Double.parseDouble(value);
2154                 } catch (NumberFormatException e) {
2155                         warnings.add(Warning.FILE_INVALID_PARAMETER);
2156                         return;
2157                 }
2158
2159                 if (c instanceof FinSet) {
2160                         ((FinSet) c).setRelativePosition(type);
2161                         c.setPositionValue(pos);
2162                 } else if (c instanceof LaunchLug) {
2163                         ((LaunchLug) c).setRelativePosition(type);
2164                         c.setPositionValue(pos);
2165                 } else if (c instanceof InternalComponent) {
2166                         ((InternalComponent) c).setRelativePosition(type);
2167                         c.setPositionValue(pos);
2168                 } else {
2169                         warnings.add(Warning.FILE_INVALID_PARAMETER);
2170                 }
2171
2172         }
2173 }
2174
2175
2176 class FinTabPositionSetter extends DoubleSetter {
2177
2178         public FinTabPositionSetter() {
2179                 super(Reflection.findMethod(FinSet.class, "setTabShift", double.class));
2180         }
2181
2182         @Override
2183         public void set(RocketComponent c, String s, HashMap<String, String> attributes,
2184                         WarningSet warnings) {
2185
2186                 if (!(c instanceof FinSet)) {
2187                         throw new IllegalStateException("FinTabPositionSetter called for component " + c);
2188                 }
2189
2190                 String relative = attributes.get("relativeto");
2191                 FinSet.TabRelativePosition position =
2192                                 (TabRelativePosition) DocumentConfig.findEnum(relative,
2193                                                 FinSet.TabRelativePosition.class);
2194
2195                 if (position != null) {
2196
2197                         ((FinSet) c).setTabRelativePosition(position);
2198
2199                 } else {
2200                         if (relative == null) {
2201                                 warnings.add("Required attribute 'relativeto' not found for fin tab position.");
2202                         } else {
2203                                 warnings.add("Illegal attribute value '" + relative + "' encountered.");
2204                         }
2205                 }
2206
2207                 super.set(c, s, attributes, warnings);
2208         }
2209
2210
2211 }
2212
2213
2214 class ClusterConfigurationSetter implements Setter {
2215
2216         @Override
2217         public void set(RocketComponent component, String value, HashMap<String, String> attributes,
2218                         WarningSet warnings) {
2219
2220                 if (!(component instanceof Clusterable)) {
2221                         warnings.add("Illegal component defined as cluster.");
2222                         return;
2223                 }
2224
2225                 ClusterConfiguration config = null;
2226                 for (ClusterConfiguration c : ClusterConfiguration.CONFIGURATIONS) {
2227                         if (c.getXMLName().equals(value)) {
2228                                 config = c;
2229                                 break;
2230                         }
2231                 }
2232
2233                 if (config == null) {
2234                         warnings.add("Illegal cluster configuration specified.");
2235                         return;
2236                 }
2237
2238                 ((Clusterable) component).setClusterConfiguration(config);
2239         }
2240 }