1 | /* |
---|
2 | * ORXONOX - the hottest 3D action shooter ever to exist |
---|
3 | * |
---|
4 | * |
---|
5 | * License notice: |
---|
6 | * |
---|
7 | * This program is free software; you can redistribute it and/or |
---|
8 | * modify it under the terms of the GNU General Public License |
---|
9 | * as published by the Free Software Foundation; either version 2 |
---|
10 | * of the License, or (at your option) any later version. |
---|
11 | * |
---|
12 | * This program is distributed in the hope that it will be useful, |
---|
13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
---|
14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
---|
15 | * GNU General Public License for more details. |
---|
16 | * |
---|
17 | * You should have received a copy of the GNU General Public License |
---|
18 | * along with this program; if not, write to the Free Software |
---|
19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. |
---|
20 | * |
---|
21 | * Author: |
---|
22 | * Fabian 'x3n' Landau |
---|
23 | * Co-authors: |
---|
24 | * ... |
---|
25 | * |
---|
26 | */ |
---|
27 | |
---|
28 | /** |
---|
29 | @file ClassTreeMask.cc |
---|
30 | @brief Implementation of the ClassTreeMask, ClassTreeMaskNode and ClassTreeMaskIterator classes. |
---|
31 | */ |
---|
32 | |
---|
33 | #include "ClassTreeMask.h" |
---|
34 | #include "Identifier.h" |
---|
35 | #include "BaseObject.h" |
---|
36 | |
---|
37 | namespace orxonox |
---|
38 | { |
---|
39 | // ############################### |
---|
40 | // ### ClassTreeMaskNode ### |
---|
41 | // ############################### |
---|
42 | /** |
---|
43 | @brief Constructor: Creates the node, sets the subclass and the rule. |
---|
44 | @param subclass The subclass the rule refers to |
---|
45 | @param bIncluded The rule: included (true) or excluded (false) |
---|
46 | */ |
---|
47 | ClassTreeMaskNode::ClassTreeMaskNode(const Identifier* subclass, bool bIncluded) |
---|
48 | { |
---|
49 | this->subclass_ = subclass; |
---|
50 | this->bIncluded_ = bIncluded; |
---|
51 | } |
---|
52 | |
---|
53 | /** |
---|
54 | @brief Destructor: Deletes all subnodes. |
---|
55 | */ |
---|
56 | ClassTreeMaskNode::~ClassTreeMaskNode() |
---|
57 | { |
---|
58 | // Go through the list of all subnodes and delete them |
---|
59 | this->deleteAllSubnodes(); |
---|
60 | } |
---|
61 | |
---|
62 | /** |
---|
63 | @brief Sets the rule for the node to "included". |
---|
64 | @param overwrite True = overwrite previously added rules for inheriting classes |
---|
65 | */ |
---|
66 | void ClassTreeMaskNode::include(bool overwrite) |
---|
67 | { |
---|
68 | this->setIncluded(true, overwrite); |
---|
69 | } |
---|
70 | |
---|
71 | /** |
---|
72 | @brief Sets the rule for the node to "excluded". |
---|
73 | @param overwrite True = overwrite previously added rules for inheriting classes |
---|
74 | */ |
---|
75 | void ClassTreeMaskNode::exclude(bool overwrite) |
---|
76 | { |
---|
77 | this->setIncluded(false, overwrite); |
---|
78 | } |
---|
79 | |
---|
80 | /** |
---|
81 | @brief Sets the rule for the node to a given value and erases all following rules. |
---|
82 | @param bIncluded The rule: included (true) or excluded (false) |
---|
83 | @param overwrite True = overwrite previously added rules for inheriting classes |
---|
84 | */ |
---|
85 | void ClassTreeMaskNode::setIncluded(bool bIncluded, bool overwrite) |
---|
86 | { |
---|
87 | if (overwrite) |
---|
88 | this->deleteAllSubnodes(); |
---|
89 | |
---|
90 | this->bIncluded_ = bIncluded; |
---|
91 | } |
---|
92 | |
---|
93 | /** |
---|
94 | @brief Adds a new subnode to the list of subnodes. |
---|
95 | @param subnode The new subnode |
---|
96 | */ |
---|
97 | void ClassTreeMaskNode::addSubnode(ClassTreeMaskNode* subnode) |
---|
98 | { |
---|
99 | this->subnodes_.insert(this->subnodes_.end(), subnode); |
---|
100 | } |
---|
101 | |
---|
102 | /** |
---|
103 | @brief Tells if the rule is "included" or not. |
---|
104 | @return The rule: true = included, false = excluded |
---|
105 | */ |
---|
106 | bool ClassTreeMaskNode::isIncluded() const |
---|
107 | { |
---|
108 | return this->bIncluded_; |
---|
109 | } |
---|
110 | |
---|
111 | /** |
---|
112 | @brief Tells if the rule is "excluded" or not. |
---|
113 | @return The inverted rule: true = excluded, false = included |
---|
114 | */ |
---|
115 | bool ClassTreeMaskNode::isExcluded() const |
---|
116 | { |
---|
117 | return (!this->bIncluded_); |
---|
118 | } |
---|
119 | |
---|
120 | /** |
---|
121 | @brief Returns the Identifier of the class the rule refers to. |
---|
122 | @return The Identifier representing the class |
---|
123 | */ |
---|
124 | const Identifier* ClassTreeMaskNode::getClass() const |
---|
125 | { |
---|
126 | return this->subclass_; |
---|
127 | } |
---|
128 | |
---|
129 | /** |
---|
130 | @brief Deletes all subnodes of this node. |
---|
131 | */ |
---|
132 | void ClassTreeMaskNode::deleteAllSubnodes() |
---|
133 | { |
---|
134 | // Go through the list of all subnodes and delete them |
---|
135 | for (std::list<ClassTreeMaskNode*>::iterator it = this->subnodes_.begin(); it != this->subnodes_.end(); ) |
---|
136 | delete (*(it++)); |
---|
137 | |
---|
138 | // Clear the list |
---|
139 | this->subnodes_.clear(); |
---|
140 | } |
---|
141 | |
---|
142 | |
---|
143 | // ############################### |
---|
144 | // ### ClassTreeMaskIterator ### |
---|
145 | // ############################### |
---|
146 | /** |
---|
147 | @brief Constructor: Initializes the iterator by creating a helper-list with the root-node and putting it to the stack. |
---|
148 | @param node The root-node |
---|
149 | */ |
---|
150 | ClassTreeMaskIterator::ClassTreeMaskIterator(ClassTreeMaskNode* node) |
---|
151 | { |
---|
152 | // Create a list and put the root-note into it |
---|
153 | std::list<ClassTreeMaskNode*>::iterator it = this->rootlist_.insert(this->rootlist_.end(), node); |
---|
154 | |
---|
155 | // Put the iterator to the only element of the list and the corresponding end()-iterator on the stack |
---|
156 | this->nodes_.push(std::pair<std::list<ClassTreeMaskNode*>::iterator, std::list<ClassTreeMaskNode*>::iterator>(it, this->rootlist_.end())); |
---|
157 | } |
---|
158 | |
---|
159 | /** |
---|
160 | @brief Destructor: Does nothing. |
---|
161 | */ |
---|
162 | ClassTreeMaskIterator::~ClassTreeMaskIterator() |
---|
163 | { |
---|
164 | } |
---|
165 | |
---|
166 | /** |
---|
167 | @brief Iterates through the rule-tree. |
---|
168 | @return A reference to the iterator itself |
---|
169 | */ |
---|
170 | ClassTreeMaskIterator& ClassTreeMaskIterator::operator++() |
---|
171 | { |
---|
172 | // Check if the actual node has subnodes |
---|
173 | if ((*this->nodes_.top().first)->subnodes_.begin() != (*this->nodes_.top().first)->subnodes_.end()) |
---|
174 | { |
---|
175 | // Yes it has: Push an iterator, pointing at the first subnode, on the stack |
---|
176 | this->nodes_.push(std::pair<std::list<ClassTreeMaskNode*>::iterator, std::list<ClassTreeMaskNode*>::iterator>((*this->nodes_.top().first)->subnodes_.begin(), (*this->nodes_.top().first)->subnodes_.end())); |
---|
177 | } |
---|
178 | else |
---|
179 | { |
---|
180 | // No subnodes, meaning we reached a leaf: Go to the next node |
---|
181 | do |
---|
182 | { |
---|
183 | // Iterate one step in the current list |
---|
184 | ++this->nodes_.top().first; |
---|
185 | |
---|
186 | // Check if we reached the end of the list (the second item in the stored pair always represents the end) |
---|
187 | if (this->nodes_.top().first == this->nodes_.top().second) |
---|
188 | { |
---|
189 | // Yes we've reached the end: Pop the list-iterator from the stack |
---|
190 | this->nodes_.pop(); |
---|
191 | |
---|
192 | // Continue with the loop, meaning: Try to iterate through the previous list |
---|
193 | continue; |
---|
194 | } |
---|
195 | |
---|
196 | // If we reached this point, we aren't yet at the end of the list: Leave the loop |
---|
197 | break; |
---|
198 | } while (!this->nodes_.empty()); // Stop looping if the stack is empty, meaning: We've iterated to the end |
---|
199 | } |
---|
200 | |
---|
201 | // Finally return a reference to the iterator itself |
---|
202 | return *this; |
---|
203 | } |
---|
204 | |
---|
205 | /** |
---|
206 | @brief Returns a pointer to the ClassTreeMaskNode whereon the iterator points. |
---|
207 | @return The pointer to the node |
---|
208 | */ |
---|
209 | ClassTreeMaskNode* ClassTreeMaskIterator::operator*() const |
---|
210 | { |
---|
211 | return (*this->nodes_.top().first); |
---|
212 | } |
---|
213 | |
---|
214 | /** |
---|
215 | @brief Returns a pointer to the ClassTreeMaskNode whereon the iterator points. |
---|
216 | @return The pointer to the node |
---|
217 | */ |
---|
218 | ClassTreeMaskNode* ClassTreeMaskIterator::operator->() const |
---|
219 | { |
---|
220 | return (*this->nodes_.top().first); |
---|
221 | } |
---|
222 | |
---|
223 | /** |
---|
224 | @brief Returns true if the stack is empty, meaning we've reached the end of the tree. |
---|
225 | @return True if we've reached the end of the tree |
---|
226 | */ |
---|
227 | ClassTreeMaskIterator::operator bool() |
---|
228 | { |
---|
229 | return (!this->nodes_.empty()); |
---|
230 | } |
---|
231 | |
---|
232 | /** |
---|
233 | @brief Compares the current node with the given one and returns true if they match. |
---|
234 | @param compare The node to compare with |
---|
235 | @return The result of the comparison (true if they match) |
---|
236 | */ |
---|
237 | bool ClassTreeMaskIterator::operator==(ClassTreeMaskNode* compare) |
---|
238 | { |
---|
239 | if (!this->nodes_.empty()) |
---|
240 | return ((*this->nodes_.top().first) == compare); |
---|
241 | else |
---|
242 | return (compare == 0); |
---|
243 | } |
---|
244 | |
---|
245 | /** |
---|
246 | @brief Compares the current node with the given one and returns true if they don't match. |
---|
247 | @param compare The node to compare with |
---|
248 | @return The result of the comparison (true if they don't match) |
---|
249 | */ |
---|
250 | bool ClassTreeMaskIterator::operator!=(ClassTreeMaskNode* compare) |
---|
251 | { |
---|
252 | if (!this->nodes_.empty()) |
---|
253 | return ((*this->nodes_.top().first) != compare); |
---|
254 | else |
---|
255 | return (compare != 0); |
---|
256 | } |
---|
257 | |
---|
258 | |
---|
259 | // ############################### |
---|
260 | // ### ClassTreeMask ### |
---|
261 | // ############################### |
---|
262 | /** |
---|
263 | @brief Constructor: Adds the root-node of the tree with the first rule ("include everything"). |
---|
264 | */ |
---|
265 | ClassTreeMask::ClassTreeMask() |
---|
266 | { |
---|
267 | this->root_ = new ClassTreeMaskNode(ClassManager<BaseObject>::getIdentifier(), true); |
---|
268 | } |
---|
269 | |
---|
270 | /** |
---|
271 | @brief Copyconstructor: Adds the root-node of the tree with the first rule ("include everything") and adds all rules from the other mask. |
---|
272 | @param other The other mask |
---|
273 | */ |
---|
274 | ClassTreeMask::ClassTreeMask(const ClassTreeMask& other) |
---|
275 | { |
---|
276 | this->root_ = new ClassTreeMaskNode(ClassManager<BaseObject>::getIdentifier(), true); |
---|
277 | for (ClassTreeMaskIterator it = other.root_; it; ++it) |
---|
278 | this->add(it->getClass(), it->isIncluded(), false); |
---|
279 | } |
---|
280 | |
---|
281 | /** |
---|
282 | @brief Destructor: Deletes the root node (which will delete all subnodes too). |
---|
283 | */ |
---|
284 | ClassTreeMask::~ClassTreeMask() |
---|
285 | { |
---|
286 | delete this->root_; |
---|
287 | } |
---|
288 | |
---|
289 | /** |
---|
290 | @brief Adds a new "include" rule for a given subclass to the mask. |
---|
291 | @param subclass The subclass |
---|
292 | @param overwrite True = overwrite previously added rules for inheriting classes |
---|
293 | @param clean True = clean the tree after adding the new rule |
---|
294 | */ |
---|
295 | void ClassTreeMask::include(const Identifier* subclass, bool overwrite, bool clean) |
---|
296 | { |
---|
297 | this->add(subclass, true, overwrite, clean); |
---|
298 | } |
---|
299 | |
---|
300 | /** |
---|
301 | @brief Adds a new "exclude" rule for a given subclass to the mask. |
---|
302 | @param subclass The subclass |
---|
303 | @param overwrite True = overwrite previously added rules for inheriting classes |
---|
304 | @param clean True = clean the tree after adding the new rule |
---|
305 | */ |
---|
306 | void ClassTreeMask::exclude(const Identifier* subclass, bool overwrite, bool clean) |
---|
307 | { |
---|
308 | this->add(subclass, false, overwrite, clean); |
---|
309 | } |
---|
310 | |
---|
311 | /** |
---|
312 | @brief Adds a new rule for a given subclass to the mask. |
---|
313 | @param subclass The subclass |
---|
314 | @param bInclude The rule: include (true) or exclude (false) |
---|
315 | @param overwrite True = overwrite previously added rules for inheriting classes |
---|
316 | @param clean True = clean the tree after adding the new rule |
---|
317 | */ |
---|
318 | void ClassTreeMask::add(const Identifier* subclass, bool bInclude, bool overwrite, bool clean) |
---|
319 | { |
---|
320 | // Check if the given subclass is a child of our root-class |
---|
321 | if (subclass->isA(this->root_->getClass())) |
---|
322 | { |
---|
323 | // Yes it is: Just add the rule to the three |
---|
324 | this->add(this->root_, subclass, bInclude, overwrite); |
---|
325 | } |
---|
326 | else |
---|
327 | { |
---|
328 | // No it's not: Search for classes inheriting from the given class and add the rules for them |
---|
329 | for (std::list<const Identifier*>::const_iterator it = subclass->getDirectChildrenBegin(); it != subclass->getDirectChildrenEnd(); ++it) |
---|
330 | if ((*it)->isA(this->root_->getClass())) |
---|
331 | if (overwrite || (!this->nodeExists(*it))) // If we don't want to overwrite, only add nodes that don't already exist |
---|
332 | this->add(this->root_, *it, bInclude, overwrite); |
---|
333 | } |
---|
334 | |
---|
335 | // Clean the rule-tree |
---|
336 | if (clean) |
---|
337 | this->clean(); |
---|
338 | } |
---|
339 | |
---|
340 | /** |
---|
341 | @brief Adds a new rule for a given subclass to a node of the internal rule-tree (recursive function). |
---|
342 | @param node The node |
---|
343 | @param subclass The subclass |
---|
344 | @param bInclude The rule: include (true) or exclude (false) |
---|
345 | @param overwrite True = overwrite previously added rules for inheriting classes |
---|
346 | */ |
---|
347 | void ClassTreeMask::add(ClassTreeMaskNode* node, const Identifier* subclass, bool bInclude, bool overwrite) |
---|
348 | { |
---|
349 | // Check if the current node contains exactly the subclass we want to add |
---|
350 | if (subclass == node->getClass()) |
---|
351 | { |
---|
352 | // We're at the right place, just change the mask and return |
---|
353 | node->setIncluded(bInclude, overwrite); |
---|
354 | return; |
---|
355 | } |
---|
356 | else if (subclass->isA(node->getClass())) |
---|
357 | { |
---|
358 | // Search for an already existing node, containing the subclass we want to add |
---|
359 | for (std::list<ClassTreeMaskNode*>::iterator it = node->subnodes_.begin(); it != node->subnodes_.end(); ++it) |
---|
360 | { |
---|
361 | if (subclass->isA((*it)->getClass())) |
---|
362 | { |
---|
363 | // We've found an existing node -> delegate the work with a recursive function-call and return |
---|
364 | this->add(*it, subclass, bInclude, overwrite); |
---|
365 | return; |
---|
366 | } |
---|
367 | } |
---|
368 | |
---|
369 | // There is no existing node satisfying our needs -> we create a new node |
---|
370 | ClassTreeMaskNode* newnode = new ClassTreeMaskNode(subclass, bInclude); |
---|
371 | |
---|
372 | // Search for nodes that should actually be subnodes of our new node and erase them |
---|
373 | for (std::list<ClassTreeMaskNode*>::iterator it = node->subnodes_.begin(); it != node->subnodes_.end(); ) |
---|
374 | { |
---|
375 | if ((*it)->getClass()->isChildOf(subclass)) |
---|
376 | { |
---|
377 | // We've found a subnode: add it to the new node and erase it from the current node |
---|
378 | if (!overwrite) |
---|
379 | newnode->addSubnode(*it); |
---|
380 | else |
---|
381 | delete (*it); |
---|
382 | |
---|
383 | node->subnodes_.erase(it++); |
---|
384 | } |
---|
385 | else |
---|
386 | { |
---|
387 | ++it; |
---|
388 | } |
---|
389 | } |
---|
390 | |
---|
391 | // Finally add the new node as a subnode to the current node |
---|
392 | node->addSubnode(newnode); |
---|
393 | } |
---|
394 | } |
---|
395 | |
---|
396 | /** |
---|
397 | @brief Adds a new "include" rule for a single subclass. The new rule doesn't change the mask for inheriting classes. |
---|
398 | @param subclass The subclass |
---|
399 | @param clean True = clean the tree after adding the new rule |
---|
400 | */ |
---|
401 | void ClassTreeMask::includeSingle(const Identifier* subclass, bool clean) |
---|
402 | { |
---|
403 | this->addSingle(subclass, true, clean); |
---|
404 | } |
---|
405 | |
---|
406 | /** |
---|
407 | @brief Adds a new "exclude" rule for a single subclass. The new rule doesn't change the mask for inheriting classes. |
---|
408 | @param subclass The subclass |
---|
409 | @param clean True = clean the tree after adding the new rule |
---|
410 | */ |
---|
411 | void ClassTreeMask::excludeSingle(const Identifier* subclass, bool clean) |
---|
412 | { |
---|
413 | this->addSingle(subclass, false, clean); |
---|
414 | } |
---|
415 | |
---|
416 | /** |
---|
417 | @brief Adds a new rule for a single subclass. The new rule doesn't change the mask for inheriting classes. |
---|
418 | @param bInclude The rule: include (true) or exclude (false) |
---|
419 | @param subclass The subclass |
---|
420 | @param clean True = clean the tree after adding the new rule |
---|
421 | */ |
---|
422 | void ClassTreeMask::addSingle(const Identifier* subclass, bool bInclude, bool clean) |
---|
423 | { |
---|
424 | for (std::list<const Identifier*>::const_iterator it = subclass->getDirectChildrenBegin(); it != subclass->getDirectChildrenEnd(); ++it) |
---|
425 | this->add(*it, this->isIncluded(*it), false, false); |
---|
426 | |
---|
427 | this->add(subclass, bInclude, false, clean); |
---|
428 | } |
---|
429 | |
---|
430 | /** |
---|
431 | @brief Resets the mask to "include everything". |
---|
432 | */ |
---|
433 | void ClassTreeMask::reset() |
---|
434 | { |
---|
435 | delete this->root_; |
---|
436 | this->root_ = new ClassTreeMaskNode(ClassManager<BaseObject>::getIdentifier(), true); |
---|
437 | } |
---|
438 | |
---|
439 | /** |
---|
440 | @brief Tells if a given subclass is included or not. |
---|
441 | @param subclass The subclass |
---|
442 | @return Included (true) or excluded (false) |
---|
443 | */ |
---|
444 | bool ClassTreeMask::isIncluded(const Identifier* subclass) const |
---|
445 | { |
---|
446 | return this->isIncluded(this->root_, subclass); |
---|
447 | } |
---|
448 | |
---|
449 | /** |
---|
450 | @brief Tells if a given subclass of a node in the rule-tree is included or not (recursive function). |
---|
451 | @param node The node |
---|
452 | @param subclass The subclass |
---|
453 | @return Included (true) or excluded (false) |
---|
454 | */ |
---|
455 | bool ClassTreeMask::isIncluded(ClassTreeMaskNode* node, const Identifier* subclass) const |
---|
456 | { |
---|
457 | // Check if the searched subclass is of the same type as the class in the current node or a derivative |
---|
458 | if (subclass->isA(node->getClass())) |
---|
459 | { |
---|
460 | // Check for the special case |
---|
461 | if (subclass == node->getClass()) |
---|
462 | return node->isIncluded(); |
---|
463 | |
---|
464 | // Go through the list of subnodes and look for a node containing the searched subclass and delegate the request by a recursive function-call. |
---|
465 | for (std::list<ClassTreeMaskNode*>::iterator it = node->subnodes_.begin(); it != node->subnodes_.end(); ++it) |
---|
466 | if (subclass->isA((*it)->getClass())) |
---|
467 | return isIncluded(*it, subclass); |
---|
468 | |
---|
469 | // There is no subnode containing our class -> the rule of the current node takes in effect |
---|
470 | return node->isIncluded(); |
---|
471 | } |
---|
472 | else |
---|
473 | { |
---|
474 | // The class is not included in the mask: return false |
---|
475 | return false; |
---|
476 | } |
---|
477 | } |
---|
478 | |
---|
479 | /** |
---|
480 | @brief Tells if a given subclass is excluded or not. |
---|
481 | @param subclass The subclass |
---|
482 | @return The inverted rule: Excluded (true) or included (false) |
---|
483 | */ |
---|
484 | bool ClassTreeMask::isExcluded(const Identifier* subclass) const |
---|
485 | { |
---|
486 | return (!this->isIncluded(subclass)); |
---|
487 | } |
---|
488 | |
---|
489 | /** |
---|
490 | @brief Removes all unneeded rules that don't change the information of the mask. |
---|
491 | */ |
---|
492 | void ClassTreeMask::clean() |
---|
493 | { |
---|
494 | this->clean(this->root_); |
---|
495 | } |
---|
496 | |
---|
497 | /** |
---|
498 | @brief Removes all unneded rules that don't change the information of a node of a mask (recursive function). |
---|
499 | @param node The node |
---|
500 | */ |
---|
501 | void ClassTreeMask::clean(ClassTreeMaskNode* node) |
---|
502 | { |
---|
503 | // Iterate through all subnodes of the given node |
---|
504 | for (std::list<ClassTreeMaskNode*>::iterator it = node->subnodes_.begin(); it != node->subnodes_.end(); ) |
---|
505 | { |
---|
506 | // Recursive function-call: Clean the subnode |
---|
507 | this->clean(*it); |
---|
508 | |
---|
509 | // Now check if the subnode contains the same rule as the current node |
---|
510 | if ((*it)->isIncluded() == node->isIncluded()) |
---|
511 | { |
---|
512 | // It does: Move all subnodes of the redundant subnode to the current node |
---|
513 | node->subnodes_.insert(node->subnodes_.end(), (*it)->subnodes_.begin(), (*it)->subnodes_.end()); |
---|
514 | (*it)->subnodes_.clear(); |
---|
515 | |
---|
516 | // Remove the redundant subnode from the current node |
---|
517 | node->subnodes_.erase(it++); |
---|
518 | } |
---|
519 | else |
---|
520 | { |
---|
521 | // The subnode is necessary: Move on to the next subnode |
---|
522 | ++it; |
---|
523 | } |
---|
524 | } |
---|
525 | } |
---|
526 | |
---|
527 | /** |
---|
528 | @brief Checks if a node for the given subclass exists. |
---|
529 | @param subclass The Identifier of the subclass |
---|
530 | @return True = the node exists |
---|
531 | */ |
---|
532 | bool ClassTreeMask::nodeExists(const Identifier* subclass) |
---|
533 | { |
---|
534 | for (ClassTreeMaskIterator it = this->root_; it; ++it) |
---|
535 | if ((*it)->getClass() == subclass) |
---|
536 | return true; |
---|
537 | |
---|
538 | return false; |
---|
539 | } |
---|
540 | |
---|
541 | /** |
---|
542 | @brief Assignment operator: Adds all rules of the other mask. |
---|
543 | @param other The other mask |
---|
544 | @return A reference to the mask itself |
---|
545 | */ |
---|
546 | ClassTreeMask& ClassTreeMask::operator=(const ClassTreeMask& other) |
---|
547 | { |
---|
548 | // Make a copy to avoid troubles with self-assignments (like A = A). |
---|
549 | ClassTreeMask temp(other); |
---|
550 | |
---|
551 | // Removes all current rules |
---|
552 | this->reset(); |
---|
553 | |
---|
554 | // Copy all rules from the other mask |
---|
555 | for (ClassTreeMaskIterator it = temp.root_; it; ++it) |
---|
556 | this->add(it->getClass(), it->isIncluded(), false, false); |
---|
557 | |
---|
558 | // Return a reference to the mask itself |
---|
559 | return (*this); |
---|
560 | } |
---|
561 | |
---|
562 | /** |
---|
563 | @brief Prefix operator + does nothing. |
---|
564 | @return A reference to the mask itself |
---|
565 | */ |
---|
566 | ClassTreeMask& ClassTreeMask::operator+() |
---|
567 | { |
---|
568 | return (*this); |
---|
569 | } |
---|
570 | |
---|
571 | /** |
---|
572 | @brief Prefix operator - inverts the mask. |
---|
573 | @return The inverted mask |
---|
574 | */ |
---|
575 | ClassTreeMask ClassTreeMask::operator-() const |
---|
576 | { |
---|
577 | return (!(*this)); |
---|
578 | } |
---|
579 | |
---|
580 | /** |
---|
581 | @brief Adds two masks in the sense of a union (all classes that are included in at least one of the masks will be included in the resulting mask too). |
---|
582 | @param other The mask to unite with |
---|
583 | @return The union |
---|
584 | */ |
---|
585 | ClassTreeMask ClassTreeMask::operator+(const ClassTreeMask& other) const |
---|
586 | { |
---|
587 | // Create a new mask |
---|
588 | ClassTreeMask newmask; |
---|
589 | |
---|
590 | // Add all nodes from the first mask, calculate the rule with the or-operation |
---|
591 | for (ClassTreeMaskIterator it = this->root_; it; ++it) |
---|
592 | { |
---|
593 | const Identifier* subclass = it->getClass(); |
---|
594 | newmask.add(subclass, this->isIncluded(subclass) || other.isIncluded(subclass), false, false); |
---|
595 | } |
---|
596 | |
---|
597 | // Add all nodes from the second mask, calculate the rule with the or-operation |
---|
598 | for (ClassTreeMaskIterator it = other.root_; it; ++it) |
---|
599 | { |
---|
600 | const Identifier* subclass = it->getClass(); |
---|
601 | newmask.add(subclass, this->isIncluded(subclass) || other.isIncluded(subclass), false, false); |
---|
602 | } |
---|
603 | |
---|
604 | // Drop all redundant rules |
---|
605 | newmask.clean(); |
---|
606 | |
---|
607 | // Return the new mask |
---|
608 | return newmask; |
---|
609 | } |
---|
610 | |
---|
611 | /** |
---|
612 | @brief Intersects two masks (only classes that are included in both masks will be included in the resulting mask too). |
---|
613 | @param other The mask to intersect with |
---|
614 | @return The intersection |
---|
615 | */ |
---|
616 | ClassTreeMask ClassTreeMask::operator*(const ClassTreeMask& other) const |
---|
617 | { |
---|
618 | // Create a new mask |
---|
619 | ClassTreeMask newmask; |
---|
620 | |
---|
621 | // Add all nodes from the first mask, calculate the rule with the and-operation |
---|
622 | for (ClassTreeMaskIterator it = this->root_; it; ++it) |
---|
623 | { |
---|
624 | const Identifier* subclass = it->getClass(); |
---|
625 | newmask.add(subclass, this->isIncluded(subclass) && other.isIncluded(subclass), false, false); |
---|
626 | } |
---|
627 | |
---|
628 | // Add all nodes from the second mask, calculate the rule with the and-operation |
---|
629 | for (ClassTreeMaskIterator it = other.root_; it; ++it) |
---|
630 | { |
---|
631 | const Identifier* subclass = it->getClass(); |
---|
632 | newmask.add(subclass, this->isIncluded(subclass) && other.isIncluded(subclass), false, false); |
---|
633 | } |
---|
634 | |
---|
635 | // Drop all redundant rules |
---|
636 | newmask.clean(); |
---|
637 | |
---|
638 | // Return the new mask |
---|
639 | return newmask; |
---|
640 | } |
---|
641 | |
---|
642 | /** |
---|
643 | @brief Removes all elements of the second mask from the first mask (all classes that are included in the first mask stay included, except those that are included in the second mask too). |
---|
644 | @param other The mask to subtract. |
---|
645 | @return The difference |
---|
646 | */ |
---|
647 | ClassTreeMask ClassTreeMask::operator-(const ClassTreeMask& other) const |
---|
648 | { |
---|
649 | return ((*this) * (!other)); |
---|
650 | } |
---|
651 | |
---|
652 | /** |
---|
653 | @brief Inverts the mask (all included classes are now excluded and vice versa). |
---|
654 | @return The complement |
---|
655 | */ |
---|
656 | ClassTreeMask ClassTreeMask::operator!() const |
---|
657 | { |
---|
658 | // Create a new mask |
---|
659 | ClassTreeMask newmask; |
---|
660 | |
---|
661 | // Add all nodes from the other mask, inverting the rules |
---|
662 | for (ClassTreeMaskIterator it = this->root_; it; ++it) |
---|
663 | { |
---|
664 | const Identifier* subclass = it->getClass(); |
---|
665 | newmask.add(subclass, !this->isIncluded(subclass), false, false); |
---|
666 | } |
---|
667 | |
---|
668 | // Return the new mask |
---|
669 | return newmask; |
---|
670 | } |
---|
671 | |
---|
672 | /** |
---|
673 | @brief Unites this mask with another mask. |
---|
674 | @param other The other mask |
---|
675 | @return A reference to the mask itself |
---|
676 | */ |
---|
677 | ClassTreeMask& ClassTreeMask::operator+=(const ClassTreeMask& other) |
---|
678 | { |
---|
679 | (*this) = (*this) + other; |
---|
680 | return (*this); |
---|
681 | } |
---|
682 | |
---|
683 | /** |
---|
684 | @brief Intersects this mask with another mask. |
---|
685 | @param other The other mask |
---|
686 | @return A reference to the mask itself |
---|
687 | */ |
---|
688 | ClassTreeMask& ClassTreeMask::operator*=(const ClassTreeMask& other) |
---|
689 | { |
---|
690 | (*this) = (*this) * other; |
---|
691 | return (*this); |
---|
692 | } |
---|
693 | |
---|
694 | /** |
---|
695 | @brief Subtracts another mask from this mask. |
---|
696 | @param other The other mask |
---|
697 | @return A reference to the mask itself |
---|
698 | */ |
---|
699 | ClassTreeMask& ClassTreeMask::operator-=(const ClassTreeMask& other) |
---|
700 | { |
---|
701 | (*this) = (*this) - other; |
---|
702 | return (*this); |
---|
703 | } |
---|
704 | |
---|
705 | /** |
---|
706 | @brief Intersects two masks (only classes that are included in both masks will be included in the resulting mask too). |
---|
707 | @param other The mask to intersect with |
---|
708 | @return The intersection |
---|
709 | */ |
---|
710 | ClassTreeMask ClassTreeMask::operator&(const ClassTreeMask& other) const |
---|
711 | { |
---|
712 | return ((*this) * other); |
---|
713 | } |
---|
714 | |
---|
715 | /** |
---|
716 | @brief Adds two masks in the sense of a union (all classes that are included in at least one of the masks will be included in the resulting mask too). |
---|
717 | @param other The mask to unite with |
---|
718 | @return The union |
---|
719 | */ |
---|
720 | ClassTreeMask ClassTreeMask::operator|(const ClassTreeMask& other) const |
---|
721 | { |
---|
722 | return ((*this) + other); |
---|
723 | } |
---|
724 | |
---|
725 | /** |
---|
726 | @brief Joins two masks in the sense of a xor (exclusivity) operation (all classes that are included in exactly one of the masks, but not in both, will be included in the resulting mask too). |
---|
727 | @param other The mask to join with |
---|
728 | @return The result |
---|
729 | */ |
---|
730 | ClassTreeMask ClassTreeMask::operator^(const ClassTreeMask& other) const |
---|
731 | { |
---|
732 | // Create a new mask |
---|
733 | ClassTreeMask newmask; |
---|
734 | |
---|
735 | // Add all nodes from the first mask, calculate the rule with the xor-operation |
---|
736 | for (ClassTreeMaskIterator it = this->root_; it; ++it) |
---|
737 | { |
---|
738 | const Identifier* subclass = it->getClass(); |
---|
739 | newmask.add(subclass, this->isIncluded(subclass) XOR other.isIncluded(subclass), false, false); |
---|
740 | } |
---|
741 | |
---|
742 | // Add all nodes from the second mask, calculate the rule with the xor-operation |
---|
743 | for (ClassTreeMaskIterator it = other.root_; it; ++it) |
---|
744 | { |
---|
745 | const Identifier* subclass = it->getClass(); |
---|
746 | newmask.add(subclass, this->isIncluded(subclass) XOR other.isIncluded(subclass), false, false); |
---|
747 | } |
---|
748 | |
---|
749 | // Drop all redundant rules |
---|
750 | newmask.clean(); |
---|
751 | |
---|
752 | // Return the new mask |
---|
753 | return newmask; |
---|
754 | } |
---|
755 | |
---|
756 | /** |
---|
757 | @brief Inverts the mask (all included classes are now excluded and vice versa). |
---|
758 | @return The complement |
---|
759 | */ |
---|
760 | ClassTreeMask ClassTreeMask::operator~() const |
---|
761 | { |
---|
762 | return (!(*this)); |
---|
763 | } |
---|
764 | |
---|
765 | /** |
---|
766 | @brief Intersects this mask with another mask (and-operation) |
---|
767 | @param other The other mask |
---|
768 | @return A reference to the mask itself |
---|
769 | */ |
---|
770 | ClassTreeMask& ClassTreeMask::operator&=(const ClassTreeMask& other) |
---|
771 | { |
---|
772 | (*this) = (*this) & other; |
---|
773 | return (*this); |
---|
774 | } |
---|
775 | |
---|
776 | /** |
---|
777 | @brief Unites this mask with another mask (or-operation). |
---|
778 | @param other The other mask |
---|
779 | @return A reference to the mask itself |
---|
780 | */ |
---|
781 | ClassTreeMask& ClassTreeMask::operator|=(const ClassTreeMask& other) |
---|
782 | { |
---|
783 | (*this) = (*this) | other; |
---|
784 | return (*this); |
---|
785 | } |
---|
786 | |
---|
787 | /** |
---|
788 | @brief Joins this mask with another mask with a xor-operation. |
---|
789 | @param other The other mask |
---|
790 | @return A reference to the mask itself |
---|
791 | */ |
---|
792 | ClassTreeMask& ClassTreeMask::operator^=(const ClassTreeMask& other) |
---|
793 | { |
---|
794 | (*this) = (*this) ^ other; |
---|
795 | return (*this); |
---|
796 | } |
---|
797 | |
---|
798 | /** |
---|
799 | @brief Converts the content of a mask into a human readable string and puts it on the ostream. |
---|
800 | @param out The ostream |
---|
801 | @param mask The mask |
---|
802 | @return A reference to the ostream |
---|
803 | */ |
---|
804 | std::ostream& operator<<(std::ostream& out, const ClassTreeMask& mask) |
---|
805 | { |
---|
806 | // Iterate through all rules |
---|
807 | for (ClassTreeMaskIterator it = mask.root_; it; ++it) |
---|
808 | { |
---|
809 | // Calculate the prefix: + means included, - means excluded |
---|
810 | if (it->isIncluded()) |
---|
811 | out << "+"; |
---|
812 | else |
---|
813 | out << "-"; |
---|
814 | |
---|
815 | // Put the name of the corresponding class on the stream |
---|
816 | out << it->getClass()->getName() << " "; |
---|
817 | } |
---|
818 | |
---|
819 | return out; |
---|
820 | } |
---|
821 | } |
---|