CellModelTableCellRenderer.java

Go to the documentation of this file.

/*
 * Autopsy Forensic Browser
 *
 * Copyright 2020 Basis Technology Corp.
 * Contact: carrier <at> sleuthkit <dot> org
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.sleuthkit.autopsy.datasourcesummary.uiutils;

import java.awt.Component;
import java.awt.Insets;
import javax.swing.BorderFactory;
import javax.swing.JLabel;
import javax.swing.JTable;
import javax.swing.border.Border;
import javax.swing.table.DefaultTableCellRenderer;
import org.apache.commons.lang3.StringUtils;

public class CellModelTableCellRenderer extends DefaultTableCellRenderer {

    private static final long serialVersionUID = 1L;

    public enum HorizontalAlign {
        LEFT(JLabel.LEFT),
        CENTER(JLabel.CENTER),
        RIGHT(JLabel.RIGHT);

        private final int jlabelAlignment;

        HorizontalAlign(int jlabelAlignment) {
            this.jlabelAlignment = jlabelAlignment;
        }

        int getJLabelAlignment() {
            return this.jlabelAlignment;
        }
    }

    public interface CellModel {

        String getText();

        String getTooltip();

        HorizontalAlign getHorizontalAlignment();

        Insets getInsets();
    }

    public static class DefaultCellModel implements CellModel {

        private final String text;
        private String tooltip;
        private HorizontalAlign horizontalAlignment;
        private Insets insets;

        public DefaultCellModel(String text) {
            this.text = text;
            this.tooltip = text;
        }

        @Override
        public String getText() {
            return text;
        }

        @Override
        public String getTooltip() {
            return tooltip;
        }

        public DefaultCellModel setTooltip(String tooltip) {
            this.tooltip = tooltip;
            return this;
        }

        @Override
        public HorizontalAlign getHorizontalAlignment() {
            return horizontalAlignment;
        }

        public DefaultCellModel setHorizontalAlignment(HorizontalAlign alignment) {
            this.horizontalAlignment = alignment;
            return this;
        }

        @Override
        public Insets getInsets() {
            return insets;
        }

        public DefaultCellModel setInsets(Insets insets) {
            this.insets = insets;
            return this;
        }

        @Override
        public String toString() {
            return getText();
        }
    }

    private static final int DEFAULT_ALIGNMENT = JLabel.LEFT;
    private static final Border DEFAULT_BORDER = BorderFactory.createEmptyBorder(1, 5, 1, 5);

    @Override
    public Component getTableCellRendererComponent(JTable table, Object value,
            boolean isSelected, boolean hasFocus, int row, int column) {

        JLabel c = (JLabel) super.getTableCellRendererComponent(table, value, isSelected, hasFocus, row, column);
        if (value instanceof CellModel) {
            return getTableCellRendererComponent(c, (CellModel) value);
        } else {
            return c;
        }

    }

    protected Component getTableCellRendererComponent(JLabel defaultCell, CellModel cellModel) {
        // sets the text for the cell or null if not present.
        String text = cellModel.getText();
        if (StringUtils.isNotBlank(text)) {
            defaultCell.setText(text);
        } else {
            defaultCell.setText(null);
        }

        // sets the tooltip for the cell if present.
        String tooltip = cellModel.getTooltip();
        if (StringUtils.isNotBlank(tooltip)) {
            defaultCell.setToolTipText(tooltip);
        } else {
            defaultCell.setToolTipText(null);
        }

        // sets the padding for cell text within the cell.
        Insets insets = cellModel.getInsets();
        if (insets != null) {
            defaultCell.setBorder(BorderFactory.createEmptyBorder(insets.top, insets.left, insets.bottom, insets.right));
        } else {
            defaultCell.setBorder(DEFAULT_BORDER);
        }

        // sets the JLabel alignment (left, center, right) or default alignment
        // if no alignment is specified
        int alignment = (cellModel.getHorizontalAlignment() == null)
                ? DEFAULT_ALIGNMENT
                : cellModel.getHorizontalAlignment().getJLabelAlignment();
        defaultCell.setHorizontalAlignment(alignment);

        return defaultCell;
    }
}