/*
 * Copyright 2007 Google Inc.
 *
 * 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.test.metric;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class ClassCost {
  public static final String CLASS_NAME = "class";

  public static class CostComparator implements java.util.Comparator<ClassCost> {
    private final CostModel costModel;

    public CostComparator(CostModel costModel) {
      this.costModel = costModel;
    }

    public int compare(ClassCost c1, ClassCost c2) {
      int diff = (costModel.computeClass(c2) - costModel.computeClass(c1));
      return diff == 0 ? c1.className.compareTo(c2.className) : diff;
    }
  }

  private final List<MethodCost> methods;
  private final String className;

  public ClassCost(String className, List<MethodCost> methods) {
    this.className = className;
    this.methods = methods;
  }

  public MethodCost getMethodCost(String methodName) {
    for (MethodCost cost : methods) {
      if (cost.getMethodName().equals(methodName)) {
        return cost;
      }
    }
    throw new IllegalArgumentException("Method '" + methodName
        + "' does not exist.");
  }

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

  public String getClassName() {
    return className;
  }

  public String getPackageName() {

    return getClassName().lastIndexOf('.') == -1 ? "" : getClassName().substring(0, getClassName().lastIndexOf('.'));
  }

  public List<MethodCost> getMethods() {
    return methods;
  }

  // TODO: delete
  public long getTotalComplexityCost() {
    long totalCost = 0;
    for (MethodCost methodCost : getMethods()) {
      totalCost += methodCost.getTotalCost().getCyclomaticComplexityCost();
    }
    return totalCost;
  }

  // TODO: delete
  public long getHighestMethodComplexityCost() {
    long cost = 0;
    for (MethodCost methodCost : getMethods()) {
      if (methodCost.getTotalCost().getCyclomaticComplexityCost() > cost) {
        cost = methodCost.getTotalCost().getCyclomaticComplexityCost();
      }
    }
    return cost;
  }

  // TODO: delete
  public long getTotalGlobalCost() {
    long totalCost = 0;
    for (MethodCost methodCost : getMethods()) {
      totalCost += methodCost.getTotalCost().getGlobalCost();
    }
    return totalCost;
  }

  // TODO: delete
  public long getHighestMethodGlobalCost() {
    long cost = 0;
    for (MethodCost methodCost : getMethods()) {
      if (methodCost.getTotalCost().getGlobalCost() > cost) {
        cost = methodCost.getTotalCost().getGlobalCost();
      }
    }
    return cost;
  }

  public Map<String, Object> getAttributes() {
    HashMap<String, Object> map = new HashMap<String, Object>();
    map.put(CLASS_NAME, className);
    return map;
  }

}