FourMomenta.h

Go to the documentation of this file.

/*  YAP - Yet another PWA toolkit
    Copyright 2015, Technische Universitaet Muenchen,
    Authors: Daniel Greenwald, Johannes Rauch

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */


#ifndef yap_FourMomenta_
#define yap_FourMomenta_

#include "fwd/CachedValue.h"
#include "fwd/DataPoint.h"
#include "fwd/FinalStateParticle.h"
#include "fwd/FourVector.h"
#include "fwd/MassAxes.h"
#include "fwd/Model.h"
#include "fwd/ParticleCombination.h"
#include "fwd/StatusManager.h"

#include "StaticDataAccessor.h"

#include <cmath>
#include <iostream>
#include <memory>
#include <vector>

namespace yap {

class FourMomenta : public StaticDataAccessor
{
public:

    FourMomenta(Model& m);

    bool consistent() const;

    virtual void calculate(DataPoint& d, StatusManager& sm) const override;


    FourVector<double> p(const DataPoint& d, const std::shared_ptr<ParticleCombination>& pc) const;

    double m2(const DataPoint& d, const std::shared_ptr<ParticleCombination>& pc) const
    { return pow(m(d, pc), 2); }

    double m(const DataPoint& d, const std::shared_ptr<ParticleCombination>& pc) const;

    const FourVector<double> totalMomentum(const DataPoint& d) const;

    const std::vector<FourVector<double> > finalStateMomenta(const DataPoint& d) const;

    std::shared_ptr<RealCachedValue> mass()
    { return M_; }

    std::shared_ptr<RealCachedValue> mass() const
    { return M_; }

    std::shared_ptr<FourVectorCachedValue> momentum()
    { return P_; }

    std::shared_ptr<FourVectorCachedValue> momentum() const
    { return P_; }


    std::ostream& printMasses(const DataPoint& d, std::ostream& os = std::cout) const;

    friend class Model;

    friend class DataAccessor;

protected:

    void setFinalStateMomenta(DataPoint& d, const std::vector<FourVector<double> >& P, StatusManager& sm) const;

    void addParticleCombination(std::shared_ptr<ParticleCombination> pc) override;

    void pruneSymmetrizationIndices() override
    {}

private:

    int TotalIndex_;

    std::vector<int> FSPIndices_;

    std::shared_ptr<FourVectorCachedValue> P_;

    std::shared_ptr<RealCachedValue> M_;

};

std::vector<FourVector<double> > calculate_four_momenta(double initial_mass, const FinalStateParticleVector& FPSs,
                                                        const MassAxes& axes, const std::vector<double>& squared_masses);

std::vector<FourVector<double> > calculate_four_momenta(double initial_mass, const Model& M,
                                                        const MassAxes& axes, const std::vector<double>& squared_masses);

}
#endif