MONITOR/alieventserver/AliEventServerReconstruction.cxx

   1 // Author: Mihai Niculescu 2013
   2
   3 /**************************************************************************
   4  * Copyright(c) 1998-2013, ALICE Experiment at CERN, all rights reserved. *
   5  * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
   6  * full copyright notice.                                                 *
   7  **************************************************************************/
   8
   9 #include <TEnv.h>
  10 #include <TSystem.h>
  11 #include <TThread.h>
  12 #include <TXMLEngine.h>
  13
  14 #include <AliLog.h>
  15 #include <AliESDEvent.h>
  16 #include <AliCDBManager.h>
  17 #include <AliGRPPreprocessor.h>
  18 #include <AliReconstruction.h>
  19 #include <AliTPCRecoParam.h>
  20
  21 #include <iostream>
  22 #include <sstream>
  23
  24 #include "AliEventServerUtil.h"
  25 #include "AliEventServerReconstruction.h"
  26 #include "AliStorageEventManager.h"
  27 #include "AliStorageTypes.h"
  28
  29 #include "AliESDEvent.h"
  30 #include "AliESDtrack.h"
  31 #include "AliTrackPointArray.h"
  32 #include "AliESDfriendTrack.h"
  33 #include "AliExternalTrackParam.h"
  34 #include "AliTrackerBase.h"
  35 #include "AliTracker.h"
  36
  37 using namespace std;
  38
  39 AliEventServerReconstruction::AliEventServerReconstruction()
  40         : TQObject(),
  41           fAliReco(0),
  42           fCDBmanager(0),
  43           fCurrentRunId(0),
  44           fIsListenning(kFALSE),
  45           fSettings(0),
  46           fHost(0),
  47           fRecoThread(0),
  48           fRecoIsRunning(false),
  49           fRecoWasInitialized(false)
  50 {}
  51
  52 AliEventServerReconstruction::~AliEventServerReconstruction()
  53 {
  54         Close();
  55         if(fSettings){delete fSettings;fSettings=0;}
  56         if(fAliReco){delete fAliReco;fAliReco=0;}
  57 }
  58
  59 void AliEventServerReconstruction::Close()
  60 {
  61         if(fIsListenning)
  62         {
  63                 Info("AliRecoServer::Close", "Closing Server");
  64                 StopReconstruction();
  65                 fIsListenning = kFALSE;
  66         }
  67 }
  68
  69 Bool_t AliEventServerReconstruction::StartReconstruction(Int_t run, const char* input)
  70 {
  71   cout<<"Start of run:"<<run<<endl;
  72         fCurrentRunId = run;
  73
  74         // re-read settings
  75         if(fSettings){delete fSettings;fSettings=0;}
  76         fSettings = new TEnv(AliEventServerUtil::GetPathToServerConf());
  77
  78         TString recoBaseDir = fSettings->GetValue("server.saveRecoDir",DEFAULT_SERVER_SAVE_RECO_DIR);
  79
  80         // Create directories and logfile
  81         TString logFile = Form("%s/log/run%d.log",recoBaseDir.Data(),run);
  82         Info("DoStart","Reconstruction log will be written to %s",logFile.Data());
  83         if( gSystem->RedirectOutput(logFile.Data())!=0)
  84         {
  85                 printf(Form("AliRecoServer::StartReconstruction [] Error while trying to redirect output to [%s]. Exiting...", logFile.Data()) );
  86                 return kFALSE;
  87         }
  88         gSystem->cd(recoBaseDir.Data());
  89
  90
  91         TString gdcs;
  92         if (RetrieveGRP(run,gdcs) <= 0 || gdcs.IsNull()){return kFALSE;}
  93
  94         gSystem->mkdir(Form("run%d", run));
  95         gSystem->cd(Form("run%d", run));
  96
  97         // Create Reco and Reco Thread
  98         cout<<"Setup reco will be called"<<endl;
  99         SetupReco(input);
 100
 101         fHost = (const char*)Form("%s:%d", fSettings->GetValue("server.host", DEFAULT_SERVER_HOST), fSettings->GetValue("server.port", DEFAULT_SERVER_PORT));
 102
 103         cout<<"Creating new thread"<<endl;
 104         fRecoThread = new TThread("AliEventServerReconstruction",Dispatch, (void*)this);
 105         fRecoThread->Run();
 106         fIsListenning = kTRUE;
 107         fRecoIsRunning=true;
 108         cout<<"Reco started"<<endl;
 109         return true;
 110 }
 111
 112 bool AliEventServerReconstruction::StopReconstruction()
 113 {
 114   cout<<"Reco server -- StopPeconstruction() called"<<endl;
 115   if(!fRecoIsRunning || !fRecoThread)
 116     {
 117       cout<<"Reco is not running. No need to stop it."<<endl;
 118       return true;
 119     }
 120   if(!fRecoWasInitialized)
 121     {
 122       cout<<"Reco is under initialization. Wait until it's finished"<<endl;
 123
 124       return false;
 125     }
 126   cout<<"killing thread"<<endl;
 127   fRecoThread->Kill();
 128   cout<<"thread killed"<<endl;
 129   delete fRecoThread;
 130   fRecoThread=0;
 131   cout<<"Reco server -- thread removed"<<endl;
 132   // Emit("Stopped()");
 133
 134   cout<<"Reco server -- terminating reconstruction"<<endl;
 135   fAliReco->SlaveTerminate();
 136   if (fAliReco->GetAbort() != TSelector::kContinue) return false;
 137   fAliReco->Terminate();
 138   if (fAliReco->GetAbort() != TSelector::kContinue) return false;
 139
 140   if(fAliReco){delete fAliReco;fAliReco=0;}
 141   cout<<"Reco server -- deleting CDBManager"<<endl;
 142   if(fCDBmanager){fCDBmanager->Destroy();fCDBmanager=0;}
 143   cout<<"Reco server -- recontruction stopped"<<endl;
 144   fRecoIsRunning=false;
 145   fRecoWasInitialized=false;
 146   return true;
 147 }
 148
 149 void AliEventServerReconstruction::ReconstructionHandle()
 150 {
 151         TThread::SetCancelAsynchronous();
 152         TThread::SetCancelOn();
 153
 154         if(!fAliReco) return;
 155
 156         AliStorageEventManager *eventManager = AliStorageEventManager::GetEventManagerInstance();
 157         eventManager->CreateSocket(EVENTS_SERVER_PUB);
 158         eventManager->CreateSocket(XML_PUB);
 159         cout<<"Sockets created"<<endl;
 160         fAliReco->Begin(NULL);
 161         cout<<"Reco began"<<endl;
 162         if (fAliReco->GetAbort() != TSelector::kContinue) return;
 163         fAliReco->SlaveBegin(NULL);
 164         cout<<"Slave began"<<endl;
 165         if (fAliReco->GetAbort() != TSelector::kContinue) return;
 166
 167         fRecoWasInitialized=true;
 168         //******* The loop over events
 169         Int_t iEvent = 0;
 170         AliESDEvent* event;
 171         while (fAliReco->HasNextEventAfter(iEvent))
 172         {
 173                 // check if process has enough resources
 174           cout<<"Event server -- checking resources"<<endl;
 175                 if (!fAliReco->HasEnoughResources(iEvent)) break;
 176                 cout<<"Event server -- resources checked"<<endl;
 177                 Bool_t status = fAliReco->ProcessEvent(iEvent);
 178                 cout<<"Event server -- event processed"<<endl;
 179
 180                 if (status)
 181                 {
 182                         event = fAliReco->GetESDEvent();
 183                         cout<<"Event server -- sending event"<<endl;
 184                         eventManager->Send(event,EVENTS_SERVER_PUB);
 185                         cout<<"Event server -- sending event as xml"<<endl;
 186                         eventManager->SendAsXml(event,XML_PUB);
 187                         cout<<"Event server -- xml sent"<<endl;
 188                 }
 189                 else
 190                 {
 191                   cout<<"Event server -- aborting"<<endl;
 192                         fAliReco->Abort("ProcessEvent",TSelector::kAbortFile);
 193                 }
 194                 cout<<"Event server -- cleaning event"<<endl;
 195                 fAliReco->CleanProcessedEvent();
 196                 cout<<"Event server -- event cleaned"<<endl;
 197                 iEvent++;
 198         }
 199         StopReconstruction();
 200 }
 201
 202 Int_t AliEventServerReconstruction::RetrieveGRP(UInt_t run, TString &gdc)
 203 {
 204         if(!fSettings) return (-1);
 205
 206         // Retrieve GRP entry for given run from aldaqdb.
 207         TString dbHost = fSettings->GetValue("logbook.host", DEFAULT_LOGBOOK_HOST);
 208         Int_t dbPort =  fSettings->GetValue("logbook.port", DEFAULT_LOGBOOK_PORT);
 209         TString dbName =  fSettings->GetValue("logbook.db", DEFAULT_LOGBOOK_DB);
 210         TString user =  fSettings->GetValue("logbook.user", DEFAULT_LOGBOOK_USER);
 211         TString password = fSettings->GetValue("logbook.pass", DEFAULT_LOGBOOK_PASS);
 212
 213         Int_t ret=AliGRPPreprocessor::ReceivePromptRecoParameters(run, dbHost.Data(),
 214                                                                   dbPort, dbName.Data(),
 215                                                                   user.Data(), password.Data(),
 216                                                                   Form("local://%s",gSystem->pwd()),
 217                                                                   gdc);
 218
 219         if(ret>0) Info("RetrieveGRP","Last run of the same type is: %d",ret);
 220         else if(ret==0) Warning("RetrieveGRP","No previous run of the same type found");
 221         else if(ret<0) Error("Retrieve","Error code while retrieving GRP parameters returned: %d",ret);
 222         return(ret);
 223 }
 224
 225 void AliEventServerReconstruction::SetupReco(const char* input)
 226 {
 227         if(!fSettings) return;
 228
 229         //AliTPCRecoParam::SetUseTimeCalibration(kFALSE); //-- !probably should be set from conf file!
 230
 231         printf(Form("=========================[local://%s/..]===========\n",gSystem->pwd()));
 232
 233         /* Settings CDB */
 234         fCDBmanager = AliCDBManager::Instance();
 235
 236         fCDBmanager->SetDefaultStorage(fSettings->GetValue("cdb.defaultStorage", DEFAULT_CDB_STORAGE));
 237         fCDBmanager->SetSpecificStorage(fSettings->GetValue( "cdb.specificStoragePath1", DEFAULT_CDB_SPEC_STORAGE_PATH1),
 238                                     fSettings->GetValue( "cdb.specificStorageValue1", DEFAULT_CDB_SPEC_STORAGE_VALUE1));
 239
 240         fCDBmanager->SetSpecificStorage(fSettings->GetValue( "cdb.specificStoragePath2", DEFAULT_CDB_SPEC_STORAGE_PATH2),
 241                                     fSettings->GetValue( "cdb.specificStorageValue2", DEFAULT_CDB_SPEC_STORAGE_VALUE2));
 242
 243         fCDBmanager->SetSpecificStorage(fSettings->GetValue( "cdb.specificStoragePath3", DEFAULT_CDB_SPEC_STORAGE_PATH3),
 244                                     fSettings->GetValue( "cdb.specificStorageValue3", DEFAULT_CDB_SPEC_STORAGE_VALUE3));
 245
 246         /* Reconstruction settings */
 247         if(!fAliReco)fAliReco = new AliReconstruction();
 248
 249         // QA options
 250         //rec->SetRunQA(fSettings->GetValue( "qa.runDetectors", DEFAULT_QA_RUN));
 251         //rec->SetRunGlobalQA(fSettings->GetValue( "qa.runGlobal", DEFAULT_QA_RUN_GLOBAL));
 252         fAliReco->SetQARefDefaultStorage(fSettings->GetValue( "qa.defaultStorage",DEFAULT_QAREF_STORAGE)) ;
 253         //rec->SetRunPlaneEff(fSettings->GetValue( "reco.runPlaneEff", DEFAULT_RECO_RUN_PLANE_EFF));
 254
 255         fAliReco->SetRunQA(":");
 256         fAliReco->SetRunGlobalQA(false);
 257         fAliReco->SetRunPlaneEff(false);
 258
 259         // AliReconstruction settings
 260         fAliReco->SetWriteESDfriend(fSettings->GetValue( "reco.writeESDfriend", DEFAULT_RECO_WRITE_ESDF));
 261         fAliReco->SetWriteAlignmentData(fSettings->GetValue( "reco.writeAlignment",DEFAULT_RECO_WRITE_ALIGN));
 262         fAliReco->SetInput(input); // reconstruct data from this input
 263         fAliReco->SetRunReconstruction(fSettings->GetValue( "reco.detectors", DEFAULT_RECO_DETECTORS));
 264         fAliReco->SetUseTrackingErrorsForAlignment("ITS"); //-- !should be set from conf file!
 265
 266         // switch off cleanESD
 267         fAliReco->SetCleanESD(fSettings->GetValue( "reco.cleanESD",DEFAULT_RECO_CLEAN_ESD));
 268
 269         // init reco for given run
 270         fAliReco->InitRun(input);
 271 }