PsdGeometricDigitizerAlgo.h

Go to the documentation of this file.

/*
 * PsdGeometricDigitizerAlgo.h
 *
 *  Created on: 11 Jan 2019
 *      Author: Nicola Mori
 */

#ifndef HERD_PSDGEOMETRICDIGITIZERALGO_H_
#define HERD_PSDGEOMETRICDIGITIZERALGO_H_

// HerdSoftware headers
#include "dataobjects/PsdGeoParamsColl.h"
#include "dataobjects/PsdHitsColl.h"
#include "dataobjects/PsdParticleHitsColl.h"

// EventAnalysis headers
#include "algorithm/Algorithm.h"

using namespace EA;

namespace Herd {

// clang-format off
// clang-format on

class PsdGeometricDigitizerAlgo : public Algorithm {
public:
  PsdGeometricDigitizerAlgo(const std::string &name);

  bool Initialize() override;

  bool Process() override;

  std::vector<std::string> FreeObjects(const std::vector<std::string> &objs, Memory::Status memStatus) override;

private:
  // Parameters
  AxesArray<int> _topAggrFactor, _latAggrFactor;

  PsdHitsColl _aggrPsdHitsColl;                            // New aggregated hits
  PsdParticleHitsColl _aggrPsdParticleHitsColl;            // New aggregated particle hits
  observer_ptr<PsdGeoParamsColl> _origPsdGeoParamsColl;    // Original parameters
  std::shared_ptr<PsdGeoParamsColl> _aggrPsdGeoParamsColl; // New aggregated parameters

  StorePtr _evStore, _globStore;

  void _Aggregate(const std::vector<PsdGeoParams> &originalGeoParams, const PsdHits &originalHits,
                  observer_ptr<const PsdParticleHits> originalParticleHits, const AxesArray<int> &aggrFactors,
                  const std::vector<PsdGeoParams> &aggregatedGeoParams, PsdHits &aggregatedHits,
                  PsdParticleHits &aggregatedParticleHits);
  // Utility variables for calling _AggregateTiles
  SidesArray<std::array<bool, 2>> _tileCooIncreasesWithID; // true if the tile coordinate increases with the tile ID.
  SidesArray<RefFrame::Axis> _tileMainAggrAxis, _tileSecAggrAxis;

  CooArray<unsigned char> _AggregatedIndexes(const CooArray<unsigned char> &origIndexes,
                                             const AxesArray<int> &aggrFactors);
};

} // namespace Herd

#endif /* HERD_PSDGEOMETRICDIGITIZERALGO_H_ */