/* FreeRDP: A Remote Desktop Protocol Client

 * Routines to set a chunk of memory to a constant.

 * 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 <freerdp/config.h>

#include <string.h>

#include <freerdp/types.h>

#include <freerdp/primitives.h>

#include "prim_internal.h"

#include "prim_set.h"

/* ========================================================================= */

static pstatus_t general_set_8u(BYTE val, BYTE* WINPR_RESTRICT pDst, UINT32 len)

{

  memset((void*)pDst, (int)val, (size_t)len);

  return PRIMITIVES_SUCCESS;

}

/* ------------------------------------------------------------------------- */

static pstatus_t general_zero(void* WINPR_RESTRICT pDst, size_t len)

{

  memset(pDst, 0, len);

  return PRIMITIVES_SUCCESS;

}

/* ========================================================================= */

static pstatus_t general_set_32s(INT32 val, INT32* WINPR_RESTRICT pDst, UINT32 len)

{

  INT32* dptr = pDst;

  size_t span = 0;

  size_t remaining = 0;

  if (len < 256)

  {

    while (len--)

      *dptr++ = val;

    return PRIMITIVES_SUCCESS;

  }

  /* else quadratic growth memcpy algorithm */

  span = 1;

  *dptr = val;

  remaining = len - 1;

  primitives_t* prims = primitives_get();

  while (remaining)

  {

    size_t thiswidth = span;

    if (thiswidth > remaining)

      thiswidth = remaining;

    const size_t s = thiswidth << 2;

    WINPR_ASSERT(thiswidth <= INT32_MAX);

    const pstatus_t rc = prims->copy_8u((BYTE*)dptr, (BYTE*)(dptr + span), (INT32)s);

    if (rc != PRIMITIVES_SUCCESS)

      return rc;

    remaining -= thiswidth;

    span <<= 1;

  }

  return PRIMITIVES_SUCCESS;

}

/* ------------------------------------------------------------------------- */

static pstatus_t general_set_32u(UINT32 val, UINT32* WINPR_RESTRICT pDst, UINT32 len)

{

  UINT32* dptr = pDst;

  size_t span = 0;

  size_t remaining = 0;

  primitives_t* prims = NULL;

  if (len < 256)

  {

    while (len--)

      *dptr++ = val;

    return PRIMITIVES_SUCCESS;

  }

  /* else quadratic growth memcpy algorithm */

  span = 1;

  *dptr = val;

  remaining = len - 1;

  prims = primitives_get();

  while (remaining)

  {

    size_t thiswidth = span;

    if (thiswidth > remaining)

      thiswidth = remaining;

    const size_t s = thiswidth << 2;

    WINPR_ASSERT(thiswidth <= INT32_MAX);

    const pstatus_t rc = prims->copy_8u((BYTE*)dptr, (BYTE*)(dptr + span), (INT32)s);

    if (rc != PRIMITIVES_SUCCESS)

      return rc;

    remaining -= thiswidth;

    span <<= 1;

  }

  return PRIMITIVES_SUCCESS;

}

/* ------------------------------------------------------------------------- */

void primitives_init_set(primitives_t* WINPR_RESTRICT prims)

{

  /* Start with the default. */

  prims->set_8u = general_set_8u;

  prims->set_32s = general_set_32s;

  prims->set_32u = general_set_32u;

  prims->zero = general_zero;

}

void primitives_init_set_opt(primitives_t* WINPR_RESTRICT prims)

{

  primitives_init_set(prims);

  primitives_init_set_sse2(prims);

}