/*
  Copyright   : The FASTMC and SPHMC Collaboration
  Author      : Ionut Cristian Arsene 
  Affiliation : Oslo University, Norway & Institute for Space Sciences, Bucharest, Romania
  e-mail      : i.c.arsene@fys.uio.no
  Date        : 2007/05/30

  This class is using the particle and decay lists provided by the 
  THERMINATOR (Computer Physics Communications 174 669 (2006)) and
  SHARE (Computer Physics Communications 167 229 (2005)) collaborations.
*/

#ifndef DECAY_CHANNEL
#include "DecayChannel.h"
#endif

#include <iostream>
using std::cout;
using std::endl;

DecayChannel::DecayChannel() :
  fMotherPDG(kNonsensePDG),
  fBranchingRatio(0.0),
  fNDaughters(0)
{
  for(Int_t i=0; i<kMaxDaughters; i++)
    fDaughtersPDG[i] = kNonsensePDG;
}

DecayChannel::DecayChannel(const DecayChannel &copy):
  fMotherPDG(copy.fMotherPDG),
  fBranchingRatio(copy.fBranchingRatio),
  fNDaughters(copy.fNDaughters)
{
  for(Int_t i=0; i<fNDaughters; i++)
    fDaughtersPDG[i] = copy.fDaughtersPDG[i];
}

DecayChannel::DecayChannel(Int_t mother, Double_t branching, Int_t nDaughters, Int_t *daughters):
  fMotherPDG(mother),
  fBranchingRatio(branching),
  fNDaughters(0)
{
  for(Int_t i=0; i<nDaughters; i++) {
    if(i >= kMaxDaughters) {
      cout << "ERROR in DecayChannel explicit constructor: " << endl;
      cout << "Number of daughters bigger than the maximum allowed one (" << kMaxDaughters << ") !!" << endl;
    }
    fDaughtersPDG[fNDaughters++] = *(daughters+i);
  }
}

void DecayChannel::SetDaughters(Int_t *daughters, Int_t n) {
  for(Int_t i=0; i<n; i++) {
    if(i >= kMaxDaughters) {
      cout << "ERROR in DecayChannel::SetDaughters() :" << endl;
      cout << "Number of daughters bigger than the maximum allowed one (" << kMaxDaughters << ") !!" << endl;
    }
    fDaughtersPDG[fNDaughters++] = *(daughters+i);
  }
}

void DecayChannel::AddDaughter(Int_t pdg) {
  if(fNDaughters >= kMaxDaughters) {
    cout << "ERROR in DecayChannel::AddDaughter() :" << endl;
    cout << "Number of daughters is already >= than the maximum allowed one (" << kMaxDaughters << ") !!" << endl;
  }
  fDaughtersPDG[fNDaughters++] = pdg;
}

Int_t DecayChannel::GetDaughterPDG(Int_t i) {
  if((i >= fNDaughters) || (i<0)) {
    cout << "ERROR in DecayChannel::GetDaughterPDG() :" << endl;
    cout << "Daughter index required is too big or less than zero!! There are only " << fNDaughters << " secondaries in this channel !!" << endl;
    return kNonsensePDG;
  }
  return fDaughtersPDG[i];
}