summaryrefslogtreecommitdiff
path: root/python/openvino/runtime/common/format_reader/format_reader_ptr.h
blob: eb9bf8ed87c08dfdd5e2475454879217d43ec585 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
// Copyright (C) 2018-2022 Intel Corporation
// SPDX-License-Identifier: Apache-2.0
//

/**
 * \brief Implementation of smart pointer for Reader class
 * \file format_reader_ptr.h
 */
#pragma once

#include <functional>
#include <memory>

#include "format_reader.h"

namespace FormatReader {
class ReaderPtr {
public:
    explicit ReaderPtr(const char* imageName) : reader(CreateFormatReader(imageName)) {}
    /**
     * @brief dereference operator overload
     * @return Reader
     */
    Reader* operator->() const noexcept {
        return reader.get();
    }

    /**
     * @brief dereference operator overload
     * @return Reader
     */
    Reader* operator*() const noexcept {
        return reader.get();
    }

    Reader* get() {
        return reader.get();
    }

protected:
    std::unique_ptr<Reader> reader;
};
}  // namespace FormatReader