Add new method.

2 files changed, 16 insertions, 0 deletions

diff --git a/src/dcp.cc b/src/dcp.cc
index fb0f2625..f99d3433 100644
--- a/src/dcp.cc
+++ b/src/dcp.cc
@@ -65,6 +65,7 @@
 
 using std::string;
 using std::list;
+using std::vector;
 using std::cout;
 using std::make_pair;
 using std::map;
@@ -504,3 +505,16 @@ DCP::assets () const
 
 	return assets;
 }
+
+/** Given a list of files that make up 1 or more DCPs, return the DCP directories */
+vector<boost::filesystem::path>
+DCP::directories_from_files (vector<boost::filesystem::path> files)
+{
+	vector<boost::filesystem::path> d;
+	BOOST_FOREACH (boost::filesystem::path i, files) {
+		if (i.filename() == "ASSETMAP" || i.filename() == "ASSETMAP.xml") {
+			d.push_back (i.parent_path ());
+		}
+	}
+	return d;
+}
diff --git a/src/dcp.h b/src/dcp.h
index e2919a38..3cd2c2cb 100644
--- a/src/dcp.h
+++ b/src/dcp.h
@@ -123,6 +123,8 @@ public:
 		return _standard;
 	}
 
+	static std::vector<boost::filesystem::path> directories_from_files (std::vector<boost::filesystem::path> files);
+
 private:
 
 	/** Write the PKL file.