<filename>src/main/java/com/google/security/zynamics/binnavi/Gui/GraphWindows/JGraphTab.java
// Copyright 2011-2016 Google LLC
//
// 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 com.google.security.zynamics.binnavi.Gui.GraphWindows;

import java.awt.Color;
import java.awt.Component;
import java.util.LinkedHashMap;

import javax.swing.JTabbedPane;

import com.google.common.base.Preconditions;
import com.google.common.base.Predicates;
import com.google.common.collect.Iterables;
import com.google.security.zynamics.binnavi.Gui.CNameShortener;
import com.google.security.zynamics.binnavi.Gui.GraphWindows.ButtonTab.ButtonTabComponent;
import com.google.security.zynamics.binnavi.Gui.GraphWindows.ButtonTab.IButtonTabListener;

/**
 * The tab component that is used to display graph panels in graph windows.
 */
public final class JGraphTab extends JTabbedPane {
  /**
   * Used for serialization.
   */
  private static final long serialVersionUID = 6366240643759928623L;

  /**
   * Listens on clicks on the X in tabs
   */
  private final InternalTabClickListener m_listener = new InternalTabClickListener();

  /**
   * Parent window of the tab component
   */
  private final CGraphWindow m_parent;

  /**
   * 
   */
  private final LinkedHashMap<Integer, Integer> moduleIdCount =
      new LinkedHashMap<Integer, Integer>();

  /**
   * Creates a new graph tab component.
   * 
   * @param parent Parent window of the tab component.
   */
  public JGraphTab(final CGraphWindow parent) {
    super(JTabbedPane.TOP, JTabbedPane.SCROLL_TAB_LAYOUT);
    m_parent = Preconditions.checkNotNull(parent, "IE01638: Parent argument can not be null");
  }

  @Override
  public synchronized void addTab(final String title, final Component component) {
    Preconditions.checkNotNull(title, "IE01639: Title argument can not be null");
    Preconditions.checkNotNull(component, "IE01640: Component argument can not be null");

    super.addTab(title, component);
    setSelectedComponent(component);

    final int moduleId =
        ((IGraphPanel) component).getModel().getViewContainer().getModules().get(0)
            .getConfiguration().getId();
    if (!moduleIdCount.containsKey(moduleId)) {
      moduleIdCount.put(moduleId, 1);
    } else {
      moduleIdCount.put(moduleId, moduleIdCount.get(moduleId) + 1);
    }

    final ButtonTabComponent buttonTab = new ButtonTabComponent(this);
    buttonTab.addListener(m_listener);

    setTabComponentAt(getSelectedIndex(), buttonTab);
  }

  @Override
  public Color getBackgroundAt(final int index) {
    final IGraphPanel panel = ((IGraphPanel) this.getComponentAt(index));
    if (moduleIdCount.size() > 1) {
      return selectTabBackGroundColor(panel.getModel().getViewContainer().getModules().get(0)
          .getConfiguration().getId());
    } else {
      return super.getBackgroundAt(index);
    }
  }

  /**
   * Updates the register headers if the name of a view changed.
   */
  public void updateRegisterHeaders() {
    for (int i = 0; i < getTabCount(); i++) {
      final IGraphPanel component = (IGraphPanel) getComponentAt(i);
      setTitleAt(i, CNameShortener.shorten(component.getModel().getGraph().getRawView()));
    }
  }

  /**
   * Listener responsible for handling clicks on the X button in tabs.
   */
  private class InternalTabClickListener implements IButtonTabListener {
    @Override
    public boolean closing(final ButtonTabComponent btc) {
      final int openWindows = getTabCount();
      for (int i = 0; i < openWindows; i++) {
        if (btc == getTabComponentAt(i)) {
          final IGraphPanel panel = (IGraphPanel) getComponentAt(i);

          if (CPanelCloser.closeTab(m_parent, panel)) {
            btc.removeListener(m_listener);
            final int moduleId =
                panel.getModel().getViewContainer().getModules().get(0).getConfiguration().getId();
            if (moduleIdCount.get(moduleId) == 1) {
              moduleIdCount.remove(moduleId);
            } else {
              moduleIdCount.put(moduleId, moduleIdCount.get(moduleId) - 1);
            }
            return true;
          } else {
            return false;
          }
        }
      }

      return true;
    }
  }

  private Color selectTabBackGroundColor(final int seed) {

    final int insertionPosition =
        Iterables.indexOf(moduleIdCount.keySet(), Predicates.equalTo(seed));

    switch (insertionPosition) {
      case 0:
        return Color.getHSBColor((float) 0.55, (float) 0.2, (float) 0.8);
      case 1:
        return Color.getHSBColor((float) 0.6, (float) 0.2, (float) 0.8);
      case 2:
        return Color.getHSBColor((float) 0.65, (float) 0.2, (float) 0.8);
      case 3:
        return Color.getHSBColor((float) 0.7, (float) 0.2, (float) 0.8);
      case 4:
        return Color.getHSBColor((float) 0.75, (float) 0.2, (float) 0.8);
      case 5:
        return Color.getHSBColor((float) 0.8, (float) 0.2, (float) 0.8);
      case 6:
        return Color.getHSBColor((float) 0.85, (float) 0.2, (float) 0.8);
      case 7:
        return Color.getHSBColor((float) 0.9, (float) 0.2, (float) 0.8);
      case 8:
        return Color.getHSBColor((float) 0.95, (float) 0.2, (float) 0.8);
      case 9:
        return Color.getHSBColor(1, (float) 0.2, (float) 0.8);
      case 10:
        return Color.getHSBColor((float) 0.05, (float) 0.2, (float) 0.8);
      case 11:
        return Color.getHSBColor((float) 0.1, (float) 0.2, (float) 0.8);
      case 12:
        return Color.getHSBColor((float) 0.15, (float) 0.2, (float) 0.8);
      case 13:
        return Color.getHSBColor((float) 0.2, (float) 0.2, (float) 0.8);
      case 14:
        return Color.getHSBColor((float) 0.25, (float) 0.2, (float) 0.8);
      case 15:
        return Color.getHSBColor((float) 0.3, (float) 0.2, (float) 0.8);
      case 16:
        return Color.getHSBColor((float) 0.35, (float) 0.2, (float) 0.8);
      case 17:
        return Color.getHSBColor((float) 0.4, (float) 0.2, (float) 0.8);
      case 18:
        return Color.getHSBColor((float) 0.45, (float) 0.2, (float) 0.8);
      case 19:
        return Color.getHSBColor((float) 0.5, (float) 0.2, (float) 0.8);
      default:
        return Color.WHITE;
    }
  }
}
