NvElementProfiler.h

前往此文件的文档。

/*
 * 版权所有 (c) 2016, NVIDIA CORPORATION。保留所有权利。
 *
 * 在源代码和二进制形式中再发布和使用，无论是否
 * 修改，均被允许，但须遵守以下条件
 * 满足：
 * * 源代码的再发布必须保留上述版权
 * 声明、此条件列表和以下免责声明。
 * * 二进制形式的再发布必须在上述版权
 * 声明、此条件列表和以下免责声明中复制，
 * 在文档和/或随发行版提供的其他材料中。
 * * 未经事先书面许可，NVIDIA CORPORATION 的名称及其
 * 贡献者的姓名不得用于背书或推销产品
 * 从本软件衍生而来。
 *
 * 本软件由版权持有者“按原样”提供，并且任何
 * 明示或暗示的担保，包括但不限于
 * 对适销性和针对特定用途的适用性的暗示担保
 * 均不承担责任。在任何情况下，版权所有者或
 * 贡献者均不对任何直接、间接、附带、特殊、
 * 惩戒性或后果性损害（包括但不限于
 * 采购替代商品或服务；使用、数据或
 * 利润损失；或业务中断）负责，无论何种原因以及基于何种理论
 * 的责任，无论是合同、严格责任还是侵权
 * （包括疏忽或其他原因）以任何方式因使用
 * 本软件而引起，即使已被告知可能发生此类损害。
 */

#ifndef __NV_ELEMENT_PROFILER_H__
#define __NV_ELEMENT_PROFILER_H__

#include <iostream>
#include <pthread.h>
#include <map>
#include <stdint.h>
#include <sys/time.h>

class NvElementProfiler {
public
 typedef int ProfilerField;
 static const ProfilerField PROFILER_FIELD_NONE = 0;
 static const ProfilerField PROFILER_FIELD_TOTAL_UNITS = 1;
 static const ProfilerField PROFILER_FIELD_LATE_UNITS = 2;
 static const ProfilerField PROFILER_FIELD_LATENCIES = 4;
 static const ProfilerField PROFILER_FIELD_FPS = 8;
 static const ProfilerField PROFILER_FIELD_ALL = (PROFILER_FIELD_FPS << 1) - 1;
 typedef struct {
 ProfilerField valid_fields;

 uint64_t average_latency_usec;
 uint64_t min_latency_usec;
 uint64_t max_latency_usec;

 uint64_t total_processed_units;
 uint64_t num_late_units;

 float average_fps;

 struct timeval profiling_time;
 } NvElementProfilerData;

 void getProfilerData(NvElementProfilerData &data);

 void printProfilerData(std::ostream &out_stream = std::cout);

 uint64_t startProcessing();

 void finishProcessing(uint64_t id, bool is_late);

 void enableProfiling(bool reset_data);

 void disableProfiling();
private
 void reset();

 pthread_mutex_t profiler_lock;
 bool enabled;
 const ProfilerField valid_fields;
 struct NvElementProfilerDataInternal : NvElementProfilerData {
 struct timeval start_time;

 struct timeval stop_time;

 struct timeval accumulated_time;

 uint64_t total_latency;
 } data_int;

 std::map<uint64_t, struct timeval> unit_start_time_queue;

 uint64_t unit_id_counter;
 NvElementProfiler(ProfilerField fields);

 NvElementProfiler(const NvElementProfiler& that);
 void operator=(NvElementProfiler const&);

 ~NvElementProfiler();

 friend class NvElement;
};

#endif