Data2D.h

Go to the documentation of this file.

/********************************************************************
 *  Copyright (C) 2014 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 __DATA2D__
#define __DATA2D__
 
#include "Data.h"
 
namespace cbl {
 
  namespace data {
  
    class Data2D : public Data
    {
    protected:
 
      
      std::vector<double> m_x;
 
      std::vector<double> m_y;
 
      int m_xsize;
 
      int m_ysize;
 
 
    public:
 
      
      Data2D ()
    : Data(DataType::_2D_) {}
 
      Data2D (const std::string input_file, const int skip_nlines=0, const std::vector<int> column={1, 2}, const std::vector<int> column_data={}, const std::vector<int> column_errors={}, const std::vector<int> column_edges={})    
    : Data(DataType::_2D_)
      { read(input_file, skip_nlines, column, column_data, column_errors, column_edges); }
      
      Data2D (const std::vector<double> x, const std::vector<double> y, const std::vector<std::vector<double>> data, const std::vector<double> bin_edges_x={}, const std::vector<double> bin_edges_y={});
 
      Data2D (const std::vector<double> x, const std::vector<double> y, const std::vector<std::vector<double>> data, const std::vector<std::vector<double>> error, const std::vector<double> bin_edges_x={}, const std::vector<double> bin_edges_y={}); 
 
      Data2D (const std::vector<double> x, const std::vector<double> y, const std::vector<double> data, const std::vector<double> error, const std::vector<double> bin_edges_x={}, const std::vector<double> bin_edges_y={}); 
 
      Data2D (const std::vector<double> x, const std::vector<double> y, const std::vector<double> data, const std::vector<std::vector<double>> covariance, const std::vector<double> bin_edges_x={}, const std::vector<double> bin_edges_y={}); 
 
      ~Data2D () = default;
 
      std::shared_ptr<Data> as_factory ();
 
      
      
      int xsize () const override
      { return m_xsize; }
 
      int ysize () const override
      { return m_ysize; }
 
      double xx (const int i) const override
      { return m_x[i]; }
 
      std::vector<double> xx () const override { return m_x; }  
 
      double yy (const int i) const override
      { return m_y[i]; }
 
      std::vector<double> yy () const override
    { return m_y; }  
 
      std::vector<std::vector<double>> IndipendentVariable (const int i=-1, const int j=-1) const override
    { (void)i; (void)j;  return {m_x, m_y}; } 
 
      double data (const int i, const int j) const override
      { return m_data[j+i*m_ysize]; } 
 
      void get_data (std::vector<std::vector<double>> &data) const override;
 
      double error (const int i, const int j) const override
      { return m_error[j+i*m_ysize]; } 
 
      void get_error (std::vector<std::vector<double>> &error) const override;
 
 
      
      
      void set_xx (const std::vector<double> x) override { m_x = x; m_xsize = m_x.size();}
 
      void set_yy (const std::vector<double> y) override { m_y = y; m_ysize = m_y.size(); }
      
      void set_edges_xx (const std::vector<double> edges) override { m_edges_xx = edges; }
      
      void set_edges_yy (const std::vector<double> edges) override { m_edges_yy = edges; }
 
      
      
      
      virtual void read (const std::string input_file, const int skip_nlines=0, const std::vector<int> column={1, 2}, const std::vector<int> column_data={}, const std::vector<int> column_errors={}, const std::vector<int> column_edges={}) override;
 
      virtual void Print (const int precision=4) const override;
      
      void write (const std::string dir, const std::string file, const std::string header, const bool full, const int prec=4, const int ww=8, const int rank=0) const;
          
      void write_covariance (const std::string dir, const std::string file, const int precision=10) const override;
 
 
 
      std::shared_ptr<Data> cut (const double xmin, const double xmax, const double ymin, const double ymax) const;
 
 
    };
 
  }
}
 
#endif