summaryrefslogtreecommitdiff
path: root/code/test/vid/client.c
blob: 017967bdcb5e8d922c99ddbe41a505b272d04976 (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
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
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>

#include "vpx/vpx_decoder.h"
#include "vpx/vp8cx.h"
#include "tools.h"
#include "display.h"

#include "core.h"
#include "vconn/vconn.h"
#include "util.h"

#define CTYPE_TEST      0
#define MTYPE_MSG       8

#define FRAG_BUF_SIZE   8192
#define RBUF_MSG_SIZE   128

ECPContext ctx;
ECPSocket sock;
ECPConnHandler handler;

ECPNode node;
ECPConnection conn;

ECPVConnection vconn;
ECPNode vconn_node;

vpx_codec_ctx_t codec;

ECPRBRecv rbuf_recv;
ECPRBMessage rbuf_r_msg[RBUF_MSG_SIZE];
ECPFragIter frag_iter;
unsigned char frag_buffer[FRAG_BUF_SIZE];

SDLCanvas sdl_canvas;

ssize_t handle_msg(ECPConnection *conn, ecp_seq_t sq, unsigned char t, unsigned char *f, ssize_t sz, ECP2Buffer *b) {
    vpx_codec_iter_t iter = NULL;
    vpx_image_t *img = NULL;

    if (vpx_codec_decode(&codec, f, (unsigned int)sz, NULL, 0)) {
        fprintf(stderr, "\n%lu\n", sz);
        fprintf(stderr, "ERROR!\n");
        // die_codec(&codec, "Failed to decode frame.");
    }

    while ((img = vpx_codec_get_frame(&codec, &iter)) != NULL) {
        if (!vpx_img_write(img, sdl_canvas.yuvBuffer, sdl_canvas.yPlaneSz + 2 * sdl_canvas.uvPlaneSz)) die_codec(NULL, "Failed to write image.");
    }
    sdl_display_frame(&sdl_canvas);
    
    return sz;
}

static void usage(char *arg) {
    fprintf(stderr, "Usage: %s <node.pub> <vcs.pub>\n", arg);
    exit(1);
}

int main(int argc, char *argv[]) {
    int rv;
    
    if (argc != 3) usage(argv[0]);
    
    rv = ecp_init(&ctx);
    fprintf(stderr, "ecp_init RV:%d\n", rv);

    if (!rv) rv = ecp_conn_handler_init(&handler);
    if (!rv) {
        handler.msg[MTYPE_MSG] = handle_msg;
        ctx.handler[CTYPE_TEST] = &handler;
    }
    
    if (!rv) rv = ecp_sock_create(&sock, &ctx, NULL);
    fprintf(stderr, "ecp_sock_create RV:%d\n", rv);

    if (!rv) rv = ecp_sock_open(&sock, NULL);
    fprintf(stderr, "ecp_sock_open RV:%d\n", rv);
    
    if (!rv) rv = ecp_start_receiver(&sock);
    fprintf(stderr, "ecp_start_receiver RV:%d\n", rv);

    if (!rv) rv = ecp_util_node_load(&ctx, &node, argv[1]);
    fprintf(stderr, "ecp_util_node_load RV:%d\n", rv);

    if (!rv) rv = ecp_util_node_load(&ctx, &vconn_node, argv[2]);
    printf("ecp_util_node_load RV:%d\n", rv);

    if (!rv) rv = ecp_conn_create(&conn, &sock, CTYPE_TEST);
    fprintf(stderr, "ecp_conn_create RV:%d\n", rv);

    if (!rv) rv = ecp_rbuf_create(&conn, NULL, NULL, 0, &rbuf_recv, rbuf_r_msg, RBUF_MSG_SIZE);
    fprintf(stderr, "ecp_rbuf_create RV:%d\n", rv);
    
    if (!rv) {
        ecp_frag_iter_init(&frag_iter, frag_buffer, FRAG_BUF_SIZE);
        rbuf_recv.frag_iter = &frag_iter;
    }

    const char *codec_arg = "vp9";
    const VpxInterface *decoder = get_vpx_decoder_by_name(codec_arg);
    if (!decoder) die_codec(NULL, "Unknown input codec.");

    fprintf(stderr, "Using %s\n", vpx_codec_iface_name(decoder->codec_interface()));

    if (vpx_codec_dec_init(&codec, decoder->codec_interface(), NULL, 0))
      die_codec(&codec, "Failed to initialize decoder.");
    
    sdl_open(&sdl_canvas, 640, 480);
    
    // if (!rv) rv = ecp_conn_open(&conn, &node);
    // fprintf(stderr, "ecp_conn_open RV:%d\n", rv);

    if (!rv) rv = ecp_vconn_open(&conn, &node, &vconn, &vconn_node, 1);
    printf("ecp_vconn_open RV:%d\n", rv);

    sdl_loop();
    sdl_close(&sdl_canvas);
    if (vpx_codec_destroy(&codec)) die_codec(&codec, "Failed to destroy codec");
}