#ifndef __RdRecShowerParameterStorageMap_H
#define __RdRecShowerParameterStorageMap_H
#include <TObject.h>
#include <map>
#include <utility>
#include <stdexcept>
#include <sstream>
#include <cmath>
#include "ShowerRRecDataQuantities.h"

class RdRecShowerParameterStorageMap : public TObject {
  public:
    typedef revt::ShowerRRecDataQuantities ParameterKey;
    typedef std::pair<revt::ShowerRRecDataQuantities, revt::ShowerRRecDataQuantities> CovarianceKey;

    typedef std::map<ParameterKey, double> ParameterMap;
    typedef std::map<CovarianceKey, double> CovarianceMap;

//     RdRecShowerParameterStorageMap();
//     virtual ~RdRecShowerParameterStorageMap();

    void
    SetParameter(const ParameterKey index, const double value)
    {
      fParameterMap[index] = value;
    }

    void
    SetParameterCovariance(const ParameterKey index1, const ParameterKey index2, const double value)
    {
      fCovarianceMap[GenCovarianceKey(index1, index2)] = value;
    }

    void
    SetParameterError(const ParameterKey index, const double value)
    {
      SetParameterCovariance(index, index, value * value);
    }

    double
    GetParameter(const ParameterKey index)
      const
    {
      const ParameterMap::const_iterator it = fParameterMap.find(index);
      if (it != fParameterMap.end()) {
        return it->second;
      }
      else {
        std::stringstream sstr;
        sstr << "Shower Parameter " << index << " not set";
        throw std::out_of_range(sstr.str().c_str());
      }
    }

    double
    GetParameterCovariance(const ParameterKey index1,
                           const ParameterKey index2)
      const
    {
      const CovarianceMap::const_iterator it =
        fCovarianceMap.find(GenCovarianceKey(index1, index2));

      if (it != fCovarianceMap.end()) {
        return it->second;
      }
      else {
        std::stringstream sstr;
        sstr << "Shower ParameterCovariance " << index1 << ", " << index2 << " not set";
        throw std::out_of_range(sstr.str().c_str());
      }
    }

    double
    GetParameterError(const ParameterKey index)
      const
    {
      return std::sqrt(GetParameterCovariance(index, index));
    }

    bool
    HasParameter(const ParameterKey index)
      const
    {
      return fParameterMap.find(index) != fParameterMap.end();
    }

    bool
    HasParameterCovariance(const ParameterKey index1,
                           const ParameterKey index2)
      const
    {
      return fCovarianceMap.find(GenCovarianceKey(index1, index2))
        != fCovarianceMap.end();
    }

  private:

    ParameterMap fParameterMap;
    CovarianceMap fCovarianceMap;

    /// helper function to create the key for the covariance map from the parameters using sorting
    inline
    CovarianceKey
    GenCovarianceKey(const ParameterKey param1,
                     const ParameterKey param2)
    const
    {
      if (long(param1) < long(param2))
        return std::make_pair(param1, param2);
      else
        return std::make_pair(param2, param1);
    }

  ClassDef(RdRecShowerParameterStorageMap, 3);
};


#endif