/*
 * Copyright (c) 2016 Etnaviv Project
 *
 * 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
 * THE AUTHORS OR COPYRIGHT HOLDERS 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.
 *
 * Authors:
 *    Rob Clark <robclark@freedesktop.org>
 *    Christian Gmeiner <christian.gmeiner@gmail.com>
 */

#include "pipe/p_screen.h"
#include "util/u_inlines.h"

#include "etnaviv_context.h"
#include "etnaviv_query.h"
#include "etnaviv_query_hw.h"
#include "etnaviv_query_sw.h"
#include "etnaviv_query_pm.h"

static struct pipe_query *
etna_create_query(struct pipe_context *pctx, unsigned query_type,
                  unsigned index)
{
   struct etna_context *ctx = etna_context(pctx);
   struct etna_query *q;

   q = etna_sw_create_query(ctx, query_type);
   if (!q)
      q = etna_hw_create_query(ctx, query_type);
   if (!q)
      q = etna_pm_create_query(ctx, query_type);

   return (struct pipe_query *)q;
}

static void
etna_destroy_query(struct pipe_context *pctx, struct pipe_query *pq)
{
   struct etna_query *q = etna_query(pq);

   q->funcs->destroy_query(etna_context(pctx), q);
}

static bool
etna_begin_query(struct pipe_context *pctx, struct pipe_query *pq)
{
   struct etna_query *q = etna_query(pq);
   bool ret;

   if (q->active)
      return false;

   ret = q->funcs->begin_query(etna_context(pctx), q);
   q->active = ret;

   return ret;
}

static bool
etna_end_query(struct pipe_context *pctx, struct pipe_query *pq)
{
   struct etna_query *q = etna_query(pq);

   if (!q->active)
      return false;

   q->funcs->end_query(etna_context(pctx), q);
   q->active = false;

   return true;
}

static bool
etna_get_query_result(struct pipe_context *pctx, struct pipe_query *pq,
                      bool wait, union pipe_query_result *result)
{
   struct etna_query *q = etna_query(pq);

   if (q->active)
      return false;

   util_query_clear_result(result, q->type);

   return q->funcs->get_query_result(etna_context(pctx), q, wait, result);
}

static int
etna_get_driver_query_info(struct pipe_screen *pscreen, unsigned index,
                           struct pipe_driver_query_info *info)
{
   int nr_sw_queries = etna_sw_get_driver_query_info(pscreen, 0, NULL);
   int nr_pm_queries = etna_pm_get_driver_query_info(pscreen, 0, NULL);

   if (!info)
      return nr_sw_queries + nr_pm_queries;

   if (index < nr_sw_queries)
      return etna_sw_get_driver_query_info(pscreen, index, info);

   return etna_pm_get_driver_query_info(pscreen, index - nr_sw_queries, info);
}

static int
etna_get_driver_query_group_info(struct pipe_screen *pscreen, unsigned index,
                                 struct pipe_driver_query_group_info *info)
{
   int nr_sw_groups = etna_sw_get_driver_query_group_info(pscreen, 0, NULL);
   int nr_pm_groups = etna_pm_get_driver_query_group_info(pscreen, 0, NULL);

   if (!info)
      return nr_sw_groups + nr_pm_groups;

   if (index < nr_sw_groups)
      return etna_sw_get_driver_query_group_info(pscreen, index, info);

   return etna_pm_get_driver_query_group_info(pscreen, index, info);
}

static void
etna_set_active_query_state(struct pipe_context *pctx, bool enable)
{
   struct etna_context *ctx = etna_context(pctx);

   if (enable) {
      list_for_each_entry(struct etna_hw_query, hq, &ctx->active_hw_queries, node)
         etna_hw_query_resume(hq, ctx);
   } else {
      list_for_each_entry(struct etna_hw_query, hq, &ctx->active_hw_queries, node)
         etna_hw_query_suspend(hq, ctx);
   }
}

void
etna_query_screen_init(struct pipe_screen *pscreen)
{
   pscreen->get_driver_query_info = etna_get_driver_query_info;
   pscreen->get_driver_query_group_info = etna_get_driver_query_group_info;
}

void
etna_query_context_init(struct pipe_context *pctx)
{
   pctx->create_query = etna_create_query;
   pctx->destroy_query = etna_destroy_query;
   pctx->begin_query = etna_begin_query;
   pctx->end_query = etna_end_query;
   pctx->get_query_result = etna_get_query_result;
   pctx->set_active_query_state = etna_set_active_query_state;
}