fluxengine/lib/fluxsource/fluxsource.cc

#include "globals.h"
#include "flags.h"
#include "fluxsource/fluxsource.h"
#include "fluxmap.h"
#include "lib/config.pb.h"
#include "proto.h"
#include "utils.h"

std::unique_ptr<FluxSource> FluxSource::create(const FluxSourceProto& config)
{
    switch (config.type())
    {
        case FluxSourceProto::DRIVE:
            return createHardwareFluxSource(config.drive());

        case FluxSourceProto::ERASE:
            return createEraseFluxSource(config.erase());

        case FluxSourceProto::KRYOFLUX:
            return createKryofluxFluxSource(config.kryoflux());

        case FluxSourceProto::TEST_PATTERN:
            return createTestPatternFluxSource(config.test_pattern());

        case FluxSourceProto::SCP:
            return createScpFluxSource(config.scp());

        case FluxSourceProto::A2R:
            return createA2rFluxSource(config.a2r());

        case FluxSourceProto::CWF:
            return createCwfFluxSource(config.cwf());

        case FluxSourceProto::FLUX:
            return createFl2FluxSource(config.fl2());

        case FluxSourceProto::FLX:
            return createFlxFluxSource(config.flx());

        default:
            error("bad input disk configuration");
            return std::unique_ptr<FluxSource>();
    }
}

class TrivialFluxSourceIterator : public FluxSourceIterator
{
public:
    TrivialFluxSourceIterator(
        TrivialFluxSource* fluxSource, int track, int head):
        _fluxSource(fluxSource),
        _track(track),
        _head(head)
    {
    }

    bool hasNext() const override
    {
        return !!_fluxSource;
    }

    std::unique_ptr<const Fluxmap> next() override
    {
        auto fluxmap = _fluxSource->readSingleFlux(_track, _head);
        _fluxSource = nullptr;
        return fluxmap;
    }

private:
    TrivialFluxSource* _fluxSource;
    int _track;
    int _head;
};

std::unique_ptr<FluxSourceIterator> TrivialFluxSource::readFlux(
    int track, int head)
{
    return std::make_unique<TrivialFluxSourceIterator>(this, track, head);
}