Быстрое контролируемое копирование из istream в ostream - PullRequest
Новая
       16

Быстрое контролируемое копирование из istream в ostream

15 голосов
/ 31 октября 2010

Мне нужно скопировать несколько байтов из istream в ostream, я знаю, что я могу выполнить эту копию двумя способами. 

myostream << myistream.rdbuf();

и 

copy( istreambuf_iterator<char>(myistream),
      istreambuf_iterator<char>(),
      ostreambuf_iterator<char>(myostream)
);

Я обнаружил, что версия rdbuf как минимум в два раза быстрее, чем copy.
Я еще не нашел способ копирования, скажем, 100 байт, но, поскольку копируемый размер, вероятно, будет достаточно большим, я хотел бы иметь возможность использовать версию rdbuf, если возможно.

Как ограничить эти копии определенным количеством байтов?

1 Ответ

11 голосов
/ 31 октября 2010

Можете ли вы использовать 0x? Если это так, то вы можете использовать copy_n: 

copy_n( istreambuf_iterator<char>(myistream),
        100,
        ostreambuf_iterator<char>(myostream)
);

РЕДАКТИРОВАТЬ 1:

Я знаю, что вы, вероятно, ищете решение для библиотеки, и вы, вероятно, могли бы выяснить это самостоятельно. Но если вы не подумали о чем-то подобном, вот что я бы сделал (если бы у меня не было copy_n): 

void stream_copy_n(std::istream & in, std::size_t count, std::ostream & out)
{
    const std::size_t buffer_size = 4096;
    char buffer[buffer_size];
    while(count > buffer_size)
    {
        in.read(buffer, buffer_size);
        out.write(buffer, buffer_size);
        count -= buffer_size;
    }

    in.read(buffer, count);
    out.write(buffer, count);
}
...