FieldParameterDrivenDateIntervalDetector.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.CalculusFieldElement;
  21. import org.hipparchus.Field;
  22. import org.hipparchus.util.FastMath;
  23. import org.orekit.errors.OrekitException;
  24. import org.orekit.errors.OrekitMessages;
  25. import org.orekit.propagation.FieldSpacecraftState;
  26. import org.orekit.propagation.events.handlers.FieldEventHandler;
  27. import org.orekit.propagation.events.handlers.FieldStopOnEvent;
  28. import org.orekit.time.AbsoluteDate;
  29. import org.orekit.utils.DateDriver;
  30. import org.orekit.utils.ParameterDriver;
  31. import org.orekit.utils.ParameterObserver;

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

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

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

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

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

  56.     /** Reference interval start driver. */
  57.     private DateDriver start;

  58.     /** Reference interval stop driver. */
  59.     private DateDriver stop;

  60.     /** Median date driver. */
  61.     private DateDriver median;

  62.     /** Duration driver. */
  63.     private ParameterDriver duration;

  64.     /** Build a new instance.
  65.      * @param field field to which the elements belong
  66.      * @param prefix prefix to use for parameter drivers names
  67.      * @param refMedian reference interval median date
  68.      * @param refDuration reference duration
  69.      */
  70.     public FieldParameterDrivenDateIntervalDetector(final Field<T> field, final String prefix,
  71.                                                     final AbsoluteDate refMedian, final double refDuration) {
  72.         this(field, prefix,
  73.              refMedian.shiftedBy(-0.5 * refDuration),
  74.              refMedian.shiftedBy(+0.5 * refDuration));
  75.     }

  76.     /** Build a new instance.
  77.      * @param field field to which the elements belong
  78.      * @param prefix prefix to use for parameter drivers names
  79.      * @param refStart reference interval start date
  80.      * @param refStop reference interval stop date
  81.      */
  82.     public FieldParameterDrivenDateIntervalDetector(final Field<T> field, final String prefix,
  83.                                                     final AbsoluteDate refStart, final AbsoluteDate refStop) {
  84.         this(field.getZero().newInstance(DEFAULT_MAXCHECK),
  85.              field.getZero().newInstance(DEFAULT_THRESHOLD),
  86.              DEFAULT_MAX_ITER,
  87.              new FieldStopOnEvent<FieldParameterDrivenDateIntervalDetector<T>, T>(),
  88.              new DateDriver(refStart, prefix + START_SUFFIX, true),
  89.              new DateDriver(refStop, prefix + STOP_SUFFIX, false),
  90.              new DateDriver(refStart.shiftedBy(0.5 * refStop.durationFrom(refStart)), prefix + MEDIAN_SUFFIX, true),
  91.              new ParameterDriver(prefix + DURATION_SUFFIX, refStop.durationFrom(refStart), 1.0, 0.0, Double.POSITIVE_INFINITY));
  92.     }

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

  117.         // set up delegation between drivers
  118.         replaceBindingObserver(start,    new StartObserver());
  119.         replaceBindingObserver(stop,     new StopObserver());
  120.         replaceBindingObserver(median,   new MedianObserver());
  121.         replaceBindingObserver(duration, new DurationObserver());

  122.     }

  123.     /** Replace binding observers.
  124.      * @param driver driver for whose binding observers should be replaced
  125.      * @param bindingObserver new binding observer
  126.      */
  127.     private void replaceBindingObserver(final ParameterDriver driver, final BindingObserver bindingObserver) {

  128.         // remove the previous binding observers
  129.         final List<ParameterObserver> original = driver.
  130.                                                  getObservers().
  131.                                                  stream().
  132.                                                  filter(observer -> observer instanceof FieldParameterDrivenDateIntervalDetector.BindingObserver).
  133.                                                  collect(Collectors.toList());
  134.         original.forEach(observer -> driver.removeObserver(observer));

  135.         driver.addObserver(bindingObserver);

  136.     }

  137.     /** {@inheritDoc} */
  138.     @Override
  139.     protected FieldParameterDrivenDateIntervalDetector<T> create(final T newMaxCheck, final T newThreshold, final int newMaxIter,
  140.                                                                  final FieldEventHandler<? super FieldParameterDrivenDateIntervalDetector<T>, T> newHandler) {
  141.         return new FieldParameterDrivenDateIntervalDetector<>(newMaxCheck, newThreshold, newMaxIter, newHandler,
  142.                                                               start, stop, median, duration);
  143.     }

  144.     /** Get the driver for start date.
  145.      * <p>
  146.      * Note that the start date is automatically adjusted if either
  147.      * {@link #getMedianDriver() median date} or {@link #getDurationDriver() duration}
  148.      * are {@link ParameterDriver#isSelected() selected} and changed.
  149.      * </p>
  150.      * @return driver for start date
  151.      */
  152.     public DateDriver getStartDriver() {
  153.         return start;
  154.     }

  155.     /** Get the driver for stop date.
  156.      * <p>
  157.      * Note that the stop date is automatically adjusted if either
  158.      * {@link #getMedianDriver() median date} or {@link #getDurationDriver() duration}
  159.      * are {@link ParameterDriver#isSelected() selected} changed.
  160.      * </p>
  161.      * @return driver for stop date
  162.      */
  163.     public DateDriver getStopDriver() {
  164.         return stop;
  165.     }

  166.     /** Get the driver for median date.
  167.      * <p>
  168.      * Note that the median date is automatically adjusted if either
  169.      * {@link #getStartDriver()} start date or {@link #getStopDriver() stop date}
  170.      * are {@link ParameterDriver#isSelected() selected} changed.
  171.      * </p>
  172.      * @return driver for median date
  173.      */
  174.     public DateDriver getMedianDriver() {
  175.         return median;
  176.     }

  177.     /** Get the driver for duration.
  178.      * <p>
  179.      * Note that the duration is automatically adjusted if either
  180.      * {@link #getStartDriver()} start date or {@link #getStopDriver() stop date}
  181.      * are {@link ParameterDriver#isSelected() selected} changed.
  182.      * </p>
  183.      * @return driver for duration
  184.      */
  185.     public ParameterDriver getDurationDriver() {
  186.         return duration;
  187.     }

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

  204.     /** Base observer. */
  205.     private abstract class BindingObserver implements ParameterObserver {

  206.         /** {@inheritDoc} */
  207.         @Override
  208.         public void valueChanged(final double previousValue, final ParameterDriver driver) {
  209.             if (driver.isSelected()) {
  210.                 setDelta(driver.getValue() - previousValue);
  211.             }
  212.         }

  213.         /** {@inheritDoc} */
  214.         @Override
  215.         public void selectionChanged(final boolean previousSelection, final ParameterDriver driver) {
  216.             if ((start.isSelected()  || stop.isSelected()) &&
  217.                 (median.isSelected() || duration.isSelected())) {
  218.                 throw new OrekitException(OrekitMessages.INCONSISTENT_SELECTION,
  219.                                           start.getName(), stop.getName(),
  220.                                           median.getName(), duration.getName());
  221.             }
  222.         }

  223.         /** Change a value.
  224.          * @param delta change of value
  225.          */
  226.         protected abstract void setDelta(double delta);

  227.     }

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

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

  246.     /** Observer for median date. */
  247.     private class MedianObserver extends BindingObserver {
  248.         /** {@inheritDoc} */
  249.         @Override
  250.         protected void setDelta(final double delta) {
  251.             start.setValue(start.getValue() + delta);
  252.             stop.setValue(stop.getValue() + delta);
  253.         }
  254.     }

  255.     /** Observer for duration. */
  256.     private class DurationObserver extends BindingObserver {
  257.         /** {@inheritDoc} */
  258.         @Override
  259.         protected void setDelta(final double delta) {
  260.             start.setValue(start.getValue() - 0.5 * delta);
  261.             stop.setValue(stop.getValue() + 0.5 * delta);
  262.         }
  263.     }

  264. }