TrackFitterSpec.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 "TrackFitterSpec.h"
 
#include <stdexcept>
#include <list>
#include <vector>
 
#include <gsl/span>
 
#include "Framework/ConfigParamRegistry.h"
#include "Framework/ControlService.h"
#include "Framework/DataProcessorSpec.h"
#include "Framework/Lifetime.h"
#include "Framework/Output.h"
#include "Framework/Task.h"
#include "Framework/Logger.h"
 
#include "DataFormatsMCH/ROFRecord.h"
#include "DataFormatsMCH/TrackMCH.h"
#include "DataFormatsMCH/Cluster.h"
#include "MCHTracking/TrackParam.h"
#include "MCHTracking/Track.h"
#include "MCHTracking/TrackExtrap.h"
#include "MCHTracking/TrackFitter.h"
 
namespace o2
{
namespace mch
{
 
using namespace std;
using namespace o2::framework;
 
class TrackFitterTask
{
 public:
  //_________________________________________________________________________________________________
  void init(framework::InitContext& ic)
  {
    LOG(info) << "initializing track fitter";
    auto l3Current = ic.options().get<float>("l3Current");
    auto dipoleCurrent = ic.options().get<float>("dipoleCurrent");
    mTrackFitter.initField(l3Current, dipoleCurrent);
    mTrackFitter.smoothTracks(true);
    TrackExtrap::useExtrapV2();
  }
 
  //_________________________________________________________________________________________________
  void run(framework::ProcessingContext& pc)
  {
 
    // get the input ROFs, tracks and attached clusters
    auto rofsIn = pc.inputs().get<gsl::span<ROFRecord>>("rofsin");
    auto tracksIn = pc.inputs().get<gsl::span<TrackMCH>>("tracksin");
    auto clustersIn = pc.inputs().get<gsl::span<Cluster>>("clustersin");
 
    // create the output messages for ROFs, refitted tracks and attached clusters
    auto& rofsOut = pc.outputs().make<std::vector<ROFRecord>>(OutputRef{"rofsout"});
    auto& tracksOut = pc.outputs().make<std::vector<TrackMCH>>(OutputRef{"tracksout"});
    auto& clustersOut = pc.outputs().make<std::vector<Cluster>>(OutputRef{"clustersout"});
 
    rofsOut.reserve(rofsIn.size());
    for (const auto& rof : rofsIn) {
 
      // loop over tracks of the current ROF
      int trackOffset(tracksOut.size());
      for (const auto& mchTrack : tracksIn.subspan(rof.getFirstIdx(), rof.getNEntries())) {
 
        // get the clusters attached to the track
        auto trackClusters = clustersIn.subspan(mchTrack.getFirstClusterIdx(), mchTrack.getNClusters());
 
        // create the internal track
        Track track{};
        for (const auto& cluster : trackClusters) {
          track.createParamAtCluster(cluster);
        }
 
        // refit the track
        try {
          mTrackFitter.fit(track);
        } catch (exception const& e) {
          LOG(error) << "Track fit failed: " << e.what();
          continue;
        }
 
        // propagate the parameters to the MID
        TrackParam paramAtMID(track.last());
        if (!TrackExtrap::extrapToMID(paramAtMID)) {
          LOG(error) << "propagation to MID failed --> track discarded";
          continue;
        }
 
        // write the refitted track and attached clusters (same as those of the input track)
        const auto& param = track.first();
        tracksOut.emplace_back(param.getZ(), param.getParameters(), param.getCovariances(),
                               param.getTrackChi2(), clustersOut.size(), track.getNClusters(),
                               paramAtMID.getZ(), paramAtMID.getParameters(), paramAtMID.getCovariances(),
                               mchTrack.getTimeMUS());
        clustersOut.insert(clustersOut.end(), trackClusters.begin(), trackClusters.end());
      }
 
      // write the current ROF with references to the associated tracks
      rofsOut.emplace_back(rof.getBCData(), trackOffset, tracksOut.size() - trackOffset, rof.getBCWidth());
    }
  }
 
 private:
  TrackFitter mTrackFitter{}; 
};
 
//_________________________________________________________________________________________________
o2::framework::DataProcessorSpec getTrackFitterSpec(const char* specName)
{
  return DataProcessorSpec{
    specName,
    Inputs{InputSpec{"rofsin", "MCH", "TRACKROFSIN", 0, Lifetime::Timeframe},
           InputSpec{"tracksin", "MCH", "TRACKSIN", 0, Lifetime::Timeframe},
           InputSpec{"clustersin", "MCH", "TRACKCLUSTERSIN", 0, Lifetime::Timeframe}},
    Outputs{OutputSpec{OutputLabel{"rofsout"}, "MCH", "TRACKROFS", 0, Lifetime::Timeframe},
            OutputSpec{OutputLabel{"tracksout"}, "MCH", "TRACKS", 0, Lifetime::Timeframe},
            OutputSpec{OutputLabel{"clustersout"}, "MCH", "TRACKCLUSTERS", 0, Lifetime::Timeframe}},
    AlgorithmSpec{adaptFromTask<TrackFitterTask>()},
    Options{{"l3Current", VariantType::Float, -30000.0f, {"L3 current"}},
            {"dipoleCurrent", VariantType::Float, -6000.0f, {"Dipole current"}}}};
}
 
} // namespace mch
} // namespace o2