// RecoTrackNtuple.cpp

#include "RecoTrackNtuple.h"
#include "trffit/RTrackMatch.h"

using std::vector;
using std::string;

using trf::RTrackList;
using trf::RTrackMatch;
using trf::RecoTrackNtuple;

//**********************************************************************
// Local definitions.
//**********************************************************************

namespace {

// Ntuple names.
vector<string> names;

int name_count = 0;

}  // end unnamed namespoace

//**********************************************************************
// Member functions.
//**********************************************************************

// constructor

RecoTrackNtuple::RecoTrackNtuple() {

}

//**********************************************************************

// Fill tuple from a reco match.

void RecoTrackNtuple::fill_match(const RTrackMatch& match) {

  // Fetch the matches.
  RTrackList rtracks1 = match.get_matched1();
  RTrackList rtracks2 = match.get_matched2_in_order1();

  // Loop over matched tracks and fill tuple.
  assert( rtracks1.size() == rtracks2.size() );
  RTrackList::const_iterator itr1 = rtracks1.begin();
  for ( RTrackList::const_iterator itr2=rtracks2.begin();
        itr2!=rtracks2.end(); ++itr2 )
    fill_reco_pair( **itr2, **itr1++ );

  // Fill tuple for unmatched reco tracks from the first set.
  RTrackList rmiss1 = match.get_unmatched1();
  for ( RTrackList::const_iterator itrr=rmiss1.begin();
        itrr!=rmiss1.end(); ++itrr ) {
    fill_reco1( **itrr );
  }

  // Fill tuple for unmatched reco tracks from the second set.
  RTrackList rmiss2 = match.get_unmatched1();
  for ( RTrackList::const_iterator itrr=rmiss2.begin();
        itrr!=rmiss2.end(); ++itrr ) {
    fill_reco2( **itrr );
  }

}

//**********************************************************************

// Fill ntuple from first list of (unmatched) reconstructed tracks.

void RecoTrackNtuple::fill_list1(const RTrackList& rtrks) {
  for ( RTrackList::const_iterator itrr=rtrks.begin();
        itrr!=rtrks.end(); ++itrr ) fill_reco1(**itrr);
}

//**********************************************************************

// Fill ntuple from second list of (unmatched) reconstructed tracks.

void RecoTrackNtuple::fill_list2(const RTrackList& rtrks) {
  for ( RTrackList::const_iterator itrr=rtrks.begin();
        itrr!=rtrks.end(); ++itrr ) fill_reco2(**itrr);
}

//**********************************************************************