// @(#)root/hist:$Name:  $:$Id: TPrincipal.h,v 1.11 2004/06/09 13:40:02 brun Exp $
// Author: Christian Holm Christensen    1/8/2000

/*************************************************************************
 * Copyright (C) 1995-2004, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/

#ifndef ROOT_TPrincipal
#define ROOT_TPrincipal

#ifndef ROOT_TNamed
#include "TNamed.h"
#endif
#ifndef ROOT_TVectorD
#include "TVectorD.h"
#endif
#ifndef ROOT_TMatrixD
#include "TMatrixD.h"
#endif
#ifndef ROOT_TList
#include "TList.h"
#endif

class TPrincipal : public TNamed {

protected:
  Int_t       fNumberOfDataPoints;   // Number of data points
  Int_t       fNumberOfVariables;    // Number of variables

  TVectorD    fMeanValues;           // Mean value over all data points
  TVectorD    fSigmas;               // vector of sigmas
  TMatrixD    fCovarianceMatrix;     // Covariance matrix

  TMatrixD    fEigenVectors;         // Eigenvector matrix of trans
  TVectorD    fEigenValues;          // Eigenvalue vector of trans

  TVectorD    fOffDiagonal;          // elements of the tridiagonal

  TVectorD    fUserData;             // Vector of original data points

  Double_t    fTrace;                // Trace of covarience matrix

  TList      *fHistograms;           // List of histograms

  Bool_t      fIsNormalised;         // Normalize matrix?
  Bool_t      fStoreData;            // Should we store input data?

  void        MakeNormalised();
  void        MakeRealCode(const char *filename, const char *prefix, Option_t *option="");

public:
  TPrincipal();
  virtual ~TPrincipal();
  TPrincipal(Int_t nVariables, Option_t *opt="ND");

  virtual void       AddRow(const Double_t *x);
  virtual void       Browse(TBrowser *b);
  virtual void       Clear(Option_t *option="");
  const TMatrixD    *GetCovarianceMatrix() const {return &fCovarianceMatrix;}
  const TVectorD    *GetEigenValues() const      {return &fEigenValues;}
  const TMatrixD    *GetEigenVectors() const     {return &fEigenVectors;}
  TList             *GetHistograms() const {return fHistograms;}
  const TVectorD    *GetMeanValues() const       {return &fMeanValues;}
  const Double_t    *GetRow(Int_t row);
  const TVectorD    *GetSigmas() const           {return &fSigmas;}
  const TVectorD    *GetUserData() const         {return &fUserData;}
  Bool_t             IsFolder() const { return kTRUE;}
  virtual void       MakeCode(const char *filename ="pca", Option_t *option="");  // *MENU*
  virtual void       MakeHistograms(const char *name = "pca", Option_t *option="epsdx"); // *MENU*
  virtual void       MakeMethods(const char *classname = "PCA", Option_t *option=""); // *MENU*
  virtual void       MakePrincipals();            // *MENU*
  virtual void       P2X(const Double_t *p, Double_t *x, Int_t nTest);
  virtual void       Print(Option_t *opt="MSE") const;         // *MENU*
  virtual void       SumOfSquareResiduals(const Double_t *x, Double_t *s);
  void               Test(Option_t *option="");       // *MENU*
  virtual void       X2P(const Double_t *x, Double_t *p);

  ClassDef(TPrincipal,2) // Principal Components Analysis
}
;

#endif
