f639e4e902
* Moved inference_engine samples to cpp folder * Fixed documentations links * Fixed installation * Fixed scripts * Fixed cmake script * Try to fix install * Fixed samples * Some fix
42 lines
898 B
C++
42 lines
898 B
C++
// Copyright (C) 2018-2021 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#include <format_reader.h>
|
|
|
|
#include <iostream>
|
|
|
|
#include "MnistUbyte.h"
|
|
#include "bmp.h"
|
|
#include "opencv_wrapper.h"
|
|
|
|
using namespace FormatReader;
|
|
|
|
std::vector<Registry::CreatorFunction> Registry::_data;
|
|
|
|
Register<MnistUbyte> MnistUbyte::reg;
|
|
#ifdef USE_OPENCV
|
|
Register<OCVReader> OCVReader::reg;
|
|
#else
|
|
Register<BitMap> BitMap::reg;
|
|
#endif
|
|
|
|
Reader* Registry::CreateReader(const char* filename) {
|
|
for (auto maker : _data) {
|
|
Reader* ol = maker(filename);
|
|
if (ol != nullptr && ol->size() != 0)
|
|
return ol;
|
|
if (ol != nullptr)
|
|
delete ol;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
void Registry::RegisterReader(CreatorFunction f) {
|
|
_data.push_back(f);
|
|
}
|
|
|
|
FORMAT_READER_API(Reader*) CreateFormatReader(const char* filename) {
|
|
return Registry::CreateReader(filename);
|
|
}
|