JGeometry3D/JLegendre.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <iomanip>

#include "TRandom3.h"

#include "JMath/JLegendre.hh"
#include "JMath/JMathSupportkit.hh"

#include "JGeometry3D/JQuaternion3D.hh"
#include "JGeometry3D/JEigen3D.hh"

#include "JTools/JQuantile.hh"

#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"


/**
 * \file
 *
 * Example program to test Legendre polynome.
 * \author mdejong
 */
int main(int argc, char**argv)
{
  using namespace std;
  using namespace JPP;

  const double xmin = -10.0;
  const double xmax = +10.0;

  typedef JLegendre<JQuaternion3D>  JLegendre_t;

  unsigned int  numberOfEvents;
  unsigned int  numberOfBins;
  JLegendre_t   f1(xmin, xmax);
  int           debug;

  try {

    JParser<> zap("Example program to test Legendre polynome.");

    zap['n'] = make_field(numberOfEvents)   =  0;
    zap['N'] = make_field(numberOfBins)     = 10;
    zap['L'] = make_field(f1);
    zap['d'] = make_field(debug)            = 3;

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


  for (JLegendre_t::iterator i = f1.begin(); i != f1.end(); ++i) {

    i->normalise();

    STATUS("function: " << *i << endl);
  }


  map<double, JQuaternion3D> data;

  for (unsigned int i = 0; i != numberOfBins; ++i) {

    const double x = xmin + i * (xmax - xmin) / (numberOfBins - 1);

    data[x] = f1(x);
  }


  for (map<double, JQuaternion3D>::const_iterator i = data.begin(); i != data.end(); ++i) {
    DEBUG("data: " 
          << FIXED(7,3) << i->first  << " "
          << i->second               << "; " 
          << 360.0 * acos(i->second.getA()) / PI << endl);
  }


  const size_t N = 2;

  JLegendre<JQuaternion3D, N> g1(data.begin(), data.end());

  for (size_t n = 0; n != N + 1; ++n) {

    STATUS("Legendre: " << setw(2) << n << ' ' << g1[n]);

    if (n < f1.size()) {
      STATUS(' ' << FIXED(7,3) << getAngle(f1[n], g1[n]));
    }

    STATUS(endl);
  }


  if (numberOfEvents > 0) {

    JQuantile Q;

    for (unsigned int i = 0; i != numberOfEvents; ++i) {

      const double        x = gRandom->Uniform(xmin, xmax);
      const JQuaternion3D y = f1(x);
      const JQuaternion3D z = g1(x);

      Q.put(getAngle(y,z));
    }

    Q.print(cout);
  }

  return 0;
}