cbf_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 19, 2011
//     Author: Eugen Wintersberger <eugen.wintersberger@desy.de>
//
//
#pragma once

#include<iostream>
#include<fstream>
#include<sstream>
#include<cstdio>
#include<cstdlib>
#include<vector>

#include<boost/regex.hpp>

#include <pni/core/types.hpp>

#include "../image_reader.hpp"
#include "dectris_reader.hpp"
#include "types.hpp"

namespace pni{
namespace io{


    class cbf_reader: public image_reader 
    {
        private:
            cbf::vendor_id _detector_vendor;  
            std::vector<image_info> _image_info;  
            std::streampos _data_offset;         
            cbf::compression_id _compression_type; 

            //-----------------------------------------------------------------
            void _parse_file();
            
        public:
            //=================constructors and destructor==================
            cbf_reader();

            //-----------------------------------------------------------------
            cbf_reader(const pni::core::string &fname);

            //-----------------------------------------------------------------
            virtual ~cbf_reader();

            //-----------------------------------------------------------------
            cbf_reader(const cbf_reader &r) = delete;

            //====================assignment operators=======================
            cbf_reader &operator=(cbf_reader &&r);

            cbf_reader &operator=(const cbf_reader &r) = delete;

            //----------------------------------------------------------------
            virtual void close()
            {
                //close the stream
                data_reader::close();
                //reset data offset
                _data_offset = 0;
                //clear the _image_info vector
                _image_info.clear();
            }

            //-----------------------------------------------------------------
            virtual void open()
            {
                close();
                data_reader::open();
                _parse_file();
            }
           
            //-----------------------------------------------------------------
            virtual size_t nimages() const 
            { 
                return _image_info.size(); 
            }
           
            //-----------------------------------------------------------------
            virtual image_info info(size_t i) const 
            {
                return _image_info[i];
            }

            //-----------------------------------------------------------------
            template<typename CTYPE> CTYPE image(size_t i,size_t c=0);
           
            //-----------------------------------------------------------------
            template<typename CTYPE> 
            void image(CTYPE &array,size_t i,size_t c=0);

    };

    //-------------------------------------------------------------------------
    template<typename CTYPE> CTYPE cbf_reader::image(size_t i,size_t c) 
    {
        using namespace pni::core;
        image_info info = _image_info[i];
        CTYPE data;
        try
        {
            data=CTYPE(info.npixels());
        }
        catch(...)
        {
            throw memory_allocation_error(EXCEPTION_RECORD,
                    "Allocation of container for image data failed!");
        }

        try
        {
            image(data,i,c);
        }
        catch(file_error &error)
        {
            //propagate exception
            error.append(EXCEPTION_RECORD);
            throw error;
        }

        return data;
    }

    //-------------------------------------------------------------------------
    template<typename CTYPE> 
        void cbf_reader::image(CTYPE &data,size_t i,size_t c) 
    {
        using namespace pni::core;
        //load image information and throw exception if image and container size
        //to not match
        image_info inf = _image_info[i];
        if(data.size()!= inf.npixels())
        {
            std::stringstream ss;
            ss<<"Container size ("<<data.size()<<") does not match image ";
            ss<<"size ("<<inf.npixels()<<")!";
            throw size_mismatch_error(EXCEPTION_RECORD,ss.str());
        }

        //load the channel information
        image_channel_info channel = inf.get_channel(c);

        if(_detector_vendor == cbf::vendor_id::DECTRIS)
        {
            if(channel.type_id() == type_id_t::INT16)
                //read 16Bit signed data
                cbf::dectris_reader::read_data_byte_offset<int16>(
                        _get_stream(),inf,data);
            if(channel.type_id() == type_id_t::INT32)
                //read 32Bit signed data
                cbf::dectris_reader::read_data_byte_offset<int32>(
                    _get_stream(),inf,data);
            else
            {
                file_error error(EXCEPTION_RECORD,
                        "No data reader for this data type!");
                throw error;
            }

        }
        else
            throw file_error(EXCEPTION_RECORD,"Unknown detector vendor!");

    }
//end of namespace
}
}