[u/mrichter/AliRoot.git] / HLT / EVE / AliHLTEveMultCorr.cxx

#include "AliHLTEveMultCorr.h"
#include "TH1F.h"
#include "TH2F.h"
#include <iostream>
#include <TCanvas.h>
#include "AliHLTHOMERBlockDesc.h"
#include "TList.h"

AliHLTEveMultCorr::AliHLTEveMultCorr(const char* name): AliHLTEveBase(name)
        ,fVzeroMultCanvas(0)
        ,fZdcMultCanvas(0)
        ,fTpcMultCanvas(0)
        ,fCorrCanvas(0)
        ,fEtCorrCanvas(0)
        ,fZdcVzeroSpdCorrCanvas(0)
{

}

AliHLTEveMultCorr::~AliHLTEveMultCorr()
{
}

void AliHLTEveMultCorr::ResetElements()
{

}

void AliHLTEveMultCorr::UpdateElements()
{
    fVzeroMultCanvas->Update();
    fZdcMultCanvas->Update();
    fTpcMultCanvas->Update();
    fCorrCanvas->Update();
    fEtCorrCanvas->Update();
    fZdcVzeroSpdCorrCanvas->Update();

}

void AliHLTEveMultCorr::ProcessBlock(AliHLTHOMERBlockDesc* block)
{
    TList *hlist = dynamic_cast<TList*>(block->GetTObject());
    if (hlist)
    {

        if (hlist->Contains("fVzeroMult")) // These are the correlation histograms
        {
            TH1F *oneDf = 0;
            TH2F *twoDf = 0;
	    int cd = 1;

            //VZERO multiplicity hists
            if (!fVzeroMultCanvas) 
	    {
	      fVzeroMultCanvas = CreateCanvas("V0 Mult", "V0 Multiplicities");
	      fVzeroMultCanvas->Divide(2, 2);
	    }
            oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fVzeroMult"));
            AddHistogramToCanvas(oneDf, fVzeroMultCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fVzeroMultAC"));
            AddHistogramToCanvas(twoDf, fVzeroMultCanvas, cd);
	    cd++;
            oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fVzeroFlaggedMult"));
            AddHistogramToCanvas(oneDf, fVzeroMultCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fVzeroFlaggedMultAC"));
            AddHistogramToCanvas(twoDf, fVzeroMultCanvas, cd);

	    cd = 1;
            //ZDC multiplicity hists
            if (!fZdcMultCanvas) 
	    {
	      fZdcMultCanvas = CreateCanvas("ZDC Mult", "ZDC Multiplicities");
	      fZdcMultCanvas->Divide(2, 4);
	    }
            oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcEzdc"));
            AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
	    cd++;
            oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcEzn"));
            AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
	    cd++;
            oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcEzp"));
            AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
	    cd++;
            oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcEzem"));
            AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
	    cd++;
            oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcNpart"));
            AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
	    cd++;
            oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcB"));
            AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fZdcEzemEzdc"));
            AddHistogramToCanvas(twoDf, fZdcMultCanvas, cd);

	    cd = 1;
            // TPC multiplicity
            if (!fTpcMultCanvas) 
	    {
	      fTpcMultCanvas = CreateCanvas("TPC Mult", "TPC Multiplicities");
	      fTpcMultCanvas->Divide(2);
	    }
            oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fTpcNch2"));
            AddHistogramToCanvas(oneDf, fTpcMultCanvas, cd);
	    cd++;
            oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fTpcNch3"));
            AddHistogramToCanvas(oneDf, fTpcMultCanvas, cd);

	    cd = 1;
            // Correlations
            if (!fCorrCanvas) 
	    {
	      fCorrCanvas = CreateCanvas("Correlations", "Multiplicity Correlations");
	      fCorrCanvas->Divide(2, 3);
	    }
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzdcNch"));
            AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzeroNch"));
            AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrSpdTpcNch"));
            AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzemNch"));
            AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzdcVzero"));
            AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzemVzero"));
            AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
	    
	    cd = 1;
            // ET Correlations
            if (!fEtCorrCanvas) 
	    {
	      fEtCorrCanvas = CreateCanvas("E_{T} corr", "E_{T} Correlations");
	      fEtCorrCanvas->Divide(2, 3);
	    }
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrZdcTotEvsPhosTotEt"));
            AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrZdcTotEvsEmcalTotEt"));
            AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrZdcTotEvsTotEt"));
            AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzerovsPhosTotEt"));
            AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzerovsEmcalTotEt"));
            AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzerovsTotEt"));
            AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);

	    cd = 1;
            // ZDC,V0 vs SPD
            if (!fZdcVzeroSpdCorrCanvas) 
	    {
	      fZdcVzeroSpdCorrCanvas = CreateCanvas("ZDC/V0 vs SPD", "ZDC/V0 vs. SPD");
	      fZdcVzeroSpdCorrCanvas->Divide(2, 3);
	      
	    }
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzeroSpd"));
            AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzeroASpd"));
            AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzeroCSpd"));
            AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzdcSpd"));
            AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzdcASpd"));
            AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
	    cd++;
            twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzdcCSpd"));
            AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
        }
        else
        {
            std::cout << "ERROR: This block does not contain what you think it contains!" << std::endl;
        }
    }
    else
    {
      std::cout << "ERROR: There is no TList object in the block" << std::endl;
    }


}

void AliHLTEveMultCorr::AddHistogramsToCanvas(AliHLTHOMERBlockDesc* /*block*/, TCanvas* /*canvas*/, Int_t& /*cdCount*/)
{
}

void AliHLTEveMultCorr::AddHistogramToCanvas(TH1* hist, TCanvas* canvas, Int_t& cdCount)
{
    if (hist)
    {
        if (!strcmp(hist->ClassName(), "TH1F"))
        {
	    canvas->cd(cdCount);
            dynamic_cast<TH1F*>(hist)->Draw();
        }
        else if (!strcmp(hist->ClassName(), "TH2F"))
        {
	    canvas->cd(cdCount);
            dynamic_cast<TH2F*>(hist)->Draw("COLZ");
        }
        else
        {
            std::cout << "I don't want histograms of type " << hist->ClassName() << std::endl;
        }
    }
    else
    {
      std::cout << "ERROR (AddHistogramsToCanvas): You gave me a null pointer" << std::endl;
    }
}
Commit	Line	Data
2127802b	1	#include "AliHLTEveMultCorr.h"
	2	#include "TH1F.h"
	3	#include "TH2F.h"
	4	#include <iostream>
	5	#include <TCanvas.h>
	6	#include "AliHLTHOMERBlockDesc.h"
	7	#include "TList.h"
	8
	9	AliHLTEveMultCorr::AliHLTEveMultCorr(const char* name): AliHLTEveBase(name)
	10	,fVzeroMultCanvas(0)
	11	,fZdcMultCanvas(0)
	12	,fTpcMultCanvas(0)
	13	,fCorrCanvas(0)
	14	,fEtCorrCanvas(0)
	15	,fZdcVzeroSpdCorrCanvas(0)
	16	{
	17
	18	}
	19
	20	AliHLTEveMultCorr::~AliHLTEveMultCorr()
	21	{
	22	}
	23
	24	void AliHLTEveMultCorr::ResetElements()
	25	{
	26
	27	}
	28
	29	void AliHLTEveMultCorr::UpdateElements()
	30	{
	31	fVzeroMultCanvas->Update();
	32	fZdcMultCanvas->Update();
	33	fTpcMultCanvas->Update();
	34	fCorrCanvas->Update();
	35	fEtCorrCanvas->Update();
	36	fZdcVzeroSpdCorrCanvas->Update();
	37
	38	}
	39
	40	void AliHLTEveMultCorr::ProcessBlock(AliHLTHOMERBlockDesc* block)
	41	{
	42	TList hlist = dynamic_cast<TList>(block->GetTObject());
	43	if (hlist)
	44	{
	45
	46	if (hlist->Contains("fVzeroMult")) // These are the correlation histograms
	47	{
	48	TH1F *oneDf = 0;
	49	TH2F *twoDf = 0;
	50	int cd = 1;
	51
	52	//VZERO multiplicity hists
	53	if (!fVzeroMultCanvas)
	54	{
	55	fVzeroMultCanvas = CreateCanvas("V0 Mult", "V0 Multiplicities");
	56	fVzeroMultCanvas->Divide(2, 2);
	57	}
	58	oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fVzeroMult"));
	59	AddHistogramToCanvas(oneDf, fVzeroMultCanvas, cd);
	60	cd++;
	61	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fVzeroMultAC"));
	62	AddHistogramToCanvas(twoDf, fVzeroMultCanvas, cd);
	63	cd++;
	64	oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fVzeroFlaggedMult"));
65	AddHistogramToCanvas(oneDf, fVzeroMultCanvas, cd);
66	cd++;
67	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fVzeroFlaggedMultAC"));
68	AddHistogramToCanvas(twoDf, fVzeroMultCanvas, cd);
69
70	cd = 1;
71	//ZDC multiplicity hists
72	if (!fZdcMultCanvas)
73	{
74	fZdcMultCanvas = CreateCanvas("ZDC Mult", "ZDC Multiplicities");
75	fZdcMultCanvas->Divide(2, 4);
76	}
77	oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcEzdc"));
78	AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
79	cd++;
80	oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcEzn"));
81	AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
82	cd++;
83	oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcEzp"));
84	AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
85	cd++;
86	oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcEzem"));
87	AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
88	cd++;
89	oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcNpart"));
90	AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
91	cd++;
92	oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fZdcB"));
93	AddHistogramToCanvas(oneDf, fZdcMultCanvas, cd);
94	cd++;
95	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fZdcEzemEzdc"));
96	AddHistogramToCanvas(twoDf, fZdcMultCanvas, cd);
97
98	cd = 1;
99	// TPC multiplicity
100	if (!fTpcMultCanvas)
101	{
102	fTpcMultCanvas = CreateCanvas("TPC Mult", "TPC Multiplicities");
103	fTpcMultCanvas->Divide(2);
104	}
105	oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fTpcNch2"));
106	AddHistogramToCanvas(oneDf, fTpcMultCanvas, cd);
107	cd++;
108	oneDf = dynamic_cast<TH1F*>(hlist->FindObject("fTpcNch3"));
109	AddHistogramToCanvas(oneDf, fTpcMultCanvas, cd);
110
111	cd = 1;
112	// Correlations
113	if (!fCorrCanvas)
114	{
115	fCorrCanvas = CreateCanvas("Correlations", "Multiplicity Correlations");
116	fCorrCanvas->Divide(2, 3);
117	}
118	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzdcNch"));
119	AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
120	cd++;
121	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzeroNch"));
122	AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
123	cd++;
124	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrSpdTpcNch"));
125	AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
126	cd++;
127	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzemNch"));
128	AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
129	cd++;
130	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzdcVzero"));
131	AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
132	cd++;
133	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzemVzero"));
134	AddHistogramToCanvas(twoDf, fCorrCanvas, cd);
135
136	cd = 1;
137	// ET Correlations
138	if (!fEtCorrCanvas)
139	{
140	fEtCorrCanvas = CreateCanvas("E_{T} corr", "E_{T} Correlations");
141	fEtCorrCanvas->Divide(2, 3);
142	}
143	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrZdcTotEvsPhosTotEt"));
144	AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);
145	cd++;
146	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrZdcTotEvsEmcalTotEt"));
147	AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);
148	cd++;
149	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrZdcTotEvsTotEt"));
150	AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);
151	cd++;
152	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzerovsPhosTotEt"));
153	AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);
154	cd++;
155	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzerovsEmcalTotEt"));
156	AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);
157	cd++;
158	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzerovsTotEt"));
159	AddHistogramToCanvas(twoDf, fEtCorrCanvas, cd);
160
161	cd = 1;
162	// ZDC,V0 vs SPD
163	if (!fZdcVzeroSpdCorrCanvas)
164	{
165	fZdcVzeroSpdCorrCanvas = CreateCanvas("ZDC/V0 vs SPD", "ZDC/V0 vs. SPD");
166	fZdcVzeroSpdCorrCanvas->Divide(2, 3);
167
168	}
169	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzeroSpd"));
170	AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
171	cd++;
172	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzeroASpd"));
173	AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
174	cd++;
175	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrVzeroCSpd"));
176	AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
177	cd++;
178	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzdcSpd"));
179	AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
180	cd++;
181	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzdcASpd"));
182	AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
183	cd++;
184	twoDf = dynamic_cast<TH2F*>(hlist->FindObject("fCorrEzdcCSpd"));
185	AddHistogramToCanvas(twoDf, fZdcVzeroSpdCorrCanvas, cd);
186	}
187	else
188	{
189	std::cout << "ERROR: This block does not contain what you think it contains!" << std::endl;
190	}
191	}
192	else
193	{
194	std::cout << "ERROR: There is no TList object in the block" << std::endl;
195	}
196
197
198	}
199
200	void AliHLTEveMultCorr::AddHistogramsToCanvas(AliHLTHOMERBlockDesc* /block/, TCanvas* /canvas/, Int_t& /cdCount/)
201	{
202	}
203
204	void AliHLTEveMultCorr::AddHistogramToCanvas(TH1* hist, TCanvas* canvas, Int_t& cdCount)
205	{
206	if (hist)
207	{
208	if (!strcmp(hist->ClassName(), "TH1F"))
209	{
210	canvas->cd(cdCount);
211	dynamic_cast<TH1F*>(hist)->Draw();
212	}
213	else if (!strcmp(hist->ClassName(), "TH2F"))
214	{
215	canvas->cd(cdCount);
216	dynamic_cast<TH2F*>(hist)->Draw("COLZ");
217	}
218	else
219	{
220	std::cout << "I don't want histograms of type " << hist->ClassName() << std::endl;
221	}
222	}
223	else
224	{
225	std::cout << "ERROR (AddHistogramsToCanvas): You gave me a null pointer" << std::endl;
226	}
227	}
228