Subversion Repositories planix.SVN

#include <u.h>
#include <libc.h>
#include <draw.h>
#include <keyboard.h>
#include <event.h>

// 2048.
// y increases down, x increases right.

// square board of this size
#define BOARDSIZE 4

// square boarder
#define BORDER 4

typedef struct Tile Tile;

struct Tile {
        int value;
        Image *color;
};

Tile grid[BOARDSIZE][BOARDSIZE];
Tile *grid1 = &grid[0][0];
int score = 0;
int gameover = 0;

Font *bigfont;

Image *colors[13];

Image *
val2col(int val)
{
        int i;

        if(val == 0)
                return colors[0];

        for(i = 32; (val & (1<<i)) == 0; i--)
                ;

        if(i < nelem(colors))
                return colors[i];

        return colors[nelem(colors)-1];
}


void
initcolors(void)
{
        int i;

        int cols[] = {
                0xcdc0b4ff, 0xeee4daff, 0xede0c8ff, 0xf2b179ff, 0xf59563ff,
                0xf67c5fff, 0xf65e36ff, 0xedcf72ff, 0xedcc61ff,
                0xedc850ff, 0xedc53fff, 0xedc22eff, 0x3c3a32ff };

        for(i = 0; i < nelem(cols); i++) {
                colors[i] = allocimage(display, Rect(0, 0, 1, 1), screen->chan, 1, cols[i]);
        }
}

void
initboard(void)
{
        int x, y, x2, y2;

        for(y = 0; y < BOARDSIZE; y++) {
                for(x = 0; x < BOARDSIZE; x++) {
                        grid[y][x].value = 0;
                        grid[y][x].color = val2col(0);
                }
        }

        y = nrand(BOARDSIZE);
        x = nrand(BOARDSIZE);

        grid[y][x].value = 2;
        grid[y][x].color = val2col(2);;

        y2 = nrand(BOARDSIZE);
        x2 = nrand(BOARDSIZE);

        while(y == y2 || x == x2) {
                y2 = nrand(BOARDSIZE);
                x2 = nrand(BOARDSIZE);
        }

        grid[y2][x2].value = 2;
        grid[y2][x2].color = val2col(2);;

        score = 0;
}

void
redraw(void)
{
        Rectangle r;
        Point p, sp;
        int x, y; // grid indexes
        int width, height; // of each tile
        char buf[64];

        Point strp;
        int strwid;

        r = insetrect(screen->r, BORDER);
        sp = r.min;
        r.min.y += font->height;
        p = r.min;
        
        width = Dx(r) / BOARDSIZE;
        height = Dy(r) / BOARDSIZE;

        draw(screen, screen->r, display->black, nil, ZP);

        snprint(buf, sizeof buf, "Score: %d", score);
        string(screen, sp, display->white, ZP, font, buf);

        for(y = 0; y < BOARDSIZE; y++) {
                for(x = 0; x < BOARDSIZE; x++) {
                        r = Rect(p.x + (x * width), p.y + (y * height), p.x + (x * width) + width, p.y + (y * height) + height);
                        r = insetrect(r, BORDER);
                        draw(screen, r, grid[y][x].color, nil, ZP);

                        if(grid[y][x].value == 0)
                                continue;

                        snprint(buf, sizeof buf, "%d", grid[y][x].value);
                        strwid = stringwidth(bigfont, buf);
                        strp = Pt(Dx(r)/2, Dy(r)/2);
                        strp.x -= strwid / 2;
                        strp.y -= font->height / 2;
                        
                        string(screen, addpt(r.min, strp), display->black, ZP, bigfont, buf);
                }
        }
}

void
addrandom(void)
{
        int x, y, nfree;
        Tile **tfree;

        nfree = 0;

        for(x = 0; x < BOARDSIZE; x++) {
                for(y = 0; y < BOARDSIZE; y++) {
                        if(grid[x][y].value == 0)
                                nfree++;
                }
        }

        if(nfree > 0) {
                tfree = mallocz(sizeof(Tile*) * nfree, 1);
                nfree = 0;
                for(x = 0; x < BOARDSIZE; x++) {
                        for(y = 0; y < BOARDSIZE; y++) {
                                if(grid[x][y].value == 0)
                                        tfree[nfree++] = &grid[x][y];
                        }
                }

                // 10% chance of 4.
                if(nrand(10) == 0)
                        x = 4;
                else
                        x = 2;

                nfree = nrand(nfree);

                tfree[nfree]->value = x;
                tfree[nfree]->color = val2col(x);

                free(tfree);
        }
}

void
buildrow(Tile *out[BOARDSIZE], int n, int dir)
{
        int i, start, add;
        add = start = 0;

        switch(dir) {
        case Kright:
                add = 1;
                start = n*BOARDSIZE;
                break;
        case Kleft:
                add = -1;
                start = n*BOARDSIZE + (BOARDSIZE-1);
                break;
        case Kdown:
                add = BOARDSIZE;
                start = n;
                break;
        case Kup:
                add = -BOARDSIZE;
                start = (BOARDSIZE-1)*BOARDSIZE + n;
                break;
        }

        for(i = 0; i < BOARDSIZE; i++) {
                out[i] = &grid1[add*i + start];
        }
}

int
merge(Tile *row[BOARDSIZE])
{
        int i, j, moved;
        Tile nrow[BOARDSIZE];

        for(i = 0; i < BOARDSIZE; i++) {
                nrow[i].value = 0;
                nrow[i].color = val2col(0);
        }

        j = BOARDSIZE-1;
        for(i = j; i >= 0; i--) {
                if(row[i]->value == 0) {
                        ;
                } else if(nrow[j].value == 0) {
                        nrow[j] = *row[i];
                } else if(nrow[j].value == row[i]->value) {
                        score += nrow[j].value *= 2;
                        nrow[j].color = val2col(nrow[j].value);
                        j--;
                } else {
                        nrow[--j] = *row[i];
                }
        }

        moved = 0;
        for(i = 0; i < BOARDSIZE; i++) {
                if(nrow[i].value != row[i]->value)
                        moved = 1;

                *row[i] = nrow[i];
        }

        return moved;
}

void
shift(int dir)
{
        Tile *cur[BOARDSIZE];
        int i;
        int moved = 0;

        for(i = 0; i < BOARDSIZE; i++) {
                //print("%d\n", i);
                buildrow(cur, i, dir);
                moved = merge(cur) || moved;
        }

        if(moved) {
                addrandom();
                // check gameover here
        }
}

void
main(int argc, char **argv)
{
        char *fontname;

        USED(argc);
        USED(argv);

        if(initdraw(0, 0, "2048") < 0)
                sysfatal("initdraw: %r");

        fontname = "/lib/font/bit/fixed/unicode.10x20.font";
        if((bigfont = openfont(display, fontname)) == nil)
                sysfatal("can't open font %s: %r", fontname);

        einit(Ekeyboard|Emouse);
        srand(time(0));

        initcolors();
        initboard();

        redraw();

        while(1) {
                int key;
                redraw();
                flushimage(display, 1);

                switch(key = ekbd()) {
                case 0x7f: // p9p lacks Kdel
                case 'q':
                        // quit
                        exits(0);
                        break;
                case ' ':
                        // reset board
                        initboard();
                case Kup:
                case Kdown:
                case Kleft:
                case Kright:
                        shift(key);
                        break;
                }
        }
}

void
eresized(int new)
{
        if(new && getwindow(display, Refnone) < 0)
                sysfatal("getwindow: %r");

        redraw();
}