Table.java

package ilarkesto.pdf.itext;

import ilarkesto.pdf.ACell;
import ilarkesto.pdf.APdfElement;
import ilarkesto.pdf.ARow;
import ilarkesto.pdf.ATable;

import java.awt.Color;
import java.util.ArrayList;
import java.util.List;

import com.lowagie.text.Element;
import com.lowagie.text.pdf.PdfPCell;
import com.lowagie.text.pdf.PdfPTable;

public class Table extends ATable implements ItextElement {

	private List<Cell> cells = new ArrayList<Cell>();

	@Override
	public ACell cell() {
		Cell c = new Cell(this);
		Float defaultCellPadding = getDefaultCellPadding();
		if (defaultCellPadding != null) {
			c.setPadding(defaultCellPadding);
		}
		cells.add(c);
		return c;
	}

	public Element getITextElement() {
		float[] cellWidths = getCellWidths();
		PdfPTable t = cellWidths == null ? new PdfPTable(getColumnCount()) : new PdfPTable(cellWidths);

		t.setHorizontalAlignment(PdfPTable.ALIGN_LEFT);

		Float width = getWidth();
		if (width != null) t.setWidthPercentage(width);

		for (Cell cell : cells) {
			t.addCell((PdfPCell) cell.getITextElement());
		}

		return t;
	}

	@Override
	public ATable createCellBorders(Color color, float width) {
		float[] cellWidths = getCellWidths();
		int cols = cellWidths == null ? getColumnCount() : cellWidths.length;
		int col = 0;
		int row = 0;
		for (Cell cell : cells) {
			if (row == 0) cell.setBorderTop(color, width);
			cell.setBorderRight(color, width);
			cell.setBorderBottom(color, width);
			if (col == 0) cell.setBorderLeft(color, width);
			col++;
			if (col >= cols) {
				col = 0;
				row++;
			}
		}
		return this;
	}

	@Override
	public ARow row() {
		return new ARow(this);
	}

	@Override
	public ARow row(Object... cellTexts) {
		ARow row = row();
		for (Object text : cellTexts)
			row.cell(text);
		return row;
	}

	// --- dependencies ---

	public Table(APdfElement parent) {
		super(parent);
	}

}