UserCode/RootMacros/plotCorrespondingHists.C

/*************************************************
  Automatically plots histograms from two or more
  files onto the same plot and saves them.

  This first looks for ALL histograms in the first
  file then and plots the corresponding histograms 
  from the rest of the files onto the sample plot.

  Images are saved into a directory structure 
  that copies the structure of the root files.
  
  Can be run from a bash prompt as well:
    root -b -l -q "plotCorrespondingHists.C(\"fileA.root\",\"fileB.root\")"
    root -b -l -q "plotCorrespondingHists.C(\"fileA.root\",\"fileB.root\",\"fileC.root\")"

  Michael B. Anderson
  Sept 29, 2009
*************************************************/


// *******************************************
// Variables
TString imageType = "gif";
TString outputFolder = "HistogramsTogether/";
int plotWidth   = 480;
int plotHeight  = 360;
bool plotLogY = false;
bool areaNormalize = true;
// Turn off stats box                                                                                         
gStyle->SetOptStat(0);
// End of Variables
// *******************************************

void plotCorrespondingHists(TString fileName1, TString fileName2) {
  vector<TString> fileNames;
  fileNames.push_back( fileName1 ) ;
  fileNames.push_back( fileName2 ) ;
  plotCorrespondingHists( fileNames );
}

void plotCorrespondingHists(TString fileName1, TString fileName2, TString fileName3) {
  vector<TString> fileNames;
  fileNames.push_back( fileName1 ) ;
  fileNames.push_back( fileName2 ) ;
  fileNames.push_back( fileName3 ) ;
  plotCorrespondingHists( fileNames );
}

void plotCorrespondingHists(TString fileName1, TString fileName2, TString fileName3, TString fileName4) {
  vector<TString> fileNames;
  fileNames.push_back( fileName1 ) ;
  fileNames.push_back( fileName2 ) ;
  fileNames.push_back( fileName3 ) ;
  fileNames.push_back( fileName4 ) ;
  plotCorrespondingHists( fileNames );
}

void plotCorrespondingHists(vector<TString> fileName) {

  TString firstFileName = fileName[0];

  if (firstFileName == "" ) return;

  // Open all the files & create a list of Labels
  TList *fileList = new TList();
  TList *listOfLabels = new TList();
  for (int i=0; i<fileName.size(); i++) {
    cout << "Opening " << fileName[i] << " ..." << endl;
    TFile *tempFile = new TFile(fileName[i], "READ");
    if ( tempFile->IsZombie() ) {
      cout << "Error opening " << fileName[i] << endl;
      return;
    }
    fileList->Add( tempFile );
    listOfLabels->Add( new TObjString(fileName[i].ReplaceAll(".root","")) );
  }

  // Create a list of all the histograms in the file
  TList *listOfHistograms = new TList();
  TFile *firstFile = (TFile*)fileList->First();
  gSystem->MakeDirectory(outputFolder);
  recurseOverKeys( firstFile, listOfHistograms );

  cout << "Found " << listOfHistograms->GetSize() << " histograms in " << firstFileName << endl;

  // Loop over histograms, then loop over files
  //  plot the same histogram from all the files together
  //  save the canvas
  TCanvas *c1 = new TCanvas("c1", "c1",0,0,plotWidth,plotHeight);
  TObjString *currentHistString = (TObjString*)listOfHistograms->First();
  for (int i_hist=0; i_hist<listOfHistograms->GetSize(); i_hist++) {
    TFile *current_file = (TFile*)fileList->First();

    // Loop over the files, building list of histograms
    TList *listOfHistsToPlot = new TList();
    for (int i_file=0; i_file<fileName.size(); i_file++) {
      TH1* htemp;
      //cout << currentHistString->GetString() << endl;
      TString currentHistName = currentHistString->GetString();
      if (currentHistName[0] == '/') {
        currentHistName = currentHistName(1,currentHistName.Length());
      }
      current_file->GetObject(currentHistName, htemp);

      if ( !htemp ) {
        cout << "Error getting " << currentHistString->GetString() << " from " << current_file->GetName() << endl;
        return;
      }

      listOfHistsToPlot->Add(htemp);
      current_file = (TFile*)fileList->After( current_file );
    }

    // Created a list of histograms to plot, now plot and save file
    plotHists(listOfHistsToPlot, listOfLabels, c1);
    c1->SaveAs(outputFolder+currentHistString->GetString()+"."+imageType);

    currentHistString = (TObjString*)listOfHistograms->After(currentHistString);
  }

  // Close all the files
  TFile *current_file = (TFile*)fileList->First();
  for (int i=0; i<fileName.size(); i++) {
    cout << "Closing " << fileName[i] << " ..." << endl;
    current_file->Close();
    current_file = (TFile*)fileList->After( current_file );
  }
}

void recurseOverKeys( const TDirectory *target, TList *listOfHistograms) {

  // Figure out where we are
  TString path( (char*)strstr( target->GetPath(), ":" ) );
  path.Remove( 0, 2 );

  TDirectory *current_sourcedir = gDirectory;

  TKey *key;
  TIter nextkey(current_sourcedir->GetListOfKeys());

  while (key = (TKey*)nextkey()) {

    obj = key->ReadObj();

    // Check if this is a 1D histogram or a directory
    if (obj->IsA()->InheritsFrom("TH1F")) {

      TH1F *htemp = (TH1F*)obj;
      TString histName = htemp->GetName();
      TObjString *histToAdd = new TObjString(path+"/"+histName);
      listOfHistograms->Add(histToAdd);

    } else if ( obj->IsA()->InheritsFrom( "TDirectory" ) ) {
      // it's a subdirectory

      const TDirectory *tempDir = (TDirectory*)obj;
      current_sourcedir->cd ( tempDir->GetName() );

      gSystem->MakeDirectory(outputFolder+path+"/"+tempDir->GetName());

      // obj still knows its depth within the target file via
      // GetPath(), so we can still figure out where we are in the recursion
      recurseOverKeys( tempDir , listOfHistograms );

    } // end of IF a TDriectory
  }
} 

void plotHists(TList *histogramsToPlot, 
               TList *histogramLabels,
               TCanvas *c1) {


  //*************************************************
  // Variables
  vector<int> histColors;
  histColors.push_back(kBlack); // change colors as you like
  histColors.push_back(kBlue);
  histColors.push_back(kRed);
  histColors.push_back(kGreen-1);

  float histLineWidth = 2.0;

  bool useMarkers = false;         // search online for TAttMarker
  vector<int> histMarkerStyles;    // to see all available markers
  histMarkerStyles.push_back( 20 );
  histMarkerStyles.push_back( 22 );
  histMarkerStyles.push_back( 21 );
  histMarkerStyles.push_back( 23 );
  float histMarkerSize = 0.9;

  // END of Variables
  //*************************************************

  // Create copies of histograms provided
  //  and area normalize, if requested
  TList *hists = new TList();
  TH1* currentHistogram = (TH1*)histogramsToPlot->First();
  for (int i=0; i<histogramsToPlot->GetSize(); i++) {
    TH1* tempCurrentClone = (TH1*)currentHistogram->Clone();
    hists->Add( tempCurrentClone );
    if (areaNormalize) {
      Double_t integral = tempCurrentClone->Integral();
      if (integral>0.0) tempCurrentClone->Scale(1/integral);
    }
    currentHistogram = (TH1*)histogramsToPlot->After( currentHistogram );
  }


  // Set histogram plotting variables
  //  and add them to the histogram stack & legend
  THStack *tempStack = new THStack();
  TLegend *infoBox = new TLegend(0.75, 0.83, 0.99, 0.99, "");
  currentHistogram = (TH1*)hists->First();
  TObjString *currentHistString = (TObjString*)histogramLabels->First();
  for (int i=0; i<histogramsToPlot->GetSize(); i++) {

    currentHistogram->SetLineColor(histColors[i]);
    currentHistogram->SetLineWidth(histLineWidth);

    if (useMarkers) {
      currentHistogram->SetMarkerStyle(histMarkerStyles[i]);
      currentHistogram->SetMarkerColor(histColors[i]);
      currentHistogram->SetMarkerSize(histMarkerSize);
    }

    infoBox->AddEntry(currentHistogram,currentHistString->GetString(),"LP");
    currentHistString = (TObjString*)histogramLabels->After( currentHistString);

    tempStack->Add(currentHistogram);

    currentHistogram = (TH1*)hists->After( currentHistogram );
  }

  // Draw the stack of histograms
  tempStack->Draw("nostack");

  // Set title/label sizes and offsets
  tempStack->GetXaxis()->SetTitleOffset(0.9);
  tempStack->GetYaxis()->SetTitleOffset(1.2);
  tempStack->GetXaxis()->SetLabelSize(0.04);
  tempStack->GetYaxis()->SetLabelSize(0.05);
  tempStack->GetXaxis()->SetTitleSize(0.06);
  tempStack->GetYaxis()->SetTitleSize(0.05);

  // Set axis titles
  TString title = ((TH1*)hists->First())->GetTitle();
  TString xTitle = ((TH1*)hists->First())->GetXaxis()->GetTitle();
  TString yTitle = ((TH1*)hists->First())->GetYaxis()->GetTitle();
  tempStack->SetTitle(title+";"+xTitle+";"+yTitle);

  // Draw legend
  infoBox->SetShadowColor(0);  // 0 = transparent
  //infoBox->SetLineColor(0);
  infoBox->SetFillColor(kWhite);
  //infoBox->SetFillStyle(0);
  infoBox->Draw();

  c1->SetLogy(plotLogY);
}
Revision:	1.2
Committed:	Wed Oct 28 01:47:36 2009 UTC (15 years, 6 months ago) by anderson
Content type:	text/plain
Branch:	MAIN
CVS Tags:	HEAD
Changes since 1.1:	+17 -3 lines
Log Message:	Added check for non-existing file. Added fix for histograms in main dir
#	User	Rev	Content
1	anderson	1.1	/*************************************************
2			Automatically plots histograms from two or more
3			files onto the same plot and saves them.
4
5			This first looks for ALL histograms in the first
6			file then and plots the corresponding histograms
7			from the rest of the files onto the sample plot.
8
9			Images are saved into a directory structure
10			that copies the structure of the root files.
11
12			Can be run from a bash prompt as well:
13			root -b -l -q "plotCorrespondingHists.C(\"fileA.root\",\"fileB.root\")"
14			root -b -l -q "plotCorrespondingHists.C(\"fileA.root\",\"fileB.root\",\"fileC.root\")"
15
16			Michael B. Anderson
17			Sept 29, 2009
18			*************************************************/
19
20
21
22			// *******************************************
23			// Variables
24			TString imageType = "gif";
25			TString outputFolder = "HistogramsTogether/";
26			int plotWidth = 480;
27			int plotHeight = 360;
28			bool plotLogY = false;
29	anderson	1.2	bool areaNormalize = true;
30	anderson	1.1	// Turn off stats box
31			gStyle->SetOptStat(0);
32			// End of Variables
33			// *******************************************
34
35			void plotCorrespondingHists(TString fileName1, TString fileName2) {
36			vector<TString> fileNames;
37			fileNames.push_back( fileName1 ) ;
38			fileNames.push_back( fileName2 ) ;
39			plotCorrespondingHists( fileNames );
40			}
41
42			void plotCorrespondingHists(TString fileName1, TString fileName2, TString fileName3) {
43			vector<TString> fileNames;
44			fileNames.push_back( fileName1 ) ;
45			fileNames.push_back( fileName2 ) ;
46			fileNames.push_back( fileName3 ) ;
47			plotCorrespondingHists( fileNames );
48			}
49
50			void plotCorrespondingHists(TString fileName1, TString fileName2, TString fileName3, TString fileName4) {
51			vector<TString> fileNames;
52			fileNames.push_back( fileName1 ) ;
53			fileNames.push_back( fileName2 ) ;
54			fileNames.push_back( fileName3 ) ;
55			fileNames.push_back( fileName4 ) ;
56			plotCorrespondingHists( fileNames );
57			}
58
59			void plotCorrespondingHists(vector<TString> fileName) {
60
61			TString firstFileName = fileName[0];
62
63			if (firstFileName == "" ) return;
64
65			// Open all the files & create a list of Labels
66			TList *fileList = new TList();
67			TList *listOfLabels = new TList();
68			for (int i=0; i<fileName.size(); i++) {
69			cout << "Opening " << fileName[i] << " ..." << endl;
70	anderson	1.2	TFile *tempFile = new TFile(fileName[i], "READ");
71			if ( tempFile->IsZombie() ) {
72			cout << "Error opening " << fileName[i] << endl;
73			return;
74			}
75			fileList->Add( tempFile );
76	anderson	1.1	listOfLabels->Add( new TObjString(fileName[i].ReplaceAll(".root","")) );
77			}
78
79			// Create a list of all the histograms in the file
80			TList *listOfHistograms = new TList();
81			TFile firstFile = (TFile)fileList->First();
82			gSystem->MakeDirectory(outputFolder);
83			recurseOverKeys( firstFile, listOfHistograms );
84
85			cout << "Found " << listOfHistograms->GetSize() << " histograms in " << firstFileName << endl;
86
87			// Loop over histograms, then loop over files
88			// plot the same histogram from all the files together
89			// save the canvas
90			TCanvas *c1 = new TCanvas("c1", "c1",0,0,plotWidth,plotHeight);
91			TObjString currentHistString = (TObjString)listOfHistograms->First();
92			for (int i_hist=0; i_hist<listOfHistograms->GetSize(); i_hist++) {
93			TFile current_file = (TFile)fileList->First();
94
95			// Loop over the files, building list of histograms
96			TList *listOfHistsToPlot = new TList();
97			for (int i_file=0; i_file<fileName.size(); i_file++) {
98			TH1* htemp;
99			//cout << currentHistString->GetString() << endl;
100	anderson	1.2	TString currentHistName = currentHistString->GetString();
101			if (currentHistName[0] == '/') {
102			currentHistName = currentHistName(1,currentHistName.Length());
103			}
104			current_file->GetObject(currentHistName, htemp);
105
106			if ( !htemp ) {
107			cout << "Error getting " << currentHistString->GetString() << " from " << current_file->GetName() << endl;
108			return;
109			}
110	anderson	1.1
111			listOfHistsToPlot->Add(htemp);
112			current_file = (TFile*)fileList->After( current_file );
113			}
114
115			// Created a list of histograms to plot, now plot and save file
116			plotHists(listOfHistsToPlot, listOfLabels, c1);
117			c1->SaveAs(outputFolder+currentHistString->GetString()+"."+imageType);
118
119			currentHistString = (TObjString*)listOfHistograms->After(currentHistString);
120			}
121
122			// Close all the files
123			TFile current_file = (TFile)fileList->First();
124			for (int i=0; i<fileName.size(); i++) {
125			cout << "Closing " << fileName[i] << " ..." << endl;
126			current_file->Close();
127			current_file = (TFile*)fileList->After( current_file );
128			}
129			}
130
131			void recurseOverKeys( const TDirectory target, TList listOfHistograms) {
132
133			// Figure out where we are
134			TString path( (char*)strstr( target->GetPath(), ":" ) );
135			path.Remove( 0, 2 );
136
137			TDirectory *current_sourcedir = gDirectory;
138
139			TKey *key;
140			TIter nextkey(current_sourcedir->GetListOfKeys());
141
142			while (key = (TKey*)nextkey()) {
143
144			obj = key->ReadObj();
145
146			// Check if this is a 1D histogram or a directory
147			if (obj->IsA()->InheritsFrom("TH1F")) {
148
149			TH1F htemp = (TH1F)obj;
150			TString histName = htemp->GetName();
151			TObjString *histToAdd = new TObjString(path+"/"+histName);
152			listOfHistograms->Add(histToAdd);
153
154			} else if ( obj->IsA()->InheritsFrom( "TDirectory" ) ) {
155			// it's a subdirectory
156
157			const TDirectory tempDir = (TDirectory)obj;
158			current_sourcedir->cd ( tempDir->GetName() );
159
160			gSystem->MakeDirectory(outputFolder+path+"/"+tempDir->GetName());
161
162			// obj still knows its depth within the target file via
163			// GetPath(), so we can still figure out where we are in the recursion
164			recurseOverKeys( tempDir , listOfHistograms );
165
166			} // end of IF a TDriectory
167			}
168			}
169
170			void plotHists(TList *histogramsToPlot,
171			TList *histogramLabels,
172			TCanvas *c1) {
173
174
175			//*************************************************
176			// Variables
177			vector<int> histColors;
178			histColors.push_back(kBlack); // change colors as you like
179			histColors.push_back(kBlue);
180			histColors.push_back(kRed);
181			histColors.push_back(kGreen-1);
182
183			float histLineWidth = 2.0;
184
185			bool useMarkers = false; // search online for TAttMarker
186			vector<int> histMarkerStyles; // to see all available markers
187			histMarkerStyles.push_back( 20 );
188			histMarkerStyles.push_back( 22 );
189			histMarkerStyles.push_back( 21 );
190			histMarkerStyles.push_back( 23 );
191			float histMarkerSize = 0.9;
192
193			// END of Variables
194			//*************************************************
195
196			// Create copies of histograms provided
197			// and area normalize, if requested
198			TList *hists = new TList();
199			TH1* currentHistogram = (TH1*)histogramsToPlot->First();
200			for (int i=0; i<histogramsToPlot->GetSize(); i++) {
201			TH1* tempCurrentClone = (TH1*)currentHistogram->Clone();
202			hists->Add( tempCurrentClone );
203			if (areaNormalize) {
204			Double_t integral = tempCurrentClone->Integral();
205			if (integral>0.0) tempCurrentClone->Scale(1/integral);
206			}
207			currentHistogram = (TH1*)histogramsToPlot->After( currentHistogram );
208			}
209
210
211			// Set histogram plotting variables
212			// and add them to the histogram stack & legend
213			THStack *tempStack = new THStack();
214			TLegend *infoBox = new TLegend(0.75, 0.83, 0.99, 0.99, "");
215			currentHistogram = (TH1*)hists->First();
216			TObjString currentHistString = (TObjString)histogramLabels->First();
217			for (int i=0; i<histogramsToPlot->GetSize(); i++) {
218
219			currentHistogram->SetLineColor(histColors[i]);
220			currentHistogram->SetLineWidth(histLineWidth);
221
222			if (useMarkers) {
223			currentHistogram->SetMarkerStyle(histMarkerStyles[i]);
224			currentHistogram->SetMarkerColor(histColors[i]);
225			currentHistogram->SetMarkerSize(histMarkerSize);
226			}
227
228			infoBox->AddEntry(currentHistogram,currentHistString->GetString(),"LP");
229			currentHistString = (TObjString*)histogramLabels->After( currentHistString);
230
231			tempStack->Add(currentHistogram);
232
233			currentHistogram = (TH1*)hists->After( currentHistogram );
234			}
235
236			// Draw the stack of histograms
237			tempStack->Draw("nostack");
238
239			// Set title/label sizes and offsets
240			tempStack->GetXaxis()->SetTitleOffset(0.9);
241			tempStack->GetYaxis()->SetTitleOffset(1.2);
242			tempStack->GetXaxis()->SetLabelSize(0.04);
243			tempStack->GetYaxis()->SetLabelSize(0.05);
244			tempStack->GetXaxis()->SetTitleSize(0.06);
245			tempStack->GetYaxis()->SetTitleSize(0.05);
246
247			// Set axis titles
248			TString title = ((TH1*)hists->First())->GetTitle();
249			TString xTitle = ((TH1*)hists->First())->GetXaxis()->GetTitle();
250			TString yTitle = ((TH1*)hists->First())->GetYaxis()->GetTitle();
251			tempStack->SetTitle(title+";"+xTitle+";"+yTitle);
252
253			// Draw legend
254			infoBox->SetShadowColor(0); // 0 = transparent
255			//infoBox->SetLineColor(0);
256			infoBox->SetFillColor(kWhite);
257			//infoBox->SetFillStyle(0);
258			infoBox->Draw();
259
260			c1->SetLogy(plotLogY);
261			}