All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • com.davidbracewell.string.TableFormatter Maven / Gradle / Ivy

    There is a newer version: 0.5
    Show newest version
    package com.davidbracewell.string;

import com.davidbracewell.conversion.Cast;
import com.davidbracewell.io.resource.Resource;
import com.davidbracewell.io.resource.StringResource;
import lombok.NonNull;

import java.io.IOException;
import java.io.PrintStream;
import java.io.Serializable;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;

/**
 * The type Table formatter.
 *
 * @author David B. Bracewell
 */
public class TableFormatter implements Serializable {
  private static final long serialVersionUID = 1L;
  private static final int MIN_CELL_WIDTH = 5;
  private static final DecimalFormat longNumberFormatter = new DecimalFormat("0.0E0");
  private static final DecimalFormat normalNumberFormatter = new DecimalFormat("0.000#####");
  private List header = new ArrayList<>();
  private List> content = new LinkedList<>();
  private int longestCell = 2;
  private int longestRow;
  private double maxNumber;
  private String title;

  /**
   * Clear.
   */
  public void clear() {
    this.header.clear();
    this.content.clear();
    this.longestCell = 2;
    this.longestRow = 0;
    this.title = null;
    this.maxNumber = 0;
  }

  /**
   * Title table formatter.
   *
   * @param title the title
   * @return the table formatter
   */
  public TableFormatter title(String title) {
    this.title = title;
    return this;
  }

  /**
   * Header table formatter.
   *
   * @param collection the collection
   * @return the table formatter
   */
  public TableFormatter header(Collection collection) {
    header.addAll(collection);
    longestCell = (int) Math.max(longestCell, collection.stream().mapToDouble(o -> o.toString().length() + 2).max().orElse(0));
    return this;
  }

  private int length(Number number) {
    if (number instanceof Long || number instanceof Integer || number instanceof Short) {
      return Math.min(longNumberFormatter.format(number).length(), number.toString().length());
    }
    return Math.min(longNumberFormatter.format(number).length(), normalNumberFormatter.format(number).length());
  }

  /**
   * Content table formatter.
   *
   * @param collection the collection
   * @return the table formatter
   */
  public TableFormatter content(Collection collection) {
    this.content.add(new ArrayList<>(collection));
    longestCell = (int) Math.max(longestCell, collection.stream()
      .mapToDouble(o -> {
          if (o instanceof Number) {
            return Math.max(MIN_CELL_WIDTH, length(Cast.as(o)));
          } else {
            return Math.max(MIN_CELL_WIDTH, o.toString().length() + 2);
          }
        }
      ).max().orElse(0)
    );
    longestRow = Math.max(longestRow, collection.size());
    return this;
  }

  private String middleCMBar(String hbar, int nC) {
    StringBuilder builder = new StringBuilder();
    builder.append("├").append(hbar);
    for (int i = 1; i < nC; i++) {
      builder.append("┼").append(hbar);
    }
    builder.append("┤");
    return builder.toString();
  }

  private String convert(Object o, int longestCell) {
    if (o instanceof Number) {
      Number number = Cast.as(o);
      if (number instanceof Long || number instanceof Integer || number instanceof Short) {
        String numString = Long.toString(number.longValue());
        return numString.length() <= longestCell ? numString : longNumberFormatter.format(number);
      } else {
        String numString = normalNumberFormatter.format(number);
        return numString.length() <= longestCell ? numString : longNumberFormatter.format(number);
      }
    }
    return StringUtils.abbreviate(o.toString(), longestCell - 2);
  }

  private void printRow(PrintStream stream, List row, int longestCell, int longestRow) {
    while (row.size() < longestRow) {
      row.add(StringUtils.EMPTY);
    }
    stream.printf("│%s", StringUtils.center(convert(row.get(0), longestCell), longestCell));
    for (int i = 1; i < longestRow; i++) {
      stream.printf("│%s", StringUtils.center(convert(row.get(i), longestCell), longestCell));
    }
    stream.println("│");
  }

  /**
   * Print the table to the give PrintStream .
   *
   * @param stream the print stream to write to
   */
  public void print(@NonNull PrintStream stream) {
    String horizontalBar = StringUtils.repeat("─", longestCell);
    String hline = middleCMBar(horizontalBar, longestRow);
    maxNumber = Math.pow(10, longestCell - 2);
    longestRow = Math.max(longestRow, header.size());

    if (!StringUtils.isNullOrBlank(title)) {
      stream.println(
        StringUtils.center(title, (longestCell * longestRow) + longestRow + 1)
      );
    }

    stream.printf("┌%s", horizontalBar);
    for (int i = 1; i < longestRow; i++) {
      stream.printf("┬%s", horizontalBar);
    }
    stream.println("┐");

    if (header.size() > 0) {
      printRow(stream, header, longestCell, longestRow);
      stream.println(hline);
    }

    for (int r = 0; r < content.size(); r++) {
      printRow(stream, content.get(r), longestCell, longestRow);
      if (r + 1 < content.size()) {
        stream.println(hline);
      }
    }


    stream.printf("└%s", horizontalBar);
    for (int i = 1; i < longestRow; i++) {
      stream.printf("┴%s", horizontalBar);
    }
    stream.println("┘");

  }

  /**
   * Writes the table to a resource.
   *
   * @param resource the resource to write to
   * @return the resource written to
   * @throws IOException Something went wrong writing to the resource
   */
  public Resource write(@NonNull Resource resource) throws IOException {
    Resource stringResource = new StringResource();
    try (PrintStream printStream = new PrintStream(stringResource.outputStream())) {
      print(printStream);
    }
    resource.write(stringResource.readToString());
    return resource;
  }


}// END OF TableFormatter
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy