Data1D_extra.h

Go to the documentation of this file.

/********************************************************************
 *  Copyright (C) 2014 by Federico Marulli                          *
 *  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 __DATA1D_EXTRA__
#define __DATA1D_EXTRA__
 
#include "Data1D.h"
 
namespace cbl {
 
  namespace data {
    
    class Data1D_extra : public Data1D
    {
      
    protected:
 
      
      std::vector<std::vector<double>> m_extra_info;
 
 
      
    public:
 
      
      Data1D_extra ()
    : Data1D() { set_dataType(DataType::_1D_extra_); }
 
      Data1D_extra (const int n_extra_info, const std::string input_file, const int skip_nlines=0, const std::vector<int> column={1}, const std::vector<int> column_data={}, const std::vector<int> column_errors={}, const std::vector<int> column_edges={})
    : Data1D()
      { m_extra_info.resize(n_extra_info); read(input_file, skip_nlines, column, column_data, column_errors, column_edges); set_dataType(DataType::_1D_extra_); }
      
      Data1D_extra (const std::vector<double> x, const std::vector<double> data, const std::vector<std::vector<double>> extra_info, const std::vector<double> bin_edges_x={})
    : Data1D(x, data, -1, bin_edges_x), m_extra_info(extra_info)
      { set_dataType(DataType::_1D_extra_); }
 
      Data1D_extra (const std::vector<double> x, const std::vector<double> data, const std::vector<double> error, const std::vector<std::vector<double>> extra_info, const std::vector<double> bin_edges_x={})
    : Data1D(x, data, error, -1, bin_edges_x), m_extra_info(extra_info)
      { set_dataType(DataType::_1D_extra_); }
      
      Data1D_extra (const std::vector<double> x, const std::vector<double> data, const std::vector<std::vector<double>> covariance, const std::vector<std::vector<double>> extra_info, const std::vector<double> bin_edges_x={})
    : Data1D(x, data, covariance, -1, bin_edges_x), m_extra_info(extra_info)
      { set_dataType(DataType::_1D_extra_); }
 
      virtual ~Data1D_extra () = default;
 
      std::shared_ptr<Data> as_factory () { return move(std::unique_ptr<Data1D_extra>(this)); }
 
 
 
 
      double extra_info (const int i, const int j) const override { return m_extra_info[i][j]; }
 
      std::vector<std::vector<double>> extra_info () const override { return m_extra_info; }
 
      std::vector<std::vector<double>> IndipendentVariable (const int i=-1, const int j=-1) const 
      {
    (void)j;
    std::vector<std::vector<double>> iv;
    iv.push_back(((i>0) ? m_extra_info[i] : m_x));
    return iv;
      }
 
 
      
      
      void set_extra_info (const std::vector<std::vector<double>> extra_info) override { m_extra_info = extra_info; }
 
      
      
      
      virtual void read (const std::string input_file, const int skip_nlines=0, const std::vector<int> column={1}, 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;
      
      virtual void write (const std::string dir, const std::string file, const std::string header, const int prec=4, const int ww=8, const int rank=0) const override;
 
 
 
      
      std::shared_ptr<Data> cut (const double xmin, const double xmax) const override;
 
      std::shared_ptr<Data> cut (const std::vector<bool> mask) const;
 
      
    };
    
  }
}
 
#endif