/*

 * Copyright (c) 2007, 2008, 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.d3d;

import java.awt.Composite;

import java.awt.Transparency;

import java.awt.geom.AffineTransform;

import java.awt.image.AffineTransformOp;

import java.awt.image.BufferedImage;

import java.awt.image.BufferedImageOp;

import java.lang.ref.WeakReference;

import sun.java2d.ScreenUpdateManager;

import sun.java2d.SurfaceData;

import sun.java2d.loops.Blit;

import sun.java2d.loops.CompositeType;

import sun.java2d.loops.GraphicsPrimitive;

import sun.java2d.loops.GraphicsPrimitiveMgr;

import sun.java2d.loops.ScaledBlit;

import sun.java2d.loops.SurfaceType;

import sun.java2d.loops.TransformBlit;

import sun.java2d.pipe.Region;

import sun.java2d.pipe.RenderBuffer;

import sun.java2d.pipe.RenderQueue;

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

import sun.java2d.windows.GDIWindowSurfaceData;

class D3DBlitLoops {

    static void register() {

        Blit blitIntArgbPreToSurface =

            new D3DSwToSurfaceBlit(SurfaceType.IntArgbPre,

                                   D3DSurfaceData.ST_INT_ARGB_PRE);

        Blit blitIntArgbPreToTexture =

            new D3DSwToTextureBlit(SurfaceType.IntArgbPre,

                                   D3DSurfaceData.ST_INT_ARGB_PRE);

        GraphicsPrimitive[] primitives = {

            // prevent D3DSurface -> Screen blits

            new D3DSurfaceToGDIWindowSurfaceBlit(),

            new D3DSurfaceToGDIWindowSurfaceScale(),

            new D3DSurfaceToGDIWindowSurfaceTransform(),

            // surface->surface ops

            new D3DSurfaceToSurfaceBlit(),

            new D3DSurfaceToSurfaceScale(),

            new D3DSurfaceToSurfaceTransform(),

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

            new D3DRTTSurfaceToSurfaceBlit(),

            new D3DRTTSurfaceToSurfaceScale(),

            new D3DRTTSurfaceToSurfaceTransform(),

            // surface->sw ops

            new D3DSurfaceToSwBlit(SurfaceType.IntArgb,

                                   D3DSurfaceData.ST_INT_ARGB),

            // sw->surface ops

            blitIntArgbPreToSurface,

            new D3DSwToSurfaceBlit(SurfaceType.IntArgb,

                                   D3DSurfaceData.ST_INT_ARGB),

            new D3DSwToSurfaceBlit(SurfaceType.IntRgb,

                                   D3DSurfaceData.ST_INT_RGB),

            new D3DSwToSurfaceBlit(SurfaceType.IntBgr,

                                   D3DSurfaceData.ST_INT_BGR),

            new D3DSwToSurfaceBlit(SurfaceType.ThreeByteBgr,

                                   D3DSurfaceData.ST_3BYTE_BGR),

            new D3DSwToSurfaceBlit(SurfaceType.Ushort565Rgb,

                                   D3DSurfaceData.ST_USHORT_565_RGB),

            new D3DSwToSurfaceBlit(SurfaceType.Ushort555Rgb,

                                   D3DSurfaceData.ST_USHORT_555_RGB),

            new D3DSwToSurfaceBlit(SurfaceType.ByteIndexed,

                                   D3DSurfaceData.ST_BYTE_INDEXED),

            // REMIND: we don't have a native sw loop to back this loop up

//            new D3DSwToSurfaceBlit(SurfaceType.ByteIndexedBm,

//                                   D3DSurfaceData.ST_BYTE_INDEXED_BM),

            new D3DGeneralBlit(D3DSurfaceData.D3DSurface,

                               CompositeType.AnyAlpha,

                               blitIntArgbPreToSurface),

            new D3DSwToSurfaceScale(SurfaceType.IntArgb,

                                    D3DSurfaceData.ST_INT_ARGB),

            new D3DSwToSurfaceScale(SurfaceType.IntArgbPre,

                                    D3DSurfaceData.ST_INT_ARGB_PRE),

            new D3DSwToSurfaceScale(SurfaceType.IntRgb,

                                    D3DSurfaceData.ST_INT_RGB),

            new D3DSwToSurfaceScale(SurfaceType.IntBgr,

                                    D3DSurfaceData.ST_INT_BGR),

            new D3DSwToSurfaceScale(SurfaceType.ThreeByteBgr,

                                    D3DSurfaceData.ST_3BYTE_BGR),

            new D3DSwToSurfaceScale(SurfaceType.Ushort565Rgb,

                                    D3DSurfaceData.ST_USHORT_565_RGB),

            new D3DSwToSurfaceScale(SurfaceType.Ushort555Rgb,

                                    D3DSurfaceData.ST_USHORT_555_RGB),

            new D3DSwToSurfaceScale(SurfaceType.ByteIndexed,

                                    D3DSurfaceData.ST_BYTE_INDEXED),

            // REMIND: we don't have a native sw loop to back this loop up

//            new D3DSwToSurfaceScale(SurfaceType.ByteIndexedBm,

//                                    D3DSurfaceData.ST_BYTE_INDEXED_BM),

            new D3DSwToSurfaceTransform(SurfaceType.IntArgb,

                                        D3DSurfaceData.ST_INT_ARGB),

            new D3DSwToSurfaceTransform(SurfaceType.IntArgbPre,

                                        D3DSurfaceData.ST_INT_ARGB_PRE),

            new D3DSwToSurfaceTransform(SurfaceType.IntRgb,

                                        D3DSurfaceData.ST_INT_RGB),

            new D3DSwToSurfaceTransform(SurfaceType.IntBgr,

                                        D3DSurfaceData.ST_INT_BGR),

            new D3DSwToSurfaceTransform(SurfaceType.ThreeByteBgr,

                                        D3DSurfaceData.ST_3BYTE_BGR),

            new D3DSwToSurfaceTransform(SurfaceType.Ushort565Rgb,

                                        D3DSurfaceData.ST_USHORT_565_RGB),

            new D3DSwToSurfaceTransform(SurfaceType.Ushort555Rgb,

                                        D3DSurfaceData.ST_USHORT_555_RGB),

            new D3DSwToSurfaceTransform(SurfaceType.ByteIndexed,

                                        D3DSurfaceData.ST_BYTE_INDEXED),

            // REMIND: we don't have a native sw loop to back this loop up

//            new D3DSwToSurfaceTransform(SurfaceType.ByteIndexedBm,

//                                        D3DSurfaceData.ST_BYTE_INDEXED_BM),

            // texture->surface ops

            new D3DTextureToSurfaceBlit(),

            new D3DTextureToSurfaceScale(),

            new D3DTextureToSurfaceTransform(),

            // sw->texture ops

            blitIntArgbPreToTexture,

            new D3DSwToTextureBlit(SurfaceType.IntRgb,

                                   D3DSurfaceData.ST_INT_RGB),

            new D3DSwToTextureBlit(SurfaceType.IntArgb,

                                   D3DSurfaceData.ST_INT_ARGB),

            new D3DSwToTextureBlit(SurfaceType.IntBgr,

                                   D3DSurfaceData.ST_INT_BGR),

            new D3DSwToTextureBlit(SurfaceType.ThreeByteBgr,

                                   D3DSurfaceData.ST_3BYTE_BGR),

            new D3DSwToTextureBlit(SurfaceType.Ushort565Rgb,

                                   D3DSurfaceData.ST_USHORT_565_RGB),

            new D3DSwToTextureBlit(SurfaceType.Ushort555Rgb,

                                   D3DSurfaceData.ST_USHORT_555_RGB),

            new D3DSwToTextureBlit(SurfaceType.ByteIndexed,

                                   D3DSurfaceData.ST_BYTE_INDEXED),

            // REMIND: we don't have a native sw loop to back this loop up

//            new D3DSwToTextureBlit(SurfaceType.ByteIndexedBm,

//                                   D3DSurfaceData.ST_BYTE_INDEXED_BM),

            new D3DGeneralBlit(D3DSurfaceData.D3DTexture,

                               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.)

     */

    private static final int OFFSET_SRCTYPE = 16;

    private static final int OFFSET_HINT    =  8;

    private static final int OFFSET_TEXTURE =  3;

    private static final int OFFSET_RTT     =  2;

    private static final int OFFSET_XFORM   =  1;

    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 |= D3DContext.SRC_IS_OPAQUE;

        }

        D3DSurfaceData d3dDst = (D3DSurfaceData)dstData;

        D3DRenderQueue rq = D3DRenderQueue.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);

            if (texture) {

                // make sure we have a current context before uploading

                // the sysmem data to the texture object

                D3DContext.setScratchSurface(d3dDst.getContext());

            } else {

                D3DContext.validateContext(d3dDst, d3dDst,

                                           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();

        }

        if (d3dDst.getType() == D3DSurfaceData.WINDOW) {

            // flush immediately when copying to the screen to improve

            // responsiveness of applications using VI or BI backbuffers

            D3DScreenUpdateManager mgr =

                (D3DScreenUpdateManager)ScreenUpdateManager.getInstance();

            mgr.runUpdateNow();

        }

    }

    /**

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

     * from the D3DBufImgOps.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 |= D3DContext.SRC_IS_OPAQUE;

        }

        D3DSurfaceData d3dDst = (D3DSurfaceData)dstData;

        D3DRenderQueue rq = D3DRenderQueue.getInstance();

        boolean rtt = false;

        rq.lock();

        try {

            D3DSurfaceData d3dSrc = (D3DSurfaceData)srcData;

            int srctype = d3dSrc.getType();

            D3DSurfaceData srcCtxData = d3dSrc;

            if (srctype == D3DSurfaceData.TEXTURE) {

                rtt = false;

            } else {

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

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

                // of surface from a regular texture object

                rtt = true;

            }

            D3DContext.validateContext(srcCtxData, d3dDst,

                                       clip, comp, xform, null, null,

                                       ctxflags);

            if (biop != null) {

                D3DBufImgOps.enableBufImgOp(rq, d3dSrc, 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) {

                D3DBufImgOps.disableBufImgOp(rq, biop);

            }

        } finally {

            rq.unlock();

        }

        if (rtt && (d3dDst.getType() == D3DSurfaceData.WINDOW)) {

            // 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

            D3DScreenUpdateManager mgr =

                (D3DScreenUpdateManager)ScreenUpdateManager.getInstance();

            mgr.runUpdateNow();

        }

    }

}

class D3DSurfaceToSurfaceBlit extends Blit {

    D3DSurfaceToSurfaceBlit() {

        super(D3DSurfaceData.D3DSurface,

              CompositeType.AnyAlpha,

              D3DSurfaceData.D3DSurface);

    }

    public void Blit(SurfaceData src, SurfaceData dst,

                     Composite comp, Region clip,

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

    {

        D3DBlitLoops.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 D3DSurfaceToSurfaceScale extends ScaledBlit {

    D3DSurfaceToSurfaceScale() {

        super(D3DSurfaceData.D3DSurface,

              CompositeType.AnyAlpha,

              D3DSurfaceData.D3DSurface);

    }

    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)

    {

        D3DBlitLoops.IsoBlit(src, dst,

                             null, null,

                             comp, clip, null,

                             AffineTransformOp.TYPE_NEAREST_NEIGHBOR,

                             sx1, sy1, sx2, sy2,

                             dx1, dy1, dx2, dy2,

                             false);

    }

}

class D3DSurfaceToSurfaceTransform extends TransformBlit {

    D3DSurfaceToSurfaceTransform() {

        super(D3DSurfaceData.D3DSurface,

              CompositeType.AnyAlpha,

              D3DSurfaceData.D3DSurface);

    }

    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)

    {

        D3DBlitLoops.IsoBlit(src, dst,

                             null, null,

                             comp, clip, at, hint,

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

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

                             false);

    }

}

class D3DRTTSurfaceToSurfaceBlit extends Blit {

    D3DRTTSurfaceToSurfaceBlit() {

        super(D3DSurfaceData.D3DSurfaceRTT,

              CompositeType.AnyAlpha,

              D3DSurfaceData.D3DSurface);

    }

    public void Blit(SurfaceData src, SurfaceData dst,

                     Composite comp, Region clip,

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

    {

        D3DBlitLoops.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 D3DRTTSurfaceToSurfaceScale extends ScaledBlit {

    D3DRTTSurfaceToSurfaceScale() {

        super(D3DSurfaceData.D3DSurfaceRTT,

              CompositeType.AnyAlpha,

              D3DSurfaceData.D3DSurface);

    }

    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)

    {

        D3DBlitLoops.IsoBlit(src, dst,

                             null, null,

                             comp, clip, null,

                             AffineTransformOp.TYPE_NEAREST_NEIGHBOR,

                             sx1, sy1, sx2, sy2,

                             dx1, dy1, dx2, dy2,

                             true);

    }

}

class D3DRTTSurfaceToSurfaceTransform extends TransformBlit {

    D3DRTTSurfaceToSurfaceTransform() {

        super(D3DSurfaceData.D3DSurfaceRTT,

              CompositeType.AnyAlpha,

              D3DSurfaceData.D3DSurface);

    }

    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)

    {

        D3DBlitLoops.IsoBlit(src, dst,

                             null, null,

                             comp, clip, at, hint,

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

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

                             true);

    }

}

class D3DSurfaceToSwBlit extends Blit {