initial commit
[dosdemo] / src / dos / gfx.c
1 #ifndef GFX_H_
2 #define GFX_H_
3
4 #include <stdio.h>
5 #include <stdlib.h>
6 #include <string.h>
7 #include <limits.h>
8 #include "vbe.h"
9 #include "dpmi.h"
10 #include "logger.h"
11
12 #define REALPTR(s, o)   (void*)(((uint32_t)(s) << 4) + (uint32_t)(o))
13 #define VBEPTR(x)               REALPTR(((x) & 0xffff0000) >> 16, (x) & 0xffff)
14 #define VBEPTR_SEG(x)   (((x) & 0xffff0000) >> 16)
15 #define VBEPTR_OFF(x)   ((x) & 0xffff)
16
17 #define SAME_BPP(a, b)  \
18         ((a) == (b) || (a) == 16 && (b) == 15 || (a) == 15 && (b) == 16 || (a) == 32 && (b) == 24 || (a) == 24 && (b) == 32)
19
20 static unsigned int make_mask(int sz, int pos);
21
22 static struct vbe_info *vbe_info;
23 static struct vbe_mode_info *mode_info;
24 static int pal_bits = 6;
25
26 void *set_video_mode(int xsz, int ysz, int bpp)
27 {
28         int i;
29         uint16_t *modes, best = 0;
30         unsigned int fbsize;
31
32         /* check for VBE2 support and output some info */
33         if(!vbe_info) {
34                 if(!(vbe_info = vbe_get_info())) {
35                         fprintf(stderr, "VESA BIOS Extensions not available\n");
36                         return 0;
37                 }
38
39                 printlog("VBE Version: %x.%x\n", vbe_info->version >> 8, vbe_info->version & 0xff);
40                 if(vbe_info->version < 0x200) {
41                         fprintf(stderr, "This program requires VBE 2.0 or greater. Try running UniVBE\n");
42                         return 0;
43                 }
44
45                 printlog("Graphics adapter: %s, %s (%s)\n", VBEPTR(vbe_info->oem_vendor_name_ptr),
46                                 VBEPTR(vbe_info->oem_product_name_ptr), VBEPTR(vbe_info->oem_product_rev_ptr));
47                 printlog("Video memory: %dkb\n", vbe_info->total_mem << 6);
48
49                 modes = VBEPTR(vbe_info->vid_mode_ptr);
50         }
51
52         for(i=0; i<1024; i++) { /* impose an upper limit to avoid inf-loops */
53                 if(modes[i] == 0xffff) {
54                         break;  /* reached the end */
55                 }
56
57                 mode_info = vbe_get_mode_info(modes[i] | VBE_MODE_LFB);
58                 if(!mode_info || mode_info->xres != xsz || mode_info->yres != ysz) {
59                         continue;
60                 }
61                 if(SAME_BPP(mode_info->bpp, bpp)) {
62                         best = modes[i];
63                 }
64         }
65
66         if(best) {
67                 mode_info = vbe_get_mode_info(best);
68         } else {
69                 fprintf(stderr, "Requested video mode (%dx%d %dbpp) is unavailable\n", xsz, ysz, bpp);
70                 return 0;
71         }
72
73         if(vbe_set_mode(best | VBE_MODE_LFB) == -1) {
74                 fprintf(stderr, "Failed to set video mode %dx%d %dbpp\n", mode_info->xres, mode_info->yres, mode_info->bpp);
75                 return 0;
76         }
77
78         /* attempt to set 8 bits of color per component in palettized modes */
79         /*if(bpp <= 8) {
80                 pal_bits = vbe_set_palette_bits(8);
81                 printlog("palette bits per color primary: %d\n", pal_bits);
82         }
83         */
84
85         fbsize = xsz * ysz * mode_info->num_img_pages * (bpp / CHAR_BIT);
86         return (void*)dpmi_mmap(mode_info->fb_addr, fbsize);
87 }
88
89 int set_text_mode(void)
90 {
91         vbe_set_mode(0x3);
92         return 0;
93 }
94
95 int get_color_depth(void)
96 {
97         if(!mode_info) {
98                 return -1;
99         }
100         return mode_info->bpp;
101 }
102
103 int get_color_bits(int *rbits, int *gbits, int *bbits)
104 {
105         if(!mode_info) {
106                 return -1;
107         }
108         *rbits = mode_info->rmask_size;
109         *gbits = mode_info->gmask_size;
110         *bbits = mode_info->bmask_size;
111         return 0;
112 }
113
114 int get_color_mask(unsigned int *rmask, unsigned int *gmask, unsigned int *bmask)
115 {
116         if(!mode_info) {
117                 return -1;
118         }
119         *rmask = make_mask(mode_info->rmask_size, mode_info->rpos);
120         *gmask = make_mask(mode_info->gmask_size, mode_info->gpos);
121         *bmask = make_mask(mode_info->bmask_size, mode_info->bpos);
122         return 0;
123 }
124
125 int get_color_shift(int *rshift, int *gshift, int *bshift)
126 {
127         if(!mode_info) {
128                 return -1;
129         }
130         *rshift = mode_info->rpos;
131         *gshift = mode_info->gpos;
132         *bshift = mode_info->bpos;
133         return 0;
134 }
135
136 void set_palette(int idx, int r, int g, int b)
137 {
138         int col[3];
139         col[0] = r;
140         col[1] = g;
141         col[2] = b;
142         vbe_set_palette(idx, col, 1, pal_bits);
143 }
144
145 void wait_vsync(void)
146 {
147         __asm {
148                 mov dx, 0x3da
149         l1:
150                 in al, dx
151                 and al, 0x8
152                 jnz l1
153         l2:
154                 in al, dx
155                 and al, 0x8
156                 jz l2
157         }
158 }
159
160 static unsigned int make_mask(int sz, int pos)
161 {
162         unsigned int i, mask = 0;
163
164         for(i=0; i<sz; i++) {
165                 mask |= 1 << i;
166         }
167         return mask << pos;
168 }
169
170
171 #endif  /* GFX_H_ */