/*
- Copyright (C) 2013-2014 Carl Hetherington <cth@carlh.net>
+ Copyright (C) 2013-2020 Carl Hetherington <cth@carlh.net>
- This program is free software; you can redistribute it and/or modify
+ This file is part of DCP-o-matic.
+
+ DCP-o-matic is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
- This program is distributed in the hope that it will be useful,
+ DCP-o-matic is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
- along with this program; if not, write to the Free Software
- Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ along with DCP-o-matic. If not, see <http://www.gnu.org/licenses/>.
*/
* @brief FileGroup class.
*/
-#include <cstdio>
-#include <sndfile.h>
#include "file_group.h"
#include "exceptions.h"
#include "cross.h"
+#include "compose.hpp"
+#include "dcpomatic_assert.h"
+#include <sndfile.h>
+#include <cstdio>
using std::vector;
-using std::cout;
/** Construct a FileGroup with no files */
FileGroup::FileGroup ()
: _current_path (0)
, _current_file (0)
+ , _current_size (0)
+ , _position (0)
{
}
FileGroup::FileGroup (boost::filesystem::path p)
: _current_path (0)
, _current_file (0)
+ , _current_size (0)
{
_paths.push_back (p);
ensure_open_path (0);
/* Already open */
return;
}
-
+
if (_current_file) {
fclose (_current_file);
}
_current_path = p;
_current_file = fopen_boost (_paths[_current_path], "rb");
if (_current_file == 0) {
- throw OpenFileError (_paths[_current_path]);
+ throw OpenFileError (_paths[_current_path], errno, OpenFileError::READ);
}
+ _current_size = boost::filesystem::file_size (_paths[_current_path]);
}
int64_t
FileGroup::seek (int64_t pos, int whence) const
{
- /* Convert pos to `full_pos', which is an offset from the start
- of all the files.
- */
- int64_t full_pos = 0;
switch (whence) {
case SEEK_SET:
- full_pos = pos;
+ _position = pos;
break;
case SEEK_CUR:
- for (size_t i = 0; i < _current_path; ++i) {
- full_pos += boost::filesystem::file_size (_paths[i]);
- }
-#ifdef DCPOMATIC_WINDOWS
- full_pos += _ftelli64 (_current_file);
-#else
- full_pos += ftell (_current_file);
-#endif
- full_pos += pos;
+ _position += pos;
break;
case SEEK_END:
- full_pos = length() - pos;
+ _position = length() - pos;
break;
}
- /* Seek to full_pos */
+ /* Find an offset within one of the files, if _position is within a file */
size_t i = 0;
- int64_t sub_pos = full_pos;
- while (i < _paths.size ()) {
+ int64_t sub_pos = _position;
+ while (i < _paths.size()) {
boost::uintmax_t len = boost::filesystem::file_size (_paths[i]);
- if (sub_pos < int64_t (len)) {
+ if (sub_pos < int64_t(len)) {
break;
}
- sub_pos -= len;
+ sub_pos -= int64_t(len);
++i;
}
- if (i == _paths.size ()) {
- return -1;
+ if (i < _paths.size()) {
+ ensure_open_path (i);
+ dcpomatic_fseek (_current_file, sub_pos, SEEK_SET);
+ } else {
+ ensure_open_path (_paths.size() - 1);
+ dcpomatic_fseek (_current_file, _current_size, SEEK_SET);
}
- ensure_open_path (i);
- dcpomatic_fseek (_current_file, sub_pos, SEEK_SET);
- return full_pos;
+ return _position;
}
/** Try to read some data from the current position into a buffer.
* @param buffer Buffer to write data into.
* @param amount Number of bytes to read.
- * @return Number of bytes read, or -1 in the case of error.
+ * @return Number of bytes read.
*/
int
FileGroup::read (uint8_t* buffer, int amount) const
{
int read = 0;
while (true) {
- int const this_time = fread (buffer + read, 1, amount - read, _current_file);
+
+ bool eof = false;
+ size_t to_read = amount - read;
+
+ DCPOMATIC_ASSERT (_current_file);
+
+#ifdef DCPOMATIC_WINDOWS
+ int64_t const current_position = _ftelli64 (_current_file);
+ if (current_position == -1) {
+ to_read = 0;
+ eof = true;
+ } else if ((current_position + to_read) > _current_size) {
+ to_read = _current_size - current_position;
+ eof = true;
+ }
+#else
+ long const current_position = ftell(_current_file);
+ if ((current_position + to_read) > _current_size) {
+ to_read = _current_size - current_position;
+ eof = true;
+ }
+#endif
+
+ int const this_time = fread (buffer + read, 1, to_read, _current_file);
read += this_time;
+ _position += this_time;
if (read == amount) {
/* Done */
break;
}
- /* See if there is another file to use */
- if ((_current_path + 1) >= _paths.size()) {
- break;
+ if (ferror(_current_file)) {
+ throw FileError (String::compose("fread error %1", errno), _paths[_current_path]);
+ }
+
+ if (eof) {
+ /* See if there is another file to use */
+ if ((_current_path + 1) >= _paths.size()) {
+ break;
+ }
+ ensure_open_path (_current_path + 1);
}
- ensure_open_path (_current_path + 1);
}
return read;