de.tsl2.nano.logictable.LogicTable Maven / Gradle / Ivy

Go to download
/*
 * File: $HeadURL$
 * Id  : $Id$
 * 
 * created by: Thomas Schneider, Thomas Schneider
 * created on: Nov 28, 2012
 * 
 * Copyright: (c) Thomas Schneider 2012, all rights reserved
 */
package de.tsl2.nano.logictable;

import java.text.Format;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import de.tsl2.nano.collection.TableList;
import de.tsl2.nano.core.util.CollectionUtil;
import de.tsl2.nano.core.util.StringUtil;

/**
 * An Extension of {@link TableList} providing logic expressions to be filled into cells like in excel.
 * 
 * Header-columns and row-ids are flexible but as default like in excel.
 * 
 * @author Thomas Schneider, Thomas Schneider
 * @version $Revision$
 */
@SuppressWarnings("unchecked")
public class LogicTable, ID> extends TableList {
    public static final String ERROR = "ERROR:";

    /** serialVersionUID */
    private static final long serialVersionUID = -7404922627823743263L;

    String regexpHeader = "[A-Z]+";
    transient Map valueMap;

    /**
     * constructor
     * 
     * @param cols
     * @param rows
     */
    public LogicTable(String name, int cols, int rows) {
        this(name, cols);
        fill(rows);
        valueMap = createValueMap();
    }

    /**
     * constructor
     * 
     * @param header
     */
    public LogicTable(H... header) {
        super(header);
    }

    public LogicTable(Object... header) {
        super((H[]) CollectionUtil.copyOfRange(header, 0, header.length, DefaultHeader[].class));
    }

    /**
     * see {@link TableList#TableList(Class, int)}
     */
    public LogicTable(String name, int columnCount) {
        super((H[]) createStandardHeader(columnCount));
        this.name = name;
        valueMap = createValueMap();
    }

    public Map getValueMap() {
        if (valueMap == null)
            valueMap = createValueMap();
        return valueMap;
    }

    protected static Object createDefaultHeader(Object source) {
        if (source instanceof Integer)
            return new DefaultHeader((Integer) source);
        else
            return new DefaultHeader(source.toString());
    }

    /**
     * createStandardHeader
     * 
     * @param columnCount
     * @return array of DefaultHeader (downcasted to Object[] to be compilable on standard-jdk)
     */
    private static final Object[] createStandardHeader(int columnCount) {
        DefaultHeader[] h = new DefaultHeader[columnCount];
        for (int i = 0; i < columnCount; i++) {
            h[i] = new DefaultHeader(i);
        }
        return h;
    }

    public boolean isFormula(Object cell) {
        return String.valueOf(cell).startsWith(EquationSolver.EQUATION);
    }

    /**
     * sets a new value through given column/row-id as strings
     * 
     * @param row row-id to be found through this string
     * @param column header-column to be found through this string
     * @param value value to set
     */
    public void set(String row, String column, Object value) {
        set(findRow(row), findHeader(column), value);
    }

    /**
     * gets a value through given column/row-id as strings
     * 
     * @param row row-id to be found through this string
     * @param column header-column to be found through this string
     * @return cell value
     */
    public Object get(String row, String column) {
        return get(findRow(row), findHeader(column));
    }

    /**
     * {@inheritDoc}
     */
    @Override
    public Object get(int row, int column) {
        if (row == -1)
            if (column == -1)
                return CELL_ROOT;
            else
                return String.valueOf(getHeader()[column]);
        else if (column == -1)
            return getRowID(row);
        Object e = super.get(row, column);
        if (e instanceof String) {
            String expression = (String) e;
            if (expression.startsWith(EquationSolver.EQUATION) || expression.startsWith(EquationSolver.ACTION)) {
                try {
                    EquationSolver solver = new EquationSolver(null, getValueMap());
                    return solver.eval(expression.substring(1));
                } catch (Exception ex) {
                    return ERROR + ex.toString() + "<<<" + e;
                }
            }
        }
        return e;
    }

    private H findHeader(String expression) {
        for (int i = 0; i < header.length; i++) {
            String h = header[i].toString();
            h = StringUtil.extract(h, regexpHeader);
            if (h.equals(expression)) {
                return header[i];
            }
        }
        throw new IllegalArgumentException("no header entry found for: " + expression);
    }

    private ID findRow(String expression) {
        for (int i = 0; i < rows.size(); i++) {
            ID r = getRowID(i);
            String rt = r.toString();
            if (rt.equals(expression)) {
                return r;
            }
        }
        throw new IllegalArgumentException("no rowID entry found for: " + expression);
    }

    /**
     * create a simple readable map, transforming a key to header/row information
     * 
     * @return new created readable map
     */
    public Map createValueMap() {
        final TableList _this = this;
        return new Map() {
            @Override
            public int size() {
                return _this.size() * getColumnCount();
            }

            @Override
            public boolean isEmpty() {
                return size() == 0;
            }

            @Override
            public boolean containsKey(Object key) {
                return get(key) != null;
            }

            @Override
            public boolean containsValue(Object value) {
                throw new UnsupportedOperationException();
            }

            @Override
            public Object get(Object key) {
                try {
                    TEntry e = transform(key);
                    return _this.get(e.value, e.key);
                } catch (Exception ex) {
                    System.err.println("Cell " + key + ": " + ex); //TODO print to log
                    return null;
                }
            }

            @Override
            public Object put(String key, Object value) {
                TEntry e = transform(key);
                return _this.set(e.value, e.key, String.valueOf(value));
            }

            @Override
            public Object remove(Object key) {
                TEntry e = transform(key);
                return _this.remove(e.value);
            }

            @Override
            public void putAll(Map m) {
            	for (String k: m.keySet() ) {
            		put(k, m.get(k));
            	}
            }

            @Override
            public void clear() {
                throw new UnsupportedOperationException();
            }

            @Override
            public Set keySet() {
            	Set keyset = new HashSet();
                for (int r = 0; r < _this.getRowCount(); r++) {
					for (int c = 0; c < _this.getColumnCount(); c++) {
						keyset.add(createKey(r, c));
					}
				}
                return keyset;
            }

            @Override
            public Collection