ConditionalIterator.h

Go to the documentation of this file.

#ifndef TIL_CONDITIONAL_ITERATOR_H
#define TIL_CONDITIONAL_ITERATOR_H

// includes from TIL library
#include "til/til_common.h"

// namespace
namespace til
{

        template < class TIterator, class BoolFunctor >
        class ConditionalIterator : public TIterator
        {
        public: // constructors & destructor
                
                ConditionalIterator(const typename TIterator::TImage &im, const BoolFunctor &boolFunctor = BoolFunctor()) : TIterator(im), m_boolFunctor(boolFunctor) {};
                
                ConditionalIterator(typename TIterator::TImage &im, const BoolFunctor &boolFunctor = BoolFunctor()) : TIterator(im), m_boolFunctor(boolFunctor) {};

        public: // functions

                bool next()
                {
                        do
                        {
                                // iterate if you can, but stop if you gotta stop!
                                if (!this->TIterator::next()) return false;
                        }
                        // Not a positive point? Then iterate again
                        while (!m_boolFunctor((*this)));
                        
                        // Okay, we found a positive point, and we are not at the end
                        return true;
                }

        private:
                BoolFunctor m_boolFunctor;
        };
}

#endif