package com.basic.security.utils;

import java.nio.ByteBuffer;

public class RotateUtil {
    byte[] yuv3;
    byte[] yuv2;
    byte[] yuv1;

    public static ByteBuffer[] nv21ToI420(byte[] data, int width, int height) {
        ByteBuffer byteBufferArray[] = new ByteBuffer[3];
        int total = width * height;
        byte[] ret = new byte[total];
        byte[] u = new byte[total / 4];
        byte[] v = new byte[total / 4];
        ByteBuffer bufferY = ByteBuffer.wrap(ret, 0, total);
        ByteBuffer bufferU = ByteBuffer.wrap(u, 0, total / 4);
        ByteBuffer bufferV = ByteBuffer.wrap(v, 0, total / 4);
//
//        ByteBuffer bufferY = ByteBuffer.wrap(ret, 0, total);
//        ByteBuffer bufferU = ByteBuffer.wrap(ret, total, total / 4);
//        ByteBuffer bufferV = ByteBuffer.wrap(ret, total + total / 4, total / 4);
        bufferY.put(data, 0, total);
        for (int i = total; i < data.length; i += 2) {
            bufferV.put(data[i]);
            bufferU.put(data[i + 1]);
        }
        byteBufferArray[0] = bufferY;
        byteBufferArray[1] = bufferU;
        byteBufferArray[2] = bufferV;
//        return ret;
        return byteBufferArray;
    }

    public static void mirror(byte[] src, int w, int h) { //src是原始yuv数组
        int i;
        int index;
        byte temp;
        int a, b;
        //mirror y
        for (i = 0; i < h; i++) {
            a = i * w;
            b = (i + 1) * w - 1;
            while (a < b) {
                temp = src[a];
                src[a] = src[b];
                src[b] = temp;
                a++;
                b--;
            }
        }
        // mirror u and v
        index = w * h;
        for (i = 0; i < h / 2; i++) {
            a = i * w;
            b = (i + 1) * w - 2;
            while (a < b) {
                temp = src[a + index];
                src[a + index] = src[b + index];
                src[b + index] = temp;
                temp = src[a + index + 1];
                src[a + index + 1] = src[b + index + 1];
                src[b + index + 1] = temp;
                a += 2;
                b -= 2;
            }
        }
    }

    public byte[] rotateYUV420Degree90(byte[] data, int imageWidth, int imageHeight) {
        if (yuv3 == null) {
            yuv3 = new byte[imageWidth * imageHeight * 3 / 2];
        }
        int i = 0;
        for (int x = 0; x < imageWidth; x++) {
            for (int y = imageHeight - 1; y >= 0; y--) {
                yuv3[i] = data[y * imageWidth + x];
                i++;
            }
        }
        i = imageWidth * imageHeight * 3 / 2 - 1;
        for (int x = imageWidth - 1; x > 0; x = x - 2) {
            for (int y = 0; y < imageHeight / 2; y++) {
                yuv3[i] = data[(imageWidth * imageHeight) + (y * imageWidth) + x];
                i--;
                yuv3[i] = data[(imageWidth * imageHeight) + (y * imageWidth) + (x - 1)];
                i--;
            }
        }
        return yuv3;
    }

    public byte[] rotateYUV420Degree180(byte[] data, int imageWidth, int imageHeight) {
        if (yuv2 == null) {
            yuv2 = new byte[imageWidth * imageHeight * 3 / 2];
        }
        int i = 0;
        int count = 0;
        for (i = imageWidth * imageHeight - 1; i >= 0; i--) {
            yuv2[count] = data[i];
            count++;
        }
        i = imageWidth * imageHeight * 3 / 2 - 1;
        for (i = imageWidth * imageHeight * 3 / 2 - 1; i >= imageWidth
                * imageHeight; i -= 2) {
            yuv2[count++] = data[i - 1];
            yuv2[count++] = data[i];
        }
        return yuv2;
    }

    public byte[] rotateYUV420Degree270(byte[] data, int imageWidth,
                                        int imageHeight) {
        if (yuv1 == null) {
            yuv1 = new byte[imageWidth * imageHeight * 3 / 2];
        }
        int nWidth = 0, nHeight = 0;
        int wh = 0;
        int uvHeight = 0;
        if (imageWidth != nWidth || imageHeight != nHeight) {
            nWidth = imageWidth;
            nHeight = imageHeight;
            wh = imageWidth * imageHeight;
            uvHeight = imageHeight >> 1;
        }
        int k = 0;
        for (int i = 0; i < imageWidth; i++) {
            int nPos = 0;
            for (int j = 0; j < imageHeight; j++) {
                yuv1[k] = data[nPos + i];
                k++;
                nPos += imageWidth;
            }
        }
        for (int i = 0; i < imageWidth; i += 2) {
            int nPos = wh;
            for (int j = 0; j < uvHeight; j++) {
                yuv1[k] = data[nPos + i];
                yuv1[k + 1] = data[nPos + i + 1];
                k += 2;
                nPos += imageWidth;
            }
        }
        return rotateYUV420Degree180(yuv1, imageWidth, imageHeight);
    }

    private byte[] rotateYUVDegree270(byte[] data, int imageWidth, int imageHeight) {
        byte[] yuv = new byte[imageWidth * imageHeight * 3 / 2];
        // Rotate the Y luma
        int i = 0;
        for (int x = imageWidth - 1; x >= 0; x--) {
            for (int y = 0; y < imageHeight; y++) {
                yuv[i] = data[y * imageWidth + x];
                i++;
            }
        }// Rotate the U and V color components
        i = imageWidth * imageHeight;
        for (int x = imageWidth - 1; x > 0; x = x - 2) {
            for (int y = 0; y < imageHeight / 2; y++) {
                yuv[i] = data[(imageWidth * imageHeight) + (y * imageWidth) + (x - 1)];
                i++;
                yuv[i] = data[(imageWidth * imageHeight) + (y * imageWidth) + x];
                i++;
            }
        }
        return yuv;
    }
}