X-Git-Url: http://git.code-monkey.de/?p=pulseview.git;a=blobdiff_plain;f=pv%2Fview%2Ftracetreeitemowner.cpp;h=c87619524574e02f7039ae60b4d8434a592d74f6;hp=af11ad2688b8d6f61c670d5398fd60ed470a4cf0;hb=6f925ba9d6faf1077b73c5a5808259576081716a;hpb=c373f82810ad9c5376a7370118de9dd587ee0e43

diff --git a/pv/view/tracetreeitemowner.cpp b/pv/view/tracetreeitemowner.cpp
index af11ad2..c876195 100644
--- a/pv/view/tracetreeitemowner.cpp
+++ b/pv/view/tracetreeitemowner.cpp
@@ -14,8 +14,7 @@
  * GNU General Public License for more details.
  *
  * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
+ * along with this program; if not, see <http://www.gnu.org/licenses/>.
  */
 
 #include <cassert>
@@ -25,8 +24,9 @@
 #include "trace.hpp"
 
 using std::dynamic_pointer_cast;
-using std::max;
+using std::find;
 using std::make_pair;
+using std::max;
 using std::min;
 using std::pair;
 using std::set;
@@ -35,19 +35,15 @@ using std::static_pointer_cast;
 using std::vector;
 
 namespace pv {
-namespace view {
+namespace views {
+namespace TraceView {
 
-vector< shared_ptr<ViewItem> >& TraceTreeItemOwner::child_items()
+const ViewItemOwner::item_list& TraceTreeItemOwner::child_items() const
 {
 	return items_;
 }
 
-const vector< shared_ptr<ViewItem> >& TraceTreeItemOwner::child_items() const
-{
-	return items_;
-}
-
-vector< std::shared_ptr<TraceTreeItem> >
+vector< shared_ptr<TraceTreeItem> >
 TraceTreeItemOwner::trace_tree_child_items() const
 {
 	vector< shared_ptr<TraceTreeItem> > items;
@@ -70,7 +66,7 @@ void TraceTreeItemOwner::clear_child_items()
 	items_.clear();
 }
 
-void TraceTreeItemOwner::add_child_item(std::shared_ptr<TraceTreeItem> item)
+void TraceTreeItemOwner::add_child_item(shared_ptr<TraceTreeItem> item)
 {
 	assert(!item->owner());
 	item->set_owner(this);
@@ -79,11 +75,11 @@ void TraceTreeItemOwner::add_child_item(std::shared_ptr<TraceTreeItem> item)
 	extents_changed(true, true);
 }
 
-void TraceTreeItemOwner::remove_child_item(std::shared_ptr<TraceTreeItem> item)
+void TraceTreeItemOwner::remove_child_item(shared_ptr<TraceTreeItem> item)
 {
 	assert(item->owner() == this);
 	item->set_owner(nullptr);
-	auto iter = std::find(items_.begin(), items_.end(), item);
+	auto iter = find(items_.begin(), items_.end(), item);
 	assert(iter != items_.end());
 	items_.erase(iter);
 
@@ -92,13 +88,16 @@ void TraceTreeItemOwner::remove_child_item(std::shared_ptr<TraceTreeItem> item)
 
 pair<int, int> TraceTreeItemOwner::v_extents() const
 {
-	pair<int, int> extents(INT_MAX, INT_MIN);
+	bool has_children = false;
 
+	pair<int, int> extents(INT_MAX, INT_MIN);
 	for (const shared_ptr<TraceTreeItem> t : trace_tree_child_items()) {
 		assert(t);
 		if (!t->enabled())
 			continue;
 
+		has_children = true;
+
 		const int child_offset = t->layout_v_offset();
 		const pair<int, int> child_extents = t->v_extents();
 		extents.first = min(child_extents.first + child_offset,
@@ -107,12 +106,33 @@ pair<int, int> TraceTreeItemOwner::v_extents() const
 			extents.second);
 	}
 
+	if (!has_children)
+		extents = make_pair(0, 0);
+
 	return extents;
 }
 
+bool TraceTreeItemOwner::reassign_bgcolour_states(bool next_bgcolour_state)
+{
+	vector< shared_ptr<TraceTreeItem> > items = trace_tree_child_items();
+
+	// Sort items according to vertical position
+	sort(items.begin(), items.end(),
+		[](const shared_ptr<TraceTreeItem> a, const shared_ptr<TraceTreeItem> b) {
+		return a->layout_v_offset() > b->layout_v_offset(); });
+
+	for (const shared_ptr<TraceTreeItem> item : items) {
+		item->set_bgcolour_state(next_bgcolour_state);
+		next_bgcolour_state = !next_bgcolour_state;
+	}
+
+	return next_bgcolour_state;
+}
+
 void TraceTreeItemOwner::restack_items()
 {
 }
 
-} // view
-} // pv
+} // namespace TraceView
+} // namespace views
+} // namespace pv