/*

 * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package sun.java2d.metal;

import sun.java2d.SurfaceData;

import sun.java2d.loops.*;

import sun.java2d.pipe.Region;

import sun.java2d.pipe.RenderBuffer;

import sun.java2d.pipe.RenderQueue;

import sun.java2d.pipe.hw.AccelSurface;

import java.awt.*;

import java.awt.geom.AffineTransform;

import java.awt.image.AffineTransformOp;

import java.awt.image.BufferedImage;

import java.awt.image.BufferedImageOp;

import java.lang.annotation.Native;

import java.lang.ref.WeakReference;

import static sun.java2d.pipe.BufferedOpCodes.BLIT;

import static sun.java2d.pipe.BufferedOpCodes.SURFACE_TO_SW_BLIT;

final class MTLBlitLoops {

    static void register() {

        Blit blitIntArgbPreToSurface =

                new MTLSwToSurfaceBlit(SurfaceType.IntArgbPre,

                        MTLSurfaceData.PF_INT_ARGB_PRE);

        Blit blitIntArgbPreToTexture =

                new MTLSwToTextureBlit(SurfaceType.IntArgbPre,

                        MTLSurfaceData.PF_INT_ARGB_PRE);

        TransformBlit transformBlitIntArgbPreToSurface =

                new MTLSwToSurfaceTransform(SurfaceType.IntArgbPre,

                        MTLSurfaceData.PF_INT_ARGB_PRE);

        MTLSurfaceToSwBlit blitSurfaceToIntArgbPre =

                new MTLSurfaceToSwBlit(SurfaceType.IntArgbPre,

                        MTLSurfaceData.PF_INT_ARGB_PRE);

        GraphicsPrimitive[] primitives = {

                // surface->surface ops

                new MTLSurfaceToSurfaceBlit(),

                new MTLSurfaceToSurfaceScale(),

                new MTLSurfaceToSurfaceTransform(),

                // render-to-texture surface->surface ops

                new MTLRTTSurfaceToSurfaceBlit(),

                new MTLRTTSurfaceToSurfaceScale(),

                new MTLRTTSurfaceToSurfaceTransform(),

                // surface->sw ops

                new MTLSurfaceToSwBlit(SurfaceType.IntArgb,

                        MTLSurfaceData.PF_INT_ARGB),

                blitSurfaceToIntArgbPre,

                // sw->surface ops

                blitIntArgbPreToSurface,

                new MTLSwToSurfaceBlit(SurfaceType.IntRgb,

                        MTLSurfaceData.PF_INT_RGB),

                new MTLSwToSurfaceBlit(SurfaceType.IntRgbx,

                        MTLSurfaceData.PF_INT_RGBX),

                new MTLSwToSurfaceBlit(SurfaceType.IntBgr,

                        MTLSurfaceData.PF_INT_BGR),

                new MTLSwToSurfaceBlit(SurfaceType.IntBgrx,

                        MTLSurfaceData.PF_INT_BGRX),

                new MTLSwToSurfaceBlit(SurfaceType.ThreeByteBgr,

                        MTLSurfaceData.PF_3BYTE_BGR),

                new MTLSwToSurfaceBlit(SurfaceType.Ushort565Rgb,

                        MTLSurfaceData.PF_USHORT_565_RGB),

                new MTLSwToSurfaceBlit(SurfaceType.Ushort555Rgb,

                        MTLSurfaceData.PF_USHORT_555_RGB),

                new MTLSwToSurfaceBlit(SurfaceType.Ushort555Rgbx,

                        MTLSurfaceData.PF_USHORT_555_RGBX),

                new MTLSwToSurfaceBlit(SurfaceType.ByteGray,

                        MTLSurfaceData.PF_BYTE_GRAY),

                new MTLSwToSurfaceBlit(SurfaceType.UshortGray,

                        MTLSurfaceData.PF_USHORT_GRAY),

                new MTLGeneralBlit(MTLSurfaceData.MTLSurface,

                        CompositeType.AnyAlpha,

                        blitIntArgbPreToSurface),

                new MTLAnyCompositeBlit(MTLSurfaceData.MTLSurface,

                        blitSurfaceToIntArgbPre,

                        blitSurfaceToIntArgbPre,

                        blitIntArgbPreToSurface),

                new MTLAnyCompositeBlit(SurfaceType.Any,

                        null,

                        blitSurfaceToIntArgbPre,

                        blitIntArgbPreToSurface),

                new MTLSwToSurfaceScale(SurfaceType.IntRgb,

                        MTLSurfaceData.PF_INT_RGB),

                new MTLSwToSurfaceScale(SurfaceType.IntRgbx,

                        MTLSurfaceData.PF_INT_RGBX),

                new MTLSwToSurfaceScale(SurfaceType.IntBgr,

                        MTLSurfaceData.PF_INT_BGR),

                new MTLSwToSurfaceScale(SurfaceType.IntBgrx,

                        MTLSurfaceData.PF_INT_BGRX),

                new MTLSwToSurfaceScale(SurfaceType.ThreeByteBgr,

                        MTLSurfaceData.PF_3BYTE_BGR),

                new MTLSwToSurfaceScale(SurfaceType.Ushort565Rgb,

                        MTLSurfaceData.PF_USHORT_565_RGB),

                new MTLSwToSurfaceScale(SurfaceType.Ushort555Rgb,

                        MTLSurfaceData.PF_USHORT_555_RGB),

                new MTLSwToSurfaceScale(SurfaceType.Ushort555Rgbx,

                        MTLSurfaceData.PF_USHORT_555_RGBX),

                new MTLSwToSurfaceScale(SurfaceType.ByteGray,

                        MTLSurfaceData.PF_BYTE_GRAY),

                new MTLSwToSurfaceScale(SurfaceType.UshortGray,

                        MTLSurfaceData.PF_USHORT_GRAY),

                new MTLSwToSurfaceScale(SurfaceType.IntArgbPre,

                        MTLSurfaceData.PF_INT_ARGB_PRE),

                new MTLSwToSurfaceTransform(SurfaceType.IntRgb,

                        MTLSurfaceData.PF_INT_RGB),

                new MTLSwToSurfaceTransform(SurfaceType.IntRgbx,

                        MTLSurfaceData.PF_INT_RGBX),

                new MTLSwToSurfaceTransform(SurfaceType.IntBgr,

                        MTLSurfaceData.PF_INT_BGR),

                new MTLSwToSurfaceTransform(SurfaceType.IntBgrx,

                        MTLSurfaceData.PF_INT_BGRX),

                new MTLSwToSurfaceTransform(SurfaceType.ThreeByteBgr,

                        MTLSurfaceData.PF_3BYTE_BGR),

                new MTLSwToSurfaceTransform(SurfaceType.Ushort565Rgb,

                        MTLSurfaceData.PF_USHORT_565_RGB),

                new MTLSwToSurfaceTransform(SurfaceType.Ushort555Rgb,

                        MTLSurfaceData.PF_USHORT_555_RGB),

                new MTLSwToSurfaceTransform(SurfaceType.Ushort555Rgbx,

                        MTLSurfaceData.PF_USHORT_555_RGBX),

                new MTLSwToSurfaceTransform(SurfaceType.ByteGray,

                        MTLSurfaceData.PF_BYTE_GRAY),

                new MTLSwToSurfaceTransform(SurfaceType.UshortGray,

                        MTLSurfaceData.PF_USHORT_GRAY),

                transformBlitIntArgbPreToSurface,

                new MTLGeneralTransformedBlit(transformBlitIntArgbPreToSurface),

                // texture->surface ops

                new MTLTextureToSurfaceBlit(),

                new MTLTextureToSurfaceScale(),

                new MTLTextureToSurfaceTransform(),

                // sw->texture ops

                blitIntArgbPreToTexture,

                new MTLSwToTextureBlit(SurfaceType.IntRgb,

                        MTLSurfaceData.PF_INT_RGB),

                new MTLSwToTextureBlit(SurfaceType.IntRgbx,

                        MTLSurfaceData.PF_INT_RGBX),

                new MTLSwToTextureBlit(SurfaceType.IntBgr,

                        MTLSurfaceData.PF_INT_BGR),

                new MTLSwToTextureBlit(SurfaceType.IntBgrx,

                        MTLSurfaceData.PF_INT_BGRX),

                new MTLSwToTextureBlit(SurfaceType.ThreeByteBgr,

                        MTLSurfaceData.PF_3BYTE_BGR),

                new MTLSwToTextureBlit(SurfaceType.Ushort565Rgb,

                        MTLSurfaceData.PF_USHORT_565_RGB),

                new MTLSwToTextureBlit(SurfaceType.Ushort555Rgb,

                        MTLSurfaceData.PF_USHORT_555_RGB),

                new MTLSwToTextureBlit(SurfaceType.Ushort555Rgbx,

                        MTLSurfaceData.PF_USHORT_555_RGBX),

                new MTLSwToTextureBlit(SurfaceType.ByteGray,

                        MTLSurfaceData.PF_BYTE_GRAY),

                new MTLSwToTextureBlit(SurfaceType.UshortGray,

                        MTLSurfaceData.PF_USHORT_GRAY),

                new MTLGeneralBlit(MTLSurfaceData.MTLTexture,

                        CompositeType.SrcNoEa,

                        blitIntArgbPreToTexture),

        };

        GraphicsPrimitiveMgr.register(primitives);

    }

    /**

     * The following offsets are used to pack the parameters in

     * createPackedParams().  (They are also used at the native level when

     * unpacking the params.)

     */

    @Native private static final int OFFSET_SRCTYPE = 16;

    @Native private static final int OFFSET_HINT    =  8;

    @Native private static final int OFFSET_TEXTURE =  3;

    @Native private static final int OFFSET_RTT     =  2;

    @Native private static final int OFFSET_XFORM   =  1;

    @Native private static final int OFFSET_ISOBLIT =  0;

    /**

     * Packs the given parameters into a single int value in order to save

     * space on the rendering queue.

     */

    private static int createPackedParams(boolean isoblit, boolean texture,

                                          boolean rtt, boolean xform,

                                          int hint, int srctype)

    {

        return

                ((srctype           << OFFSET_SRCTYPE) |

                        (hint              << OFFSET_HINT   ) |

                        ((texture ? 1 : 0) << OFFSET_TEXTURE) |

                        ((rtt     ? 1 : 0) << OFFSET_RTT    ) |

                        ((xform   ? 1 : 0) << OFFSET_XFORM  ) |

                        ((isoblit ? 1 : 0) << OFFSET_ISOBLIT));

    }

    /**

     * Enqueues a BLIT operation with the given parameters.  Note that the

     * RenderQueue lock must be held before calling this method.

     */

    private static void enqueueBlit(RenderQueue rq,

                                    SurfaceData src, SurfaceData dst,

                                    int packedParams,

                                    int sx1, int sy1,

                                    int sx2, int sy2,

                                    double dx1, double dy1,

                                    double dx2, double dy2)

    {

        // assert rq.lock.isHeldByCurrentThread();

        RenderBuffer buf = rq.getBuffer();

        rq.ensureCapacityAndAlignment(72, 24);

        buf.putInt(BLIT);

        buf.putInt(packedParams);

        buf.putInt(sx1).putInt(sy1);

        buf.putInt(sx2).putInt(sy2);

        buf.putDouble(dx1).putDouble(dy1);

        buf.putDouble(dx2).putDouble(dy2);

        buf.putLong(src.getNativeOps());

        buf.putLong(dst.getNativeOps());

    }

    static void Blit(SurfaceData srcData, SurfaceData dstData,

                     Composite comp, Region clip,

                     AffineTransform xform, int hint,

                     int sx1, int sy1,

                     int sx2, int sy2,

                     double dx1, double dy1,

                     double dx2, double dy2,

                     int srctype, boolean texture)

    {

        int ctxflags = 0;

        if (srcData.getTransparency() == Transparency.OPAQUE) {

            ctxflags |= MTLContext.SRC_IS_OPAQUE;

        }

        MTLRenderQueue rq = MTLRenderQueue.getInstance();

        rq.lock();

        try {

            // make sure the RenderQueue keeps a hard reference to the

            // source (sysmem) SurfaceData to prevent it from being

            // disposed while the operation is processed on the QFT

            rq.addReference(srcData);

            MTLSurfaceData oglDst = (MTLSurfaceData)dstData;

            if (texture) {

                // make sure we have a current context before uploading

                // the sysmem data to the texture object

                MTLGraphicsConfig gc = oglDst.getMTLGraphicsConfig();

                MTLContext.setScratchSurface(gc);

            } else {

                MTLContext.validateContext(oglDst, oglDst,

                        clip, comp, xform, null, null,

                        ctxflags);

            }

            int packedParams = createPackedParams(false, texture,

                    false, xform != null,

                    hint, srctype);

            enqueueBlit(rq, srcData, dstData,

                    packedParams,

                    sx1, sy1, sx2, sy2,

                    dx1, dy1, dx2, dy2);

            // always flush immediately, since we (currently) have no means

            // of tracking changes to the system memory surface

            rq.flushNow();

        } finally {

            rq.unlock();

        }

    }

    /**

     * Note: The srcImg and biop parameters are only used when invoked

     * from the MTLBufImgOps.renderImageWithOp() method; in all other cases,

     * this method can be called with null values for those two parameters,

     * and they will be effectively ignored.

     */

    static void IsoBlit(SurfaceData srcData, SurfaceData dstData,

                        BufferedImage srcImg, BufferedImageOp biop,

                        Composite comp, Region clip,

                        AffineTransform xform, int hint,

                        int sx1, int sy1,

                        int sx2, int sy2,

                        double dx1, double dy1,

                        double dx2, double dy2,

                        boolean texture)

    {

        int ctxflags = 0;

        if (srcData.getTransparency() == Transparency.OPAQUE) {

            ctxflags |= MTLContext.SRC_IS_OPAQUE;

        }

        MTLRenderQueue rq = MTLRenderQueue.getInstance();

        rq.lock();

        try {

            MTLSurfaceData oglSrc = (MTLSurfaceData)srcData;

            MTLSurfaceData oglDst = (MTLSurfaceData)dstData;

            int srctype = oglSrc.getType();

            boolean rtt;

            MTLSurfaceData srcCtxData;

            if (srctype == MTLSurfaceData.TEXTURE) {

                // the source is a regular texture object; we substitute

                // the destination surface for the purposes of making a

                // context current

                rtt = false;

                srcCtxData = oglDst;

            } else {

                // the source is a pbuffer, backbuffer, or render-to-texture

                // surface; we set rtt to true to differentiate this kind

                // of surface from a regular texture object

                rtt = true;

                if (srctype == AccelSurface.RT_TEXTURE) {

                    srcCtxData = oglDst;

                } else {

                    srcCtxData = oglSrc;

                }

            }

            MTLContext.validateContext(srcCtxData, oglDst,

                    clip, comp, xform, null, null,

                    ctxflags);

            if (biop != null) {

                MTLBufImgOps.enableBufImgOp(rq, oglSrc, srcImg, biop);

            }

            int packedParams = createPackedParams(true, texture,

                    rtt, xform != null,

                    hint, 0 /*unused*/);

            enqueueBlit(rq, srcData, dstData,

                    packedParams,

                    sx1, sy1, sx2, sy2,

                    dx1, dy1, dx2, dy2);

            if (biop != null) {

                MTLBufImgOps.disableBufImgOp(rq, biop);

            }

            if (rtt && oglDst.isOnScreen()) {

                // we only have to flush immediately when copying from a

                // (non-texture) surface to the screen; otherwise Swing apps

                // might appear unresponsive until the auto-flush completes

                rq.flushNow();

            }

        } finally {

            rq.unlock();

        }

    }

}

class MTLSurfaceToSurfaceBlit extends Blit {

    MTLSurfaceToSurfaceBlit() {

        super(MTLSurfaceData.MTLSurface,

                CompositeType.AnyAlpha,

                MTLSurfaceData.MTLSurface);

    }

    public void Blit(SurfaceData src, SurfaceData dst,

                     Composite comp, Region clip,

                     int sx, int sy, int dx, int dy, int w, int h)

    {

        MTLBlitLoops.IsoBlit(src, dst,

                null, null,

                comp, clip, null,

                AffineTransformOp.TYPE_NEAREST_NEIGHBOR,

                sx, sy, sx+w, sy+h,

                dx, dy, dx+w, dy+h,

                false);

    }

}

class MTLSurfaceToSurfaceScale extends ScaledBlit {

    MTLSurfaceToSurfaceScale() {

        super(MTLSurfaceData.MTLSurface,

                CompositeType.AnyAlpha,

                MTLSurfaceData.MTLSurface);

    }

    public void Scale(SurfaceData src, SurfaceData dst,

                      Composite comp, Region clip,

                      int sx1, int sy1,

                      int sx2, int sy2,

                      double dx1, double dy1,

                      double dx2, double dy2)

    {

        MTLBlitLoops.IsoBlit(src, dst,

                null, null,

                comp, clip, null,

                AffineTransformOp.TYPE_NEAREST_NEIGHBOR,

                sx1, sy1, sx2, sy2,

                dx1, dy1, dx2, dy2,

                false);

    }

}

class MTLSurfaceToSurfaceTransform extends TransformBlit {

    MTLSurfaceToSurfaceTransform() {

        super(MTLSurfaceData.MTLSurface,

                CompositeType.AnyAlpha,

                MTLSurfaceData.MTLSurface);

    }

    public void Transform(SurfaceData src, SurfaceData dst,

                          Composite comp, Region clip,

                          AffineTransform at, int hint,

                          int sx, int sy, int dx, int dy,

                          int w, int h)

    {

        MTLBlitLoops.IsoBlit(src, dst,

                null, null,

                comp, clip, at, hint,

                sx, sy, sx+w, sy+h,

                dx, dy, dx+w, dy+h,

                false);

    }

}

class MTLRTTSurfaceToSurfaceBlit extends Blit {

    MTLRTTSurfaceToSurfaceBlit() {

        super(MTLSurfaceData.MTLSurfaceRTT,

                CompositeType.AnyAlpha,

                MTLSurfaceData.MTLSurface);

    }

    public void Blit(SurfaceData src, SurfaceData dst,

                     Composite comp, Region clip,

                     int sx, int sy, int dx, int dy, int w, int h)

    {

        MTLBlitLoops.IsoBlit(src, dst,

                null, null,

                comp, clip, null,

                AffineTransformOp.TYPE_NEAREST_NEIGHBOR,

                sx, sy, sx+w, sy+h,

                dx, dy, dx+w, dy+h,

                true);

    }

}

class MTLRTTSurfaceToSurfaceScale extends ScaledBlit {

    MTLRTTSurfaceToSurfaceScale() {

        super(MTLSurfaceData.MTLSurfaceRTT,

                CompositeType.AnyAlpha,

                MTLSurfaceData.MTLSurface);

    }

    public void Scale(SurfaceData src, SurfaceData dst,

                      Composite comp, Region clip,

                      int sx1, int sy1,

                      int sx2, int sy2,

                      double dx1, double dy1,

                      double dx2, double dy2)

    {

        MTLBlitLoops.IsoBlit(src, dst,

                null, null,

                comp, clip, null,

                AffineTransformOp.TYPE_NEAREST_NEIGHBOR,

                sx1, sy1, sx2, sy2,

                dx1, dy1, dx2, dy2,

                true);

    }

}

class MTLRTTSurfaceToSurfaceTransform extends TransformBlit {

    MTLRTTSurfaceToSurfaceTransform() {

        super(MTLSurfaceData.MTLSurfaceRTT,

                CompositeType.AnyAlpha,

                MTLSurfaceData.MTLSurface);

    }

    public void Transform(SurfaceData src, SurfaceData dst,

                          Composite comp, Region clip,

                          AffineTransform at, int hint,

                          int sx, int sy, int dx, int dy, int w, int h)

    {

        MTLBlitLoops.IsoBlit(src, dst,

                null, null,

                comp, clip, at, hint,

                sx, sy, sx+w, sy+h,

                dx, dy, dx+w, dy+h,

                true);

    }

}

final class MTLSurfaceToSwBlit extends Blit {

    private final int typeval;

    private WeakReference<SurfaceData> srcTmp;

    // destination will actually be ArgbPre or Argb

    MTLSurfaceToSwBlit(final SurfaceType dstType, final int typeval) {

        super(MTLSurfaceData.MTLSurface,

                CompositeType.SrcNoEa,

                dstType);

        this.typeval = typeval;

    }

    private synchronized void complexClipBlit(SurfaceData src, SurfaceData dst,