libStatGen Software
1
|
Class for easily reading/writing files without having to worry about file type (uncompressed, gzip, bgzf) when reading. More...
#include <InputFile.h>
Public Types | |
enum | ifileCompression { DEFAULT, UNCOMPRESSED, GZIP, BGZF } |
Compression to use when writing a file & decompression used when reading a file from stdin. More... | |
Public Member Functions | |
InputFile () | |
Default constructor. | |
~InputFile () | |
Destructor. | |
InputFile (const char *filename, const char *mode, InputFile::ifileCompression compressionMode=InputFile::DEFAULT) | |
Constructor for opening a file. | |
void | bufferReads (unsigned int bufferSize=DEFAULT_BUFFER_SIZE) |
Set the buffer size for reading from files so that bufferSize bytes are read at a time and stored until accessed by another read call. | |
void | disableBuffering () |
Disable read buffering. | |
int | ifclose () |
Close the file. | |
int | ifread (void *buffer, unsigned int size) |
Read size bytes from the file into the buffer. | |
int | readTilChar (const std::string &stopChars, std::string &stringRef) |
Read until the specified characters, returning which character was found causing the stop, -1 returned for EOF, storing the other read characters into the specified string. | |
int | readTilChar (const std::string &stopChars) |
Read until the specified characters, returning which character was found causing the stop, -1 returned for EOF, dropping all read chars. | |
int | discardLine () |
Read until the end of the line, discarding the characters, returning -1 returned for EOF and returning 0 if the end of the line was found. | |
int | readLine (std::string &line) |
Read, appending the characters into the specified string until new line or EOF is found, returning -1 if EOF is found first and 0 if new line is found first. | |
int | readTilTab (std::string &field) |
Read, appending the characters into the specified string until tab, new line, or EOF is found, returning -1 if EOF is found first, 0 if new line is found first, or 1 if a tab is found first. | |
int | ifgetc () |
Get a character from the file. | |
bool | ifgetline (void *voidBuffer, size_t max) |
Get a line from the file. | |
void | ifrewind () |
Reset to the beginning of the file. | |
int | ifeof () const |
Check to see if we have reached the EOF. | |
unsigned int | ifwrite (const void *buffer, unsigned int size) |
Write the specified buffer into the file. | |
bool | isOpen () const |
Returns whether or not the file was successfully opened. | |
int64_t | iftell () |
Get current position in the file. | |
bool | ifseek (int64_t offset, int origin) |
Seek to the specified offset from the origin. | |
const char * | getFileName () const |
Get the filename that is currently opened. | |
void | setAttemptRecovery (bool flag=false) |
Enable (default) or disable recovery. | |
bool | attemptRecoverySync (bool(*checkSignature)(void *data), int length) |
bool | openFile (const char *filename, const char *mode, InputFile::ifileCompression compressionMode) |
Protected Member Functions | |
int | readFromFile (void *buffer, unsigned int size) |
Protected Attributes | |
FileType * | myFileTypePtr |
unsigned int | myAllocatedBufferSize |
char * | myFileBuffer |
int | myBufferIndex |
int | myCurrentBufferSize |
std::string | myFileName |
Static Protected Attributes | |
static const unsigned int | DEFAULT_BUFFER_SIZE = 65536 |
Class for easily reading/writing files without having to worry about file type (uncompressed, gzip, bgzf) when reading.
It hides the low level file operations/structure from the user, allowing them to generically open and operate on a file using the same interface without knowing the file format (standard uncompressed, gzip, or bgzf). For writing, the user must specify the file type. There is a typedef IFILE which is InputFile* and setup to mimic FILE including global methods that take IFILE as a parameter.
Definition at line 36 of file InputFile.h.
Compression to use when writing a file & decompression used when reading a file from stdin.
Any other read checks the file to determine how to uncompress it.
DEFAULT |
Check the extension, if it is ".gz", treat as gzip, otherwise treat it as UNCOMPRESSED. |
UNCOMPRESSED |
uncompressed file. |
GZIP |
gzip file. |
BGZF |
bgzf file. |
Definition at line 44 of file InputFile.h.
{ DEFAULT, ///< Check the extension, if it is ".gz", treat as gzip, otherwise treat it as UNCOMPRESSED. UNCOMPRESSED, ///< uncompressed file. GZIP, ///< gzip file. BGZF ///< bgzf file. };
InputFile::InputFile | ( | const char * | filename, |
const char * | mode, | ||
InputFile::ifileCompression | compressionMode = InputFile::DEFAULT |
||
) |
Constructor for opening a file.
filename | file to open |
mode | same format as fopen: "r" for read & "w" for write. |
compressionMode | set the type of file to open for writing or for reading from stdin (when reading files, the compression type is determined by reading the file). |
Definition at line 28 of file InputFile.cpp.
{ // XXX duplicate code myAttemptRecovery = false; myFileTypePtr = NULL; myBufferIndex = 0; myCurrentBufferSize = 0; myAllocatedBufferSize = DEFAULT_BUFFER_SIZE; myFileBuffer = new char[myAllocatedBufferSize]; myFileName.clear(); openFile(filename, mode, compressionMode); }
void InputFile::bufferReads | ( | unsigned int | bufferSize = DEFAULT_BUFFER_SIZE | ) | [inline] |
Set the buffer size for reading from files so that bufferSize bytes are read at a time and stored until accessed by another read call.
This improves performance over reading the file small bits at a time. Buffering reads disables the tell call for bgzf files. Any previous values in the buffer will be deleted.
bufferSize | number of bytes to read/buffer at a time, turn off read buffering by setting bufferSize = 1; |
Definition at line 83 of file InputFile.h.
Referenced by disableBuffering().
{ // If the buffer size is the same, do nothing. if(bufferSize == myAllocatedBufferSize) { return; } // Delete the previous buffer. if(myFileBuffer != NULL) { delete[] myFileBuffer; } myBufferIndex = 0; myCurrentBufferSize = 0; // The buffer size must be at least 1 so one character can be // read and ifgetc can just assume reading into the buffer. if(bufferSize < 1) { bufferSize = 1; } myFileBuffer = new char[bufferSize]; myAllocatedBufferSize = bufferSize; if(myFileTypePtr != NULL) { if(bufferSize == 1) { myFileTypePtr->setBuffered(false); } else { myFileTypePtr->setBuffered(true); } } }
int InputFile::discardLine | ( | ) |
Read until the end of the line, discarding the characters, returning -1 returned for EOF and returning 0 if the end of the line was found.
Definition at line 95 of file InputFile.cpp.
References ifgetc().
{ int charRead = 0; // Loop until the character was not found in the stop characters. while((charRead != EOF) && (charRead != '\n')) { charRead = ifgetc(); } // First Check for EOF. If EOF is found, just return -1 if(charRead == EOF) { return(-1); } return(0); }
const char* InputFile::getFileName | ( | ) | const [inline] |
Get the filename that is currently opened.
Definition at line 473 of file InputFile.h.
Referenced by SamFile::ReadBamIndex().
{
return(myFileName.c_str());
}
int InputFile::ifclose | ( | ) | [inline] |
Close the file.
Definition at line 133 of file InputFile.h.
Referenced by ifclose().
{ if (myFileTypePtr == NULL) { return EOF; } int result = myFileTypePtr->close(); delete myFileTypePtr; myFileTypePtr = NULL; myFileName.clear(); return result; }
int InputFile::ifeof | ( | ) | const [inline] |
Check to see if we have reached the EOF.
Definition at line 386 of file InputFile.h.
Referenced by ifeof(), readLine(), and readTilTab().
{ // Not EOF if we are not at the end of the buffer. if (myBufferIndex < myCurrentBufferSize) { // There are still available bytes in the buffer, so NOT EOF. return false; } else { if (myFileTypePtr == NULL) { // No myFileTypePtr, so not eof (return 0). return 0; } // exhausted our buffer, so check the file for eof. return myFileTypePtr->eof(); } }
int InputFile::ifgetc | ( | ) | [inline] |
Get a character from the file.
Read a character from the internal buffer, or if the end of the buffer has been reached, read from the file into the buffer and return index 0.
Definition at line 324 of file InputFile.h.
Referenced by discardLine(), ifgetc(), ifgetline(), operator>>(), readLine(), readTilChar(), and readTilTab().
{ if (myBufferIndex >= myCurrentBufferSize) { // at the last index, read a new buffer. myCurrentBufferSize = readFromFile(myFileBuffer, myAllocatedBufferSize); myBufferIndex = 0; // If the buffer index is still greater than or equal to the // myCurrentBufferSize, then we failed to read the file - return EOF. // NB: This only needs to be checked when myCurrentBufferSize // is changed. Simplify check - readFromFile returns zero on EOF if (myCurrentBufferSize == 0) { return(EOF); } } return(myFileBuffer[myBufferIndex++]); }
bool InputFile::ifgetline | ( | void * | voidBuffer, |
size_t | max | ||
) | [inline] |
Get a line from the file.
buffer | the buffer into which data is to be placed |
max | the maximum size of the buffer, in bytes |
Definition at line 347 of file InputFile.h.
References ifgetc().
Referenced by ifgetline().
int InputFile::ifread | ( | void * | buffer, |
unsigned int | size | ||
) | [inline] |
Read size bytes from the file into the buffer.
buffer | pointer to memory at least size bytes big to write the data into. |
size | number of bytes to be read |
Definition at line 153 of file InputFile.h.
Referenced by ifread().
{ // There are 2 cases: // 1) There are already size available bytes in buffer. // 2) There are not size bytes in buffer. // Determine the number of available bytes in the buffer. unsigned int availableBytes = myCurrentBufferSize - myBufferIndex; int returnSize = 0; // Case 1: There are already size available bytes in buffer. if (size <= availableBytes) { // Just copy from the buffer, increment the index and return. memcpy(buffer, myFileBuffer+myBufferIndex, size); // Increment the buffer index. myBufferIndex += size; returnSize = size; } // Case 2: There are not size bytes in buffer. else { // Check to see if there are some bytes in the buffer. if (availableBytes > 0) { // Size > availableBytes > 0 // Copy the available bytes into the buffer. memcpy(buffer, myFileBuffer+myBufferIndex, availableBytes); } // So far availableBytes have been copied into the read buffer. returnSize = availableBytes; // Increment myBufferIndex by what was read. myBufferIndex += availableBytes; unsigned int remainingSize = size - availableBytes; // Check if the remaining size is more or less than the // max buffer size. if(remainingSize < myAllocatedBufferSize) { // the remaining size is not the full buffer, but read // a full buffer worth of data anyway. myCurrentBufferSize = readFromFile(myFileBuffer, myAllocatedBufferSize); // Check for an error. if(myCurrentBufferSize <= 0) { // No more data was successfully read, so check to see // if any data was copied to the return buffer at all. if( returnSize == 0) { // No data has been copied at all into the // return read buffer, so just return the value // returned from readFromFile. returnSize = myCurrentBufferSize; // Otherwise, returnSize is already set to the // available bytes that was already copied (so no // else statement is needed). } // Set myBufferIndex & myCurrentBufferSize to 0. myCurrentBufferSize = 0; myBufferIndex = 0; } else { // Successfully read more data. // Check to see how much was copied. int copySize = remainingSize; if(copySize > myCurrentBufferSize) { // Not the entire requested amount was read // (either from EOF or there was a partial read due to // an error), so set the copySize to what was read. copySize = myCurrentBufferSize; } // Now copy the rest of the bytes into the buffer. memcpy((char*)buffer+availableBytes, myFileBuffer, copySize); // set the buffer index to the location after what we are // returning as read. myBufferIndex = copySize; returnSize += copySize; } } else { // More remaining to be read than the max buffer size, so just // read directly into the output buffer. int readSize = readFromFile((char*)buffer + availableBytes, remainingSize); // Already used the buffer, so "clear" it. myCurrentBufferSize = 0; myBufferIndex = 0; if(readSize <= 0) { // No more data was successfully read, so check to see // if any data was copied to the return buffer at all. if(returnSize == 0) { // No data has been copied at all into the // return read buffer, so just return the value // returned from readFromFile. returnSize = readSize; // Otherwise, returnSize is already set to the // available bytes that was already copied (so no // else statement is needed). } } else { // More data was read, so increment the return count. returnSize += readSize; } } } return(returnSize); }
bool InputFile::ifseek | ( | int64_t | offset, |
int | origin | ||
) | [inline] |
Seek to the specified offset from the origin.
offset | offset into the file to move to (must be from a tell call) |
origin | can be any of the following: Note: not all are valid for all filetypes. SEEK_SET - Beginning of file SEEK_CUR - Current position of the file pointer SEEK_END - End of file |
Definition at line 457 of file InputFile.h.
Referenced by ifseek().
{ if (myFileTypePtr == NULL) { // No myFileTypePtr, so return false - could not seek. return false; } // TODO - may be able to seek within the buffer if applicable. // Reset buffering since a seek is being done. myBufferIndex = 0; myCurrentBufferSize = 0; return myFileTypePtr->seek(offset, origin); }
int64_t InputFile::iftell | ( | ) | [inline] |
Get current position in the file.
Definition at line 436 of file InputFile.h.
Referenced by iftell().
{ if (myFileTypePtr == NULL) { // No myFileTypePtr, so return false - could not seek. return -1; } int64_t pos = myFileTypePtr->tell(); pos -= (myCurrentBufferSize - myBufferIndex); return(pos); }
unsigned int InputFile::ifwrite | ( | const void * | buffer, |
unsigned int | size | ||
) | [inline] |
Write the specified buffer into the file.
buffer | buffer containing size bytes to write to the file. |
size | number of bytes to write |
Definition at line 411 of file InputFile.h.
Referenced by ifwrite(), and operator<<().
{ if (myFileTypePtr == NULL) { // No myFileTypePtr, so return 0 - nothing written. return 0; } return myFileTypePtr->write(buffer, size); }
bool InputFile::isOpen | ( | ) | const [inline] |
Returns whether or not the file was successfully opened.
Definition at line 423 of file InputFile.h.
Referenced by ifopen(), FastQFile::isOpen(), SamFile::IsOpen(), GlfHeader::read(), SamRecord::setBufferFromFile(), GlfHeader::write(), and SamRecord::writeRecordBuffer().
{ // It is open if the myFileTypePtr is set and says it is open. if ((myFileTypePtr != NULL) && myFileTypePtr->isOpen()) { return true; } // File was not successfully opened. return false; }
int InputFile::readLine | ( | std::string & | line | ) |
Read, appending the characters into the specified string until new line or EOF is found, returning -1 if EOF is found first and 0 if new line is found first.
The new line and EOF are not written into the specified string.
line | reference to a string that the read characters should be apppended to (does not include the new line or eof). |
Definition at line 112 of file InputFile.cpp.
int InputFile::readTilChar | ( | const std::string & | stopChars, |
std::string & | stringRef | ||
) |
Read until the specified characters, returning which character was found causing the stop, -1 returned for EOF, storing the other read characters into the specified string.
Note: If stopChars is just '
', readLine is faster and if stopChars is just '
' and '', readTilTab is faster.
stopChars | characters to stop reading when they are hit. |
stringRef | reference to a string that the read characters should be apppended to (does not include the stopchar). |
Definition at line 44 of file InputFile.cpp.
References ifgetc().
{ int charRead = 0; size_t pos = std::string::npos; // Loop until the character was not found in the stop characters. while(pos == std::string::npos) { charRead = ifgetc(); // First Check for EOF. If EOF is found, just return -1 if(charRead == EOF) { return(-1); } // Try to find the character in the stopChars. pos = stopChars.find(charRead); if(pos == std::string::npos) { // Didn't find a stop character and it is not an EOF, // so add it to the string. stringRef += charRead; } } return(pos); }
int InputFile::readTilChar | ( | const std::string & | stopChars | ) |
Read until the specified characters, returning which character was found causing the stop, -1 returned for EOF, dropping all read chars.
Note: If stopChars is just '
', discardLine is faster.
stopChars | characters to stop reading when they are hit. |
Definition at line 73 of file InputFile.cpp.
References ifgetc().
{ int charRead = 0; size_t pos = std::string::npos; // Loop until the character was not found in the stop characters. while(pos == std::string::npos) { charRead = ifgetc(); // First Check for EOF. If EOF is found, just return -1 if(charRead == EOF) { return(-1); } // Try to find the character in the stopChars. pos = stopChars.find(charRead); } return(pos); }
int InputFile::readTilTab | ( | std::string & | field | ) |
Read, appending the characters into the specified string until tab, new line, or EOF is found, returning -1 if EOF is found first, 0 if new line is found first, or 1 if a tab is found first.
The tab, new line, and EOF are not written into the specified string.
field | reference to a string that the read characters should be apppended to (does not include the tab, new line, or eof). |
Definition at line 133 of file InputFile.cpp.
void InputFile::setAttemptRecovery | ( | bool | flag = false | ) | [inline] |
Enable (default) or disable recovery.
When true, we can attach a myFileTypePtr that implements a recovery capable decompressor. This requires that the caller be able to catch the exception XXX "blah blah blah".
Definition at line 485 of file InputFile.h.
Referenced by SamFile::OpenForRead().
{ myAttemptRecovery = flag; }