DataCorr.h

Go to the documentation of this file.

#ifndef ANP_DATACORR_H
#define ANP_DATACORR_H

// $Id: DataCorr.h,v 1.2 2007/12/14 02:40:27 rustem Exp $
//
// Compute linear correlation coefficients
//

// C++
#include <map>

// Local
#include "PhysicsNtuple/DataItem.h"
#include "PhysicsNtuple/KeyPair.h"

namespace Anp
{
   namespace Corr
   {
      struct Data
      {
         Data() : sum_x(0.0), sum_y(0.0), sum_xx(0.0), sum_xy(0.0), sum_yy(0.0) {}
         double sum_x;
         double sum_y;
         double sum_xx;
         double sum_xy;
         double sum_yy;
      };

      typedef KeyPair<short> Key;
      typedef std::map<Key, Data> DataMap;
      typedef std::map<Key, double> CorrMap;
   }
   
   class DataCorr
   {
   public:
      
      DataCorr();
      ~DataCorr();

      unsigned int Fill(DataIter ibeg, DataIter iend);

      unsigned int Size() const;

      const Corr::CorrMap GetCorrCoef() const;

      void Reset();

   private:
      
      unsigned int fSampleSize;

      Corr::DataMap fData;
   };

   //
   // Inlined member functions
   //
   
   inline unsigned int DataCorr::Size() const { return fSampleSize; }
}

#endif

---