image_arithmetic.hpp

// RAWK Copyright (C) 2014 Robert Kooima
//
// 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 WITH-
// OUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
// more details.

#ifndef IMAGE_ARITHMETIC_HPP
#define IMAGE_ARITHMETIC_HPP

//------------------------------------------------------------------------------


class sum : public image
{
public:

    sum(image *L, image *R) : image(L, R) { }

    virtual double get(int i, int j, int k) const
    {
        return L->get(i, j, k)
             + R->get(i, j, k);
    }

    virtual void doc(std::ostream& out) const
    {
        out << "sum";
    }
};

//------------------------------------------------------------------------------


class difference : public image
{
public:

    difference(image *L, image *R) : image(L, R) { }

    virtual double get(int i, int j, int k) const
    {
        return L->get(i, j, k)
             - R->get(i, j, k);
    }

    virtual void doc(std::ostream& out) const
    {
        out << "difference";
    }
};

//------------------------------------------------------------------------------


class multiply : public image
{
public:

    multiply(image *L, image *R) : image(L, R) { }

    virtual double get(int i, int j, int k) const
    {
        if (double v = L->get(i, j, k))
            return v * R->get(i, j, k);
        else
            return 0.0;
    }

    virtual void doc(std::ostream& out) const
    {
        out << "multiply";
    }
};

//------------------------------------------------------------------------------

#endif