MultiShapeletBasis.h

Go to the documentation of this file.

// -*- lsst-c++ -*-
/*
 * LSST Data Management System
 * Copyright 2008-2013 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_SHAPELET_MultiShapeletBasis_h_INCLUDED
#define LSST_SHAPELET_MultiShapeletBasis_h_INCLUDED

#include "lsst/shapelet/MultiShapeletFunction.h"

namespace lsst { namespace shapelet {

class MultiShapeletBasisComponent {
public:

    MultiShapeletBasisComponent(double radius, int order, ndarray::Array<double const,2,2> const & matrix);

    double getRadius() const { return _radius; }

    int getOrder() const { return _order; }

    ndarray::Array<double const,2,2> getMatrix() const { return _matrix; }

private:
    friend class MultiShapeletBasis;

    double _radius;
    int _order;
    ndarray::Array<double const,2,2> _matrix;
};

class MultiShapeletBasis {
public:
    typedef MultiShapeletBasisComponent Component;
    typedef std::vector<Component> ComponentVector;
    typedef ComponentVector::const_iterator Iterator;

    explicit MultiShapeletBasis(int size);

    int getSize() const { return _size; }

    int getComponentCount() const { return _components.size(); }

    Iterator begin() const { return _components.begin(); }
    Iterator end() const { return _components.end(); }

    void addComponent(double radius, int order, ndarray::Array<double const,2,2> const & matrix);

    void scale(double factor);

    void normalize();

    void merge(MultiShapeletBasis const & other);

    MultiShapeletFunction makeFunction(
        afw::geom::ellipses::Ellipse const & ellipse,
        ndarray::Array<double const,1,1> const & coefficients
    ) const;

private:
    int _size;
    ComponentVector _components;
};

}} // namespace lsst::shapelet

#endif // !LSST_SHAPELET_MultiShapeletBasis_h_INCLUDED