99df448669
* refactor: add clang style check for samples * fix: add .clang-format for ie * fix: style check for missing headers * refactor: remove cpplint for IE samples * fix: setw is not a member of std for classification_results.hpp * feat: add indent after ifdefine * feat: set up google style for IE samples * fix indents for w_dirent headers * fix: include issues for utils.cpp due to clang-format
42 lines
897 B
C++
42 lines
897 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_wraper.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);
|
|
}
|