#include <iostream>
#include <iomanip>

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

#include "JAAnet/JParticleTypes.hh"

#include "JOscProb/JOscChannel.hh"
#include "JOscProb/JOscProbToolkit.hh"


/**
 * \file
 * Auxiliary program to test JOscChannel.
 *
 * \author bjung
 */
int main(int argc, char **argv)
{
  using namespace std;
  using namespace JPP;
  int                     debug;

  try {
    
    JParser<> zap;
    
    zap['d'] = make_field(debug)                                                                 = 3;

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

  // Test comparisons
  
  JOscChannel channel1(JFlavour_t::MUON,
		       JFlavour_t::TAU,
		       JChargeParity_t::PARTICLE);

  NOTICE("Channel1:\n" << channel1 << endl);

  JOscChannel channel2(JFlavour_t::ELECTRON,
		       JFlavour_t::MUON,
		       JChargeParity_t::ANTIPARTICLE);

  NOTICE("Channel2:\n" << channel2 << endl);  

  JOscChannel channel3(JFlavour_t::MUON,
		       JFlavour_t::TAU,
		       JChargeParity_t::ANTIPARTICLE);

  NOTICE("Channel3:\n" << channel3 << endl);  

  ASSERT(channel1 > channel2);
  ASSERT(channel3 > channel2);

  // Test auxiliary functions

  ASSERT(getFlavour((int) TRACK_TYPE_ANTINUE)   == JFlavour_t::ELECTRON);
  ASSERT(getFlavour((int) TRACK_TYPE_ANTINUMU)  == JFlavour_t::MUON);
  ASSERT(getFlavour((int) TRACK_TYPE_ANTINUTAU) == JFlavour_t::TAU);
  ASSERT(getFlavour((int) TRACK_TYPE_NUE)       == JFlavour_t::ELECTRON);
  ASSERT(getFlavour((int) TRACK_TYPE_NUMU)      == JFlavour_t::MUON);
  ASSERT(getFlavour((int) TRACK_TYPE_NUTAU)     == JFlavour_t::TAU);

  ASSERT(getOscProbFlavour(JFlavour_t::ELECTRON) == OscProbFlavour_t::ELECTRON);
  ASSERT(getOscProbFlavour(JFlavour_t::MUON)     == OscProbFlavour_t::MUON);
  ASSERT(getOscProbFlavour(JFlavour_t::TAU)      == OscProbFlavour_t::TAU);  

  ASSERT(getChargeParity((int) TRACK_TYPE_ANTINUE)   == JChargeParity_t::ANTIPARTICLE);
  ASSERT(getChargeParity((int) TRACK_TYPE_ANTINUMU)  == JChargeParity_t::ANTIPARTICLE);
  ASSERT(getChargeParity((int) TRACK_TYPE_ANTINUTAU) == JChargeParity_t::ANTIPARTICLE);
  ASSERT(getChargeParity((int) TRACK_TYPE_NUE)       == JChargeParity_t::PARTICLE);
  ASSERT(getChargeParity((int) TRACK_TYPE_NUMU)      == JChargeParity_t::PARTICLE);
  ASSERT(getChargeParity((int) TRACK_TYPE_NUTAU)     == JChargeParity_t::PARTICLE);
}