libetherpix

1) /*

2)  * EtherPix library

3)  *

4)  * Copyright 2010-2011 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 <intern/config.h>
28) #include <intern/constants.h>
29) #include <intern/net.h>
30) #include <intern/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)   p_display->sock = socket(AF_INET, SOCK_DGRAM | SOCK_NONBLOCK, IPPROTO_UDP);

82)   if (!etp_sock_is_valid(p_display->sock)) {
83)     etp_display_free(p_display);

84)     if (p_msg_func) {
85)       char errmsg[256];

86)       etp_sock_get_last_error(errmsg, sizeof(errmsg));
87)       p_msg_func(p_msg_ctx, etp_msg_type_err,

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

90)     return NULL;
91)   }
92) 
93)   /* bind socket */
94)   if (bind(p_display->sock, (struct sockaddr *)&p_display->bind_addr,
95)            sizeof (p_display->bind_addr))) {

96)     etp_display_free(p_display);

97)     if (p_msg_func) {
98)       char errmsg[256];

99)       etp_sock_get_last_error(errmsg, sizeof(errmsg));
100)       p_msg_func(p_msg_ctx, etp_msg_type_err,

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

103)                  (unsigned int)ntohs(p_display->bind_addr.sin_port), errmsg);
104)     }

105)     return NULL;
106)   }
107) 
108)   /* clear display */
109) 

110)   etp_display_data_clear(p_display);

111) 
112)   return p_display;
113) }
114) 
115) /**

116)  * \brief free a EtherPix display

117)  *

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

119)  */

120) void etp_display_free(etp_display_t *p_display)

121) {
122)   int i;
123) 
124)   if (p_display) {
125) 
126)     /* free distributors */

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

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

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

130)         free(p_distri->p_pixels);
131)         free(p_distri->p_msg_buf);
132)         free(p_distri);
133)       }
134)     }
135) 
136)     /* close network connection */

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

138)       closesocket(p_display->sock);
139) 
140)     /* free display structutre */
141)     free(p_display);
142) 
143)   } /* if (p_display) */
144) }
145) 
146) /**
147)  * \brief get size of display
148)  *

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

150)  * \param[out] p_width width of display
151)  * \param[out] p_height height of display
152)  */

153) void etp_display_get_size(etp_display_t *p_display,

154)                           unsigned int *p_width, unsigned int *p_height)
155) {
156)   *p_width = (unsigned int)p_display->size.x;
157)   *p_height = (unsigned int)p_display->size.y;
158) }
159) 
160) /**

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

162)  *
163)  * clears the stored image data,
164)  * stored image data can later be sent to the distributors using

165)  * etp_display_send()

166)  *

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

168)  */

169) void etp_display_data_clear(etp_display_t *p_display)

170) {

171)   etp_u8_t black[3] = { 0, 0, 0 };
172)   etp_display_data_fmt(p_display, black, 0, 0,

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

174)                        etp_pixfmt_rgb24);

175) }
176) 
177) /**

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

179)  *
180)  * updates (part of) the stored image data,
181)  * stored image data can later be sent to the distributors using

182)  * etp_display_send()

183)  *

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

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

186)  * \param[in] stride_x stride between two pixels in X direction
187)  * \param[in] stride_y stride between two pixels in Y direction
188)  * \param[in] x X coordinate of left side of rectangular area
189)  * \param[in] y Y coordinate of top side of rectangular area
190)  * \param[in] width with of rectangular area
191)  * \param[in] height height of rectangular area

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

193)  */

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

195)                           int stride_x, int stride_y,
196)                           unsigned int x, unsigned int y,
197)                           unsigned int width, unsigned int height,

198)                           etp_pixfmt_t pixfmt)

199) {

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

201)   etp_distri_t *p_distri;
202)   etp_u8_t *dest, *src;
203)   etp_pixel_t *p_pixel;

204)   int rx, ry;
205) 
206)   /* set data for all distributors */

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

208)     p_distri = p_display->distri_ptrs[distri];
209)     if (p_distri) {
210) 
211)       /* set pointer to start of RGB data for pixels in message buffer
212)          (header is already set up and stays the same) */

213)       dest = p_distri->p_msg_buf + ETP_MCUF_HDR_LEN;

214) 
215)       /* get RGB data of pixels for every output */
216)       for (out = 0, i = 0; out < p_distri->output_cnt; out++) {
217)         for (pix = 0; pix < p_distri->pixel_cnt; pix++, i++) {
218)           p_pixel = &p_distri->p_pixels[i];
219) 
220)           /* get pixel coordinates relative to rectangular area of image */
221)           rx = p_pixel->x - (int)x;
222)           ry = p_pixel->y - (int)y;
223)           /* check if pixel is within rectangular area of image */
224)           if (rx >= 0 && (unsigned int)rx < width
225)               && ry >= 0 && (unsigned int)ry < height) {
226)             /* get pixel data and map it */
227)             src = (p_data + rx * stride_x + ry * stride_y);

228)             switch (pixfmt) {

229)               case etp_pixfmt_rgb24:

230)                 dest[0] = p_distri->mapping[0].table[src[0]];
231)                 dest[1] = p_distri->mapping[1].table[src[1]];
232)                 dest[2] = p_distri->mapping[2].table[src[2]];
233)                 break;

234)               case etp_pixfmt_bgr24:

235)                 dest[0] = p_distri->mapping[0].table[src[2]];
236)                 dest[1] = p_distri->mapping[1].table[src[1]];
237)                 dest[2] = p_distri->mapping[2].table[src[0]];
238)                 break;
239)               default:
240)                 dest[0] = 0;
241)                 dest[1] = 0;
242)                 dest[2] = 0;
243)             }

244)           }
245) 
246)           dest += 3;
247)         } /* for (pix ...) */
248)       } /* for (out ...) */
249) 
250)     } /* if (p_distri) */
251)   } /* for (distri ...) */
252) }
253) 

254) /**

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

256)  *

257)  * see etp_display_data_fmt for documentation
258)  * pixfmt is fixed to etp_pixfmt_rgb24

259)  */

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

261)                       int stride_x, int stride_y,
262)                       unsigned int x, unsigned int y,
263)                       unsigned int width, unsigned int height)
264) {

265)   etp_display_data_fmt(p_display, p_data, stride_x, stride_y,
266)                        x, y, width, height, etp_pixfmt_rgb24);

267) }
268) 

269) /**
270)  * \brief send image data to distributors
271)  *
272)  * sends the currently stored image data to all configured distributors
273)  *

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

275)  */

276) void etp_display_send(etp_display_t *p_display)

277) {
278)   unsigned int distri;

279)   etp_distri_t *p_distri;

280) 
281)   /* send data to all distributors */

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

283)     p_distri = p_display->distri_ptrs[distri];
284)     if (p_distri) {
285) 
286)       /* send message as UDP packet */

287)       sendto(p_display->sock, (char *)p_distri->p_msg_buf, p_distri->msg_len,

288)              0, (struct sockaddr *)&p_distri->addr, sizeof (p_distri->addr));