isr.h

Go to the documentation of this file.

// -*- LSST-C++ -*- 
 
/*
 * LSST Data Management System
 * Copyright 2016 LSST Corporation.
 *
 * This product includes software developed by the
 * LSST Project (http://www.lsst.org/).
 *
 * 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 LSST License Statement and
 * the GNU General Public License along with this program.  If not,
 * see <http://www.lsstcorp.org/LegalNotices/>.
 */
 
#ifndef LSST_IP_ISR_ISR_H
#define LSST_IP_ISR_ISR_H
 
#include <memory>
#include <string>
#include <vector>
#include <cmath>
 
#include "lsst/afw/math.h"
#include "lsst/afw/math/Statistics.h"
#include "lsst/afw/image.h"
#include "lsst/pex/exceptions/Exception.h"
 
namespace lsst {
namespace ip {
namespace isr {
 
    template <typename ImageT, typename MaskT=lsst::afw::image::MaskPixel>
    class CountMaskedPixels {
    public:
        typedef typename lsst::afw::image::MaskedImage<ImageT>::x_iterator x_iterator;
        CountMaskedPixels() :
            _count(0) {} ;
        virtual ~CountMaskedPixels() {};
 
        // Clear the accumulator
        void reset() { _count = 0; }
 
        // Count pixels
        void apply(lsst::afw::image::MaskedImage<ImageT> const& image,
                   MaskT bitmask) {
            reset();
            for (int y = 0; y != image.getHeight(); ++y) {
                for (x_iterator ptr = image.row_begin(y); ptr != image.row_end(y); ++ptr) {
                    if ( ((*ptr).mask() & bitmask) == bitmask ) {
                        _count += 1;
                    }
                }
            }
        }
 
        // Return the total counts
        int getCount() const { return _count; }
 
    private:
        int _count;
    };
 
    template <typename PixelT>
    size_t maskNans(
        afw::image::MaskedImage<PixelT> const& mi, 
        afw::image::MaskPixel maskVal,  
        afw::image::MaskPixel allow=0 
        );
 
 
    template<typename ImagePixelT>
    std::vector<double> fitOverscanImage(
        lsst::afw::image::MaskedImage<ImagePixelT> const& overscan,
        std::vector<std::string> badPixelMask,
        bool isTransposed
        );
 
}}} // namespace lsst::ip::isr
 
#endif // !defined(LSST_IP_ISR_ISR_H)