src/asset.cc

   1 /*
   2     Copyright (C) 2014 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
  32 using std::string;
  33 using boost::function;
  34 using boost::optional;
  35 using namespace dcp;
  36
  37 /** Create an Asset with a randomly-generated ID */
  38 Asset::Asset ()
  39 {
  40
  41 }
  42
  43 /** Create an Asset from a given file.
  44  *  @param file File name.
  45  */
  46 Asset::Asset (boost::filesystem::path file)
  47         : _file (file)
  48 {
  49
  50 }
  51
  52 void
  53 Asset::write_to_pkl (xmlpp::Node* node, Standard standard) const
  54 {
  55         DCP_ASSERT (!_file.empty ());
  56
  57         xmlpp::Node* asset = node->add_child ("Asset");
  58         asset->add_child("Id")->add_child_text ("urn:uuid:" + _id);
  59         asset->add_child("AnnotationText")->add_child_text (_id);
  60         asset->add_child("Hash")->add_child_text (hash ());
  61         asset->add_child("Size")->add_child_text (raw_convert<string> (boost::filesystem::file_size (_file)));
  62         asset->add_child("Type")->add_child_text (pkl_type (standard));
  63 }
  64
  65 void
  66 Asset::write_to_assetmap (xmlpp::Node* node, boost::filesystem::path root) const
  67 {
  68         DCP_ASSERT (!_file.empty ());
  69
  70         xmlpp::Node* asset = node->add_child ("Asset");
  71         asset->add_child("Id")->add_child_text ("urn:uuid:" + _id);
  72         xmlpp::Node* chunk_list = asset->add_child ("ChunkList");
  73         xmlpp::Node* chunk = chunk_list->add_child ("Chunk");
  74
  75         optional<boost::filesystem::path> path = relative_to_root (
  76                 boost::filesystem::canonical (root),
  77                 boost::filesystem::canonical (_file)
  78                 );
  79
  80         if (!path) {
  81                 throw MiscError (String::compose ("Asset %1 is not within the directory %2", _file, root));
  82         }
  83
  84         chunk->add_child("Path")->add_child_text (path.get().string ());
  85         chunk->add_child("VolumeIndex")->add_child_text ("1");
  86         chunk->add_child("Offset")->add_child_text ("0");
  87         chunk->add_child("Length")->add_child_text (raw_convert<string> (boost::filesystem::file_size (_file)));
  88 }
  89
  90 string
  91 Asset::hash (function<void (float)> progress) const
  92 {
  93         DCP_ASSERT (!_file.empty ());
  94
  95         if (_hash.empty ()) {
  96                 _hash = make_digest (_file, progress);
  97         }
  98
  99         return _hash;
 100 }
 101
 102 bool
 103 Asset::equals (boost::shared_ptr<const Asset> other, EqualityOptions, NoteHandler note) const
 104 {
 105         if (_hash != other->_hash) {
 106                 note (DCP_ERROR, "Asset: hashes differ");
 107                 return false;
 108         }
 109
 110         return true;
 111 }
 112
 113 /** Set the file that holds this asset on disk.  Calling this function
 114  *  clears this object's store of its hash, so you should call ::hash
 115  *  after this.
 116  *
 117  *  @param file New file's path.
 118  */
 119 void
 120 Asset::set_file (boost::filesystem::path file) const
 121 {
 122         _file = boost::filesystem::absolute (file);
 123         _hash.clear ();
 124 }
 125