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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
// $Id: File.h 1897 2008-10-08 23:51:26Z hieuhoang1972 $
/* ---------------------------------------------------------------- */
/* Copyright 2005 (c) by RWTH Aachen - Lehrstuhl fuer Informatik VI */
/* Richard Zens */
/* ---------------------------------------------------------------- */
#ifndef FILE_H_
#define FILE_H_
#include <cstdio>
#include <iostream>
#include <vector>
#include <cassert>
#include "UserMessage.h"
#include "TypeDef.h"
#include "Util.h"
namespace Moses
{
#ifdef WIN32
#define OFF_T __int64
#define FTELLO(file) _ftelli64(file)
#define FSEEKO(file, offset, origin) _fseeki64(file, offset, origin)
#else
#define OFF_T off_t
#define FTELLO(f) ftello(f)
#define FSEEKO(file, offset, origin) fseeko(file, offset, origin)
#endif
static const OFF_T InvalidOffT=-1;
// WARNING:
// these functions work only for bitwise read/write-able types
template<typename T> inline size_t fWrite(FILE* f,const T& t) {
if(fwrite(&t,sizeof(t),1,f)!=1) {TRACE_ERR("ERROR:: fwrite!\n");abort();}
return sizeof(t);
}
template<typename T> inline void fRead(FILE* f,T& t) {
if(fread(&t,sizeof(t),1,f)!=1) {TRACE_ERR("ERROR: fread!\n");abort();}
}
template<typename T> inline size_t fWrite(FILE* f,const T* b,const T* e) {
UINT32 s=std::distance(b,e);size_t rv=fWrite(f,s);
if(fwrite(b,sizeof(T),s,f)!=s) {TRACE_ERR("ERROR: fwrite!\n");abort();}
return rv+sizeof(T)*s;
}
template<typename T> inline size_t fWrite(FILE* f,const T b,const T e) {
UINT32 s=std::distance(b,e);size_t rv=fWrite(f,s);
if(fwrite(&(*b),sizeof(T),s,f)!=s) {TRACE_ERR("ERROR: fwrite!\n");abort();}
return rv+sizeof(T)*s;
}
template<typename C> inline size_t fWriteVector(FILE* f,const C& v) {
UINT32 s=v.size();
size_t rv=fWrite(f,s);
if(fwrite(&v[0],sizeof(typename C::value_type),s,f)!=s) {TRACE_ERR("ERROR: fwrite!\n");abort();}
return rv+sizeof(typename C::value_type)*s;
}
template<typename C> inline void fReadVector(FILE* f, C& v) {
UINT32 s;fRead(f,s);
v.resize(s);
size_t r=fread(&(*v.begin()),sizeof(typename C::value_type),s,f);
if(r!=s) {TRACE_ERR("ERROR: freadVec! "<<r<<" "<<s<<"\n");abort();}
}
inline size_t fWriteString(FILE* f,const char* e, UINT32 s) {
size_t rv=fWrite(f,s);
if(fwrite(e,sizeof(char),s,f)!=s) {TRACE_ERR("ERROR:: fwrite!\n");abort();}
return rv+sizeof(char)*s;
}
inline void fReadString(FILE* f,std::string& e) {
UINT32 s;fRead(f,s);
char* a=new char[s+1];
if(fread(a,sizeof(char),s,f)!=s) {TRACE_ERR("ERROR: fread!\n");abort();}
a[s]='\0';
e.assign(a);
}
inline size_t fWriteStringVector(FILE* f,const std::vector<std::string>& v) {
UINT32 s=v.size();
size_t totrv=fWrite(f,s);
for (size_t i=0;i<s;i++){ totrv+=fWriteString(f,v.at(i).c_str(),v.at(i).size()); }
return totrv;
}
inline void fReadStringVector(FILE* f, std::vector<std::string>& v) {
UINT32 s;fRead(f,s);v.resize(s);
for (size_t i=0;i<s;i++){ fReadString(f,v.at(i)); }
}
inline OFF_T fTell(FILE* f) {return FTELLO(f);}
inline void fSeek(FILE* f,OFF_T o) {
if(FSEEKO(f,o,SEEK_SET)<0) {
TRACE_ERR("ERROR: could not fseeko position "<<o<<"\n");
if(o==InvalidOffT) TRACE_ERR("You tried to seek for 'InvalidOffT'!\n");
abort();
}
}
inline FILE* fOpen(const char* fn,const char* m) {
if(FILE* f=fopen(fn,m))
return f;
else {
assert(false);
UserMessage::Add(std::string("ERROR: could not open file ") + fn + " with mode " + m + "\n");
return NULL;
}
}
inline void fClose(FILE* f) {fclose(f);} // for consistent function names only
}
#endif
|
