/**
* File: weatherIntentParser.h
*
* Author: Kevin Karol
* Date:   5/10/2018
*
* Description: Translates the contents of a weather response user intent into terms that
* can be used for decision making/response internally
*
* Copyright: Anki, Inc. 2018
**/

#include "engine/aiComponent/behaviorComponent/weatherIntents/weatherIntentParser.h"

#include "util/console/consoleInterface.h"
#include "util/logging/DAS.h"
#include "util/logging/logging.h"
#include "util/string/stringUtils.h"

#include <exception>
#include <iomanip>

namespace Anki {
namespace Vector {

// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
WeatherIntentParser::WeatherIntentParser(const RobotDataLoader::WeatherResponseMap* weatherResponseMap,
                                         const Json::Value& conditionRemaps)
: _weatherResponseMap(weatherResponseMap)
, _conditionRemaps(conditionRemaps)
{

}


// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
bool WeatherIntentParser::IsForecast(const UserIntent_WeatherResponse& weatherIntent) const
{
  return (weatherIntent.isForecast == "true") ||
         (weatherIntent.isForecast == "True");
}


// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
bool WeatherIntentParser::IsFahrenheit(const UserIntent_WeatherResponse& weatherIntent) const
{
  return weatherIntent.temperatureUnit == "F" ||
         weatherIntent.temperatureUnit == "f";
}


// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
WeatherConditionType WeatherIntentParser::GetCondition(const UserIntent_WeatherResponse& weatherIntent) const
{
  std::string str;
  std::transform(weatherIntent.condition.begin(), weatherIntent.condition.end(),
                 std::back_inserter(str), [](const char c) { return std::tolower(c); });

  auto iter = _weatherResponseMap->find(str);
  if(iter != _weatherResponseMap->end()){
    return _conditionRemaps.GetRemappedCondition(*this, weatherIntent, iter->second);
  }else{
    PRINT_NAMED_ERROR("WeatherIntentParser.GetCondition.NoConditionMatch",
                      "No weather condition found for %s",
                      str.c_str());
    return WeatherConditionType::Count;
  }
}


// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
tm WeatherIntentParser::GetLocalDateTime(const UserIntent_WeatherResponse& weatherIntent) const
{
  tm localTime;
  strptime(weatherIntent.localDateTime.c_str(), "%Y-%m-%dT%H:%M%S-", &localTime);
  return localTime;
}


// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
bool WeatherIntentParser::GetRawTemperature(const UserIntent_WeatherResponse& weatherIntent,
                                            int& outTemp) const
{
  if(weatherIntent.temperature.empty()){
    return false;
  }
  try{
    outTemp = std::stoi(weatherIntent.temperature);
  }catch(const std::invalid_argument& ia){
    return false;
  }
  return true;
}

// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
bool WeatherIntentParser::GetTemperatureF(const UserIntent_WeatherResponse& weatherIntent,
                                          float& outTempF) const
{
  int rawTemp;
  if( !GetRawTemperature(weatherIntent, rawTemp) ) {
    return false;
  }

  const bool isFahrenheit = IsFahrenheit(weatherIntent);

  if( isFahrenheit ) {
    outTempF = rawTemp;
  }
  else {
    outTempF = ConvertTempCToF(rawTemp);
  }

  return true;
}

// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
float WeatherIntentParser::ConvertTempCToF(const float tempC)
{
  const float ret = tempC * 1.8f + 32.0f;
  return ret;
}

// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
void WeatherIntentParser::SendDASEventForResponse(const UserIntent_WeatherResponse& intent) const
{
  const bool isForecast = IsForecast(intent);
  const bool isFahrenheit = IsFahrenheit(intent);
  auto condition = GetCondition(intent);
  int temp = 0;
  const bool tempOK = GetRawTemperature(intent, temp);

  DASMSG(weather_response, "behavior.weather.response", "The robot is responding to a weather request");
  DASMSG_SET(s1, WeatherConditionTypeToString(condition), "displayed weather condition");
  DASMSG_SET(s2, intent.condition, "raw condition response from cloud");
  DASMSG_SET(i1, isForecast ? 1 : 0, "1 if request is for a forecast, 0 otherwise");
  if( tempOK ) {
    DASMSG_SET(i2, temp, "Returned temperature (in units specified in s3)");
    DASMSG_SET(s3, isFahrenheit ? "F" : "C", "temperature units (F or C, null if no temperature)");
  }
  DASMSG_SEND();
}


} // namespace Vector
} // namespace Anki