(svn r20451) -Codechange: Add helper function Window::GetScrollbar() to get the Scrollbar belonging to a scrollbar widget.

1 files changed, 33 insertions, 0 deletions

diff --git a/src/window.cpp b/src/window.cpp
index 3081bc268..a002de09b 100644
--- a/src/window.cpp
+++ b/src/window.cpp
@@ -131,6 +131,39 @@ void Scrollbar::SetCapacityFromWidget(Window *w, int widget, int padding)
 }
 
 /**
+ * Return the Scrollbar to a widget index.
+ * @param widnum Scrollbar widget index
+ * @return Scrollbar to the widget
+ */
+const Scrollbar *Window::GetScrollbar(uint widnum) const
+{
+	const NWidgetLeaf *wid = this->GetWidget<NWidgetLeaf>(widnum);
+	switch (wid->type) {
+		case WWT_HSCROLLBAR: return &this->hscroll;
+		case WWT_SCROLLBAR:  return &this->vscroll;
+		case WWT_SCROLL2BAR: return &this->vscroll2;
+		default: NOT_REACHED();
+	}
+}
+
+/**
+ * Return the Scrollbar to a widget index.
+ * @param widnum Scrollbar widget index
+ * @return Scrollbar to the widget
+ */
+Scrollbar *Window::GetScrollbar(uint widnum)
+{
+	NWidgetLeaf *wid = this->GetWidget<NWidgetLeaf>(widnum);
+	switch (wid->type) {
+		case WWT_HSCROLLBAR: return &this->hscroll;
+		case WWT_SCROLLBAR:  return &this->vscroll;
+		case WWT_SCROLL2BAR: return &this->vscroll2;
+		default: NOT_REACHED();
+	}
+}
+
+
+/**
  * Set the window that has the focus
  * @param w The window to set the focus on
  */