#ifndef COMMON_HPP
#define COMMON_HPP

#include <cmath>

template<typename A, typename B>
float displacement(const A& a, const B& b){
    return std::sqrt(pow(a.x()-b.x(), 2) +
                     pow(a.y()-b.y(), 2) +
                     pow(a.z()-b.z(), 2));
}

template<typename T>
float rho(const T& t){
    return std::sqrt(pow(t.x(),2)+pow(t.y(),2));
}

float pseudorapidity(float z, float r){
    float theta = atan2(r, z);
    return -log(tan(theta/2.0));
}

float pseudorapidity(float x, float y, float z){
    float r = sqrt(x*x + y*y);
    return pseudorapidity(z, r);
}

template<typename T>
float pseudorapidity(const T &t){
    return pseudorapidity(t.x(), t.y(), t.z());
}

float pseudorapidityP(float px, float py, float pz){
    float p_mag = sqrt(px*px + py*py + pz*pz);
    return 0.5*log((p_mag+pz)/(p_mag-pz));
}

template<typename T>
float pseudorapidityP(const T &t){
    return pseudorapidityP(t.px(), t.py(), t.pz());
}

#endif // COMMON_HPP