AstrometryFit.h

Go to the documentation of this file.

// -*- LSST-C++ -*-
/*
 * This file is part of jointcal.
 *
 * Developed for the LSST Data Management System.
 * This product includes software developed by the LSST Project
 * (https://www.lsst.org).
 * See the COPYRIGHT file at the top-level directory of this distribution
 * for details of code ownership.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#ifndef LSST_JOINTCAL_ASTROMETRY_FIT_H
#define LSST_JOINTCAL_ASTROMETRY_FIT_H

#include <string>
#include <iostream>
#include <sstream>

#include "lsst/jointcal/Associations.h"
#include "lsst/jointcal/CcdImage.h"
#include "lsst/jointcal/Chi2.h"
#include "lsst/jointcal/Eigenstuff.h"
#include "lsst/jointcal/FitterBase.h"
#include "lsst/jointcal/Tripletlist.h"
#include "lsst/jointcal/AstrometryModel.h"

namespace lsst {
namespace jointcal {

class AstrometryFit : public FitterBase {
public:
    AstrometryFit(std::shared_ptr<Associations> associations,
                  std::shared_ptr<AstrometryModel> astrometryModel, double posError);

    AstrometryFit(AstrometryFit const &) = delete;
    AstrometryFit(AstrometryFit &&) = delete;
    AstrometryFit &operator=(AstrometryFit const &) = delete;
    AstrometryFit &operator=(AstrometryFit &&) = delete;

    void assignIndices(std::string const &whatToFit) override;

    void freezeErrorTransform() { _astrometryModel->freezeErrorTransform(); }

    void offsetParams(Eigen::VectorXd const &delta) override;

    std::shared_ptr<AstrometryModel> getModel() const { return _astrometryModel; }

    void checkStuff();

protected:
    void saveChi2MeasContributions(std::string const &filename) const override;

    void saveChi2RefContributions(std::string const &filename) const override;

private:
    bool _fittingDistortions, _fittingPos, _fittingRefrac, _fittingPM;
    std::shared_ptr<AstrometryModel> _astrometryModel;
    double _referenceColor, _sigCol;  // average and r.m.s color
    double _refractionCoefficient;    // fit parameter
    Eigen::Index _refracPosInMatrix;  // where it stands
    double _JDRef;                    // average Julian date

    // counts in parameter subsets.
    std::size_t _nParDistortions;
    std::size_t _nParPositions;
    std::size_t _nParRefrac;

    double _posError;  // constant term on error on position (in pixel unit)

    void leastSquareDerivativesMeasurement(CcdImage const &ccdImage, TripletList &tripletList,
                                           Eigen::VectorXd &grad,
                                           MeasuredStarList const *msList = nullptr) const override;

    void leastSquareDerivativesReference(FittedStarList const &fittedStarList, TripletList &tripletList,
                                         Eigen::VectorXd &grad) const override;

    void accumulateStatImageList(CcdImageList const &ccdImageList, Chi2Accumulator &accum) const override;

    void accumulateStatRefStars(Chi2Accumulator &accum) const override;

    void getIndicesOfMeasuredStar(MeasuredStar const &measuredStar,
                                  IndexVector &indices) const override;

    Point transformFittedStar(FittedStar const &fittedStar, AstrometryTransform const &sky2TP,
                              Point const &refractionVector, double refractionCoeff, double mjd) const;

    void accumulateStatImage(CcdImage const &ccdImage, Chi2Accumulator &accum) const;
};
}  // namespace jointcal
}  // namespace lsst
#endif  // LSST_JOINTCAL_ASTROMETRY_FIT_H