From 12e20e180062dbc338e775bd1f5ec7a2af91df7f Mon Sep 17 00:00:00 2001 From: Carl Hetherington Date: Fri, 17 Jul 2015 00:41:04 +0100 Subject: Improve image filename sorter. --- src/lib/image_filename_sorter.cc | 40 +++++++++++++++++++++++++++------------- 1 file changed, 27 insertions(+), 13 deletions(-) (limited to 'src') diff --git a/src/lib/image_filename_sorter.cc b/src/lib/image_filename_sorter.cc index 5a3e1dcec..037446398 100644 --- a/src/lib/image_filename_sorter.cc +++ b/src/lib/image_filename_sorter.cc @@ -27,35 +27,49 @@ class ImageFilenameSorter public: bool operator() (boost::filesystem::path a, boost::filesystem::path b) { - boost::optional na = extract_number (a); - boost::optional nb = extract_number (b); - if (!na || !nb) { - return a.string() < b.string(); + std::vector na = extract_numbers (a); + std::vector nb = extract_numbers (b); + + std::vector::const_iterator i = na.begin (); + std::vector::const_iterator j = nb.begin (); + + while (true) { + if (i == na.end () || j == nb.end ()) { + return false; + } + + if (*i != *j) { + return *i < *j; + } + + ++i; + ++j; } - return na.get() < nb.get(); + /* NOT REACHED */ + return false; } private: - boost::optional extract_number (boost::filesystem::path p) + std::vector extract_numbers (boost::filesystem::path p) { p = p.leaf (); + std::vector numbers; std::string number; for (size_t i = 0; i < p.string().size(); ++i) { if (isdigit (p.string()[i])) { number += p.string()[i]; - } else { - if (!number.empty ()) { - break; - } + } else if (!number.empty ()) { + numbers.push_back (raw_convert (number)); + number.clear (); } } - if (number.empty ()) { - return boost::optional (); + if (!number.empty ()) { + numbers.push_back (raw_convert (number)); } - return raw_convert (number); + return numbers; } }; -- cgit v1.2.3