/*
- Copyright (C) 2013-2014 Carl Hetherington <cth@carlh.net>
+ Copyright (C) 2013-2021 Carl Hetherington <cth@carlh.net>
This file is part of DCP-o-matic.
*/
+
/** @file src/lib/file_group.cc
* @brief FileGroup class.
*/
-#include <cstdio>
-#include <sndfile.h>
-#include "file_group.h"
-#include "exceptions.h"
+
+#include "compose.hpp"
#include "cross.h"
-#include <iostream>
+#include "dcpomatic_assert.h"
+#include "exceptions.h"
+#include "file_group.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)
{
}
+
/** Construct a FileGroup with a single file */
FileGroup::FileGroup (boost::filesystem::path p)
- : _current_path (0)
- , _current_file (0)
{
_paths.push_back (p);
ensure_open_path (0);
seek (0, SEEK_SET);
}
+
/** Construct a FileGroup with multiple files */
FileGroup::FileGroup (vector<boost::filesystem::path> const & p)
: _paths (p)
- , _current_path (0)
- , _current_file (0)
{
ensure_open_path (0);
seek (0, SEEK_SET);
}
-/** Destroy a FileGroup, closing any open file */
-FileGroup::~FileGroup ()
-{
- if (_current_file) {
- fclose (_current_file);
- }
-}
void
FileGroup::set_paths (vector<boost::filesystem::path> const & p)
seek (0, SEEK_SET);
}
+
/** Ensure that the given path index in the content is the _current_file */
void
FileGroup::ensure_open_path (size_t p) const
}
if (_current_file) {
- fclose (_current_file);
+ _current_file->close();
}
_current_path = p;
- _current_file = fopen_boost (_paths[_current_path], "rb");
- if (_current_file == 0) {
- throw OpenFileError (_paths[_current_path], errno, true);
+ _current_file = dcp::File(_paths[_current_path], "rb");
+ if (!_current_file) {
+ 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);
+ _current_file->seek(sub_pos, SEEK_SET);
+ } else {
+ ensure_open_path (_paths.size() - 1);
+ _current_file->seek(_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
{
+ DCPOMATIC_ASSERT (_current_file);
+
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);
+
+ auto const current_position = _current_file->tell();
+ 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;
+ }
+
+ int const this_time = _current_file->read(buffer + read, 1, to_read);
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 (_current_file->error()) {
+ 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;
}
+
/** @return Combined length of all the files */
int64_t
FileGroup::length () const