开发者

How to write a class capable of foreach

开发者 https://www.devze.com 2022-12-09 05:50 出处:网络
It\'s been a while since Visual Studio added support for a foreach extension that works like vector<int> v(3)

It's been a while since Visual Studio added support for a foreach extension that works like

vector<int> v(3)
for each (int i in开发者_如何学Go v) {
  printf("%d\n",i);
}

I want to know how to make any class able to use foreach. Do I need to implement some interface?


for each statement in VC++, when used on a non-managed class:

for each (T x in xs)
{
    ...
}

is just syntactic sugar for this:

for (auto iter = xs.begin(), end = xs.end(); iter != end; ++iter)
{
     T x = *iter;
}

Where auto means that type of variable is deduced automatically from type of initializer.

In other words, you need to provide begin() and end() methods on your class that would return begin and end input iterators for it.

Here is an example of class that wraps an istream and allows you to iterate over all lines in it:

#include <istream>
#include <iostream>
#include <fstream>
#include <string>


class lines
{
public:

    class line_iterator
    {
    public:

        line_iterator() : in(0)
        {
        }

        line_iterator(std::istream& in) : in(&in)
        {
            ++*this;
        }

        line_iterator& operator++ ()
        {
            getline(*in, line);
            return *this;
        }

        line_iterator operator++ (int)
        {
            line_iterator result = *this;
            ++*this;
            return result;
        }

        const std::string& operator* () const
        {
            return line;
        }

        const std::string& operator-> () const
        {
            return line;
        }

        friend bool operator== (const line_iterator& lhs, const line_iterator& rhs)
        {
            return (lhs.in == rhs.in) ||
                   (lhs.in == 0 && rhs.in->eof()) ||
                   (rhs.in == 0 && lhs.in->eof());
        }

        friend bool operator!= (const line_iterator& lhs, const line_iterator& rhs)
        {
            return !(lhs == rhs);
        }

    private:

        std::istream* const in;
        std::string line;
    };


    lines(std::istream& in) : in(in)
    {
    }

    line_iterator begin() const
    {
        return line_iterator(in);
    }

    line_iterator end() const
    {
        return line_iterator();
    }

private:

    std::istream& in;
};


int main()
{
    std::ifstream f(__FILE__);
    for each (std::string line in lines(f))
    {
        std::cout << line << std::endl;
    }
}

Note that implementation of line_iterator is actually somewhat bigger than the minimum needed by for each; however, it is the minimum implementation that conforms to input iterator requirements, and thus this class is also usable with all STL algorithms that work on input iterators, such as std::for_each, std::find etc.


You can use this

std::for_each


foreach isn't part of the C++ language as far as I know. It's in C# though. Also, I think STL and/or Boost has a foreach method. Perhaps you're thinking about that one?


Your class should inherit IEnumerable to use foreach

0

精彩评论

暂无评论...
验证码 换一张
取 消