package org.jcodec.common;
import static java.lang.System.currentTimeMillis;
import static org.junit.Assert.assertArrayEquals;

import org.jcodec.common.io.NIOUtils;
import org.junit.Test;

import java.io.File;
import java.lang.System;
import java.nio.ByteBuffer;

/**
 * This class is part of JCodec ( www.jcodec.org ) This software is distributed
 * under FreeBSD License
 * 
 * @author The JCodec project
 * 
 */
public class TestByteBufferUtil {

    @Test
    public void testSearch() {

        byte[] marker = new byte[] { 0, 0, 1 };

        ByteBuffer buf = ByteBuffer.wrap(new byte[] { 10, 11, 12, 0, 0, 0, 0, 0, 1, 53, 23, 13, 0, 0, 12, 0, 0, 1, 13,
                0, 23, 0, 0, 23, 0, 0, 1, 0, 0, 2, 0, 0, 0, 0, 0, 0, 0, 0, 2, 3, 4 });

        assertArrayEquals(new byte[] { 10, 11, 12, 0, 0, 0 },
                NIOUtils.toArray(NIOUtils.search(buf, 0, marker)));
        assertArrayEquals(new byte[] { 0, 0, 1, 53, 23, 13, 0, 0, 12 },
                NIOUtils.toArray(NIOUtils.search(buf, 1, marker)));
        assertArrayEquals(new byte[] { 0, 0, 1, 13, 0, 23, 0, 0, 23 },
                NIOUtils.toArray(NIOUtils.search(buf, 1, marker)));
        assertArrayEquals(new byte[] { 0, 0, 1, 0, 0, 2, 0, 0, 0, 0, 0, 0, 0, 0, 2, 3, 4 },
                NIOUtils.toArray(NIOUtils.search(buf, 1, marker)));
    }
    
    
    @Test
    public void testSliceVsPut() throws Exception {
        ByteBuffer rawFrame = NIOUtils.fetchFromFile(new File("src/test/resources/mkv/single-frame01.vp8"));
        ByteBuffer newFrame = ByteBuffer.allocate(rawFrame.limit());
        
        long start = currentTimeMillis();
        for (int i=0;i < 10E7; i++){
            newFrame.put(rawFrame);
            newFrame.flip();
        }
        System.out.println((currentTimeMillis()-start)+"ms for put");
        
        start = currentTimeMillis();
        for (int i=0;i < 10E7; i++){
            newFrame = rawFrame.slice();
        }
        System.out.println((currentTimeMillis()-start)+"ms for slice");
    }
}