TwoPointCorrelation_wedges.h

Go to the documentation of this file.

/********************************************************************
 *  Copyright (C) 2010 by Federico Marulli and Alfonso Veropalumbo  *
 *  federico.marulli3@unibo.it                                      *
 *                                                                  *
 *  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.,                                      *
 *  59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.       *
 ********************************************************************/
 
#ifndef __TWOPOINTWED__
#define __TWOPOINTWED__
 
 
#include "TwoPointCorrelation2D_polar.h"
 
 
// ===================================================================================================
 
 
namespace cbl {
 
  namespace measure {
  
    namespace twopt {
 
      class TwoPointCorrelation_wedges : public TwoPointCorrelation2D_polar {
 
      protected:
 
    int m_nWedges;
 
    std::vector<std::vector<double>> m_mu_integral_limits;
    
    std::shared_ptr<data::Data> Wedges (const std::vector<double> rr, const std::vector<double> mu, const std::vector<std::vector<double> > xi, const std::vector<std::vector<double> > error_xi) override;
 
    std::shared_ptr<data::Data> data_with_extra_info (const std::vector<double> rad, const std::vector<double> wedges, const std::vector<double> error) const;
      
    void measurePoisson (const std::string dir_output_pairs = par::defaultString, const std::vector<std::string> dir_input_pairs={}, const bool count_dd=true, const bool count_rr=true, const bool count_dr=true, const bool tcount=true, const Estimator estimator=Estimator::_LandySzalay_, const double fact=0.1) override;
 
    void measureJackknife (const std::string dir_output_pairs = par::defaultString, const std::vector<std::string> dir_input_pairs={}, const std::string dir_output_resample = par::defaultString, const bool count_dd=true, const bool count_rr=true, const bool count_dr=true, const bool tcount=true, const Estimator estimator=Estimator::_LandySzalay_, const double fact=0.1) override;
 
    void measureBootstrap (const int nMocks, const std::string dir_output_pairs = par::defaultString, const std::vector<std::string> dir_input_pairs={}, const std::string dir_output_resample = par::defaultString, const bool count_dd=true, const bool count_rr=true, const bool count_dr=true, const bool tcount=true, const Estimator estimator=Estimator::_LandySzalay_, const double fact=0.1, const int seed=3213) override;
 
    std::vector<std::shared_ptr<data::Data> > XiJackknife (const std::vector<std::shared_ptr<pairs::Pair> > dd, const std::vector<std::shared_ptr<pairs::Pair> > rr) override;
 
    std::vector<std::shared_ptr<data::Data> > XiJackknife (const std::vector<std::shared_ptr<pairs::Pair> > dd, const std::vector<std::shared_ptr<pairs::Pair> > rr, const std::vector<std::shared_ptr<pairs::Pair> > dr) override;
 
    std::vector<std::shared_ptr<data::Data> > XiBootstrap (const int nMocks, const std::vector<std::shared_ptr<pairs::Pair> > dd, const std::vector<std::shared_ptr<pairs::Pair> > rr, const int seed=3213) override;
 
    std::vector<std::shared_ptr<data::Data> > XiBootstrap (const int nMocks, const std::vector<std::shared_ptr<pairs::Pair> > dd, const std::vector<std::shared_ptr<pairs::Pair> > rr, const std::vector<std::shared_ptr<pairs::Pair> > dr, const int seed=3213) override;
 
      public:
 
 
    TwoPointCorrelation_wedges () { m_twoPType = TwoPType::_wedges_; }
    
    TwoPointCorrelation_wedges (catalogue::Catalogue data, catalogue::Catalogue random, const BinType binType_rad, const double rMin, const double rMax, const int nbins_rad, const double shift_rad, const int nWedges=2, const int nbins_mu=50, const double shift_mu=0.5, const std::vector<std::vector<double>> mu_integral_limits={{0., 0.5}, {0.5, 1.}}, const CoordinateUnits angularUnits=CoordinateUnits::_radians_, std::function<double(double)> angularWeight=nullptr, const bool compute_extra_info=false, const double random_dilution_fraction=1.)
      : TwoPointCorrelation2D_polar(data, random, binType_rad, rMin, rMax, nbins_rad, shift_rad, BinType::_linear_, 0., 1., nbins_mu, shift_mu, angularUnits, angularWeight, compute_extra_info, random_dilution_fraction)
      {
        m_twoPType = TwoPType::_wedges_;
        
        if (nWedges<2)
          cbl::ErrorCBL("nWedges must be larger than 2!", "TwoPointCorrelation_wedges", "TwoPointCorrelation_wedges.h");
 
        if (int(mu_integral_limits.size())!=nWedges)
          cbl::ErrorCBL("mu_integral_limits.size()="+cbl::conv(mu_integral_limits.size(), cbl::par::fINT)+" must be equal to nWedges="+cbl::conv(nWedges, cbl::par::fINT), "TwoPointCorrelation_wedges", "TwoPointCorrelation_wedges.h");
        
        m_nWedges = nWedges;
        m_mu_integral_limits = mu_integral_limits;
      }
 
    TwoPointCorrelation_wedges (catalogue::Catalogue data, catalogue::Catalogue random, const BinType binType_rad, const double rMin, const double rMax, const double binSize_rad, const double shift_rad, const int nWedges=2, const double binSize_mu=1., const double shift_mu=0.5, const std::vector<std::vector<double>> mu_integral_limits={{0., 0.5}, {0.5, 1.}}, const CoordinateUnits angularUnits=CoordinateUnits::_radians_, std::function<double(double)> angularWeight=nullptr, const bool compute_extra_info=false, const double random_dilution_fraction=1.)
      : TwoPointCorrelation2D_polar(data, random, binType_rad, rMin, rMax, binSize_rad, shift_rad, BinType::_linear_, 0., 1., binSize_mu, shift_mu, angularUnits, angularWeight, compute_extra_info, random_dilution_fraction)
      {
        m_twoPType = TwoPType::_wedges_;
 
        if (nWedges<2)
          cbl::ErrorCBL("nWedges must be larger than 2!", "TwoPointCorrelation_wedges", "TwoPointCorrelation_wedges.h");
 
        if (int(mu_integral_limits.size())!=nWedges)
          cbl::ErrorCBL("mu_integral_limits.size()="+cbl::conv(mu_integral_limits.size(), cbl::par::fINT)+" must be equal to nWedges="+cbl::conv(nWedges, cbl::par::fINT), "TwoPointCorrelation_wedges", "TwoPointCorrelation_wedges.h");
        
        m_nWedges = nWedges;
        m_mu_integral_limits = mu_integral_limits;
      }
 
    ~TwoPointCorrelation_wedges () = default;
 
 
    
    int nWedges () const
    { return m_nWedges; }
    
    std::vector<std::vector<double>> mu_integral_limits () const
    { return m_mu_integral_limits; }
    
    std::vector<double> xx () const override;
 
    std::vector<double> xiPerpendicular () const override;
 
    std::vector<double> errorPerpendicular () const override;
 
    std::vector<double> xiParallel () const override;
 
    std::vector<double> errorParallel () const override;
 
    std::vector<double> yy () const 
      { cbl::ErrorCBL("", "yy", "TwoPointCorrelation_wedges.h"); std::vector<double> vv; return vv; }
 
    std::vector<double> xi1D () const
      { cbl::ErrorCBL("", "xi1D", "TwoPointCorrelation_wedges.h"); std::vector<double> vv; return vv; }
 
    std::vector<double> error1D () const
      { cbl::ErrorCBL("", "error1D", "TwoPointCorrelation_wedges.h"); std::vector<double> vv; return vv; }
 
    std::vector<std::vector<double> > xi2D () const 
      { cbl::ErrorCBL("", "xi2D", "TwoPointCorrelation_wedges.h"); std::vector<std::vector<double> > vv; return vv; }
 
    std::vector<std::vector<double> > error2D () const 
      { cbl::ErrorCBL("", "error2D", "TwoPointCorrelation_wedges.h"); std::vector<std::vector<double> > vv; return vv; }
 
 
    
 
    void measure (const ErrorType errorType=ErrorType::_Poisson_, const std::string dir_output_pairs=par::defaultString, const std::vector<std::string> dir_input_pairs={},  const std::string dir_output_resample=par::defaultString, const int nMocks=0, const bool count_dd=true, const bool count_rr=true, const bool count_dr=true, const bool tcount=true, const Estimator estimator=Estimator::_LandySzalay_, const double fact=0.1, const int seed=3213) override;
 
 
 
 
    void read (const std::string dir, const std::string file) override
    { (void)dir; (void)file; ErrorCBL("", "read", "TwoPointCorrelation_wedges.h", glob::ExitCode::_workInProgress_); }  
 
    void write (const std::string dir=par::defaultString, const std::string file=par::defaultString, const int rank=0) const override;
 
 
      
 
    void read_covariance (const std::string dir, const std::string file) override;
 
    void write_covariance (const std::string dir, const std::string file) const override;
 
    void compute_covariance (const std::vector<std::shared_ptr<data::Data>> xi, const bool JK) override;
 
    void compute_covariance (const std::vector<std::string> file, const bool JK) override;
 
 
      };
    }
  }
}
 
#endif