teapot.c

Go to the documentation of this file.

/*
# _____     ___ ____     ___ ____
#  ____|   |    ____|   |        | |____|
# |     ___|   |____ ___|    ____| |    \    PS2DEV Open Source Project.
#-----------------------------------------------------------------------
# (c) 2005 Naomi Peori <naomi@peori.ca>
# Licenced under Academic Free License version 2.0
# Review ps2sdk README & LICENSE files for further details.
#
*/
 
#include <kernel.h>
#include <stdlib.h>
#include <tamtypes.h>
#include <math3d.h>
 
#include <packet.h>
 
#include <dma_tags.h>
#include <gif_tags.h>
#include <gs_psm.h>
 
#include <dma.h>
 
#include <graph.h>
 
#include <draw.h>
#include <draw3d.h>
 
#include "mesh_data.c"
 
VECTOR camera_position = { 0.00f, 0.00f, 100.00f, 1.00f };
VECTOR camera_rotation = { 0.00f, 0.00f,   0.00f, 1.00f };
 
VECTOR *temp_normals;
VECTOR *temp_lights;
VECTOR *temp_colours;
VECTOR *temp_vertices;
 
xyz_t *xyz;
color_t *rgbaq;
 
int light_count = 4;
 
VECTOR light_direction[4] = {
        {  0.00f,  0.00f,  0.00f, 1.00f },
        {  1.00f,  0.00f, -1.00f, 1.00f },
        {  0.00f,  1.00f, -1.00f, 1.00f },
        { -1.00f, -1.00f, -1.00f, 1.00f }
};
 
VECTOR light_colour[4] = {
        { 0.00f, 0.00f, 0.00f, 1.00f },
        { 1.00f, 0.00f, 0.00f, 1.00f },
        { 0.30f, 0.30f, 0.30f, 1.00f },
        { 0.50f, 0.50f, 0.50f, 1.00f }
};
 
int light_type[4] = {
        LIGHT_AMBIENT,
        LIGHT_DIRECTIONAL,
        LIGHT_DIRECTIONAL,
        LIGHT_DIRECTIONAL
};
 
void init_gs(framebuffer_t *frame, zbuffer_t *z)
{
 
        // Define a 32-bit 640x512 framebuffer.
        frame->width = 640;
        frame->height = 512;
        frame->mask = 0;
        frame->psm = GS_PSM_32;
 
        // Allocate some vram for our framebuffer.
        frame->address = graph_vram_allocate(frame->width,frame->height, frame->psm, GRAPH_ALIGN_PAGE);
 
        frame++;
 
        frame->width = 640;
        frame->height = 512;
        frame->mask = 0;
        frame->psm = GS_PSM_32;
 
        // Allocate some vram for our framebuffer.
        frame->address = graph_vram_allocate(frame->width,frame->height, frame->psm, GRAPH_ALIGN_PAGE);
 
        // Enable the zbuffer.
        z->enable = DRAW_ENABLE;
        z->mask = 0;
        z->method = ZTEST_METHOD_GREATER_EQUAL;
        z->zsm = GS_ZBUF_32;
        z->address = graph_vram_allocate(frame->width,frame->height,z->zsm, GRAPH_ALIGN_PAGE);
 
        // Initialize the screen and tie the first framebuffer to the read circuits.
        graph_initialize(frame->address,frame->width,frame->height,frame->psm,0,0);
 
}
 
void init_drawing_environment(framebuffer_t *frame, zbuffer_t *z)
{
 
        packet_t *packet = packet_init(20,PACKET_NORMAL);
 
        // This is our generic qword pointer.
        qword_t *q = packet->data;
 
        // This will setup a default drawing environment.
        q = draw_setup_environment(q,0,frame,z);
 
        // Now reset the primitive origin to 2048-width/2,2048-height/2.
        q = draw_primitive_xyoffset(q,0,(2048-320),(2048-256));
 
        // Finish setting up the environment.
        q = draw_finish(q);
 
        // Now send the packet, no need to wait since it's the first.
        dma_channel_send_normal(DMA_CHANNEL_GIF,packet->data,q - packet->data, 0, 0);
        dma_wait_fast();
 
        free(packet);
 
}
 
void flip_buffers(packet_t *flip,framebuffer_t *frame)
{
 
        qword_t *q = flip->data;
 
        q = draw_framebuffer(q,0,frame);
        q = draw_finish(q);
 
        dma_wait_fast();
        dma_channel_send_normal_ucab(DMA_CHANNEL_GIF,flip->data,q - flip->data, 0);
 
        draw_wait_finish();
 
}
 
qword_t *render_teapot(qword_t *q,MATRIX view_screen, VECTOR object_position, VECTOR object_rotation, prim_t *prim, color_t *color, framebuffer_t *frame, zbuffer_t *z)
{
 
        int i;
 
        qword_t *dmatag;
 
        MATRIX local_world;
        MATRIX local_light;
        MATRIX world_view;
 
        MATRIX local_screen;
 
        // Now grab our qword pointer and increment past the dmatag.
        dmatag = q;
        q++;
 
        // Spin the teapot a bit.
        object_rotation[0] += 0.008f; while (object_rotation[0] > 3.14f) { object_rotation[0] -= 6.28f; }
        object_rotation[1] += 0.112f; while (object_rotation[1] > 3.14f) { object_rotation[1] -= 6.28f; }
 
        // Create the local_world matrix.
        create_local_world(local_world, object_position, object_rotation);
 
        // Create the local_light matrix.
        create_local_light(local_light, object_rotation);
 
        // Create the world_view matrix.
        create_world_view(world_view, camera_position, camera_rotation);
 
        // Create the local_screen matrix.
        create_local_screen(local_screen, local_world, world_view, view_screen);
 
        // Calculate the normal values.
        calculate_normals(temp_normals, vertex_count, normals, local_light);
 
        // Calculate the lighting values.
        calculate_lights(temp_lights, vertex_count, temp_normals, light_direction, light_colour, light_type, light_count);
 
        // Calculate the colour values after lighting.
        calculate_colours(temp_colours, vertex_count, colours, temp_lights);
 
        // Calculate the vertex values.
        calculate_vertices(temp_vertices, vertex_count, vertices, local_screen);
 
        // Convert floating point vertices to fixed point and translate to center of screen.
        draw_convert_xyz(xyz, 2048, 2048, 32, vertex_count, (vertex_f_t*)temp_vertices);
 
        // Convert floating point colours to fixed point.
        draw_convert_rgbq(rgbaq, vertex_count, (vertex_f_t*)temp_vertices, (color_f_t*)temp_colours, color->a);
 
        // Draw the triangles using triangle primitive type.
        q = draw_prim_start(q,0,prim,color);
 
        for(i = 0; i < points_count; i++)
        {
                q->dw[0] = rgbaq[points[i]].rgbaq;
                q->dw[1] = xyz[points[i]].xyz;
                q++;
        }
 
        q = draw_prim_end(q,2,DRAW_RGBAQ_REGLIST);
 
        // Define our dmatag for the dma chain.
        DMATAG_CNT(dmatag,q-dmatag-1,0,0,0);
 
 
        return q;
 
}
 
int render(framebuffer_t *frame, zbuffer_t *z)
{
 
        int context = 0;
 
        // Packets for doublebuffering dma sends
        packet_t *packets[2];
        packet_t *current;
 
        // This packet is special for framebuffer switching
        packet_t *flip_pkt;
 
        qword_t *q;
        qword_t *dmatag;
 
        prim_t prim;
        color_t color;
 
        MATRIX view_screen;
 
        packets[0] = packet_init(40000,PACKET_NORMAL);
        packets[1] = packet_init(40000,PACKET_NORMAL);
 
        // Uncached accelerated
        flip_pkt = packet_init(3,PACKET_UCAB);
 
        VECTOR object_position = { 0.00f, 0.00f, 0.00f, 1.00f };
        VECTOR object_rotation = { 0.00f, 0.00f, 0.00f, 1.00f };
 
        // Define the triangle primitive we want to use.
        prim.type = PRIM_TRIANGLE;
        prim.shading = PRIM_SHADE_GOURAUD;
        prim.mapping = DRAW_DISABLE;
        prim.fogging = DRAW_DISABLE;
        prim.blending = DRAW_ENABLE;
        prim.antialiasing = DRAW_DISABLE;
        prim.mapping_type = DRAW_DISABLE;
        prim.colorfix = PRIM_UNFIXED;
 
        color.r = 0x80;
        color.g = 0x80;
        color.b = 0x80;
        color.a = 0x80;
        color.q = 1.0f;
 
        // Allocate calculation space.
        temp_normals  = memalign(128, sizeof(VECTOR) * vertex_count);
        temp_lights   = memalign(128, sizeof(VECTOR) * vertex_count);
        temp_colours  = memalign(128, sizeof(VECTOR) * vertex_count);
        temp_vertices = memalign(128, sizeof(VECTOR) * vertex_count);
 
        // Allocate register space.
        xyz   = memalign(128, sizeof(u64) * vertex_count);
        rgbaq = memalign(128, sizeof(u64) * vertex_count);
 
        // Create the view_screen matrix.
        create_view_screen(view_screen, graph_aspect_ratio(), -3.00f, 3.00f, -3.00f, 3.00f, 1.00f, 2000.00f);
 
        for (;;)
        {
 
                current = packets[context];
 
                q = current->data;
 
                dmatag = q;
                q++;
 
                // Clear framebuffer without any pixel testing.
                q = draw_disable_tests(q,0,z);
                q = draw_clear(q,0,2048.0f-320.0f,2048.0f-256.0f,frame->width,frame->height,0x00,0x00,0x00);
                q = draw_enable_tests(q,0,z);
 
                DMATAG_CNT(dmatag,q-dmatag - 1,0,0,0);
 
                //render teapots
                color.a = 0x40;
                object_position[0] = 30.0f;
                q = render_teapot(q, view_screen, object_position, object_rotation, &prim, &color, frame, z);
 
                object_position[0] = -30.0f;
                q = render_teapot(q, view_screen, object_position, object_rotation, &prim, &color, frame, z);
 
                color.a = 0x80;
                object_position[0] = 0.0f;
                object_position[1] = -20.0f;
                q = render_teapot(q, view_screen, object_position, object_rotation, &prim, &color, frame, z);
 
                object_position[1] = 20.0f;
                q = render_teapot(q, view_screen, object_position, object_rotation, &prim, &color, frame, z);
 
                object_position[0] = 0.0f;
                object_position[1] = 0.0f;
 
                dmatag = q;
                q++;
 
                q = draw_finish(q);
 
                DMATAG_END(dmatag,q-dmatag-1,0,0,0);
 
                // Now send our current dma chain.
                dma_wait_fast();
                dma_channel_send_chain(DMA_CHANNEL_GIF,current->data, q - current->data, 0, 0);
 
                // Either block until a vsync, or keep rendering until there's one available.
                graph_wait_vsync();
 
                draw_wait_finish();
                graph_set_framebuffer_filtered(frame[context].address,frame[context].width,frame[context].psm,0,0);
 
                // Switch context.
                context ^= 1;
 
                // We need to flip buffers outside of the chain, for some reason,
                // so we use a separate small packet.
                flip_buffers(flip_pkt,&frame[context]);
 
        }
 
        free(packets[0]);
        free(packets[1]);
 
}
 
int main(int argc, char **argv)
{
 
        // The buffers to be used.
        framebuffer_t frame[2];
        zbuffer_t z;
 
        // Init GIF dma channel.
        dma_channel_initialize(DMA_CHANNEL_GIF,NULL,0);
        dma_channel_fast_waits(DMA_CHANNEL_GIF);
 
        // Init the GS, framebuffer, and zbuffer.
        init_gs(frame, &z);
 
        // Init the drawing environment and framebuffer.
        init_drawing_environment(frame,&z);
 
        // Render teapots
        render(frame,&z);
 
        // Sleep
        SleepThread();
 
        // End program.
        return 0;
 
}