src/gui/itemviews/qbsptree.cpp
changeset 0 1918ee327afb
child 4 3b1da2848fc7
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/gui/itemviews/qbsptree.cpp	Mon Jan 11 14:00:40 2010 +0000
@@ -0,0 +1,145 @@
+/****************************************************************************
+**
+** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
+** All rights reserved.
+** Contact: Nokia Corporation (qt-info@nokia.com)
+**
+** This file is part of the QtGui module of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:LGPL$
+** No Commercial Usage
+** This file contains pre-release code and may not be distributed.
+** You may use this file in accordance with the terms and conditions
+** contained in the Technology Preview License Agreement accompanying
+** this package.
+**
+** GNU Lesser General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU Lesser
+** General Public License version 2.1 as published by the Free Software
+** Foundation and appearing in the file LICENSE.LGPL included in the
+** packaging of this file.  Please review the following information to
+** ensure the GNU Lesser General Public License version 2.1 requirements
+** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
+**
+** In addition, as a special exception, Nokia gives you certain additional
+** rights.  These rights are described in the Nokia Qt LGPL Exception
+** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
+**
+** If you have questions regarding the use of this file, please contact
+** Nokia at qt-info@nokia.com.
+**
+**
+**
+**
+**
+**
+**
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#include "qbsptree_p.h"
+
+QT_BEGIN_NAMESPACE
+
+QBspTree::QBspTree() : depth(6), visited(0) {}
+
+void QBspTree::create(int n, int d)
+{
+    // simple heuristics to find the best tree depth
+    if (d == -1) {
+        int c;
+        for (c = 0; n; ++c)
+            n = n / 10;
+        depth = c << 1;
+    } else {
+        depth = d;
+    }
+    depth = qMax(depth, uint(1));
+
+    nodes.resize((1 << depth) - 1); // resize to number of nodes
+    leaves.resize(1 << depth); // resize to number of leaves
+}
+
+void QBspTree::destroy()
+{
+    leaves.clear();
+    nodes.clear();
+}
+
+void QBspTree::climbTree(const QRect &rect, callback *function, QBspTreeData data)
+{
+    if (nodes.isEmpty())
+        return;
+    ++visited;
+    climbTree(rect, function, data, 0);
+}
+
+void QBspTree::climbTree(const QRect &area, callback *function, QBspTreeData data, int index)
+{
+    if (index >= nodes.count()) { // the index points to a leaf
+        Q_ASSERT(!nodes.isEmpty());
+        function(leaf(index - nodes.count()), area, visited, data);
+        return;
+    }
+
+    Node::Type t = (Node::Type) nodes.at(index).type;
+
+    int pos = nodes.at(index).pos;
+    int idx = firstChildIndex(index);
+    if (t == Node::VerticalPlane) {
+        if (area.left() < pos)
+            climbTree(area, function, data, idx); // back
+        if (area.right() >= pos)
+            climbTree(area, function, data, idx + 1); // front
+    } else {
+        if (area.top() < pos)
+            climbTree(area, function, data, idx); // back
+        if (area.bottom() >= pos)
+            climbTree(area, function, data, idx + 1); // front
+    }
+}
+
+void QBspTree::init(const QRect &area, int depth, NodeType type, int index)
+{
+    Node::Type t = Node::None; // t should never have this value
+    if (type == Node::Both) // if both planes are specified, use 2d bsp
+        t = (depth & 1) ? Node::HorizontalPlane : Node::VerticalPlane;
+    else
+        t = type;
+    QPoint center = area.center();
+    nodes[index].pos = (t == Node::VerticalPlane ? center.x() : center.y());
+    nodes[index].type = t;
+
+    QRect front = area;
+    QRect back = area;
+
+    if (t == Node::VerticalPlane) {
+        front.setLeft(center.x());
+        back.setRight(center.x() - 1); // front includes the center
+    } else { // t == Node::HorizontalPlane
+        front.setTop(center.y());
+        back.setBottom(center.y() - 1);
+    }
+
+    int idx = firstChildIndex(index);
+    if (--depth) {
+        init(back, depth, type, idx);
+        init(front, depth, type, idx + 1);
+    }
+}
+
+void QBspTree::insert(QVector<int> &leaf, const QRect &, uint, QBspTreeData data)
+{
+    leaf.append(data.i);
+}
+
+void QBspTree::remove(QVector<int> &leaf, const QRect &, uint, QBspTreeData data)
+{
+    int i = leaf.indexOf(data.i);
+    if (i != -1)
+        leaf.remove(i);
+}
+
+QT_END_NAMESPACE