Pick up effect and effect color.

[libdcp.git] / src / util.cc

diff --git a/src/util.cc b/src/util.cc
index e805f3eb876010c61e6c4b57678b2c87f714f7a6..7138a075db5e29ab401a16fba7e631cc2ce0821f 100644 (file)
--- a/src/util.cc
+++ b/src/util.cc
@@ -17,20 +17,26 @@
 
 */
 
+/** @file  src/util.cc
+ *  @brief Utility methods.
+ */
+
 #include <stdexcept>
 #include <sstream>
+#include <iostream>
 #include <iomanip>
+#include <boost/filesystem.hpp>
 #include <openssl/sha.h>
 #include "KM_util.h"
 #include "KM_fileio.h"
 #include "AS_DCP.h"
 #include "util.h"
+#include "exceptions.h"
+#include "types.h"
 
 using namespace std;
+using namespace boost;
 
-/** Create a UUID.
- *  @return UUID.
- */
 string
 libdcp::make_uuid ()
 {
@@ -41,22 +47,21 @@ libdcp::make_uuid ()
        return string (buffer);
 }
 
-/** Create a digest for a file.
- *  @param filename File name.
- *  @return Digest.
- */
 string
-libdcp::make_digest (string filename)
+libdcp::make_digest (string filename, sigc::signal1<void, float>* progress)
 {
+       int const file_size = filesystem::file_size (filename);
+       
        Kumu::FileReader reader;
        if (ASDCP_FAILURE (reader.OpenRead (filename.c_str ()))) {
-               throw runtime_error ("could not open file to compute digest");
+               throw FileError ("could not open file to compute digest", filename);
        }
        
        SHA_CTX sha;
        SHA1_Init (&sha);
        
        Kumu::ByteString read_buffer (65536);
+       int done = 0;
        while (1) {
                ui32_t read = 0;
                Kumu::Result_t r = reader.Read (read_buffer.Data(), read_buffer.Capacity(), &read);
@@ -64,10 +69,15 @@ libdcp::make_digest (string filename)
                if (r == Kumu::RESULT_ENDOFFILE) {
                        break;
                } else if (ASDCP_FAILURE (r)) {
-                       throw runtime_error ("could not read file to compute digest");
+                       throw FileError ("could not read file to compute digest", filename);
                }
                
                SHA1_Update (&sha, read_buffer.Data(), read);
+               done += read;
+
+               if (progress) {
+                       (*progress) (0.5 + (0.5 * done / file_size));
+               }
        }
 
        byte_t byte_buffer[20];
@@ -75,6 +85,72 @@ libdcp::make_digest (string filename)
 
        stringstream s;
        char digest[64];
-       s << setfill('0') << setw(36) << Kumu::base64encode (byte_buffer, 20, digest, 64);
-       return s.str ();
+       return Kumu::base64encode (byte_buffer, 20, digest, 64);
+}
+
+string
+libdcp::content_kind_to_string (ContentKind kind)
+{
+       switch (kind) {
+       case FEATURE:
+               return "feature";
+       case SHORT:
+               return "short";
+       case TRAILER:
+               return "trailer";
+       case TEST:
+               return "test";
+       case TRANSITIONAL:
+               return "transitional";
+       case RATING:
+               return "rating";
+       case TEASER:
+               return "teaser";
+       case POLICY:
+               return "policy";
+       case PUBLIC_SERVICE_ANNOUNCEMENT:
+               return "psa";
+       case ADVERTISEMENT:
+               return "advertisement";
+       }
+
+       assert (false);
+}
+
+libdcp::ContentKind
+libdcp::content_kind_from_string (string type)
+{
+       if (type == "feature") {
+               return FEATURE;
+       } else if (type == "short") {
+               return SHORT;
+       } else if (type == "trailer") {
+               return TRAILER;
+       } else if (type == "test") {
+               return TEST;
+       } else if (type == "transitional") {
+               return TRANSITIONAL;
+       } else if (type == "rating") {
+               return RATING;
+       } else if (type == "teaser") {
+               return TEASER;
+       } else if (type == "policy") {
+               return POLICY;
+       } else if (type == "psa") {
+               return PUBLIC_SERVICE_ANNOUNCEMENT;
+       } else if (type == "advertisement") {
+               return ADVERTISEMENT;
+       }
+
+       assert (false);
+}
+               
+bool
+libdcp::ends_with (string big, string little)
+{
+       if (little.size() > big.size()) {
+               return false;
+       }
+
+       return big.compare (big.length() - little.length(), little.length(), little) == 0;
 }