Add missing spell_picker files

1 files changed, 23 insertions, 0 deletions

diff --git a/crawl-ref/source/spl-pick.cc b/crawl-ref/source/spl-pick.cc
new file mode 100644
index 0000000000..698018ebd3
--- /dev/null
+++ b/crawl-ref/source/spl-pick.cc
@@ -0,0 +1,23 @@
+/**
+ * @file
+ * @brief Functions for picking from lists of spells.
+**/
+
+#include "AppHdr.h"
+
+#include "spl-pick.h"
+
+spell_type spell_picker::pick_with_veto(const spell_entry *weights,
+                                        int level, spell_type none,
+                                        spell_pick_vetoer vetoer)
+{
+    veto_func = vetoer;
+    return pick(weights, level, none);
+}
+
+// Veto specialisation for the spell_picker class; this simply calls the
+// stored veto function. Can subclass further for more complex veto behaviour.
+bool spell_picker::veto(spell_type mon)
+{
+    return veto_func && veto_func(mon);
+}