Blame view

test/main.h 1.37 KB
09c2d08c   Hu Chunming   arm交付版
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
  /**
  * @file utils.h
  *
  * Copyright (C) 2020. Huawei Technologies Co., Ltd. All rights reserved.
  *
  * This program is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  */
  #pragma once
  #include <iostream>
  #include <unistd.h>
  #include <dirent.h>
  #include <fstream>
  #include <cstring>
  #include <vector>
  #include <sys/types.h>
  #include <sys/stat.h>
  #include <map>
  #include <cstdint>
  #include "acl/acl.h"
  #include "acl/ops/acl_dvpp.h"
  
  #define INFO_LOG(fmt, args...) fprintf(stdout, "[INFO]  " fmt "\n", ##args)
  #define WARN_LOG(fmt, args...) fprintf(stdout, "[WARN]  " fmt "\n", ##args)
  #define ERROR_LOG(fmt, args...) fprintf(stderr, "[ERROR]  " fmt "\n", ##args)
  
  typedef enum Result {
      SUCCESS = 0,
      FAILED = 1
  } Result;
  
  typedef struct PicDesc {
      std::string picName;
      int width;
      int height;
  }PicDesc;
  
  typedef struct CropPicDesc {
      std::string picName;
      int left;
      int top;
      int width;
      int height;
  }CropPicDesc;
  
  int32_t deviceId_;
  aclrtContext context_;
  aclrtStream stream_;
  
  aclvdecChannelDesc *vdecChannelDesc_;
  acldvppStreamDesc *streamInputDesc_;
  acldvppPicDesc *picOutputDesc_;
  acldvppChannelDesc *dvppChannelDesc_;
  uint32_t inBufferSize;
  uint32_t inputWidth;
  uint32_t inputHeight;
  aclrtRunMode runMode;
  
  PicDesc inPicDesc;
  CropPicDesc outPicDesc;