PlotterProgram.cpp

//Program adopted from "DataFitter", primary use for creating Figure 4 in Smolec. May be formed for general use later on?

//V1: Making program able to create figure 5 in smolec for sets A B C and D.

# include <iostream>
# include <fstream>
# include <math.h>
# include <iomanip>
# include <cmath>
# include <stdlib.h>
# include <cstdlib>
//# include <fstream.h>
# include <string.h>
# include <string>
//# include <dos.h> //For Sleep() 

# include "TROOT.h"
# include "TFile.h"
# include "TTree.h"
# include "TBrowser.h"
# include "TH1.h"
# include "TH2.h"
# include "TH3.h"
# include "TRandom.h"

//Gauss user defined: [0]*exp((-(x-[1])*(x-[1]))/(2*[2]*[2]))


int main(){
	
	//////Controls//////
	
	int NumOfInputFiles = 4; //Number of input files plots are made for, useful for SetA through D.
	
	int NumOfLines_1 = 387; //Number of lines in each input file including header if there is one 
	int NumOfLines_2 = 387;
	int NumOfLines_3 = 387;
	int NumOfLines_4 = 380;
	
	char PNGFileName[50] = "Fig4_m06_L50"; //PNG output file name, do not include .png
	char RootFileName[50] = "Fig4_m06_L50"; //Root output file name, do not include .root
	
	char inputFileName1[50] = "ResonanceData_Fig5_FU_FO_SetA.dat"; 
	char inputFileName2[50] = "ResonanceData_Fig5_FU_FO_SetB.dat"; 
	char inputFileName3[50] = "ResonanceData_Fig5_FU_FO_SetC.dat"; 
	char inputFileName4[50] = "ResonanceData_Fig5_FU_FO_SetD.dat"; 
	
	char Label_1[10] = "SetA"; //Labels associated with input files, will be used as suffix to outfile names
	char Label_2[10] = "SetB";
	char Label_3[10] = "SetC";
	char Label_4[10] = "SetD";
	
	//Top plot of Fig4//
	char PlotTitle_1[50] = "Frequency"; //Title of the plot
	char xTitle_1[50] = "Teff"; //Title for x-axis, first column of data file
	char yTitle_1[50] = "Frequency"; //Title for y-axis, second column of data file
	
	double xLowerLimit_1 = 4000; //Limits for top plot in Fig4
	double xUpperLimit_1 = 8000;
	double yLowerLimit_1 = 0;
	double yUpperLimit_1 = 5.0;
	
	//Bottom plot of Fig4//
	char PlotTitle_2[50] = "GrowthRate"; //Title of the plot
	char xTitle_2[50] = "Teff"; //Title for x-axis, first column of data file
	char yTitle_2[50] = "Growth Rate"; //Title for y-axis, second column of data file
	
	double xLowerLimit_2 = 4000; //Limits for bottom plot in Fig4
	double xUpperLimit_2 = 8000;
	double yLowerLimit_2 = -2.0;
	double yUpperLimit_2 = 0.5;
	
	bool DataWithHeaders = true; //If true, will account for headers by using getline function
	bool SavePlotAsPNG = true; //If true, will save plot with file name above.
	bool SavePlotAsRootFile = true; //If true, will save plot in root file with name above.
	
	////////////////////////////
	
	//////Variables//////
	
	ifstream inFile;
	ofstream outFile;
	
	int NumOfLines;
	
	double xMax;
	double xMin;
	double yMax;
	double yMin;
	
	char line[100];
	char header[100];
	char Full_RootFileName[50];
	char Full_PNGFileName[50];
	
	//Dummy Variables//
	
	double dummy1, dummy2, dummy3, dummy4, dummy5, dummy6, dummy7;
	
	////////////////////////////
	
	//////Main Program//////
	for(int f = 0; f < NumOfInputFiles; f++){
		
		//////Variables//////
		
		if(f == 0){
			const int nArray = NumOfLines_1 + 1;
		}
		if(f == 1){
			const int nArray = NumOfLines_2 + 1;
		}
		if(f == 2){
			const int nArray = NumOfLines_3 + 1;
		}
		if(f == 3){
			const int nArray = NumOfLines_4 + 1;
		}
		
		double Teff[nArray];           
		double freq_FU[nArray];
		double freq_FO[nArray];          
		double freq_2O[nArray];
		double freq_ThirdO[nArray];         
		double freq_ForthO[nArray];          
		double growth_FU[nArray];
		double growth_FO[nArray];
		double growth_2O[nArray];
		double growth_ThirdO[nArray];
		double growth_ForthO[nArray];
		
		double freq_BG1[nArray];
		double freq_BG2[nArray];
		double freq_BG3[nArray];
		double freq_BG4[nArray];
		double freq_BG5[nArray];
		
		char inputFileName[50] = "";
		
		/////////////////////
		
		if(f == 0){
			strcpy(inputFileName,inputFileName1);
			if(DataWithHeaders){
				NumOfLines = NumOfLines_1 - 1;
			}
			else{NumOfLines = NumOfLines_1;}
		}
		if(f == 1){
			strcpy(inputFileName,inputFileName2);
			if(DataWithHeaders){
				NumOfLines = NumOfLines_2 - 1;
			}
			else{NumOfLines = NumOfLines_2;}
		}
		if(f == 2){
			strcpy(inputFileName,inputFileName3);
			if(DataWithHeaders){
				NumOfLines = NumOfLines_3 - 1;
			}
			else{NumOfLines = NumOfLines_3;}
		}
		if(f == 3){
			strcpy(inputFileName,inputFileName4);
			if(DataWithHeaders){
				NumOfLines = NumOfLines_4 - 1;
			}
			else{NumOfLines = NumOfLines_4;}
		}
		
		cout<<"Input file name: "<<inputFileName<<endl;
		
		//Getting x,y from file//
		inFile.open(inputFileName,ios::in);
		if(inFile.is_open()){
			cout<<"Input file was opened successfully"<<endl;
		}
		else{cout<<"Input file not found, check name?"<<endl;}
		
		if(inFile.good()){
			if(DataWithHeaders){
				getline(inFile,header);
				cout<<header<<endl;
			}

			for(int i = 0; i < NumOfLines; i++){
				inFile>>Teff[i]>>freq_FU[i]>>freq_FO[i]>>freq_2O[i]>>freq_ThirdO[i]>>freq_ForthO[i]>>growth_FU[i]>>growth_FO[i]>>growth_2O[i]>>growth_ThirdO[i]>>growth_ForthO[i];
				//inFile.ignore(50, '\n'); //This allows for only the defined lines to be read (unless data file is 50+ columns long) 
				cout<<setw(5)<<Teff[i]<<setw(10)<<freq_FU[i]<<setw(10)<<freq_FO[i]<<setw(10)<<freq_2O[i]<<setw(10)<<freq_ThirdO[i]<<setw(10)<<freq_ForthO[i]<<setw(10)<<growth_FU[i]<<setw(10)<<growth_FO[i]<<setw(10)<<growth_2O[i]<<setw(10)<<growth_ThirdO[i]<<setw(10)<<growth_ForthO[i]<<endl;
				
				freq_BG1[i] = 1.5*freq_FU[i];
				freq_BG2[i] = 2.0*freq_FU[i];
				freq_BG3[i] = 2.5*freq_FU[i];
				freq_BG4[i] = 3.5*freq_FU[i];
				freq_BG5[i] = 4.5*freq_FU[i];
				
				
			}
			cout<<"************* End of file ********************"<<endl;
		}
		inFile.close();
	
	
		//////Graphics//////
		
		//Output file name creation//
		if(f == 0){
			strcpy(Full_RootFileName, RootFileName);
			strcat(Full_RootFileName, "_");
			strcat(Full_RootFileName, Label_1);
			strcat(Full_RootFileName, ".root");
			
			strcpy(Full_PNGFileName, PNGFileName);
			strcat(Full_PNGFileName, "_");
			strcat(Full_PNGFileName, Label_1);
			strcat(Full_PNGFileName, ".png");
		}
		if(f == 1){
			strcpy(Full_RootFileName, RootFileName);
			strcat(Full_RootFileName, "_");
			strcat(Full_RootFileName, Label_2);
			strcat(Full_RootFileName, ".root");
			
			strcpy(Full_PNGFileName, PNGFileName);
			strcat(Full_PNGFileName, "_");
			strcat(Full_PNGFileName, Label_2);
			strcat(Full_PNGFileName, ".png");
		}
		if(f == 2){
			strcpy(Full_RootFileName, RootFileName);
			strcat(Full_RootFileName, "_");
			strcat(Full_RootFileName, Label_3);
			strcat(Full_RootFileName, ".root");
			
			strcpy(Full_PNGFileName, PNGFileName);
			strcat(Full_PNGFileName, "_");
			strcat(Full_PNGFileName, Label_3);
			strcat(Full_PNGFileName, ".png");
		}
		if(f == 3){
			strcpy(Full_RootFileName, RootFileName);
			strcat(Full_RootFileName, "_");
			strcat(Full_RootFileName, Label_4);
			strcat(Full_RootFileName, ".root");
			
			strcpy(Full_PNGFileName, PNGFileName);
			strcat(Full_PNGFileName, "_");
			strcat(Full_PNGFileName, Label_4);
			strcat(Full_PNGFileName, ".png");
		}
		
		//Plot creation//
		TCanvas *c1 = new TCanvas("c1","Canvas 1",200,10,900,700);
			c1->Divide(1,2);
			c1->SetFillColor(10);
			c1->SetGrid();
			c1->GetFrame()->SetFillColor(10);
			c1->GetFrame()->SetBorderSize(12);
			c1->Range(0,0,1,1);

			gStyle->SetOptStat(0);
			
		c1->cd(1);
			TH1F *hr1 = c1_1->DrawFrame(xLowerLimit_1,yLowerLimit_1,xUpperLimit_1,yUpperLimit_1);
				hr1->SetTitle(PlotTitle_1);
				hr1->SetXTitle(xTitle_1);
				hr1->SetYTitle(yTitle_1);
				
			//1.5vF//
			TGraphErrors *gr6 = new TGraphErrors(NumOfLines,Teff,freq_BG1,0,0);
				gr6->SetMarkerColor(kBlack);
				gr6->SetMarkerSize(0.5);
				gr6->SetMarkerStyle(7);
				gr6->SetFillColor(kWhite);
				gr6->SetName("gr6");
				gr6->Draw("P");
				
			//2.0vF//
			TGraphErrors *gr7 = new TGraphErrors(NumOfLines,Teff,freq_BG2,0,0);
				gr7->SetMarkerColor(kBlack);
				gr7->SetMarkerSize(0.5);
				gr7->SetMarkerStyle(7);
				gr7->SetFillColor(kWhite);
				gr7->SetName("gr7");
				gr7->Draw("PSAME");
			
			//2.5vF//
			TGraphErrors *gr8 = new TGraphErrors(NumOfLines,Teff,freq_BG3,0,0);
				gr8->SetMarkerColor(kBlack);
				gr8->SetMarkerSize(0.5);
				gr8->SetMarkerStyle(7);
				gr8->SetName("gr8");
				gr8->Draw("PSAME");
			
			//3.5vF//
			TGraphErrors *gr9 = new TGraphErrors(NumOfLines,Teff,freq_BG4,0,0);
				gr9->SetMarkerColor(kBlack);
				gr9->SetMarkerSize(0.5);
				gr9->SetMarkerStyle(7);
				gr9->SetFillColor(kWhite);
				gr9->SetName("gr9");
				gr9->Draw("PSAME");
			
			//4.5vF//
			TGraphErrors *gr10 = new TGraphErrors(NumOfLines,Teff,freq_BG5,0,0);
				gr10->SetMarkerColor(kBlack);
				gr10->SetMarkerSize(0.5);
				gr10->SetMarkerStyle(7);
				gr10->SetFillColor(kWhite);
				gr10->SetName("gr10");
				gr10->Draw("PSAME");
				
			//FU//
			TGraphErrors *gr1 = new TGraphErrors(NumOfLines,Teff,freq_FU,0,0);
				gr1->SetMarkerColor(kRed);
				gr1->SetMarkerSize(0.5);
				gr1->SetMarkerStyle(20);
				gr1->SetFillColor(kWhite);
				gr1->SetName("gr1");
				gr1->Draw("P");
				
			//FO//	
			TGraphErrors *gr2 = new TGraphErrors(NumOfLines,Teff,freq_FO,0,0);
				gr2->SetMarkerColor(kGreen);
				gr2->SetMarkerSize(0.5);
				gr2->SetMarkerStyle(24);
				gr2->SetFillColor(kWhite);
				gr2->SetName("gr2");
				gr2->Draw("PSAME");
			
			//2O//
			TGraphErrors *gr3 = new TGraphErrors(NumOfLines,Teff,freq_2O,0,0);
				gr3->SetMarkerColor(kBlue);
				gr3->SetMarkerSize(0.5);
				gr3->SetMarkerStyle(21);
				gr3->SetFillColor(kWhite);
				gr3->SetName("gr3");
				gr3->Draw("PSAME");
				
			//ThirdO//
			TGraphErrors *gr4 = new TGraphErrors(NumOfLines,Teff,freq_ThirdO,0,0);
				gr4->SetMarkerColor(kMagenta);
				gr4->SetMarkerSize(0.5);
				gr4->SetMarkerStyle(25);
				gr4->SetFillColor(kWhite);
				gr4->SetName("gr4");
				gr4->Draw("PSAME");
				
			//ForthO//
			TGraphErrors *gr5 = new TGraphErrors(NumOfLines,Teff,freq_ForthO,0,0);
				gr5->SetMarkerColor(kCyan);
				gr5->SetMarkerSize(0.5);
				gr5->SetMarkerStyle(22);
				gr5->SetFillColor(kWhite);
				gr5->SetName("gr5");
				gr5->Draw("PSAME");	
				
			TLegend *leg = new TLegend(0.1,0.7,0.28,0.9);
				leg->AddEntry("gr1","F","p");
				leg->AddEntry("gr2","1O","p");
				leg->AddEntry("gr3","2O","p");
				leg->AddEntry("gr4","3O","p");
				leg->AddEntry("gr5","4O","p");
				leg->Draw();
				
			
		c1->cd(2);
			TH1F *hr2 = c1_2->DrawFrame(xLowerLimit_2,yLowerLimit_2,xUpperLimit_2,yUpperLimit_2);
				hr2->SetTitle(PlotTitle_2);
				hr2->SetXTitle(xTitle_2);
				hr2->SetYTitle(yTitle_2);
				
			TGraphErrors *gr11 = new TGraphErrors(NumOfLines,Teff,growth_FU,0,0);
				gr11->SetMarkerColor(kRed);
				gr11->SetMarkerSize(0.5);
				gr11->SetMarkerStyle(20);
				gr11->SetName("gr11");
				gr11->Draw("P");
				
			TGraphErrors *gr12 = new TGraphErrors(NumOfLines,Teff,growth_FO,0,0);
				gr12->SetMarkerColor(kGreen);
				gr12->SetMarkerSize(0.5);
				gr12->SetMarkerStyle(24);
				gr12->SetName("gr12");
				gr12->Draw("PSAME");
				
			TGraphErrors *gr13 = new TGraphErrors(NumOfLines,Teff,growth_2O,0,0);
				gr13->SetMarkerColor(kBlue);
				gr13->SetMarkerSize(0.5);
				gr13->SetMarkerStyle(21);
				gr13->SetName("gr13");
				gr13->Draw("PSAME");
				
			TGraphErrors *gr14 = new TGraphErrors(NumOfLines,Teff,growth_ThirdO,0,0);
				gr14->SetMarkerColor(kMagenta);
				gr14->SetMarkerSize(0.5);
				gr14->SetMarkerStyle(25);
				gr14->SetName("gr14");
				gr14->Draw("PSAME");
				
			TGraphErrors *gr15 = new TGraphErrors(NumOfLines,Teff,growth_ForthO,0,0);
				gr15->SetMarkerColor(kCyan);
				gr15->SetMarkerSize(0.5);
				gr15->SetMarkerStyle(22);
				gr15->SetName("gr15");
				gr15->Draw("PSAME");
		
		
		if(SavePlotAsPNG){
			c1->SaveAs(Full_PNGFileName);
		}
		if(SavePlotAsRootFile){
			TFile outfile1(Full_RootFileName, "RECREATE");
			c1->Write();
			outfile1.Close();
		}
		
	}
	
	return 0;
	
}