#include "Stntuple/run1/StnRun1InitDataBlocks.hh"
#include "Stntuple/run1/TStnRun1Event.hh"
#include "Stntuple/obj/TStnJet.hh"
#include "Stntuple/obj/TStnJetBlock.hh"


//_____________________________________________________________________________
Int_t StnRun1InitJet(TStnJet* Jet, StnJetsBlock_t* R1Jet, Int_t I) 
{
  //  init jet object from a UC standard Ntuple
  //  use the same format for the fat jets, however zero the missing part
  //  flag = cone*10

  Jet->fMomentum.SetXYZT(R1Jet->Jtp4[I][0],R1Jet->Jtp4[I][1],
			 R1Jet->Jtp4[I][2],R1Jet->Jtp4[I][3]);

  Jet->fType      = TStnJet::kJet;
  Jet->fEt        = R1Jet->Jtet[I];   
  Jet->fM         = R1Jet->Jtm[I];    
  Jet->fEveta     = R1Jet->Jteveta[I];
  Jet->fDteta     = R1Jet->Jtdteta[I];
  Jet->fPhi       = R1Jet->Jtphi[I];  
  Jet->fEmf       = R1Jet->Jtemf[I];  
  Jet->fWord      = R1Jet->Jtword[I]; 
  Jet->fCorfm     = R1Jet->Jtcorfm[I];
  Jet->fCorfd     = R1Jet->Jtcorfd[I];
  Jet->fCorfa     = R1Jet->Jtcorfa[I];
  Jet->fNtr       = R1Jet->Jtntr[I];  
  Jet->fNtrp      = R1Jet->Jtntrp[I]; 
  Jet->fEp        = R1Jet->Jtep[I];   
  Jet->fNjtrk0    = R1Jet->Jtnjtrk0[I];
  Jet->fTag       = R1Jet->Jttag[I];  
  Jet->fEgrd      = R1Jet->Jtegrd[I]; 
  Jet->fVbpb      = R1Jet->Jvbpb[I];  
  Jet->fPbpb      = R1Jet->Jpbpb[I];  
  Jet->fScbpb     = R1Jet->Scbpb[I];  
  Jet->fScnpb     = R1Jet->Scnpb[I];  
  Jet->fPppb      = R1Jet->Jpppb[I];  
  Jet->fPnpb      = R1Jet->Jpnpb[I];  
  Jet->fTau1      = R1Jet->Jtau1[I];  
  Jet->fTau2      = R1Jet->Jtau2[I];  
  Jet->fZvtx      = R1Jet->Jtzvtx[I]; 
  Jet->fDz        = R1Jet->Jtdzvtx[I];
  Jet->fPtout     = R1Jet->Jt_ptout[I];
  Jet->fVpt.Set(20,R1Jet->Jtvpt[I]);
  Jet->fNovpt     = R1Jet->Jtnovpt[I];

  Jet->fMatch     = -1;

  return 0;
}



//_____________________________________________________________________________
int StnRun1InitJetBlock(TStnDataBlock*  Block, TStnEvent*  Event, Int_t Mode) 
{
  // initialize jet data block starting from Run1 UC Stntuple

  TStnRun1Event* event = (TStnRun1Event*) Event;

  TStnJetBlock* blk = (TStnJetBlock*) Block;
  blk->Clear();

  StnJetsBlock_t*    r1jet    = event->JetsBlock();
  StnGeneralBlock_t* r1header = event->GeneralBlock();

  blk->fNJet10    = r1header->Njet10;  
  blk->fNJet15    = r1header->Njet15;  
  blk->fNJet20    = r1header->Njet20;  
  blk->fNJet25    = r1header->Njet25;  

  for (int i=0; i<r1jet->Njt; i++) {
    TStnJet* jet = blk->NewJet();
    StnRun1InitJet(jet,r1jet,i);
  }
  return 0;
}