#include "CbmAnaDimuonCandidate.h"
#include "CbmAnaMuonCandidate.h"
#include "CbmStsTrack.h"
#include "CbmKFTrack.h"
#include "TLorentzVector.h"

CbmAnaDimuonCandidate::CbmAnaDimuonCandidate(
    CbmAnaMuonCandidate muPlus, 
    CbmAnaMuonCandidate muMinus){
  
  fMuPlus  = CbmAnaMuonCandidate(muPlus);
  fMuMinus = CbmAnaMuonCandidate(muMinus);
}

TLorentzVector CbmAnaDimuonCandidate::GetMomentumMC(){
  TLorentzVector* p1 = fMuPlus.GetMomentumMC();
  TLorentzVector* p2 = fMuMinus.GetMomentumMC();
  TLorentzVector p = (*p1)+(*p2);
  return p;
}

TLorentzVector CbmAnaDimuonCandidate::GetMomentumRC(){
  TLorentzVector* p1 = fMuPlus.GetMomentumRC();
  TLorentzVector* p2 = fMuMinus.GetMomentumRC();
  TLorentzVector p = (*p1)+(*p2);
  return p;
}


ClassImp(CbmAnaDimuonCandidate);