[u/mrichter/AliRoot.git] / ITS / AliITSDigitizer.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/
 
/*
$Log$
Revision 1.6  2002/10/22 14:45:34  alibrary
Introducing Riostream.h

Revision 1.5  2002/10/14 14:57:00  hristov
Merging the VirtualMC branch to the main development branch (HEAD)

Revision 1.3.4.1  2002/06/10 17:51:14  hristov
Merged with v3-08-02

Revision 1.4  2002/04/24 22:08:12  nilsen
New ITS Digitizer/merger with two macros. One to make SDigits (old way) and
one to run the  merger (modified for Jiri).

Revision 1.3  2002/03/25 10:48:55  nilsen
New ITS SDigit merging with region of interest cut. Update for changes in
AliDigitizer. Additional optimization should be done.

Revision 1.2  2002/03/15 17:26:40  nilsen
New SDigit version of ITS Digitizer.

Revision 1.1  2001/11/27 16:27:28  nilsen
Adding AliITSDigitizer class to do merging and digitization . Based on the
TTask method. AliITSDigitizer class added to the Makefile and ITSLinkDef.h
file. The following files required minor changes. AliITS, added functions
SetHitsAddressBranch, MakeBranchInTreeD and modified MakeBranchD.
AliITSsimulationSDD.cxx needed a Tree independent way of returning back to
the original Root Directory in function Compress1D. Now it uses gDirectory.

Revision 1.2  2002/03/01  E. Lopez
Digitization changed to start from SDigits instead of Hits.
The SDigits are reading as TClonesArray of AliITSpListItem
*/

#include <stdlib.h>
#include <Riostream.h>
#include <TObjArray.h>
#include <TTree.h>
#include <TBranch.h>
#include <TFile.h>

#include <AliRun.h>
#include <AliRunDigitizer.h>

#include "AliITSDigitizer.h"
#include "AliITSpList.h"
#include "AliITSmodule.h"
#include "AliITSsimulation.h"
#include "AliITSDetType.h"
#include "AliITSgeom.h"

ClassImp(AliITSDigitizer)

//______________________________________________________________________
AliITSDigitizer::AliITSDigitizer() : AliDigitizer(){
    // Default constructor. Assign fITS since it is never written out from
    // here. 
    // Inputs:
    //      Option_t * opt   Not used
    // Outputs:
    //      none.
    // Return:
    //      A blank AliITSDigitizer class.

    fITS      = 0;
    fActive   = 0;
    fRoif     = -1;
    fRoiifile = 0;
    fInit     = kFALSE;
}
//______________________________________________________________________
AliITSDigitizer::AliITSDigitizer(AliRunDigitizer *mngr) : AliDigitizer(mngr){
    // Standard constructor. Assign fITS since it is never written out from
    // here. 
    // Inputs:
    //      Option_t * opt   Not used
    // Outputs:
    //      none.
    // Return:
    //      An AliItSDigitizer class.

    fITS      = 0;
    fActive   = 0;
    fRoif     = -1;
    fRoiifile = 0;
    fInit     = kFALSE;
}
//______________________________________________________________________
AliITSDigitizer::~AliITSDigitizer(){
    // Default destructor. 
    // Inputs:
    //      Option_t * opt   Not used
    // Outputs:
    //      none.
    // Return:
    //      none.

    fITS = 0; // don't delete fITS. Done else where.
    if(fActive) delete[] fActive;
}
//______________________________________________________________________
Bool_t AliITSDigitizer::Init(){
    // Initialization. Set up region of interest, if switched on, and
    // loads ITS and ITSgeom.
    // Inputs:
    //      none.
    // Outputs:
    //      none.
    // Return:
    //      none.

    fInit = kTRUE; // Assume for now init will work.
    if(!gAlice) {
	fITS      = 0;
	fRoiifile = 0;
	fInit     = kFALSE;
	Warning("Init","gAlice not found");
	return fInit;
    } // end if
    fITS = (AliITS *)(gAlice->GetDetector("ITS"));
    if(!fITS){
	fRoiifile = 0;
	fInit     = kFALSE;
	Warning("Init","ITS not found");
	return fInit;
    } else if(fITS->GetITSgeom()){
	//cout << "fRoif,fRoiifile="<<fRoif<<" "<<fRoiifile<<endl;
	fActive = new Bool_t[fITS->GetITSgeom()->GetIndexMax()];
    } else{
	fRoiifile = 0;
	fInit     = kFALSE;
	Warning("Init","ITS geometry not found");
	return fInit;
    } // end if
    // fActive needs to be set to a default all kTRUE value
    for(Int_t i=0;i<fITS->GetITSgeom()->GetIndexMax();i++) fActive[i] = kTRUE;
/*  This will not work from Init. ts is aways returned as zero.
    TTree *ts;
    if(fRoif>=0 && fRoiifile>=0 && fRoiifile<GetManager()->GetNinputs()){
	ts = GetManager()->GetInputTreeS(fRoiifile);
	if(!ts){
	    if(!gAlice) ts = gAlice->TreeS();
	    if(!ts){
		cout <<"The TTree TreeS needed to set by region not found."
		    " No region of interest cut will be applied."<< endl;
		return fInit;
	    } // end if
	} // end if
	cout << "calling SetByReionOfInterest ts="<< ts <<endl;
	SetByRegionOfInterest(ts);
    } // end if
*/
    return fInit;
}
//______________________________________________________________________
void AliITSDigitizer::Exec(Option_t* opt){
    // Main digitization function. 
    // Inputs:
    //      Option_t * opt   list of sub detector to digitize. =0 all.
    // Outputs:
    //      none.
    // Return:
    //      none.

    char name[20] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
    char *all;
    const char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),
                          strstr(opt,"SSD")};
    if( !det[0] && !det[1] && !det[2] ) all = "All";
    else all = 0;
    AliITSsimulation *sim      = 0;
    AliITSDetType    *iDetType = 0;
    static Bool_t    setDef    = kTRUE;

    if(!fInit){
	Error("Exec","Init not succesfull, aborting.");
	return;
    } // end if

    if( setDef ) fITS->SetDefaultSimulation();
    setDef = kFALSE;
    sprintf(name,"%s",fITS->GetName());

    Int_t nfiles = GetManager()->GetNinputs();
    Int_t event  = GetManager()->GetOutputEventNr();
    Int_t size   = fITS->GetITSgeom()->GetIndexMax();
    Int_t module,id,ifiles,mask;
    Bool_t lmod;
    Int_t *fl = new Int_t[nfiles];
    fl[0] = fRoiifile;
    mask = 1;
    for(id=0;id<nfiles;id++) if(id!=fRoiifile){
	// just in case fRoiifile!=0.
	fl[mask] = id;
	mask++;
    } // end for,if
    TClonesArray * sdig = new TClonesArray( "AliITSpListItem",1000 );
    
    // Digitize
    fITS->MakeBranchInTreeD(GetManager()->GetTreeD());
    if(fRoif!=0) Info("AliITSDigitizer","Region of Interest digitization selected");
    else Info("AliITSDigitizer","No Region of Interest selected. Digitizing everything");
    //cout <<"fActive="<<fActive<<" fRoif="<<fRoif;
    if(fActive==0) fRoif = 0; // fActive array must be define for RIO cuts.
    //cout <<" fRoif="<<fRoif<<endl;

    for(module=0; module<size; module++ ){
	if(fRoif!=0) if(!fActive[module]) continue;
        id = fITS->GetITSgeom()->GetModuleType(module);
        if(!all && !det[id]) continue;
        iDetType = fITS->DetType( id );
        sim      = (AliITSsimulation*)iDetType->GetSimulationModel();
        if(!sim) {
            Error( "Exec", "The simulation class was not instanciated!" );
            exit(1);
        } // end if !sim

        // Fill the module with the sum of SDigits
        sim->InitSimulationModule(module, event);
	//cout << "Module=" << module;
        for(ifiles=0; ifiles<nfiles; ifiles++ ){
	    if(fRoif!=0) if(!fActive[module]) continue;
	    //cout <<" fl[ifiles=" << ifiles << "]=" << fl[ifiles];
            TTree *treeS = GetManager()->GetInputTreeS(fl[ifiles]);
            if( !(treeS && fITS->GetSDigits()) ) continue;   
            TBranch *brchSDigits = treeS->GetBranch( name );
            if( brchSDigits ) {
                brchSDigits->SetAddress( &sdig ); 
            } else {
                Error( "Exec", "branch ITS not found in TreeS, input file %d ",
                       ifiles );
		return;
            } // end if brchSDigits
            sdig->Clear();
            mask = GetManager()->GetMask(ifiles);
            // add summable digits to module
            brchSDigits->GetEvent( module );
            lmod = sim->AddSDigitsToModule(sdig,mask);
	    if(ifiles==0){
		fActive[module] = lmod;
	    } // end if
	    //cout << " fActive["<<module<<"]=";
	    //if(fActive[module]) cout << "kTRUE";
	    //else cout << "kFALSE";
        } // end for ifiles
	//cout << " end ifiles loop" << endl;
        // Digitize current module sum(SDigits)->Digits
        sim->FinishSDigitiseModule();

        // fills all branches - wasted disk space
        GetManager()->GetTreeD()->Fill();
        fITS->ResetDigits();
    } // end for module
    //cout << "end modules loop"<<endl;

    GetManager()->GetTreeD()->AutoSave();

    delete[] fl;
    sdig->Clear();
    delete sdig;
    for(Int_t i=0;i<fITS->GetITSgeom()->GetIndexMax();i++) fActive[i] = kTRUE;
    return;
}
//______________________________________________________________________
void AliITSDigitizer::SetByRegionOfInterest(TTree *ts){
    // Scans through the ITS branch of the SDigits tree, ts, for modules
    // which have SDigits in them. For these modules, a flag is set to
    // digitize only these modules. The value of fRoif determines how many
    // neighboring modules will also be turned on. fRoif=0 will turn on only
    // those modules with SDigits in them. fRoif=1 will turn on, in addition,
    // those modules that are +-1 module from the one with the SDigits. And
    // So on. This last feature is not supported yet.
    // Inputs:
    //      TTree *ts  The tree in which the existing SDigits will define the
    //                 region of interest.
    // Outputs:
    //      none.
    // Return:
    //      none.
    Int_t m,nm,i;

    if(fRoif==0) return;
    if(ts==0) return;
    TBranch *brchSDigits = ts->GetBranch(fITS->GetName());
    TClonesArray * sdig = new TClonesArray( "AliITSpListItem",1000 );
    //cout << "Region of Interest ts="<<ts<<" brchSDigits="<<brchSDigits<<" sdig="<<sdig<<endl;

    if( brchSDigits ) {
	brchSDigits->SetAddress( &sdig );
    } else {
	Error( "SetByRegionOfInterest","branch ITS not found in TreeS");
	return;
    } // end if brchSDigits

    nm = fITS->GetITSgeom()->GetIndexMax();
    for(m=0;m<nm;m++){
	//cout << " fActive["<<m<<"]=";
	fActive[m] = kFALSE; // Not active by default
	sdig->Clear();
	brchSDigits->GetEvent(m);
	if(sdig->GetLast()>=0) for(i=0;i<sdig->GetLast();i++){
	    // activate the necessary modules
	    if(((AliITSpList*)sdig->At(m))->GetpListItem(i)->GetSignal()>0.0){ // Must have non zero signal.
		fActive[m] = kTRUE;
		break;
	    } // end if
	} // end if. end for i.
	//cout << fActive[m];
	//cout << endl;
    } // end for m
    Info("AliITSDigitizer","Digitization by Region of Interest selected");
    sdig->Clear();
    delete sdig;
    return;
}
Commit	Line	Data
9ad8b5dd	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15
	16	/*
	17	$Log$
f8d9a5b8	18	Revision 1.6 2002/10/22 14:45:34 alibrary
	19	Introducing Riostream.h
	20
4ae5bbc4	21	Revision 1.5 2002/10/14 14:57:00 hristov
	22	Merging the VirtualMC branch to the main development branch (HEAD)
	23
b9d0a01d	24	Revision 1.3.4.1 2002/06/10 17:51:14 hristov
	25	Merged with v3-08-02
	26
	27	Revision 1.4 2002/04/24 22:08:12 nilsen
	28	New ITS Digitizer/merger with two macros. One to make SDigits (old way) and
	29	one to run the merger (modified for Jiri).
	30
fd04285a	31	Revision 1.3 2002/03/25 10:48:55 nilsen
	32	New ITS SDigit merging with region of interest cut. Update for changes in
	33	AliDigitizer. Additional optimization should be done.
	34
fca85276	35	Revision 1.2 2002/03/15 17:26:40 nilsen
	36	New SDigit version of ITS Digitizer.
	37
f1888ca5	38	Revision 1.1 2001/11/27 16:27:28 nilsen
	39	Adding AliITSDigitizer class to do merging and digitization . Based on the
	40	TTask method. AliITSDigitizer class added to the Makefile and ITSLinkDef.h
	41	file. The following files required minor changes. AliITS, added functions
	42	SetHitsAddressBranch, MakeBranchInTreeD and modified MakeBranchD.
fca85276	43	AliITSsimulationSDD.cxx needed a Tree independent way of returning back to
f1888ca5	44	the original Root Directory in function Compress1D. Now it uses gDirectory.
	45
	46	Revision 1.2 2002/03/01 E. Lopez
fca85276	47	Digitization changed to start from SDigits instead of Hits.
f1888ca5	48	The SDigits are reading as TClonesArray of AliITSpListItem
9ad8b5dd	49	*/
	50
	51	#include <stdlib.h>
4ae5bbc4	52	#include <Riostream.h>
9ad8b5dd	53	#include <TObjArray.h>
	54	#include <TTree.h>
	55	#include <TBranch.h>
fca85276	56	#include <TFile.h>
9ad8b5dd	57
	58	#include <AliRun.h>
	59	#include <AliRunDigitizer.h>
	60
	61	#include "AliITSDigitizer.h"
fd04285a	62	#include "AliITSpList.h"
9ad8b5dd	63	#include "AliITSmodule.h"
	64	#include "AliITSsimulation.h"
	65	#include "AliITSDetType.h"
	66	#include "AliITSgeom.h"
	67
	68	ClassImp(AliITSDigitizer)
	69
	70	//______________________________________________________________________
	71	AliITSDigitizer::AliITSDigitizer() : AliDigitizer(){
	72	// Default constructor. Assign fITS since it is never written out from
	73	// here.
	74	// Inputs:
	75	// Option_t * opt Not used
	76	// Outputs:
	77	// none.
	78	// Return:
	79	// A blank AliITSDigitizer class.
	80
fd04285a	81	fITS = 0;
	82	fActive = 0;
	83	fRoif = -1;
fca85276	84	fRoiifile = 0;
fd04285a	85	fInit = kFALSE;
9ad8b5dd	86	}
	87	//______________________________________________________________________
	88	AliITSDigitizer::AliITSDigitizer(AliRunDigitizer *mngr) : AliDigitizer(mngr){
	89	// Standard constructor. Assign fITS since it is never written out from
	90	// here.
	91	// Inputs:
	92	// Option_t * opt Not used
	93	// Outputs:
	94	// none.
	95	// Return:
	96	// An AliItSDigitizer class.
	97
fd04285a	98	fITS = 0;
	99	fActive = 0;
	100	fRoif = -1;
	101	fRoiifile = 0;
	102	fInit = kFALSE;
9ad8b5dd	103	}
	104	//______________________________________________________________________
	105	AliITSDigitizer::~AliITSDigitizer(){
	106	// Default destructor.
	107	// Inputs:
	108	// Option_t * opt Not used
	109	// Outputs:
	110	// none.
	111	// Return:
	112	// none.
	113
	114	fITS = 0; // don't delete fITS. Done else where.
f1888ca5	115	if(fActive) delete[] fActive;
9ad8b5dd	116	}
9ad8b5dd	117	//______________________________________________________________________
9ad8b5dd	118	Bool_t AliITSDigitizer::Init(){
fd04285a	119	// Initialization. Set up region of interest, if switched on, and
fd04285a	120	// loads ITS and ITSgeom.
9ad8b5dd	121	// Inputs:
	122	// none.
	123	// Outputs:
	124	// none.
	125	// Return:
	126	// none.
fca85276	127
fd04285a	128	fInit = kTRUE; // Assume for now init will work.
fca85276	129	if(!gAlice) {
fca85276	130	fITS = 0;
fca85276	131	fRoiifile = 0;
fd04285a	132	fInit = kFALSE;
	133	Warning("Init","gAlice not found");
	134	return fInit;
fca85276	135	} // end if
	136	fITS = (AliITS *)(gAlice->GetDetector("ITS"));
	137	if(!fITS){
fca85276	138	fRoiifile = 0;
fd04285a	139	fInit = kFALSE;
	140	Warning("Init","ITS not found");
	141	return fInit;
fca85276	142	} else if(fITS->GetITSgeom()){
fd04285a	143	//cout << "fRoif,fRoiifile="<<fRoif<<" "<<fRoiifile<<endl;
fca85276	144	fActive = new Bool_t[fITS->GetITSgeom()->GetIndexMax()];
fca85276	145	} else{
fca85276	146	fRoiifile = 0;
fd04285a	147	fInit = kFALSE;
	148	Warning("Init","ITS geometry not found");
	149	return fInit;
fca85276	150	} // end if
	151	// fActive needs to be set to a default all kTRUE value
	152	for(Int_t i=0;i<fITS->GetITSgeom()->GetIndexMax();i++) fActive[i] = kTRUE;
fd04285a	153	/* This will not work from Init. ts is aways returned as zero.
fd04285a	154	TTree *ts;
fca85276	155	if(fRoif>=0 && fRoiifile>=0 && fRoiifile<GetManager()->GetNinputs()){
fca85276	156	ts = GetManager()->GetInputTreeS(fRoiifile);
fd04285a	157	if(!ts){
	158	if(!gAlice) ts = gAlice->TreeS();
	159	if(!ts){
	160	cout <<"The TTree TreeS needed to set by region not found."
	161	" No region of interest cut will be applied."<< endl;
	162	return fInit;
	163	} // end if
	164	} // end if
	165	cout << "calling SetByReionOfInterest ts="<< ts <<endl;
fca85276	166	SetByRegionOfInterest(ts);
	167	} // end if
	168	*/
fd04285a	169	return fInit;
9ad8b5dd	170	}
	171	//______________________________________________________________________
	172	void AliITSDigitizer::Exec(Option_t* opt){
fca85276	173	// Main digitization function.
9ad8b5dd	174	// Inputs:
fca85276	175	// Option_t * opt list of sub detector to digitize. =0 all.
9ad8b5dd	176	// Outputs:
	177	// none.
	178	// Return:
	179	// none.
f1888ca5	180
9ad8b5dd	181	char name[20] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
	182	char *all;
	183	const char *det[3] = {strstr(opt,"SPD"),strstr(opt,"SDD"),
	184	strstr(opt,"SSD")};
f1888ca5	185	if( !det[0] && !det[1] && !det[2] ) all = "All";
9ad8b5dd	186	else all = 0;
	187	AliITSsimulation *sim = 0;
	188	AliITSDetType *iDetType = 0;
f1888ca5	189	static Bool_t setDef = kTRUE;
9ad8b5dd	190
fd04285a	191	if(!fInit){
fd04285a	192	Error("Exec","Init not succesfull, aborting.");
f1888ca5	193	return;
9ad8b5dd	194	} // end if
9ad8b5dd	195
f1888ca5	196	if( setDef ) fITS->SetDefaultSimulation();
f1888ca5	197	setDef = kFALSE;
fd04285a	198	sprintf(name,"%s",fITS->GetName());
9ad8b5dd	199
f1888ca5	200	Int_t nfiles = GetManager()->GetNinputs();
	201	Int_t event = GetManager()->GetOutputEventNr();
	202	Int_t size = fITS->GetITSgeom()->GetIndexMax();
fd04285a	203	Int_t module,id,ifiles,mask;
	204	Bool_t lmod;
	205	Int_t *fl = new Int_t[nfiles];
	206	fl[0] = fRoiifile;
	207	mask = 1;
	208	for(id=0;id<nfiles;id++) if(id!=fRoiifile){
	209	// just in case fRoiifile!=0.
	210	fl[mask] = id;
	211	mask++;
	212	} // end for,if
f1888ca5	213	TClonesArray * sdig = new TClonesArray( "AliITSpListItem",1000 );
f1888ca5	214
fd04285a	215	// Digitize
fd04285a	216	fITS->MakeBranchInTreeD(GetManager()->GetTreeD());
f8d9a5b8	217	if(fRoif!=0) Info("AliITSDigitizer","Region of Interest digitization selected");
	218	else Info("AliITSDigitizer","No Region of Interest selected. Digitizing everything");
	219	//cout <<"fActive="<<fActive<<" fRoif="<<fRoif;
	220	if(fActive==0) fRoif = 0; // fActive array must be define for RIO cuts.
	221	//cout <<" fRoif="<<fRoif<<endl;
f1888ca5	222
fd04285a	223	for(module=0; module<size; module++ ){
f8d9a5b8	224	if(fRoif!=0) if(!fActive[module]) continue;
fd04285a	225	id = fITS->GetITSgeom()->GetModuleType(module);
fd04285a	226	if(!all && !det[id]) continue;
f1888ca5	227	iDetType = fITS->DetType( id );
9ad8b5dd	228	sim = (AliITSsimulation*)iDetType->GetSimulationModel();
fca85276	229	if(!sim) {
f1888ca5	230	Error( "Exec", "The simulation class was not instanciated!" );
9ad8b5dd	231	exit(1);
9ad8b5dd	232	} // end if !sim
fca85276	233
f1888ca5	234	// Fill the module with the sum of SDigits
fd04285a	235	sim->InitSimulationModule(module, event);
	236	//cout << "Module=" << module;
	237	for(ifiles=0; ifiles<nfiles; ifiles++ ){
f8d9a5b8	238	if(fRoif!=0) if(!fActive[module]) continue;
fd04285a	239	//cout <<" fl[ifiles=" << ifiles << "]=" << fl[ifiles];
fd04285a	240	TTree *treeS = GetManager()->GetInputTreeS(fl[ifiles]);
f1888ca5	241	if( !(treeS && fITS->GetSDigits()) ) continue;
	242	TBranch *brchSDigits = treeS->GetBranch( name );
	243	if( brchSDigits ) {
	244	brchSDigits->SetAddress( &sdig );
	245	} else {
	246	Error( "Exec", "branch ITS not found in TreeS, input file %d ",
	247	ifiles );
fd04285a	248	return;
f1888ca5	249	} // end if brchSDigits
f1888ca5	250	sdig->Clear();
fd04285a	251	mask = GetManager()->GetMask(ifiles);
f1888ca5	252	// add summable digits to module
f1888ca5	253	brchSDigits->GetEvent( module );
fd04285a	254	lmod = sim->AddSDigitsToModule(sdig,mask);
	255	if(ifiles==0){
	256	fActive[module] = lmod;
	257	} // end if
	258	//cout << " fActive["<<module<<"]=";
	259	//if(fActive[module]) cout << "kTRUE";
	260	//else cout << "kFALSE";
f1888ca5	261	} // end for ifiles
fd04285a	262	//cout << " end ifiles loop" << endl;
fca85276	263	// Digitize current module sum(SDigits)->Digits
f1888ca5	264	sim->FinishSDigitiseModule();
f1888ca5	265
9ad8b5dd	266	// fills all branches - wasted disk space
	267	GetManager()->GetTreeD()->Fill();
	268	fITS->ResetDigits();
	269	} // end for module
fd04285a	270	//cout << "end modules loop"<<endl;
	271
	272	GetManager()->GetTreeD()->AutoSave();
	273
	274	delete[] fl;
f1888ca5	275	sdig->Clear();
f1888ca5	276	delete sdig;
fd04285a	277	for(Int_t i=0;i<fITS->GetITSgeom()->GetIndexMax();i++) fActive[i] = kTRUE;
fd04285a	278	return;
9ad8b5dd	279	}
fca85276	280	//______________________________________________________________________
	281	void AliITSDigitizer::SetByRegionOfInterest(TTree *ts){
	282	// Scans through the ITS branch of the SDigits tree, ts, for modules
	283	// which have SDigits in them. For these modules, a flag is set to
	284	// digitize only these modules. The value of fRoif determines how many
	285	// neighboring modules will also be turned on. fRoif=0 will turn on only
	286	// those modules with SDigits in them. fRoif=1 will turn on, in addition,
	287	// those modules that are +-1 module from the one with the SDigits. And
	288	// So on. This last feature is not supported yet.
	289	// Inputs:
	290	// TTree *ts The tree in which the existing SDigits will define the
	291	// region of interest.
	292	// Outputs:
	293	// none.
	294	// Return:
	295	// none.
fd04285a	296	Int_t m,nm,i;
	297
	298	if(fRoif==0) return;
	299	if(ts==0) return;
fca85276	300	TBranch *brchSDigits = ts->GetBranch(fITS->GetName());
fca85276	301	TClonesArray * sdig = new TClonesArray( "AliITSpListItem",1000 );
fd04285a	302	//cout << "Region of Interest ts="<<ts<<" brchSDigits="<<brchSDigits<<" sdig="<<sdig<<endl;
fca85276	303
fca85276	304	if( brchSDigits ) {
fd04285a	305	brchSDigits->SetAddress( &sdig );
fca85276	306	} else {
	307	Error( "SetByRegionOfInterest","branch ITS not found in TreeS");
	308	return;
	309	} // end if brchSDigits
	310
	311	nm = fITS->GetITSgeom()->GetIndexMax();
	312	for(m=0;m<nm;m++){
fd04285a	313	//cout << " fActive["<<m<<"]=";
fca85276	314	fActive[m] = kFALSE; // Not active by default
	315	sdig->Clear();
	316	brchSDigits->GetEvent(m);
fd04285a	317	if(sdig->GetLast()>=0) for(i=0;i<sdig->GetLast();i++){
	318	// activate the necessary modules
	319	if(((AliITSpList*)sdig->At(m))->GetpListItem(i)->GetSignal()>0.0){ // Must have non zero signal.
fca85276	320	fActive[m] = kTRUE;
fd04285a	321	break;
fca85276	322	} // end if
fd04285a	323	} // end if. end for i.
	324	//cout << fActive[m];
	325	//cout << endl;
fca85276	326	} // end for m
f8d9a5b8	327	Info("AliITSDigitizer","Digitization by Region of Interest selected");
fca85276	328	sdig->Clear();
fca85276	329	delete sdig;
fd04285a	330	return;
fca85276	331	}