#ifndef sibyageonstep_h
#define sibyageonstep_h

#include "sionstep.h"

/**
 * \class SIByAgeOnStep
 * \brief This is the class used to calculate a likelihood score by fitting the model population to age based survey index data
 */
class SIByAgeOnStep : public SIOnStep {
public:
  /**
   * \brief This is the SIByAgeOnStep constructor
   * \param infile is the CommentStream to read the survey index data from
   * \param areas is the IntMatrix of areas that the survey index can take place on
   * \param ages is the IntMatrix of ages from the survey index data
   * \param areaindex is the CharPtrVector of area identifier text strings
   * \param ageindex is the CharPtrVector of age identifier text strings
   * \param TimeInfo is the TimeClass for the current model
   * \param datafilename is the name of the file containing the survey index data
   * \param givenname is the name of the SIByAgeOnStep likelihood component
   * \param bio is the flag to denote whether the index should be based on the biomass or not
   */
  SIByAgeOnStep(CommentStream& infile, const IntMatrix& areas,
    const IntMatrix& ages, const CharPtrVector& areaindex,
    const CharPtrVector& ageindex, const TimeClass* const TimeInfo,
    const char* datafilename, const char* givenname, int bio);
  /**
   * \brief This is the default SIByAgeOnStep destructor
   */
  virtual ~SIByAgeOnStep();
  /**
   * \brief This function will sum the survey index by age data
   * \param TimeInfo is the TimeClass for the current model
   */
  virtual void Sum(const TimeClass* const TimeInfo);
  /**
   * \brief This will select the fleets and stocks required to calculate the survey index likelihood score
   * \param Fleets is the FleetPtrVector of all the available fleets
   * \param Stocks is the StockPtrVector of all the available stocks
   */
  virtual void setFleetsAndStocks(FleetPtrVector& Fleets, StockPtrVector& Stocks);
  /**
   * \brief This function will print the SIByAgeOnStep information
   * \param outfile is the ofstream that all the model information gets sent to
   */
  virtual void Print(ofstream& outfile) const { aggregator->Print(outfile); };
protected:
  /**
   * \brief This is the StockAggregator used to collect information about the relevant stocks for the survey index data
   */
  StockAggregator* aggregator;
};

#endif