--- /dev/null
+/**************************************************************************
+ * Copyright(c) 1998-2007, 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. *
+ **************************************************************************/
+
+/* $Id$ */
+
+//-------------------------------------------------------------------------
+// Event handler for multiple VEvent input.
+// This class handles multiple inputs for event mixing.
+// Author: Andreas Morsch, CERN
+//-------------------------------------------------------------------------
+
+#include "AliMultiEventInputHandler.h"
+#include "AliVEvent.h"
+#include "AliAODEvent.h"
+#include "AliESDEvent.h"
+#include "AliVEventPool.h"
+#include "AliLog.h"
+#include <TObjArray.h>
+#include <TTree.h>
+
+
+ClassImp(AliMultiEventInputHandler)
+
+AliMultiEventInputHandler::AliMultiEventInputHandler() :
+ AliInputEventHandler(),
+ fBufferSize(0),
+ fFormat(1),
+ fNBuffered(0),
+ fIndex(0),
+ fCurrentBin(0),
+ fTree(0),
+ fEventPool(0),
+ fEventBuffer(0)
+{
+ // Default constructor
+}
+
+//______________________________________________________________________________
+AliMultiEventInputHandler::AliMultiEventInputHandler(Int_t size, Int_t format) :
+ AliInputEventHandler(),
+ fBufferSize(size),
+ fFormat(format),
+ fNBuffered(0),
+ fIndex(0),
+ fCurrentBin(0),
+ fTree(0),
+ fEventPool(0),
+ fEventBuffer(new AliVEvent*[size])
+{
+ // Default constructor
+ for (Int_t i = 0; i < size; i++)
+ if (fFormat == 1) {
+ fEventBuffer[i] = new AliAODEvent();
+ } else if (fFormat == 0) {
+ fEventBuffer[i] = new AliESDEvent();
+ } else{
+ AliWarning(Form("Unknown Format %5d", fFormat));
+ }
+}
+
+//______________________________________________________________________________
+AliMultiEventInputHandler::AliMultiEventInputHandler(const char* name, const char* title, Int_t size, Int_t format):
+ AliInputEventHandler(name, title),
+ fBufferSize(size),
+ fFormat(format),
+ fNBuffered(0),
+ fIndex(0),
+ fCurrentBin(0),
+ fTree(0),
+ fEventPool(0),
+ fEventBuffer(new AliVEvent*[size])
+{
+ // Constructor
+ for (Int_t i = 0; i < size; i++)
+ if (fFormat == 1) {
+ fEventBuffer[i] = new AliAODEvent();
+ } else if (fFormat == 0) {
+ fEventBuffer[i] = new AliESDEvent();
+ } else{
+ AliWarning(Form("Unknown Format %5d", fFormat));
+ }
+}
+
+//______________________________________________________________________________
+AliMultiEventInputHandler::~AliMultiEventInputHandler()
+{
+// Destructor
+}
+
+Bool_t AliMultiEventInputHandler::Init(TTree* tree, Option_t* /*opt*/)
+{
+ // Initialisation necessary for each new tree
+ fTree = tree;
+ if (!fTree) return kFALSE;
+ // Get pointer to AOD event
+ fEventBuffer[0]->ReadFromTree(fTree, "");
+ fIndex = 0;
+ fNBuffered = 1;
+ return kTRUE;
+}
+
+Bool_t AliMultiEventInputHandler::BeginEvent(Long64_t /*entry*/)
+{
+ // Actions before analysis of each event
+ //
+ // Reset the number of events buffered for this bin to 0
+ if (fCurrentBin != (fEventPool->BinNumber())) {
+ fCurrentBin = fEventPool->BinNumber();
+ fNBuffered = 0;
+ }
+ return kTRUE;
+}
+
+Bool_t AliMultiEventInputHandler::FinishEvent()
+{
+ //
+ // Connect the next event in the buffer to the tree
+ fIndex++;
+
+ fIndex %= fBufferSize;
+ AliInfo(Form("Connecting buffer entry %5d", fIndex));
+
+ fEventBuffer[fIndex]->ReadFromTree(fTree, "reconnect");
+
+ fNBuffered++;
+ if (fNBuffered > fBufferSize) fNBuffered = fBufferSize;
+
+ return (kTRUE);
+}
+
+
+AliVEvent* AliMultiEventInputHandler::GetEvent(Int_t iev) const
+{
+ // Get event number iev from buffer
+ if ((iev < 0) || (iev >= fBufferSize))
+ {
+ AliWarning(Form("Event number out of range: %10d", iev));
+ return 0;
+ }
+
+ iev = fIndex - (fBufferSize - 1 - iev);
+ if (iev < 0) iev += fBufferSize;
+ AliInfo(Form("Event index in buffer is %5d", iev));
+ return (fEventBuffer[iev]);
+}
+
--- /dev/null
+#ifndef ALIMULTIEVENTINPUTHANDLER_H
+#define ALIMULTIEVENTINPUTHANDLER_H
+/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice */
+
+/* $Id$ */
+
+//----------------------------------------------------------------------------
+// Multi VEvent Input Handler realisation of the AliVEventHandler interface.
+// This class handles multiple events for mixing.
+// Author: Andreas Morsch, CERN
+//----------------------------------------------------------------------------
+
+#include "AliInputEventHandler.h"
+class AliVEventPool;
+class AliVEvent;
+
+class AliMultiEventInputHandler : public AliInputEventHandler {
+
+ public:
+ AliMultiEventInputHandler();
+ AliMultiEventInputHandler(Int_t size, Int_t format = 1);
+ AliMultiEventInputHandler(const char* name, const char* title, Int_t size, Int_t format = 1);
+ virtual ~AliMultiEventInputHandler();
+ void SetBufferSize(Int_t size) {fBufferSize = size;}
+ void SetEventPool(AliVEventPool* pool) {fEventPool = pool;}
+ Int_t GetBufferSize() const {return fBufferSize;}
+ Int_t GetNBuffered() const {return fNBuffered;}
+ Bool_t IsBufferReady() const {return (fNBuffered >= (fBufferSize -1));}
+ Bool_t IsFreshBuffer() const {return (fIndex == (fBufferSize - 1));}
+ AliVEventPool *GetEventPool() const {return fEventPool;}
+ virtual AliVEvent *GetEvent() const {return 0;}
+ virtual AliVEvent *GetEvent(Int_t iev) const;
+ AliVEvent *GetLatestEvent() const {return fEventBuffer[fIndex];}
+ // From the interface
+ virtual Bool_t Init(Option_t* /*opt*/) {return kTRUE;}
+ virtual Bool_t Init(TTree* tree, Option_t* /*opt*/);
+ virtual Bool_t FinishEvent();
+ virtual Bool_t BeginEvent(Long64_t /*entry*/);
+
+ private:
+ AliMultiEventInputHandler(const AliMultiEventInputHandler& handler);
+ AliMultiEventInputHandler& operator=(const AliMultiEventInputHandler& handler);
+ private:
+ Int_t fBufferSize; // Size of the buffer
+ Int_t fFormat; // 0: ESD 1: AOD
+ Int_t fNBuffered; // Number of events actually buffered
+ Int_t fIndex; // Pointer to most recent event
+ Int_t fCurrentBin; // Current bin from the pool
+ TTree* fTree; // Pointer to the tree
+ AliVEventPool* fEventPool; // Pointer to the pool
+ AliVEvent** fEventBuffer; // The event buffer
+ ClassDef(AliMultiEventInputHandler, 1);
+};
+
+#endif
git://git.uio.no / u / mrichter / AliRoot.git / commitdiff