/** @file
  OcBlitLib - Library to perform blt operations on a frame buffer.

  Copyright (c) 2007 - 2018, Intel Corporation. All rights reserved.<BR>
  Copyright (c) 2021, vit9696. All rights reserved.<BR>

  SPDX-License-Identifier: BSD-2-Clause-Patent
**/

#include "BlitInternal.h"

#include <Library/BaseLib.h>
#include <Library/BaseMemoryLib.h>
#include <Library/DebugLib.h>

RETURN_STATUS
BlitLibVideoToBuffer0 (
  IN     OC_BLIT_CONFIGURE           *Configure,
  OUT EFI_GRAPHICS_OUTPUT_BLT_PIXEL  *BltBuffer,
  IN     UINTN                       SourceX,
  IN     UINTN                       SourceY,
  IN     UINTN                       DestinationX,
  IN     UINTN                       DestinationY,
  IN     UINTN                       Width,
  IN     UINTN                       Height,
  IN     UINTN                       DeltaPixels
  )
{
  UINT32  *Source;
  UINT32  *Destination;
  UINT32  *SourceWalker;
  UINT32  *DestinationWalker;
  UINTN   IndexX;
  UINTN   WidthInBytes;
  UINTN   PixelsPerScanLine;
  UINT32  Uint32;

  WidthInBytes      = Width * BYTES_PER_PIXEL;
  PixelsPerScanLine = Configure->PixelsPerScanLine;

  Destination = (UINT32 *)BltBuffer
                + DestinationY * DeltaPixels + DestinationX;
  Source = (UINT32 *)Configure->FrameBuffer
           + SourceY * PixelsPerScanLine + SourceX;
  if (Configure->PixelFormat == PixelBlueGreenRedReserved8BitPerColor) {
    while (Height > 0) {
      CopyMem (Destination, Source, WidthInBytes);
      Source      += PixelsPerScanLine;
      Destination += DeltaPixels;
      Height--;
    }
  } else {
    while (Height > 0) {
      DestinationWalker = (UINT32 *)Configure->LineBuffer;
      SourceWalker      = (UINT32 *)Source;
      for (IndexX = 0; IndexX < Width; IndexX++) {
        Uint32               = *SourceWalker++;
        *DestinationWalker++ =
          (UINT32)(
                   (((Uint32 << Configure->PixelShl[0]) >> Configure->PixelShr[0]) &
                    Configure->PixelMasks.RedMask) |
                   (((Uint32 << Configure->PixelShl[1]) >> Configure->PixelShr[1]) &
                    Configure->PixelMasks.GreenMask) |
                   (((Uint32 << Configure->PixelShl[2]) >> Configure->PixelShr[2]) &
                    Configure->PixelMasks.BlueMask)
                   );
      }

      CopyMem (Destination, Configure->LineBuffer, WidthInBytes);
      Source      += PixelsPerScanLine;
      Destination += DeltaPixels;
      Height--;
    }
  }

  return EFI_SUCCESS;
}

RETURN_STATUS
BlitLibVideoToBuffer90 (
  IN     OC_BLIT_CONFIGURE           *Configure,
  OUT EFI_GRAPHICS_OUTPUT_BLT_PIXEL  *BltBuffer,
  IN     UINTN                       SourceX,
  IN     UINTN                       SourceY,
  IN     UINTN                       DestinationX,
  IN     UINTN                       DestinationY,
  IN     UINTN                       Width,
  IN     UINTN                       Height,
  IN     UINTN                       DeltaPixels
  )
{
  UINT32  *Source;
  UINT32  *Destination;
  UINT32  *SourceWalker;
  UINT32  *DestinationWalker;
  UINTN   IndexX;
  UINTN   PixelsPerScanLine;
  UINT32  Uint32;

  PixelsPerScanLine = Configure->PixelsPerScanLine;

  Destination = (UINT32 *)BltBuffer
                + DestinationY * DeltaPixels + DestinationX;
  Source = (UINT32 *)Configure->FrameBuffer
           + (Configure->Width - SourceY - 1);

  if (Configure->PixelFormat == PixelBlueGreenRedReserved8BitPerColor) {
    while (Height > 0) {
      DestinationWalker = Destination;
      SourceWalker      = Source;
      for (IndexX = SourceX; IndexX < SourceX + Width; IndexX++) {
        *DestinationWalker++ = SourceWalker[IndexX * PixelsPerScanLine];
      }

      Source--;
      Destination += DeltaPixels;
      Height--;
    }
  } else {
    while (Height > 0) {
      DestinationWalker = Destination;
      SourceWalker      = Source;
      for (IndexX = SourceX; IndexX < SourceX + Width; IndexX++) {
        Uint32               = SourceWalker[IndexX * PixelsPerScanLine];
        *DestinationWalker++ =
          (UINT32)(
                   (((Uint32 << Configure->PixelShl[0]) >> Configure->PixelShr[0]) &
                    Configure->PixelMasks.RedMask) |
                   (((Uint32 << Configure->PixelShl[1]) >> Configure->PixelShr[1]) &
                    Configure->PixelMasks.GreenMask) |
                   (((Uint32 << Configure->PixelShl[2]) >> Configure->PixelShr[2]) &
                    Configure->PixelMasks.BlueMask)
                   );
      }

      Source--;
      Destination += DeltaPixels;
      Height--;
    }
  }

  return EFI_SUCCESS;
}

RETURN_STATUS
BlitLibVideoToBuffer180 (
  IN     OC_BLIT_CONFIGURE           *Configure,
  OUT EFI_GRAPHICS_OUTPUT_BLT_PIXEL  *BltBuffer,
  IN     UINTN                       SourceX,
  IN     UINTN                       SourceY,
  IN     UINTN                       DestinationX,
  IN     UINTN                       DestinationY,
  IN     UINTN                       Width,
  IN     UINTN                       Height,
  IN     UINTN                       DeltaPixels
  )
{
  UINT32  *Source;
  UINT32  *Destination;
  UINT32  *SourceWalker;
  UINT32  *DestinationWalker;
  UINTN   IndexX;
  UINTN   PixelsPerScanLine;
  UINT32  Uint32;

  PixelsPerScanLine = Configure->PixelsPerScanLine;

  SourceX = Configure->Width  - SourceX - Width;
  SourceY = Configure->Height - SourceY - Height;

  Destination = (UINT32 *)BltBuffer
                + DestinationY * DeltaPixels + DestinationX;
  Source = (UINT32 *)Configure->FrameBuffer
           + (SourceY + (Height - 1)) * PixelsPerScanLine + SourceX;

  if (Configure->PixelFormat == PixelBlueGreenRedReserved8BitPerColor) {
    while (Height > 0) {
      DestinationWalker = Destination;
      SourceWalker      = Source + (Width - 1);
      for (IndexX = 0; IndexX < Width; IndexX++) {
        *DestinationWalker++ = *SourceWalker--;
      }

      Source      -= PixelsPerScanLine;
      Destination += DeltaPixels;
      Height--;
    }
  } else {
    while (Height > 0) {
      DestinationWalker = Destination;
      SourceWalker      = Source + (Width - 1);
      for (IndexX = 0; IndexX < Width; IndexX++) {
        Uint32               = *SourceWalker--;
        *DestinationWalker++ =
          (UINT32)(
                   (((Uint32 << Configure->PixelShl[0]) >> Configure->PixelShr[0]) &
                    Configure->PixelMasks.RedMask) |
                   (((Uint32 << Configure->PixelShl[1]) >> Configure->PixelShr[1]) &
                    Configure->PixelMasks.GreenMask) |
                   (((Uint32 << Configure->PixelShl[2]) >> Configure->PixelShr[2]) &
                    Configure->PixelMasks.BlueMask)
                   );
      }

      Source      -= PixelsPerScanLine;
      Destination += DeltaPixels;
      Height--;
    }
  }

  return EFI_SUCCESS;
}

RETURN_STATUS
BlitLibVideoToBuffer270 (
  IN     OC_BLIT_CONFIGURE           *Configure,
  OUT EFI_GRAPHICS_OUTPUT_BLT_PIXEL  *BltBuffer,
  IN     UINTN                       SourceX,
  IN     UINTN                       SourceY,
  IN     UINTN                       DestinationX,
  IN     UINTN                       DestinationY,
  IN     UINTN                       Width,
  IN     UINTN                       Height,
  IN     UINTN                       DeltaPixels
  )
{
  UINT32  *Source;
  UINT32  *Destination;
  UINT32  *SourceWalker;
  UINT32  *DestinationWalker;
  UINTN   IndexX;
  UINTN   LastX;
  UINTN   PixelsPerScanLine;
  UINT32  Uint32;

  PixelsPerScanLine = Configure->PixelsPerScanLine;

  Destination = (UINT32 *)BltBuffer
                + DestinationY * DeltaPixels + DestinationX;
  Source = (UINT32 *)Configure->FrameBuffer
           + SourceY;

  if (Configure->PixelFormat == PixelBlueGreenRedReserved8BitPerColor) {
    while (Height > 0) {
      DestinationWalker = Destination;
      SourceWalker      = Source;
      LastX             = Configure->Height - SourceX - 1;
      for (IndexX = 0; IndexX < Width; IndexX++) {
        *DestinationWalker++ = SourceWalker[(LastX - IndexX) * PixelsPerScanLine];
      }

      Source++;
      Destination += DeltaPixels;
      Height--;
    }
  } else {
    while (Height > 0) {
      DestinationWalker = Destination;
      SourceWalker      = Source;
      LastX             = Configure->Height - SourceX - 1;
      for (IndexX = 0; IndexX < Width; IndexX++) {
        Uint32               = SourceWalker[(LastX - IndexX) * PixelsPerScanLine];
        *DestinationWalker++ =
          (UINT32)(
                   (((Uint32 << Configure->PixelShl[0]) >> Configure->PixelShr[0]) &
                    Configure->PixelMasks.RedMask) |
                   (((Uint32 << Configure->PixelShl[1]) >> Configure->PixelShr[1]) &
                    Configure->PixelMasks.GreenMask) |
                   (((Uint32 << Configure->PixelShl[2]) >> Configure->PixelShr[2]) &
                    Configure->PixelMasks.BlueMask)
                   );
      }

      Source++;
      Destination += DeltaPixels;
      Height--;
    }
  }

  return EFI_SUCCESS;
}