ParameterDriversList.java

  1. /* Copyright 2002-2017 CS Systèmes d'Information
  2.  * Licensed to CS Systèmes d'Information (CS) under one or more
  3.  * contributor license agreements.  See the NOTICE file distributed with
  4.  * this work for additional information regarding copyright ownership.
  5.  * CS licenses this file to You under the Apache License, Version 2.0
  6.  * (the "License"); you may not use this file except in compliance with
  7.  * the License.  You may obtain a copy of the License at
  8.  *
  9.  *   http://www.apache.org/licenses/LICENSE-2.0
  10.  *
  11.  * Unless required by applicable law or agreed to in writing, software
  12.  * distributed under the License is distributed on an "AS IS" BASIS,
  13.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14.  * See the License for the specific language governing permissions and
  15.  * limitations under the License.
  16.  */
  17. package org.orekit.utils;

  18. import java.util.ArrayList;
  19. import java.util.Collections;
  20. import java.util.Comparator;
  21. import java.util.Iterator;
  22. import java.util.List;

  23. import org.orekit.errors.OrekitException;
  24. import org.orekit.errors.OrekitExceptionWrapper;
  25. import org.orekit.time.AbsoluteDate;


  26. /** Class managing several {@link ParameterDriver parameter drivers},
  27.  * taking care of duplicated names.
  28.  * <p>
  29.  * Once parameter drivers sharing the same name have been added to
  30.  * an instance of this class, they are permanently bound together and
  31.  * also bound to the {@link #getDrivers() delegating driver} that
  32.  * manages them. This means that if drivers {@code d1}, {@code d2}...
  33.  * {@code dn} are added to the list and both correspond to parameter
  34.  * name "P", then {@link #getDrivers()} will return a list containing
  35.  * a delegating driver {@code delegateD} for the same name "P".
  36.  * Afterwards, whenever either {@link ParameterDriver#setValue(double)}
  37.  * or {@link ParameterDriver#setReferenceDate(AbsoluteDate)} is called
  38.  * on any of the {@code n+1} instances {@code d1}, {@code d2}... {@code dn}
  39.  * or {@code delegateD}, the call will be automatically forwarded to the
  40.  * {@code n} remaining instances, hence ensuring they remain consistent
  41.  * with each other.
  42.  * </p>
  43.  * @author Luc Maisonobe
  44.  * @since 8.0
  45.  */
  46. public class ParameterDriversList {

  47.     /** Managed drivers. */
  48.     private final List<DelegatingDriver> delegating;

  49.     /** Creates an empty list.
  50.      */
  51.     public ParameterDriversList() {
  52.         this.delegating = new ArrayList<DelegatingDriver>();
  53.     }

  54.     /** Add a driver.
  55.      * <p>
  56.      * If the driver is already present, it will not be added.
  57.      * If another driver managing the same parameter is present,
  58.      * both drivers will be managed together, existing drivers
  59.      * being set to the value of the last driver added (i.e.
  60.      * each addition overrides the parameter value).
  61.      * </p>
  62.      * @param driver driver to add
  63.      * @exception OrekitException if an existing driver for the
  64.      * same parameter throws one when its value is reset using the
  65.      * new driver value
  66.      */
  67.     public void add(final ParameterDriver driver) throws OrekitException {

  68.         for (final DelegatingDriver d : delegating) {
  69.             if (d.getName().equals(driver.getName())) {
  70.                 // the parameter is already managed by existing drivers

  71.                 for (final ParameterDriver existing : d.drivers) {
  72.                     if (existing == driver) {
  73.                         // the driver is already known, don't duplicate it
  74.                         return;
  75.                     }
  76.                 }

  77.                 // this is a new driver for an already managed parameter
  78.                 d.add(driver);
  79.                 return;

  80.             }
  81.         }

  82.         // this is the first driver we have for this parameter name
  83.         delegating.add(new DelegatingDriver(driver));

  84.     }

  85.     /** Sort the parameters lexicographically.
  86.      */
  87.     public void sort() {
  88.         Collections.sort(delegating, new Comparator<DelegatingDriver>() {
  89.             /** {@inheritDoc} */
  90.             @Override
  91.             public int compare(final DelegatingDriver d1, final DelegatingDriver d2) {
  92.                 return d1.getName().compareTo(d2.getName());
  93.             }
  94.         });
  95.     }

  96.     /** Filter parameters to keep only one type of selection status.
  97.      * @param selected if true, only {@link ParameterDriver#isSelected()
  98.      * selected} parameters will be kept, the other ones will be removed
  99.      */
  100.     public void filter(final boolean selected) {
  101.         for (final Iterator<DelegatingDriver> iterator = delegating.iterator(); iterator.hasNext();) {
  102.             if (iterator.next().isSelected() != selected) {
  103.                 iterator.remove();
  104.             }
  105.         }
  106.     }

  107.     /** Get the number of parameters with different names.
  108.      * @return number of parameters with different names
  109.      */
  110.     public int getNbParams() {
  111.         return delegating.size();
  112.     }

  113.     /** Get delegating drivers for all parameters.
  114.      * <p>
  115.      * The delegating drivers are <em>not</em> the same as
  116.      * the drivers added to the list, but they delegate to them.
  117.      * </p>
  118.      * <p>
  119.      * All delegating drivers manage parameters with different names.
  120.      * </p>
  121.      * @return unmodifiable view of the list of delegating drivers
  122.      */
  123.     public List<DelegatingDriver> getDrivers() {
  124.         return Collections.unmodifiableList(delegating);
  125.     }

  126.     /** Specialized driver delegating to several other managing
  127.      * the same parameter name.
  128.      */
  129.     public static class DelegatingDriver extends ParameterDriver {

  130.         /** Drivers managing the same parameter. */
  131.         private final List<ParameterDriver> drivers;

  132.         /** Observer for propagating changes between all drivers. */
  133.         private final ChangesForwarder forwarder;

  134.         /** Simple constructor.
  135.          * @param driver first driver in the series
  136.          * @exception OrekitException if first drivers throws one
  137.          */
  138.         DelegatingDriver(final ParameterDriver driver) throws OrekitException {
  139.             super(driver.getName(), driver.getReferenceValue(),
  140.                   driver.getScale(), driver.getMinValue(), driver.getMaxValue());
  141.             drivers = new ArrayList<ParameterDriver>();
  142.             drivers.add(driver);

  143.             setValue(driver.getValue());
  144.             setReferenceDate(driver.getReferenceDate());
  145.             setSelected(driver.isSelected());

  146.             // when the value or reference date of the delegating driver change,
  147.             // all underlying drivers must reproduce the change
  148.             forwarder = new ChangesForwarder();

  149.             addObserver(forwarder);
  150.             driver.addObserver(forwarder);

  151.         }

  152.         /** Add a driver.
  153.          * @param driver driver to add
  154.          * @exception OrekitException if an existing drivers cannot be set to the same value
  155.          */
  156.         private void add(final ParameterDriver driver)
  157.             throws OrekitException {

  158.             setValue(driver.getValue());
  159.             setReferenceDate(driver.getReferenceDate());

  160.             // if any of the drivers is selected, all must be selected
  161.             if (isSelected()) {
  162.                 driver.setSelected(true);
  163.             } else {
  164.                 setSelected(driver.isSelected());
  165.             }

  166.             driver.addObserver(forwarder);
  167.             drivers.add(driver);

  168.         }

  169.         /** Get the raw drivers to which this one delegates.
  170.          * <p>
  171.          * These raw drivers all manage the same parameter name.
  172.          * </p>
  173.          * @return raw drivers to which this one delegates
  174.          */
  175.         public List<ParameterDriver> getRawDrivers() {
  176.             return Collections.unmodifiableList(drivers);
  177.         }

  178.         /** Local observer for propagating changes, avoiding infinite recursion. */
  179.         private class ChangesForwarder implements ParameterObserver {

  180.             /** Root of the current update chain. */
  181.             private ParameterDriver root;

  182.             /** Depth of the current update chain. */
  183.             private int depth;

  184.             /** {@inheritDoc} */
  185.             @Override
  186.             public void valueChanged(final double previousValue, final ParameterDriver driver)
  187.                 throws OrekitException {
  188.                 try {
  189.                     updateAll(driver, d -> {
  190.                         try {
  191.                             d.setValue(driver.getValue());
  192.                         } catch (OrekitException oe) {
  193.                             throw new OrekitExceptionWrapper(oe);
  194.                         }
  195.                     });
  196.                 } catch (OrekitExceptionWrapper oew) {
  197.                     throw oew.getException();
  198.                 }
  199.             }

  200.             /** {@inheritDoc} */
  201.             @Override
  202.             public void referenceDateChanged(final AbsoluteDate previousReferenceDate, final ParameterDriver driver) {
  203.                 updateAll(driver, d -> d.setReferenceDate(driver.getReferenceDate()));
  204.             }

  205.             /** {@inheritDoc} */
  206.             @Override
  207.             public void nameChanged(final String previousName, final ParameterDriver driver) {
  208.                 updateAll(driver, d -> d.setName(driver.getName()));
  209.             }

  210.             /** {@inheritDoc} */
  211.             @Override
  212.             public void selectionChanged(final boolean previousSelection, final ParameterDriver driver) {
  213.                 updateAll(driver, d -> d.setSelected(driver.isSelected()));
  214.             }

  215.             /** Update all bound parameters.
  216.              * @param driver driver triggering the update
  217.              * @param updater updater to use
  218.              */
  219.             private void updateAll(final ParameterDriver driver, final Updater updater) {

  220.                 final boolean firstCall = depth++ == 0;
  221.                 if (firstCall) {
  222.                     root = driver;
  223.                 }

  224.                 if (driver == DelegatingDriver.this) {
  225.                     // propagate change downwards, which will trigger recursive calls
  226.                     for (final ParameterDriver d : drivers) {
  227.                         if (d != root) {
  228.                             updater.update(d);
  229.                         }
  230.                     }
  231.                 } else if (firstCall) {
  232.                     // first call started from an underlying driver, propagate change upwards
  233.                     updater.update(DelegatingDriver.this);
  234.                 }

  235.                 if (--depth == 0) {
  236.                     // this is the end of the root call
  237.                     root = null;
  238.                 }

  239.             }

  240.         }

  241.     }

  242.     /** Interface for updating parameters. */
  243.     @FunctionalInterface
  244.     private interface Updater {
  245.         /** Update a driver.
  246.          * @param driver driver to update
  247.          */
  248.         void update(ParameterDriver driver);
  249.     }

  250. }