File: pforeach.h

package info (click to toggle)
vcmi 1.6.5%2Bdfsg-2
  • links: PTS, VCS
  • area: contrib
  • in suites: forky, sid, trixie
  • size: 32,060 kB
  • sloc: cpp: 238,971; python: 265; sh: 224; xml: 157; ansic: 78; objc: 61; makefile: 49
file content (50 lines) | stat: -rw-r--r-- 997 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#pragma once

#include "Engine/Nullkiller.h"

namespace NKAI
{

template<typename TFunc>
void pforeachTilePos(const int3 & mapSize, TFunc fn)
{
	for(int z = 0; z < mapSize.z; ++z)
	{
		tbb::parallel_for(tbb::blocked_range<size_t>(0, mapSize.x), [&](const tbb::blocked_range<size_t> & r)
			{
				int3 pos(0, 0, z);

				for(pos.x = r.begin(); pos.x != r.end(); ++pos.x)
				{
					for(pos.y = 0; pos.y < mapSize.y; ++pos.y)
					{
						fn(pos);
					}
				}
			});
	}
}

template<typename TFunc>
void pforeachTilePaths(const int3 & mapSize, const Nullkiller * ai, TFunc fn)
{
	for(int z = 0; z < mapSize.z; ++z)
	{
		tbb::parallel_for(tbb::blocked_range<size_t>(0, mapSize.x), [&](const tbb::blocked_range<size_t> & r)
			{
				int3 pos(0, 0, z);
				std::vector<AIPath> paths;

				for(pos.x = r.begin(); pos.x != r.end(); ++pos.x)
				{
					for(pos.y = 0; pos.y < mapSize.y; ++pos.y)
					{
						ai->pathfinder->calculatePathInfo(paths, pos);
						fn(pos, paths);
					}
				}
			});
	}
}

}