Back to home page

sPhenix code displayed by LXR

 
 

    


Warning, file /analysis/LightFlavorRatios/bco_correction/skim_KFParticle_trees.C was not indexed or was modified since last indexation (in which case cross-reference links may be missing, inaccurate or erroneous).

0001 #include "V0DuplicateReader.h"
0002 
0003 void skim_KFParticle_trees()
0004 {
0005   TFile* Ks_file = TFile::Open("/sphenix/tg/tg01/hf/cdean/LF_analysis/data_nTuples/output_Kshort_run3pp_looseCuts_20260608.root");
0006   TFile* lambda_file = TFile::Open("/sphenix/tg/tg01/hf/cdean/LF_analysis/data_nTuples/output_Lambda0_run3pp_looseCuts_20260608.root");
0007 
0008   TTree* ks_tree = (TTree*)Ks_file->Get("DecayTree");
0009   TTree* lambda_tree = (TTree*)lambda_file->Get("DecayTree");
0010 
0011   V0DuplicateReader ks_reader(ks_tree, V0DuplicateReader::ParticleType::K0s);
0012   V0DuplicateReader lambda_reader(lambda_tree, V0DuplicateReader::ParticleType::Lambda);
0013 
0014   TFile* Ks_outfile = new TFile("/sphenix/tg/tg01/hf/mjpeters/LightFlavorProduction/bcoSkimmedDataSample/Kshort_run3pp_looseCuts_20260608_bco_skimmed.root","RECREATE");
0015   TTree* ks_outtree = (TTree*)ks_tree->CloneTree(0);
0016 
0017   TFile* lambda_outfile = new TFile("/sphenix/tg/tg01/hf/mjpeters/LightFlavorProduction/bcoSkimmedDataSample/Lambda_run3pp_looseCuts_20260608_bco_skimmed.root","RECREATE");
0018   TTree* lambda_outtree = (TTree*)lambda_tree->CloneTree(0);
0019 
0020   ks_reader.enableDeltaBCOCut(0, 350);
0021   lambda_reader.enableDeltaBCOCut(0, 350);
0022 
0023   for(Long64_t i = 0; i < ks_reader.entries(); ++i)
0024   {
0025     if(i % 10000 == 0) std::cout << "processing BCO for Kshorts entry " << i << " / " << ks_reader.entries() << std::endl;
0026     ks_reader.loadEntry(i);
0027 
0028     if(ks_reader.passesDeltaBCOCut() && ks_reader.isCurrentEntryUnique())
0029     {
0030       ks_outtree->Fill();
0031     }
0032   }
0033 
0034   for(Long64_t i = 0; i < lambda_reader.entries(); ++i)
0035   {
0036     if(i % 10000 == 0) std::cout << "processing BCO for lambda entry " << i << " / " << lambda_reader.entries() << std::endl;
0037     lambda_reader.loadEntry(i);
0038 
0039     if(lambda_reader.passesDeltaBCOCut() && lambda_reader.isCurrentEntryUnique())
0040     {
0041       lambda_outtree->Fill();
0042     }
0043   }
0044 
0045   Ks_outfile->cd();
0046   ks_outtree->Write();
0047   lambda_outfile->cd();
0048   lambda_outtree->Write();
0049 }