ParameterDriversList.java

  1. /* Copyright 2002-2018 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.         final DelegatingDriver existingHere = findByName(driver.getName());
  69.         final DelegatingDriver alreadyBound = getAssociatedDelegatingDriver(driver);

  70.         if (existingHere != null) {
  71.             if (alreadyBound != null) {
  72.                 // ensure we don't get intermixed change forwarders that call each other recursively
  73.                 existingHere.setForwarder(alreadyBound.getForwarder());
  74.             } else {
  75.                 // this is a new driver for an already managed parameter
  76.                 existingHere.add(driver);
  77.             }
  78.         } else {
  79.             if (alreadyBound != null) {
  80.                 // the driver is new here, but already bound to other drivers in other lists
  81.                 delegating.add(alreadyBound);
  82.             } else {
  83.                 // this is the first driver we have for this parameter name
  84.                 final DelegatingDriver created   = new DelegatingDriver(driver);
  85.                 final ChangesForwarder forwarder = new ChangesForwarder(created);
  86.                 created.setForwarder(forwarder);
  87.                 delegating.add(created);
  88.             }
  89.         }

  90.     }

  91.     /** Get a {@link DelegatingDriver delegating driver} bound to a driver.
  92.      * @param driver driver to check
  93.      * @return a {@link DelegatingDriver delegating driver} bound to a driver, or
  94.      * null if this driver is not associated with any {@link DelegatingDriver delegating driver}
  95.      * @since 9.1
  96.      */
  97.     private DelegatingDriver getAssociatedDelegatingDriver(final ParameterDriver driver) {
  98.         for (final ParameterObserver observer : driver.getObservers()) {
  99.             if (observer instanceof ChangesForwarder) {
  100.                 return ((ChangesForwarder) observer).getDelegatingDriver();
  101.             }
  102.         }
  103.         return null;
  104.     }

  105.     /** Find  a {@link DelegatingDriver delegating driver} by name.
  106.      * @param name name to check
  107.      * @return a {@link DelegatingDriver delegating driver} managing this parameter name
  108.      * @since 9.1
  109.      */
  110.     public DelegatingDriver findByName(final String name) {
  111.         for (final DelegatingDriver d : delegating) {
  112.             if (d.getName().equals(name)) {
  113.                 return d;
  114.             }
  115.         }
  116.         return null;
  117.     }

  118.     /** Sort the parameters lexicographically.
  119.      */
  120.     public void sort() {
  121.         Collections.sort(delegating, new Comparator<DelegatingDriver>() {
  122.             /** {@inheritDoc} */
  123.             @Override
  124.             public int compare(final DelegatingDriver d1, final DelegatingDriver d2) {
  125.                 return d1.getName().compareTo(d2.getName());
  126.             }
  127.         });
  128.     }

  129.     /** Filter parameters to keep only one type of selection status.
  130.      * @param selected if true, only {@link ParameterDriver#isSelected()
  131.      * selected} parameters will be kept, the other ones will be removed
  132.      */
  133.     public void filter(final boolean selected) {
  134.         for (final Iterator<DelegatingDriver> iterator = delegating.iterator(); iterator.hasNext();) {
  135.             if (iterator.next().isSelected() != selected) {
  136.                 iterator.remove();
  137.             }
  138.         }
  139.     }

  140.     /** Get the number of parameters with different names.
  141.      * @return number of parameters with different names
  142.      */
  143.     public int getNbParams() {
  144.         return delegating.size();
  145.     }

  146.     /** Get delegating drivers for all parameters.
  147.      * <p>
  148.      * The delegating drivers are <em>not</em> the same as
  149.      * the drivers added to the list, but they delegate to them.
  150.      * </p>
  151.      * <p>
  152.      * All delegating drivers manage parameters with different names.
  153.      * </p>
  154.      * @return unmodifiable view of the list of delegating drivers
  155.      */
  156.     public List<DelegatingDriver> getDrivers() {
  157.         return Collections.unmodifiableList(delegating);
  158.     }

  159.     /** Specialized driver delegating to several other managing
  160.      * the same parameter name.
  161.      */
  162.     public static class DelegatingDriver extends ParameterDriver {

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

  165.         /** Observer for propagating changes between all drivers. */
  166.         private ChangesForwarder forwarder;

  167.         /** Simple constructor.
  168.          * @param driver first driver in the series
  169.          * @exception OrekitException if first drivers throws one
  170.          */
  171.         DelegatingDriver(final ParameterDriver driver) throws OrekitException {
  172.             super(driver.getName(), driver.getReferenceValue(),
  173.                   driver.getScale(), driver.getMinValue(), driver.getMaxValue());
  174.             drivers = new ArrayList<ParameterDriver>();
  175.             drivers.add(driver);

  176.             setValue(driver.getValue());
  177.             setReferenceDate(driver.getReferenceDate());
  178.             setSelected(driver.isSelected());

  179.         }

  180.         /** Set the changes forwarder.
  181.          * @param forwarder new changes forwarder
  182.          * @exception OrekitException if forwarder generates one
  183.          * @since 9.1
  184.          */
  185.         void setForwarder(final ChangesForwarder forwarder)
  186.             throws OrekitException {

  187.             // remove the previous observer if any
  188.             if (this.forwarder != null) {
  189.                 removeObserver(this.forwarder);
  190.                 for (final ParameterDriver driver : drivers) {
  191.                     driver.removeObserver(this.forwarder);
  192.                 }
  193.             }

  194.             // add the new observer
  195.             this.forwarder = forwarder;
  196.             addObserver(forwarder);
  197.             for (final ParameterDriver driver : drivers) {
  198.                 driver.addObserver(forwarder);
  199.             }

  200.         }

  201.         /** Get the changes forwarder.
  202.          * @return changes forwarder
  203.          */
  204.         ChangesForwarder getForwarder() {
  205.             return forwarder;
  206.         }

  207.         /** Add a driver.
  208.          * @param driver driver to add
  209.          * @exception OrekitException if an existing drivers cannot be set to the same value
  210.          */
  211.         private void add(final ParameterDriver driver)
  212.             throws OrekitException {

  213.             for (final ParameterDriver d : drivers) {
  214.                 if (d == driver) {
  215.                     // the driver is already known, don't add it again
  216.                     return;
  217.                 }
  218.             }

  219.             setValue(driver.getValue());
  220.             setReferenceDate(driver.getReferenceDate());

  221.             // if any of the drivers is selected, all must be selected
  222.             if (isSelected()) {
  223.                 driver.setSelected(true);
  224.             } else {
  225.                 setSelected(driver.isSelected());
  226.             }

  227.             driver.addObserver(forwarder);
  228.             drivers.add(driver);

  229.         }

  230.         /** Get the raw drivers to which this one delegates.
  231.          * <p>
  232.          * These raw drivers all manage the same parameter name.
  233.          * </p>
  234.          * @return raw drivers to which this one delegates
  235.          */
  236.         public List<ParameterDriver> getRawDrivers() {
  237.             return Collections.unmodifiableList(drivers);
  238.         }

  239.     }

  240.     /** Local observer for propagating changes, avoiding infinite recursion. */
  241.     private static class ChangesForwarder implements ParameterObserver {

  242.         /** DelegatingDriver we are associated with. */
  243.         private final DelegatingDriver delegating;

  244.         /** Root of the current update chain. */
  245.         private ParameterDriver root;

  246.         /** Depth of the current update chain. */
  247.         private int depth;

  248.         /** Simple constructor.
  249.          * @param delegating delegatingDriver we are associated with
  250.          */
  251.         ChangesForwarder(final DelegatingDriver delegating) {
  252.             this.delegating = delegating;
  253.         }

  254.         /** Get the {@link DelegatingDriver} associated with this instance.
  255.          * @return {@link DelegatingDriver} associated with this instance
  256.          * @since 9.1
  257.          */
  258.         DelegatingDriver getDelegatingDriver() {
  259.             return delegating;
  260.         }

  261.         /** {@inheritDoc} */
  262.         @Override
  263.         public void valueChanged(final double previousValue, final ParameterDriver driver)
  264.             throws OrekitException {
  265.             try {
  266.                 updateAll(driver, d -> {
  267.                     try {
  268.                         d.setValue(driver.getValue());
  269.                     } catch (OrekitException oe) {
  270.                         throw new OrekitExceptionWrapper(oe);
  271.                     }
  272.                 });
  273.             } catch (OrekitExceptionWrapper oew) {
  274.                 throw oew.getException();
  275.             }
  276.         }

  277.         /** {@inheritDoc} */
  278.         @Override
  279.         public void referenceDateChanged(final AbsoluteDate previousReferenceDate, final ParameterDriver driver) {
  280.             updateAll(driver, d -> d.setReferenceDate(driver.getReferenceDate()));
  281.         }

  282.         /** {@inheritDoc} */
  283.         @Override
  284.         public void nameChanged(final String previousName, final ParameterDriver driver) {
  285.             updateAll(driver, d -> d.setName(driver.getName()));
  286.         }

  287.         /** {@inheritDoc} */
  288.         @Override
  289.         public void selectionChanged(final boolean previousSelection, final ParameterDriver driver) {
  290.             updateAll(driver, d -> d.setSelected(driver.isSelected()));
  291.         }

  292.         /** Update all bound parameters.
  293.          * @param driver driver triggering the update
  294.          * @param updater updater to use
  295.          */
  296.         private void updateAll(final ParameterDriver driver, final Updater updater) {

  297.             final boolean firstCall = depth++ == 0;
  298.             if (firstCall) {
  299.                 root = driver;
  300.             }

  301.             if (driver == getDelegatingDriver()) {
  302.                 // propagate change downwards, which will trigger recursive calls
  303.                 for (final ParameterDriver d : delegating.drivers) {
  304.                     if (d != root) {
  305.                         updater.update(d);
  306.                     }
  307.                 }
  308.             } else if (firstCall) {
  309.                 // first call started from an underlying driver, propagate change upwards
  310.                 updater.update(getDelegatingDriver());
  311.             }

  312.             if (--depth == 0) {
  313.                 // this is the end of the root call
  314.                 root = null;
  315.             }

  316.         }

  317.     }

  318.     /** Interface for updating parameters. */
  319.     @FunctionalInterface
  320.     private interface Updater {
  321.         /** Update a driver.
  322.          * @param driver driver to update
  323.          */
  324.         void update(ParameterDriver driver);
  325.     }

  326. }