ParameterDrivenDateIntervalDetector.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.propagation.events;

  18. import java.util.List;
  19. import java.util.stream.Collectors;

  20. import org.hipparchus.util.FastMath;
  21. import org.orekit.errors.OrekitException;
  22. import org.orekit.errors.OrekitMessages;
  23. import org.orekit.propagation.SpacecraftState;
  24. import org.orekit.propagation.events.handlers.EventHandler;
  25. import org.orekit.propagation.events.handlers.StopOnDecreasing;
  26. import org.orekit.time.AbsoluteDate;
  27. import org.orekit.utils.DateDriver;
  28. import org.orekit.utils.ParameterDriver;
  29. import org.orekit.utils.ParameterObserver;

  30. /** Detector for date intervals that may be offset thanks to parameter drivers.
  31.  * <p>
  32.  * Two dual views can be used for date intervals: either start date/stop date or
  33.  * median date/duration. {@link #getStartDriver() start}/{@link #getStopDriver() stop}
  34.  * drivers and {@link #getMedianDriver() median}/{@link #getDurationDriver() duration}
  35.  * drivers work in pair. Both drivers in one pair can be selected and their changes will
  36.  * be propagated to the other pair, but attempting to select drivers in both
  37.  * pairs at the same time will trigger an exception. Changing the value of a driver
  38.  * that is not selected should be avoided as it leads to inconsistencies between the pairs.
  39.  * </p>
  40.  * @see org.orekit.propagation.Propagator#addEventDetector(EventDetector)
  41.  * @author Luc Maisonobe
  42.  * @since 11.1
  43.  */
  44. public class ParameterDrivenDateIntervalDetector extends AbstractDetector<ParameterDrivenDateIntervalDetector> {

  45.     /** Default suffix for start driver. */
  46.     public static final String START_SUFFIX = "_START";

  47.     /** Default suffix for stop driver. */
  48.     public static final String STOP_SUFFIX = "_STOP";

  49.     /** Default suffix for median driver. */
  50.     public static final String MEDIAN_SUFFIX = "_MEDIAN";

  51.     /** Default suffix for duration driver. */
  52.     public static final String DURATION_SUFFIX = "_DURATION";

  53.     /** Reference interval start driver. */
  54.     private DateDriver start;

  55.     /** Reference interval stop driver. */
  56.     private DateDriver stop;

  57.     /** Median date driver. */
  58.     private DateDriver median;

  59.     /** Duration driver. */
  60.     private ParameterDriver duration;

  61.     /** Build a new instance.
  62.      * @param prefix prefix to use for parameter drivers names
  63.      * @param refMedian reference interval median date
  64.      * @param refDuration reference duration
  65.      */
  66.     public ParameterDrivenDateIntervalDetector(final String prefix,
  67.                                                final AbsoluteDate refMedian, final double refDuration) {
  68.         this(prefix,
  69.              refMedian.shiftedBy(-0.5 * refDuration),
  70.              refMedian.shiftedBy(+0.5 * refDuration));
  71.     }

  72.     /** Build a new instance.
  73.      * @param prefix prefix to use for parameter drivers names
  74.      * @param refStart reference interval start date
  75.      * @param refStop reference interval stop date
  76.      */
  77.     public ParameterDrivenDateIntervalDetector(final String prefix,
  78.                                                final AbsoluteDate refStart, final AbsoluteDate refStop) {
  79.         this(0.5 * refStop.durationFrom(refStart), 1.0e-10, DEFAULT_MAX_ITER,
  80.              new StopOnDecreasing<ParameterDrivenDateIntervalDetector>(),
  81.              new DateDriver(refStart, prefix + START_SUFFIX, true),
  82.              new DateDriver(refStop, prefix + STOP_SUFFIX, false),
  83.              new DateDriver(refStart.shiftedBy(0.5 * refStop.durationFrom(refStart)), prefix + MEDIAN_SUFFIX, true),
  84.              new ParameterDriver(prefix + DURATION_SUFFIX, refStop.durationFrom(refStart), 1.0, 0.0, Double.POSITIVE_INFINITY));
  85.     }

  86.     /** Private constructor with full parameters.
  87.      * <p>
  88.      * This constructor is private as users are expected to use the builder
  89.      * API with the various {@code withXxx()} methods to set up the instance
  90.      * in a readable manner without using a huge amount of parameters.
  91.      * </p>
  92.      * @param maxCheck maximum checking interval (s)
  93.      * @param threshold convergence threshold (s)
  94.      * @param maxIter maximum number of iterations in the event time search
  95.      * @param handler event handler to call at event occurrences
  96.      * @param start reference interval start driver
  97.      * @param stop reference interval stop driver
  98.      * @param median median date driver
  99.      * @param duration duration driver
  100.      */
  101.     private ParameterDrivenDateIntervalDetector(final double maxCheck, final double threshold, final int maxIter,
  102.                                                 final EventHandler<? super ParameterDrivenDateIntervalDetector> handler,
  103.                                                 final DateDriver start, final DateDriver stop,
  104.                                                 final DateDriver median, final ParameterDriver duration) {
  105.         super(maxCheck, threshold, maxIter, handler);
  106.         this.start    = start;
  107.         this.stop     = stop;
  108.         this.median   = median;
  109.         this.duration = duration;

  110.         // set up delegation between drivers
  111.         replaceBindingObserver(start,    new StartObserver());
  112.         replaceBindingObserver(stop,     new StopObserver());
  113.         replaceBindingObserver(median,   new MedianObserver());
  114.         replaceBindingObserver(duration, new DurationObserver());

  115.     }

  116.     /** Replace binding observers.
  117.      * @param driver driver for whose binding observers should be replaced
  118.      * @param bindingObserver new binding observer
  119.      */
  120.     private void replaceBindingObserver(final ParameterDriver driver, final BindingObserver bindingObserver) {

  121.         // remove the previous binding observers
  122.         final List<ParameterObserver> original = driver.
  123.                                                  getObservers().
  124.                                                  stream().
  125.                                                  filter(observer -> observer instanceof ParameterDrivenDateIntervalDetector.BindingObserver).
  126.                                                  collect(Collectors.toList());
  127.         original.forEach(observer -> driver.removeObserver(observer));

  128.         driver.addObserver(bindingObserver);

  129.     }

  130.     /** {@inheritDoc} */
  131.     @Override
  132.     protected ParameterDrivenDateIntervalDetector create(final double newMaxCheck, final double newThreshold, final int newMaxIter,
  133.                                                          final EventHandler<? super ParameterDrivenDateIntervalDetector> newHandler) {
  134.         return new ParameterDrivenDateIntervalDetector(newMaxCheck, newThreshold, newMaxIter, newHandler,
  135.                                                        start, stop, median, duration);
  136.     }

  137.     /** Get the driver for start date.
  138.      * <p>
  139.      * Note that the start date is automatically adjusted if either
  140.      * {@link #getMedianDriver() median date} or {@link #getDurationDriver() duration}
  141.      * are {@link ParameterDriver#isSelected() selected} and changed.
  142.      * </p>
  143.      * @return driver for start date
  144.      */
  145.     public DateDriver getStartDriver() {
  146.         return start;
  147.     }

  148.     /** Get the driver for stop date.
  149.      * <p>
  150.      * Note that the stop date is automatically adjusted if either
  151.      * {@link #getMedianDriver() median date} or {@link #getDurationDriver() duration}
  152.      * are {@link ParameterDriver#isSelected() selected} changed.
  153.      * </p>
  154.      * @return driver for stop date
  155.      */
  156.     public DateDriver getStopDriver() {
  157.         return stop;
  158.     }

  159.     /** Get the driver for median date.
  160.      * <p>
  161.      * Note that the median date is automatically adjusted if either
  162.      * {@link #getStartDriver()} start date or {@link #getStopDriver() stop date}
  163.      * are {@link ParameterDriver#isSelected() selected} changed.
  164.      * </p>
  165.      * @return driver for median date
  166.      */
  167.     public DateDriver getMedianDriver() {
  168.         return median;
  169.     }

  170.     /** Get the driver for duration.
  171.      * <p>
  172.      * Note that the duration is automatically adjusted if either
  173.      * {@link #getStartDriver()} start date or {@link #getStopDriver() stop date}
  174.      * are {@link ParameterDriver#isSelected() selected} changed.
  175.      * </p>
  176.      * @return driver for duration
  177.      */
  178.     public ParameterDriver getDurationDriver() {
  179.         return duration;
  180.     }

  181.     /** Compute the value of the switching function.
  182.      * <p>
  183.      * The function is positive for dates within the interval defined
  184.      * by applying the parameter drivers shifts to reference dates,
  185.      * and negative for dates outside of this interval. Note that
  186.      * if Δt_start - Δt_stop is less than ref_stop.durationFrom(ref_start),
  187.      * then the interval degenerates to empty and the function never
  188.      * reaches positive values.
  189.      * </p>
  190.      * @param s the current state information: date, kinematics, attitude
  191.      * @return value of the switching function
  192.      */
  193.     public double g(final SpacecraftState s) {
  194.         return FastMath.min(s.getDate().durationFrom(start.getDate()),
  195.                             stop.getDate().durationFrom(s.getDate()));
  196.     }

  197.     /** Base observer. */
  198.     private abstract class BindingObserver implements ParameterObserver {

  199.         /** {@inheritDoc} */
  200.         @Override
  201.         public void valueChanged(final double previousValue, final ParameterDriver driver) {
  202.             if (driver.isSelected()) {
  203.                 setDelta(driver.getValue() - previousValue);
  204.             }
  205.         }

  206.         /** {@inheritDoc} */
  207.         @Override
  208.         public void selectionChanged(final boolean previousSelection, final ParameterDriver driver) {
  209.             if ((start.isSelected()  || stop.isSelected()) &&
  210.                 (median.isSelected() || duration.isSelected())) {
  211.                 throw new OrekitException(OrekitMessages.INCONSISTENT_SELECTION,
  212.                                           start.getName(), stop.getName(),
  213.                                           median.getName(), duration.getName());
  214.             }
  215.         }

  216.         /** Change a value.
  217.          * @param delta change of value
  218.          */
  219.         protected abstract void setDelta(double delta);

  220.     }

  221.     /** Observer for start date. */
  222.     private class StartObserver extends BindingObserver {
  223.         /** {@inheritDoc} */
  224.         @Override
  225.         protected void setDelta(final double delta) {
  226.             median.setValue(median.getValue() + 0.5 * delta);
  227.             duration.setValue(duration.getValue() - delta);
  228.         }
  229.     }

  230.     /** Observer for stop date. */
  231.     private class StopObserver extends BindingObserver {
  232.         /** {@inheritDoc} */
  233.         @Override
  234.         protected void setDelta(final double delta) {
  235.             median.setValue(median.getValue() + 0.5 * delta);
  236.             duration.setValue(duration.getValue() + delta);
  237.         }
  238.     }

  239.     /** Observer for median date. */
  240.     private class MedianObserver extends BindingObserver {
  241.         /** {@inheritDoc} */
  242.         @Override
  243.         protected void setDelta(final double delta) {
  244.             start.setValue(start.getValue() + delta);
  245.             stop.setValue(stop.getValue() + delta);
  246.         }
  247.     }

  248.     /** Observer for duration. */
  249.     private class DurationObserver extends BindingObserver {
  250.         /** {@inheritDoc} */
  251.         @Override
  252.         protected void setDelta(final double delta) {
  253.             start.setValue(start.getValue() - 0.5 * delta);
  254.             stop.setValue(stop.getValue() + 0.5 * delta);
  255.         }
  256.     }

  257. }