ParameterDriversList.java

  1. /* Copyright 2002-2022 CS GROUP
  2.  * Licensed to CS GROUP (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.time.AbsoluteDate;


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

  45.     /** Managed drivers. */
  46.     private final List<DelegatingDriver> delegating;

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

  52.     /** Add a driver.
  53.      * <p>
  54.      * If the driver is already present, it will not be added.
  55.      * If another driver managing the same parameter is present,
  56.      * both drivers will be managed together, existing drivers
  57.      * being set to the value of the last driver added (i.e.
  58.      * each addition overrides the parameter value).
  59.      * </p>
  60.      * @param driver driver to add
  61.      */
  62.     public void add(final ParameterDriver driver) {

  63.         final DelegatingDriver existingHere = findByName(driver.getName());
  64.         final DelegatingDriver alreadyBound = getAssociatedDelegatingDriver(driver);

  65.         if (existingHere != null) {
  66.             if (alreadyBound != null) {
  67.                 // merge the two delegating drivers
  68.                 existingHere.merge(alreadyBound);
  69.             } else {
  70.                 // this is a new driver for an already managed parameter
  71.                 existingHere.add(driver);
  72.             }
  73.         } else {
  74.             if (alreadyBound != null) {
  75.                 // the driver is new here, but already bound to other drivers in other lists
  76.                 delegating.add(alreadyBound);
  77.                 alreadyBound.addOwner(this);
  78.             } else {
  79.                 // this is the first driver we have for this parameter name
  80.                 delegating.add(new DelegatingDriver(this, driver));
  81.             }
  82.         }

  83.     }

  84.     /** Get a {@link DelegatingDriver delegating driver} bound to a driver.
  85.      * @param driver driver to check
  86.      * @return a {@link DelegatingDriver delegating driver} bound to a driver, or
  87.      * null if this driver is not associated with any {@link DelegatingDriver delegating driver}
  88.      * @since 9.1
  89.      */
  90.     private DelegatingDriver getAssociatedDelegatingDriver(final ParameterDriver driver) {
  91.         for (final ParameterObserver observer : driver.getObservers()) {
  92.             if (observer instanceof ChangesForwarder) {
  93.                 return ((ChangesForwarder) observer).getDelegatingDriver();
  94.             }
  95.         }
  96.         return null;
  97.     }

  98.     /** Replace a {@link DelegatingDriver delegating driver}.
  99.      * @param oldDelegating delegating driver to replace
  100.      * @param newDelegating new delegating driver to use
  101.      * @since 10.1
  102.      */
  103.     private void replaceDelegating(final DelegatingDriver oldDelegating, final DelegatingDriver newDelegating) {
  104.         for (int i = 0; i < delegating.size(); ++i) {
  105.             if (delegating.get(i) == oldDelegating) {
  106.                 delegating.set(i, newDelegating);
  107.             }
  108.         }
  109.     }

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

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

  134.     /** Filter parameters to keep only one type of selection status.
  135.      * @param selected if true, only {@link ParameterDriver#isSelected()
  136.      * selected} parameters will be kept, the other ones will be removed
  137.      */
  138.     public void filter(final boolean selected) {
  139.         for (final Iterator<DelegatingDriver> iterator = delegating.iterator(); iterator.hasNext();) {
  140.             final DelegatingDriver delegatingDriver = iterator.next();
  141.             if (delegatingDriver.isSelected() != selected) {
  142.                 iterator.remove();
  143.                 delegatingDriver.removeOwner(this);
  144.             }
  145.         }
  146.     }

  147.     /** Get the number of parameters with different names.
  148.      * @return number of parameters with different names
  149.      */
  150.     public int getNbParams() {
  151.         return delegating.size();
  152.     }

  153.     /** Get delegating drivers for all parameters.
  154.      * <p>
  155.      * The delegating drivers are <em>not</em> the same as
  156.      * the drivers added to the list, but they delegate to them.
  157.      * </p>
  158.      * <p>
  159.      * All delegating drivers manage parameters with different names.
  160.      * </p>
  161.      * @return unmodifiable view of the list of delegating drivers
  162.      */
  163.     public List<DelegatingDriver> getDrivers() {
  164.         return Collections.unmodifiableList(delegating);
  165.     }

  166.     /** Specialized driver delegating to several other managing
  167.      * the same parameter name.
  168.      */
  169.     public static class DelegatingDriver extends ParameterDriver {

  170.         /** Lists owning this delegating driver. */
  171.         private final List<ParameterDriversList> owners;

  172.         /** Observer for propagating changes between all drivers. */
  173.         private ChangesForwarder forwarder;

  174.         /** Simple constructor.
  175.          * @param owner list owning this delegating driver
  176.          * @param driver first driver in the series
  177.          */
  178.         DelegatingDriver(final ParameterDriversList owner, final ParameterDriver driver) {
  179.             super(driver.getName(), driver.getReferenceValue(),
  180.                   driver.getScale(), driver.getMinValue(), driver.getMaxValue());

  181.             owners = new ArrayList<>();
  182.             addOwner(owner);

  183.             setValue(driver.getValue());
  184.             setReferenceDate(driver.getReferenceDate());
  185.             setSelected(driver.isSelected());

  186.             // set up a change forwarder observing both the raw driver and the delegating driver
  187.             this.forwarder = new ChangesForwarder(this, driver);
  188.             addObserver(forwarder);
  189.             driver.addObserver(forwarder);

  190.         }

  191.         /** Add an owner for this delegating driver.
  192.          * @param owner owner to add
  193.          */
  194.         void addOwner(final ParameterDriversList owner) {
  195.             owners.add(owner);
  196.         }

  197.         /** Remove one owner of this driver.
  198.          * @param owner owner to remove delegating driver from
  199.          * @since 10.1
  200.          */
  201.         private void removeOwner(final ParameterDriversList owner) {
  202.             for (final Iterator<ParameterDriversList> iterator = owners.iterator(); iterator.hasNext();) {
  203.                 if (iterator.next() == owner) {
  204.                     iterator.remove();
  205.                 }
  206.             }
  207.         }

  208.         /** Add a driver.
  209.          * @param driver driver to add
  210.          */
  211.         private void add(final ParameterDriver driver) {

  212.             setValue(driver.getValue());
  213.             setReferenceDate(driver.getReferenceDate());

  214.             // if any of the drivers is selected, all must be selected
  215.             if (isSelected()) {
  216.                 driver.setSelected(true);
  217.             } else {
  218.                 setSelected(driver.isSelected());
  219.             }

  220.             driver.addObserver(forwarder);
  221.             forwarder.add(driver);

  222.         }

  223.         /** Merge another instance.
  224.          * <p>
  225.          * After merging, the other instance is merely empty and preserved
  226.          * only as a child of the current instance. Changes are therefore
  227.          * still forwarded to it, but it is itself not responsible anymore
  228.          * for forwarding change.
  229.          * </p>
  230.          * @param other instance to merge
  231.          */
  232.         private void merge(final DelegatingDriver other) {

  233.             if (other.forwarder == forwarder) {
  234.                 // we are attempting to merge an instance with either itself
  235.                 // or an already embedded one, just ignore the request
  236.                 return;
  237.             }

  238.             // synchronize parameter
  239.             setValue(other.getValue());
  240.             setReferenceDate(other.getReferenceDate());
  241.             if (isSelected()) {
  242.                 other.setSelected(true);
  243.             } else {
  244.                 setSelected(other.isSelected());
  245.             }

  246.             // move around drivers
  247.             for (final ParameterDriver otherDriver : other.forwarder.getDrivers()) {
  248.                 // as drivers are added one at a time and always refer back to a single
  249.                 // DelegatingDriver (through the ChangesForwarder), they cannot be
  250.                 // referenced by two different DelegatingDriver. We can blindly move
  251.                 // around all drivers, there cannot be any duplicates
  252.                 forwarder.add(otherDriver);
  253.                 otherDriver.replaceObserver(other.forwarder, forwarder);
  254.             }

  255.             // forwarding is now delegated to current instance
  256.             other.replaceObserver(other.forwarder, forwarder);
  257.             other.forwarder = forwarder;

  258.             // replace merged instance with current instance in former owners
  259.             for (final ParameterDriversList otherOwner : other.owners) {
  260.                 owners.add(otherOwner);
  261.                 otherOwner.replaceDelegating(other, this);
  262.             }

  263.         }

  264.         /** Get the raw drivers to which this one delegates.
  265.          * <p>
  266.          * These raw drivers all manage the same parameter name.
  267.          * </p>
  268.          * @return raw drivers to which this one delegates
  269.          */
  270.         public List<ParameterDriver> getRawDrivers() {
  271.             return Collections.unmodifiableList(forwarder.getDrivers());
  272.         }

  273.     }

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

  276.         /** DelegatingDriver we are associated with. */
  277.         private final DelegatingDriver delegating;

  278.         /** Drivers synchronized together by the instance. */
  279.         private final List<ParameterDriver> drivers;

  280.         /** Root of the current update chain. */
  281.         private ParameterDriver root;

  282.         /** Depth of the current update chain. */
  283.         private int depth;

  284.         /** Simple constructor.
  285.          * @param delegating delegatingDriver we are associated with
  286.          * @param driver first driver in the series
  287.          */
  288.         ChangesForwarder(final DelegatingDriver delegating, final ParameterDriver driver) {
  289.             this.delegating = delegating;
  290.             this.drivers    = new ArrayList<>();
  291.             drivers.add(driver);
  292.         }

  293.         /** Get the {@link DelegatingDriver} associated with this instance.
  294.          * @return {@link DelegatingDriver} associated with this instance
  295.          * @since 9.1
  296.          */
  297.         DelegatingDriver getDelegatingDriver() {
  298.             return delegating;
  299.         }

  300.         /** Add a driver to the list synchronized together by the instance.
  301.          * @param driver driver to add
  302.          * @since 10.1
  303.          */
  304.         void add(final ParameterDriver driver) {
  305.             drivers.add(driver);
  306.         }

  307.         /** Get the drivers synchronized together by the instance.
  308.          * @return drivers synchronized together by the instance.
  309.          * @since 10.1
  310.          */
  311.         public List<ParameterDriver> getDrivers() {
  312.             return drivers;
  313.         }

  314.         /** {@inheritDoc} */
  315.         @Override
  316.         public void valueChanged(final double previousValue, final ParameterDriver driver) {
  317.             updateAll(driver, d -> d.setValue(driver.getValue()));
  318.         }

  319.         /** {@inheritDoc} */
  320.         @Override
  321.         public void referenceDateChanged(final AbsoluteDate previousReferenceDate, final ParameterDriver driver) {
  322.             updateAll(driver, d -> d.setReferenceDate(driver.getReferenceDate()));
  323.         }

  324.         /** {@inheritDoc} */
  325.         @Override
  326.         public void nameChanged(final String previousName, final ParameterDriver driver) {
  327.             updateAll(driver, d -> d.setName(driver.getName()));
  328.         }

  329.         /** {@inheritDoc} */
  330.         @Override
  331.         public void selectionChanged(final boolean previousSelection, final ParameterDriver driver) {
  332.             updateAll(driver, d -> d.setSelected(driver.isSelected()));
  333.         }

  334.         /** {@inheritDoc} */
  335.         @Override
  336.         public void referenceValueChanged(final double previousReferenceValue, final ParameterDriver driver) {
  337.             updateAll(driver, d -> d.setReferenceValue(driver.getReferenceValue()));
  338.         }

  339.         /** {@inheritDoc} */
  340.         @Override
  341.         public void minValueChanged(final double previousMinValue, final ParameterDriver driver) {
  342.             updateAll(driver, d -> d.setMinValue(driver.getMinValue()));
  343.         }

  344.         /** {@inheritDoc} */
  345.         @Override
  346.         public void maxValueChanged(final double previousMaxValue, final ParameterDriver driver) {
  347.             updateAll(driver, d -> d.setMaxValue(driver.getMaxValue()));
  348.         }

  349.         /** {@inheritDoc} */
  350.         @Override
  351.         public void scaleChanged(final double previousScale, final ParameterDriver driver) {
  352.             updateAll(driver, d -> d.setScale(driver.getScale()));
  353.         }

  354.         /** Update all bound parameters.
  355.          * @param driver driver triggering the update
  356.          * @param updater updater to use
  357.          */
  358.         private void updateAll(final ParameterDriver driver, final Updater updater) {

  359.             final boolean firstCall = depth++ == 0;
  360.             if (firstCall) {
  361.                 root = driver;
  362.             }

  363.             if (driver == getDelegatingDriver()) {
  364.                 // propagate change downwards, which will trigger recursive calls
  365.                 for (final ParameterDriver d : drivers) {
  366.                     if (d != root) {
  367.                         updater.update(d);
  368.                     }
  369.                 }
  370.             } else if (firstCall) {
  371.                 // first call started from an underlying driver, propagate change upwards
  372.                 updater.update(getDelegatingDriver());
  373.             }

  374.             if (--depth == 0) {
  375.                 // this is the end of the root call
  376.                 root = null;
  377.             }

  378.         }

  379.     }

  380.     /** Interface for updating parameters. */
  381.     @FunctionalInterface
  382.     private interface Updater {
  383.         /** Update a driver.
  384.          * @param driver driver to update
  385.          */
  386.         void update(ParameterDriver driver);
  387.     }

  388. }