BlockEncoder.h

Go to the documentation of this file.

/****
 * ArchiveStream.h
 *
 * Copyright 2021 mikee47 <mike@sillyhouse.net>
 *
 * This file is part of the IFS Library
 *
 * This library is free software: you can redistribute it and/or modify it under the terms of the
 * GNU General Public License as published by the Free Software Foundation, version 3 or later.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with this library.
 * If not, see <https://www.gnu.org/licenses/>.
 *
 ****/
 
#pragma once
 
#include <Data/Stream/DataSourceStream.h>
 
namespace IFS::FWFS
{
class IBlockEncoder
{
public:
    virtual ~IBlockEncoder() = default;
 
    virtual IDataSourceStream* getNextStream() = 0;
};
 
class BasicEncoder : public IBlockEncoder
{
public:
    BasicEncoder(IDataSourceStream* stream) : stream(stream)
    {
    }
 
    IDataSourceStream* getNextStream() override
    {
        if(done) {
            stream.reset();
            return nullptr;
        }
 
        done = true;
        return stream.get();
    }
 
protected:
    std::unique_ptr<IDataSourceStream> stream;
    bool done{false};
};
 
} // namespace IFS::FWFS