ph_scattering_matrix.h

#ifndef PH_SCATTERING_MATRIX_H
#define PH_SCATTERING_MATRIX_H
 
#include "interaction_3ph.h"
#include "phonon_h0.h"
#include "base_ph_scattering_matrix.h"
#include "vector_bte.h"
 
class PhScatteringMatrix : virtual public BasePhScatteringMatrix {
 public:
  PhScatteringMatrix(Context &context_, StatisticsSweep &statisticsSweep_,
                     BaseBandStructure &innerBandStructure_,
                     BaseBandStructure &outerBandStructure_,
                     PhononH0 *phononH0_ = nullptr);
 
 protected:
 
  PhononH0 *phononH0;
 
  // implementation of the scattering matrix
  void builder(std::shared_ptr<VectorBTE> linewidth,
               std::vector<VectorBTE> &inPopulations,
               std::vector<VectorBTE> &outPopulations) override;
 
 // friend functions for adding scattering rates,
 // these live in ph_scattering.cpp
 // TODO write docstrings for these
 
  friend void addPhPhScattering(BasePhScatteringMatrix &matrix, Context &context,
                                std::vector<VectorBTE> &inPopulations,
                                std::vector<VectorBTE> &outPopulations,
                                std::vector<std::tuple<std::vector<int>, int>> qPairIterator,
                                Eigen::MatrixXd &innerBose, Eigen::MatrixXd &outerBose,
                                BaseBandStructure &innerBandStructure,
                                BaseBandStructure &outerBandStructure,
                                PhononH0& phononH0,
                                Interaction3Ph& coupling3Ph,
                                std::shared_ptr<VectorBTE> linewidth);
 
  friend void addIsotopeScattering(BasePhScatteringMatrix &matrix, Context &context,
                                std::vector<VectorBTE> &inPopulations,
                                std::vector<VectorBTE> &outPopulations, 
                                std::vector<std::tuple<std::vector<int>, int>> qPairIterator,
                                Eigen::MatrixXd &innerBose, Eigen::MatrixXd &outerBose,
                                BaseBandStructure &innerBandStructure,
                                BaseBandStructure &outerBandStructure,
                                std::shared_ptr<VectorBTE> linewidth);
 
  friend void addPhElScattering(BasePhScatteringMatrix &matrix, Context &context,
                                std::vector<VectorBTE> &inPopulations,
                                std::vector<VectorBTE> &outPopulations,
                                BaseBandStructure &phBandStructure,
                                BaseBandStructure &elBandStructure,
                                StatisticsSweep& statisticsSweep, 
                                InteractionElPhWan &couplingElPhWan,
                                std::shared_ptr<VectorBTE> linewidth);
 
};
 
#endif