TerminalColor.java

package com.jediterm.terminal;

import org.jetbrains.annotations.Nullable;

import java.awt.*;

/**
 * @author traff
 */
public class TerminalColor {
  public static final TerminalColor BLACK = index(0);
  public static final TerminalColor WHITE = index(15);

  private int myColorIndex;
  private int myR;
  private int myG;
  private int myB;

  public TerminalColor(int index) {
    myColorIndex = index;
    myR = -1;
    myG = -1;
    myB = -1;
  }

  public TerminalColor(int r, int g, int b) {
    myColorIndex = -1;
    myR = r;
    myG = g;
    myB = b;
  }

  public static TerminalColor index(int index) {
    return new TerminalColor(index);
  }

  public static TerminalColor rgb(int r, int g, int b) {
    return new TerminalColor(r, g, b);
  }

  public boolean isIndexed() {
    return myColorIndex != -1;
  }

  public Color toAwtColor() {
    if (isIndexed()) {
      throw new IllegalArgumentException("Color is indexed color so a palette is needed");
    }

    return new Color(myR, myG, myB);
  }

  public int getIndex() {
    return myColorIndex;
  }

  @Override
  public boolean equals(Object o) {
    if (this == o) return true;
    if (o == null || getClass() != o.getClass()) return false;

    TerminalColor that = (TerminalColor) o;

    if (isIndexed()) {
      if (!that.isIndexed()) return false;
      if (myColorIndex != that.myColorIndex) return false;
    } else {
      if (that.isIndexed()) {
        return false;
      }
      if (myB != that.myB) return false;
      if (myG != that.myG) return false;
      if (myR != that.myR) return false;
    }
    return true;
  }

  @Override
  public int hashCode() {
    int result = myColorIndex;
    result = 31 * result + myR;
    result = 31 * result + myG;
    result = 31 * result + myB;
    return result;
  }

  @Nullable
  public static TerminalColor awt(@Nullable Color color) {
    if (color == null) {
      return null;
    }
    return rgb(color.getRed(), color.getGreen(), color.getBlue());
  }
}