使c自定义foreach宏更干净

2019年8月6日 132次阅读

我有一个自定义的c宏来模拟foreach循环：

#define foreach(TYPE, ELEMENT, COLLECTION_TYPE, COLLECTION)\
for(COLLECTION_TYPE::iterator ELEMENT##__MACRO_TEMP_IT = COLLECTION.begin(); ELEMENT##__MACRO_TEMP_IT != COLLECTION.end(); ELEMENT##__MACRO_TEMP_IT++) \
{ TYPE ELEMENT = *(ELEMENT##__MACRO_TEMP_IT);

我知道有另外的方法可以做一个foreach循环 – 无论是使用C 11,STL,Qt还是Boost的foreach,但我只是为了体验而尝试自定义解决方案.

问题是,这个宏需要以2个大括号(“}}”结尾或者省略第一个大括号并以一个结尾,如下所示：

foreach(int, i, std::list<int>, indexes)
{
   //do stuff
}}

要么

foreach(int, i, std::list<int>, indexes)

   //do stuff
}

我想知道：有没有一个智能的宏解决方法,所以可以使用这个宏如下？

foreach(int, i, std::list<int>, indexes)
{
   //do stuff
}

最佳答案

#define foreach(TYPE, ELEMENT, COLLECTION_TYPE, COLLECTION)\
for(COLLECTION_TYPE::iterator ELEMENT##MACRO_TEMP_IT = (COLLECTION).begin(); ELEMENT##MACRO_TEMP_IT != (COLLECTION).end(); ++ELEMENT##MACRO_TEMP_IT)\
for(bool ELEMENT##MACRO_B = true; ELEMENT##MACRO_B;)\
for(TYPE ELEMENT = *(ELEMENT##MACRO_TEMP_IT); ELEMENT##MACRO_B; ELEMENT##MACRO_B = false)

测试：

#include <iostream>
#include <list>

int main()
{
    std::list<int> indexes{1, 2, 3};

    foreach(int, i, std::list<int>, indexes)
    {
        std::cout << i << std::endl;
    }

    foreach(int, i, std::list<int>, indexes)
        std::cout << i << std::endl;

    std::list<std::list<int>> listOfLists{{1, 2}, {3, 4}};
    foreach(std::list<int>&, li, std::list<std::list<int>>, listOfLists)
        foreach(int, i, std::list<int>, li)
            std::cout << i << std::endl;
}

DEMO

点赞