ProjectiveCylinder.cpp

Go to the documentation of this file.

/*
 * ProjectiveCylinder.cpp
 *
 *  Created on: Oct 31, 2013
 *      Author: Christian Grefe, CERN
 */

#include "DDSegmentation/ProjectiveCylinder.h"
#include "DDSegmentation/SegmentationUtil.h"

#define _USE_MATH_DEFINES
#include <cmath>

namespace DD4hep {
namespace DDSegmentation {

using Util::thetaFromXYZ;
using Util::phiFromXYZ;

ProjectiveCylinder::ProjectiveCylinder(const std::string& cellEncoding) :
        CylindricalSegmentation(cellEncoding) {
        // define type and description
        _type = "ProjectiveCylinder";
        _description = "Projective segmentation in the global coordinates";

        // register all necessary parameters
        registerParameter("theta_bins", "Number of bins theta", _thetaBins, 1);
        registerParameter("phi_bins", "Number of bins phi", _phiBins, 1);
        registerParameter("offset_theta", "Angular offset in theta", _offsetTheta, 0., SegmentationParameter::AngleUnit, true);
        registerParameter("offset_phi", "Angular offset in phi", _offsetPhi, 0., SegmentationParameter::AngleUnit, true);
        registerIdentifier("identifier_theta", "Cell ID identifier for theta", _thetaID, "theta");
        registerIdentifier("identifier_phi", "Cell ID identifier for phi", _phiID, "phi");
}


ProjectiveCylinder::ProjectiveCylinder(BitField64* decode) :    CylindricalSegmentation(decode) {
        // define type and description
        _type = "ProjectiveCylinder";
        _description = "Projective segmentation in the global coordinates";

        // register all necessary parameters
        registerParameter("theta_bins", "Number of bins theta", _thetaBins, 1);
        registerParameter("phi_bins", "Number of bins phi", _phiBins, 1);
        registerParameter("offset_theta", "Angular offset in theta", _offsetTheta, 0., SegmentationParameter::AngleUnit, true);
        registerParameter("offset_phi", "Angular offset in phi", _offsetPhi, 0., SegmentationParameter::AngleUnit, true);
        registerIdentifier("identifier_theta", "Cell ID identifier for theta", _thetaID, "theta");
        registerIdentifier("identifier_phi", "Cell ID identifier for phi", _phiID, "phi");
}

ProjectiveCylinder::~ProjectiveCylinder() {

}

Vector3D ProjectiveCylinder::position(const CellID& cID) const {
        _decoder->setValue(cID);
        return Util::positionFromRThetaPhi(1.0, theta(), phi());
}

CellID ProjectiveCylinder::cellID(const Vector3D& /* localPosition */, const Vector3D& globalPosition, const VolumeID& vID) const {
        _decoder->setValue(vID);
        double lTheta = thetaFromXYZ(globalPosition);
        double lPhi = phiFromXYZ(globalPosition);
        (*_decoder)[_thetaID] = positionToBin(lTheta, M_PI / (double) _thetaBins, _offsetTheta);
        (*_decoder)[_phiID] = positionToBin(lPhi, 2 * M_PI / (double) _phiBins, _offsetPhi);
        return _decoder->getValue();
}

double ProjectiveCylinder::theta() const {
        CellID thetaIndex = (*_decoder)[_thetaID].value();
        return M_PI * ((double) thetaIndex + 0.5) / (double) _thetaBins;
}
double ProjectiveCylinder::phi() const {
        CellID phiIndex = (*_decoder)[_phiID].value();
        return 2. * M_PI * ((double) phiIndex + 0.5) / (double) _phiBins;
}

double ProjectiveCylinder::theta(const CellID& cID) const {
        _decoder->setValue(cID);
        CellID thetaIndex = (*_decoder)[_thetaID].value();
        return M_PI * ((double) thetaIndex + 0.5) / (double) _thetaBins;
}
double ProjectiveCylinder::phi(const CellID& cID) const {
        _decoder->setValue(cID);
        CellID phiIndex = (*_decoder)[_phiID].value();
        return 2. * M_PI * ((double) phiIndex + 0.5) / (double) _phiBins;
}

REGISTER_SEGMENTATION(ProjectiveCylinder)

} /* namespace DDSegmentation */
} /* namespace DD4hep */