Add missing spell_picker files

1 files changed, 31 insertions, 0 deletions

diff --git a/crawl-ref/source/spl-pick.h b/crawl-ref/source/spl-pick.h
new file mode 100644
index 0000000000..c9dbc03fdb
--- /dev/null
+++ b/crawl-ref/source/spl-pick.h
@@ -0,0 +1,31 @@
+/**
+ * @file
+ * @brief Functions for picking from lists of spells.
+**/
+
+#ifndef SPL_PICK_H
+#define SPL_PICK_H
+
+#include "externs.h"
+#include "random-pick.h"
+
+#define spell_entry random_pick_entry<spell_type>
+typedef bool (*spell_pick_vetoer)(spell_type);
+
+// Subclass a spell picker using the random_picker template
+class spell_picker : public random_picker<spell_type, NUM_SPELLS>
+{
+public:
+    spell_picker(spell_pick_vetoer _veto = nullptr) : veto_func(_veto) { };
+
+    spell_type pick_with_veto(const spell_entry *weights, int level,
+                              spell_type none,
+                              spell_pick_vetoer veto_func = nullptr);
+
+    virtual bool veto(spell_type spell);
+
+protected:
+    spell_pick_vetoer veto_func;
+};
+
+#endif