/*
 * ome.io.nio.itests.PlaneIOUnitTest
 *
 *   Copyright 2006 University of Dundee. All rights reserved.
 *   Use is subject to license terms supplied in LICENSE.txt
 */
package ome.io.nio.itests;

import java.io.File;
import java.io.IOException;

import ome.io.nio.DimensionsOutOfBoundsException;
import ome.io.nio.PixelBuffer;
import ome.util.PixelData;
import ome.io.nio.PixelsService;
import ome.model.core.Pixels;
import ome.server.itests.AbstractManagedContextTest;

import org.apache.commons.codec.binary.Hex;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

/**
 * @author callan
 * 
 */
public class PlaneWriteUnitTest extends AbstractManagedContextTest {
    private PixelsService service;

    private Pixels pixels;

    private PixbufIOFixture baseFixture;

    private PixelBuffer pixbuf;

    private byte[] testPlane;

    private String originalDigest;

    private String ROOT;

    private byte[] getTestPlane() {
        if (testPlane == null) {
            Long planeSize = pixbuf.getPlaneSize();
            testPlane = new byte[planeSize.intValue()];

            for (int i = 0; i < planeSize; i++) {
                if (i % 2 == 0) {
                    testPlane[i] = -128;
                } else {
                    testPlane[i] = 126; // Changed to not match
                    // PixelService.nullPlane -j.m
                }
            }

            byte[] md = Helper.calculateMessageDigest(testPlane);
            originalDigest = Hex.encodeHexString(md);
        }

        return testPlane;
    }

    @Test
    public void testLowerPlaneWrite() throws IOException,
            DimensionsOutOfBoundsException {
        byte[] testPlane = getTestPlane();
        pixbuf.setPlane(testPlane, 0, 0, 0);

        pixbuf = service.getPixelBuffer(pixels);
        PixelData plane = pixbuf.getPlane(0, 0, 0);
        assertNotNull(plane);
        byte[] newMD = Helper.calculateMessageDigest(plane.getData());

        assertEquals(originalDigest, Hex.encodeHexString(newMD));
    }

    @Test
    public void testUpperPlaneWrite() throws IOException,
            DimensionsOutOfBoundsException {
        int z = pixels.getSizeZ() - 1;
        int c = pixels.getSizeC() - 1;
        int t = pixels.getSizeT() - 1;

        byte[] testPlane = getTestPlane();
        pixbuf.setPlane(testPlane, z, c, t);

        pixbuf = service.getPixelBuffer(pixels);
        PixelData plane = pixbuf.getPlane(z, c, t);
        assertNotNull(plane);
        byte[] newMD = Helper.calculateMessageDigest(plane.getData());

        assertEquals(originalDigest, Hex.encodeHexString(newMD));
    }

    @BeforeMethod
    protected void setup() throws Exception {

        ROOT = getOmeroDataDir();

        // Create set up the base fixture which sets up the database for us
        baseFixture = new PixbufIOFixture(this.iContainer, this.iUpdate);
        pixels = baseFixture.setUp();

        // "Our" fixture which creates the planes needed for this test case.
        service = new PixelsService(ROOT);
        pixbuf = service.createPixelBuffer(pixels);
    }

    @BeforeMethod
    protected void tearDown() throws Exception {
        // Tear down the resources create in this fixture
        String path = pixbuf.getPath();
        File f = new File(path);
        f.delete();

        // Tear down the resources created as part of the base fixture
        baseFixture.tearDown();
    }
}