mirror of
https://github.com/davidgiven/fluxengine.git
synced 2025-10-24 11:11:02 -07:00
f24e4029b4
which is now honoured. Fix a bunch of bugs in some of the flux sources and sinks. The converter now actually works, maybe.
92 lines
2.3 KiB
C++
92 lines
2.3 KiB
C++
#include "lib/core/globals.h"
|
|
#include "lib/config/config.h"
|
|
#include "lib/config/flags.h"
|
|
#include "lib/fluxsource/fluxsource.h"
|
|
#include "lib/data/fluxmap.h"
|
|
#include "lib/config/config.pb.h"
|
|
#include "lib/config/proto.h"
|
|
#include "lib/core/utils.h"
|
|
|
|
std::unique_ptr<FluxSource> FluxSource::create(Config& config)
|
|
{
|
|
if (!config.hasFluxSource())
|
|
error("no flux source configured");
|
|
auto fluxSource = create(config->flux_source());
|
|
globalConfig().base()->MergeFrom(fluxSource->getExtraConfig());
|
|
return fluxSource;
|
|
}
|
|
|
|
std::unique_ptr<FluxSource> FluxSource::create(const FluxSourceProto& config)
|
|
{
|
|
switch (config.type())
|
|
{
|
|
case FLUXTYPE_DRIVE:
|
|
return createHardwareFluxSource(config.drive());
|
|
|
|
case FLUXTYPE_ERASE:
|
|
return createEraseFluxSource(config.erase());
|
|
|
|
case FLUXTYPE_KRYOFLUX:
|
|
return createKryofluxFluxSource(config.kryoflux());
|
|
|
|
case FLUXTYPE_TEST_PATTERN:
|
|
return createTestPatternFluxSource(config.test_pattern());
|
|
|
|
case FLUXTYPE_SCP:
|
|
return createScpFluxSource(config.scp());
|
|
|
|
case FLUXTYPE_A2R:
|
|
return createA2rFluxSource(config.a2r());
|
|
|
|
case FLUXTYPE_CWF:
|
|
return createCwfFluxSource(config.cwf());
|
|
|
|
case FLUXTYPE_DMK:
|
|
return createDmkFluxSource(config.dmk());
|
|
|
|
case FLUXTYPE_FLUX:
|
|
return createFl2FluxSource(config.fl2());
|
|
|
|
case FLUXTYPE_FLX:
|
|
return createFlxFluxSource(config.flx());
|
|
|
|
default:
|
|
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);
|
|
}
|