ThermalModelVDWCanonicalStrangeness.h

Go to the documentation of this file.

/*
 * Thermal-FIST package
 * 
 * Copyright (c) 2018-2019 Volodymyr Vovchenko
 *
 * GNU General Public License (GPLv3 or later)
 */
#ifndef THERMALMODELVDWCANONICALSTRANGENESS_H
#define THERMALMODELVDWCANONICALSTRANGENESS_H

#include <map>
#include <iostream>

#include "HRGBase/ThermalModelCanonicalStrangeness.h"
#include "HRGVDW/ThermalModelVDW.h"

namespace thermalfist {

  class ThermalModelVDWCanonicalStrangeness : public ThermalModelCanonicalStrangeness
  {
  public:
    ThermalModelVDWCanonicalStrangeness(ThermalParticleSystem *TPS, const ThermalModelParameters& params = ThermalModelParameters());

    virtual ~ThermalModelVDWCanonicalStrangeness(void);
    
    void FillVirialEV(const std::vector< std::vector<double> > & bij = std::vector< std::vector<double> >(0));

    // Override functions begin

    void FillVirial(const std::vector<double> & ri = std::vector<double>(0));

    void FillAttraction(const std::vector< std::vector<double> > & aij = std::vector< std::vector<double> >(0));
    
    void SetVirial(int i, int j, double b) { if (i >= 0 && i < static_cast<int>(m_Virial.size()) && j >= 0 && j < static_cast<int>(m_Virial[i].size())) m_Virial[i][j] = b; }
    
    void SetAttraction(int i, int j, double a) { if (i >= 0 && i < static_cast<int>(m_Attr.size()) && j >= 0 && j < static_cast<int>(m_Attr[i].size()))     m_Attr[i][j] = a; }

    double VirialCoefficient(int i, int j) const;

    double AttractionCoefficient(int i, int j) const;

    virtual void ReadInteractionParameters(const std::string &filename);

    virtual void WriteInteractionParameters(const std::string &filename);

    virtual void CalculateDensitiesGCE();

    virtual void CalculateEnergyDensitiesGCE();

    virtual void CalculatePressuresGCE();

    virtual void CalculatePrimordialDensities();

    virtual double CalculateEnergyDensity();

    virtual double CalculateEntropyDensity();

    virtual double CalculatePressure();

    virtual bool IsConservedChargeCanonical(ConservedCharge::Name charge) const;
    
    // Override functions end

  protected:
    void PrepareModelVDW();
    
    void ClearModelVDW();       
  
    virtual void CalculateSums(const std::vector<double> &  Vcs);

    virtual double MuShift(int id) const;

    ThermalModelVDWFull *m_modelVDW; 
    std::vector< std::vector<double> > m_Virial; 
    std::vector< std::vector<double> > m_Attr;   
    double m_PNS;                             
    std::vector<double> m_MuStar;         
    std::vector<double> m_Suppression;      
  };

} // namespace thermalfist

#endif // THERMALMODELEVCANONICALSTRANGENESS_H