JMEstimator.cc File Reference

Example program to plot various M-Estimators. More...

#include <string>
#include <iostream>
#include <iomanip>
#include "TROOT.h"
#include "TFile.h"
#include "TH1D.h"
#include "JFit/JMEstimator.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"

Go to the source code of this file.

Functions
int	main (int argc, char **argv)

Detailed Description

Example program to plot various M-Estimators.

Author

mdejong

Definition in file JMEstimator.cc.

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 21 of file JMEstimator.cc.

{
  using namespace std;

  string              outputFile;
  string              mestimator;
  bool                debug;

  try {

    JParser<> zap("Example program to plot various M-Estimators.");

    zap['o'] = make_field(outputFile)  = "mestimator.root";
    zap['M'] = make_field(mestimator)  = "normal", "lorentzian", "tukey", "linear"; 
    zap['d'] = make_field(debug);

    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }


  using namespace JPP;

  JMEstimator* me = NULL;

  if      (mestimator == "normal") 
    me = new JMEstimatorNormal();
  else if (mestimator == "lorentzian") 
    me = new JMEstimatorLorentzian();
  else if (mestimator == "tukey") 
    me = new JMEstimatorTukey(5.0);
  else if (mestimator == "linear") 
    me = new JMEstimatorLinear();
  else
    FATAL("Missing M-Estimator.");
  

  TFile out(outputFile.c_str(), "recreate");

  TH1D h0("rho", NULL, 4000, -10.0, +10.0);
  TH1D h1("psi", NULL, 4000, -10.0, +10.0);

  for (int i = 1; i <= h0.GetNbinsX(); ++i) {

    const double x   = h0.GetBinCenter(i);

    const double rho = me->getRho(x);
    const double psi = me->getPsi(x);

    h0.SetBinContent(i, rho);
    h1.SetBinContent(i, psi);
  }

  delete me;

  out.Write();
  out.Close();
}