LaserTrackFilterSpec.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#include <algorithm>
#include <iterator>
 
#include "DataFormatsTPC/TrackTPC.h"
#include "TPCCalibration/CalibLaserTracks.h"
#include "Framework/Task.h"
#include "Framework/ConfigParamRegistry.h"
#include "Framework/ControlService.h"
#include "Framework/WorkflowSpec.h"
 
using namespace o2::framework;
 
namespace o2::tpc
{
 
class LaserTrackFilterDevice : public o2::framework::Task
{
 public:
  void init(o2::framework::InitContext& ic) final
  {
  }
 
  void run(o2::framework::ProcessingContext& pc) final
  {
    const auto tracks = pc.inputs().get<gsl::span<TrackTPC>>("tracks");
    std::copy_if(tracks.begin(), tracks.end(), std::back_inserter(mLaserTracks),
                 [this](const auto& track) { return isLaserTrackCandidate(track); });
 
    LOGP(info, "Filtered {} laser track candidates out of {} total tpc tracks", mLaserTracks.size(), tracks.size());
 
    sendOutput(pc.outputs());
  }
 
  void endOfStream(o2::framework::EndOfStreamContext& ec) final
  {
  }
 
 private:
  std::vector<TrackTPC> mLaserTracks;
 
  //________________________________________________________________
  void sendOutput(DataAllocator& output)
  {
    output.snapshot(Output{"TPC", "LASERTRACKS", 0}, mLaserTracks);
    mLaserTracks.clear();
  }
 
  bool isLaserTrackCandidate(const TrackTPC& track)
  {
    if (track.getP() < 1) {
      return false;
    }
 
    if (track.getNClusters() < 80) {
      //return false;
    }
 
    if (track.hasBothSidesClusters()) {
      return false;
    }
 
    const auto& parOutLtr = track.getOuterParam();
    if (parOutLtr.getX() < 220) {
      return false;
    }
 
    const int side = track.hasCSideClusters();
    if (!CalibLaserTracks::hasNearbyLaserRod(parOutLtr, side)) {
      return false;
    }
 
    return true;
  }
};
 
DataProcessorSpec getLaserTrackFilter()
{
  using device = o2::tpc::LaserTrackFilterDevice;
 
  std::vector<OutputSpec> outputs;
  outputs.emplace_back("TPC", "LASERTRACKS", 0, o2::framework::Lifetime::Timeframe);
 
  return DataProcessorSpec{
    "tpc-laser-track-filter",
    Inputs{{"tracks", "TPC", "TRACKS", 0}},
    outputs,
    AlgorithmSpec{adaptFromTask<device>()},
    Options{}};
}
 
} // namespace o2::tpc