mirror of
https://github.com/moses-smt/mosesdecoder.git
synced 2024-08-16 15:00:33 +03:00
175 lines
3.7 KiB
C++
175 lines
3.7 KiB
C++
// $Id$
|
|
|
|
/* ---------------------------------------------------------------- */
|
|
/* Copyright 2005 (c) by RWTH Aachen - Lehrstuhl fuer Informatik VI */
|
|
/* Richard Zens */
|
|
/* ---------------------------------------------------------------- */
|
|
#ifndef moses_File_h
|
|
#define moses_File_h
|
|
|
|
#include <cstdio>
|
|
#include <iostream>
|
|
#include <sstream>
|
|
#include <vector>
|
|
#include "util/exception.hh"
|
|
#include "util/string_stream.hh"
|
|
#include "TypeDef.h"
|
|
#include "Util.h"
|
|
|
|
namespace Moses
|
|
{
|
|
|
|
#ifdef WIN32
|
|
#ifdef __MINGW32__
|
|
#define OFF_T __int64
|
|
#define FTELLO(f) ftello64(f)
|
|
#define FSEEKO(file, offset, origin) fseeko64(file, offset, origin)
|
|
#else
|
|
#define OFF_T __int64
|
|
#define FTELLO(file) _ftelli64(file)
|
|
#define FSEEKO(file, offset, origin) _fseeki64(file, offset, origin)
|
|
#endif
|
|
|
|
#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) {
|
|
UTIL_THROW2("ERROR:: fwrite!");
|
|
}
|
|
return sizeof(t);
|
|
}
|
|
|
|
template<typename T> inline void fRead(FILE* f,T& t)
|
|
{
|
|
if(fread(&t,sizeof(t),1,f)!=1) {
|
|
UTIL_THROW2("ERROR: fread!");
|
|
}
|
|
}
|
|
|
|
template<typename T> inline size_t fWrite(FILE* f,const T* b,const T* e)
|
|
{
|
|
uint32_t s=std::distance(b,e);
|
|
size_t rv=fWrite(f,s);
|
|
if(fwrite(b,sizeof(T),s,f)!=s) {
|
|
UTIL_THROW2("ERROR: fwrite!");
|
|
}
|
|
return rv+sizeof(T)*s;
|
|
}
|
|
|
|
template<typename T> inline size_t fWrite(FILE* f,const T b,const T e)
|
|
{
|
|
uint32_t s=std::distance(b,e);
|
|
size_t rv=fWrite(f,s);
|
|
if(fwrite(&(*b),sizeof(T),s,f)!=s) {
|
|
UTIL_THROW2("ERROR: fwrite!");
|
|
}
|
|
return rv+sizeof(T)*s;
|
|
}
|
|
|
|
template<typename C> inline size_t fWriteVector(FILE* f,const C& v)
|
|
{
|
|
uint32_t s=v.size();
|
|
size_t rv=fWrite(f,s);
|
|
if(fwrite(&v[0],sizeof(typename C::value_type),s,f)!=s) {
|
|
UTIL_THROW2("ERROR: fwrite!");
|
|
}
|
|
return rv+sizeof(typename C::value_type)*s;
|
|
}
|
|
|
|
template<typename C> inline void fReadVector(FILE* f, C& v)
|
|
{
|
|
uint32_t s;
|
|
fRead(f,s);
|
|
v.resize(s);
|
|
size_t r=fread(&(*v.begin()),sizeof(typename C::value_type),s,f);
|
|
if(r!=s) {
|
|
UTIL_THROW2("ERROR: freadVec! "<<r<<" "<<s);
|
|
}
|
|
}
|
|
|
|
inline size_t fWriteString(FILE* f,const char* e, uint32_t s)
|
|
{
|
|
size_t rv=fWrite(f,s);
|
|
if(fwrite(e,sizeof(char),s,f)!=s) {
|
|
UTIL_THROW2("ERROR:: fwrite!");
|
|
}
|
|
return rv+sizeof(char)*s;
|
|
}
|
|
|
|
inline void fReadString(FILE* f,std::string& e)
|
|
{
|
|
uint32_t s;
|
|
fRead(f,s);
|
|
char* a=new char[s+1];
|
|
if(fread(a,sizeof(char),s,f)!=s) {
|
|
UTIL_THROW2("ERROR: fread!");
|
|
}
|
|
a[s]='\0';
|
|
e.assign(a);
|
|
delete[](a);
|
|
}
|
|
|
|
inline size_t fWriteStringVector(FILE* f,const std::vector<std::string>& v)
|
|
{
|
|
uint32_t 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_t 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) {
|
|
util::StringStream strme;
|
|
strme << "ERROR: could not fseeko position " << o <<"\n";
|
|
if(o==InvalidOffT) strme << "You tried to seek for 'InvalidOffT'!\n";
|
|
UTIL_THROW2(strme.str());
|
|
}
|
|
}
|
|
|
|
inline FILE* fOpen(const char* fn,const char* m)
|
|
{
|
|
if(FILE* f=fopen(fn,m))
|
|
return f;
|
|
else {
|
|
UTIL_THROW(util::Exception, "ERROR: could not open file " << fn << " with mode " << m);
|
|
return NULL;
|
|
}
|
|
}
|
|
inline void fClose(FILE* f)
|
|
{
|
|
fclose(f); // for consistent function names only
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|