TYTrajet.h

Go to the documentation of this file.

/*
 * Copyright (C) <2012> <EDF-R&D> <FRANCE>
 * 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 2 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, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
 
#ifndef __TY_TRAJET__
#define __TY_TRAJET__
 
#include "TYChemin.h"
#include "Tympan/models/common/3d.h"
#include "Tympan/models/common/atmospheric_conditions.h"
#include "Tympan/models/solver/entities.hpp"
#include "Tympan/models/common/acoustic_path.h"
#include <gtest/gtest_prod.h>
 
class TYTrajet
{
 
    // Methods
public:
    TYTrajet(tympan::AcousticSource& asrc_, tympan::AcousticReceptor& arcpt_);
 
    TYTrajet(const TYTrajet& other);
    virtual ~TYTrajet();
 
    TYTrajet& operator=(const TYTrajet& other);
    bool operator==(const TYTrajet& other) const;
    bool operator!=(const TYTrajet& other) const;
 
    void reset();
 
    double getDistance()
    {
        return _distance;
    }
    const double getDistance() const
    {
        return _distance;
    }
 
    void setDistance(const double& distance)
    {
        _distance = distance;
    }
 
    size_t getNbChemins()
    {
        return _chemins.size();
    }
 
    TYTabChemin& getChemins()
    {
        return _chemins;
    }
 
    TYTabChemin& getCheminsDirect()
    {
        return _cheminsDirect;
    }
 
    void addChemin(const TYChemin& chemin);
 
    void addCheminDirect(const TYChemin& chemin);
 
    TYChemin getChemin(int index)
    {
        return _chemins.at(index);
    }
 
    OSpectre getPNoOp();
 
    void setPtSetPtR(const OPoint3D& pt1, const OPoint3D& pt2)
    {
        _ptS = pt1;
        _ptR = pt2;
    }
    void getPtSetPtR(OPoint3D& pt1, OPoint3D& pt2)
    {
        pt1 = _ptS;
        pt2 = _ptR;
    }
 
    void setPtSetPtRfromOSeg3D(const OSegment3D& seg)
    {
        _ptS = seg._ptA;
        _ptR = seg._ptB;
    }
    void getPtSetPtRfromOSeg3D(OSegment3D& seg) const
    {
        seg._ptA = _ptS, seg._ptB = _ptR;
    }
 
    // TYSpectre& getSpectre() { return _sLP; }
    // const TYSpectre getSpectre() const { return _sLP; }
    // void setSpectre(const TYSpectre& spectre) { _sLP = spectre; }
    OSpectre& getSpectre()
    {
        return _sLP;
    }
    const OSpectre getSpectre() const
    {
        return _sLP;
    }
    void setSpectre(const OSpectre& spectre)
    {
        _sLP = spectre;
    }
 
    OSpectre getPEnergetique(const AtmosphericConditions& atmos);
 
    OSpectre getPInterference(const AtmosphericConditions& atmos);
 
    // Get the tab of rays
    std::vector<acoustic_path*>& get_tab_rays();
 
private:
    OSpectre correctTiers(const OSpectreComplex& si, const OSpectreComplex& sj,
                          const AtmosphericConditions& atmos, const double& ri, const double& rj) const;
    void build_tab_rays();
    FRIEND_TEST(test_TYTrajet, getPInterference);
    FRIEND_TEST(test_TYTrajet, getPEnergetique);
 
public:
    tympan::AcousticSource& asrc;
    tympan::source_idx asrc_idx;
 
    tympan::AcousticReceptor& arcpt;
    tympan::receptor_idx arcpt_idx;
 
    // Members
protected:
    OPoint3D _ptS;
 
    OPoint3D _ptR;
 
    TYTabChemin _chemins;
 
    TYTabChemin _cheminsDirect;
 
    double _distance;
 
    // TYSpectre _sLP;
    OSpectre _sLP;
 
    std::vector<acoustic_path*> _tabRays;
};
#endif // __TY_TRAJET__