AliEmcalTrackSelection.h

Go to the documentation of this file.

/************************************************************************************
 * Copyright (C) 2017, Copyright Holders of the ALICE Collaboration                 *
 * All rights reserved.                                                             *
 *                                                                                  *
 * Redistribution and use in source and binary forms, with or without               *
 * modification, are permitted provided that the following conditions are met:      *
 *     * Redistributions of source code must retain the above copyright             *
 *       notice, this list of conditions and the following disclaimer.              *
 *     * Redistributions in binary form must reproduce the above copyright          *
 *       notice, this list of conditions and the following disclaimer in the        *
 *       documentation and/or other materials provided with the distribution.       *
 *     * Neither the name of the <organization> nor the                             *
 *       names of its contributors may be used to endorse or promote products       *
 *       derived from this software without specific prior written permission.      *
 *                                                                                  *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND  *
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED    *
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE           *
 * DISCLAIMED. IN NO EVENT SHALL ALICE COLLABORATION BE LIABLE FOR ANY              *
 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES       *
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;     *
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND      *
 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT       *
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS    *
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.                     *
 ************************************************************************************/
#ifndef ALIEMCALTRACKSELECTION_H_
#define ALIEMCALTRACKSELECTION_H_

#include <TObject.h>
#include <TBits.h>
#include "AliEmcalTrackSelResultPtr.h"

class TClonesArray;
class TList;
class TObjArray;
class AliVCuts;
class AliVEvent;
class AliVTrack;

namespace PWG {
namespace EMCAL {

class AliEmcalCutBase;

}
}

class AliEmcalManagedObject : public TObject {
public:
  AliEmcalManagedObject();

  AliEmcalManagedObject(TObject *object, bool owner = true);

  AliEmcalManagedObject(const AliEmcalManagedObject &ref);

  AliEmcalManagedObject &operator=(const AliEmcalManagedObject &ref);

  virtual ~AliEmcalManagedObject() { Cleanup(); }

  operator bool() const { return fManagedObject != NULL; }

  bool IsOwner() const { return fOwner; }

  void SetOwner(bool owner = true) { fOwner = owner; }

  void SetObject(TObject *managedObject, bool owner = true) {
    Cleanup();
    fManagedObject = managedObject;
    fOwner = owner;
  }

  TObject *GetObject() { return fManagedObject; }

protected:

  void Cleanup();

private:
  Bool_t            fOwner;             
  TObject           *fManagedObject;    

  ClassDef(AliEmcalManagedObject, 1);
};

class AliEmcalTrackSelection : public TObject {
public:

  enum ETrackFilterType_t {
    kNoTrackFilter = 0,   
    kCustomTrackFilter,   
    kHybridTracks,        
    kTPCOnlyTracks,       
    kITSPureTracks,       
    kHybridTracks2010wNoRefit,  
    kHybridTracks2010woNoRefit, 
    kHybridTracks2011wNoRefit,  
    kHybridTracks2011woNoRefit, 
    kHybridTracks2018TRD        
  };

  AliEmcalTrackSelection();

  AliEmcalTrackSelection(const AliEmcalTrackSelection &ref);

  AliEmcalTrackSelection &operator=(const AliEmcalTrackSelection &ref);

  virtual ~AliEmcalTrackSelection();

  TObjArray *GetAcceptedTracks(const TClonesArray * const tracks);

  TObjArray *GetAcceptedTracks(const AliVEvent *const event);

  virtual PWG::EMCAL::AliEmcalTrackSelResultPtr IsTrackAccepted(AliVTrack * const trk) = 0;

  virtual void GenerateTrackCuts(ETrackFilterType_t type, const char* period = "") = 0;

  void AddTrackCuts(AliVCuts *cuts);

  void AddTrackCuts(PWG::EMCAL::AliEmcalCutBase *cuts);

  void AddTrackCuts(TObjArray *cuts);

  Int_t GetNumberOfCutObjects() const;

  PWG::EMCAL::AliEmcalCutBase *GetTrackCuts(Int_t icut);

  void SetSelectionModeAny() { fSelectionModeAny = kTRUE ; }

  void SetSelectionModeAll() { fSelectionModeAny = kFALSE; }

  virtual void SaveQAObjects(TList *outputList) {}

protected:
  TObjArray    *fListOfTracks;         
  TObjArray    *fListOfCuts;           
  Bool_t        fSelectionModeAny;     


};

#endif /* ALIEMCALTRACKSELECTION_H_ */