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__