[u/mrichter/AliRoot.git] / EVE / EveDet / AliEveITSScaledModule.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.                                                 *
 **************************************************************************/

#include "AliEveITSScaledModule.h"

#include <AliITSsegmentationSPD.h>
#include <AliITSsegmentationSDD.h>
#include <AliITSsegmentationSSD.h>

#include <AliITSdigitSPD.h>
#include <AliITSdigitSDD.h>
#include <AliITSdigitSSD.h>

#include <TMath.h>
#include <TClonesArray.h>

//==============================================================================
//==============================================================================
// AliEveDigitScaleInfo
//==============================================================================

//______________________________________________________________________________
//
// Encapsulates current state of scaling and agglomeration.

ClassImp(AliEveDigitScaleInfo)

AliEveDigitScaleInfo::AliEveDigitScaleInfo():
  fScale(1),
  fStatType (kSTAverage),
  fSyncPalette(kFALSE)
{
}

void AliEveDigitScaleInfo::ScaleChanged(Int_t s)
{
  fScale = s;

  AliEveITSScaledModule* sm;
  std::list<TEveElement*>::iterator i = fBackRefs.begin();
  while (i != fBackRefs.end())
  {
    sm = dynamic_cast<AliEveITSScaledModule*>(*i);
    if(sm) sm->LoadQuads();
    ++i;
  }
}

void AliEveDigitScaleInfo::StatTypeChanged(Int_t t)
{
  fStatType = t;
  fSyncPalette = kTRUE;

  AliEveITSScaledModule* sm;
  std::list<TEveElement*>::iterator i = fBackRefs.begin();
  while (i != fBackRefs.end())
  {
    sm = dynamic_cast<AliEveITSScaledModule*>(*i);
    if (sm) sm->SetQuadValues();
    ++i;
  }
}

//______________________________________________________________________________
// ScaledDigit_t
//

AliEveITSScaledModule::ScaledDigit_t::ScaledDigit_t() :
  TObject(),
  fN(0),
  fSum(0), fSqrSum(0),
  fMinI(-1), fMinJ(-1), fMaxI(-1), fMaxJ(-1)
{
}

AliEveITSScaledModule::ScaledDigit_t::ScaledDigit_t(Int_t di, Int_t dj) :
  TObject(),
  fN(0),
  fSum(0), fSqrSum(0),
  fMinI(di), fMinJ(dj), fMaxI(di), fMaxJ(dj)
{
}

void AliEveITSScaledModule::ScaledDigit_t::Dump() const
{
  printf("N %d, sum %f, sqr_sum %f", fN, fSum, fSqrSum);
}


//==============================================================================
//==============================================================================
// AliEveITSScaledModule
//==============================================================================

//______________________________________________________________________________
//
// Visualization of an ITS module with digits aggregated
// on a grid of pre-defined size.

ClassImp(AliEveITSScaledModule)

AliEveITSScaledModule::AliEveITSScaledModule(Int_t gid, AliEveITSDigitsInfo* info, AliEveDigitScaleInfo* si):
  AliEveITSModule("AliEveITSScaledModule", "AliEveITSScaledModule"),
  fDigitsMap(),
  fNx(-1),
  fNz(-1),
  fNCx(-1),
  fNCz(-1),
  fScaleInfo(si)
{
  SetOwnIds(kTRUE);

  SetDigitsInfo(info);
  SetID(gid);
  fScaleInfo->IncRefCount(this);
}

AliEveITSScaledModule::~AliEveITSScaledModule()
{
  fScaleInfo->DecRefCount(this);
}

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

void AliEveITSScaledModule::LoadQuads()
{
  // Here we still use 'z' for the name of axial coordinates.
  // The transforamtion matrix aplied rotates y -> z.
  // We need this as TEveQuadSet offers optimized treatment for
  // quads in the x-y plane.

  TClonesArray *digits;
  Float_t       x, z, zo, dpx, dpz; // orig cells size, pos
  Int_t         i, j, ndigits;   // orig cells idx
  Int_t         c1, c2;          // original coordinates

  Int_t id;
  std::map<Int_t, Int_t> dmap;
  std::map<Int_t, Int_t>::iterator miter;
  digits  = fInfo->GetDigits(fID, fDetID);
  ndigits = digits->GetEntriesFast();

  ScaledDigit_t* sd;
  Int_t scale = fScaleInfo->GetScale() -1;
  switch(fDetID)
  {
    case 0:
    {
      // SPD
      Reset(kQT_RectangleXZFixedY, kFALSE, 32);

      fNCz = fInfo->fSPDScaleZ[scale];
      fNCx = fInfo->fSPDScaleX[scale];
      fNz  = Int_t(fInfo->fSegSPD->Npz()/fNCz);
      fNx  = Int_t(fInfo->fSegSPD->Npx()/fNCx);
      dpz = 2*fDz/fNz;
      dpx = 2*fDx/fNx;
      //printf("SPD orig cells (%d, %d) (%d, %d)\n", fInfo->fSegSPD->Npx(), fInfo->fSegSPD->Npz(), Nx, Nz);

      AliITSdigitSPD *od ;
      for (Int_t k=0; k<ndigits; ++k)
      {
	od = (AliITSdigitSPD*) digits->UncheckedAt(k);

	fInfo->GetSPDLocalZ(od->GetCoord1(),zo);
        c1 = od->GetCoord1(); c2 = od->GetCoord2();
	i = Int_t((zo+fDz)/dpz);
	j = Int_t((od->GetCoord2()*fNx)/fInfo->fSegSPD->Npx());
	id = j*fNx + i;

        miter = dmap.find(id);
	if(miter == dmap.end())
	{
          dmap[id] = fPlex.Size();
          z = dpz*(i) - fDz;
          x = dpx*(j) - fDx;
          AddQuad(x, z, dpx, dpz);
          sd = new ScaledDigit_t(c1, c2);
	  QuadId(sd);
	}
        else
	{
	  sd = dynamic_cast<ScaledDigit_t*>(GetId(miter->second));
          if(c1 < sd->fMinI)
	    sd->fMinI = c1;
	  else if( c1 > sd->fMaxI)
            sd->fMaxI = c1;

          if(c2 < sd->fMinJ)
	    sd->fMinJ = c2;
	  else if( c2 > sd->fMaxJ)
	    sd->fMaxJ = c2;
	}

	sd->fN++;
	sd->fSum  += od->GetSignal();
	sd->fSqrSum += od->GetSignal()*od->GetSignal();
      }
      break;
    }
    case 1:
    {
      // SDD
      Reset(kQT_RectangleXZFixedY, kFALSE, 32);

      fNCz = fInfo->fSDDScaleZ[scale];
      fNCx = fInfo->fSDDScaleX[scale];
      fNz  = Int_t(fInfo->fSegSDD->Npz()/fNCz);
      fNx  = Int_t(fInfo->fSegSDD->Npx()/fNCx);
      dpz = 2*fDz/fNz;
      dpx = 2*fDx/fNx;

      AliITSdigitSDD *od=0;
      for (Int_t k=0; k<ndigits; k++) {
	od=(AliITSdigitSDD*)digits->UncheckedAt(k);
	fInfo->fSegSDD->DetToLocal(od->GetCoord2(), od->GetCoord1(),x,z);
	z+= fDz;
	x+= fDx;
	i = Int_t(z/dpz);
	j = Int_t(x/dpx);
	//printf("Mod %d coord %d,%d out of %d,%d :: ORIG coord %d,%d out of %d,%d \n",fID,
	//       i,j,Nz,Nx,od->GetCoord1(),od->GetCoord2(),fInfo->fSegSDD->Npz(),fInfo->fSegSDD->Npx());

	id = j*fNx + i;
	c1 = od->GetCoord1(); c2 = od->GetCoord2();

	miter = dmap.find(id);
	if(miter == dmap.end())
	{
	  dmap[id] = fPlex.Size();
	  z = dpz*(i) - fDz;
	  x = dpx*(j) - fDx;
	  AddQuad(x, z, dpx, dpz);
	  sd = new ScaledDigit_t(od->GetCoord1(),od->GetCoord2());
	  QuadId(sd);
	}
	else
	{
	  sd = dynamic_cast<ScaledDigit_t*>(GetId(miter->second));
	  if(c1 < sd->fMinI)
	    sd->fMinI = c1;
	  else if( c1 > sd->fMaxI)
	    sd->fMaxI = c1;

	  if(c2 < sd->fMinJ)
	    sd->fMinJ = c2;
	  else if( c2 > sd->fMaxJ)
	    sd->fMaxJ = c2;
	}
	sd->fN++;
	sd->fSum  += od->GetSignal();
	sd->fSqrSum += od->GetSignal()*od->GetSignal();
      }
      break;
    }
    case 2:
    {
      // SSD
      Reset(kQT_LineXZFixedY, kFALSE, 32);

      AliITSsegmentationSSD* seg = fInfo->fSegSSD;
      Float_t ap, an; // positive/negative angles -> offsets
      seg->Angles(ap, an);
      ap =   TMath::Tan(ap) * fDz;
      an = - TMath::Tan(an) * fDz;

      fNCx  = fInfo->fSSDScale[scale];
      fNz  = 1;
      fNx  = Int_t(fInfo->fSegSSD->Npx()/fNCx);
      dpz = 2*fDz/fNz;
      dpx = 2*fDx/fNx;

      AliITSdigitSSD *od=0;
      for (Int_t k=0; k<ndigits; k++) {
	od=(AliITSdigitSSD*)digits->UncheckedAt(k);
	if(od->GetCoord1() == 1)
	  i = 1; // p side
	else
	  i= -1; // n side
	j = Int_t(od->GetCoord2()/fNCx);
	c1 = od->GetCoord1(); c2 = od->GetCoord2();
	id = j*i;

	miter = dmap.find(id);
	ScaledDigit_t* sd;
	if(miter == dmap.end())
	{
	  // printf("orig digit %d,%d scaled %d,%d \n",od->GetCoord1(),od->GetCoord2(),i,j);
	  dmap[id] = fPlex.Size();
	  z = dpz*(i) - fDz;
	  x = dpx*(j) - fDx;
	  Float_t a = ( od->GetCoord1() == 1) ? ap : an;
	  AddLine(x-a, -fDz, 2*a, 2*fDz);

	  sd = new ScaledDigit_t(c1, c2);
	  QuadId(sd);
	}
	else
	{
	  sd = dynamic_cast<ScaledDigit_t*>(GetId(miter->second));
	  if(c1 < sd->fMinI)
	    sd->fMinI = c1;
	  else if( c1 > sd->fMaxI)
	    sd->fMaxI = c1;

	  if(c2 < sd->fMinJ)
	    sd->fMinJ = c2;
	  else if( c2 > sd->fMaxJ)
	    sd->fMaxJ = c2;
	}
	sd->fN++;
	sd->fSum  += od->GetSignal();
	sd->fSqrSum += od->GetSignal()*od->GetSignal();
      } // for digits
      break;
    } // end case 2
  } // end switch

  SetQuadValues();
  RefitPlex();
}

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

void AliEveITSScaledModule::SetQuadValues()
{
  if(fScaleInfo->GetSyncPalette()) SyncPalette();

  Int_t num = fPlex.Size();
  for (Int_t i = 0 ; i < num; i++)
  {
    ScaledDigit_t* sd = dynamic_cast<ScaledDigit_t*>(GetId(i));
    Int_t v = 0;
    switch(fScaleInfo->GetStatType())
    {
      using namespace TMath;

      case AliEveDigitScaleInfo::kSTOccup:
	v = Nint((100.0*sd->fN) / (fNCx*fNCz));
	break;
      case AliEveDigitScaleInfo::kSTAverage:
	v = Nint((Double_t) sd->fSum / sd->fN);
	break;
      case AliEveDigitScaleInfo::kSTRms:
	v = Nint(Sqrt(sd->fSqrSum) / sd->fN);
	break;
    }
    DigitBase_t* qb = GetDigit(i);
    qb->fValue = v;
  }
}

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

void AliEveITSScaledModule::SyncPalette()
{
  // printf("AliEveITSScaledModule::SyncPalette()\n");
  if(fScaleInfo->GetStatType() == AliEveDigitScaleInfo::kSTOccup)
  {
    // SPD
    AliEveITSModule::fgSPDPalette->SetLimits(0, 100);
    AliEveITSModule::fgSPDPalette->SetMinMax(0, 100);

    // SDD
    AliEveITSModule::fgSDDPalette->SetLimits(0, 100);
    AliEveITSModule::fgSDDPalette->SetMinMax(0, 100);

    // SSD
    AliEveITSModule::fgSSDPalette->SetLimits(0, 100);
    AliEveITSModule::fgSDDPalette->SetMinMax(0, 100);
  }
  else
  {
    AliEveITSDigitsInfo& di = *fInfo;
    // SPD
    AliEveITSModule::fgSPDPalette->SetLimits(0, di.fSPDHighLim);
    AliEveITSModule::fgSPDPalette->SetMinMax(di.fSPDMinVal, di.fSPDMaxVal);

    // SDD
    AliEveITSModule::fgSDDPalette->SetLimits(0, di.fSDDHighLim);
    AliEveITSModule::fgSDDPalette->SetMinMax(di.fSDDMinVal, di.fSDDMaxVal);

    // SSD
    AliEveITSModule::fgSSDPalette->SetLimits(0, di.fSSDHighLim);
    AliEveITSModule::fgSSDPalette->SetMinMax(di.fSSDMinVal, di.fSSDMaxVal);
  }

  fScaleInfo->SetSyncPalette(kFALSE);
}

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

void AliEveITSScaledModule::GetScaleData(Int_t& cnx, Int_t& cnz, Int_t& total) const
{
  cnx   = fNx;
  cnz   = fNz;
  total = cnx*cnz;
}

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

void  AliEveITSScaledModule::DigitSelected(Int_t idx)
{
  // Override control-click from TEveQuadSet
  printf("AliEveITSScaledModule::DigitSelected "); Print();

  DigitBase_t* qb  = GetDigit(idx);
  TObject* obj  = qb->fId.GetObject();
  ScaledDigit_t* sd = dynamic_cast<ScaledDigit_t*>(obj);
  TClonesArray *digits = fInfo->GetDigits(fID, fDetID);
  Int_t ndigits = digits->GetEntriesFast();

  printf("%d digits in cell scaleX = %d,  scaleZ = %d \n", sd->fN, fNCx, fNCz);

  Int_t il = 0;
  for(Int_t k=0; k<ndigits; k++)
  {
    AliITSdigit *d = (AliITSdigit*) digits->UncheckedAt(k);

    if(d->GetCoord1()>=sd->fMinI && d->GetCoord1()<=sd->fMaxI &&
       d->GetCoord2()>=sd->fMinJ && d->GetCoord2()<=sd->fMaxJ)
    {
      printf("%3d, %3d: %3d", d->GetCoord1(), d->GetCoord2(), d->GetSignal());
      printf(" | ");
      il++;
      if(il>5) {
	printf("\n");
	il = 0;
      }
    }
  }
  if(il) printf("\n");
}
Commit	Line	Data
d810d0de	1	// $Id$
d810d0de	2	// Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
3b868b75	3
d810d0de	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	**************************************************************************/
	9
	10	#include "AliEveITSScaledModule.h"
3b868b75	11
0879c50b	12	#include <AliITSsegmentationSPD.h>
	13	#include <AliITSsegmentationSDD.h>
	14	#include <AliITSsegmentationSSD.h>
	15
3b868b75	16	#include <AliITSdigitSPD.h>
	17	#include <AliITSdigitSDD.h>
	18	#include <AliITSdigitSSD.h>
d810d0de	19
0879c50b	20	#include <TMath.h>
a15e6d7d	21	#include <TClonesArray.h>
3b868b75	22
a15e6d7d	23	//==============================================================================
a15e6d7d	24	//==============================================================================
d810d0de	25	// AliEveDigitScaleInfo
a15e6d7d	26	//==============================================================================
	27
	28	//______________________________________________________________________________
3b868b75	29	//
a15e6d7d	30	// Encapsulates current state of scaling and agglomeration.
698e2c9b	31
d810d0de	32	ClassImp(AliEveDigitScaleInfo)
3b868b75	33
d810d0de	34	AliEveDigitScaleInfo::AliEveDigitScaleInfo():
3b868b75	35	fScale(1),
a15e6d7d	36	fStatType (kSTAverage),
32e219c2	37	fSyncPalette(kFALSE)
3b868b75	38	{
	39	}
	40
d810d0de	41	void AliEveDigitScaleInfo::ScaleChanged(Int_t s)
3b868b75	42	{
32e219c2	43	fScale = s;
51346b82	44
d810d0de	45	AliEveITSScaledModule* sm;
84aff7a4	46	std::list<TEveElement*>::iterator i = fBackRefs.begin();
32e219c2	47	while (i != fBackRefs.end())
32e219c2	48	{
d810d0de	49	sm = dynamic_cast<AliEveITSScaledModule>(i);
32e219c2	50	if(sm) sm->LoadQuads();
	51	++i;
	52	}
3b868b75	53	}
3b868b75	54
d810d0de	55	void AliEveDigitScaleInfo::StatTypeChanged(Int_t t)
3b868b75	56	{
32e219c2	57	fStatType = t;
32e219c2	58	fSyncPalette = kTRUE;
51346b82	59
d810d0de	60	AliEveITSScaledModule* sm;
84aff7a4	61	std::list<TEveElement*>::iterator i = fBackRefs.begin();
32e219c2	62	while (i != fBackRefs.end())
32e219c2	63	{
d810d0de	64	sm = dynamic_cast<AliEveITSScaledModule>(i);
a15e6d7d	65	if (sm) sm->SetQuadValues();
32e219c2	66	++i;
32e219c2	67	}
3b868b75	68	}
3b868b75	69
57ffa5fb	70	//______________________________________________________________________________
a15e6d7d	71	// ScaledDigit_t
3b868b75	72	//
c76ea574	73
a15e6d7d	74	AliEveITSScaledModule::ScaledDigit_t::ScaledDigit_t() :
3b868b75	75	TObject(),
a15e6d7d	76	fN(0),
	77	fSum(0), fSqrSum(0),
	78	fMinI(-1), fMinJ(-1), fMaxI(-1), fMaxJ(-1)
3b868b75	79	{
	80	}
	81
a15e6d7d	82	AliEveITSScaledModule::ScaledDigit_t::ScaledDigit_t(Int_t di, Int_t dj) :
3b868b75	83	TObject(),
a15e6d7d	84	fN(0),
	85	fSum(0), fSqrSum(0),
	86	fMinI(di), fMinJ(dj), fMaxI(di), fMaxJ(dj)
3b868b75	87	{
	88	}
	89
a15e6d7d	90	void AliEveITSScaledModule::ScaledDigit_t::Dump() const
3b868b75	91	{
a15e6d7d	92	printf("N %d, sum %f, sqr_sum %f", fN, fSum, fSqrSum);
3b868b75	93	}
3b868b75	94
698e2c9b	95
a15e6d7d	96	//==============================================================================
a15e6d7d	97	//==============================================================================
d810d0de	98	// AliEveITSScaledModule
a15e6d7d	99	//==============================================================================
698e2c9b	100
698e2c9b	101	//______________________________________________________________________________
3b868b75	102	//
698e2c9b	103	// Visualization of an ITS module with digits aggregated
698e2c9b	104	// on a grid of pre-defined size.
3b868b75	105
d810d0de	106	ClassImp(AliEveITSScaledModule)
3b868b75	107
d810d0de	108	AliEveITSScaledModule::AliEveITSScaledModule(Int_t gid, AliEveITSDigitsInfo* info, AliEveDigitScaleInfo* si):
d810d0de	109	AliEveITSModule("AliEveITSScaledModule", "AliEveITSScaledModule"),
c76ea574	110	fDigitsMap(),
3b868b75	111	fNx(-1),
	112	fNz(-1),
	113	fNCx(-1),
03dfc4a7	114	fNCz(-1),
03dfc4a7	115	fScaleInfo(si)
3b868b75	116	{
3b868b75	117	SetOwnIds(kTRUE);
	118
	119	SetDigitsInfo(info);
51346b82	120	SetID(gid);
32e219c2	121	fScaleInfo->IncRefCount(this);
3b868b75	122	}
3b868b75	123
d810d0de	124	AliEveITSScaledModule::~AliEveITSScaledModule()
3b868b75	125	{
32e219c2	126	fScaleInfo->DecRefCount(this);
3b868b75	127	}
3b868b75	128
57ffa5fb	129	/******************************************************************************/
3b868b75	130
d810d0de	131	void AliEveITSScaledModule::LoadQuads()
3b868b75	132	{
	133	// Here we still use 'z' for the name of axial coordinates.
	134	// The transforamtion matrix aplied rotates y -> z.
84aff7a4	135	// We need this as TEveQuadSet offers optimized treatment for
3b868b75	136	// quads in the x-y plane.
	137
	138	TClonesArray *digits;
	139	Float_t x, z, zo, dpx, dpz; // orig cells size, pos
	140	Int_t i, j, ndigits; // orig cells idx
	141	Int_t c1, c2; // original coordinates
	142
	143	Int_t id;
fd31e9de	144	std::map<Int_t, Int_t> dmap;
fd31e9de	145	std::map<Int_t, Int_t>::iterator miter;
3b868b75	146	digits = fInfo->GetDigits(fID, fDetID);
	147	ndigits = digits->GetEntriesFast();
	148
a15e6d7d	149	ScaledDigit_t* sd;
03dfc4a7	150	Int_t scale = fScaleInfo->GetScale() -1;
3b868b75	151	switch(fDetID)
3b868b75	152	{
51346b82	153	case 0:
51346b82	154	{
3b868b75	155	// SPD
84aff7a4	156	Reset(kQT_RectangleXZFixedY, kFALSE, 32);
3b868b75	157
	158	fNCz = fInfo->fSPDScaleZ[scale];
	159	fNCx = fInfo->fSPDScaleX[scale];
	160	fNz = Int_t(fInfo->fSegSPD->Npz()/fNCz);
	161	fNx = Int_t(fInfo->fSegSPD->Npx()/fNCx);
51346b82	162	dpz = 2*fDz/fNz;
3b868b75	163	dpx = 2*fDx/fNx;
	164	//printf("SPD orig cells (%d, %d) (%d, %d)\n", fInfo->fSegSPD->Npx(), fInfo->fSegSPD->Npz(), Nx, Nz);
	165
	166	AliITSdigitSPD *od ;
	167	for (Int_t k=0; k<ndigits; ++k)
	168	{
	169	od = (AliITSdigitSPD*) digits->UncheckedAt(k);
	170
	171	fInfo->GetSPDLocalZ(od->GetCoord1(),zo);
	172	c1 = od->GetCoord1(); c2 = od->GetCoord2();
	173	i = Int_t((zo+fDz)/dpz);
	174	j = Int_t((od->GetCoord2()*fNx)/fInfo->fSegSPD->Npx());
	175	id = j*fNx + i;
51346b82	176
3b868b75	177	miter = dmap.find(id);
3b868b75	178	if(miter == dmap.end())
51346b82	179	{
3b868b75	180	dmap[id] = fPlex.Size();
	181	z = dpz*(i) - fDz;
	182	x = dpx*(j) - fDx;
	183	AddQuad(x, z, dpx, dpz);
a15e6d7d	184	sd = new ScaledDigit_t(c1, c2);
3b868b75	185	QuadId(sd);
3b868b75	186	}
51346b82	187	else
3b868b75	188	{
a15e6d7d	189	sd = dynamic_cast<ScaledDigit_t*>(GetId(miter->second));
	190	if(c1 < sd->fMinI)
	191	sd->fMinI = c1;
	192	else if( c1 > sd->fMaxI)
	193	sd->fMaxI = c1;
	194
	195	if(c2 < sd->fMinJ)
	196	sd->fMinJ = c2;
	197	else if( c2 > sd->fMaxJ)
	198	sd->fMaxJ = c2;
3b868b75	199	}
51346b82	200
a15e6d7d	201	sd->fN++;
	202	sd->fSum += od->GetSignal();
	203	sd->fSqrSum += od->GetSignal()*od->GetSignal();
3b868b75	204	}
	205	break;
	206	}
51346b82	207	case 1:
51346b82	208	{
3b868b75	209	// SDD
84aff7a4	210	Reset(kQT_RectangleXZFixedY, kFALSE, 32);
3b868b75	211
3b868b75	212	fNCz = fInfo->fSDDScaleZ[scale];
51346b82	213	fNCx = fInfo->fSDDScaleX[scale];
3b868b75	214	fNz = Int_t(fInfo->fSegSDD->Npz()/fNCz);
	215	fNx = Int_t(fInfo->fSegSDD->Npx()/fNCx);
	216	dpz = 2*fDz/fNz;
	217	dpx = 2*fDx/fNx;
	218
	219	AliITSdigitSDD *od=0;
	220	for (Int_t k=0; k<ndigits; k++) {
	221	od=(AliITSdigitSDD*)digits->UncheckedAt(k);
	222	fInfo->fSegSDD->DetToLocal(od->GetCoord2(), od->GetCoord1(),x,z);
	223	z+= fDz;
	224	x+= fDx;
	225	i = Int_t(z/dpz);
	226	j = Int_t(x/dpx);
	227	//printf("Mod %d coord %d,%d out of %d,%d :: ORIG coord %d,%d out of %d,%d \n",fID,
	228	// i,j,Nz,Nx,od->GetCoord1(),od->GetCoord2(),fInfo->fSegSDD->Npz(),fInfo->fSegSDD->Npx());
	229
	230	id = j*fNx + i;
	231	c1 = od->GetCoord1(); c2 = od->GetCoord2();
51346b82	232
3b868b75	233	miter = dmap.find(id);
3b868b75	234	if(miter == dmap.end())
51346b82	235	{
3b868b75	236	dmap[id] = fPlex.Size();
	237	z = dpz*(i) - fDz;
	238	x = dpx*(j) - fDx;
	239	AddQuad(x, z, dpx, dpz);
a15e6d7d	240	sd = new ScaledDigit_t(od->GetCoord1(),od->GetCoord2());
3b868b75	241	QuadId(sd);
3b868b75	242	}
51346b82	243	else
3b868b75	244	{
a15e6d7d	245	sd = dynamic_cast<ScaledDigit_t*>(GetId(miter->second));
	246	if(c1 < sd->fMinI)
	247	sd->fMinI = c1;
	248	else if( c1 > sd->fMaxI)
	249	sd->fMaxI = c1;
	250
	251	if(c2 < sd->fMinJ)
	252	sd->fMinJ = c2;
	253	else if( c2 > sd->fMaxJ)
	254	sd->fMaxJ = c2;
3b868b75	255	}
a15e6d7d	256	sd->fN++;
	257	sd->fSum += od->GetSignal();
	258	sd->fSqrSum += od->GetSignal()*od->GetSignal();
3b868b75	259	}
3b868b75	260	break;
51346b82	261	}
	262	case 2:
	263	{
	264	// SSD
84aff7a4	265	Reset(kQT_LineXZFixedY, kFALSE, 32);
3b868b75	266
51346b82	267	AliITSsegmentationSSD* seg = fInfo->fSegSSD;
3b868b75	268	Float_t ap, an; // positive/negative angles -> offsets
	269	seg->Angles(ap, an);
	270	ap = TMath::Tan(ap) * fDz;
	271	an = - TMath::Tan(an) * fDz;
	272
51346b82	273	fNCx = fInfo->fSSDScale[scale];
3b868b75	274	fNz = 1;
	275	fNx = Int_t(fInfo->fSegSSD->Npx()/fNCx);
	276	dpz = 2*fDz/fNz;
	277	dpx = 2*fDx/fNx;
	278
	279	AliITSdigitSSD *od=0;
	280	for (Int_t k=0; k<ndigits; k++) {
	281	od=(AliITSdigitSSD*)digits->UncheckedAt(k);
51346b82	282	if(od->GetCoord1() == 1)
3b868b75	283	i = 1; // p side
51346b82	284	else
3b868b75	285	i= -1; // n side
	286	j = Int_t(od->GetCoord2()/fNCx);
	287	c1 = od->GetCoord1(); c2 = od->GetCoord2();
51346b82	288	id = j*i;
3b868b75	289
3b868b75	290	miter = dmap.find(id);
a15e6d7d	291	ScaledDigit_t* sd;
3b868b75	292	if(miter == dmap.end())
51346b82	293	{
3b868b75	294	// printf("orig digit %d,%d scaled %d,%d \n",od->GetCoord1(),od->GetCoord2(),i,j);
	295	dmap[id] = fPlex.Size();
	296	z = dpz*(i) - fDz;
	297	x = dpx*(j) - fDx;
	298	Float_t a = ( od->GetCoord1() == 1) ? ap : an;
	299	AddLine(x-a, -fDz, 2a, 2fDz);
	300
a15e6d7d	301	sd = new ScaledDigit_t(c1, c2);
3b868b75	302	QuadId(sd);
3b868b75	303	}
51346b82	304	else
3b868b75	305	{
a15e6d7d	306	sd = dynamic_cast<ScaledDigit_t*>(GetId(miter->second));
	307	if(c1 < sd->fMinI)
	308	sd->fMinI = c1;
	309	else if( c1 > sd->fMaxI)
	310	sd->fMaxI = c1;
	311
	312	if(c2 < sd->fMinJ)
	313	sd->fMinJ = c2;
	314	else if( c2 > sd->fMaxJ)
	315	sd->fMaxJ = c2;
3b868b75	316	}
a15e6d7d	317	sd->fN++;
	318	sd->fSum += od->GetSignal();
	319	sd->fSqrSum += od->GetSignal()*od->GetSignal();
3b868b75	320	} // for digits
	321	break;
	322	} // end case 2
	323	} // end switch
	324
	325	SetQuadValues();
	326	RefitPlex();
	327	}
	328
57ffa5fb	329	/******************************************************************************/
3b868b75	330
d810d0de	331	void AliEveITSScaledModule::SetQuadValues()
3b868b75	332	{
32e219c2	333	if(fScaleInfo->GetSyncPalette()) SyncPalette();
32e219c2	334
a15e6d7d	335	Int_t num = fPlex.Size();
a15e6d7d	336	for (Int_t i = 0 ; i < num; i++)
3b868b75	337	{
a15e6d7d	338	ScaledDigit_t* sd = dynamic_cast<ScaledDigit_t*>(GetId(i));
3b868b75	339	Int_t v = 0;
32e219c2	340	switch(fScaleInfo->GetStatType())
	341	{
	342	using namespace TMath;
	343
a15e6d7d	344	case AliEveDigitScaleInfo::kSTOccup:
a15e6d7d	345	v = Nint((100.0sd->fN) / (fNCxfNCz));
32e219c2	346	break;
a15e6d7d	347	case AliEveDigitScaleInfo::kSTAverage:
a15e6d7d	348	v = Nint((Double_t) sd->fSum / sd->fN);
32e219c2	349	break;
a15e6d7d	350	case AliEveDigitScaleInfo::kSTRms:
a15e6d7d	351	v = Nint(Sqrt(sd->fSqrSum) / sd->fN);
32e219c2	352	break;
3b868b75	353	}
84aff7a4	354	DigitBase_t* qb = GetDigit(i);
3b868b75	355	qb->fValue = v;
	356	}
	357	}
	358
57ffa5fb	359	/******************************************************************************/
3b868b75	360
d810d0de	361	void AliEveITSScaledModule::SyncPalette()
51346b82	362	{
d810d0de	363	// printf("AliEveITSScaledModule::SyncPalette()\n");
a15e6d7d	364	if(fScaleInfo->GetStatType() == AliEveDigitScaleInfo::kSTOccup)
32e219c2	365	{
32e219c2	366	// SPD
d810d0de	367	AliEveITSModule::fgSPDPalette->SetLimits(0, 100);
d810d0de	368	AliEveITSModule::fgSPDPalette->SetMinMax(0, 100);
51346b82	369
32e219c2	370	// SDD
d810d0de	371	AliEveITSModule::fgSDDPalette->SetLimits(0, 100);
d810d0de	372	AliEveITSModule::fgSDDPalette->SetMinMax(0, 100);
32e219c2	373
32e219c2	374	// SSD
d810d0de	375	AliEveITSModule::fgSSDPalette->SetLimits(0, 100);
d810d0de	376	AliEveITSModule::fgSDDPalette->SetMinMax(0, 100);
32e219c2	377	}
	378	else
	379	{
a15e6d7d	380	AliEveITSDigitsInfo& di = *fInfo;
32e219c2	381	// SPD
a15e6d7d	382	AliEveITSModule::fgSPDPalette->SetLimits(0, di.fSPDHighLim);
a15e6d7d	383	AliEveITSModule::fgSPDPalette->SetMinMax(di.fSPDMinVal, di.fSPDMaxVal);
51346b82	384
32e219c2	385	// SDD
a15e6d7d	386	AliEveITSModule::fgSDDPalette->SetLimits(0, di.fSDDHighLim);
a15e6d7d	387	AliEveITSModule::fgSDDPalette->SetMinMax(di.fSDDMinVal, di.fSDDMaxVal);
32e219c2	388
32e219c2	389	// SSD
a15e6d7d	390	AliEveITSModule::fgSSDPalette->SetLimits(0, di.fSSDHighLim);
a15e6d7d	391	AliEveITSModule::fgSSDPalette->SetMinMax(di.fSSDMinVal, di.fSSDMaxVal);
32e219c2	392	}
	393
	394	fScaleInfo->SetSyncPalette(kFALSE);
	395	}
	396
57ffa5fb	397	/******************************************************************************/
32e219c2	398
a15e6d7d	399	void AliEveITSScaledModule::GetScaleData(Int_t& cnx, Int_t& cnz, Int_t& total) const
32e219c2	400	{
a15e6d7d	401	cnx = fNx;
a15e6d7d	402	cnz = fNz;
32e219c2	403	total = cnx*cnz;
	404	}
	405
57ffa5fb	406	/******************************************************************************/
32e219c2	407
d810d0de	408	void AliEveITSScaledModule::DigitSelected(Int_t idx)
3b868b75	409	{
84aff7a4	410	// Override control-click from TEveQuadSet
d810d0de	411	printf("AliEveITSScaledModule::DigitSelected "); Print();
3b868b75	412
84aff7a4	413	DigitBase_t* qb = GetDigit(idx);
3b868b75	414	TObject* obj = qb->fId.GetObject();
a15e6d7d	415	ScaledDigit_t* sd = dynamic_cast<ScaledDigit_t*>(obj);
3b868b75	416	TClonesArray *digits = fInfo->GetDigits(fID, fDetID);
	417	Int_t ndigits = digits->GetEntriesFast();
	418
a15e6d7d	419	printf("%d digits in cell scaleX = %d, scaleZ = %d \n", sd->fN, fNCx, fNCz);
3b868b75	420
3b868b75	421	Int_t il = 0;
51346b82	422	for(Int_t k=0; k<ndigits; k++)
3b868b75	423	{
	424	AliITSdigit d = (AliITSdigit) digits->UncheckedAt(k);
	425
a15e6d7d	426	if(d->GetCoord1()>=sd->fMinI && d->GetCoord1()<=sd->fMaxI &&
a15e6d7d	427	d->GetCoord2()>=sd->fMinJ && d->GetCoord2()<=sd->fMaxJ)
3b868b75	428	{
	429	printf("%3d, %3d: %3d", d->GetCoord1(), d->GetCoord2(), d->GetSignal());
	430	printf(" \| ");
	431	il++;
	432	if(il>5) {
	433	printf("\n");
	434	il = 0;
	435	}
	436	}
	437	}
	438	if(il) printf("\n");
	439	}