|| || || || || || || ||

---

RangeIterator.h

Go to the documentation of this file.

#ifndef Impala_Util_RangeIterator_h
#define Impala_Util_RangeIterator_h

#include "Util/Iterable.h"

namespace Impala
{
namespace Util
{

class RangeIterator
{
public:
    RangeIterator()
    {
        mDone = false;
    }

    Iterable* GetIterable(const std::string& name)
    {
        for(int i=0 ; i<mRanges.size() ; i++)
            if(mRanges[i]->GetName() == name)
                return mRanges[i];
        return 0;
    }

    int GetTotalIterations()
    {
        int total = 1;
        for(int i=0 ; i<mRanges.size() ; i++)
            total *= mRanges[i]->GetSteps();
        return total;
    }

    int GetIteration()
    {
        return mCurrentIteration;
    }

    void AddRange(const std::string& name, const std::string& range)
    {
        Util::Iterable* param = Util::ParseRangeDefinition(name, range);
        if(param == 0)
            return;

        for(int i=0 ; i<mRanges.size() ; i++)
            if(mRanges[i]->GetName() == name)
            {
                delete mRanges[i];
                mRanges[i] = param;
                return;
            }
        mRanges.push_back(param);
    }

    void SetIteration(int iteration)
    {
        if(iteration >= GetTotalIterations())
            ILOG_ERROR("argument too big");

        mDone = false;

        for(int i=0 ; i<mRanges.size() ; i++)
        {
            Util::Iterable* curr = mRanges[i];
            int range = curr->GetSteps();
            if(range  > 0)
            {
                curr->SetIteration(iteration % range);
                iteration /= range;
            }
        }
    }

    void GetParameters(PropertySet* properties)
    {
        for(int i=0 ; i<mRanges.size() ; i++)
            properties->Add(mRanges[i]->GetName(), mRanges[i]->GetValue());
    }

    void Next()
    {
        int i;
        for(i=0 ; i<mRanges.size() ; i++)
        {
            Util::Iterable* curr = mRanges[i];
            curr->Next();
            if(curr->Done())
                curr->SetIteration(0);
            else
                break;
        }
        if(i==mRanges.size())
            mDone = true;
    }

    bool IsDone()
    {
        return mDone;
    }

private:
    void InitialiseDistribution(int thisCpu, int nrCpus)
    {
        // for each loop (parameter range) assign a number of cpus and continue with the loop that contains this cpu.
        mCurrentIteration = (GetTotalIterations() * thisCpu) / nrCpus;
        mLastIteration = ((GetTotalIterations() * (thisCpu+1)) / nrCpus) - 1;
    }

    int mCurrentIteration;
    int mLastIteration;
    bool mDone;

    std::vector<Iterable*> mRanges;
    ILOG_VAR_DEC;
};

ILOG_VAR_INIT(RangeIterator, Impala.Util);

std::ostream& operator<<(std::ostream& os, RangeIterator& ri)
{
    PropertySet ps;
    ri.GetParameters(&ps);
    ps.Print(os);
    return os;
}

}//namespace Util
}//namespace Impala

#endif

---