/* Copyright 2002-2022 CS GROUP
    * Licensed to CS GROUP (CS) under one or more
    * contributor license agreements.  See the NOTICE file distributed with
    * this work for additional information regarding copyright ownership.
    * CS licenses this file to You under the Apache License, Version 2.0
    * (the "License"); you may not use this file except in compliance with
    * the License.  You may obtain a copy of the License at
    *
    *   http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  package org.orekit.forces.maneuvers.trigger;
  
  import java.util.ArrayList;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  
  import org.hipparchus.CalculusFieldElement;
  import org.hipparchus.Field;
  import org.orekit.propagation.FieldSpacecraftState;
  import org.orekit.propagation.SpacecraftState;
  import org.orekit.time.AbsoluteDate;
  import org.orekit.time.FieldAbsoluteDate;
  import org.orekit.utils.TimeSpanMap;
  
  /** Base class for triggers.
   * @author Luc Maisonobe
   * @since 11.1
   */
  public abstract class AbstractManeuverTriggers implements ManeuverTriggers {
  
      /** Firing time spans. */
      private TimeSpanMap<Boolean> firings;
  
      /** Propagation direction. */
      private boolean forward;
  
      /** Resetters for the maneuver triggers. */
      private final List<ManeuverTriggersResetter> resetters;
  
      /** Cached field-based resetters. */
      private final transient Map<Field<? extends CalculusFieldElement<?>>, List<FieldManeuverTriggersResetter<?>>> fieldResetters;
  
      /** Simple constructor.
       */
      protected AbstractManeuverTriggers() {
          this.firings        = new TimeSpanMap<>(Boolean.FALSE);
          this.resetters      = new ArrayList<>();
          this.fieldResetters = new HashMap<>();
      }
  
      /** {@inheritDoc} */
      @Override
      public void init(final SpacecraftState initialState, final AbsoluteDate target) {
  
          forward = target.isAfterOrEqualTo(initialState);
          firings = new TimeSpanMap<>(Boolean.FALSE);
          for (final ManeuverTriggersResetter r : resetters) {
              r.init(initialState, target);
          }
  
          if (isFiringOnInitialState(initialState, forward)) {
              if (forward) {
                  firings.addValidAfter(Boolean.TRUE, initialState.getDate(), false);
              } else {
                  firings.addValidBefore(Boolean.TRUE, initialState.getDate(), false);
              }
          }
  
      }
  
      /** {@inheritDoc} */
      @SuppressWarnings("unchecked")
      @Override
      public <T extends CalculusFieldElement<T>> void init(final FieldSpacecraftState<T> initialState, final FieldAbsoluteDate<T> target) {
  
          forward = target.isAfterOrEqualTo(initialState);
          firings = new TimeSpanMap<>(Boolean.FALSE);
          // check if we already have resetters for this field
          final List<FieldManeuverTriggersResetter<?>> list = fieldResetters.get(initialState.getDate().getField());
          if (list != null) {
              for (FieldManeuverTriggersResetter<?> r : list) {
                  ((FieldManeuverTriggersResetter<T>) r).init(initialState, target);
              }
          }
  
          if (isFiringOnInitialState(initialState.toSpacecraftState(), forward)) {
              if (forward) {
                  firings.addValidAfter(Boolean.TRUE, initialState.getDate().toAbsoluteDate(), false);
              } else {
                  firings.addValidBefore(Boolean.TRUE, initialState.getDate().toAbsoluteDate(), false);
              }
          }
 
     }
 
     /**
      * Method to check if the thruster is firing on initialization. can be called by
      * sub classes
      *
      * @param initialState initial spacecraft state
      * @param isForward if true, propagation will be in the forward direction
      * @return true if firing in propagation direction
      */
     protected abstract boolean isFiringOnInitialState(SpacecraftState initialState, boolean isForward);
 
     /** {@inheritDoc} */
     @Override
     public boolean isFiring(final AbsoluteDate date, final double[] parameters) {
         return firings.get(date);
     }
 
     /** {@inheritDoc} */
     @Override
     public <S extends CalculusFieldElement<S>> boolean isFiring(final FieldAbsoluteDate<S> date, final S[] parameters) {
         return firings.get(date.toAbsoluteDate());
     }
 
     /** Get the firings detected during last propagation.
      * @return firings detected during last propagation
      */
     public TimeSpanMap<Boolean> getFirings() {
         return firings;
     }
 
     /** Add a resetter.
      * @param resetter resetter to add
      */
     public void addResetter(final ManeuverTriggersResetter resetter) {
         resetters.add(resetter);
     }
 
     /** Add a resetter.
      * @param field field to which the state belongs
      * @param resetter resetter to add
      * @param <T> type of the field elements
      */
     public <T extends CalculusFieldElement<T>> void addResetter(final Field<T> field, final FieldManeuverTriggersResetter<T> resetter) {
 
         // check if we already have resetters for this field
         List<FieldManeuverTriggersResetter<?>> list = fieldResetters.get(field);
         if (list == null) {
             list = new ArrayList<>();
             fieldResetters.put(field, list);
         }
 
         // add the resetter to the list
         list.add(resetter);
 
     }
 
     /** Initialize resetters.
      * @param initialState initial state
      * @param target target date for the propagation
      */
     protected void initializeResetters(final SpacecraftState initialState, final AbsoluteDate target) {
         for (final ManeuverTriggersResetter r : resetters) {
             r.init(initialState, target);
         }
     }
 
     /** Notify resetters.
      * @param state spacecraft state at trigger date (before applying the maneuver)
      * @param start if true, the trigger is the start of the maneuver
      */
     protected void notifyResetters(final SpacecraftState state, final boolean start) {
         for (final ManeuverTriggersResetter r : resetters) {
             r.maneuverTriggered(state, start);
         }
     }
 
     /** Apply resetters.
      * @param state spacecraft state at trigger date
      * @return reset state
      */
     protected SpacecraftState applyResetters(final SpacecraftState state) {
         SpacecraftState reset = state;
         for (final ManeuverTriggersResetter r : resetters) {
             reset = r.resetState(reset);
         }
         return reset;
     }
 
     /** Initialize resetters.
      * @param initialState initial state
      * @param target target date for the propagation
      * @param <T> type of the field elements
      */
     protected <T extends CalculusFieldElement<T>> void initializeResetters(final FieldSpacecraftState<T> initialState, final FieldAbsoluteDate<T> target) {
         final List<FieldManeuverTriggersResetter<?>> list = fieldResetters.get(initialState.getDate().getField());
         if (list != null) {
             for (final FieldManeuverTriggersResetter<?> r : list) {
                 @SuppressWarnings("unchecked")
                 final FieldManeuverTriggersResetter<T> tr = (FieldManeuverTriggersResetter<T>) r;
                 tr.init(initialState, target);
             }
         }
     }
 
     /** Notify resetters.
      * @param state spacecraft state at trigger date (before applying the maneuver)
      * @param start if true, the trigger is the start of the maneuver
      * @param <T> type of the field elements
      */
     protected <T extends CalculusFieldElement<T>> void notifyResetters(final FieldSpacecraftState<T> state, final boolean start) {
         final List<FieldManeuverTriggersResetter<?>> list = fieldResetters.get(state.getDate().getField());
         if (list != null) {
             for (final FieldManeuverTriggersResetter<?> r : list) {
                 @SuppressWarnings("unchecked")
                 final FieldManeuverTriggersResetter<T> tr = (FieldManeuverTriggersResetter<T>) r;
                 tr.maneuverTriggered(state, start);
             }
         }
     }
 
     /** Apply resetters.
      * @param state spacecraft state at trigger date
      * @param <T> type of the field elements
      * @return reset state
      */
     protected <T extends CalculusFieldElement<T>> FieldSpacecraftState<T>
         applyResetters(final FieldSpacecraftState<T> state) {
         FieldSpacecraftState<T> reset = state;
         final List<FieldManeuverTriggersResetter<?>> list = fieldResetters.get(state.getDate().getField());
         if (list != null) {
             for (final FieldManeuverTriggersResetter<?> r : list) {
                 @SuppressWarnings("unchecked")
                 final FieldManeuverTriggersResetter<T> tr = (FieldManeuverTriggersResetter<T>) r;
                 reset = tr.resetState(reset);
             }
         }
         return reset;
     }
 
 }