package org.apache.tuscany.sca.layout;

import static org.junit.Assert.*;
import junit.framework.Assert;

import org.apache.tuscany.sca.impl.layout.ComponentEntity;
import org.apache.tuscany.sca.impl.layout.CompositeEntity;
import org.apache.tuscany.sca.impl.layout.Entity;
import org.apache.tuscany.sca.impl.layout.LayoutBuilder;
import org.junit.Before;
import org.junit.Test;

public class LayoutBuilderTest {

	LayoutBuilder lb;
	Entity[] ents;
	Entity parent;
	int[][] conns;
	
	@Before
	public void setUp() throws Exception {
		parent = new CompositeEntity("composite");
		parent.setX(200);
		parent.setY(100);
		
		ents = new Entity[5];
		for(int i=0; i<ents.length ;i++){
			ents[i] = new ComponentEntity();
			ents[i].setId(i);
			ents[i].setParent(parent);
		}
		
		
	}

	@Test
	public void testPlaceEntities() throws Exception {
		//setUp();
		Assert.assertEquals(5, ents.length);
		Assert.assertEquals(200, ents[0].getStartPosition());
		
		conns = new int[5][5];
		for(int i=0; i<conns.length ;i++){
			for(int j=0; j<conns.length ;j++){
				//int x = Math.getExponent(Math.random());
				if(i==j-1)
					conns[i][j] = 1;
				else
					conns[i][j] = 0;
			}
		}
		
		lb = new LayoutBuilder(ents, conns);
		
		ents = lb.placeEntities();
		
		Assert.assertEquals(5, ents.length);
		Assert.assertEquals(0, lb.getStartEnt().getId());
		
		Assert.assertEquals(0, ents[0].getLevel());
		Assert.assertEquals(0, ents[1].getLevel());
		Assert.assertEquals(0, ents[2].getLevel());
		Assert.assertEquals(0, ents[3].getLevel());
		Assert.assertEquals(0, ents[4].getLevel());
		
		Assert.assertEquals(0, ents[0].getLane());
		Assert.assertEquals(1, ents[1].getLane());
		Assert.assertEquals(2, ents[2].getLane());
		Assert.assertEquals(3, ents[3].getLane());
		Assert.assertEquals(4, ents[4].getLane());
		
		
	}
	
	@Test
	public void testPlaceEntities1() throws Exception {
		
		conns = new int[5][5];
		for(int i=0; i<conns.length ;i++){
			for(int j=0; j<conns.length ;j++){
				//int x = Math.getExponent(Math.random());
				if(i==j-1 || i==j-4)
					conns[i][j] = 1;
				else
					conns[i][j] = 0;
			}
		}
		conns[3][4] = 0;
		
		lb = new LayoutBuilder(ents, conns);
		
		ents = lb.placeEntities();
		
		Assert.assertEquals(5, ents.length);
		Assert.assertEquals(0, lb.getStartEnt().getId());
		
		Assert.assertEquals(0, ents[0].getLevel());
		Assert.assertEquals(0, ents[1].getLevel());
		Assert.assertEquals(0, ents[2].getLevel());
		Assert.assertEquals(0, ents[3].getLevel());
		Assert.assertEquals(1, ents[4].getLevel());
		
		Assert.assertEquals(0, ents[0].getLane());
		Assert.assertEquals(1, ents[1].getLane());
		Assert.assertEquals(2, ents[2].getLane());
		Assert.assertEquals(3, ents[3].getLane());
		Assert.assertEquals(1, ents[4].getLane());
		
	}

	@Test
	public void testPlaceEntities2() throws Exception {
		
		conns = new int[5][5];
		for(int i=0; i<conns.length ;i++){
			for(int j=0; j<conns.length ;j++){
				if(i*j>2 && i*j<=6)
					conns[i][j] = 1;
				else
					conns[i][j] = 0;
			}
		}
		
		lb = new LayoutBuilder(ents, conns);
		
		ents = lb.placeEntities();
		
		Assert.assertEquals(5, ents.length);
		Assert.assertEquals(1, lb.getStartEnt().getId());
		
		Assert.assertEquals(1, ents[0].getLevel());
		Assert.assertEquals(0, ents[1].getLevel());
		Assert.assertEquals(0, ents[2].getLevel());
		Assert.assertEquals(0, ents[3].getLevel());
		Assert.assertEquals(1, ents[4].getLevel());
		
		Assert.assertEquals(0, ents[0].getLane());
		Assert.assertEquals(0, ents[1].getLane());
		Assert.assertEquals(2, ents[2].getLane());
		Assert.assertEquals(1, ents[3].getLane());
		Assert.assertEquals(1, ents[4].getLane());
		
	}
}