summaryrefslogtreecommitdiffstats
path: root/src/grabbers/me4.c
blob: 6cd72ae90114a5a69578b58268f04dce19ce9497 (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
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
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229

#include <stdlib.h>
#include <string.h>
#include <clser.h>
#include <fgrab_struct.h>
#include <fgrab_prototyp.h>

#include "uca.h"
#include "uca-grabber.h"

struct fg_apc_data {
    Fg_Struct               *fg;
    dma_mem                 *mem;
    uint32_t                timeout;

    /* End-user related callback variables */
    uca_cam_grab_callback   callback;
    void                    *meta_data;
    void                    *user;
};

struct uca_sisofg_map_t {
    enum uca_grabber_constants uca_prop;
    int  fg_id;
    bool interpret_data;    /**< is data a constant or a value? */
};

static struct uca_sisofg_map_t uca_to_fg[] = {
    /* properties */
    { UCA_GRABBER_WIDTH,            FG_WIDTH,               false },
    { UCA_GRABBER_HEIGHT,           FG_HEIGHT,              false },
    { UCA_GRABBER_OFFSET_X,         FG_XOFFSET,             false },
    { UCA_GRABBER_OFFSET_Y,         FG_YOFFSET,             false },
    { UCA_GRABBER_EXPOSURE,         FG_EXPOSURE,            false },
    { UCA_GRABBER_TIMEOUT,          FG_TIMEOUT,             false },
    { UCA_GRABBER_TRIGGER_MODE,     FG_TRIGGERMODE,         true},
    { UCA_GRABBER_FORMAT,           FG_FORMAT,              true},
    { UCA_GRABBER_CAMERALINK_TYPE,  FG_CAMERA_LINK_CAMTYP,  true },

    /* values */
    { UCA_FORMAT_GRAY8,             FG_GRAY,                false },
    { UCA_FORMAT_GRAY16,            FG_GRAY16,              false },
    { UCA_CL_8BIT_FULL_8,           FG_CL_8BIT_FULL_8,      false },
    { UCA_CL_8BIT_FULL_10,          FG_CL_8BIT_FULL_10,     false },
    { UCA_TRIGGER_FREERUN,          FREE_RUN,               false },
    { UCA_GRABBER_INVALID,          0,                      false }
};

#define GET_FG(grabber) (((struct fg_apc_data *) grabber->user)->fg)
#define GET_MEM(grabber) (((struct fg_apc_data *) grabber->user)->mem)

uint32_t uca_me4_destroy(struct uca_grabber *grabber)
{
    if (grabber != NULL) {
        Fg_FreeMemEx(GET_FG(grabber), GET_MEM(grabber));
        Fg_FreeGrabber(GET_FG(grabber));
    }
    return UCA_NO_ERROR;
}

static struct uca_sisofg_map_t *uca_me4_find_property(enum uca_grabber_constants property)
{
    int i = 0;
    /* Find a valid frame grabber id for the property */
    while (uca_to_fg[i].uca_prop != UCA_GRABBER_INVALID) {
        if (uca_to_fg[i].uca_prop == property)
            return &uca_to_fg[i];
        i++;
    }
    return NULL;
}

uint32_t uca_me4_set_property(struct uca_grabber *grabber, enum uca_grabber_constants property, void *data)
{
    struct uca_sisofg_map_t *fg_prop = uca_me4_find_property(property);
    if (fg_prop == NULL)
        return UCA_ERR_PROP_INVALID;

    switch (property) {
        case UCA_GRABBER_TIMEOUT:
            ((struct fg_apc_data *) grabber->user)->timeout = *((uint32_t *) data);
            break;

        default:
            break;
    }

    if (fg_prop->interpret_data) {
        /* Data is not a value but a SiSo specific constant that we need to
         * translate to Silicon Software speak. Therefore, we try to find it in
         * the map. */
        struct uca_sisofg_map_t *constant = uca_me4_find_property(*((uint32_t *) data));
        if (constant != NULL)
            return Fg_setParameter(GET_FG(grabber), fg_prop->fg_id, &constant->fg_id, PORT_A) == FG_OK ? UCA_NO_ERROR : UCA_ERR_PROP_INVALID;
        return UCA_ERR_PROP_INVALID;
    }
    else
        return Fg_setParameter(GET_FG(grabber), fg_prop->fg_id, data, PORT_A) == FG_OK ? UCA_NO_ERROR : UCA_ERR_PROP_GENERAL;
}

uint32_t uca_me4_get_property(struct uca_grabber *grabber, enum uca_grabber_constants property, void *data)
{
    struct uca_sisofg_map_t *fg_prop = uca_me4_find_property(property);
    if (fg_prop == NULL)
        return UCA_ERR_PROP_INVALID;

    /* FIXME: translate data back to UCA_ normalized constants */
    return Fg_getParameter(GET_FG(grabber), fg_prop->fg_id, data, PORT_A) == FG_OK ? UCA_NO_ERROR : UCA_ERR_PROP_GENERAL;
}

uint32_t uca_me4_alloc(struct uca_grabber *grabber, uint32_t pixel_size, uint32_t n_buffers)
{
    dma_mem *mem = GET_MEM(grabber);
    /* If buffers are already allocated, we are freeing every buffer and start
     * again. */
    if (mem != NULL) 
        Fg_FreeMemEx(GET_FG(grabber), mem);

    uint32_t width, height;
    uca_me4_get_property(grabber, UCA_GRABBER_WIDTH, &width);
    uca_me4_get_property(grabber, UCA_GRABBER_HEIGHT, &height);

    mem = Fg_AllocMemEx(GET_FG(grabber), n_buffers*width*height*pixel_size, n_buffers);
    if (mem != NULL) {
        ((struct fg_apc_data *) grabber->user)->mem = mem;
        return UCA_NO_ERROR;
    }
    return UCA_ERR_PROP_GENERAL;
}

uint32_t uca_me4_acquire(struct uca_grabber *grabber, int32_t n_frames)
{
    if (GET_MEM(grabber) == NULL)
        return UCA_ERR_GRABBER_NOMEM;

    int flag = grabber->asynchronous ? ACQ_STANDARD : ACQ_BLOCK;
    n_frames = n_frames < 0 ? GRAB_INFINITE : n_frames;
    if (Fg_AcquireEx(GET_FG(grabber), 0, n_frames, flag, GET_MEM(grabber)) == FG_OK)
        return UCA_NO_ERROR;

    return UCA_ERR_GRABBER_ACQUIRE;
}

uint32_t uca_me4_stop_acquire(struct uca_grabber *grabber)
{
    if (GET_MEM(grabber) != NULL)
        if (Fg_stopAcquireEx(GET_FG(grabber), 0, GET_MEM(grabber), STOP_SYNC) != FG_OK)
            return UCA_ERR_PROP_GENERAL;
    return UCA_NO_ERROR;
}

uint32_t uca_me4_grab(struct uca_grabber *grabber, void **buffer, uint64_t *frame_number)
{
    static frameindex_t last_frame = 0;
    struct fg_apc_data *me4 = (struct fg_apc_data *) grabber->user;

    if (grabber->asynchronous)
        last_frame = Fg_getLastPicNumberEx(me4->fg, PORT_A, me4->mem);
    else 
        last_frame = Fg_getLastPicNumberBlockingEx(me4->fg, last_frame+1, PORT_A, me4->timeout, me4->mem);

    if (last_frame <= 0)
        return UCA_ERR_PROP_GENERAL;

    *frame_number = (uint64_t) last_frame;
    *buffer = Fg_getImagePtrEx(me4->fg, last_frame, PORT_A, me4->mem);
    return UCA_NO_ERROR;
}

static int uca_me4_callback(frameindex_t frame, struct fg_apc_data *apc)
{
    apc->callback(frame, Fg_getImagePtrEx(apc->fg, frame, PORT_A, apc->mem), apc->meta_data, apc->user);
    return 0;
}

uint32_t uca_me4_register_callback(struct uca_grabber *grabber, uca_cam_grab_callback callback, void *meta_data, void *user)
{
    if (grabber->callback == NULL) {
        grabber->callback = callback;

        struct fg_apc_data *apc_data = (struct fg_apc_data *) grabber->user;
        apc_data->callback = callback;
        apc_data->meta_data = meta_data;
        apc_data->user = user;

        struct FgApcControl ctrl;
        ctrl.version = 0;
        ctrl.data = (struct fg_apc_data *) grabber->user;
        ctrl.func = &uca_me4_callback;
        ctrl.flags = FG_APC_DEFAULTS;
        ctrl.timeout = 1;

        if (Fg_registerApcHandler(GET_FG(grabber), PORT_A, &ctrl, FG_APC_CONTROL_BASIC) != FG_OK)
            return UCA_ERR_GRABBER_CALLBACK_REGISTRATION_FAILED;
    }
    else
        return UCA_ERR_GRABBER_CALLBACK_ALREADY_REGISTERED;

    return UCA_NO_ERROR;
}

uint32_t uca_me4_init(struct uca_grabber **grabber)
{
    Fg_Struct *fg = Fg_Init("libFullAreaGray8.so", 0);
    if (fg == NULL)
        return UCA_ERR_GRABBER_NOT_FOUND;

    struct uca_grabber *uca = (struct uca_grabber *) malloc(sizeof(struct uca_grabber));
    memset(uca, 0, sizeof(struct uca_grabber));

    struct fg_apc_data *me4 = (struct fg_apc_data *) malloc(sizeof(struct fg_apc_data));
    memset(me4, 0, sizeof(struct fg_apc_data));
    me4->fg  = fg;

    Fg_getParameter(fg, FG_TIMEOUT, &me4->timeout, PORT_A);

    uca->user = me4;
    uca->destroy = &uca_me4_destroy;
    uca->set_property = &uca_me4_set_property;
    uca->get_property = &uca_me4_get_property;
    uca->alloc = &uca_me4_alloc;
    uca->acquire = &uca_me4_acquire;
    uca->stop_acquire = &uca_me4_stop_acquire;
    uca->grab = &uca_me4_grab;
    uca->register_callback = &uca_me4_register_callback;
    
    *grabber = uca;
    return UCA_NO_ERROR;
}