/home/palakars/workspace-inprogress/code-in-progress/aitools3/aitools3-aq-invertedindex-cpp/src/writer/PostlistWriter.cpp

Go to the documentation of this file.

#include "PostlistWriter.hpp"
#include "System.hpp"

namespace aitools {
namespace invertedindex {

PostlistWriter::PostlistWriter()
:       file_(NULL)
{}

PostlistWriter::PostlistWriter(const bfs::path& path)
throw (std::invalid_argument)
:       file_(NULL)
{
        open(path);
}

PostlistWriter::~PostlistWriter()
{
        close();
}

void
PostlistWriter::close()
{
        if (is_open())
        {
                path_ = bfs::path();
                System::fclose(file_);
                file_ = NULL;
        }
}

bool
PostlistWriter::is_open() const
{
        return file_ != NULL;
}

void
PostlistWriter::open(const bfs::path& path)
throw (std::runtime_error, std::invalid_argument)
{
        if (is_open())
        {
                Exception::throw_runtime_error("File already open");
        }
        file_ = System::fopen(path, "wb");
        path_ = path;
}

const bfs::path&
PostlistWriter::path() const
{
        return path_;
}

void
PostlistWriter::write(Iterator::SharedPointer iterator)
{
        assert(iterator);
        iterator->rewind();
        System::fwrite((char*)&iterator->header(),
                        Iterator::sizeof_header, 1, file_);
        if (!iterator->value_sizes().empty())
        {
                System::fwrite((char*)iterator->value_sizes().data(),
                                sizeof(iterator->value_sizes().front()),
                                iterator->value_sizes().size(), file_);
        }
        do
        {
                System::fwrite((char*)iterator->chunk().buffer->data(), 1,
                                iterator->chunk().buffer->size(), file_);
        }
        while (iterator->swap());
}

bool
PostlistWriter::ready() const
{
        return is_open() && std::feof(file_) == 0;
}

uint64_t
PostlistWriter::tell()
{
        return is_open() ? System::ftell(file_) : 0;
}

} // namespace invertedindex
} // namespace aitools

---