00001
00002
00003
00004
00005
00006
00007 #ifndef moses_File_h
00008 #define moses_File_h
00009
00010 #include <cstdio>
00011 #include <iostream>
00012 #include <sstream>
00013 #include <vector>
00014 #include "util/exception.hh"
00015 #include "util/string_stream.hh"
00016 #include "TypeDef.h"
00017 #include "Util.h"
00018
00019 namespace Moses
00020 {
00021
00022 #ifdef WIN32
00023 #ifdef __MINGW32__
00024 #define OFF_T __int64
00025 #define FTELLO(f) ftello64(f)
00026 #define FSEEKO(file, offset, origin) fseeko64(file, offset, origin)
00027 #else
00028 #define OFF_T __int64
00029 #define FTELLO(file) _ftelli64(file)
00030 #define FSEEKO(file, offset, origin) _fseeki64(file, offset, origin)
00031 #endif
00032
00033 #else
00034 #define OFF_T off_t
00035 #define FTELLO(f) ftello(f)
00036 #define FSEEKO(file, offset, origin) fseeko(file, offset, origin)
00037 #endif
00038
00039 static const OFF_T InvalidOffT=-1;
00040
00041
00042
00043
00044 template<typename T> inline size_t fWrite(FILE* f,const T& t)
00045 {
00046 if(fwrite(&t,sizeof(t),1,f)!=1) {
00047 UTIL_THROW2("ERROR:: fwrite!");
00048 }
00049 return sizeof(t);
00050 }
00051
00052 template<typename T> inline void fRead(FILE* f,T& t)
00053 {
00054 if(fread(&t,sizeof(t),1,f)!=1) {
00055 UTIL_THROW2("ERROR: fread!");
00056 }
00057 }
00058
00059 template<typename T> inline size_t fWrite(FILE* f,const T* b,const T* e)
00060 {
00061 uint32_t s=std::distance(b,e);
00062 size_t rv=fWrite(f,s);
00063 if(fwrite(b,sizeof(T),s,f)!=s) {
00064 UTIL_THROW2("ERROR: fwrite!");
00065 }
00066 return rv+sizeof(T)*s;
00067 }
00068
00069 template<typename T> inline size_t fWrite(FILE* f,const T b,const T e)
00070 {
00071 uint32_t s=std::distance(b,e);
00072 size_t rv=fWrite(f,s);
00073 if(fwrite(&(*b),sizeof(T),s,f)!=s) {
00074 UTIL_THROW2("ERROR: fwrite!");
00075 }
00076 return rv+sizeof(T)*s;
00077 }
00078
00079 template<typename C> inline size_t fWriteVector(FILE* f,const C& v)
00080 {
00081 uint32_t s=v.size();
00082 size_t rv=fWrite(f,s);
00083 if(fwrite(&v[0],sizeof(typename C::value_type),s,f)!=s) {
00084 UTIL_THROW2("ERROR: fwrite!");
00085 }
00086 return rv+sizeof(typename C::value_type)*s;
00087 }
00088
00089 template<typename C> inline void fReadVector(FILE* f, C& v)
00090 {
00091 uint32_t s;
00092 fRead(f,s);
00093 v.resize(s);
00094 size_t r=fread(&(*v.begin()),sizeof(typename C::value_type),s,f);
00095 if(r!=s) {
00096 UTIL_THROW2("ERROR: freadVec! "<<r<<" "<<s);
00097 }
00098 }
00099
00100 inline size_t fWriteString(FILE* f,const char* e, uint32_t s)
00101 {
00102 size_t rv=fWrite(f,s);
00103 if(fwrite(e,sizeof(char),s,f)!=s) {
00104 UTIL_THROW2("ERROR:: fwrite!");
00105 }
00106 return rv+sizeof(char)*s;
00107 }
00108
00109 inline void fReadString(FILE* f,std::string& e)
00110 {
00111 uint32_t s;
00112 fRead(f,s);
00113 char* a=new char[s+1];
00114 if(fread(a,sizeof(char),s,f)!=s) {
00115 UTIL_THROW2("ERROR: fread!");
00116 }
00117 a[s]='\0';
00118 e.assign(a);
00119 delete[](a);
00120 }
00121
00122 inline size_t fWriteStringVector(FILE* f,const std::vector<std::string>& v)
00123 {
00124 uint32_t s=v.size();
00125 size_t totrv=fWrite(f,s);
00126 for (size_t i=0; i<s; i++) {
00127 totrv+=fWriteString(f,v.at(i).c_str(),v.at(i).size());
00128 }
00129 return totrv;
00130 }
00131
00132 inline void fReadStringVector(FILE* f, std::vector<std::string>& v)
00133 {
00134 uint32_t s;
00135 fRead(f,s);
00136 v.resize(s);
00137
00138 for (size_t i=0; i<s; i++) {
00139 fReadString(f,v.at(i));
00140 }
00141 }
00142
00143 inline OFF_T fTell(FILE* f)
00144 {
00145 return FTELLO(f);
00146 }
00147
00148 inline void fSeek(FILE* f,OFF_T o)
00149 {
00150 if(FSEEKO(f,o,SEEK_SET)<0) {
00151 util::StringStream strme;
00152 strme << "ERROR: could not fseeko position " << o <<"\n";
00153 if(o==InvalidOffT) strme << "You tried to seek for 'InvalidOffT'!\n";
00154 UTIL_THROW2(strme.str());
00155 }
00156 }
00157
00158 inline FILE* fOpen(const char* fn,const char* m)
00159 {
00160 if(FILE* f=fopen(fn,m))
00161 return f;
00162 else {
00163 UTIL_THROW(util::Exception, "ERROR: could not open file " << fn << " with mode " << m);
00164 return NULL;
00165 }
00166 }
00167 inline void fClose(FILE* f)
00168 {
00169 fclose(f);
00170 }
00171
00172 }
00173
00174 #endif