1 package net.sf.openrocket.gui.dialogs.motor.thrustcurve;
3 import java.awt.BasicStroke;
5 import java.awt.Component;
6 import java.awt.Cursor;
9 import java.awt.Rectangle;
10 import java.awt.event.ActionEvent;
11 import java.awt.event.ActionListener;
12 import java.awt.event.MouseAdapter;
13 import java.awt.event.MouseEvent;
14 import java.util.ArrayList;
15 import java.util.Collections;
16 import java.util.List;
17 import java.util.prefs.Preferences;
19 import javax.swing.BorderFactory;
20 import javax.swing.DefaultComboBoxModel;
21 import javax.swing.JCheckBox;
22 import javax.swing.JComboBox;
23 import javax.swing.JComponent;
24 import javax.swing.JLabel;
25 import javax.swing.JLayeredPane;
26 import javax.swing.JList;
27 import javax.swing.JPanel;
28 import javax.swing.JScrollPane;
29 import javax.swing.JSeparator;
30 import javax.swing.JTable;
31 import javax.swing.JTextArea;
32 import javax.swing.JTextField;
33 import javax.swing.ListCellRenderer;
34 import javax.swing.ListSelectionModel;
35 import javax.swing.RowFilter;
36 import javax.swing.SwingUtilities;
37 import javax.swing.event.DocumentEvent;
38 import javax.swing.event.DocumentListener;
39 import javax.swing.event.ListSelectionEvent;
40 import javax.swing.event.ListSelectionListener;
41 import javax.swing.table.TableModel;
42 import javax.swing.table.TableRowSorter;
44 import net.miginfocom.swing.MigLayout;
45 import net.sf.openrocket.database.ThrustCurveMotorSet;
46 import net.sf.openrocket.database.ThrustCurveMotorSetDatabase;
47 import net.sf.openrocket.gui.components.StyledLabel;
48 import net.sf.openrocket.gui.components.StyledLabel.Style;
49 import net.sf.openrocket.gui.dialogs.motor.CloseableDialog;
50 import net.sf.openrocket.gui.dialogs.motor.MotorSelector;
51 import net.sf.openrocket.gui.util.GUIUtil;
52 import net.sf.openrocket.gui.util.Icons;
53 import net.sf.openrocket.gui.util.SwingPreferences;
54 import net.sf.openrocket.l10n.Translator;
55 import net.sf.openrocket.logging.LogHelper;
56 import net.sf.openrocket.motor.Motor;
57 import net.sf.openrocket.motor.MotorDigest;
58 import net.sf.openrocket.motor.ThrustCurveMotor;
59 import net.sf.openrocket.startup.Application;
60 import net.sf.openrocket.unit.UnitGroup;
61 import net.sf.openrocket.util.BugException;
62 import net.sf.openrocket.utils.MotorCorrelation;
64 import org.jfree.chart.ChartColor;
65 import org.jfree.chart.ChartFactory;
66 import org.jfree.chart.ChartPanel;
67 import org.jfree.chart.JFreeChart;
68 import org.jfree.chart.axis.ValueAxis;
69 import org.jfree.chart.plot.PlotOrientation;
70 import org.jfree.chart.plot.XYPlot;
71 import org.jfree.chart.title.TextTitle;
72 import org.jfree.data.xy.XYSeries;
73 import org.jfree.data.xy.XYSeriesCollection;
75 public class ThrustCurveMotorSelectionPanel extends JPanel implements MotorSelector {
76 private static final LogHelper log = Application.getLogger();
77 private static final Translator trans = Application.getTranslator();
79 private static final double MOTOR_SIMILARITY_THRESHOLD = 0.95;
81 private static final int SHOW_ALL = 0;
82 private static final int SHOW_SMALLER = 1;
83 private static final int SHOW_EXACT = 2;
84 private static final String[] SHOW_DESCRIPTIONS = {
86 trans.get("TCMotorSelPan.SHOW_DESCRIPTIONS.desc1"),
87 //// Show motors with diameter less than that of the motor mount
88 trans.get("TCMotorSelPan.SHOW_DESCRIPTIONS.desc2"),
89 //// Show motors with diameter equal to that of the motor mount
90 trans.get("TCMotorSelPan.SHOW_DESCRIPTIONS.desc3")
92 private static final int SHOW_MAX = 2;
94 private static final int ZOOM_ICON_POSITION_NEGATIVE_X = 50;
95 private static final int ZOOM_ICON_POSITION_POSITIVE_Y = 12;
97 private static final Paint[] CURVE_COLORS = ChartColor.createDefaultPaintArray();
99 private static final Color NO_COMMENT_COLOR = Color.GRAY;
100 private static final Color WITH_COMMENT_COLOR = Color.BLACK;
102 private static final ThrustCurveMotorComparator MOTOR_COMPARATOR = new ThrustCurveMotorComparator();
106 private final List<ThrustCurveMotorSet> database;
108 private final double diameter;
109 private CloseableDialog dialog = null;
112 private final ThrustCurveMotorDatabaseModel model;
113 private final JTable table;
114 private final TableRowSorter<TableModel> sorter;
116 private final JCheckBox hideSimilarBox;
118 private final JTextField searchField;
119 private String[] searchTerms = new String[0];
122 private final JLabel curveSelectionLabel;
123 private final JComboBox curveSelectionBox;
124 private final DefaultComboBoxModel curveSelectionModel;
126 private final JLabel totalImpulseLabel;
127 private final JLabel avgThrustLabel;
128 private final JLabel maxThrustLabel;
129 private final JLabel burnTimeLabel;
130 private final JLabel launchMassLabel;
131 private final JLabel emptyMassLabel;
132 private final JLabel dataPointsLabel;
133 private final JLabel digestLabel;
135 private final JTextArea comment;
136 private final Font noCommentFont;
137 private final Font withCommentFont;
139 private final JFreeChart chart;
140 private final ChartPanel chartPanel;
141 private final JLabel zoomIcon;
143 private final JComboBox delayBox;
145 private ThrustCurveMotor selectedMotor;
146 private ThrustCurveMotorSet selectedMotorSet;
147 private double selectedDelay;
153 * @param current the currently selected ThrustCurveMotor, or <code>null</code> for none.
154 * @param delay the currently selected ejection charge delay.
155 * @param diameter the diameter of the motor mount.
157 public ThrustCurveMotorSelectionPanel(ThrustCurveMotor current, double delay, double diameter) {
158 super(new MigLayout("fill", "[grow][]"));
160 this.diameter = diameter;
163 // Construct the database (adding the current motor if not in the db already)
164 List<ThrustCurveMotorSet> db;
165 // TODO - ugly blind cast.
166 db = ((ThrustCurveMotorSetDatabase) Application.getMotorSetDatabase()).getMotorSets();
168 // If current motor is not found in db, add a new ThrustCurveMotorSet containing it
169 if (current != null) {
170 selectedMotor = current;
171 for (ThrustCurveMotorSet motorSet : db) {
172 if (motorSet.getMotors().contains(current)) {
173 selectedMotorSet = motorSet;
177 if (selectedMotorSet == null) {
178 db = new ArrayList<ThrustCurveMotorSet>(db);
179 ThrustCurveMotorSet extra = new ThrustCurveMotorSet();
180 extra.addMotor(current);
181 selectedMotorSet = extra;
183 Collections.sort(db);
195 panel = new JPanel(new MigLayout("fill"));
196 this.add(panel, "grow");
201 //// Select rocket motor:
202 label = new StyledLabel(trans.get("TCMotorSelPan.lbl.Selrocketmotor"), Style.BOLD);
203 panel.add(label, "spanx, wrap para");
205 // Diameter selection
206 JComboBox filterComboBox = new JComboBox(SHOW_DESCRIPTIONS);
207 filterComboBox.addActionListener(new ActionListener() {
209 public void actionPerformed(ActionEvent e) {
210 JComboBox cb = (JComboBox) e.getSource();
211 int sel = cb.getSelectedIndex();
212 if ((sel < 0) || (sel > SHOW_MAX))
216 sorter.setRowFilter(new MotorRowFilterAll());
220 sorter.setRowFilter(new MotorRowFilterSmaller());
224 sorter.setRowFilter(new MotorRowFilterExact());
228 throw new BugException("Invalid selection mode sel=" + sel);
230 Application.getPreferences().putChoice("MotorDiameterMatch", sel);
231 scrollSelectionVisible();
234 panel.add(filterComboBox, "spanx, growx, wrap rel");
236 //// Hide very similar thrust curves
237 hideSimilarBox = new JCheckBox(trans.get("TCMotorSelPan.checkbox.hideSimilar"));
238 GUIUtil.changeFontSize(hideSimilarBox, -1);
239 hideSimilarBox.setSelected(Application.getPreferences().getBoolean(net.sf.openrocket.startup.Preferences.MOTOR_HIDE_SIMILAR, true));
240 hideSimilarBox.addActionListener(new ActionListener() {
242 public void actionPerformed(ActionEvent e) {
243 Application.getPreferences().putBoolean(net.sf.openrocket.startup.Preferences.MOTOR_HIDE_SIMILAR, hideSimilarBox.isSelected());
247 panel.add(hideSimilarBox, "gapleft para, spanx, growx, wrap para");
250 // Motor selection table
251 model = new ThrustCurveMotorDatabaseModel(database);
252 table = new JTable(model);
255 // Set comparators and widths
256 table.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
257 sorter = new TableRowSorter<TableModel>(model);
258 for (int i = 0; i < ThrustCurveMotorColumns.values().length; i++) {
259 ThrustCurveMotorColumns column = ThrustCurveMotorColumns.values()[i];
260 sorter.setComparator(i, column.getComparator());
261 table.getColumnModel().getColumn(i).setPreferredWidth(column.getWidth());
263 table.setRowSorter(sorter);
265 // Set selection and double-click listeners
266 table.getSelectionModel().addListSelectionListener(new ListSelectionListener() {
268 public void valueChanged(ListSelectionEvent e) {
269 int row = table.getSelectedRow();
271 row = table.convertRowIndexToModel(row);
272 ThrustCurveMotorSet motorSet = model.getMotorSet(row);
273 log.user("Selected table row " + row + ": " + motorSet);
274 if (motorSet != selectedMotorSet) {
275 select(selectMotor(motorSet));
278 log.user("Selected table row " + row + ", nothing selected");
282 table.addMouseListener(new MouseAdapter() {
284 public void mouseClicked(MouseEvent e) {
285 if (e.getButton() == MouseEvent.BUTTON1 && e.getClickCount() == 2) {
286 if (dialog != null) {
294 JScrollPane scrollpane = new JScrollPane();
295 scrollpane.setViewportView(table);
296 panel.add(scrollpane, "grow, width :500:, height :300:, spanx, wrap para");
301 // Motor mount diameter label
302 //// Motor mount diameter:
303 label = new StyledLabel(trans.get("TCMotorSelPan.lbl.Motormountdia")+ " " +
304 UnitGroup.UNITS_MOTOR_DIMENSIONS.getDefaultUnit().toStringUnit(diameter));
305 panel.add(label, "gapright 30lp, spanx, split");
311 label = new StyledLabel(trans.get("TCMotorSelPan.lbl.Search"));
312 panel.add(label, "");
314 searchField = new JTextField();
315 searchField.getDocument().addDocumentListener(new DocumentListener() {
317 public void changedUpdate(DocumentEvent e) {
322 public void insertUpdate(DocumentEvent e) {
327 public void removeUpdate(DocumentEvent e) {
331 private void update() {
332 String text = searchField.getText().trim();
333 String[] split = text.split("\\s+");
334 ArrayList<String> list = new ArrayList<String>();
335 for (String s : split) {
336 s = s.trim().toLowerCase();
337 if (s.length() > 0) {
341 searchTerms = list.toArray(new String[0]);
343 scrollSelectionVisible();
346 panel.add(searchField, "growx, wrap");
351 this.add(panel, "grow");
352 this.add(new JSeparator(JSeparator.VERTICAL), "growy, gap para para");
353 panel = new JPanel(new MigLayout("fill"));
357 // Thrust curve selection
358 //// Select thrust curve:
359 curveSelectionLabel = new JLabel(trans.get("TCMotorSelPan.lbl.Selectthrustcurve"));
360 panel.add(curveSelectionLabel);
362 curveSelectionModel = new DefaultComboBoxModel();
363 curveSelectionBox = new JComboBox(curveSelectionModel);
364 curveSelectionBox.setRenderer(new CurveSelectionRenderer(curveSelectionBox.getRenderer()));
365 curveSelectionBox.addActionListener(new ActionListener() {
367 public void actionPerformed(ActionEvent e) {
368 Object value = curveSelectionBox.getSelectedItem();
370 select(((MotorHolder) value).getMotor());
374 panel.add(curveSelectionBox, "growx, wrap para");
380 // Ejection charge delay:
381 panel.add(new JLabel(trans.get("TCMotorSelPan.lbl.Ejectionchargedelay")));
383 delayBox = new JComboBox();
384 delayBox.setEditable(true);
385 delayBox.addActionListener(new ActionListener() {
387 public void actionPerformed(ActionEvent e) {
388 JComboBox cb = (JComboBox) e.getSource();
389 String sel = (String) cb.getSelectedItem();
391 if (sel.equalsIgnoreCase(trans.get("TCMotorSelPan.equalsIgnoreCase.None"))) {
392 selectedDelay = Motor.PLUGGED;
395 selectedDelay = Double.parseDouble(sel);
396 } catch (NumberFormatException ignore) {
402 panel.add(delayBox, "growx, wrap rel");
403 //// (Number of seconds or \"None\")
404 panel.add(new StyledLabel(trans.get("TCMotorSelPan.lbl.NumberofsecondsorNone"), -3), "skip, wrap para");
408 panel.add(new JSeparator(), "spanx, growx, wrap para");
414 panel.add(new JLabel(trans.get("TCMotorSelPan.lbl.Totalimpulse")));
415 totalImpulseLabel = new JLabel();
416 panel.add(totalImpulseLabel, "wrap");
419 panel.add(new JLabel(trans.get("TCMotorSelPan.lbl.Avgthrust")));
420 avgThrustLabel = new JLabel();
421 panel.add(avgThrustLabel, "wrap");
424 panel.add(new JLabel(trans.get("TCMotorSelPan.lbl.Maxthrust")));
425 maxThrustLabel = new JLabel();
426 panel.add(maxThrustLabel, "wrap");
429 panel.add(new JLabel(trans.get("TCMotorSelPan.lbl.Burntime")));
430 burnTimeLabel = new JLabel();
431 panel.add(burnTimeLabel, "wrap");
434 panel.add(new JLabel(trans.get("TCMotorSelPan.lbl.Launchmass")));
435 launchMassLabel = new JLabel();
436 panel.add(launchMassLabel, "wrap");
439 panel.add(new JLabel(trans.get("TCMotorSelPan.lbl.Emptymass")));
440 emptyMassLabel = new JLabel();
441 panel.add(emptyMassLabel, "wrap");
444 panel.add(new JLabel(trans.get("TCMotorSelPan.lbl.Datapoints")));
445 dataPointsLabel = new JLabel();
446 panel.add(dataPointsLabel, "wrap para");
448 if (System.getProperty("openrocket.debug.motordigest") != null) {
450 panel.add(new JLabel(trans.get("TCMotorSelPan.lbl.Digest")));
451 digestLabel = new JLabel();
452 panel.add(digestLabel, "w :300:, wrap para");
458 comment = new JTextArea(5, 5);
459 GUIUtil.changeFontSize(comment, -2);
460 withCommentFont = comment.getFont();
461 noCommentFont = withCommentFont.deriveFont(Font.ITALIC);
462 comment.setLineWrap(true);
463 comment.setWrapStyleWord(true);
464 comment.setEditable(false);
465 scrollpane = new JScrollPane(comment);
466 panel.add(scrollpane, "spanx, growx, wrap para");
472 chart = ChartFactory.createXYLineChart(
477 PlotOrientation.VERTICAL,
484 // Add the data and formatting to the plot
485 XYPlot plot = chart.getXYPlot();
487 changeLabelFont(plot.getRangeAxis(), -2);
488 changeLabelFont(plot.getDomainAxis(), -2);
491 chart.setTitle(new TextTitle(trans.get("TCMotorSelPan.title.Thrustcurve"), this.getFont()));
492 chart.setBackgroundPaint(this.getBackground());
493 plot.setBackgroundPaint(Color.WHITE);
494 plot.setDomainGridlinePaint(Color.LIGHT_GRAY);
495 plot.setRangeGridlinePaint(Color.LIGHT_GRAY);
497 chartPanel = new ChartPanel(chart,
503 chartPanel.setMouseZoomable(false);
504 chartPanel.setPopupMenu(null);
505 chartPanel.setMouseWheelEnabled(false);
506 chartPanel.setRangeZoomable(false);
507 chartPanel.setDomainZoomable(false);
509 chartPanel.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));
510 chartPanel.addMouseListener(new MouseAdapter() {
512 public void mouseClicked(MouseEvent e) {
513 if (selectedMotor == null || selectedMotorSet == null)
515 if (e.getButton() == MouseEvent.BUTTON1) {
517 List<ThrustCurveMotor> motors = getFilteredCurves();
518 ThrustCurveMotorPlotDialog plotDialog = new ThrustCurveMotorPlotDialog(motors,
519 motors.indexOf(selectedMotor),
520 SwingUtilities.getWindowAncestor(ThrustCurveMotorSelectionPanel.this));
521 plotDialog.setVisible(true);
526 JLayeredPane layer = new CustomLayeredPane();
528 layer.setBorder(BorderFactory.createLineBorder(Color.BLUE));
530 layer.add(chartPanel, (Integer) 0);
532 zoomIcon = new JLabel(Icons.ZOOM_IN);
533 zoomIcon.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));
534 layer.add(zoomIcon, (Integer) 1);
537 panel.add(layer, "width 300:300:, height 180:180:, grow, spanx");
541 this.add(panel, "grow");
546 int showMode = Application.getPreferences().getChoice(net.sf.openrocket.startup.Preferences.MOTOR_DIAMETER_FILTER, SHOW_MAX, SHOW_EXACT);
547 filterComboBox.setSelectedIndex(showMode);
550 // Update the panel data
552 selectedDelay = delay;
559 public Motor getSelectedMotor() {
560 return selectedMotor;
565 public double getSelectedDelay() {
566 return selectedDelay;
571 public JComponent getDefaultFocus() {
576 public void selectedMotor(Motor motorSelection) {
577 if (!(motorSelection instanceof ThrustCurveMotor)) {
578 log.error("Received argument that was not ThrustCurveMotor: " + motorSelection);
582 ThrustCurveMotor motor = (ThrustCurveMotor) motorSelection;
583 ThrustCurveMotorSet set = findMotorSet(motor);
585 log.error("Could not find set for motor:" + motorSelection);
589 // Store selected motor in preferences node, set all others to false
590 Preferences prefs = ((SwingPreferences) Application.getPreferences()).getNode(net.sf.openrocket.startup.Preferences.PREFERRED_THRUST_CURVE_MOTOR_NODE);
591 for (ThrustCurveMotor m : set.getMotors()) {
592 String digest = m.getDigest();
593 prefs.putBoolean(digest, m == motor);
597 public void setCloseableDialog(CloseableDialog dialog) {
598 this.dialog = dialog;
603 private void changeLabelFont(ValueAxis axis, float size) {
604 Font font = axis.getTickLabelFont();
605 font = font.deriveFont(font.getSize2D() + size);
606 axis.setTickLabelFont(font);
611 * Called when a different motor is selected from within the panel.
613 private void select(ThrustCurveMotor motor) {
614 if (selectedMotor == motor)
617 ThrustCurveMotorSet set = findMotorSet(motor);
619 throw new BugException("Could not find motor from database, motor=" + motor);
622 boolean updateDelays = (selectedMotorSet != set);
624 selectedMotor = motor;
625 selectedMotorSet = set;
633 private void updateData() {
635 if (selectedMotorSet == null) {
637 curveSelectionModel.removeAllElements();
638 curveSelectionBox.setEnabled(false);
639 curveSelectionLabel.setEnabled(false);
640 totalImpulseLabel.setText("");
641 avgThrustLabel.setText("");
642 maxThrustLabel.setText("");
643 burnTimeLabel.setText("");
644 launchMassLabel.setText("");
645 emptyMassLabel.setText("");
646 dataPointsLabel.setText("");
647 if (digestLabel != null) {
648 digestLabel.setText("");
651 chart.getXYPlot().setDataset(new XYSeriesCollection());
656 // Check which thrust curves to display
657 List<ThrustCurveMotor> motors = getFilteredCurves();
658 final int index = motors.indexOf(selectedMotor);
661 // Update the thrust curve selection box
662 curveSelectionModel.removeAllElements();
663 for (int i = 0; i < motors.size(); i++) {
664 curveSelectionModel.addElement(new MotorHolder(motors.get(i), i));
666 curveSelectionBox.setSelectedIndex(index);
668 if (motors.size() > 1) {
669 curveSelectionBox.setEnabled(true);
670 curveSelectionLabel.setEnabled(true);
672 curveSelectionBox.setEnabled(false);
673 curveSelectionLabel.setEnabled(false);
677 // Update thrust curve data
678 totalImpulseLabel.setText(UnitGroup.UNITS_IMPULSE.getDefaultUnit().toStringUnit(
679 selectedMotor.getTotalImpulseEstimate()));
680 avgThrustLabel.setText(UnitGroup.UNITS_FORCE.getDefaultUnit().toStringUnit(
681 selectedMotor.getAverageThrustEstimate()));
682 maxThrustLabel.setText(UnitGroup.UNITS_FORCE.getDefaultUnit().toStringUnit(
683 selectedMotor.getMaxThrustEstimate()));
684 burnTimeLabel.setText(UnitGroup.UNITS_SHORT_TIME.getDefaultUnit().toStringUnit(
685 selectedMotor.getBurnTimeEstimate()));
686 launchMassLabel.setText(UnitGroup.UNITS_MASS.getDefaultUnit().toStringUnit(
687 selectedMotor.getLaunchCG().weight));
688 emptyMassLabel.setText(UnitGroup.UNITS_MASS.getDefaultUnit().toStringUnit(
689 selectedMotor.getEmptyCG().weight));
690 dataPointsLabel.setText("" + (selectedMotor.getTimePoints().length - 1));
691 if (digestLabel != null) {
692 digestLabel.setText(selectedMotor.getDigest());
695 setComment(selectedMotor.getDescription());
699 XYPlot plot = chart.getXYPlot();
701 XYSeriesCollection dataset = new XYSeriesCollection();
702 for (int i = 0; i < motors.size(); i++) {
703 ThrustCurveMotor m = motors.get(i);
706 XYSeries series = new XYSeries(trans.get("TCMotorSelPan.title.Thrust"));
707 double[] time = m.getTimePoints();
708 double[] thrust = m.getThrustPoints();
710 for (int j = 0; j < time.length; j++) {
711 series.add(time[j], thrust[j]);
714 dataset.addSeries(series);
716 boolean selected = (i == index);
717 plot.getRenderer().setSeriesStroke(i, new BasicStroke(selected ? 3 : 1));
718 plot.getRenderer().setSeriesPaint(i, getColor(i));
721 plot.setDataset(dataset);
725 private List<ThrustCurveMotor> getFilteredCurves() {
726 List<ThrustCurveMotor> motors = selectedMotorSet.getMotors();
727 if (hideSimilarBox.isSelected()) {
728 List<ThrustCurveMotor> filtered = new ArrayList<ThrustCurveMotor>(motors.size());
729 for (int i = 0; i < motors.size(); i++) {
730 ThrustCurveMotor m = motors.get(i);
731 if (m.equals(selectedMotor)) {
736 double similarity = MotorCorrelation.similarity(selectedMotor, m);
737 log.debug("Motor similarity: " + similarity);
738 if (similarity < MOTOR_SIMILARITY_THRESHOLD) {
745 Collections.sort(motors, MOTOR_COMPARATOR);
751 private void setComment(String s) {
753 if (s.length() == 0) {
754 //// No description available.
755 comment.setText("No description available.");
756 comment.setFont(noCommentFont);
757 comment.setForeground(NO_COMMENT_COLOR);
760 comment.setFont(withCommentFont);
761 comment.setForeground(WITH_COMMENT_COLOR);
763 comment.setCaretPosition(0);
766 private void scrollSelectionVisible() {
767 if (selectedMotorSet != null) {
768 int index = table.convertRowIndexToView(model.getIndex(selectedMotorSet));
769 System.out.println("index=" + index);
770 table.getSelectionModel().setSelectionInterval(index, index);
771 Rectangle rect = table.getCellRect(index, 0, true);
772 rect = new Rectangle(rect.x, rect.y - 100, rect.width, rect.height + 200);
773 table.scrollRectToVisible(rect);
778 public static Color getColor(int index) {
779 return (Color) CURVE_COLORS[index % CURVE_COLORS.length];
784 * Find the ThrustCurveMotorSet that contains a motor.
786 * @param motor the motor to look for.
787 * @return the ThrustCurveMotorSet, or null if not found.
789 private ThrustCurveMotorSet findMotorSet(ThrustCurveMotor motor) {
790 for (ThrustCurveMotorSet set : database) {
791 if (set.getMotors().contains(motor)) {
802 * Select the default motor from this ThrustCurveMotorSet. This uses primarily motors
803 * that the user has previously used, and secondarily a heuristic method of selecting which
804 * thrust curve seems to be better or more reliable.
806 * @param set the motor set
807 * @return the default motor in this set
809 private ThrustCurveMotor selectMotor(ThrustCurveMotorSet set) {
810 if (set.getMotorCount() == 0) {
811 throw new BugException("Attempting to select motor from empty ThrustCurveMotorSet: " + set);
813 if (set.getMotorCount() == 1) {
814 return set.getMotors().get(0);
818 // Find which motor has been used the most recently
819 List<ThrustCurveMotor> list = set.getMotors();
820 Preferences prefs = ((SwingPreferences) Application.getPreferences()).getNode(net.sf.openrocket.startup.Preferences.PREFERRED_THRUST_CURVE_MOTOR_NODE);
821 for (ThrustCurveMotor m : list) {
822 String digest = m.getDigest();
823 if (prefs.getBoolean(digest, false)) {
828 // No motor has been used
829 Collections.sort(list, MOTOR_COMPARATOR);
835 * Set the values in the delay combo box. If <code>reset</code> is <code>true</code>
836 * then sets the selected value as the value closest to selectedDelay, otherwise
837 * leaves selection alone.
839 private void setDelays(boolean reset) {
840 if (selectedMotor == null) {
843 delayBox.setModel(new DefaultComboBoxModel(new String[] { trans.get("TCMotorSelPan.delayBox.None") }));
844 delayBox.setSelectedIndex(0);
848 List<Double> delays = selectedMotorSet.getDelays();
849 String[] delayStrings = new String[delays.size()];
850 double currentDelay = selectedDelay; // Store current setting locally
852 for (int i = 0; i < delays.size(); i++) {
854 delayStrings[i] = ThrustCurveMotor.getDelayString(delays.get(i), trans.get("TCMotorSelPan.delayBox.None"));
856 delayBox.setModel(new DefaultComboBoxModel(delayStrings));
860 // Find and set the closest value
861 double closest = Double.NaN;
862 for (int i = 0; i < delays.size(); i++) {
863 // if-condition to always become true for NaN
864 if (!(Math.abs(delays.get(i) - currentDelay) > Math.abs(closest - currentDelay))) {
865 closest = delays.get(i);
868 if (!Double.isNaN(closest)) {
869 selectedDelay = closest;
871 delayBox.setSelectedItem(ThrustCurveMotor.getDelayString(closest, trans.get("TCMotorSelPan.delayBox.None")));
873 delayBox.setSelectedItem("None");
878 selectedDelay = currentDelay;
880 delayBox.setSelectedItem(ThrustCurveMotor.getDelayString(currentDelay, trans.get("TCMotorSelPan.delayBox.None")));
890 //////////////////////
893 private class CurveSelectionRenderer implements ListCellRenderer {
895 private final ListCellRenderer renderer;
897 public CurveSelectionRenderer(ListCellRenderer renderer) {
898 this.renderer = renderer;
902 public Component getListCellRendererComponent(JList list, Object value, int index,
903 boolean isSelected, boolean cellHasFocus) {
905 Component c = renderer.getListCellRendererComponent(list, value, index, isSelected, cellHasFocus);
906 if (value instanceof MotorHolder) {
907 MotorHolder m = (MotorHolder) value;
908 c.setForeground(getColor(m.getIndex()));
920 * Abstract adapter class.
922 private abstract class MotorRowFilter extends RowFilter<TableModel, Integer> {
924 public boolean include(RowFilter.Entry<? extends TableModel, ? extends Integer> entry) {
925 int index = entry.getIdentifier();
926 ThrustCurveMotorSet m = model.getMotorSet(index);
927 return filterByDiameter(m) && filterByString(m);
930 public abstract boolean filterByDiameter(ThrustCurveMotorSet m);
933 public boolean filterByString(ThrustCurveMotorSet m) {
934 main: for (String s : searchTerms) {
935 for (ThrustCurveMotorColumns col : ThrustCurveMotorColumns.values()) {
936 String str = col.getValue(m).toString().toLowerCase();
937 if (str.indexOf(s) >= 0)
949 private class MotorRowFilterAll extends MotorRowFilter {
951 public boolean filterByDiameter(ThrustCurveMotorSet m) {
957 * Show motors smaller than the mount.
959 private class MotorRowFilterSmaller extends MotorRowFilter {
961 public boolean filterByDiameter(ThrustCurveMotorSet m) {
962 return (m.getDiameter() <= diameter + 0.0004);
967 * Show motors that fit the mount.
969 private class MotorRowFilterExact extends MotorRowFilter {
971 public boolean filterByDiameter(ThrustCurveMotorSet m) {
972 return ((m.getDiameter() <= diameter + 0.0004) && (m.getDiameter() >= diameter - 0.0015));
978 * Custom layered pane that sets the bounds of the components on every layout.
980 public class CustomLayeredPane extends JLayeredPane {
982 public void doLayout() {
983 synchronized (getTreeLock()) {
986 chartPanel.setBounds(0, 0, w, h);
987 zoomIcon.setBounds(w - ZOOM_ICON_POSITION_NEGATIVE_X, ZOOM_ICON_POSITION_POSITIVE_Y, 50, 50);