genEvents.h

// 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.

/// \file genEvents.h
/// \author Sergey Gorbunov

#ifndef GENEVENTS_H
#define GENEVENTS_H

#include "GPUCommonDef.h"
#include <cmath>

namespace o2::gpu
{
class GPUChainTracking;
struct GPUParam;
class GPUTPCGMPhysicalTrackModel;
#if !defined(GPUCA_BUILD_QA) || defined(_WIN32)
class genEvents
{
 public:
  genEvents(GPUChainTracking* rec) {}
  void InitEventGenerator() {}
  int32_t GenerateEvent(const GPUParam& sectorParam, const char* filename) { return 1; }
  void FinishEventGenerator() {}

  static void RunEventGenerator(GPUChainTracking* rec, const std::string& dir) {};
};

#else

class genEvents
{
 public:
  genEvents(GPUChainTracking* rec) : mRec(rec) {}
  void InitEventGenerator();
  int32_t GenerateEvent(const GPUParam& sectorParam, const char* filename);
  void FinishEventGenerator();

  static void RunEventGenerator(GPUChainTracking* rec, const std::string& dir);

 private:
  int32_t GetSector(double GlobalPhi);
  int32_t GetDSector(double LocalPhi);
  double GetSectorAngle(int32_t iSector);
  int32_t RecalculateSector(GPUTPCGMPhysicalTrackModel& t, int32_t& iSector);
  double GetGaus(double sigma);

  TH1F* mClusterError[3][2] = {{nullptr, nullptr}, {nullptr, nullptr}, {nullptr, nullptr}};

  struct GenCluster {
    int32_t sector;
    int32_t row;
    int32_t mcID;
    float x;
    float y;
    float z;
    uint32_t id;
  };

  const double mTwoPi = 2 * M_PI;
  const double mSectorDAngle = mTwoPi / 18.;
  const double mSectorAngleOffset = mSectorDAngle / 2;

  GPUChainTracking* mRec;
};

#endif
} // namespace o2::gpu

#endif