File: RectangleOverlapHandler.h

package info (click to toggle)
spring 106.0%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 55,260 kB
  • sloc: cpp: 543,946; ansic: 44,800; python: 12,575; java: 12,201; awk: 5,889; sh: 1,796; asm: 1,546; xml: 655; perl: 405; php: 211; objc: 194; makefile: 76; sed: 2
file content (127 lines) | stat: -rw-r--r-- 3,021 bytes parent folder | download | duplicates (3)
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/* This file is part of the Spring engine (GPL v2 or later), see LICENSE.html */

#ifndef RECTANGLE_OVERLAP_HANDLER_H
#define RECTANGLE_OVERLAP_HANDLER_H

#include <vector>
#include <bitset>

#include "System/Rectangle.h"
#include "System/creg/creg_cond.h"


/**
 * @brief CRectangleOverlapHandler
 *
 * Container & preprocessor for rectangles. It handles any overlap & merges+resizes rectangles.
 */
class CRectangleOverlapHandler
{
	CR_DECLARE_STRUCT(CRectangleOverlapHandler)

public:
	CRectangleOverlapHandler() { clear(); }
	~CRectangleOverlapHandler();

	void Process();

	size_t GetTotalArea() const;

public:
	typedef std::vector<SRectangle> container;
	typedef container::iterator iterator;
	typedef container::const_iterator const_iterator;

	bool empty() const { return (frontIdx >= rectangles.size()); }
	size_t size() const { return (rectangles.size() - frontIdx); }

	SRectangle& front() { return rectangles.at(frontIdx); }
	SRectangle& back() { return rectangles.back(); }

	void pop_front() {
		if (!empty()) {
			// leave a null-rectangle so RemoveEmptyRects will clean it up
			rectangles[frontIdx++] = {};
			return;
		}

		clear();
	}
	void push_back(const SRectangle& rect) {
		// skip zero- or negative-area rectangles
		// assert(rect.GetArea() > 0);
		if (rect.GetArea() <= 0)
			return;
		needsUpdate = true;
		rectangles.push_back(rect);
	}

	void swap(CRectangleOverlapHandler& other) {
		std::swap(rectangles, other.rectangles);
		std::swap(frontIdx, other.frontIdx);
		std::swap(needsUpdate, other.needsUpdate);
	}

	void append(CRectangleOverlapHandler& other) {
		needsUpdate = (other.needsUpdate || !empty());

		for (const SRectangle& r: other.rectangles) {
			rectangles.push_back(r);
		}

		other.clear();
	}

	void clear() {
		frontIdx = 0;
		needsUpdate = false;

		rectangles.clear();
		rectangles.reserve(512);
	}

	const_iterator cbegin() { return (rectangles.cbegin() + frontIdx); }
	const_iterator cend() { return (rectangles.cend()); }

	iterator begin() { return (rectangles.begin() + frontIdx); }
	iterator end() { return (rectangles.end()); }

private:
	void StageMerge();
	void StageOverlap();
	void StageSplitTooLarge();
	void RemoveEmptyRects() {
		size_t j = (frontIdx = 0);

		for (size_t i = j, n = rectangles.size(); i < n; i++) {
			if (rectangles[i].GetArea() <= 0)
				continue;

			rectangles[j++] = rectangles[i];
		}

		// shrink without reallocating
		rectangles.resize(j);
	}

	bool HandleMerge(SRectangle& rect1, SRectangle& rect2);
	int HandleOverlapping(SRectangle* rect1, SRectangle* rect2);
	static std::bitset<4> GetEdgesInRect(const SRectangle& rect1, const SRectangle& rect2);
	static std::bitset<4> GetSharedEdges(const SRectangle& rect1, const SRectangle& rect2);
	static bool AreMergable(const SRectangle& rect1, const SRectangle& rect2);

private:
	container rectangles;

	constexpr static int maxAreaPerRect = 500 * 500;

	static size_t statsTotalSize;
	static size_t statsOptimSize;

	size_t frontIdx = 0;

	bool needsUpdate = false;
};

#endif