Gabriele Cosmo
138 lines
4.2 KiB
C++
138 lines
4.2 KiB
C++
//
|
|
// ********************************************************************
|
|
// * DISCLAIMER *
|
|
// * *
|
|
// * The following disclaimer summarizes all the specific disclaimers *
|
|
// * of contributors to this software. The specific disclaimers,which *
|
|
// * govern, are listed with their locations in: *
|
|
// * http://cern.ch/geant4/license *
|
|
// * *
|
|
// * Neither the authors of this software system, nor their employing *
|
|
// * institutes,nor the agencies providing financial support for this *
|
|
// * work make any representation or warranty, express or implied, *
|
|
// * regarding this software system or assume any liability for its *
|
|
// * use. *
|
|
// * *
|
|
// * This code implementation is the intellectual property of the *
|
|
// * GEANT4 collaboration. *
|
|
// * By copying, distributing or modifying the Program (or any work *
|
|
// * based on the Program) you indicate your acceptance of this *
|
|
// * statement, and all its terms. *
|
|
// ********************************************************************
|
|
//
|
|
|
|
#include <string>
|
|
#include <iostream>
|
|
#include <cmath>
|
|
|
|
#include "StreamerHepRepPoint.h"
|
|
|
|
using namespace std;
|
|
using namespace HEPREP;
|
|
|
|
StreamerHepRepPoint::StreamerHepRepPoint(HepRepWriter* stream, HepRepInstance* inst, double x, double y, double z)
|
|
: StreamerHepRepAttribute(stream), instance(inst), x(x), y(y), z(z) {
|
|
|
|
if (instance == NULL) {
|
|
cerr << "HepRepPoints cannot be created without a HepRepInstance." << endl;
|
|
}
|
|
|
|
stream->write(this);
|
|
}
|
|
|
|
StreamerHepRepPoint::~StreamerHepRepPoint() {
|
|
}
|
|
|
|
HepRepInstance* StreamerHepRepPoint::getInstance() {
|
|
return instance;
|
|
}
|
|
|
|
HepRepAttValue* StreamerHepRepPoint::getAttValue(string lowerCaseName) {
|
|
HepRepAttValue* value = getAttValueFromNode(lowerCaseName);
|
|
return (value != NULL) ? value : instance->getAttValue(lowerCaseName);
|
|
}
|
|
|
|
HepRepPoint* StreamerHepRepPoint::copy(HepRepInstance*) {
|
|
return NULL;
|
|
}
|
|
|
|
double StreamerHepRepPoint::getX() {
|
|
return x;
|
|
}
|
|
|
|
double StreamerHepRepPoint::getY() {
|
|
return y;
|
|
}
|
|
|
|
double StreamerHepRepPoint::getZ() {
|
|
return z;
|
|
}
|
|
|
|
vector<double>* StreamerHepRepPoint::getXYZ(vector<double>* xyz) {
|
|
(*xyz)[0] = x;
|
|
(*xyz)[1] = y;
|
|
(*xyz)[2] = z;
|
|
return xyz;
|
|
}
|
|
|
|
double StreamerHepRepPoint::getRho() {
|
|
return sqrt(x*x + y*y);
|
|
}
|
|
|
|
double StreamerHepRepPoint::getPhi() {
|
|
return atan2(y, x);
|
|
}
|
|
|
|
double StreamerHepRepPoint::getTheta() {
|
|
return atan2(getRho(), z);
|
|
}
|
|
|
|
double StreamerHepRepPoint::getR() {
|
|
double r = getRho();
|
|
return sqrt(r*r + z*z);
|
|
}
|
|
|
|
double StreamerHepRepPoint::getEta() {
|
|
double ct = cos(getTheta());
|
|
return -0.5*log((1.-ct)/(1.+ct));
|
|
}
|
|
|
|
double StreamerHepRepPoint::getX(double xVertex, double, double) {
|
|
return x - xVertex;
|
|
}
|
|
|
|
double StreamerHepRepPoint::getY(double, double yVertex, double) {
|
|
return y - yVertex;
|
|
}
|
|
|
|
double StreamerHepRepPoint::getZ(double, double, double zVertex) {
|
|
return z - zVertex;
|
|
}
|
|
|
|
double StreamerHepRepPoint::getRho(double xVertex, double yVertex, double zVertex) {
|
|
double dx = getX(xVertex, yVertex, zVertex);
|
|
double dy = getY(xVertex, yVertex, zVertex);
|
|
return sqrt(dx*dx + dy*dy);
|
|
}
|
|
|
|
double StreamerHepRepPoint::getPhi(double xVertex, double yVertex, double zVertex) {
|
|
return atan2(getY(xVertex, yVertex, zVertex), getX(xVertex, yVertex, zVertex));
|
|
}
|
|
|
|
double StreamerHepRepPoint::getTheta(double xVertex, double yVertex, double zVertex) {
|
|
return atan2(getRho(xVertex, yVertex, zVertex), getZ(xVertex, yVertex, zVertex));
|
|
}
|
|
|
|
double StreamerHepRepPoint::getR(double xVertex, double yVertex, double zVertex) {
|
|
double dr = getRho(xVertex, yVertex, zVertex);
|
|
double dz = getZ(xVertex, yVertex, zVertex);
|
|
return sqrt(dr*dr + dz*dz);
|
|
}
|
|
|
|
double StreamerHepRepPoint::getEta(double xVertex, double yVertex, double zVertex) {
|
|
double ct = cos(getTheta(xVertex, yVertex, zVertex));
|
|
return -0.5*log((1.-ct)/(1.+ct));
|
|
}
|
|
|
|
|