src/asset.cc

   1 /*
   2     Copyright (C) 2014-2015 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 /** @file  src/asset.cc
  21  *  @brief Asset class.
  22  */
  23
  24 #include "raw_convert.h"
  25 #include "asset.h"
  26 #include "util.h"
  27 #include "exceptions.h"
  28 #include "dcp_assert.h"
  29 #include "compose.hpp"
  30 #include <libxml++/libxml++.h>
  31 #include <boost/algorithm/string.hpp>
  32
  33 using std::string;
  34 using boost::function;
  35 using boost::optional;
  36 using namespace dcp;
  37
  38 /** Create an Asset with a randomly-generated ID */
  39 Asset::Asset ()
  40 {
  41
  42 }
  43
  44 /** Create an Asset from a given file.
  45  *  @param file File name.
  46  */
  47 Asset::Asset (boost::filesystem::path file)
  48         : _file (file)
  49 {
  50
  51 }
  52
  53 Asset::Asset (string id, boost::filesystem::path file)
  54         : Object (id)
  55         , _file (file)
  56 {
  57
  58 }
  59
  60 void
  61 Asset::write_to_pkl (xmlpp::Node* node, Standard standard) const
  62 {
  63         DCP_ASSERT (!_file.empty ());
  64
  65         xmlpp::Node* asset = node->add_child ("Asset");
  66         asset->add_child("Id")->add_child_text ("urn:uuid:" + _id);
  67         asset->add_child("AnnotationText")->add_child_text (_id);
  68         asset->add_child("Hash")->add_child_text (hash ());
  69         asset->add_child("Size")->add_child_text (raw_convert<string> (boost::filesystem::file_size (_file)));
  70         asset->add_child("Type")->add_child_text (pkl_type (standard));
  71 }
  72
  73 void
  74 Asset::write_to_assetmap (xmlpp::Node* node, boost::filesystem::path root) const
  75 {
  76         DCP_ASSERT (!_file.empty ());
  77
  78         xmlpp::Node* asset = node->add_child ("Asset");
  79         asset->add_child("Id")->add_child_text ("urn:uuid:" + _id);
  80         xmlpp::Node* chunk_list = asset->add_child ("ChunkList");
  81         xmlpp::Node* chunk = chunk_list->add_child ("Chunk");
  82
  83         optional<boost::filesystem::path> path = relative_to_root (
  84                 boost::filesystem::canonical (root),
  85                 boost::filesystem::canonical (_file)
  86                 );
  87
  88         if (!path) {
  89                 throw MiscError (String::compose ("Asset %1 is not within the directory %2", _file, root));
  90         }
  91
  92         /* On Windows path.string() will contain back-slashes; replace these with
  93            forward-slashes.  XXX: perhaps there is a nicer way to do this with boost.
  94         */
  95
  96         string path_string = path.get().string ();
  97         boost::replace_all (path_string, "\\/", "/");
  98
  99         chunk->add_child("Path")->add_child_text (path_string);
 100         chunk->add_child("VolumeIndex")->add_child_text ("1");
 101         chunk->add_child("Offset")->add_child_text ("0");
 102         chunk->add_child("Length")->add_child_text (raw_convert<string> (boost::filesystem::file_size (_file)));
 103 }
 104
 105 string
 106 Asset::hash (function<void (float)> progress) const
 107 {
 108         DCP_ASSERT (!_file.empty ());
 109
 110         if (_hash.empty ()) {
 111                 _hash = make_digest (_file, progress);
 112         }
 113
 114         return _hash;
 115 }
 116
 117 bool
 118 Asset::equals (boost::shared_ptr<const Asset> other, EqualityOptions, NoteHandler note) const
 119 {
 120         if (_hash != other->_hash) {
 121                 note (DCP_ERROR, "Asset: hashes differ");
 122                 return false;
 123         }
 124
 125         return true;
 126 }
 127
 128 /** Set the file that holds this asset on disk.  Calling this function
 129  *  clears this object's store of its hash, so you should call ::hash
 130  *  after this.
 131  *
 132  *  @param file New file's path.
 133  */
 134 void
 135 Asset::set_file (boost::filesystem::path file) const
 136 {
 137         _file = boost::filesystem::absolute (file);
 138         _hash.clear ();
 139 }