Add option to specify object classes list file
[tidl/tidl-api.git] / examples / common / utils.h
1 /******************************************************************************
2  * Copyright (c) 2018, Texas Instruments Incorporated - http://www.ti.com/
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are met:
7  *     * Redistributions of source code must retain the above copyright
8  *       notice, this list of conditions and the following disclaimer.
9  *     * Redistributions in binary form must reproduce the above copyright
10  *       notice, this list of conditions and the following disclaimer in the
11  *       documentation and/or other materials provided with the distribution.
12  *     * Neither the name of Texas Instruments Incorporated nor the
13  *       names of its contributors may be used to endorse or promote products
14  *       derived from this software without specific prior written permission.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
20  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
26  * THE POSSIBILITY OF SUCH DAMAGE.
27  *****************************************************************************/
28 #pragma once
30 #include <string>
31 #include <iostream>
32 #include <fstream>
33 #include <vector>
34 #include "executor.h"
35 #include "execution_object.h"
36 #include "execution_object_pipeline.h"
37 #include "configuration.h"
39 using tidl::Executor;
40 using tidl::ExecutionObject;
41 using tidl::ExecutionObjectPipeline;
42 using tidl::Configuration;
43 using tidl::DeviceType;
45 Executor* CreateExecutor(DeviceType dt, int num, const Configuration& c,
46                          int layer_group_id);
48 bool ReadFrame(ExecutionObject*     eo,
49                int                  frame_idx,
50                const Configuration& configuration,
51                std::istream&        input_file);
53 bool ReadFrame(ExecutionObjectPipeline* eop,
54                int                      frame_idx,
55                const Configuration&     configuration,
56                std::istream&            input_file);
58 namespace tidl {
59   std::size_t GetBinaryFileSize (const std::string &F);
60   bool        ReadBinary        (const std::string &F, char* buffer, int size);
61 };
63 bool WriteFrame(const ExecutionObject* eo, std::ostream& output_file);
65 void ReportTime(const ExecutionObject* eo);
67 bool CheckFrame(const ExecutionObject* eo, const char *ref_output);
68 bool CheckFrame(const ExecutionObjectPipeline *eop, const char *ref_output);
70 const char* ReadReferenceOutput(const std::string& name);
72 void AllocateMemory(const std::vector<ExecutionObject *>& eos);
73 void FreeMemory(const std::vector<ExecutionObject *>& eos);
74 void AllocateMemory(const std::vector<ExecutionObjectPipeline *>& eops);
75 void FreeMemory(const std::vector<ExecutionObjectPipeline *>& eops);