TestPrimitivesCopy.c

/* test_copy.c
 * vi:ts=4 sw=4
 *
 * (c) Copyright 2012 Hewlett-Packard Development Company, L.P.
 * Licensed under the Apache License, Version 2.0 (the "License"); you may
 * not use this file except in compliance with the License. You may obtain
 * a copy of the License at http://www.apache.org/licenses/LICENSE-2.0.
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied. See the License for the specific language governing
 * permissions and limitations under the License.
 */
 
#include <stdio.h>
 
#include <freerdp/config.h>
#include <winpr/crypto.h>
 
#include <winpr/sysinfo.h>
#include "prim_test.h"
 
#define COPY_TESTSIZE (256 * 2 + 16 * 2 + 15 + 15)
 
/* ------------------------------------------------------------------------- */
static BOOL test_copy8u_func(void)
{
  primitives_t* prims = primitives_get();
  BYTE data[COPY_TESTSIZE + 15] = WINPR_C_ARRAY_INIT;
  if (winpr_RAND(data, sizeof(data)) < 0)
    return FALSE;
 
  for (int soff = 0; soff < 16; ++soff)
  {
    for (int doff = 0; doff < 16; ++doff)
    {
      for (int length = 1; length <= COPY_TESTSIZE - doff; ++length)
      {
        BYTE dest[COPY_TESTSIZE + 15] = WINPR_C_ARRAY_INIT;
 
        if (prims->copy_8u(data + soff, dest + doff, length) != PRIMITIVES_SUCCESS)
          return FALSE;
 
        for (int i = 0; i < length; ++i)
        {
          if (dest[i + doff] != data[i + soff])
          {
            printf("COPY8U FAIL: off=%d len=%d, dest[%d]=0x%02" PRIx8 ""
                   "data[%d]=0x%02" PRIx8 "\n",
                   doff, length, i + doff, dest[i + doff], i + soff, data[i + soff]);
            return FALSE;
          }
        }
      }
    }
  }
 
  return TRUE;
}
 
/* ------------------------------------------------------------------------- */
static BOOL test_copy8u_speed(void)
{
  BYTE src[MAX_TEST_SIZE + 4] = WINPR_C_ARRAY_INIT;
  BYTE dst[MAX_TEST_SIZE + 4] = WINPR_C_ARRAY_INIT;
 
  if (!speed_test("copy_8u", "aligned", g_Iterations, (speed_test_fkt)generic->copy_8u,
                  (speed_test_fkt)optimized->copy_8u, src, dst, MAX_TEST_SIZE))
    return FALSE;
 
  if (!speed_test("copy_8u", "unaligned", g_Iterations, (speed_test_fkt)generic->copy_8u,
                  (speed_test_fkt)optimized->copy_8u, src + 1, dst + 1, MAX_TEST_SIZE))
    return FALSE;
 
  return TRUE;
}
 
static BYTE* rand_alloc(size_t w, size_t h, size_t bpp, size_t pad, BYTE** copy)
{
  const size_t s = w * bpp + pad;
  BYTE* ptr = calloc(s, h);
  if (!ptr)
    return nullptr;
 
  if (winpr_RAND(ptr, s * h) < 0)
  {
    free(ptr);
    return nullptr;
  }
 
  if (copy)
  {
    BYTE* ptr2 = calloc(s, h);
    if (!ptr2)
    {
      free(ptr);
      return nullptr;
    }
    memcpy(ptr2, ptr, s * h);
    *copy = ptr2;
  }
  return ptr;
}
 
static size_t runcount = 0;
 
static BOOL test_copy_no_overlap_off(BOOL verbose, UINT32 srcFormat, UINT32 dstFormat, UINT32 flags,
                                     UINT32 pad, UINT32 w, UINT32 h, UINT32 dxoff, UINT32 dyoff,
                                     UINT32 sxoff, UINT32 syoff)
{
  BOOL rc = FALSE;
  primitives_t* gen = primitives_get_generic();
  primitives_t* prims = primitives_get();
  if (!gen || !prims)
    return FALSE;
 
  runcount++;
 
  WINPR_ASSERT(dxoff < w);
  WINPR_ASSERT(sxoff < w);
  WINPR_ASSERT(dyoff < h);
  WINPR_ASSERT(syoff < h);
 
  const UINT32 sbpp = FreeRDPGetBytesPerPixel(srcFormat);
  const UINT32 dbpp = FreeRDPGetBytesPerPixel(dstFormat);
 
  if (verbose)
  {
    (void)fprintf(stderr,
                  "run src: %s, dst: %s [flags 0x%08" PRIx32 "] %" PRIu32 "x%" PRIu32
                  ", soff=%" PRIu32 "x%" PRIu32 ", doff=%" PRIu32 "x%" PRIu32 ", pad=%" PRIu32
                  "\n",
                  FreeRDPGetColorFormatName(srcFormat), FreeRDPGetColorFormatName(dstFormat),
                  flags, w, h, sxoff, syoff, dxoff, dyoff, pad);
  }
 
  const UINT32 sstride = (w + sxoff) * sbpp + pad;
  const UINT32 dstride = (w + dxoff) * dbpp + pad;
  BYTE* dst2 = nullptr;
  BYTE* src2 = nullptr;
  BYTE* dst1 = rand_alloc(w + dxoff, h + dyoff, dbpp, pad, &dst2);
  BYTE* src1 = rand_alloc(w + sxoff, h + syoff, sbpp, pad, &src2);
  if (!dst1 || !dst2 || !src1 || !src2)
    goto fail;
 
  if (gen->copy_no_overlap(dst1, dstFormat, dstride, dxoff, dyoff, w, h, src1, srcFormat, sstride,
                           sxoff, syoff, nullptr, flags) != PRIMITIVES_SUCCESS)
    goto fail;
 
  if (memcmp(src1, src2, 1ULL * sstride * h) != 0)
    goto fail;
 
  if (prims->copy_no_overlap(dst2, dstFormat, dstride, dxoff, dyoff, w, h, src1, srcFormat,
                             sstride, sxoff, syoff, nullptr, flags) != PRIMITIVES_SUCCESS)
    goto fail;
 
  if (memcmp(src1, src2, 1ULL * sstride * h) != 0)
    goto fail;
 
  if (memcmp(dst1, dst2, 1ULL * dstride * h) != 0)
    goto fail;
 
  if (flags == FREERDP_KEEP_DST_ALPHA)
  {
    for (size_t y = 0; y < h; y++)
    {
      const BYTE* d1 = &dst1[(y + dyoff) * dstride];
      const BYTE* d2 = &dst2[(y + dyoff) * dstride];
      for (size_t x = 0; x < w; x++)
      {
        const UINT32 c1 = FreeRDPReadColor(&d1[(x + dxoff) * dbpp], dstFormat);
        const UINT32 c2 = FreeRDPReadColor(&d2[(x + dxoff) * dbpp], dstFormat);
        BYTE a1 = 0;
        BYTE a2 = 0;
        FreeRDPSplitColor(c1, dstFormat, nullptr, nullptr, nullptr, &a1, nullptr);
        FreeRDPSplitColor(c2, dstFormat, nullptr, nullptr, nullptr, &a2, nullptr);
        if (a1 != a2)
          goto fail;
      }
    }
  }
  rc = TRUE;
 
fail:
  if (!rc)
  {
    (void)fprintf(stderr, "failed to compare copy_no_overlap(%s -> %s [0x%08" PRIx32 "])\n",
                  FreeRDPGetColorFormatName(srcFormat), FreeRDPGetColorFormatName(dstFormat),
                  flags);
  }
  free(dst1);
  free(dst2);
  free(src1);
  free(src2);
  return rc;
}
 
static BOOL test_copy_no_overlap(BOOL verbose, UINT32 srcFormat, UINT32 dstFormat, UINT32 flags,
                                 UINT32 width, UINT32 height)
{
  BOOL rc = TRUE;
  const UINT32 mw = 4;
  const UINT32 mh = 4;
  for (UINT32 dxoff = 0; dxoff < mw; dxoff++)
  {
    for (UINT32 dyoff = 0; dyoff <= mh; dyoff++)
    {
      for (UINT32 sxoff = 0; sxoff <= mw; sxoff++)
      {
        for (UINT32 syoff = 0; syoff <= mh; syoff++)
        {
          /* We need minimum alignment of 8 bytes.
           * AVX2 can read 8 pixels (at most 8x4=32 bytes) per step
           * if we have 24bpp input that is 24 bytes with 8 bytes read
           * out of bound */
          for (UINT32 pad = 8; pad <= 12; pad++)
          {
            if (!test_copy_no_overlap_off(verbose, srcFormat, dstFormat, flags, pad,
                                          width, height, dxoff, dyoff, sxoff, syoff))
              rc = FALSE;
          }
        }
      }
    }
  }
 
  return rc;
}
 
int TestPrimitivesCopy(int argc, char* argv[])
{
  WINPR_UNUSED(argc);
  WINPR_UNUSED(argv);
 
  const BOOL verbose = argc > 1;
 
  prim_test_setup(FALSE);
 
  if (!test_copy8u_func())
    return 1;
 
  if (g_TestPrimitivesPerformance)
  {
    if (!test_copy8u_speed())
      return 1;
  }
 
  const UINT32 flags[] = {
    FREERDP_FLIP_NONE,
    FREERDP_KEEP_DST_ALPHA,
    FREERDP_FLIP_HORIZONTAL,
    FREERDP_KEEP_DST_ALPHA | FREERDP_FLIP_HORIZONTAL,
#if defined(TEST_ALL_FLAGS)
    FREERDP_FLIP_VERTICAL,
    FREERDP_FLIP_VERTICAL | FREERDP_FLIP_HORIZONTAL,
    FREERDP_KEEP_DST_ALPHA | FREERDP_FLIP_VERTICAL,
    FREERDP_KEEP_DST_ALPHA | FREERDP_FLIP_VERTICAL | FREERDP_FLIP_HORIZONTAL
#endif
  };
  const UINT32 formats[] = { PIXEL_FORMAT_BGRA32,
                           PIXEL_FORMAT_BGRX32,
                           PIXEL_FORMAT_BGR24
#if defined(TEST_ALL_FLAGS) /* Only the previous 3 have SIMD optimizations, so skip the rest */
                           ,
                           PIXEL_FORMAT_RGB24,
                           PIXEL_FORMAT_ABGR32,
                           PIXEL_FORMAT_ARGB32,
                           PIXEL_FORMAT_XBGR32,
                           PIXEL_FORMAT_XRGB32,
                           PIXEL_FORMAT_RGBA32,
                           PIXEL_FORMAT_RGBX32
#endif
  };
 
  int rc = 0;
  for (size_t z = 0; z < ARRAYSIZE(flags); z++)
  {
    const UINT32 flag = flags[z];
    for (size_t x = 0; x < ARRAYSIZE(formats); x++)
    {
      const UINT32 sformat = formats[x];
      for (size_t y = 0; y < ARRAYSIZE(formats); y++)
      {
        const UINT32 dformat = formats[y];
 
        if (!test_copy_no_overlap(verbose, sformat, dformat, flag, 21, 17))
          rc = -1;
      }
    }
  }
 
  if (verbose)
    (void)fprintf(stderr, "runcount=%" PRIuz "\n", runcount);
 
  return rc;
}