155 lines
4.8 KiB
C++
155 lines
4.8 KiB
C++
/*
|
|
==============================================================================
|
|
|
|
This file is part of the JUCE library.
|
|
Copyright (c) 2017 - ROLI Ltd.
|
|
|
|
JUCE is an open source library subject to commercial or open-source
|
|
licensing.
|
|
|
|
The code included in this file is provided under the terms of the ISC license
|
|
http://www.isc.org/downloads/software-support-policy/isc-license. Permission
|
|
To use, copy, modify, and/or distribute this software for any purpose with or
|
|
without fee is hereby granted provided that the above copyright notice and
|
|
this permission notice appear in all copies.
|
|
|
|
JUCE IS PROVIDED "AS IS" WITHOUT ANY WARRANTY, AND ALL WARRANTIES, WHETHER
|
|
EXPRESSED OR IMPLIED, INCLUDING MERCHANTABILITY AND FITNESS FOR PURPOSE, ARE
|
|
DISCLAIMED.
|
|
|
|
==============================================================================
|
|
*/
|
|
|
|
namespace juce
|
|
{
|
|
|
|
int64 juce_fileSetPosition (void* handle, int64 pos);
|
|
|
|
|
|
//==============================================================================
|
|
FileInputStream::FileInputStream (const File& f) : file (f)
|
|
{
|
|
openHandle();
|
|
}
|
|
|
|
int64 FileInputStream::getTotalLength()
|
|
{
|
|
// You should always check that a stream opened successfully before using it!
|
|
jassert (openedOk());
|
|
|
|
return file.getSize();
|
|
}
|
|
|
|
int FileInputStream::read (void* buffer, int bytesToRead)
|
|
{
|
|
// You should always check that a stream opened successfully before using it!
|
|
jassert (openedOk());
|
|
|
|
// The buffer should never be null, and a negative size is probably a
|
|
// sign that something is broken!
|
|
jassert (buffer != nullptr && bytesToRead >= 0);
|
|
|
|
auto num = readInternal (buffer, (size_t) bytesToRead);
|
|
currentPosition += (int64) num;
|
|
|
|
return (int) num;
|
|
}
|
|
|
|
bool FileInputStream::isExhausted()
|
|
{
|
|
return currentPosition >= getTotalLength();
|
|
}
|
|
|
|
int64 FileInputStream::getPosition()
|
|
{
|
|
return currentPosition;
|
|
}
|
|
|
|
bool FileInputStream::setPosition (int64 pos)
|
|
{
|
|
// You should always check that a stream opened successfully before using it!
|
|
jassert (openedOk());
|
|
|
|
if (pos != currentPosition)
|
|
currentPosition = juce_fileSetPosition (fileHandle, pos);
|
|
|
|
return currentPosition == pos;
|
|
}
|
|
|
|
//==============================================================================
|
|
#if JUCE_UNIT_TESTS
|
|
|
|
struct FileInputStreamTests : public UnitTest
|
|
{
|
|
FileInputStreamTests()
|
|
: UnitTest ("FileInputStream", "Streams")
|
|
{}
|
|
|
|
void runTest() override
|
|
{
|
|
const MemoryBlock data ("abcdefghijklmnopqrstuvwxyz", 26);
|
|
File f (File::createTempFile (".txt"));
|
|
f.appendData (data.getData(), data.getSize());
|
|
FileInputStream stream (f);
|
|
|
|
beginTest ("Read");
|
|
|
|
expectEquals (stream.getPosition(), (int64) 0);
|
|
expectEquals (stream.getTotalLength(), (int64) data.getSize());
|
|
expectEquals (stream.getNumBytesRemaining(), stream.getTotalLength());
|
|
expect (! stream.isExhausted());
|
|
|
|
size_t numBytesRead = 0;
|
|
MemoryBlock readBuffer (data.getSize());
|
|
|
|
while (numBytesRead < data.getSize())
|
|
{
|
|
numBytesRead += (size_t) stream.read (&readBuffer[numBytesRead], 3);
|
|
|
|
expectEquals (stream.getPosition(), (int64) numBytesRead);
|
|
expectEquals (stream.getNumBytesRemaining(), (int64) (data.getSize() - numBytesRead));
|
|
expect (stream.isExhausted() == (numBytesRead == data.getSize()));
|
|
}
|
|
|
|
expectEquals (stream.getPosition(), (int64) data.getSize());
|
|
expectEquals (stream.getNumBytesRemaining(), (int64) 0);
|
|
expect (stream.isExhausted());
|
|
|
|
expect (readBuffer == data);
|
|
|
|
beginTest ("Skip");
|
|
|
|
stream.setPosition (0);
|
|
expectEquals (stream.getPosition(), (int64) 0);
|
|
expectEquals (stream.getTotalLength(), (int64) data.getSize());
|
|
expectEquals (stream.getNumBytesRemaining(), stream.getTotalLength());
|
|
expect (! stream.isExhausted());
|
|
|
|
numBytesRead = 0;
|
|
const int numBytesToSkip = 5;
|
|
|
|
while (numBytesRead < data.getSize())
|
|
{
|
|
stream.skipNextBytes (numBytesToSkip);
|
|
numBytesRead += numBytesToSkip;
|
|
numBytesRead = std::min (numBytesRead, data.getSize());
|
|
|
|
expectEquals (stream.getPosition(), (int64) numBytesRead);
|
|
expectEquals (stream.getNumBytesRemaining(), (int64) (data.getSize() - numBytesRead));
|
|
expect (stream.isExhausted() == (numBytesRead == data.getSize()));
|
|
}
|
|
|
|
expectEquals (stream.getPosition(), (int64) data.getSize());
|
|
expectEquals (stream.getNumBytesRemaining(), (int64) 0);
|
|
expect (stream.isExhausted());
|
|
|
|
f.deleteFile();
|
|
}
|
|
};
|
|
|
|
static FileInputStreamTests fileInputStreamTests;
|
|
|
|
#endif
|
|
|
|
} // namespace juce
|