graphics.c

/*
 * Copyright (C) 2007 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include <stdlib.h>
#include <unistd.h>

#include <fcntl.h>
#include <stdio.h>

#include <sys/ioctl.h>
#include <sys/mman.h>
#include <sys/types.h>

#include <linux/fb.h>
#include <linux/kd.h>

#include <pixelflinger/pixelflinger.h>

#include "font_10x18.h"
#include "minui.h"

typedef struct {
    GGLSurface texture;
    unsigned cwidth;
    unsigned cheight;
    unsigned ascent;
} GRFont;

static GRFont *gr_font = 0;
static GGLContext *gr_context = 0;
static GGLSurface gr_font_texture;
static GGLSurface gr_framebuffer[2];
static GGLSurface gr_mem_surface;
static unsigned gr_active_fb = 0;

static int gr_fb_fd = -1;
static int gr_vt_fd = -1;

static struct fb_var_screeninfo vi;

static int get_framebuffer(GGLSurface *fb)
{
    int fd;
    struct fb_fix_screeninfo fi;
    void *bits;

    fd = open("/dev/graphics/fb0", O_RDWR);
    if (fd < 0) {
        perror("cannot open fb0");
        return -1;
    }

    if (ioctl(fd, FBIOGET_FSCREENINFO, &fi) < 0) {
        perror("failed to get fb0 info");
        close(fd);
        return -1;
    }

    if (ioctl(fd, FBIOGET_VSCREENINFO, &vi) < 0) {
        perror("failed to get fb0 info");
        close(fd);
        return -1;
    }

    bits = mmap(0, fi.smem_len, PROT_READ | PROT_WRITE, MAP_SHARED, fd, 0);
    if (bits == MAP_FAILED) {
        perror("failed to mmap framebuffer");
        close(fd);
        return -1;
    }
    printf("%d\n",vi.yres_virtual);

    fb->version = sizeof(*fb);
    fb->width = vi.xres;
    fb->height = vi.yres;
    fb->stride = vi.xres;
    fb->data = bits;
#if BYTES_PER_PIXEL == 4
    fb->format = GGL_PIXEL_FORMAT_RGBA_8888;
#else
    fb->format = GGL_PIXEL_FORMAT_RGB_565;
#endif

    fb++;

    fb->version = sizeof(*fb);
    fb->width = vi.xres;
    fb->height = vi.yres;
    fb->stride = vi.xres;
    fb->data = (void*) (((unsigned) bits) + vi.yres * vi.xres * BYTES_PER_PIXEL);
#if BYTES_PER_PIXEL == 4
    fb->format = GGL_PIXEL_FORMAT_RGBA_8888;
#else
    fb->format = GGL_PIXEL_FORMAT_RGB_565;
#endif

    return fd;
}

static void get_memory_surface(GGLSurface* ms) {
  ms->version = sizeof(*ms);
  ms->width = vi.xres;
  ms->height = vi.yres;
  ms->stride = vi.xres;
  ms->data = malloc(vi.xres * vi.yres * BYTES_PER_PIXEL);
#if BYTES_PER_PIXEL == 4
    ms->format = GGL_PIXEL_FORMAT_RGBA_8888;
#else
    ms->format = GGL_PIXEL_FORMAT_RGB_565;
#endif
}

static void set_active_framebuffer(unsigned n)
{
    if (n > 1) return;
//    vi.yres_virtual = vi.yres * BYTES_PER_PIXEL;
    vi.yres_virtual = vi.yres * 2;
    vi.yoffset = n * vi.yres;
    vi.bits_per_pixel = BYTES_PER_PIXEL * 8;
    if (ioctl(gr_fb_fd, FBIOPUT_VSCREENINFO, &vi) < 0) {
        perror("active fb swap failed");
    }
}

// switch to the first fb
void gr_flip_zero()
{
    GGLContext *gl = gr_context;

    /* swap front and back buffers */
    gr_active_fb = 0;

    /* copy data from the in-memory surface to the buffer we're about
     * to make active. */
    memcpy(gr_framebuffer[gr_active_fb].data, gr_mem_surface.data,
           vi.xres * vi.yres * BYTES_PER_PIXEL);

    /* inform the display driver */
    set_active_framebuffer(gr_active_fb);
}

void gr_flip(void)
{
    GGLContext *gl = gr_context;

    /* swap front and back buffers */
    gr_active_fb = (gr_active_fb + 1) & 1;

    /* copy data from the in-memory surface to the buffer we're about
     * to make active. */
    memcpy(gr_framebuffer[gr_active_fb].data, gr_mem_surface.data,
           vi.xres * vi.yres * BYTES_PER_PIXEL);

    /* inform the display driver */
    set_active_framebuffer(gr_active_fb);
}

void gr_color(unsigned char r, unsigned char g, unsigned char b, unsigned char a)
{
    GGLContext *gl = gr_context;
    GGLint color[4];
    color[0] = ((r << 8) | r) + 1;
    color[1] = ((g << 8) | g) + 1;
    color[2] = ((b << 8) | b) + 1;
    color[3] = ((a << 8) | a) + 1;
    gl->color4xv(gl, color);
}

int gr_measure(const char *s)
{
    return gr_font->cwidth * strlen(s);
}

int gr_text_size(const char *s, int *width, int *height)
{
  int maxwidth = 0;
  int maxheight = 0;
  char* str = s;
  while (str && *str) {
    if (str!=s) str++;
    maxheight++;
    int pos = strcspn(str,"\n");
    if (pos>maxwidth) {
      maxwidth = pos;
    }
    str = str+pos;
  }
  *width = maxwidth * gr_font->cwidth;
  *height = maxheight * gr_font->cheight;
  return 1;
}

// ax and ay are from -1..1 to detemine alignment of text. -1: right/bottom, 0: center, 1: left/top
// baseline: whether y is for the baseline or the top of the line
int gr_text_align(int x, int y, int ax, int ay, int baseline, const char *s)
{
  int width;
  int height;
  int fl;
  char *str;
  gr_text_size(s, &width, &height);
  if (ay==GR_TEXT_ALIGN_BOTTOM) y-=height;
  if (ay==GR_TEXT_ALIGN_CENTER) y-=height/2;
  if (!baseline) y+=gr_font->ascent; // this will be removed by gr_text
  str = malloc(strlen(s)+1);
  char *save = str;
  strcpy(str,s);
  fl = 0;
  while (str && *str) {
    if (fl) str++;
    int pos = strcspn(str,"\n");
    int mod = 0; //whether we modified the buffer or not
    if (*(str+pos)) mod = 1;
    *(str+pos) = '\0';

    int px = x;
    if (ax==GR_TEXT_ALIGN_RIGHT) px-=strlen(str)*gr_font->cwidth;
    if (ax==GR_TEXT_ALIGN_CENTER) px-=strlen(str)*gr_font->cwidth/2;

    gr_text(px,y,str);

    if (mod) *(str+pos) = '\n'; //if we did, restore the original
    str = str+pos;
    y += gr_font->cheight;
    fl = 1;
  }
  free(save);
  return 1;
}

int gr_text(int x, int y, const char *s)
{
    GGLContext *gl = gr_context;
    GRFont *font = gr_font;
    unsigned off;

    y -= font->ascent;

    // don't run this if it won't shown anyway
    if (y+((int)font->ascent)<0) return 0;
    if (y-((int)font->ascent)>gr_fb_height()) return 0;

    gl->bindTexture(gl, &font->texture);
    gl->texEnvi(gl, GGL_TEXTURE_ENV, GGL_TEXTURE_ENV_MODE, GGL_REPLACE);
    gl->texGeni(gl, GGL_S, GGL_TEXTURE_GEN_MODE, GGL_ONE_TO_ONE);
    gl->texGeni(gl, GGL_T, GGL_TEXTURE_GEN_MODE, GGL_ONE_TO_ONE);
    gl->enable(gl, GGL_TEXTURE_2D);

    while((off = *s++)) {
        off -= 32;
        if (off < 96) {
            gl->texCoord2i(gl, (off * font->cwidth) - x, 0 - y);
            gl->recti(gl, x, y, x + font->cwidth, y + font->cheight);
        }
        x += font->cwidth;
    }

    return x;
}

void gr_fill(int x, int y, int w, int h)
{
    GGLContext *gl = gr_context;
    gl->disable(gl, GGL_TEXTURE_2D);
    gl->recti(gl, x, y, w, h);
}

void gr_line(int x, int y, int x2, int y2, int w)
{
  GGLContext *gl = gr_context;
  gl->disable(gl, GGL_TEXTURE_2D);
  // GGLcoord contains Q28.4 numbers
  GGLcoord pos1[2];
  GGLcoord pos2[2];
  pos1[0] = x<<4;
  pos1[1] = y<<4;
  pos2[0] = x2<<4;
  pos2[1] = y2<<4;
  w = w<<4;
  gl->linex(gl, pos1, pos2, w);
}

void gr_point(int x, int y, int w)
{
  GGLContext *gl = gr_context;
  gl->disable(gl, GGL_TEXTURE_2D);
  // without this, pixelflinger draws a rectangle
  gl->enable(gl, GGL_AA);
  // GGLcoord contains Q28.4 numbers
  GGLcoord pos[2];
  pos[0] = x<<4;
  pos[1] = y<<4;
  w = w<<4;
  gl->pointx(gl, pos, w);
  gl->disable(gl, GGL_AA);
}

void gr_blend(int enable) {
  GGLContext *gl = gr_context;
  if (enable) {
    gl->enable(gl, GGL_BLEND);
    gl->blendFunc(gl, GGL_SRC_ALPHA, GGL_ONE_MINUS_SRC_ALPHA);
  } else {
    gl->disable(gl, GGL_BLEND);
  }
}

void gr_blit(gr_surface source, int sx, int sy, int w, int h, int dx, int dy) {
    if (gr_context == NULL) {
        return;
    }
    GGLContext *gl = gr_context;

    gl->bindTexture(gl, (GGLSurface*) source);
    gl->texEnvi(gl, GGL_TEXTURE_ENV, GGL_TEXTURE_ENV_MODE, GGL_REPLACE);
    gl->texGeni(gl, GGL_S, GGL_TEXTURE_GEN_MODE, GGL_ONE_TO_ONE);
    gl->texGeni(gl, GGL_T, GGL_TEXTURE_GEN_MODE, GGL_ONE_TO_ONE);
    gl->enable(gl, GGL_TEXTURE_2D);
    gl->texCoord2i(gl, sx - dx, sy - dy);
    gl->recti(gl, dx, dy, dx + w, dy + h);
}

// create a new surface the size of the frame buffer
gr_surface gr_create_surface(void) {
    GGLSurface* s = malloc(sizeof(GGLSurface));
    get_memory_surface(s);
    return (gr_surface *)s;
}

// free the previous surface
void gr_free_surface(gr_surface surface) {
  if (surface) {
    free(((GGLSurface*)surface)->data);
    free(surface);
  }
}

// save the active surface data to this surface
void gr_save_active_surface(gr_surface surface) {
  memcpy(((GGLSurface*)surface)->data, gr_mem_surface.data, vi.xres * vi.yres * BYTES_PER_PIXEL);
}

unsigned int gr_get_width(gr_surface surface) {
    if (surface == NULL) {
        return 0;
    }
    return ((GGLSurface*) surface)->width;
}

unsigned int gr_get_height(gr_surface surface) {
    if (surface == NULL) {
        return 0;
    }
    return ((GGLSurface*) surface)->height;
}

static void gr_init_font(void)
{
    GGLSurface *ftex;
    unsigned char *bits, *rle;
    unsigned char *in, data;

    gr_font = calloc(sizeof(*gr_font), 1);
    ftex = &gr_font->texture;

    bits = malloc(font.width * font.height);

    ftex->version = sizeof(*ftex);
    ftex->width = font.width;
    ftex->height = font.height;
    ftex->stride = font.width;
    ftex->data = (void*) bits;
    ftex->format = GGL_PIXEL_FORMAT_A_8;

    in = font.rundata;
    while((data = *in++)) {
        memset(bits, (data & 0x80) ? 255 : 0, data & 0x7f);
        bits += (data & 0x7f);
    }

    gr_font->cwidth = font.cwidth;
    gr_font->cheight = font.cheight;
    gr_font->ascent = font.cheight - 2;
}

int gr_init(void)
{
    gglInit(&gr_context);
    GGLContext *gl = gr_context;

    gr_init_font();
    gr_vt_fd = open("/dev/tty0", O_RDWR | O_SYNC);
    if (gr_vt_fd < 0) {
        // This is non-fatal; post-Cupcake kernels don't have tty0.
        perror("can't open /dev/tty0");
    } else if (ioctl(gr_vt_fd, KDSETMODE, (void*) KD_GRAPHICS)) {
        // However, if we do open tty0, we expect the ioctl to work.
        perror("failed KDSETMODE to KD_GRAPHICS on tty0");
        gr_exit();
        return -1;
    }

    gr_fb_fd = get_framebuffer(gr_framebuffer);
    if (gr_fb_fd < 0) {
        gr_exit();
        return -1;
    }

    get_memory_surface(&gr_mem_surface);

    fprintf(stderr, "framebuffer: fd %d (%d x %d)\n",
            gr_fb_fd, gr_framebuffer[0].width, gr_framebuffer[0].height);

        /* start with 0 as front (displayed) and 1 as back (drawing) */
    gr_active_fb = 0;
    set_active_framebuffer(0);
    gl->colorBuffer(gl, &gr_mem_surface);


    gl->activeTexture(gl, 0);
    gl->enable(gl, GGL_BLEND);
    gl->blendFunc(gl, GGL_SRC_ALPHA, GGL_ONE_MINUS_SRC_ALPHA);

    return 0;
}

void gr_crop(int x, int y, int x2, int y2)
{
    GGLContext *gl = gr_context;
    if ((x == y) && (x2 == y2) && (x == x2) && (x == 0)) {
      gl->disable(gl, GGL_SCISSOR_TEST);
      gl->scissor(gl,0,0,gr_fb_width(),gr_fb_height());
    } else {
      gl->enable(gl, GGL_SCISSOR_TEST);
      gl->scissor(gl,x,y,x2-x,y2-y);
    }
}

void gr_exit(void)
{
    gr_flip_zero();
    close(gr_fb_fd);
    gr_fb_fd = -1;

    free(gr_mem_surface.data);

    ioctl(gr_vt_fd, KDSETMODE, (void*) KD_TEXT);
    close(gr_vt_fd);
    gr_vt_fd = -1;
}

int gr_fb_width(void)
{
    return gr_framebuffer[0].width;
}

int gr_fb_height(void)
{
    return gr_framebuffer[0].height;
}

gr_pixel *gr_fb_data(void)
{
    return (unsigned short *) gr_mem_surface.data;
}