git.gag.com Git - debian/openrocket/blob - core/src/net/sf/openrocket/file/rocksim/export/RocksimSaver.java

   1 package net.sf.openrocket.file.rocksim.export;
   2
   3 import net.sf.openrocket.document.OpenRocketDocument;
   4 import net.sf.openrocket.document.StorageOptions;
   5 import net.sf.openrocket.file.RocketSaver;
   6 import net.sf.openrocket.file.rocksim.RocksimCommonConstants;
   7 import net.sf.openrocket.logging.LogHelper;
   8 import net.sf.openrocket.masscalc.BasicMassCalculator;
   9 import net.sf.openrocket.masscalc.MassCalculator;
  10 import net.sf.openrocket.rocketcomponent.Configuration;
  11 import net.sf.openrocket.rocketcomponent.Rocket;
  12 import net.sf.openrocket.rocketcomponent.Stage;
  13 import net.sf.openrocket.startup.Application;
  14
  15 import javax.xml.bind.JAXBContext;
  16 import javax.xml.bind.Marshaller;
  17 import java.io.BufferedWriter;
  18 import java.io.IOException;
  19 import java.io.OutputStream;
  20 import java.io.OutputStreamWriter;
  21 import java.io.StringWriter;
  22
  23 /**
  24  * This class is responsible for converting an OpenRocket design to a Rocksim design.
  25  */
  26 public class RocksimSaver extends RocketSaver {
  27
  28     /** The logger. */
  29     private static final LogHelper log = Application.getLogger();
  30
  31     /**
  32      * This method marshals an OpenRocketDocument (OR design) to Rocksim-compliant XML.
  33      *
  34      * @param doc  the OR design
  35      *
  36      * @return Rocksim-compliant XML
  37      */
  38     public String marshalToRocksim(OpenRocketDocument doc) {
  39
  40         try {
  41             JAXBContext binder = JAXBContext.newInstance(RocksimDocumentDTO.class);
  42             Marshaller marshaller = binder.createMarshaller();
  43             marshaller.setProperty(Marshaller.JAXB_FRAGMENT, Boolean.TRUE);
  44             marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, Boolean.TRUE);
  45             StringWriter sw = new StringWriter();
  46
  47             marshaller.marshal(toRocksimDocumentDTO(doc), sw);
  48             return sw.toString();
  49         } catch (Exception e) {
  50             e.printStackTrace();
  51         }
  52
  53         return null;
  54     }
  55
  56     @Override
  57     public void save(OutputStream dest, OpenRocketDocument doc, StorageOptions options) throws IOException {
  58         log.info("Saving .rkt file");
  59
  60         BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(dest, "UTF-8"));
  61         writer.write(marshalToRocksim(doc));
  62         writer.flush();
  63         writer.close();
  64     }
  65
  66     @Override
  67     public long estimateFileSize(OpenRocketDocument doc, StorageOptions options) {
  68         return marshalToRocksim(doc).length();
  69     }
  70
  71     /**
  72      * Root conversion method.  It iterates over all subcomponents.
  73      *
  74      * @param doc  the OR design
  75      *
  76      * @return a corresponding Rocksim representation
  77      */
  78     private RocksimDocumentDTO toRocksimDocumentDTO(OpenRocketDocument doc) {
  79         RocksimDocumentDTO rsd = new RocksimDocumentDTO();
  80
  81         rsd.setDesign(toRocksimDesignDTO(doc.getRocket()));
  82
  83         return rsd;
  84     }
  85
  86     private RocksimDesignDTO toRocksimDesignDTO(Rocket rocket) {
  87         RocksimDesignDTO result = new RocksimDesignDTO();
  88         result.setDesign(toRocketDesignDTO(rocket));
  89         return result;
  90     }
  91
  92     private RocketDesignDTO toRocketDesignDTO(Rocket rocket) {
  93         RocketDesignDTO result = new RocketDesignDTO();
  94
  95         MassCalculator massCalc = new BasicMassCalculator();
  96
  97             final double cg = massCalc.getCG(new Configuration(rocket), MassCalculator.MassCalcType.NO_MOTORS).x *
  98                     RocksimCommonConstants.ROCKSIM_TO_OPENROCKET_LENGTH;
  99         int stageCount = rocket.getStageCount();
 100         if (stageCount == 3) {
 101             result.setStage321CG(cg);
 102         }
 103         else if (stageCount == 2) {
 104             result.setStage32CG(cg);
 105         }
 106         else {
 107             result.setStage3CG(cg);
 108         }
 109
 110         result.setName(rocket.getName());
 111         result.setStageCount(stageCount);
 112         if (stageCount > 0) {
 113             result.setStage3(toStageDTO(rocket.getChild(0).getStage(), result, 3));
 114         }
 115         if (stageCount > 1) {
 116             result.setStage2(toStageDTO(rocket.getChild(1).getStage(), result, 2));
 117         }
 118         if (stageCount > 2) {
 119             result.setStage1(toStageDTO(rocket.getChild(2).getStage(), result, 1));
 120         }
 121         return result;
 122     }
 123
 124     private StageDTO toStageDTO(Stage stage, RocketDesignDTO designDTO, int stageNumber) {
 125         return new StageDTO(stage, designDTO, stageNumber);
 126     }
 127
 128 }