X-Git-Url: https://main.carlh.net/gitweb/?a=blobdiff_plain;f=gtk2_ardour%2Ftrack_selection.h;h=d94ec3c8c1773bf39aa1edf6c09014a0704b35a9;hb=6ee23029a338951705c589be6c61ab52099758b6;hp=0c4f364c4ecf59c34663b91ac7161a234edcf068;hpb=ed93d47e7cbf467161d89c84297e19c26039a6ff;p=ardour.git

diff --git a/gtk2_ardour/track_selection.h b/gtk2_ardour/track_selection.h
index 0c4f364c4e..d94ec3c8c1 100644
--- a/gtk2_ardour/track_selection.h
+++ b/gtk2_ardour/track_selection.h
@@ -23,6 +23,7 @@
 #include "track_view_list.h"
 #include "route_ui.h"
 #include "audio_time_axis.h"
+#include "midi_time_axis.h"
 
 class PublicEditor;
 
@@ -33,7 +34,7 @@ public:
 	TrackSelection (PublicEditor const *, TrackViewList const &);
 
 	virtual ~TrackSelection ();
-	
+
 	TrackViewList add (TrackViewList const &);
 
 	template <typename Function>
@@ -47,7 +48,19 @@ public:
 	void foreach_route_ui (Function f) {
 		for (iterator i = begin(); i != end(); ++i) {
 			RouteUI* t = dynamic_cast<RouteUI*> (*i);
-			f (t);
+			if (t) {
+				f (t);
+			}
+		}
+	}
+
+	template <typename Function>
+	void foreach_route_time_axis (Function f) {
+		for (iterator i = begin(); i != end(); ++i) {
+			RouteTimeAxisView* t = dynamic_cast<RouteTimeAxisView*> (*i);
+			if (t) {
+				f (t);
+			}
 		}
 	}
 
@@ -55,10 +68,22 @@ public:
 	void foreach_audio_time_axis (Function f) {
 		for (iterator i = begin(); i != end(); ++i) {
 			AudioTimeAxisView* t = dynamic_cast<AudioTimeAxisView*> (*i);
-			f (t);
+			if (t) {
+				f (t);
+			}
 		}
 	}
-	
+
+	template <typename Function>
+	void foreach_midi_time_axis (Function f) {
+		for (iterator i = begin(); i != end(); ++i) {
+			MidiTimeAxisView* t = dynamic_cast<MidiTimeAxisView*> (*i);
+			if (t) {
+				f (t);
+			}
+		}
+	}
+
 private:
 	PublicEditor const * _editor;
 };