NoiseCalibratorSpec.h

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.
 
 
#ifndef O2_ITS_NOISECALIBRATORSPEC
#define O2_ITS_NOISECALIBRATORSPEC
 
#include <string>
#include <TStopwatch.h>
#include "Framework/DataProcessorSpec.h"
#include "Framework/Task.h"
#include "DetectorsBase/GRPGeomHelper.h"
#include "DataFormatsDCS/DCSConfigObject.h"
 
//#define TIME_SLOT_CALIBRATION
#ifdef TIME_SLOT_CALIBRATION
#include "ITSCalibration/NoiseSlotCalibrator.h"
using CALIBRATOR = o2::its::NoiseSlotCalibrator;
#else
#include "ITSCalibration/NoiseCalibrator.h"
using CALIBRATOR = o2::its::NoiseCalibrator;
#endif
 
using namespace o2::framework;
 
namespace o2
{
 
namespace its
{
 
class NoiseCalibratorSpec : public Task
{
 public:
  enum ProcessingMode { Full,
                        Accumulate,
                        Normalize };
 
  NoiseCalibratorSpec(ProcessingMode md = ProcessingMode::Full, bool useClusters = false, std::shared_ptr<o2::base::GRPGeomRequest> req = {}) : mMode(md), mCCDBRequest(req), mUseClusters(useClusters)
  {
    mTimer.Stop();
  }
  ~NoiseCalibratorSpec() override = default;
 
  void init(InitContext& ic) final;
  void run(ProcessingContext& pc) final;
  void endOfStream(EndOfStreamContext& ec) final;
  void finaliseCCDB(ConcreteDataMatcher& matcher, void* obj) final;
  void setProcessingMode(ProcessingMode m) { mMode = m; }
  ProcessingMode getProcessingMode() const { return mMode; }
 
 private:
  void addDatabaseEntry(int chip, int row, int col);
  void sendOutput(DataAllocator& output);
  void sendAccumulatedMap(DataAllocator& output);
  void updateTimeDependentParams(ProcessingContext& pc);
  std::unique_ptr<CALIBRATOR> mCalibrator = nullptr;
  std::shared_ptr<o2::base::GRPGeomRequest> mCCDBRequest;
  unsigned int mStrobeCounter = 0;
  size_t mDataSizeStat = 0;
  size_t mNClustersProc = 0;
  int mValidityDays = 3;
  int mNPartsDone = 0; // number of accumalated parts in Normalization mode
  bool mUseClusters = false;
  bool mStopMeOnly = false; // send QuitRequest::Me instead of QuitRequest::All
  bool mRunStopRequested = false; // flag that run was stopped (ant the last output is sent)
  bool mInitOnceDone = false;     // flag that the CCDB objects which should be fetche once per run are loaded
  TStopwatch mTimer{};
  float mNoiseCutIB = -1.;
  o2::dcs::DCSconfigObject_t mNoiseMapDCS; // noisy pixels to be sent to DCS CCDB
  std::vector<int>* mConfDBmap{nullptr};
  ProcessingMode mMode = ProcessingMode::Full;
};
 
DataProcessorSpec getNoiseCalibratorSpec(bool useClusters, int pmode = 0);
 
} // namespace its
} // namespace o2
 
#endif /* O2_ITS_NOISECALIBRATORSPEC */