FillFunc.hpp

/*
 *  Copyright 2011, 2012 Lukas Jirkovsky
 *
 *  This file is part of FotoSHOCKcore.
 *
 *  FotoSHOCKcore is free software: you can redistribute it and/or modify it
 *  under the terms of the GNU Lesser General Public License as published by
 *  the Free Software Foundation, version 3 of the License.
 *
 *  FotoSHOCKcore 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 Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU Lesser General Public License
 *  along with FotoSHOCKcore.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef FILLFUNC_H
#define FILLFUNC_H

#include "PixelData.hpp"
#include "ValueType.hpp"

#include <cstring>
#include <string>

namespace FotoSHOCKcore {

template <typename Format>
class FillFunctor {
    public:
        FillFunctor(PixelData format, const std::string color) : m_format(format) {
            m_color = new Format[format.numOfBands()];
            if (color == "white") {
                // TODO: support for other color spaces
                for (int i = 0; i < format.numOfBands(); i++) {
                    switch(format.format()) {
                        #define FORMAT_DO_STH(T,U) m_color[i] = ValueTypeInfo<T>::max();
                        case_ANY;
                        #undef FORMAT_DO_STH
                    }
                }
            }

            else if (color == "black") {
                // TODO: support for other color spaces
                for (int i = 0; i < format.numOfBands(); i++) {
                    switch(format.format()) {
                        #define FORMAT_DO_STH(T,U) m_color[i] = ValueTypeInfo<T>::min();
                        case_ANY;
                        #undef FORMAT_DO_STH
                    }
                }
            }
        };

        FillFunctor(const FillFunctor<Format>& other) : m_format(other.m_format) {
            m_color = new Format[m_format.numOfBands()];
            std::memcpy(m_color, other.m_color, m_format.numOfBands() * sizeof(Format));
        }

        ~FillFunctor() {
            delete[] m_color;
        }

        void operator()(Format* /*in*/, Format* out) {
            for (int i = 0; i < m_format.numOfBands(); i++) {
                out[i] = m_color[i];
            }
        }

    private:
        const PixelData m_format;
        Format* m_color;
};

}

#endif