2 package net.sf.openrocket.preset.xml;
4 import net.sf.openrocket.database.Databases;
5 import net.sf.openrocket.material.Material;
6 import net.sf.openrocket.motor.Manufacturer;
7 import net.sf.openrocket.preset.ComponentPreset;
8 import net.sf.openrocket.preset.InvalidComponentPresetException;
9 import net.sf.openrocket.preset.TypedPropertyMap;
10 import net.sf.openrocket.unit.UnitGroup;
12 import javax.imageio.ImageIO;
13 import javax.xml.bind.DatatypeConverter;
14 import javax.xml.bind.annotation.XmlAccessType;
15 import javax.xml.bind.annotation.XmlAccessorType;
16 import javax.xml.bind.annotation.XmlAttribute;
17 import javax.xml.bind.annotation.XmlElement;
18 import javax.xml.bind.annotation.XmlInlineBinaryData;
19 import javax.xml.bind.annotation.XmlValue;
20 import javax.xml.bind.annotation.adapters.XmlAdapter;
21 import javax.xml.bind.annotation.adapters.XmlJavaTypeAdapter;
22 import java.awt.image.BufferedImage;
23 import java.io.ByteArrayInputStream;
24 import java.io.ByteArrayOutputStream;
25 import java.io.IOException;
26 import java.util.List;
29 * Base class for the external representation of all component presets.
31 @XmlAccessorType(XmlAccessType.FIELD)
32 public abstract class BaseComponentDTO {
34 @XmlElement(name = "Manufacturer")
35 private String manufacturer;
36 @XmlElement(name = "PartNumber")
37 private String partNo;
38 @XmlElement(name = "Description")
39 private String description;
40 @XmlElement(name = "Material")
41 private AnnotatedMaterialDTO material;
42 @XmlElement(name = "Mass")
43 private AnnotatedMassDTO mass;
44 @XmlElement(name="Filled")
45 private Boolean filled;
47 @XmlJavaTypeAdapter(Base64Adapter.class)
48 @XmlElement(name = "Thumbnail")
52 * Default constructor.
54 protected BaseComponentDTO() {
60 * @param preset the preset to use to pull data values out of
62 * @throws net.sf.openrocket.util.BugException thrown if the expected body tube keys are not in the preset
64 protected BaseComponentDTO(final ComponentPreset preset) {
65 setManufacturer(preset.getManufacturer().getSimpleName());
66 setPartNo(preset.getPartNo());
67 if ( preset.has(ComponentPreset.DESCRIPTION )) {
68 setDescription(preset.get(ComponentPreset.DESCRIPTION));
70 if ( preset.has(ComponentPreset.MATERIAL)) {
71 setMaterial(new AnnotatedMaterialDTO(preset.get(ComponentPreset.MATERIAL)));
73 if (preset.has(ComponentPreset.MASS)) {
74 setMass(preset.get(ComponentPreset.MASS));
76 if ( preset.has(ComponentPreset.FILLED) ) {
77 setFilled( preset.get(ComponentPreset.FILLED));
81 public String getManufacturer() {
85 public void setManufacturer(final String theManufacturer) {
86 manufacturer = theManufacturer;
89 public String getPartNo() {
93 public void setPartNo(final String thePartNo) {
97 public String getDescription() {
101 public void setDescription(final String theDescription) {
102 description = theDescription;
105 public AnnotatedMaterialDTO getMaterial() {
109 public void setMaterial(final AnnotatedMaterialDTO theMaterial) {
110 material = theMaterial;
113 public double getMass() {
114 return mass.getValue();
117 public void setMass(final AnnotatedMassDTO theMass) {
121 public void setMass(final double theMass) {
122 mass = new AnnotatedMassDTO(theMass);
125 public Boolean getFilled() {
129 public void setFilled(Boolean filled) {
130 this.filled = filled;
133 public byte[] getImageData() {
137 public void setImageData(final byte[] theImage) {
141 public BufferedImage getImage() throws IOException {
143 return ImageIO.read(new ByteArrayInputStream(image));
148 public void setImage(BufferedImage theImage) throws IOException {
149 if (theImage != null) {
150 final ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
151 ImageIO.write(theImage, "png", byteArrayOutputStream);
152 image = byteArrayOutputStream.toByteArray();
156 public abstract ComponentPreset asComponentPreset(List<MaterialDTO> materials) throws InvalidComponentPresetException;
158 void addProps(TypedPropertyMap props, List<MaterialDTO> materialList) {
159 props.put(ComponentPreset.MANUFACTURER, Manufacturer.getManufacturer(manufacturer));
160 props.put(ComponentPreset.PARTNO, partNo);
161 if ( description != null ) {
162 props.put(ComponentPreset.DESCRIPTION, description);
164 Material m = find(materialList, material);
166 props.put(ComponentPreset.MATERIAL, find(materialList, material));
168 if ( mass != null ) {
169 props.put(ComponentPreset.MASS, getMass());
171 if ( filled != null ) {
172 props.put(ComponentPreset.FILLED, getFilled());
176 protected Material find(List<MaterialDTO> materialList, AnnotatedMaterialDTO dto) {
180 for (int i = 0; i < materialList.size(); i++) {
181 MaterialDTO materialDTO = materialList.get(i);
182 if (materialDTO.getType().name().equals(dto.type) && materialDTO.getName().equals(dto.material)) {
183 return materialDTO.asMaterial();
187 // Don't have one, build one.
189 if ( "BULK".equals( dto.type ) ) {
190 return new Material.Bulk(dto.material, 0.0, true);
191 } else if ( "SURFACE".equals( dto.type ) ) {
192 return new Material.Surface(dto.material, 0.0, true);
193 } else if ( "LINE".equals( dto.type ) ) {
194 return new Material.Line(dto.material, 0.0, true);
201 static class AnnotatedMaterialDTO {
202 @XmlAttribute(name = "Type")
205 private String material;
207 AnnotatedMaterialDTO() {}
209 AnnotatedMaterialDTO(Material theMaterial) {
210 type = theMaterial.getType().name();
211 material = theMaterial.getName();
215 static class AnnotatedLengthDTO {
216 @XmlAttribute(name="Unit", required=false)
217 private String unitName = "m";
219 private double length;
221 AnnotatedLengthDTO() {}
223 AnnotatedLengthDTO( double length ) {
224 this.length = length;
227 public double getValue() {
228 return UnitGroup.UNITS_LENGTH.getUnit(unitName).fromUnit(length);
232 static class AnnotatedMassDTO {
233 @XmlAttribute(name="Unit", required=false)
234 private String unitName = "kg";
238 AnnotatedMassDTO() {}
240 AnnotatedMassDTO( double mass ) {
244 public double getValue() {
245 return UnitGroup.UNITS_MASS.getUnit(unitName).fromUnit(mass);
249 static class Base64Adapter extends XmlAdapter<String, byte[]> {
250 public byte[] unmarshal(String s) {
254 return DatatypeConverter.parseBase64Binary(s);
257 public String marshal(byte[] bytes) {
261 return DatatypeConverter.printBase64Binary(bytes);