RawReader.cpp

#include <string>
#include <iostream>
#include <fstream>
#include <vector>
#include <sstream>

#include "LifeVariables.h"
#include "RawReader.h"
#include "Error.h"
#include "osdir.h"

using namespace std;

RawReader::RawReader() {
    this->openFlag = false;
    this->openDirFlag = false;
    this->voxelData = 0;
    this->size = 0;
    fileInfoString = "";
    try {
       this->openPath(DEFAULT_PATH);       
    }
    catch(Error e) {
        cout << e.getErrorMsg();
    }
}

void RawReader::openPath(string pathName) throw(Error){
    oslink::directory dataDir(pathName);
    string tempString;
    if(dataDir != NULL) {
        // clear the existing files if there are some
        if (this->fileList.size()> 0) {
            if(this->openFlag) {
                close();
            }
            //cout << ">> clearing filelist." << endl;
            this->fileList.clear();
        }
        while(dataDir) {
            // read out the next file name
            tempString = dataDir.next();
            //  and add it to the vector iff it is a valid file name
            int pos = string::npos;
            if((pos = tempString.rfind(".dat")) != string::npos) {
                this->fd = new FileDescription();
                this->fd->name = tempString.substr(0, pos);
                this->fd->path = pathName+tempString;
                this->fileList.push_back(fd);
                //cout << ">> adding file: " << tempString << endl;
            }
            tempString = "";
        }
        // somehow it doublecounts the last directory entry -> delete the last one
        if(this->fileList.size() > 1) {
            this->fileList.pop_back();
        }
        else if(this->fileList.size() == 0) {
            this->openFlag = false;
            this->openDirFlag = false;
            throw Error("The directory '"+pathName+"' does not contain any valid files.");
        }
    }
    else {
        this->openFlag = false;
        this->openDirFlag = false;   
        throw Error("The directory '"+pathName+"' does not exist.");       
    }
    this->openFlag = false;
    this->openDirFlag = true;
}


void RawReader::open(int index) throw(Error) {
    if(index > this->fileList.size()) {
        this->openFlag = false;
        throw Error("File is not in the directory.");        
    }
    if(openFlag) {        
        this->close();
    }
        
    unsigned short *cPtr = 0; 
        long ColNum = 0;
    long RowNum = 0;
    
        // get the name of the file from the list
    string fileName = this->fileList.at(index)->path;
    
        // create an input fileStream
    this->fileStream = new ifstream(fileName.c_str(), ios::in|ios::binary);
    if(this->fileStream->is_open()) {
        //cout << ">> reading file: " << fileName << endl;
              
        // get the dimensions of the dataset
        this->fileStream->read((char*)&dimX, 2);
        this->fileStream->read((char*)&dimY, 2);
        this->fileStream->read((char*)&dimZ, 2);
        
        ostringstream infoStream;
        infoStream << dimX << " x " << dimY << " x " << dimZ;
        fileInfoString =  infoStream.str();        

        // read out the densities
        this->size = dimX*dimY*dimZ;
        //cout << ">> file dimensions: " << dimX << " " << dimY << " " << dimZ << " " << size  << endl;
            this->voxelData = new unsigned short[size];
        
        if(this->voxelData!=0) {
            this->fileStream->read((char*)voxelData, sizeof(unsigned short)*size);
            /*
            for(int c = 0; c < size; c++) { 
                this->fileStream->read((char*)&voxelData[c], 2);
            }
            */
            this->fileStream->close();
        }
        else {
            throw Error("File '"+ fileName + "' might be to big for loading."); 
        }
        
        
        this->openFlag = true;
    }
    else {
        this->openFlag = false;
        this->fileInfoString="-";
        throw Error("File '"+ fileName + "' could not be loaded."); 
    }    
}


void RawReader::close() {
    if(openFlag) {
        //cout << ">> closing opened file " << endl;
        //if(this->voxelData != 0) {
        //    delete[] this->voxelData;            
        //}
        this->openFlag = false;
        this->fileInfoString="-";
    }
}


char* RawReader::getFileNameAt(int index) throw(Error){
    if(index > fileList.size()) {
        throw Error("The filename is not in the list");
    }
    else {
        return &((FileDescription*)fileList.at(index))->name[0];
    }
}

---