libetherpix

1) /*

2)  * EtherPix library

3)  *

4)  * Copyright 2010-2017 Stefan Schuermans <stefan schuermans info>

5)  *
6)  * This program is free software: you can redistribute it and/or modify
7)  * it under the terms of the GNU General Public License as published by
8)  * the Free Software Foundation, version 3 of the License.
9)  *
10)  *
11)  * This program is distributed in the hope that it will be useful,
12)  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13)  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14)  * GNU General Public License for more details.
15)  *
16)  * You should have received a copy of the GNU Lesser General Public License
17)  * along with this program. If not, see <http://www.gnu.org/licenses/>.
18)  */
19) 
20) #include <errno.h>
21) #include <stdlib.h>
22) #include <string.h>
23) 

24) #include <etherpix/display.h>
25) #include <etherpix/msg.h>
26) #include <etherpix/types.h>

27) #include "config.h"
28) #include "constants.h"
29) #include "net.h"
30) #include "types.h"

31) 
32) /**

33)  * \brief create a new EtherPix display

34)  *
35)  * \param[in] sz_config_file name of config file to read
36)  * \param[in] p_msg_func message callback function or NULL
37)  * \param[in] p_msg_ctx user context for message callback

38)  * \return pointer to new EtherPix display on success

39)  *         or NULL on error
40)  */

41) etp_display_t *etp_display_create(const char *sz_config_file,
42)                                   etp_msg_func_p_t p_msg_func,

43)                                   void *p_msg_ctx)
44) {

45)   etp_display_t *p_display;

46) 
47)   /* basic display setup */
48) 
49)   /* create display structure */

50)   p_display = (etp_display_t *)calloc(1, sizeof (etp_display_t));

51)   if (!p_display) {
52)     if (p_msg_func)

53)       p_msg_func(p_msg_ctx, etp_msg_type_err,

54)                  "out of memory\n");
55)     return NULL;
56)   }
57) 
58)   /* set default bind address */
59)   p_display->bind_addr.sin_family = AF_INET;

60)   p_display->bind_addr.sin_port = htons(ETP_BIND_PORT);
61)   p_display->bind_addr.sin_addr.s_addr = htonl(ETP_BIND_IP);

62) 
63)   /* no socket yet */
64)   p_display->sock = -1;
65) 
66)   /* config file */
67) 
68)   /* read config file */

69)   if(etp_config_proc_file(p_display, sz_config_file,

70)                           p_msg_func, p_msg_ctx)) {

71)     etp_display_free(p_display); /* cleanup everything that config file

72)                                     processing might heave created already */

73)     if (p_msg_func)

74)       p_msg_func(p_msg_ctx, etp_msg_type_err, "reading config file failed\n");

75)     return NULL;
76)   }
77) 
78)   /* set up UDP socket */
79) 
80)   /* create socket */

81) #ifdef WINDOWS
82)   p_display->sock = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
83)   if (p_display->sock != INVALID_SOCKET) {
84)     u_long arg = 1;
85)     if (ioctlsocket(p_display->sock, FIONBIO, &arg) != 0) {
86)       closesocket(p_display->sock);
87)       p_display->sock = INVALID_SOCKET;
88)     }
89)   }
90) #else

91)   p_display->sock = socket(AF_INET, SOCK_DGRAM | SOCK_NONBLOCK, IPPROTO_UDP);

92) #endif

93)   if (!etp_sock_is_valid(p_display->sock)) {
94)     etp_display_free(p_display);

95)     if (p_msg_func) {
96)       char errmsg[256];

97)       etp_sock_get_last_error(errmsg, sizeof(errmsg));
98)       p_msg_func(p_msg_ctx, etp_msg_type_err,

99)                  "could not create UDP socket: %s\n", errmsg);
100)     }

101)     return NULL;
102)   }
103) 
104)   /* bind socket */
105)   if (bind(p_display->sock, (struct sockaddr *)&p_display->bind_addr,
106)            sizeof (p_display->bind_addr))) {

107)     etp_display_free(p_display);

108)     if (p_msg_func) {
109)       char errmsg[256];

110)       etp_sock_get_last_error(errmsg, sizeof(errmsg));
111)       p_msg_func(p_msg_ctx, etp_msg_type_err,

112)                  "could not bind UDP socket to \"%s:%u\": %s\n",
113)                  inet_ntoa(p_display->bind_addr.sin_addr),

114)                  (unsigned int)ntohs(p_display->bind_addr.sin_port), errmsg);
115)     }

116)     return NULL;
117)   }
118) 
119)   /* clear display */
120) 

121)   etp_display_data_clear(p_display);

122) 
123)   return p_display;
124) }
125) 
126) /**

127)  * \brief free a EtherPix display

128)  *

129)  * \param[in] p_display pointer to EtherPix display

130)  */

131) void etp_display_free(etp_display_t *p_display)

132) {
133)   int i;
134) 
135)   if (p_display) {
136) 
137)     /* free distributors */

138)     for (i = 0; i < ETP_DISTRI_MAX_CNT; i++) {

139)       if (p_display->distri_ptrs[i]) {

140)         etp_distri_t *p_distri = p_display->distri_ptrs[i];

141)         free(p_distri->p_pixels);
142)         free(p_distri->p_msg_buf);
143)         free(p_distri);
144)       }
145)     }
146) 
147)     /* close network connection */

148)     if (etp_sock_is_valid(p_display->sock))

149)       closesocket(p_display->sock);
150) 
151)     /* free display structutre */
152)     free(p_display);
153) 
154)   } /* if (p_display) */
155) }
156) 
157) /**
158)  * \brief get size of display
159)  *

160)  * \param[in] p_display pointer to EtherPix display

161)  * \param[out] p_width width of display
162)  * \param[out] p_height height of display
163)  */

164) void etp_display_get_size(etp_display_t *p_display,

165)                           unsigned int *p_width, unsigned int *p_height)
166) {
167)   *p_width = (unsigned int)p_display->size.x;
168)   *p_height = (unsigned int)p_display->size.y;
169) }
170) 
171) /**

172)  * \brief clear image data to output on EtherPix display

173)  *
174)  * clears the stored image data,
175)  * stored image data can later be sent to the distributors using

176)  * etp_display_send()

177)  *

178)  * \param[in] p_display pointer to EtherPix display

179)  */

180) void etp_display_data_clear(etp_display_t *p_display)

181) {

182)   etp_u8_t black[3] = { 0, 0, 0 };
183)   etp_display_data_fmt(p_display, black, 0, 0,

184)                        0, 0, p_display->size.x, p_display->size.y,

185)                        etp_pixfmt_rgb24);

186) }
187) 
188) /**

189)  * \brief set image data to output on EtherPix display

190)  *
191)  * updates (part of) the stored image data,
192)  * stored image data can later be sent to the distributors using

193)  * etp_display_send()

194)  *

195)  * \param[in] p_display pointer to EtherPix display

196)  * \param[in] p_data pointer to rectangular section of image data

197)  * \param[in] stride_x stride between two pixels in X direction
198)  * \param[in] stride_y stride between two pixels in Y direction
199)  * \param[in] x X coordinate of left side of rectangular area
200)  * \param[in] y Y coordinate of top side of rectangular area
201)  * \param[in] width with of rectangular area
202)  * \param[in] height height of rectangular area

203)  * \param[in] pixfmt format of pixel data

204)  */

205) void etp_display_data_fmt(etp_display_t *p_display, etp_u8_t *p_data,

206)                           int stride_x, int stride_y,
207)                           unsigned int x, unsigned int y,
208)                           unsigned int width, unsigned int height,

209)                           etp_pixfmt_t pixfmt)

210) {

211)   unsigned int distri, out, pix, i;

212)   etp_distri_t *p_distri;
213)   etp_u8_t *dest, *src;
214)   etp_pixel_t *p_pixel;

215)   int rx, ry;
216) 
217)   /* set data for all distributors */

218)   for (distri = 0; distri < ETP_DISTRI_MAX_CNT; distri++) {

219)     p_distri = p_display->distri_ptrs[distri];
220)     if (p_distri) {
221) 
222)       /* set pointer to start of RGB data for pixels in message buffer
223)          (header is already set up and stays the same) */

224)       dest = p_distri->p_msg_buf + ETP_MCUF_HDR_LEN;

225) 
226)       /* get RGB data of pixels for every output */
227)       for (out = 0, i = 0; out < p_distri->output_cnt; out++) {
228)         for (pix = 0; pix < p_distri->pixel_cnt; pix++, i++) {
229)           p_pixel = &p_distri->p_pixels[i];
230) 
231)           /* get pixel coordinates relative to rectangular area of image */
232)           rx = p_pixel->x - (int)x;
233)           ry = p_pixel->y - (int)y;
234)           /* check if pixel is within rectangular area of image */
235)           if (rx >= 0 && (unsigned int)rx < width
236)               && ry >= 0 && (unsigned int)ry < height) {
237)             /* get pixel data and map it */
238)             src = (p_data + rx * stride_x + ry * stride_y);

239)             switch (pixfmt) {

240)               case etp_pixfmt_rgb24:

241)                 dest[0] = p_distri->mapping[0].table[src[0]];
242)                 dest[1] = p_distri->mapping[1].table[src[1]];
243)                 dest[2] = p_distri->mapping[2].table[src[2]];
244)                 break;

245)               case etp_pixfmt_bgr24:

246)                 dest[0] = p_distri->mapping[0].table[src[2]];
247)                 dest[1] = p_distri->mapping[1].table[src[1]];
248)                 dest[2] = p_distri->mapping[2].table[src[0]];
249)                 break;
250)               default:
251)                 dest[0] = 0;
252)                 dest[1] = 0;
253)                 dest[2] = 0;
254)             }

255)           }
256) 
257)           dest += 3;
258)         } /* for (pix ...) */
259)       } /* for (out ...) */
260) 
261)     } /* if (p_distri) */
262)   } /* for (distri ...) */
263) }
264) 

265) /**

266)  * \brief set image data to output on EtherPix display

267)  *

268)  * see etp_display_data_fmt for documentation
269)  * pixfmt is fixed to etp_pixfmt_rgb24

270)  */

271) void etp_display_data(etp_display_t *p_display, etp_u8_t *p_data,

272)                       int stride_x, int stride_y,
273)                       unsigned int x, unsigned int y,
274)                       unsigned int width, unsigned int height)
275) {

276)   etp_display_data_fmt(p_display, p_data, stride_x, stride_y,
277)                        x, y, width, height, etp_pixfmt_rgb24);

278) }
279) 

280) /**
281)  * \brief send image data to distributors
282)  *
283)  * sends the currently stored image data to all configured distributors
284)  *

285)  * \param[in] p_display pointer to EtherPix display

286)  */

287) void etp_display_send(etp_display_t *p_display)

288) {

289)   unsigned int distri, addr_i;

290)   etp_distri_t *p_distri;

291) 
292)   /* send data to all distributors */

293)   for (distri = 0; distri < ETP_DISTRI_MAX_CNT; distri++) {

294)     p_distri = p_display->distri_ptrs[distri];
295)     if (p_distri) {
296) 

297)       /* send message as UDP packet to each address */
298)       for (addr_i = 0; addr_i < p_distri->addr_cnt; ++addr_i) {
299)         sendto(p_display->sock,
300)                (char *)p_distri->p_msg_buf, p_distri->msg_len, 0,
301)                (struct sockaddr *)&p_distri->addrs[addr_i],
302)                sizeof (p_distri->addrs[addr_i]));
303)       } /* for (addr_i ...) */