src/lib/copy_from_dvd_job.cc

   1 /*
   2     Copyright (C) 2012 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/copy_from_dvd_job.cc
  21  *  @brief A job to copy a film from a DVD.
  22  */
  23
  24 #include <stdio.h>
  25 #include <boost/algorithm/string.hpp>
  26 #include <boost/filesystem.hpp>
  27 #include "copy_from_dvd_job.h"
  28 #include "film_state.h"
  29 #include "dvd.h"
  30
  31 using namespace std;
  32 using namespace boost;
  33
  34 /** @param fs FilmState for the film to write DVD data into.
  35  *  @param l Log that we can write to.
  36  */
  37 CopyFromDVDJob::CopyFromDVDJob (shared_ptr<const FilmState> fs, Log* l)
  38         : Job (fs, shared_ptr<Options> (), l)
  39 {
  40
  41 }
  42
  43 string
  44 CopyFromDVDJob::name () const
  45 {
  46         return "Copy film from DVD";
  47 }
  48
  49 void
  50 CopyFromDVDJob::run ()
  51 {
  52         /* Remove any old DVD rips */
  53         filesystem::remove_all (_fs->dir ("dvd"));
  54
  55         string const dvd = find_dvd ();
  56         if (dvd.empty ()) {
  57                 set_error ("could not find DVD");
  58                 set_state (FINISHED_ERROR);
  59         }
  60
  61         vector<uint64_t> const t = dvd_titles (dvd);
  62         if (t.empty ()) {
  63                 set_error ("no titles found on DVD");
  64                 set_state (FINISHED_ERROR);
  65         }
  66
  67         int longest_title = 0;
  68         uint64_t longest_size = 0;
  69         for (vector<int>::size_type i = 0; i < t.size(); ++i) {
  70                 if (longest_size < t[i]) {
  71                         longest_size = t[i];
  72                         longest_title = i;
  73                 }
  74         }
  75
  76         stringstream c;
  77         c << "vobcopy -n " << longest_title << " -l -o \"" << _fs->dir ("dvd") << "\" 2>&1";
  78
  79         FILE* f = popen (c.str().c_str(), "r");
  80         if (f == 0) {
  81                 set_error ("could not run vobcopy command");
  82                 set_state (FINISHED_ERROR);
  83                 return;
  84         }
  85
  86         while (!feof (f)) {
  87                 char buf[256];
  88                 if (fscanf (f, "%s", buf)) {
  89                         string s (buf);
  90                         if (!s.empty () && s[s.length() - 1] == '%') {
  91                                 set_progress (atof (s.substr(0, s.length() - 1).c_str()) / 100.0);
  92                         }
  93                 }
  94         }
  95
  96         int const r = pclose (f);
  97         if (WEXITSTATUS (r) != 0) {
  98                 set_error ("call to vobcopy failed");
  99                 set_state (FINISHED_ERROR);
 100         } else {
 101                 set_state (FINISHED_OK);
 102         }
 103 }