1/*-
2 * Copyright (c) 2015 Conrad Meyer <cse.cem@gmail.com>
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 *
14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24 * SUCH DAMAGE.
25 */
26
27#include <sys/cdefs.h>
28__FBSDID("$FreeBSD$");
29
30#include <sys/param.h>
31#include <sys/callout.h>
32#include <sys/cons.h>
33#include <sys/kernel.h>
34#include <sys/smp.h>
35#include <sys/systm.h>
36#include <sys/terminal.h>
37
38#include <dev/vt/vt.h>
39
40extern const unsigned char vt_beastie_vga16[];
41extern const unsigned char vt_beastie2_vga16[];
42extern const unsigned char vt_orb_vga16[];
43
44static struct callout vt_splash_cpu_callout;
45
46static inline unsigned char
47vt_vga2bsd(unsigned char vga)
48{
49	static const unsigned char lut[8] = {
50		0,
51		4,	/* 1 and 4 swap */
52		2,
53		6,	/* 3 and 6 swap */
54		1,	/* 4 and 1 swap */
55		5,
56		3,	/* 6 and 3 swap */
57		7,
58	};
59	unsigned int bright;
60
61	bright = (vga & 0x8);
62	return (lut[vga & 0x7] | bright);
63}
64
65static void
66vt_draw_2_vga16_px(struct vt_device *vd, vt_axis_t x, vt_axis_t y,
67    unsigned char color)
68{
69
70	vd->vd_driver->vd_setpixel(vd, x, y, vt_vga2bsd(color >> 4));
71	vd->vd_driver->vd_setpixel(vd, x + 1, y, vt_vga2bsd(color & 0xf));
72}
73
74static void
75vt_draw_1_logo(struct vt_device *vd, vt_axis_t top, vt_axis_t left)
76{
77	const unsigned char rle_sent = 0x16, *data;
78	unsigned int xy, run, runcolor, i;
79
80	switch (vt_splash_cpu_style) {
81	case VT_LOGOS_DRAW_ALT_BEASTIE:
82		data = vt_beastie2_vga16;
83		break;
84	case VT_LOGOS_DRAW_ORB:
85		data = vt_orb_vga16;
86		break;
87	case VT_LOGOS_DRAW_BEASTIE:
88		/* FALLTHROUGH */
89	default:
90		data = vt_beastie_vga16;
91		break;
92	}
93
94	/* Decode basic RLE (gets us to 30-40% of uncompressed data size): */
95	for (i = 0, xy = 0; xy < vt_logo_sprite_height * vt_logo_sprite_width;) {
96		if (data[i] == rle_sent) {
97			runcolor = data[i + 1];
98			run = data[i + 2];
99
100			for (; run; run--, xy += 2)
101				vt_draw_2_vga16_px(vd,
102				    left + (xy % vt_logo_sprite_width),
103				    top + (xy / vt_logo_sprite_width),
104				    runcolor);
105
106			i += 3;
107		} else {
108			vt_draw_2_vga16_px(vd, left + (xy % vt_logo_sprite_width),
109			    top + (xy / vt_logo_sprite_width), data[i]);
110
111			i++;
112			xy += 2;
113		}
114	}
115}
116
117void
118vtterm_draw_cpu_logos(struct vt_device *vd)
119{
120	unsigned int ncpu, i;
121	vt_axis_t left;
122
123	if (vt_splash_ncpu)
124		ncpu = vt_splash_ncpu;
125	else {
126		ncpu = mp_ncpus;
127		if (ncpu < 1)
128			ncpu = 1;
129	}
130
131	if (vd->vd_driver->vd_drawrect)
132		vd->vd_driver->vd_drawrect(vd, 0, 0, vd->vd_width,
133		    vt_logo_sprite_height, 1, TC_BLACK);
134	/*
135	 * Blank is okay because we only ever draw beasties on full screen
136	 * refreshes.
137	 */
138	else if (vd->vd_driver->vd_blank)
139		vd->vd_driver->vd_blank(vd, TC_BLACK);
140
141	ncpu = MIN(ncpu, vd->vd_width / vt_logo_sprite_width);
142	for (i = 0, left = 0; i < ncpu; left += vt_logo_sprite_width, i++)
143		vt_draw_1_logo(vd, 0, left);
144}
145
146static void
147vt_fini_logos(void *dummy __unused)
148{
149	struct vt_device *vd;
150	struct vt_window *vw;
151	struct terminal *tm;
152	struct vt_font *vf;
153	struct winsize wsz;
154	term_pos_t size;
155	unsigned int i;
156
157	if (!vt_draw_logo_cpus)
158		return;
159	if (!vty_enabled(VTY_VT))
160		return;
161	if (!vt_splash_cpu)
162		return;
163
164	vd = &vt_consdev;
165	VT_LOCK(vd);
166	if ((vd->vd_flags & (VDF_DEAD | VDF_TEXTMODE)) != 0) {
167		VT_UNLOCK(vd);
168		return;
169	}
170	vt_draw_logo_cpus = 0;
171	VT_UNLOCK(vd);
172
173	for (i = 0; i < VT_MAXWINDOWS; i++) {
174		vw = vd->vd_windows[i];
175		if (vw == NULL)
176			continue;
177		tm = vw->vw_terminal;
178		vf = vw->vw_font;
179		if (vf == NULL)
180			continue;
181
182		vt_termsize(vd, vf, &size);
183		vt_winsize(vd, vf, &wsz);
184
185		/* Resize screen buffer and terminal. */
186		terminal_mute(tm, 1);
187		vtbuf_grow(&vw->vw_buf, &size, vw->vw_buf.vb_history_size);
188		terminal_set_winsize_blank(tm, &wsz, 0, NULL);
189		terminal_set_cursor(tm, &vw->vw_buf.vb_cursor);
190		terminal_mute(tm, 0);
191
192		VT_LOCK(vd);
193		vt_compute_drawable_area(vw);
194
195		if (vd->vd_curwindow == vw) {
196			vd->vd_flags |= VDF_INVALID;
197			vt_resume_flush_timer(vw, 0);
198		}
199		VT_UNLOCK(vd);
200	}
201}
202
203static void
204vt_init_logos(void *dummy)
205{
206	struct vt_device *vd;
207	struct vt_window *vw;
208	struct terminal *tm;
209	struct vt_font *vf;
210	struct winsize wsz;
211	term_pos_t size;
212
213	if (!vty_enabled(VTY_VT))
214		return;
215	if (!vt_splash_cpu)
216		return;
217
218	tm = &vt_consterm;
219	vw = tm->tm_softc;
220	if (vw == NULL)
221		return;
222	vd = vw->vw_device;
223	if (vd == NULL)
224		return;
225	vf = vw->vw_font;
226	if (vf == NULL)
227		return;
228
229	VT_LOCK(vd);
230	if ((vd->vd_flags & VDF_INITIALIZED) == 0)
231		goto out;
232	if ((vd->vd_flags & (VDF_DEAD | VDF_TEXTMODE)) != 0)
233		goto out;
234	if (vd->vd_height <= vt_logo_sprite_height)
235		goto out;
236
237	vt_draw_logo_cpus = 1;
238	VT_UNLOCK(vd);
239
240	vt_termsize(vd, vf, &size);
241	vt_winsize(vd, vf, &wsz);
242
243	/* Resize screen buffer and terminal. */
244	terminal_mute(tm, 1);
245	vtbuf_grow(&vw->vw_buf, &size, vw->vw_buf.vb_history_size);
246	terminal_set_winsize_blank(tm, &wsz, 0, NULL);
247	terminal_set_cursor(tm, &vw->vw_buf.vb_cursor);
248	terminal_mute(tm, 0);
249
250	VT_LOCK(vd);
251	vt_compute_drawable_area(vw);
252
253	if (vd->vd_curwindow == vw) {
254		vd->vd_flags |= VDF_INVALID;
255		vt_resume_flush_timer(vw, 0);
256	}
257
258	callout_init(&vt_splash_cpu_callout, 1);
259	callout_reset(&vt_splash_cpu_callout, vt_splash_cpu_duration * hz,
260	    vt_fini_logos, NULL);
261
262out:
263	VT_UNLOCK(vd);
264}
265SYSINIT(vt_logos, SI_SUB_CPU + 1, SI_ORDER_ANY, vt_init_logos, NULL);
266