/**************************************************************************
 *
 * Copyright 2008 Tungsten Graphics, Inc., Cedar Park, Texas.
 * All Rights Reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the
 * "Software"), to deal in the Software without restriction, including
 * without limitation the rights to use, copy, modify, merge, publish,
 * distribute, sub license, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so, subject to
 * the following conditions:
 *
 * The above copyright notice and this permission notice (including the
 * next paragraph) shall be included in all copies or substantial portions
 * of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
 * IN NO EVENT SHALL TUNGSTEN GRAPHICS AND/OR ITS SUPPLIERS BE LIABLE FOR
 * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 *
 **************************************************************************/

#include "util/u_memory.h"

#include "tr_dump.h"
#include "tr_state.h"
#include "tr_winsys.h"
#include "tr_texture.h"
#include "tr_screen.h"


static const char *
trace_screen_get_name(struct pipe_screen *_screen)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   const char *result;
   
   trace_dump_call_begin("pipe_screen", "get_name");
   
   trace_dump_arg(ptr, screen);

   result = screen->get_name(screen);
   
   trace_dump_ret(string, result);
   
   trace_dump_call_end();
   
   return result;
}


static const char *
trace_screen_get_vendor(struct pipe_screen *_screen)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   const char *result;
   
   trace_dump_call_begin("pipe_screen", "get_vendor");
   
   trace_dump_arg(ptr, screen);
  
   result = screen->get_vendor(screen);
   
   trace_dump_ret(string, result);
   
   trace_dump_call_end();
   
   return result;
}


static int 
trace_screen_get_param(struct pipe_screen *_screen, 
                       int param)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   int result;
   
   trace_dump_call_begin("pipe_screen", "get_param");
   
   trace_dump_arg(ptr, screen);
   trace_dump_arg(int, param);

   result = screen->get_param(screen, param);
   
   trace_dump_ret(int, result);
   
   trace_dump_call_end();
   
   return result;
}


static float 
trace_screen_get_paramf(struct pipe_screen *_screen, 
                        int param)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   float result;
   
   trace_dump_call_begin("pipe_screen", "get_paramf");
   
   trace_dump_arg(ptr, screen);
   trace_dump_arg(int, param);

   result = screen->get_paramf(screen, param);
   
   trace_dump_ret(float, result);
   
   trace_dump_call_end();
   
   return result;
}


static boolean 
trace_screen_is_format_supported(struct pipe_screen *_screen,
                                 enum pipe_format format,
                                 enum pipe_texture_target target,
                                 unsigned tex_usage, 
                                 unsigned geom_flags)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   boolean result;
   
   trace_dump_call_begin("pipe_screen", "is_format_supported");
   
   trace_dump_arg(ptr, screen);
   trace_dump_arg(format, format);
   trace_dump_arg(int, target);
   trace_dump_arg(uint, tex_usage);
   trace_dump_arg(uint, geom_flags);

   result = screen->is_format_supported(screen, format, target, tex_usage, geom_flags);
   
   trace_dump_ret(bool, result);
   
   trace_dump_call_end();
   
   return result;
}


static struct pipe_texture *
trace_screen_texture_create(struct pipe_screen *_screen,
                            const struct pipe_texture *templat)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   struct pipe_texture *result;
   
   trace_dump_call_begin("pipe_screen", "texture_create");

   trace_dump_arg(ptr, screen);
   trace_dump_arg(template, templat);

   result = screen->texture_create(screen, templat);
   
   trace_dump_ret(ptr, result);
   
   trace_dump_call_end();
   
   result = trace_texture_create(tr_scr, result);
   
   return result;
}


static struct pipe_texture *
trace_screen_texture_blanket(struct pipe_screen *_screen,
                             const struct pipe_texture *templat,
                             const unsigned *ppitch,
                             struct pipe_buffer *buffer)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   unsigned pitch = *ppitch;
   struct pipe_texture *result;

   trace_dump_call_begin("pipe_screen", "texture_blanket");

   trace_dump_arg(ptr, screen);
   trace_dump_arg(template, templat);
   trace_dump_arg(uint, pitch);
   trace_dump_arg(ptr, buffer);

   result = screen->texture_blanket(screen, templat, ppitch, buffer);
   
   trace_dump_ret(ptr, result);
   
   trace_dump_call_end();
   
   result = trace_texture_create(tr_scr, result);
   
   return result;
}


static void 
trace_screen_texture_release(struct pipe_screen *_screen,
                             struct pipe_texture **ptexture)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   struct trace_texture *tr_tex;
   struct pipe_texture *texture;
   
   assert(ptexture);
   if(*ptexture) {
      tr_tex = trace_texture(tr_scr, *ptexture);
      texture = tr_tex->texture;
      assert(texture->screen == screen);
   }
   else
      texture = NULL;
   
   if (*ptexture) {
      if (!--(*ptexture)->refcount) {
         trace_dump_call_begin("pipe_screen", "texture_destroy");
         
         trace_dump_arg(ptr, screen);
         trace_dump_arg(ptr, texture);
         
         trace_texture_destroy(tr_scr, *ptexture);
         
         trace_dump_call_end();
      }
   
      *ptexture = NULL;
   }
}


static struct pipe_surface *
trace_screen_get_tex_surface(struct pipe_screen *_screen,
                             struct pipe_texture *texture,
                             unsigned face, unsigned level,
                             unsigned zslice,
                             unsigned usage)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   struct trace_texture *tr_tex;
   struct pipe_surface *result;
   
   assert(texture);
   tr_tex = trace_texture(tr_scr, texture);
   texture = tr_tex->texture;
   assert(texture->screen == screen);
   
   trace_dump_call_begin("pipe_screen", "get_tex_surface");
   
   trace_dump_arg(ptr, screen);
   trace_dump_arg(ptr, texture);
   trace_dump_arg(uint, face);
   trace_dump_arg(uint, level);
   trace_dump_arg(uint, zslice);
   trace_dump_arg(uint, usage);

   result = screen->get_tex_surface(screen, texture, face, level, zslice, usage);

   trace_dump_ret(ptr, result);
   
   trace_dump_call_end();
   
   result = trace_surface_create(tr_tex, result);

   return result;
}


static void 
trace_screen_tex_surface_release(struct pipe_screen *_screen,
                                 struct pipe_surface **psurface)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   struct trace_texture *tr_tex;
   struct trace_surface *tr_surf;
   struct pipe_surface *surface;
   
   assert(psurface);
   if(*psurface) {
      tr_tex = trace_texture(tr_scr, (*psurface)->texture);
      tr_surf = trace_surface(tr_tex, *psurface);
      surface = tr_surf->surface;
   }
   else
      surface = NULL;
   
   if (*psurface) {
      if (!--(*psurface)->refcount) {
         trace_dump_call_begin("pipe_screen", "tex_surface_destroy");
         
         trace_dump_arg(ptr, screen);
         trace_dump_arg(ptr, surface);

         trace_surface_destroy(tr_tex, *psurface);

         trace_dump_call_end();
      }
   
      *psurface = NULL;
   }
}


static void *
trace_screen_surface_map(struct pipe_screen *_screen,
                         struct pipe_surface *surface,
                         unsigned flags)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   struct trace_texture *tr_tex;
   struct trace_surface *tr_surf;
   void *map;
   
   tr_tex = trace_texture(tr_scr, surface->texture);
   tr_surf = trace_surface(tr_tex, surface);
   surface = tr_surf->surface;

   map = screen->surface_map(screen, surface, flags);
   if(map) {
      if(flags & PIPE_BUFFER_USAGE_CPU_WRITE) {
         assert(!tr_surf->map);
         tr_surf->map = map;
      }
   }
   
   return map;
}


static void 
trace_screen_surface_unmap(struct pipe_screen *_screen,
                           struct pipe_surface *surface)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   struct trace_texture *tr_tex;
   struct trace_surface *tr_surf;
   
   tr_tex = trace_texture(tr_scr, surface->texture);
   tr_surf = trace_surface(tr_tex, surface);
   surface = tr_surf->surface;
   
   if(tr_surf->map) {
      size_t size = surface->nblocksy * surface->stride;
      
      trace_dump_call_begin("pipe_winsys", "surface_write");
      
      trace_dump_arg(ptr, screen);
      
      trace_dump_arg(ptr, surface);
      
      trace_dump_arg_begin("data");
      trace_dump_bytes(tr_surf->map, size);
      trace_dump_arg_end();

      trace_dump_arg_begin("stride");
      trace_dump_uint(surface->stride);
      trace_dump_arg_end();

      trace_dump_arg_begin("size");
      trace_dump_uint(size);
      trace_dump_arg_end();
   
      trace_dump_call_end();

      tr_surf->map = NULL;
   }

   screen->surface_unmap(screen, surface);
}


static void
trace_screen_destroy(struct pipe_screen *_screen)
{
   struct trace_screen *tr_scr = trace_screen(_screen);
   struct pipe_screen *screen = tr_scr->screen;
   
   trace_dump_call_begin("pipe_screen", "destroy");
   
   trace_dump_arg(ptr, screen);

   screen->destroy(screen);
   
   trace_dump_call_end();

   trace_dump_trace_end();

   FREE(tr_scr);
}


struct pipe_screen *
trace_screen_create(struct pipe_screen *screen)
{
   struct trace_screen *tr_scr;
   struct pipe_winsys *winsys;
   
   if(!screen)
      goto error1;

   if(!trace_dump_trace_begin())
      goto error1;

   tr_scr = CALLOC_STRUCT(trace_screen);
   if(!tr_scr)
      goto error2;

   winsys = trace_winsys_create(screen->winsys);
   if(!winsys)
      goto error3;
   
   tr_scr->base.winsys = winsys;
   tr_scr->base.destroy = trace_screen_destroy;
   tr_scr->base.get_name = trace_screen_get_name;
   tr_scr->base.get_vendor = trace_screen_get_vendor;
   tr_scr->base.get_param = trace_screen_get_param;
   tr_scr->base.get_paramf = trace_screen_get_paramf;
   tr_scr->base.is_format_supported = trace_screen_is_format_supported;
   tr_scr->base.texture_create = trace_screen_texture_create;
   tr_scr->base.texture_blanket = trace_screen_texture_blanket;
   tr_scr->base.texture_release = trace_screen_texture_release;
   tr_scr->base.get_tex_surface = trace_screen_get_tex_surface;
   tr_scr->base.tex_surface_release = trace_screen_tex_surface_release;
   tr_scr->base.surface_map = trace_screen_surface_map;
   tr_scr->base.surface_unmap = trace_screen_surface_unmap;
   
   tr_scr->screen = screen;

   trace_dump_call_begin("", "pipe_screen_create");
   trace_dump_arg_begin("winsys");
   trace_dump_ptr(screen->winsys);
   trace_dump_arg_end();
   trace_dump_ret(ptr, screen);
   trace_dump_call_end();

   return &tr_scr->base;

error3:
   FREE(tr_scr);
error2:
   trace_dump_trace_end();
error1:
   return screen;
}


struct trace_screen *
trace_screen(struct pipe_screen *screen)
{
   assert(screen);
   assert(screen->destroy == trace_screen_destroy);
   return (struct trace_screen *)screen;
}