//! \file reaction.h
//! Data for an incident neutron reaction

#ifndef OPENMC_REACTION_H
#define OPENMC_REACTION_H

#include <string>

#include "hdf5.h"

#include "openmc/particle_data.h"
#include "openmc/reaction_product.h"
#include "openmc/span.h"
#include "openmc/vector.h"

namespace openmc {

//==============================================================================
//! Data for a single reaction including cross sections (possibly at multiple
//! temperatures) and reaction products (with secondary angle-energy
//! distributions)
//==============================================================================

class Reaction {
public:
  //! Construct reaction from HDF5 data
  //! \param[in] group HDF5 group containing reaction data
  //! \param[in] temperatures Desired temperatures for cross sections
  //! \param[in] name Name of the nuclide
  explicit Reaction(
    hid_t group, const vector<int>& temperatures, std::string name);

  //! Calculate cross section given temperautre/grid index, interpolation factor
  //
  //! \param[in] i_temp Temperature index
  //! \param[in] i_grid Energy grid index
  //! \param[in] interp_factor Interpolation factor between grid points
  double xs(int64_t i_temp, int64_t i_grid, double interp_factor) const;

  //! Calculate cross section
  //
  //! \param[in] micro Microscopic cross section cache
  double xs(const NuclideMicroXS& micro) const;

  //! \brief Calculate reaction rate based on group-wise flux distribution
  //
  //! \param[in] i_temp Temperature index
  //! \param[in] energy Energy group boundaries in [eV]
  //! \param[in] flux Flux in each energy group (not normalized per eV)
  //! \param[in] grid Nuclide energy grid
  //! \return Reaction rate
  double collapse_rate(int64_t i_temp, span<const double> energy,
    span<const double> flux, const vector<double>& grid) const;

  //! Cross section at a single temperature
  struct TemperatureXS {
    int threshold;
    vector<double> value;
  };

  int mt_;                           //!< ENDF MT value
  double q_value_;                   //!< Reaction Q value in [eV]
  bool scatter_in_cm_;               //!< scattering system in center-of-mass?
  bool redundant_;                   //!< redundant reaction?
  vector<TemperatureXS> xs_;         //!< Cross section at each temperature
  vector<ReactionProduct> products_; //!< Reaction products
};

//==============================================================================
// Non-member functions
//==============================================================================

//! Return reaction name given an ENDF MT value
//
//! \param[in] mt  ENDF MT value
//! \return Name of the corresponding reaction
std::string reaction_name(int mt);

//! Return MT value for given a reaction name (including special tally MT
//! values)
//
//! \param[in] name  Reaction name
//! \return Corresponding MT number or special tally score
int reaction_tally_mt(std::string name);

//! Return ENDF MT number given a reaction name
//
//! Unlike reaction_tally_mt(), this function always returns a positive ENDF MT
//! number and never a special negative tally score.
//!
//! \param[in] name  Reaction name
//! \return Corresponding ENDF MT number
int reaction_mt(const std::string& name);

} // namespace openmc

#endif // OPENMC_REACTION_H