aboutsummaryrefslogtreecommitdiff
path: root/samples/X11/xcb_shm_image/xcb_shm_image_example.c
blob: 4f87e76d3366fd25f29587646e34f64dd7fb063d (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
#include <stdlib.h>
#include <stdio.h>

#include <sys/ipc.h>
#include <sys/shm.h>

#include <xcb/xcb.h>
#include <xcb/shm.h>
#include <xcb/xcb_image.h>

#if __ORDER_LITTLE_ENDIAN == __BYTE_ORDER__
# define NATIVE_XCB_IMAGE_ORDER    XCB_IMAGE_ORDER_LSB_FIRST
#else
# define NATIVE_XCB_IMAGE_ORDER    XCB_IMAGE_ORDER_MSB_FIRST
#endif

enum  {
	IMAGE_WIDTH  = 512,
	IMAGE_HEIGHT = 512,
	IMAGE_DEPTH  = 24,
};

static xcb_format_t const *query_xcb_format_for_depth(
	xcb_connection_t *const connection,
	unsigned depth )
{
	xcb_setup_t const *const setup = xcb_get_setup(connection);
	xcb_format_iterator_t it;
	for( it = xcb_setup_pixmap_formats_iterator(setup)
	   ; it.rem
	   ; xcb_format_next(&it)
	){
		xcb_format_t const *const format = it.data;
		if( format->depth == depth ){
			return format;
		}
	}
	return NULL;
}

typedef struct shm_xcb_image_t {
	xcb_connection_t *connection;
	xcb_image_t  *image;
	xcb_shm_seg_t shm_seg;
	int           shm_id;
} shm_xcb_image_t;

static shm_xcb_image_t *shm_xcb_image_create(
	xcb_connection_t *const connection,
	unsigned const width,
	unsigned const height,
	unsigned const depth )
{
	xcb_generic_error_t *error = NULL;

	shm_xcb_image_t *shmimg = calloc(1, sizeof(*shmimg));
	if( !shmimg ){ goto fail; }
	shmimg->connection = connection;

	xcb_format_t const *const format = query_xcb_format_for_depth(connection, depth);
	shmimg->image = xcb_image_create(
		width, height,
		XCB_IMAGE_FORMAT_Z_PIXMAP,
		format->scanline_pad,
		format->depth, format->bits_per_pixel, 0,
		NATIVE_XCB_IMAGE_ORDER,
		XCB_IMAGE_ORDER_MSB_FIRST,
		NULL, ~0, 0);
	if( !shmimg->image ){
		fprintf(stderr, "could not create X11 image structure\n");
	}
	size_t const image_segment_size = shmimg->image->stride * shmimg->image->height;

	shmimg->shm_id = shmget(IPC_PRIVATE, image_segment_size, IPC_CREAT | 0600);
	if( 0 > shmimg->shm_id ){ goto fail; }

	shmimg->image->data = shmat(shmimg->shm_id, 0, 0);
	if( (void*)-1 == (void*)(shmimg->image->data) ){ goto fail; }

	shmimg->shm_seg = xcb_generate_id(connection),
	error = xcb_request_check(connection,
		xcb_shm_attach_checked(
			connection,
			shmimg->shm_seg, shmimg->shm_id, 0) );
fail:
	if( shmimg ){
		if( shmimg->image ){
			if( shmimg->image->data && error ){
				shmdt(shmimg->image->data);
				shmimg->image->data = NULL;
			}
			if( !shmimg->image->data ){
				shmctl(shmimg->shm_id, IPC_RMID, 0);
				shmimg->shm_id = -1;
			}
		}
		if( 0 > shmimg->shm_id ){
			xcb_image_destroy(shmimg->image);
			shmimg->image = NULL;
		}
		if( !shmimg->image ){
			free(shmimg);
			shmimg = NULL;
		}
	}
	free(error);

	return shmimg;
}

static void shm_xcb_image_destroy(shm_xcb_image_t *shmimg)
{
	xcb_shm_detach(shmimg->connection, shmimg->shm_seg);
	shmdt(shmimg->image->data);
	shmctl(shmimg->shm_id, IPC_RMID, 0);
	xcb_image_destroy(shmimg->image);
	free(shmimg);
}

static void generate_image(
	shm_xcb_image_t *shmimg,
	unsigned t )
{
	for( unsigned j = 0; j < shmimg->image->height; ++j ){
		uint8_t *const line = shmimg->image->data + j * shmimg->image->stride;
		for( unsigned i = 0; i < shmimg->image->width; ++i ){
			unsigned const bytes_per_pixel = shmimg->image->bpp/8;
			uint8_t *pixel = line + i * bytes_per_pixel;

			unsigned const a = (i + t);
			unsigned const b = (j + (i >> 8) & 0xFF);
			unsigned const c = (j >> 8) & 0xFF;

			switch( bytes_per_pixel ){
			case 4: pixel[3] = 0xFF; /* fallthrough */
			case 3: pixel[2] = a & 0xFF; /* fallthrough */
			case 2: pixel[1] = b & 0xFF; /* fallthrough */
			case 1: pixel[0] = c & 0xFF; /* fallthrough */
			default: break;
			}
		}
	}
}

int main(int argc, char *argv[])
{
	/* Open the connection to the X server */
	xcb_connection_t *connection = xcb_connect(NULL, NULL);

	/* Check that X MIT-SHM is available (should be). */
	const xcb_query_extension_reply_t *shm_extension = xcb_get_extension_data(connection, &xcb_shm_id);
	if( !shm_extension || !shm_extension->present ){
		fprintf(stderr, "Query for X MIT-SHM extension failed.\n");
		return 1;
	}

	shm_xcb_image_t *shmimg = shm_xcb_image_create(connection, IMAGE_WIDTH, IMAGE_HEIGHT, IMAGE_DEPTH);
	if( !shmimg ){
		fprintf(stderr, "Creating shared memory image failed");
	}

	/* Get the first screen */
	xcb_screen_t *const screen = xcb_setup_roots_iterator(xcb_get_setup(connection)).data;

	/* Create a window */
	uint32_t const window_mask = XCB_CW_EVENT_MASK;
	uint32_t const window_values[] = { XCB_EVENT_MASK_EXPOSURE};
	xcb_drawable_t const window = xcb_generate_id(connection);
	xcb_create_window(connection,
		XCB_COPY_FROM_PARENT,          /* depth */
		window,                        /* window Id */
		screen->root,                  /* parent window */
		0, 0,                          /* x, y */
		IMAGE_WIDTH, IMAGE_HEIGHT,     /* width, height */
		0,                             /* border_width */
		XCB_WINDOW_CLASS_INPUT_OUTPUT, /* class */
		screen->root_visual,           /* visual */
		window_mask, window_values     /* masks */
	);

	/* Create black (foreground) graphic context */
	xcb_gcontext_t gc = xcb_generate_id( connection );
	uint32_t const gc_mask = XCB_GC_FOREGROUND | XCB_GC_GRAPHICS_EXPOSURES;
	uint32_t const gc_values[] = {screen->black_pixel, 0};
	xcb_create_gc(connection, gc, window, gc_mask, gc_values);

	/* Map the window on the screen and flush*/
	xcb_map_window(connection, window);
	xcb_flush(connection);

	/* Event loop */
	unsigned counter = 0;
	for( xcb_generic_event_t *event
	   ; (event = xcb_wait_for_event(connection))
	   ; free(event)
	){
		switch( event->response_type & ~0x80 ){
		case XCB_EXPOSE:
			generate_image(shmimg, counter++);
			xcb_shm_put_image(connection, window, gc,
				shmimg->image->width, shmimg->image->height, 0, 0,
				shmimg->image->width, shmimg->image->height, 0, 0,
				shmimg->image->depth, shmimg->image->format, 0, 
				shmimg->shm_seg, 0);

			/* flush the request */
			xcb_flush(connection);
			break;
		default:
			/* Unknown event type, ignore it */
			break;
		}

	}

	shm_xcb_image_destroy(shmimg);

	return 0;
}