sionstep.h

Go to the documentation of this file.

#ifndef sionstep_h
#define sionstep_h
 
#include "areatime.h"
#include "doublematrixptrvector.h"
#include "intmatrix.h"
#include "charptrvector.h"
#include "conversionindex.h"
#include "commentstream.h"
#include "actionattimes.h"
#include "agebandmatrix.h"
#include "stockaggregator.h"
#include "regressionline.h"
#include "hasname.h"
#include "gadget.h"
 
enum FitType { LINEARFIT = 1, LOGLINEARFIT, WEIGHTLINEARFIT, LOGWEIGHTLINEARFIT,
  FIXEDSLOPELINEARFIT, FIXEDSLOPELOGLINEARFIT, FIXEDSLOPEWEIGHTLINEARFIT,
  FIXEDSLOPELOGWEIGHTLINEARFIT, FIXEDINTERCEPTLINEARFIT, FIXEDINTERCEPTLOGLINEARFIT,
  FIXEDINTERCEPTWEIGHTLINEARFIT, FIXEDINTERCEPTLOGWEIGHTLINEARFIT,
  FIXEDLINEARFIT, FIXEDLOGLINEARFIT, FIXEDWEIGHTLINEARFIT, FIXEDLOGWEIGHTLINEARFIT };
enum SIType { SILENGTH = 1, SIAGE, SIFLEET, SIEFFORT, SIACOUSTIC };
 
class SIOnStep : public HasName {
public:
  SIOnStep(CommentStream& infile, const char* datafilename, const CharPtrVector& aindex,
    const TimeClass* const TimeInfo, const IntMatrix& areas, const CharPtrVector& charindex,
    const char* givenname, int bio, SIType type);
  virtual ~SIOnStep();
  virtual void Sum(const TimeClass* const TimeInfo) = 0;
  virtual void setFleetsAndStocks(FleetPtrVector& Fleets, StockPtrVector& Stocks) = 0;
  virtual void Reset();
  virtual void Print(ofstream& outfile) const = 0;
  virtual void printLikelihood(ofstream& outfile, const TimeClass* const TimeInfo);
  virtual void printSummary(ofstream& outfile, const double weight);
  virtual double calcSSE();
  FitType getType() const { return fittype; };
  SIType getSIType() const { return sitype; };
protected:
  IntMatrix Areas;
  IntVector Years;
  IntVector Steps;
  IntMatrix Ages;
  LengthGroupDivision* LgrpDiv;
  DoubleVector likelihoodValues;
  ActionAtTimes AAT;
  DoubleMatrixPtrVector obsIndex;
  DoubleMatrixPtrVector modelIndex;
  DoubleMatrixPtrVector weightIndex;
  const AgeBandMatrixPtrVector* alptr;
  SIType sitype;
  int timeindex;
  int biomass;
  int useweight;
private:
  void readSIData(CommentStream& infile, const TimeClass* const TimeInfo);
  void readSIRegressionData(CommentStream& infile);
  double slope;
  double intercept;
  DoubleMatrix slopes;
  DoubleMatrix intercepts;
  DoubleMatrix sse;
  DoubleVector tmpModel;
  DoubleVector tmpData;
  DoubleVector tmpWeight;
  CharPtrVector areaindex;
  CharPtrVector colindex;
  FitType fittype;
  Regression* LR;
};
 
#endif