From 89629a0b30f32fefbb161ce6fc2fc1773c107146 Mon Sep 17 00:00:00 2001
From: Joshua Moerman <lakseru@gmail.com>
Date: Sun, 28 Jun 2015 13:05:23 +0200
Subject: [PATCH] Forgot to add database.hpp (it was ignored)

---
 lib/database.hpp | 51 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 lib/database.hpp

diff --git a/lib/database.hpp b/lib/database.hpp
new file mode 100644
index 0000000..ace3d3b
--- /dev/null
+++ b/lib/database.hpp
@@ -0,0 +1,51 @@
+#pragma once
+
+#include <fingerprints.hpp>
+#include <image_io.hpp>
+
+#include <boost/serialization/access.hpp>
+#include <boost/serialization/vector.hpp>
+
+#include <string>
+#include <vector>
+
+template <typename Fingerprint, typename Traits = fingerprint_traits<Fingerprint>>
+struct image_database {
+	using index = size_t;
+
+	static auto fingerprint_name() { return Fingerprint::name(); }
+	auto filename(index i) const { return filenames[i]; }
+	auto fingerprint(index i) const { return fingerprints[i]; }
+	auto size() const { return fingerprints.size(); }
+
+	void add(std::string const & filename){
+		auto && image = open_image(filename);
+		auto && fingerprint = Traits::calculate(image);
+
+		filenames.push_back(filename);
+		fingerprints.push_back(fingerprint);
+	}
+
+	//! returns a list of distances along with their index
+	auto distances_for_image(av::frame const & image) const {
+		std::vector<std::pair<typename Traits::distance_type, index>> ret;
+		ret.reserve(size());
+
+		auto const pre_fingerprint = Traits::pre_calculate(image);
+		for(auto&& fingerprint : fingerprints){
+			ret.emplace_back(Traits::distance(fingerprint, pre_fingerprint), ret.size());
+		}
+		return ret;
+	}
+
+private:
+	friend class boost::serialization::access;
+	template<class Archive>
+	void serialize(Archive & ar, const unsigned int /*version*/){
+		ar & filenames;
+		ar & fingerprints;
+	}
+
+	std::vector<std::string> filenames;
+	std::vector<Fingerprint> fingerprints;
+};