Welcome to mirror list, hosted at ThFree Co, Russian Federation.

io.h - github.com/moses-smt/vowpal_wabbit.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
path: root/io.h
blob: f7a2b7f963b57b2de2c4e593997e0e538384ef42 (plain)
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
/*
Copyright (c) 2009 Yahoo! Inc.  All rights reserved.  The copyrights
embodied in the content of this file are licensed under the BSD
(revised) open source license
 */

#ifndef IO_H__
#define IO_H__

#include "v_array.h"
#include<iostream>
using namespace std;

class io_buf {
 public:
  v_array<char > space;
  int file;
  char* endloaded;
  string currentname;
  string finalname;
  
  io_buf() {
    size_t s = 1 << 16;
    reserve(space, s); 
    endloaded = space.begin;}
  void set(char *p){space.end = p;}
  size_t fill() {
    if (space.end_array - endloaded == 0)
      {
	size_t offset = endloaded - space.begin;
	reserve(space, 2 * (space.end_array - space.begin));
	endloaded = space.begin+offset;
      }
    ssize_t num_read = read(file, endloaded, space.end_array - endloaded);
    endloaded = endloaded+num_read;
    return num_read;
  }
  void flush() { 
    if (write(file, space.begin, space.index()) != (int) space.index())
      cerr << "error, failed to write to cache\n";
    space.end = space.begin; fsync(file); }
};

void buf_write(io_buf &o, char* &pointer, int n);
unsigned int buf_read(io_buf &i, char* &pointer, int n);
size_t readto(io_buf &i, char* &pointer, char terminal);
#endif