Add method to search for potential DCPs.v1.10.13

2 files changed, 44 insertions, 0 deletions

diff --git a/src/search.cc b/src/search.cc
index e7871dc6..09e4d3ce 100644
--- a/src/search.cc
+++ b/src/search.cc
@@ -93,3 +93,37 @@ dcp::find_and_resolve_cpls (vector<boost::filesystem::path> const& directories,
 	return cpls;
 }
 
+
+
+static
+void
+find_potential_dcps(boost::filesystem::path path, vector<boost::filesystem::path>& dcp)
+{
+	bool found = false;
+	try {
+		found = dcp::filesystem::exists(path / "ASSETMAP") || dcp::filesystem::exists(path / "ASSETMAP.xml");
+	} catch (...) {}
+
+	if (found) {
+		dcp.push_back(path);
+	} else if (boost::filesystem::is_directory(path)) {
+		for (auto i: boost::filesystem::directory_iterator(path)) {
+			try {
+				find_potential_dcps(i.path(), dcp);
+			} catch (...) {}
+		}
+	}
+}
+
+
+/** Search a directory recursively and return directories within that could be DCPs (because they
+ *  contain ASSETMAP or ASSETMAP.xml)
+ */
+vector<boost::filesystem::path>
+dcp::find_potential_dcps(boost::filesystem::path directory)
+{
+	vector<boost::filesystem::path> dcp;
+	::find_potential_dcps(directory, dcp);
+	return dcp;
+}
+
diff --git a/src/search.h b/src/search.h
index c7199d44..0d8a7405 100644
--- a/src/search.h
+++ b/src/search.h
@@ -32,9 +32,19 @@
 */
 
 
+#include <boost/filesystem.hpp>
+#include <memory>
+#include <vector>
+
+
 namespace dcp {
 
+
+class CPL;
+
+
 extern std::vector<std::shared_ptr<dcp::CPL>> find_and_resolve_cpls (std::vector<boost::filesystem::path> const& directories, bool tolerant);
+extern std::vector<boost::filesystem::path> find_potential_dcps(boost::filesystem::path directory);
 
 }