ARA ROOT Test BEd Software

analysis/combineFFTHistos.cxx

//Includes
#include <iostream>

#include "TFile.h"
#include "TTree.h"
#include "TH1.h"
#include "TH1D.h"
#include "TH2.h"
#include "TH2D.h"
#include "TCanvas.h"

#include "araDefines.h"

//Global variabls
TH2D *fftAvTime[RFCHANS_PER_STATION];

int main(int argc, char **argv){

  if(argc<2) {
    std::cout << "Usage\n" << argv[0] << " <input file> \n";
    std::cout << "e.g.\n" << argv[0] << " L2Histos.root summaryHisto.root\n";
    return 0;
  }

  TFile *fpIn = TFile::Open(argv[1], "UPDATE");
  
  //  fpIn->ls();

  TTree *mapTree = (TTree*) fpIn->Get("mapTimes");
  if(!mapTree) {
    std::cerr << "Can't find map Tree\n";
    return -1;
  }

  Long64_t histoTime=0;
  mapTree->SetBranchAddress("histoTime", &histoTime);
  Long64_t numEntries = mapTree->GetEntries();
  Long64_t minT=0, maxT=0;
  mapTree->GetEntry(0);
  minT=histoTime;
  mapTree->GetEntry(numEntries-1);
  maxT=histoTime;

  TH1D *tempHisto;

  //Create the output histogram

  char outHistoName[260];
  Int_t numBinsF=256;
  Double_t minF=0.0;
  Double_t maxF=1000.0;
  minF = minF - ( (maxF-minF)/numBinsF/2.0 ); // adjust histogram edges so that the bin centers
  maxF = maxF + ( (maxF-minF)/numBinsF/2.0 ); // of the histograms are aligned with graph [add bdf]
  numBinsF++;
  
  Long64_t numBinsT=numEntries;
    
  //  printf("minT %llu maxT %llu numBinsT %llu\n", minT, maxT, numBinsT);

  //  TFile *fpOut = new TFile(argv[2], "RECREATE");

  for(int antenna=0;antenna<RFCHANS_PER_STATION;antenna++){
    sprintf(outHistoName, "average_vTime_fft_antenna_%i", antenna);
    //    printf("average_vTime_fft_antenna_%i\n", antenna);
    
    fftAvTime[antenna] = new TH2D(outHistoName, outHistoName, numBinsT, minT, maxT, numBinsF, minF, maxF);
    
  }



  for(int entry=0; entry < numEntries; entry++){

    mapTree->GetEntry(entry);
    //    printf("entry %i histoTime %llu\n", entry, histoTime);
    for(int antenna=0; antenna<RFCHANS_PER_STATION; antenna++){
      char histoDir[260];
      char histoName[260];
      sprintf(histoDir, "time_%llu", histoTime);
      // sprintf(histoName, "average_fft_antenna_%i", antenna );

      sprintf(histoName, "time_%llu/average_fft_antenna_%i", histoTime, antenna );
      //      tempHisto = (TH1D*) fpIn->GetDirectory(histoDir)->Get(histoName);
      tempHisto = (TH1D*) fpIn->Get(histoName);
      if(tempHisto){

        //      std::cerr << "A\n";
        //process the histo
        for(int bin=2; bin<tempHisto->GetNbinsX()+1;bin++){
          Double_t binContent = tempHisto->GetBinContent(bin);
          Double_t binCenter =  tempHisto->GetBinCenter(bin);
          //      printf("histoTime %llu binContent %e binCenter %e\n",histoTime, binCenter, binContent);

          fftAvTime[antenna]->Fill(histoTime, binCenter, binContent);
        }
          
      }

    }
  }

 for(int antenna=0;antenna<RFCHANS_PER_STATION;antenna++){
    if(fftAvTime[antenna]){
      //      printf("%s\n", fftAvTime[antenna]->GetName());
      fftAvTime[antenna]->Write();
      fftAvTime[antenna]->GetXaxis()->SetTimeDisplay(1);
  }
    else printf("histo not present\n");
  }

  
  TCanvas *canHPol= new TCanvas("HPolSummary", "HPolSummary");
  TCanvas *canVPol=new TCanvas("VPolSummary", "VPolSummary");

  canHPol->Divide(4,2);
  canVPol->Divide(4,2);

  canHPol->cd(1);
  fftAvTime[0]->Draw("COLZ");
  canHPol->cd(2);
  fftAvTime[1]->Draw("COLZ");
  canHPol->cd(3);
  fftAvTime[5]->Draw("COLZ");
  canHPol->cd(4);
  fftAvTime[7]->Draw("COLZ");
  canHPol->cd(5);
  fftAvTime[9]->Draw("COLZ");
  canHPol->cd(6);
  fftAvTime[11]->Draw("COLZ");
  canHPol->cd(7);
  fftAvTime[12]->Draw("COLZ");
  canHPol->cd(8);
  fftAvTime[13]->Draw("COLZ");
  canVPol->cd(1);
  fftAvTime[2]->Draw("COLZ");
  canVPol->cd(2);
  fftAvTime[3]->Draw("COLZ");
  canVPol->cd(3);
  fftAvTime[4]->Draw("COLZ");
  canVPol->cd(4);
  fftAvTime[6]->Draw("COLZ");
  canVPol->cd(5);
  fftAvTime[8]->Draw("COLZ");
  canVPol->cd(6);
  fftAvTime[10]->Draw("COLZ");
  canVPol->Write();  
  canHPol->Write();  



 

   fpIn->Close();




  return 0;
}

---

UCL HEP Home   

UHE Neutrino Home   

ARA ROOT Software

Main  

Download  

Branches

3.13  

3.12  

3.11  

3.10  

3.9  

3.8  

3.7  

3.6  

3.5  

3.4  

3.3  

3.2  

TestBed