/*
 * Copyright (c) 2020 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:
 *    Christian Gmeiner <christian.gmeiner@gmail.com>
 */

#include "etnaviv_compiler.h"
#include "etnaviv_compiler_nir.h"
#include "etnaviv_debug.h"
#include "etnaviv_disk_cache.h"
#include "util/ralloc.h"

struct etna_compiler *
etna_compiler_create(const char *renderer, const struct etna_core_info *info)
{
   struct etna_compiler *compiler = rzalloc(NULL, struct etna_compiler);
   bool has_sign_floor_ceil = etna_core_has_feature(info, ETNA_FEATURE_HAS_SIGN_FLOOR_CEIL);
   bool has_sin_cos_sqrt = etna_core_has_feature(info, ETNA_FEATURE_HAS_SQRT_TRIG);

   compiler->options = (nir_shader_compiler_options) {
      .has_texture_scaling = true,
      .lower_fpow = true,
      .lower_fround_even = true,
      .lower_ftrunc = true,
      .fuse_ffma16 = true,
      .fuse_ffma32 = true,
      .fuse_ffma64 = true,
      .lower_uadd_carry = true,
      .lower_usub_borrow = true,
      .lower_ldexp = true,
      .lower_mul_high = true,
      .lower_bitops = true,
      .lower_flrp32 = true,
      .lower_fmod = true,
      .lower_vector_cmp = true,
      .lower_fdph = true,
      .lower_insert_byte = true,
      .lower_insert_word = true,
      .lower_fdiv = true, /* !specs->has_new_transcendentals */
      .lower_extract_byte = true,
      .lower_extract_word = true,
      .lower_fsign = !has_sign_floor_ceil,
      .lower_ffloor = !has_sign_floor_ceil,
      .lower_fceil = !has_sign_floor_ceil,
      .lower_fsqrt = !has_sin_cos_sqrt,
      .lower_sincos = !has_sin_cos_sqrt,
      .lower_uniforms_to_ubo = info->halti >= 2,
      .force_indirect_unrolling = nir_var_all,
      .max_unroll_iterations = 32,
      .lower_pack_32_2x16_split = true,
      .lower_pack_64_2x32_split = true,
      .lower_pack_half_2x16 = true,
      .lower_pack_snorm_2x16 = true,
      .lower_pack_split = true,
      .lower_pack_unorm_2x16 = true,
      .lower_unpack_32_2x16_split = true,
      .lower_unpack_64_2x32_split = true,
      .lower_unpack_half_2x16 = true,
      .lower_unpack_snorm_2x16 = true,
      .lower_unpack_unorm_2x16 = true,
      .lower_find_lsb = true,
      .lower_ifind_msb = true,
      .lower_ufind_msb = true,
      .has_uclz = true,
      .no_integers = info->halti < 2,
   };

   compiler->regs = etna_ra_setup(compiler);
   if (!compiler->regs) {
      ralloc_free((void *)compiler);
      compiler = NULL;
   }

   etna_disk_cache_init(compiler, renderer);

   return compiler;
}

void
etna_compiler_destroy(const struct etna_compiler *compiler)
{
   disk_cache_destroy(compiler->disk_cache);
   ralloc_free((void *)compiler);
}

const nir_shader_compiler_options *
etna_compiler_get_options(struct etna_compiler *compiler)
{
   return &compiler->options;
}
