[u/mrichter/AliRoot.git] / EVE / EveDet / AliEveTOFDigitsInfo.cxx

// $Id$
// Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007

/**************************************************************************
 * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
 * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
 * full copyright notice.                                                 *
 **************************************************************************/
//
// AliEveTOFDigitsInfo
//

#include <TEveTreeTools.h>

#include "AliEveTOFDigitsInfo.h"
#include <AliTOFdigit.h>
#include <AliTOFGeometry.h>
//#include <AliTOFDigitMap.h>

using namespace std;

//_________________________________________________________

ClassImp(AliEveTOFDigitsInfo)

  AliEveTOFDigitsInfo::AliEveTOFDigitsInfo():
    TObject(),
    TEveRefCnt(),
    fTree (0),
    fNewTree (0),
    fGeom (0),
    fTOFdigitMap(new AliTOFDigitMap())
{}
/* ******************************************************* */

AliEveTOFDigitsInfo:: ~AliEveTOFDigitsInfo()
{

  delete fGeom;
  delete fTree;
  delete fNewTree;
  delete fTOFdigitMap;

}
/* ******************************************************* */

void AliEveTOFDigitsInfo::SetTree(TTree* tree)
{
  static const TEveException eH("AliEveTOFDigitsInfo::SetTree ");

  if(fGeom == 0) {
    fGeom = new AliTOFGeometry();
  }

  fTree = tree;
  /*
  DecRefCount();
  IncRefCount();
  */
}
/* ******************************************************* */

void AliEveTOFDigitsInfo::LoadDigits()
{

  TClonesArray *digitsTOF = 0x0;
  AliTOFdigit *digs;

  fTree->SetBranchAddress("TOF",&digitsTOF);
  fTree->GetEntry(0);

  Int_t vol[5] = {-1,-1,-1,-1,-1};

  for (Int_t digitNumber=0; digitNumber<digitsTOF->GetEntries(); digitNumber++) {

    //if (digitNumber==digitsTOF->GetEntries()-1) printf(" Hello  4 -> %3i digit of %i \n", digitNumber+1, digitsTOF->GetEntries());

    digs = (AliTOFdigit*)digitsTOF->UncheckedAt(digitNumber);

    vol[0] = digs->GetSector(); // Sector Number (0-17)
    vol[1] = digs->GetPlate();  // Plate Number (0-4)
    vol[2] = digs->GetStrip();  // Strip Number (0-14/18)
    vol[3] = digs->GetPadx();   // TEvePad Number in x direction (0-47)
    vol[4] = digs->GetPadz();   // TEvePad Number in z direction (0-1)

    fTOFdigitMap->AddDigit(vol, digitNumber);
    //if (digitNumber==digitsTOF->GetEntries()-1) printf(" I am inside LoadDigits %3i \n", digitNumber);

  }

}

/* ******************************************************* */

void AliEveTOFDigitsInfo::GetDigits(Int_t nSector, Int_t nPlate,
			      Int_t nStrip, Int_t nPadZ, Int_t nPadX,
			      Int_t indexDigit[3])
{

  Int_t vol[5] = {nSector,nPlate,nStrip,nPadX,nPadZ};

  fTOFdigitMap->GetDigitIndex(vol, indexDigit);

}
/* ******************************************************* */

TClonesArray* AliEveTOFDigitsInfo::GetDigits(Int_t nSector, Int_t nPlate,
				       Int_t nStrip)
{

  Int_t newCounter = 0;
  Int_t nDigitsInVolume[3] = {-1, -1, -1};
  Int_t dummy[3] = {-1, -1, -1};
  Int_t informations[4] = {-1, -1, -1, -1};

  TClonesArray* digitsTOFnew = new TClonesArray("AliTOFdigit",  300);
  TClonesArray &ldigits = *digitsTOFnew;

  AliTOFdigit *digs;

  TClonesArray *digitsTOF = 0x0;
  fTree->SetBranchAddress("TOF",&digitsTOF);
  fTree->GetEntry(0);


  Int_t vol[5] = {nSector,nPlate,nStrip,-1,-1};

  for(Int_t iPadZ=0; iPadZ<fGeom->NpadZ(); iPadZ++){
    vol[4] = iPadZ;
    for(Int_t iPadX=0; iPadX<fGeom->NpadX(); iPadX++) {
      vol[3] = iPadX;

      //GetDigits(vol[0], vol[1], vol[2], vol[3], vol[4], nDigitsInVolume)

      fTOFdigitMap->GetDigitIndex(vol, nDigitsInVolume);

      for (Int_t ii=0; ii<3; ii++) {

	if (nDigitsInVolume[ii]>=0 ) {
	  //printf("  nDigitsInVolume[%2i]  = %3i\n ", ii, nDigitsInVolume[ii]);
	  digs = (AliTOFdigit*)digitsTOF->UncheckedAt(nDigitsInVolume[ii]);
	  informations[0] = digs->GetTdc();
	  informations[1] = digs->GetAdc();
	  informations[2] = digs->GetToT();
	  informations[3] = digs->GetTdcND();
	  new (ldigits[newCounter++]) AliTOFdigit(dummy, vol, informations);
	}

      }

      for (Int_t ii=0; ii<4; ii++) informations[ii]=-1;
      for (Int_t ii=0; ii<3; ii++) dummy[ii]=-1;
      for (Int_t ii=0; ii<3; ii++) nDigitsInVolume[ii]=-1;

    }
  }

  /*
  if (digitsTOFnew)
    printf("Sector %2i   Plate %1i  Strip %2i  -> number of digits %3i \n",
	   nSector, nPlate, nStrip, digitsTOFnew->GetEntries());
  */
  return digitsTOFnew;

}
/* ******************************************************* */

TClonesArray* AliEveTOFDigitsInfo::GetDigits(Int_t nSector)
{

  Int_t newCounter = 0;
  Int_t nDigitsInVolume[3] = {-1, -1, -1};
  Int_t dummy[3] = {-1, -1, -1};
  Int_t informations[4] = {-1, -1, -1, -1};

  Int_t nStrips=19;

  TClonesArray* digitsTOFnew = new TClonesArray("AliTOFdigit",  300);
  TClonesArray &ldigits = *digitsTOFnew;

  AliTOFdigit *digs;

  TClonesArray *digitsTOF = 0x0;
  fTree->SetBranchAddress("TOF",&digitsTOF);
  fTree->GetEntry(0);

  //Int_t nSector = 1;
  Int_t vol[5] = {nSector,-1,-1,-1,-1};

  for(Int_t iPlate=0; iPlate<fGeom->NPlates(); iPlate++){
    vol[1] = iPlate;
    if(iPlate==2) nStrips=15;
    else nStrips=19;

    for(Int_t iStrip=0; iStrip<nStrips; iStrip++){
      vol[2] = iStrip;

      for(Int_t iPadZ=0; iPadZ<fGeom->NpadZ(); iPadZ++){
	vol[4] = iPadZ;

	for(Int_t iPadX=0; iPadX<fGeom->NpadX(); iPadX++) {
	  vol[3] = iPadX;

	  //GetDigits(vol[0], vol[1], vol[2], vol[3], vol[4], nDigitsInVolume)

	  fTOFdigitMap->GetDigitIndex(vol, nDigitsInVolume);

	  for (Int_t ii=0; ii<3; ii++) {

	    if (nDigitsInVolume[ii]>=0 ) {
	      //printf("  nDigitsInVolume[%2i]  = %3i\n ", ii, nDigitsInVolume[ii]);
	      digs = (AliTOFdigit*)digitsTOF->UncheckedAt(nDigitsInVolume[ii]);
	      informations[0] = digs->GetTdc();
	      informations[1] = digs->GetAdc();
	      informations[2] = digs->GetToT();
	      informations[3] = digs->GetTdcND();
	      new (ldigits[newCounter++]) AliTOFdigit(dummy, vol, informations);
	    }

	  }

	  for (Int_t ii=0; ii<4; ii++) informations[ii]=-1;
	  for (Int_t ii=0; ii<3; ii++) dummy[ii]=-1;
	  for (Int_t ii=0; ii<3; ii++) nDigitsInVolume[ii]=-1;

	}
      }
    }
  }

  /*
  if (digitsTOFnew)
    printf("Sector %2i   Plate %1i  Strip %2i  -> number of digits %3i \n",
	   nSector, nPlate, nStrip, digitsTOFnew->GetEntries());
  */
  return digitsTOFnew;

}
/* ******************************************************* */

void AliEveTOFDigitsInfo::GetDigits()
{

  for (Int_t iSector=0; iSector<fGeom->NSectors(); iSector++) {

    fNewTree = new TTree();


  }

}
Commit	Line	Data
d810d0de	1	// $Id$
	2	// Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
	3
	4	/**************************************************************************
	5	* Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
	6	* See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for *
51346b82	7	* full copyright notice. *
d810d0de	8	**************************************************************************/
8616d353	9	//
d810d0de	10	// AliEveTOFDigitsInfo
8616d353	11	//
8616d353	12
84aff7a4	13	#include <TEveTreeTools.h>
8616d353	14
d810d0de	15	#include "AliEveTOFDigitsInfo.h"
8616d353	16	#include <AliTOFdigit.h>
28a21457	17	#include <AliTOFGeometry.h>
8616d353	18	//#include <AliTOFDigitMap.h>
d810d0de	19
8616d353	20	using namespace std;
	21
	22	//_________________________________________________________
	23
d810d0de	24	ClassImp(AliEveTOFDigitsInfo)
8616d353	25
51346b82	26	AliEveTOFDigitsInfo::AliEveTOFDigitsInfo():
8616d353	27	TObject(),
84aff7a4	28	TEveRefCnt(),
8616d353	29	fTree (0),
	30	fNewTree (0),
	31	fGeom (0),
	32	fTOFdigitMap(new AliTOFDigitMap())
	33	{}
	34	/* ******************************************************* */
	35
51346b82	36	AliEveTOFDigitsInfo:: ~AliEveTOFDigitsInfo()
8616d353	37	{
	38
	39	delete fGeom;
	40	delete fTree;
	41	delete fNewTree;
	42	delete fTOFdigitMap;
	43
	44	}
	45	/* ******************************************************* */
	46
d810d0de	47	void AliEveTOFDigitsInfo::SetTree(TTree* tree)
8616d353	48	{
d810d0de	49	static const TEveException eH("AliEveTOFDigitsInfo::SetTree ");
51346b82	50
8616d353	51	if(fGeom == 0) {
28a21457	52	fGeom = new AliTOFGeometry();
8616d353	53	}
51346b82	54
8616d353	55	fTree = tree;
	56	/*
	57	DecRefCount();
	58	IncRefCount();
	59	*/
	60	}
	61	/* ******************************************************* */
	62
d810d0de	63	void AliEveTOFDigitsInfo::LoadDigits()
8616d353	64	{
	65
	66	TClonesArray *digitsTOF = 0x0;
	67	AliTOFdigit *digs;
	68
	69	fTree->SetBranchAddress("TOF",&digitsTOF);
	70	fTree->GetEntry(0);
	71
	72	Int_t vol[5] = {-1,-1,-1,-1,-1};
	73
	74	for (Int_t digitNumber=0; digitNumber<digitsTOF->GetEntries(); digitNumber++) {
	75
	76	//if (digitNumber==digitsTOF->GetEntries()-1) printf(" Hello 4 -> %3i digit of %i \n", digitNumber+1, digitsTOF->GetEntries());
51346b82	77
8616d353	78	digs = (AliTOFdigit*)digitsTOF->UncheckedAt(digitNumber);
	79
	80	vol[0] = digs->GetSector(); // Sector Number (0-17)
	81	vol[1] = digs->GetPlate(); // Plate Number (0-4)
	82	vol[2] = digs->GetStrip(); // Strip Number (0-14/18)
84aff7a4	83	vol[3] = digs->GetPadx(); // TEvePad Number in x direction (0-47)
84aff7a4	84	vol[4] = digs->GetPadz(); // TEvePad Number in z direction (0-1)
8616d353	85
	86	fTOFdigitMap->AddDigit(vol, digitNumber);
	87	//if (digitNumber==digitsTOF->GetEntries()-1) printf(" I am inside LoadDigits %3i \n", digitNumber);
	88
	89	}
	90
	91	}
	92
	93	/* ******************************************************* */
	94
d810d0de	95	void AliEveTOFDigitsInfo::GetDigits(Int_t nSector, Int_t nPlate,
8616d353	96	Int_t nStrip, Int_t nPadZ, Int_t nPadX,
	97	Int_t indexDigit[3])
	98	{
	99
	100	Int_t vol[5] = {nSector,nPlate,nStrip,nPadX,nPadZ};
	101
	102	fTOFdigitMap->GetDigitIndex(vol, indexDigit);
	103
	104	}
	105	/* ******************************************************* */
	106
d810d0de	107	TClonesArray* AliEveTOFDigitsInfo::GetDigits(Int_t nSector, Int_t nPlate,
8616d353	108	Int_t nStrip)
	109	{
	110
	111	Int_t newCounter = 0;
	112	Int_t nDigitsInVolume[3] = {-1, -1, -1};
	113	Int_t dummy[3] = {-1, -1, -1};
	114	Int_t informations[4] = {-1, -1, -1, -1};
	115
	116	TClonesArray* digitsTOFnew = new TClonesArray("AliTOFdigit", 300);
	117	TClonesArray &ldigits = *digitsTOFnew;
	118
	119	AliTOFdigit *digs;
	120
	121	TClonesArray *digitsTOF = 0x0;
	122	fTree->SetBranchAddress("TOF",&digitsTOF);
	123	fTree->GetEntry(0);
	124
	125
	126	Int_t vol[5] = {nSector,nPlate,nStrip,-1,-1};
	127
	128	for(Int_t iPadZ=0; iPadZ<fGeom->NpadZ(); iPadZ++){
	129	vol[4] = iPadZ;
	130	for(Int_t iPadX=0; iPadX<fGeom->NpadX(); iPadX++) {
	131	vol[3] = iPadX;
	132
	133	//GetDigits(vol[0], vol[1], vol[2], vol[3], vol[4], nDigitsInVolume)
	134
	135	fTOFdigitMap->GetDigitIndex(vol, nDigitsInVolume);
	136
	137	for (Int_t ii=0; ii<3; ii++) {
	138
	139	if (nDigitsInVolume[ii]>=0 ) {
	140	//printf(" nDigitsInVolume[%2i] = %3i\n ", ii, nDigitsInVolume[ii]);
	141	digs = (AliTOFdigit*)digitsTOF->UncheckedAt(nDigitsInVolume[ii]);
	142	informations[0] = digs->GetTdc();
	143	informations[1] = digs->GetAdc();
	144	informations[2] = digs->GetToT();
	145	informations[3] = digs->GetTdcND();
	146	new (ldigits[newCounter++]) AliTOFdigit(dummy, vol, informations);
	147	}
	148
	149	}
	150
	151	for (Int_t ii=0; ii<4; ii++) informations[ii]=-1;
	152	for (Int_t ii=0; ii<3; ii++) dummy[ii]=-1;
	153	for (Int_t ii=0; ii<3; ii++) nDigitsInVolume[ii]=-1;
	154
	155	}
	156	}
	157
	158	/*
	159	if (digitsTOFnew)
	160	printf("Sector %2i Plate %1i Strip %2i -> number of digits %3i \n",
	161	nSector, nPlate, nStrip, digitsTOFnew->GetEntries());
	162	*/
	163	return digitsTOFnew;
	164
	165	}
	166	/* ******************************************************* */
	167
d810d0de	168	TClonesArray* AliEveTOFDigitsInfo::GetDigits(Int_t nSector)
8616d353	169	{
	170
	171	Int_t newCounter = 0;
	172	Int_t nDigitsInVolume[3] = {-1, -1, -1};
	173	Int_t dummy[3] = {-1, -1, -1};
	174	Int_t informations[4] = {-1, -1, -1, -1};
	175
	176	Int_t nStrips=19;
	177
	178	TClonesArray* digitsTOFnew = new TClonesArray("AliTOFdigit", 300);
	179	TClonesArray &ldigits = *digitsTOFnew;
	180
	181	AliTOFdigit *digs;
	182
	183	TClonesArray *digitsTOF = 0x0;
	184	fTree->SetBranchAddress("TOF",&digitsTOF);
	185	fTree->GetEntry(0);
	186
	187	//Int_t nSector = 1;
	188	Int_t vol[5] = {nSector,-1,-1,-1,-1};
51346b82	189
8616d353	190	for(Int_t iPlate=0; iPlate<fGeom->NPlates(); iPlate++){
	191	vol[1] = iPlate;
	192	if(iPlate==2) nStrips=15;
	193	else nStrips=19;
51346b82	194
8616d353	195	for(Int_t iStrip=0; iStrip<nStrips; iStrip++){
8616d353	196	vol[2] = iStrip;
51346b82	197
8616d353	198	for(Int_t iPadZ=0; iPadZ<fGeom->NpadZ(); iPadZ++){
	199	vol[4] = iPadZ;
	200
	201	for(Int_t iPadX=0; iPadX<fGeom->NpadX(); iPadX++) {
	202	vol[3] = iPadX;
	203
	204	//GetDigits(vol[0], vol[1], vol[2], vol[3], vol[4], nDigitsInVolume)
	205
	206	fTOFdigitMap->GetDigitIndex(vol, nDigitsInVolume);
	207
	208	for (Int_t ii=0; ii<3; ii++) {
	209
	210	if (nDigitsInVolume[ii]>=0 ) {
	211	//printf(" nDigitsInVolume[%2i] = %3i\n ", ii, nDigitsInVolume[ii]);
	212	digs = (AliTOFdigit*)digitsTOF->UncheckedAt(nDigitsInVolume[ii]);
	213	informations[0] = digs->GetTdc();
	214	informations[1] = digs->GetAdc();
	215	informations[2] = digs->GetToT();
	216	informations[3] = digs->GetTdcND();
	217	new (ldigits[newCounter++]) AliTOFdigit(dummy, vol, informations);
	218	}
	219
	220	}
	221
	222	for (Int_t ii=0; ii<4; ii++) informations[ii]=-1;
	223	for (Int_t ii=0; ii<3; ii++) dummy[ii]=-1;
	224	for (Int_t ii=0; ii<3; ii++) nDigitsInVolume[ii]=-1;
51346b82	225
8616d353	226	}
	227	}
	228	}
	229	}
	230
	231	/*
	232	if (digitsTOFnew)
	233	printf("Sector %2i Plate %1i Strip %2i -> number of digits %3i \n",
	234	nSector, nPlate, nStrip, digitsTOFnew->GetEntries());
	235	*/
	236	return digitsTOFnew;
	237
	238	}
	239	/* ******************************************************* */
	240
d810d0de	241	void AliEveTOFDigitsInfo::GetDigits()
8616d353	242	{
	243
	244	for (Int_t iSector=0; iSector<fGeom->NSectors(); iSector++) {
	245
	246	fNewTree = new TTree();
	247
	248
	249
	250
	251	}
	252
	253	}