File: FileUtils.cpp

package info (click to toggle)
pbseqlib 0~20161219-1
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 5,924 kB
  • ctags: 5,123
  • sloc: cpp: 82,727; makefile: 305; python: 239; sh: 8
file content (61 lines) | stat: -rw-r--r-- 1,548 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
#include "FileUtils.hpp"

using namespace std;

bool FileExists(string &fileName) {
    FILE *fp = fopen(fileName.c_str(),"r");
    if( fp ) {
        // exists
        fclose(fp);
        return true;
    } 
    else {
        return false;
    }
}

void CriticalOpenRead(string &fileName, ifstream &file, std::ios::openmode mode) {
    file.open(fileName.c_str(), mode | std::ios::in);
    if (!file.good()) {
        cerr << "Could not open file:"  << fileName << endl;
        exit(1);
    }
}

int OpenRead(string &fileName, ifstream &file, std::ios::openmode mode) {
    file.open(fileName.c_str(), mode | std::ios::in);
    return file.good();
}


void CriticalOpenWrite(string &fileName, ofstream &file, std::ios::openmode mode) {
    file.open(fileName.c_str(), mode | std::ios::out);
    if (!file.good()) { 
        cerr << "Could not open file: " << fileName << endl;
        exit(1);
    }
}

int OpenWrite(string &fileName, ofstream &file, std::ios::openmode mode) {
    file.open(fileName.c_str(), mode | std::ios::out);
    return file.good();
}


int CountLinesInFile(string fileName) {
    char* filePtr;
    long fileSize;
    int fileDes;
    fileDes = open(fileName.c_str(), O_RDONLY);  
    fileSize = lseek(fileDes, 0, SEEK_END);
    lseek(fileDes, 0, SEEK_SET);
    filePtr = (char*) mmap(0, fileSize, PROT_READ, MAP_PRIVATE, fileDes, 0);
    long pos;
    int numLines = 0;
    for (pos = 0; pos < fileSize; pos++, filePtr++) {
        if (*filePtr == '\n') {
            numLines++;
        }
    }
    return numLines;
}