Subversion Repositories Kolibri OS

/**************************************************************************

 *

 * Copyright © 2010 Luca Barbieri

 *

 * 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, sublicense,

 * 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 NONINFRINGEMENT. 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.

 *

 **************************************************************************/

#include 

#include "translate/translate.h"

#include "util/u_memory.h"

#include "util/u_format.h"

#include "util/u_half.h"

#include "util/u_cpu_detect.h"

#include "rtasm/rtasm_cpu.h"

/* don't use this for serious use */

static double rand_double()

{

   const double rm = (double)RAND_MAX + 1;

   double div = 1;

   double v = 0;

   unsigned i;

   for(i = 0; i < 4; ++i)

   {

      div *= rm;

      v += (double)rand() / div;

   }

   return v;

}

int main(int argc, char** argv)

{

   struct translate *(*create_fn)(const struct translate_key *key) = 0;

   struct translate_key key;

   unsigned output_format;

   unsigned input_format;

   unsigned buffer_size = 4096;

   unsigned char* buffer[5];

   unsigned char* byte_buffer;

   float* float_buffer;

   double* double_buffer;

   uint16_t *half_buffer;

   unsigned * elts;

   unsigned count = 4;

   unsigned i, j, k;

   unsigned passed = 0;

   unsigned total = 0;

   const float error = 0.03125;

   create_fn = 0;

   util_cpu_detect();

   if(argc <= 1)

   {}

   else if (!strcmp(argv[1], "generic"))

      create_fn = translate_generic_create;

   else if (!strcmp(argv[1], "x86"))

      create_fn = translate_sse2_create;

   else if (!strcmp(argv[1], "nosse"))

   {

      util_cpu_caps.has_sse = 0;

      util_cpu_caps.has_sse2 = 0;

      util_cpu_caps.has_sse3 = 0;

      util_cpu_caps.has_sse4_1 = 0;

      create_fn = translate_sse2_create;

   }

   else if (!strcmp(argv[1], "sse"))

   {

      if(!util_cpu_caps.has_sse || !rtasm_cpu_has_sse())

      {

         printf("Error: CPU doesn't support SSE (test with qemu)\n");

         return 2;

      }

      util_cpu_caps.has_sse2 = 0;

      util_cpu_caps.has_sse3 = 0;

      util_cpu_caps.has_sse4_1 = 0;

      create_fn = translate_sse2_create;

   }

   else if (!strcmp(argv[1], "sse2"))

   {

      if(!util_cpu_caps.has_sse2 || !rtasm_cpu_has_sse())

      {

         printf("Error: CPU doesn't support SSE2 (test with qemu)\n");

         return 2;

      }

      util_cpu_caps.has_sse3 = 0;

      util_cpu_caps.has_sse4_1 = 0;

      create_fn = translate_sse2_create;

   }

   else if (!strcmp(argv[1], "sse3"))

   {

      if(!util_cpu_caps.has_sse3 || !rtasm_cpu_has_sse())

      {

         printf("Error: CPU doesn't support SSE3 (test with qemu)\n");

         return 2;

      }

      util_cpu_caps.has_sse4_1 = 0;

      create_fn = translate_sse2_create;

   }

   else if (!strcmp(argv[1], "sse4.1"))

   {

      if(!util_cpu_caps.has_sse4_1 || !rtasm_cpu_has_sse())

      {

         printf("Error: CPU doesn't support SSE4.1 (test with qemu)\n");

         return 2;

      }

      create_fn = translate_sse2_create;

   }

   if (!create_fn)

   {

      printf("Usage: ./translate_test [generic|x86|nosse|sse|sse2|sse3|sse4.1]\n");

      return 2;

   }

   for (i = 1; i < Elements(buffer); ++i)

      buffer[i] = align_malloc(buffer_size, 4096);

   byte_buffer = align_malloc(buffer_size, 4096);

   float_buffer = align_malloc(buffer_size, 4096);

   double_buffer = align_malloc(buffer_size, 4096);

   half_buffer = align_malloc(buffer_size, 4096);

   elts = align_malloc(count * sizeof *elts, 4096);

   key.nr_elements = 1;

   key.element[0].input_buffer = 0;

   key.element[0].input_offset = 0;

   key.element[0].output_offset = 0;

   key.element[0].type = TRANSLATE_ELEMENT_NORMAL;

   key.element[0].instance_divisor = 0;

   srand(4359025);

   /* avoid negative values that work badly when converted to unsigned format*/

   for (i = 0; i < buffer_size; ++i)

      byte_buffer[i] = rand() & 0x7f7f7f7f;

   for (i = 0; i < buffer_size / sizeof(float); ++i)

      float_buffer[i] = (float)rand_double();

   for (i = 0; i < buffer_size / sizeof(double); ++i)

      double_buffer[i] = rand_double();

   for (i = 0; i < buffer_size / sizeof(double); ++i)

      half_buffer[i] = util_float_to_half((float) rand_double());

   for (i = 0; i < count; ++i)

      elts[i] = i;

   for (output_format = 1; output_format < PIPE_FORMAT_COUNT; ++output_format)

   {

      const struct util_format_description* output_format_desc = util_format_description(output_format);

      unsigned output_format_size;

      unsigned output_normalized = 0;

      if (!output_format_desc

            || !output_format_desc->fetch_rgba_float

            || !output_format_desc->pack_rgba_float

            || output_format_desc->colorspace != UTIL_FORMAT_COLORSPACE_RGB

            || output_format_desc->layout != UTIL_FORMAT_LAYOUT_PLAIN

            || !translate_is_output_format_supported(output_format))

         continue;

      for(i = 0; i < output_format_desc->nr_channels; ++i)

      {

         if(output_format_desc->channel[i].type != UTIL_FORMAT_TYPE_FLOAT)

            output_normalized |= (1 << output_format_desc->channel[i].normalized);

      }

      output_format_size = util_format_get_stride(output_format, 1);

      for (input_format = 1; input_format < PIPE_FORMAT_COUNT; ++input_format)

      {

         const struct util_format_description* input_format_desc = util_format_description(input_format);

         unsigned input_format_size;

         struct translate* translate[2];

         unsigned fail = 0;

         unsigned used_generic = 0;

         unsigned input_normalized = 0;

         boolean input_is_float = FALSE;

         if (!input_format_desc

               || !input_format_desc->fetch_rgba_float

               || !input_format_desc->pack_rgba_float

               || input_format_desc->colorspace != UTIL_FORMAT_COLORSPACE_RGB

               || input_format_desc->layout != UTIL_FORMAT_LAYOUT_PLAIN

               || !translate_is_output_format_supported(input_format))

            continue;

         input_format_size = util_format_get_stride(input_format, 1);

         for(i = 0; i < input_format_desc->nr_channels; ++i)

         {

            if(input_format_desc->channel[i].type == UTIL_FORMAT_TYPE_FLOAT)

            {

               input_is_float = 1;

               input_normalized |= 1 << 1;

            }

            else

               input_normalized |= (1 << input_format_desc->channel[i].normalized);

         }

         if(((input_normalized | output_normalized) == 3)

               || ((input_normalized & 1) && (output_normalized & 1)

                     && input_format_size * output_format_desc->nr_channels > output_format_size * input_format_desc->nr_channels))

            continue;

         key.element[0].input_format = input_format;

         key.element[0].output_format = output_format;

         key.output_stride = output_format_size;

         translate[0] = create_fn(&key);

         if (!translate[0])

            continue;

         key.element[0].input_format = output_format;

         key.element[0].output_format = input_format;

         key.output_stride = input_format_size;

         translate[1] = create_fn(&key);

         if(!translate[1])

         {

            used_generic = 1;

            translate[1] = translate_generic_create(&key);

            if(!translate[1])

               continue;

         }

         for(i = 1; i < 5; ++i)

            memset(buffer[i], 0xcd - (0x22 * i), 4096);

         if(input_is_float && input_format_desc->channel[0].size == 32)

            buffer[0] = (unsigned char*)float_buffer;

         else if(input_is_float && input_format_desc->channel[0].size == 64)

            buffer[0] = (unsigned char*)double_buffer;

         else if(input_is_float && input_format_desc->channel[0].size == 16)

            buffer[0] = (unsigned char*)half_buffer;

         else if(input_is_float)

            abort();

         else

            buffer[0] = byte_buffer;

         translate[0]->set_buffer(translate[0], 0, buffer[0], input_format_size, count - 1);

         translate[0]->run_elts(translate[0], elts, count, 0, 0, buffer[1]);

         translate[1]->set_buffer(translate[1], 0, buffer[1], output_format_size, count - 1);

         translate[1]->run_elts(translate[1], elts, count, 0, 0, buffer[2]);

         translate[0]->set_buffer(translate[0], 0, buffer[2], input_format_size, count - 1);

         translate[0]->run_elts(translate[0], elts, count, 0, 0, buffer[3]);

         translate[1]->set_buffer(translate[1], 0, buffer[3], output_format_size, count - 1);

         translate[1]->run_elts(translate[1], elts, count, 0, 0, buffer[4]);

         for (i = 0; i < count; ++i)

         {

            float a[4];

            float b[4];

            input_format_desc->fetch_rgba_float(a, buffer[2] + i * input_format_size, 0, 0);

            input_format_desc->fetch_rgba_float(b, buffer[4] + i * input_format_size, 0, 0);

            for (j = 0; j < count; ++j)

            {

               float d = a[j] - b[j];

               if (d > error || d < -error)

               {

                  fail = 1;

                  break;

               }

            }

         }

         printf("%s%s: %s -> %s -> %s -> %s -> %s\n",

               fail ? "FAIL" : "PASS",

               used_generic ? "[GENERIC]" : "",

               input_format_desc->name, output_format_desc->name, input_format_desc->name, output_format_desc->name, input_format_desc->name);

         if (1)

         {

            for (i = 0; i < Elements(buffer); ++i)

            {

               unsigned format_size = (i & 1) ? output_format_size : input_format_size;

               printf("%c ", (i == 2 || i == 4) ? '*' : ' ');

               for (j = 0; j < count; ++j)

               {

                  for (k = 0; k < format_size; ++k)

                  {

                     printf("%02x", buffer[i][j * format_size + k]);

                  }

                  printf(" ");

               }

               printf("\n");

            }

         }

         if (!fail)

            ++passed;

         ++total;

         if(translate[1])

            translate[1]->release(translate[1]);

         translate[0]->release(translate[0]);

      }

   }

   printf("%u/%u tests passed for translate_%s\n", passed, total, argv[1]);

   return passed != total;

}