/**
 * Copyright (C) 2009 - present by OpenGamma Inc. and the OpenGamma group of companies
 *
 * Please see distribution for license.
 */
package com.opengamma.web.server.conversion;

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

import com.opengamma.core.marketdatasnapshot.VolatilitySurfaceData;
import com.opengamma.engine.value.ValueSpecification;

/**
 * Converter for {@link VolatilitySurfaceData} objects.
 */
@SuppressWarnings("rawtypes")
public class VolatilitySurfaceDataConverter implements ResultConverter<VolatilitySurfaceData> {

  @Override
  // TODO PLAT-2249 Add field to allow transposing the display surface 
  public Object convertForDisplay(ResultConverterCache context, ValueSpecification valueSpec, VolatilitySurfaceData rawValue, ConversionMode mode) {
    @SuppressWarnings("unchecked")
    VolatilitySurfaceData<Object, Object> value = rawValue;
    Map<String, Object> result = new HashMap<String, Object>();
    
    result.put("xCount", value.getXs().length);
    result.put("yCount", value.getYs().length);
    
    if (mode == ConversionMode.FULL) {
      //TODO assuming that all surfaces are interpolated - bad
      Object[] xs = value.getXs();
      String[] xsStrings = new String[xs.length];
      for (int i = 0; i < xs.length; i++) {
        xsStrings[i] = LabelFormatter.format(xs[i]);
      }
      result.put("xs", xsStrings);
      Object[] ys = value.getYs();
      String[] ysStrings = new String[ys.length];
      for (int i = 0; i < ys.length; i++) {
        ysStrings[i] = LabelFormatter.format(ys[i]);
      }
      result.put("ys", ysStrings);
      
      
      double[][] surface = new double[ys.length][xs.length];
      boolean[][] missingValues = new boolean[ys.length][xs.length];
      // Summary view includes only the actual points of the surface
      for (int y = 0; y < ys.length; y++) {
        for (int x = 0; x < xs.length; x++) {
          Object xt = xs[x];
          Object yt = ys[y];
          Double volatility = value.getVolatility(xt, yt);
          if (volatility == null) {
            missingValues[y][x] = true;
            //Some 'obviously wrong' value in case client displays it.  Can't use NaN
            surface[y][x] = Double.MAX_VALUE;
          } else {
            surface[y][x] = volatility;
          }
        }
      }
      result.put("surface", surface);
      result.put("missingValues", missingValues);
    }
    result.put("axesLabel", "Strike \\ Expiry");
    return result;
  }

  @Override
  public Object convertForHistory(ResultConverterCache context, ValueSpecification valueSpec, VolatilitySurfaceData value) {
    return null;
  }

  @Override
  public String convertToText(ResultConverterCache context, ValueSpecification valueSpec, VolatilitySurfaceData value) {
    // Could actually serialise the surface to a string if this is an issue
    return "Volatility Surface (" + value.getXs().length + " x " + value.getYs().length + ")";
  }
  
  @Override
  public String getFormatterName() {
    return "SURFACE_DATA";
  }
  
}