Add a missing function for the locomotive import UI

[r2c2.git] / source / engineer / newtraindialog.h

diff --git a/source/engineer/newtraindialog.h b/source/engineer/newtraindialog.h
index 9cec84e4bb6b1c259fcfa1b87f6b45eae50a0ff2..f6af7f3b86dde67a9323530fdda997a55a473493 100644 (file)
--- a/source/engineer/newtraindialog.h
+++ b/source/engineer/newtraindialog.h
@@ -6,6 +6,7 @@
 #include <msp/gltk/entry.h>
 #include <msp/gltk/listdata.h>
 #include "libr2c2/catalogue.h"
+#include "libr2c2/driver.h"
 
 class Engineer;
 
@@ -23,6 +24,8 @@ private:
 public:
        NewTrainDialog(Engineer &);
 
+       void prefill(const R2C2::Driver::DetectedLocomotive &);
+
 private:
        virtual void on_response(int);
 };