MaxstARUtil.java

/*
 * Copyright 2016 Maxst, Inc. All Rights Reserved.
 */

package com.maxst.ar;

import android.content.res.AssetManager;
import android.graphics.Bitmap;
import android.graphics.BitmapFactory;

import junit.framework.Assert;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

public class MaxstARUtil {

    public static byte[] readYuvBytesFromFile(String srcFile) {
        File file = new File(srcFile);
        int size = (int) file.length();
        byte[] bytes = new byte[size];
        try {
            BufferedInputStream buf = new BufferedInputStream(new FileInputStream(file));
            buf.read(bytes, 0, bytes.length);
            buf.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return bytes;
    }

    public static void writeYuvBytesToFile(byte[] input, int width, int height, int format, String dstFile) {
        FileOutputStream fos = null;
        try {
            fos = new FileOutputStream(new File(dstFile));
            fos.write(intToByteArray(width), 0, 4);
            fos.write(intToByteArray(height), 0, 4);
            fos.write(intToByteArray(format), 0, 4);
            fos.write(input, 0, input.length);
            fos.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    private static void copyFile(InputStream in, OutputStream out) throws IOException {
        byte[] buffer = new byte[1024];
        int read;
        while ((read = in.read(buffer)) != -1) {
            out.write(buffer, 0, read);
        }
    }

    public static Bitmap getBitmapFromAsset(String fileName, AssetManager assets) {
        InputStream inputStream = null;
        try {
            inputStream = assets.open(fileName, AssetManager.ACCESS_BUFFER);

            BufferedInputStream bufferedStream = new BufferedInputStream(
                    inputStream);
            return BitmapFactory.decodeStream(bufferedStream);
        } catch (IOException e) {
            return null;
        }
    }

    public static byte[] intToByteArray(int i) {
        byte[] result = new byte[4];

        result[0] = (byte) ((i & 0xFF000000) >> 24);
        result[1] = (byte) ((i & 0x00FF0000) >> 16);
        result[2] = (byte) ((i & 0x0000FF00) >> 8);
        result[3] = (byte) ((i & 0x000000FF));

        return result;
    }

    public static int byteArrayToInt(byte[] bytes, int start) {
        Assert.assertTrue("Byte buffer length is not 4", (bytes.length - start) >= 4);
        return  (bytes[start + 3] & 0xFF) |
                (bytes[start + 2] & 0xFF) << 8 |
                (bytes[start + 1] & 0xFF) << 16 |
                (bytes[start] & 0xFF) << 24;
    }
}