EGLMinSampleJetson/main.cpp at master · cbodenst/EGLMinSampleJetson · GitHub

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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
/*
 * main.cpp
 *
 * Copyright 2016 NVIDIA Corporation.  All rights reserved.
 *
 * Please refer to the NVIDIA end user license agreement (EULA) associated
 * with this source code for terms and conditions that govern your use of
 * this software. Any use, reproduction, disclosure, or distribution of
 * this software and related documentation outside the terms of the EULA
 * is strictly prohibited.
 *
 */

//
// DESCRIPTION:   Simple EGL stream sample app
//
//

//#define EGL_EGLEXT_PROTOTYPES

#include "cuda_consumer.h"
#include "cuda_producer.h"
#include "eglstrm_common.h"
#include "cudaEGL.h"

/* ------  globals ---------*/

int main(int argc, char **argv)
{
    TestArgs args;
    CUresult curesult = CUDA_SUCCESS;
    unsigned int i, j;
    EGLint streamState = 0;

    test_cuda_consumer_s cudaConsumer;
    test_cuda_producer_s cudaProducer;

    memset(&cudaProducer, 0, sizeof(test_cuda_producer_s));
    memset(&cudaConsumer, 0, sizeof(test_cuda_consumer_s));

    // Hook up Ctrl-C handler
    if(!eglSetupExtensions()) {
        printf("SetupExtentions failed \n");
        curesult = CUDA_ERROR_UNKNOWN;
        goto done;
    }

    if(!EGLStreamInit()) {
        printf("EGLStream Init failed.\n");
        curesult = CUDA_ERROR_UNKNOWN;
        goto done;
    }
    curesult = cudaDeviceCreateProducer(&cudaProducer);
    if (curesult != CUDA_SUCCESS) {
        goto done;
    }
    curesult = cudaDeviceCreateConsumer(&cudaConsumer);
    if (curesult != CUDA_SUCCESS) {
        goto done;
    }
    if (CUDA_SUCCESS != (curesult = cuEGLStreamConsumerConnect(&(cudaConsumer.cudaConn), eglStream))) {
        printf("FAILED Connect CUDA consumer  with error %d\n", curesult);
        goto done;
    }
    else {
        printf("Connected CUDA consumer, CudaConsumer %p\n", cudaConsumer.cudaConn);
    }
    if (CUDA_SUCCESS == (curesult = cuEGLStreamProducerConnect(&(cudaProducer.cudaConn), eglStream, WIDTH, HEIGHT))) {
        printf("Connect CUDA producer Done, CudaProducer %p\n", cudaProducer.cudaConn);
    } else {
        printf("Connect CUDA producer FAILED with error %d\n", curesult);
        goto done;
    }

    // Initialize producer
    args.inputWidth        = WIDTH;
    args.inputHeight       = HEIGHT;
    args.isARGB        = 0;
    args.infile1       = "cuda_f_1.yuv";
    args.infile2       = "cuda_f_2.yuv";
    args.pitchLinearOutput = 1;

    cudaProducerInit(&cudaProducer, g_display, eglStream, &args);
    cuda_consumer_init(&cudaConsumer, &args);

    printf("main - Cuda Producer and Consumer Initialized.\n");

    printf("Running for YUV frame and Pitchlinear\n");
    curesult = cudaProducerTest(&cudaProducer, cudaProducer.fileName1);
    if (curesult != CUDA_SUCCESS) {
        printf("Cuda Producer Test failed for frame 1\n");
        goto done;
    }
    curesult = cudaConsumerTest(&cudaConsumer, cudaConsumer.outFile1);
    if (curesult != CUDA_SUCCESS) {
        printf("Cuda Consumer Test failed for frame = 1\n");
        goto done;
    }
    curesult = cudaProducerTest(&cudaProducer, cudaProducer.fileName2);
    if (curesult != CUDA_SUCCESS) {
        printf("Cuda Producer Test failed for frame = 1\n");
        goto done;
    }

    curesult = cudaConsumerTest(&cudaConsumer, cudaConsumer.outFile2);
    if (curesult != CUDA_SUCCESS) {
        printf("Cuda Consumer Test failed for frame = 2\n");
        goto done;
    }

    if (CUDA_SUCCESS != (curesult = cudaProducerDeinit(&cudaProducer))) {
        printf("Producer Disconnect FAILED. \n");
        goto done;
    }
    if(!eglQueryStreamKHR_wrapper(
                g_display,
                eglStream,
                EGL_STREAM_STATE_KHR,
                &streamState)) {
        printf("Cuda consumer, eglQueryStreamKHR EGL_STREAM_STATE_KHR failed\n");
        curesult = CUDA_ERROR_UNKNOWN;
        goto done;
    }
    if(streamState != EGL_STREAM_STATE_DISCONNECTED_KHR) {
        if (CUDA_SUCCESS != (curesult = cuda_consumer_deinit(&cudaConsumer))) {
            printf("Consumer Disconnect FAILED.\n");
            goto done;
        }
    }
    printf("Producer and Consumer Disconnected \n");

done:
    if(!eglQueryStreamKHR_wrapper(
                g_display,
                eglStream,
                EGL_STREAM_STATE_KHR,
                &streamState)) {
        printf("Cuda consumer, eglQueryStreamKHR EGL_STREAM_STATE_KHR failed\n");
        curesult = CUDA_ERROR_UNKNOWN;
    }
    if(streamState != EGL_STREAM_STATE_DISCONNECTED_KHR) {
        EGLStreamFini();
    }

    if (curesult == CUDA_SUCCESS) {
        printf("&&&& EGLStream interop test PASSED\n");
    }
    else {
        printf("&&&& EGLStream interop test FAILED\n");
    }
    return 0;
}