diff options
Diffstat (limited to 'moses/src/File.h')
-rw-r--r-- | moses/src/File.h | 122 |
1 files changed, 122 insertions, 0 deletions
diff --git a/moses/src/File.h b/moses/src/File.h new file mode 100644 index 000000000..04e061980 --- /dev/null +++ b/moses/src/File.h @@ -0,0 +1,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 + |