YieldCurveDataSelectorFudgeBuilder.java

/**
 * Copyright (C) 2013 - present by OpenGamma Inc. and the OpenGamma group of companies
 *
 * Please see distribution for license.
 */
package com.opengamma.integration.marketdata.manipulator.dsl;

import java.util.Set;
import java.util.regex.Pattern;

import org.fudgemsg.FudgeField;
import org.fudgemsg.FudgeMsg;
import org.fudgemsg.MutableFudgeMsg;
import org.fudgemsg.mapping.FudgeBuilder;
import org.fudgemsg.mapping.FudgeBuilderFor;
import org.fudgemsg.mapping.FudgeDeserializer;
import org.fudgemsg.mapping.FudgeSerializer;

import com.google.common.collect.Sets;
import com.opengamma.util.money.Currency;

/**
 *
 */
@FudgeBuilderFor(YieldCurveDataSelector.class)
public class YieldCurveDataSelectorFudgeBuilder implements FudgeBuilder<YieldCurveDataSelector> {

  private static final String CALC_CONFIGS = "calculationConfigurationNames";
  private static final String NAMES = "names";
  private static final String CURRENCIES = "currencies";
  private static final String NAME_MATCH_PATTERN = "nameMatchPattern";
  private static final String NAME_LIKE_PATTERN = "nameLikePattern";

  @Override
  public MutableFudgeMsg buildMessage(FudgeSerializer serializer, YieldCurveDataSelector selector) {
    MutableFudgeMsg msg = serializer.newMessage();
    Set<String> calcConfigNames = selector.getCalcConfigNames();
    if (calcConfigNames != null) {
      MutableFudgeMsg calcConfigsMsg = serializer.newMessage();
      for (String calcConfigName : calcConfigNames) {
        serializer.addToMessage(calcConfigsMsg, null, null, calcConfigName);
      }
      serializer.addToMessage(msg, CALC_CONFIGS, null, calcConfigsMsg);
    }
    if (selector.getNames() != null && !selector.getNames().isEmpty()) {
      MutableFudgeMsg namesMsg = serializer.newMessage();
      for (String name : selector.getNames()) {
        serializer.addToMessage(namesMsg, null, null, name);
      }
      serializer.addToMessage(msg, NAMES, null, namesMsg);
    }
    if (selector.getCurrencies() != null && !selector.getCurrencies().isEmpty()) {
      MutableFudgeMsg currenciesMsg = serializer.newMessage();
      for (Currency currency : selector.getCurrencies()) {
        serializer.addToMessage(currenciesMsg, null, null, currency.getCode());
      }
      serializer.addToMessage(msg, CURRENCIES, null, currenciesMsg);
    }
    if (selector.getNameMatchPattern() != null) {
      serializer.addToMessage(msg, NAME_MATCH_PATTERN, null, selector.getNameMatchPattern().pattern());
    }
    if (selector.getNameLikePattern() != null) {
      serializer.addToMessage(msg, NAME_LIKE_PATTERN, null, selector.getNameLikePattern().pattern());
    }
    return msg;
  }

  @Override
  public YieldCurveDataSelector buildObject(FudgeDeserializer deserializer, FudgeMsg msg) {
    Set<String> calcConfigNames;
    if (msg.hasField(CALC_CONFIGS)) {
      calcConfigNames = Sets.newHashSet();
      FudgeMsg calcConfigsMsg = msg.getMessage(CALC_CONFIGS);
      for (FudgeField field : calcConfigsMsg) {
        calcConfigNames.add(deserializer.fieldValueToObject(String.class, field));
      }
    } else {
      calcConfigNames = null;
    }
    FudgeField namesField = msg.getByName(NAMES);
    Set<String> names;
    if (namesField != null) {
      FudgeMsg namesMsg = (FudgeMsg) namesField.getValue();
      names = Sets.newHashSet();
      for (FudgeField field : namesMsg) {
        names.add(deserializer.fieldValueToObject(String.class, field));
      }
    } else {
      names = null;
    }

    FudgeField currenciesField = msg.getByName(CURRENCIES);
    Set<Currency> currencies;
    if (currenciesField != null) {
      FudgeMsg currenciesMsg = (FudgeMsg) currenciesField.getValue();
      currencies = Sets.newHashSet();
      for (FudgeField field : currenciesMsg) {
        currencies.add(Currency.of(deserializer.fieldValueToObject(String.class, field)));
      }
    } else {
      currencies = null;
    }

    Pattern nameMatchPattern;
    FudgeField namePatternField = msg.getByName(NAME_MATCH_PATTERN);
    if (namePatternField != null) {
      String regex = deserializer.fieldValueToObject(String.class, namePatternField);
      nameMatchPattern = Pattern.compile(regex);
    } else {
      nameMatchPattern = null;
    }

    Pattern nameLikePattern;
    FudgeField nameLikeField = msg.getByName(NAME_LIKE_PATTERN);
    if (nameLikeField != null) {
      String regex = deserializer.fieldValueToObject(String.class, nameLikeField);
      nameLikePattern = Pattern.compile(regex);
    } else {
      nameLikePattern = null;
    }
    return new YieldCurveDataSelector(calcConfigNames, names, currencies, nameMatchPattern, nameLikePattern);
  }
}