/**
* Copyright (C) 2009 - present by OpenGamma Inc. and the OpenGamma group of companies
*
* Please see distribution for license.
*/
package com.opengamma.financial.aggregation;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import com.opengamma.core.position.Position;
import com.opengamma.core.position.Trade;
import com.opengamma.core.position.impl.SimplePositionComparator;
/**
* Abstract aggregation function for bucketing equities and equity options by GICS code of the underlying
*/
public class CounterpartyAggregationFunction implements AggregationFunction<String> {
private static final String UNKNOWN = "N/A";
private final Comparator<Position> _comparator = new SimplePositionComparator();
public CounterpartyAggregationFunction() {
}
@Override
public String classifyPosition(Position position) {
if (position.getTrades().size() == 0) {
return UNKNOWN;
} else {
Trade trade = position.getTrades().iterator().next();
if (trade.getCounterparty() != null) {
return trade.getCounterparty().getExternalId().getValue();
} else {
return UNKNOWN;
}
}
}
@Override
public String getName() {
return "Counterparty";
}
@Override
public Collection<String> getRequiredEntries() {
return Collections.emptyList();
}
@Override
public int compare(String o1, String o2) {
if (o1.equals(UNKNOWN)) {
if (o2.equals(UNKNOWN)) {
return 0;
}
return 1;
} else if (o2.equals(UNKNOWN)) {
return -1;
}
return o1.compareTo(o2);
}
@Override
public Comparator<Position> getPositionComparator() {
return _comparator;
}
}