spreadsheet_reader.hpp

Go to the documentation of this file.

//
// (c) Copyright 2011 DESY, Eugen Wintersberger <eugen.wintersberger@desy.de>
//
// This file is part of libpniio.
//
// libpniio 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.
//
// libpniio 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 libpniio.  If not, see <http://www.gnu.org/licenses/>.
// ===========================================================================
//
// Created on: Apr 30, 2012
//     Author: Eugen Wintersberger <eugen.wintersberger@desy.de>
//
//
#pragma once

#include <vector>
#include <boost/current_function.hpp>
#include <pni/core/types.hpp>
#include <pni/core/error.hpp>
#include "data_reader.hpp"
#include "column_info.hpp"

namespace pni{
namespace io{

    class spreadsheet_reader:public data_reader
    {
        private:
            std::vector<column_info> _columns_info; 
            size_t _nrec;    

        protected:
            //============constructors and destructor==========================
            spreadsheet_reader();

            spreadsheet_reader(const spreadsheet_reader &) = delete;

            spreadsheet_reader(spreadsheet_reader &&o);

            spreadsheet_reader(const pni::core::string &n);

            //==================protected member functions======================
            virtual void _append_column(const column_info &i)
            {
                _columns_info.push_back(i);
            }

            //-----------------------------------------------------------------
            virtual column_info _get_column(size_t i) const
            {
                return _columns_info.at(i);
            }

            //-----------------------------------------------------------------
            virtual column_info _get_column(const pni::core::string &n) const
            {
                using namespace pni::core;
                size_t i=0;
                for(auto c: _columns_info)
                {
                    if(c.name() == n)
                        return _get_column(i); 

                    //increment column counter
                    i++;
                }
               
                throw key_error(EXCEPTION_RECORD,"Column ["+n+"] not found!");

                return column_info(); //just to get rid of compiler warning
            }

            //-----------------------------------------------------------------
            void _nrecords(size_t n) { _nrec = n; }


        public:
            //========================public type==============================
            typedef std::vector<column_info>::iterator iterator;
            typedef std::vector<column_info>::const_iterator const_iterator; 
            //=======================destructor================================
            virtual ~spreadsheet_reader();

            //===================assignment operators==========================
            spreadsheet_reader &operator=(const spreadsheet_reader &) = delete;

            //-----------------------------------------------------------------
            spreadsheet_reader &operator=(spreadsheet_reader &&r);

            //===============public memeber methods============================
            size_t ncolumns() const
            {
                return _columns_info.size();
            }

            //-----------------------------------------------------------------
            size_t nrecords() const { return _nrec; }

            //-----------------------------------------------------------------
            iterator begin() { return _columns_info.begin(); }

            //-----------------------------------------------------------------
            const_iterator begin() const { return _columns_info.begin(); }

            //-----------------------------------------------------------------
            iterator end() { return _columns_info.end(); }

            //-----------------------------------------------------------------
            const_iterator end() const { return _columns_info.end(); }

            //-----------------------------------------------------------------
            bool has_column(const pni::core::string &name) const;

            //-----------------------------------------------------------------
            size_t column_index(const pni::core::string &name) const;

    };


//end of namespace
}
}