tested with trace fitting with multi-thread
This commit is contained in:
parent
76976482fe
commit
83177e57c7
|
@ -20,13 +20,16 @@ Bool_t GeneralSort::Process(Long64_t entry){
|
||||||
|
|
||||||
if( entry < 1 ) printf("============================== start processing data\n");
|
if( entry < 1 ) printf("============================== start processing data\n");
|
||||||
|
|
||||||
|
if( entry > 5000) return true;
|
||||||
|
|
||||||
///initialization
|
///initialization
|
||||||
for( int i = 0; i < mapping::nDetType; i++){
|
for( int i = 0; i < mapping::nDetType; i++){
|
||||||
|
if( mapping::detNum[i] == 0 ) continue;
|
||||||
for( int j = 0; j < mapping::detNum[i]; j++){
|
for( int j = 0; j < mapping::detNum[i]; j++){
|
||||||
eE[i][j] = TMath::QuietNaN();
|
eE[i][j] = TMath::QuietNaN();
|
||||||
eT[i][j] = 0;
|
eT[i][j] = 0;
|
||||||
|
|
||||||
if( isTraceExist && traceMethod > 0){
|
if( isTraceExist && traceMethod > 1){
|
||||||
teE[i][j] = TMath::QuietNaN();
|
teE[i][j] = TMath::QuietNaN();
|
||||||
teT[i][j] = TMath::QuietNaN();
|
teT[i][j] = TMath::QuietNaN();
|
||||||
teR[i][j] = TMath::QuietNaN();
|
teR[i][j] = TMath::QuietNaN();
|
||||||
|
@ -34,7 +37,6 @@ Bool_t GeneralSort::Process(Long64_t entry){
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
b_evID->GetEntry(entry);
|
b_evID->GetEntry(entry);
|
||||||
b_multi->GetEntry(entry);
|
b_multi->GetEntry(entry);
|
||||||
b_bd->GetEntry(entry);
|
b_bd->GetEntry(entry);
|
||||||
|
@ -47,8 +49,7 @@ Bool_t GeneralSort::Process(Long64_t entry){
|
||||||
int detID = mapping::map[bd[i]][ch[i]];
|
int detID = mapping::map[bd[i]][ch[i]];
|
||||||
if( detID < 0 ) continue;
|
if( detID < 0 ) continue;
|
||||||
int detType = mapping::FindDetTypeIndex(detID);
|
int detType = mapping::FindDetTypeIndex(detID);
|
||||||
int low = (i == 0 ? 0 : mapping::detMaxID[detType-1]);
|
int reducedDetID = detID - (detID/100) * 100;
|
||||||
int reducedDetID = detID - low;
|
|
||||||
eE[detType][reducedDetID] = e[i] * mapping::detParity[detType];
|
eE[detType][reducedDetID] = e[i] * mapping::detParity[detType];
|
||||||
eT[detType][reducedDetID] = e_t[i];
|
eT[detType][reducedDetID] = e_t[i];
|
||||||
}
|
}
|
||||||
|
@ -65,7 +66,7 @@ Bool_t GeneralSort::Process(Long64_t entry){
|
||||||
|
|
||||||
for( unsigned int i = 0; i < multi; i++){
|
for( unsigned int i = 0; i < multi; i++){
|
||||||
int detID = mapping::map[bd[i]][ch[i]];
|
int detID = mapping::map[bd[i]][ch[i]];
|
||||||
|
if( detID < 0 ) continue;
|
||||||
|
|
||||||
int traceLength = tl[i];
|
int traceLength = tl[i];
|
||||||
gTrace = (TGraph*) arr->ConstructedAt(countTrace, "C");
|
gTrace = (TGraph*) arr->ConstructedAt(countTrace, "C");
|
||||||
|
@ -83,12 +84,11 @@ Bool_t GeneralSort::Process(Long64_t entry){
|
||||||
if( traceMethod == 2){
|
if( traceMethod == 2){
|
||||||
|
|
||||||
int detType = mapping::FindDetTypeIndex(detID);
|
int detType = mapping::FindDetTypeIndex(detID);
|
||||||
|
if( mapping::detNum[detType] == 0 ) continue;
|
||||||
|
|
||||||
//TODO use a blackList
|
//TODO use a blackList
|
||||||
//if( mapping::detTypeName[detType] != "rdt") continue;
|
//if( mapping::detTypeName[detType] != "rdt") continue;
|
||||||
|
|
||||||
//TODO try custom build fiting algorithm. May be faster?
|
|
||||||
gFit = new TF1("gFit", fitFunc, 0, traceLength, numPara);
|
|
||||||
gFit->SetLineColor(6);
|
|
||||||
gFit->SetRange(0, traceLength);
|
gFit->SetRange(0, traceLength);
|
||||||
|
|
||||||
gFit->SetParameter(0, e[i]);
|
gFit->SetParameter(0, e[i]);
|
||||||
|
@ -103,15 +103,13 @@ Bool_t GeneralSort::Process(Long64_t entry){
|
||||||
|
|
||||||
gTrace->Fit("gFit", "QR", "", 0, traceLength);
|
gTrace->Fit("gFit", "QR", "", 0, traceLength);
|
||||||
|
|
||||||
int low = (i == 0 ? 0 : mapping::detMaxID[detType-1]);
|
int reducedDetID = detID - (detID/100) * 100;
|
||||||
int reducedDetID = detID - low;
|
|
||||||
|
|
||||||
teE[detType][reducedDetID] = gFit->GetParameter(0);
|
teE[detType][reducedDetID] = gFit->GetParameter(0);
|
||||||
teT[detType][reducedDetID] = gFit->GetParameter(1);
|
teT[detType][reducedDetID] = gFit->GetParameter(1);
|
||||||
teR[detType][reducedDetID] = gFit->GetParameter(2);
|
teR[detType][reducedDetID] = gFit->GetParameter(2);
|
||||||
|
|
||||||
delete gFit;
|
// delete gFit;
|
||||||
gFit = nullptr;
|
// gFit = nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
//***=================== Trapezoid filter
|
//***=================== Trapezoid filter
|
||||||
|
@ -147,7 +145,7 @@ void GeneralSort::Terminate(){
|
||||||
|
|
||||||
printf("=============================== %s\n", __func__);
|
printf("=============================== %s\n", __func__);
|
||||||
|
|
||||||
DecodeOption();
|
// CleanUpMemory(); //? Should Clean?
|
||||||
|
|
||||||
if( !isParallel){
|
if( !isParallel){
|
||||||
stpWatch.Start(kFALSE);
|
stpWatch.Start(kFALSE);
|
||||||
|
@ -161,10 +159,8 @@ void GeneralSort::Terminate(){
|
||||||
saveFile = TFile::Open(saveFileName);
|
saveFile = TFile::Open(saveFileName);
|
||||||
if( saveFile->IsOpen() ){
|
if( saveFile->IsOpen() ){
|
||||||
TTree * tree = (TTree*) saveFile->FindObjectAny("gen_tree");
|
TTree * tree = (TTree*) saveFile->FindObjectAny("gen_tree");
|
||||||
int validCount = tree->GetEntries();
|
int validCount = tree->GetEntries();
|
||||||
|
|
||||||
saveFile->Close();
|
saveFile->Close();
|
||||||
|
|
||||||
printf("=========================================================================\n");
|
printf("=========================================================================\n");
|
||||||
PrintTraceMethod();
|
PrintTraceMethod();
|
||||||
printf("----- saved as \033[1;33m%s\033[0m. valid event: %d\n", saveFileName.Data() , validCount);
|
printf("----- saved as \033[1;33m%s\033[0m. valid event: %d\n", saveFileName.Data() , validCount);
|
||||||
|
@ -190,16 +186,16 @@ void GeneralSort::Begin(TTree * tree){
|
||||||
}
|
}
|
||||||
|
|
||||||
void GeneralSort::SlaveBegin(TTree * /*tree*/){
|
void GeneralSort::SlaveBegin(TTree * /*tree*/){
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void GeneralSort::SlaveTerminate(){
|
void GeneralSort::SlaveTerminate(){
|
||||||
|
printf("============= %s\n", __func__);
|
||||||
if( isParallel){
|
if( isParallel){
|
||||||
printf("%s::SaveTree\n", __func__);
|
printf("%s::SaveTree\n", __func__);
|
||||||
saveFile->cd();
|
saveFile->cd();
|
||||||
newSaveTree->Write();
|
newSaveTree->Write();
|
||||||
fOutput->Add(proofFile);
|
fOutput->Add(proofFile);
|
||||||
saveFile->Close();
|
saveFile->Close();
|
||||||
|
printf("---- closing this worker\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
|
@ -129,6 +129,8 @@ public :
|
||||||
bool isTraceExist;
|
bool isTraceExist;
|
||||||
int traceMethod;
|
int traceMethod;
|
||||||
|
|
||||||
|
void CleanUpMemory();
|
||||||
|
|
||||||
void SetTraceMethod(int methodID) {traceMethod = methodID;}
|
void SetTraceMethod(int methodID) {traceMethod = methodID;}
|
||||||
void PrintTraceMethod();
|
void PrintTraceMethod();
|
||||||
|
|
||||||
|
@ -186,6 +188,7 @@ void GeneralSort::SetUpTree(){
|
||||||
eT = new ULong64_t * [mapping::nDetType];
|
eT = new ULong64_t * [mapping::nDetType];
|
||||||
|
|
||||||
for( int i = 0 ; i < mapping::nDetType; i++){
|
for( int i = 0 ; i < mapping::nDetType; i++){
|
||||||
|
if( mapping::detNum[i] == 0 ) continue;
|
||||||
eE[i] = new Float_t[mapping::detNum[i]];
|
eE[i] = new Float_t[mapping::detNum[i]];
|
||||||
eT[i] = new ULong64_t[mapping::detNum[i]];
|
eT[i] = new ULong64_t[mapping::detNum[i]];
|
||||||
|
|
||||||
|
@ -205,6 +208,7 @@ void GeneralSort::SetUpTree(){
|
||||||
|
|
||||||
newSaveTree->Branch("trace", arr, 256000);
|
newSaveTree->Branch("trace", arr, 256000);
|
||||||
arr->BypassStreamer();
|
arr->BypassStreamer();
|
||||||
|
arr->Clear("C");
|
||||||
|
|
||||||
if( traceMethod > 1 ){
|
if( traceMethod > 1 ){
|
||||||
|
|
||||||
|
@ -213,6 +217,7 @@ void GeneralSort::SetUpTree(){
|
||||||
teR = new Float_t * [mapping::nDetType];
|
teR = new Float_t * [mapping::nDetType];
|
||||||
|
|
||||||
for( int i = 0 ; i < mapping::nDetType; i++){
|
for( int i = 0 ; i < mapping::nDetType; i++){
|
||||||
|
if( mapping::detNum[i] == 0 ) continue;
|
||||||
teE[i] = new Float_t[mapping::detNum[i]];
|
teE[i] = new Float_t[mapping::detNum[i]];
|
||||||
teT[i] = new Float_t[mapping::detNum[i]];
|
teT[i] = new Float_t[mapping::detNum[i]];
|
||||||
teR[i] = new Float_t[mapping::detNum[i]];
|
teR[i] = new Float_t[mapping::detNum[i]];
|
||||||
|
@ -226,14 +231,14 @@ void GeneralSort::SetUpTree(){
|
||||||
//TODO use a blackList to skip some trace
|
//TODO use a blackList to skip some trace
|
||||||
|
|
||||||
newSaveTree->Branch( ("w" + mapping::detTypeName[i]).c_str(), teE[i], Form("trace_%s[%d]/F", mapping::detTypeName[i].c_str(), mapping::detNum[i]));
|
newSaveTree->Branch( ("w" + mapping::detTypeName[i]).c_str(), teE[i], Form("trace_%s[%d]/F", mapping::detTypeName[i].c_str(), mapping::detNum[i]));
|
||||||
newSaveTree->Branch( ("w" + mapping::detTypeName[i]+"T").c_str(), teT[i], Form("trace_%s_time[%d]/l", mapping::detTypeName[i].c_str(), mapping::detNum[i]));
|
newSaveTree->Branch( ("w" + mapping::detTypeName[i]+"T").c_str(), teT[i], Form("trace_%s_time[%d]/F", mapping::detTypeName[i].c_str(), mapping::detNum[i]));
|
||||||
newSaveTree->Branch( ("w" + mapping::detTypeName[i]+"R").c_str(), teR[i], Form("trace_%s_rise[%d]/l", mapping::detTypeName[i].c_str(), mapping::detNum[i]));
|
newSaveTree->Branch( ("w" + mapping::detTypeName[i]+"R").c_str(), teR[i], Form("trace_%s_rise[%d]/F", mapping::detTypeName[i].c_str(), mapping::detNum[i]));
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
newSaveTree->Print("toponly"); //very important, otherwise the mac will blow up.
|
|
||||||
|
if( !isParallel ) newSaveTree->Print("toponly");
|
||||||
}
|
}
|
||||||
|
|
||||||
//^##############################################################
|
//^##############################################################
|
||||||
|
@ -258,7 +263,7 @@ void GeneralSort::DecodeOption(){
|
||||||
|
|
||||||
//^##############################################################
|
//^##############################################################
|
||||||
void GeneralSort::Init(TTree *tree){
|
void GeneralSort::Init(TTree *tree){
|
||||||
|
printf("============= %s\n", __func__);
|
||||||
// Set branch addresses and branch pointers
|
// Set branch addresses and branch pointers
|
||||||
if (!tree) return;
|
if (!tree) return;
|
||||||
fChain = tree;
|
fChain = tree;
|
||||||
|
@ -273,31 +278,28 @@ void GeneralSort::Init(TTree *tree){
|
||||||
fChain->SetBranchAddress("lowFlag", lowFlag, &b_lowFlag);
|
fChain->SetBranchAddress("lowFlag", lowFlag, &b_lowFlag);
|
||||||
fChain->SetBranchAddress("highFlag", highFlag, &b_highFlag);
|
fChain->SetBranchAddress("highFlag", highFlag, &b_highFlag);
|
||||||
|
|
||||||
TBranch * br = (TBranch *) fChain->GetListOfBranches()->FindObject("tl");
|
TBranch * br = (TBranch *) fChain->GetListOfBranches()->FindObject("traceLen");
|
||||||
if( br == NULL ){
|
if( br == NULL ){
|
||||||
printf(" ++++++++ no Trace.\n");
|
printf(" ++++++++ no Trace.\n");
|
||||||
isTraceExist = false;
|
isTraceExist = false;
|
||||||
}else{
|
}else{
|
||||||
printf(" ++++++++ Found Trace.\n");
|
printf(" ++++++++ Found Trace.\n");
|
||||||
isTraceExist = true;
|
isTraceExist = true;
|
||||||
fChain->SetBranchAddress("tl", tl, &b_tl);
|
fChain->SetBranchAddress("traceLen", tl, &b_tl);
|
||||||
fChain->SetBranchAddress("trace", trace, &b_trace);
|
fChain->SetBranchAddress("trace", trace, &b_trace);
|
||||||
}
|
}
|
||||||
|
|
||||||
NumEntries = fChain->GetEntries();
|
NumEntries = fChain->GetEntries();
|
||||||
printf( " ========== total Entry : %ld\n", NumEntries);
|
printf( " ======== total Entry : %ld\n", NumEntries);
|
||||||
|
|
||||||
//########################### Get Option
|
//########################### Get Option
|
||||||
DecodeOption();
|
DecodeOption();
|
||||||
|
|
||||||
if( isTraceExist ){
|
|
||||||
PrintTraceMethod();
|
|
||||||
}else{
|
|
||||||
printf("++++++++ no Trace found\n");
|
|
||||||
}
|
|
||||||
|
|
||||||
SetUpTree();
|
SetUpTree();
|
||||||
|
|
||||||
|
gFit = new TF1("gFit", fitFunc, 0, 1250, numPara);
|
||||||
|
gFit->SetLineColor(6);
|
||||||
|
|
||||||
printf("---- end of Init %s\n ", __func__);
|
printf("---- end of Init %s\n ", __func__);
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -315,7 +317,43 @@ void GeneralSort::PrintTraceMethod(){
|
||||||
case 3 : traceMethodStr = "Trapezoid"; break;
|
case 3 : traceMethodStr = "Trapezoid"; break;
|
||||||
default: traceMethodStr = "Unknown"; break;
|
default: traceMethodStr = "Unknown"; break;
|
||||||
}
|
}
|
||||||
printf("\033[1;33m ===== Trace method ? %s \033[m\n", traceMethodStr);
|
printf("\033[1;33m ===== Trace method ? %s (%d) \033[m\n", traceMethodStr, traceMethod);
|
||||||
|
}
|
||||||
|
|
||||||
|
void GeneralSort::CleanUpMemory(){
|
||||||
|
printf("Clean up memory");
|
||||||
|
if( traceMethod > 1 ){
|
||||||
|
for( int i = 0 ; i < mapping::nDetType; i++){
|
||||||
|
if( mapping::detNum[i] == 0 ) continue;
|
||||||
|
delete [] teE[i];
|
||||||
|
delete [] teT[i];
|
||||||
|
delete [] teR[i];
|
||||||
|
|
||||||
|
delete [] eE[i];
|
||||||
|
delete [] eT[i];
|
||||||
|
}
|
||||||
|
delete [] teE;
|
||||||
|
delete [] teT;
|
||||||
|
delete [] teR;
|
||||||
|
}
|
||||||
|
|
||||||
|
for( int i = 0 ; i < mapping::nDetType; i++){
|
||||||
|
if( mapping::detNum[i] == 0 ) continue;
|
||||||
|
delete [] eE[i];
|
||||||
|
delete [] eT[i];
|
||||||
|
}
|
||||||
|
delete [] eE;
|
||||||
|
delete [] eT;
|
||||||
|
|
||||||
|
//trace
|
||||||
|
if( arr ) delete arr ;
|
||||||
|
if( gTrace ) delete gTrace;
|
||||||
|
if( gFit ) delete gFit;
|
||||||
|
if( arrTrapezoid ) delete arrTrapezoid ;
|
||||||
|
if( gTrapezoid ) delete gTrapezoid;
|
||||||
|
|
||||||
|
printf(".... done\n");
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif // #ifdef GeneralSort_cxx
|
#endif // #ifdef GeneralSort_cxx
|
||||||
|
|
|
@ -44,7 +44,7 @@ if [ $# -ge 3 ]; then nWorker=$3; fi
|
||||||
if [ $# -ge 4 ]; then isMonitor=$4; fi
|
if [ $# -ge 4 ]; then isMonitor=$4; fi
|
||||||
|
|
||||||
|
|
||||||
TraceMethod=$((nWorker / 100));
|
temp=$((nWorker / 100));
|
||||||
if [ $temp -lt 0 ]; then
|
if [ $temp -lt 0 ]; then
|
||||||
TraceMethod=$((-1 * temp));
|
TraceMethod=$((-1 * temp));
|
||||||
fi
|
fi
|
||||||
|
|
|
@ -46,3 +46,5 @@ function ShowRunSize {
|
||||||
fi
|
fi
|
||||||
du -hc $SOLARISANADIR/data_raw/${expName}_${RUN}_*.sol
|
du -hc $SOLARISANADIR/data_raw/${expName}_${RUN}_*.sol
|
||||||
}
|
}
|
||||||
|
|
||||||
|
2Working
|
Loading…
Reference in New Issue
Block a user