package jalview.io;

import jalview.api.ColorI;
import jalview.api.FeatureColourI;
import jalview.schemes.Colour;
import jalview.schemes.FeatureColour;
import jalview.schemes.FeatureSettingsAdapter;

public class PDBFeatureSettings extends FeatureSettingsAdapter
{

  public static final String FEATURE_INSERTION = "INSERTION";

  public static final String FEATURE_RES_NUM = "RESNUM";

  @Override
  public boolean isFeatureDisplayed(String type)
  {
    return type.equalsIgnoreCase(FEATURE_INSERTION)
            || type.equalsIgnoreCase(FEATURE_RES_NUM);
  }

  @Override
  public FeatureColourI getFeatureColour(String type)
  {
    if (type.equalsIgnoreCase(FEATURE_INSERTION))
    {
      return new FeatureColour()
      {

        @Override
        public ColorI getColour()
        {
          return Colour.red;
        }
      };
    }
    return null;
  }

  /**
   * Order to render insertion after ResNum
   */
  @Override
  public int compare(String feature1, String feature2)
  {
    if (feature1.equalsIgnoreCase(FEATURE_INSERTION))
    {
      return +1;
    }
    if (feature2.equalsIgnoreCase(FEATURE_INSERTION))
    {
      return -1;
    }
    if (feature1.equalsIgnoreCase(FEATURE_RES_NUM))
    {
      return +1;
    }
    if (feature2.equalsIgnoreCase(FEATURE_RES_NUM))
    {
      return -1;
    }
    return 0;
  }
}