photmodel.h

// -*- C++ -*-
// 
// photometric calibration model.
// 
#ifndef PHOTMODEL_H
#define PHOTMODEL_H

#include <vector>
#include "matvect.h"


class Associations;
class MeasuredStar;
class FittedStar;
class CcdImage;



class CCDgStars
{
public:
  CCDgStars(unsigned int nstarmax=2000,
        unsigned int nccdmax=35);
  
  ~CCDgStars() {}
  
  //  bool          InitModel(PhotFitter const&, Vect& pars);
  
  void          InitPars(Vect& pars) const;
  
  int           NGroups() const { return 2; }
  
  unsigned int  GroupIndex(int igroup) const;
  
  unsigned int  Size(int igroup=-1) const;
  
  unsigned int  NPar(int igroup=-1) const;
  
  //  //! return the numbers of FITTED parameters
  //  unsigned int  NFittedPars(int igroup=-1) const;
  
  unsigned int  NFittedPhotFactors() const { return nfpf_; }
  
  unsigned int  NFittedStars() const { return nfs_; }
  
  bool          CanCompute(const MeasuredStar& var, Vect& pars) const;
  
  void          InitIndexTables(Vect const& pars);
  
  bool          Compute(const MeasuredStar& var, const Vect& pars,
            double& val, Vect& deriv) const;
  
  void          SetRefCCD(unsigned int ccd, double scale=1.);
  void          ReleaseRefCCD(unsigned int ccd);
  void          SetExclCCD(unsigned int ccd);
  
  //  void          SetRefStar(unsigned int star, double refflux);
  //  void          SetExclStar(unsigned int star);
  
  int           GetPhotFactorIndex(MeasuredStar const&) const;
  
  int           GetStarFluxIndex(MeasuredStar const&) const;
  
  void          UpdateFittedStar(FittedStar& fs,
                 Vect const& pars, 
                 Vect const& epars) const;
  
private:
  //  bool  hasref_;
  
  unsigned int nfpf_;
  unsigned int nfs_;
  unsigned int nstarmax_;
  unsigned int nccdmax_;
  
  Vect  ref_ccd_;
  Vect  excl_ccd_;
  
  //  Vect  ref_star_;
  //  Vect  excl_star_;
  
  Vect fitted_ccd_;
  Vect fitted_star_;
  
  Vect  photfactindex_;
  Vect  starindex_;
};



class CCDgExpStars
{
public:
  CCDgExpStars(unsigned int nstarmax=2000,
           unsigned int nccdmax=36,
           unsigned int nexpmax=1000);
  
  ~CCDgExpStars() {}
  
  //  bool         InitFit(PhotFitter const&, Vect& pars);
  
  void         InitPars(Vect& pars)  const;
  
  int          NGroups() const { return 3; }
  
  unsigned int GroupIndex(int igroup) const;

  unsigned int  Size(int igroup=-1) const;
  
  unsigned int NPar(int igroup=-1) const;
  
  unsigned int NFittedCCDPhotFactors() const { return nfccdpf_; }
  
  unsigned int NFittedStars() const { return nfs_; }
  
  unsigned int NFittedExposurePhotFactors() const { return nfexppf_; }
  
  bool         CanCompute(const MeasuredStar& var, Vect& par) const;
  
  void         InitIndexTables(Vect const& pars);
  
  bool         Compute(const MeasuredStar& var, const Vect& pars,
               double& val, Vect& deriv) const;
  
  void         SetRefCCD(unsigned int ccd, double scale=1.);
  //  void         SetExclCCD(unsigned int ccd);
  void         SetRefExp(CcdImage const&, double scale=1.);
  void         SetRefExp(int expindex, double scale=1.);
  //  void         SetExclExposure(CcdImage const&);
  void         SetRefStar(FittedStar const&, double flux);
  //  void         SetExclStar(FittedStar const&);
  
  int          GetCcdPhotFactorIndex(MeasuredStar const&) const;
  int          GetExpPhotFactorIndex(MeasuredStar const&) const;
  int          GetStarFluxIndex(MeasuredStar const&) const;
  
  void         UpdateFittedStar(FittedStar& fs, 
                Vect const& pars,
                Vect const& epars) const;
  
private:
  unsigned int nfs_;
  unsigned int nfccdpf_;
  unsigned int nfexppf_;
  
  unsigned int nstarmax_;
  unsigned int nccdmax_;
  unsigned int nexpmax_;
  
  Vect         ref_ccd_;
  Vect         ref_star_;
  Vect         ref_exp_;
  //  Vect         excl_ccd_;
  //  Vect         excl_star_;
  //  Vect         excl_exp_;
  
  Vect         ccd_pf_idx_;
  Vect         exp_pf_idx_;
  Vect         star_idx_;
};




class ExpStars
{
public:
  ExpStars(unsigned int nstarmax,
       unsigned int nccdimmax);
  
  ~ExpStars() { }
  
  void           InitPars(Vect& pars) const;
  
  int            NGroups() const { return 2; }
  
  unsigned int   GroupIndex(int igroup) const;
  
  unsigned int   Size(int igroup=-1) const;
  
  unsigned int   NPar(int igroup=-1) const;
  
  unsigned int   NFittedPhotFactors() const { return nfpf_; }
  
  unsigned int   NFittedStars() const { return nfs_; }
  
  bool           CanCompute(const MeasuredStar& var, Vect& pars) const;
  
  void           InitIndexTables(Vect const& pars);
  
  bool           Compute(const MeasuredStar& var, const Vect& pars,
             double& val, Vect& deriv) const;
  
  void           SetRefExp(CcdImage const&, double scale=1.);
  void           SetRefExp(unsigned int expindex, double scale=1.);
  
  int            GetPhotFactorIndex(MeasuredStar const&) const;
  int            GetStarFluxIndex(MeasuredStar const&) const;
  
  void           UpdateFittedStar(FittedStar& fs,
                  Vect const& pars,
                  Vect const& epars) const;
  
private:
  unsigned int nfpf_;
  unsigned int nfs_;

  unsigned int nstarmax_;
  unsigned int nexpmax_;

  
  Vect     ref_exp_;
  //  Vect     ref_star_;
  
  Vect     exp_idx_;
  Vect     star_idx_;
};


#endif