reconstruct_wbp.h

Go to the documentation of this file.

/***************************************************************************
 *
 * Authors:     Roberto Marabini roberto@cnb.csic.es
 *
 * Unidad de  Bioinformatica of Centro Nacional de Biotecnologia , CSIC
 *
 * 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
 *
 *  All comments concerning this program package may be sent to the
 *  e-mail address 'xmipp@cnb.csic.es'
 ***************************************************************************/

#include <memory>
#include "core/symmetries.h"
#include "core/xmipp_image.h"
#include "core/metadata_vec.h"
#include "reconstruction/recons.h"

class Projection;

typedef struct
{
    double x; // Projection direction (x,y,z)
    double y;
    double z;
    double count;
}
WBPInfo;

class ProgRecWbp: public ProgReconsBase
{
public:
    FileName fn_out, fn_sym, fn_sel;
    MetaDataVec SF;
    double threshold;
    int count_thr;
    int diameter;
    //int verb;
    size_t dim;
    int no_mats;
    WBPInfo * mat_g, * mat_f;
    double sampling;
    bool do_all_matrices;
    bool do_weights;
    SymList SL;
    size_t time_bar_step, time_bar_size, time_bar_done;
    std::unique_ptr<MetaDataVec::id_iterator> iter;
    Image<double> reconstructedVolume;
public:

    void readParams();

    void show();

    void defineParams();

    void run();

    virtual void showProgress();

    virtual void finishProcessing();

    void setIO(const FileName &fn_in, const FileName &fn_out);

    virtual void produceSideInfo() ;

    virtual bool getImageToProcess(size_t &objId);

    void getAnglesForImage(size_t id, double &rot, double &tilt, double &psi,
                              double &xoff, double &yoff, bool &flip, double &weight);

    void getAllMatrices(MetaData &SF) ;

    void getSampledMatrices(MetaData &SF) ;

    // Simple (i.e. unfiltered) backprojection of a single image
    void simpleBackprojection(Projection &img, MultidimArray<double> &vol,
                               int diameter) ;

    // Calculate the filter and apply it to a projection
    void filterOneImage(Projection &proj, Tabsinc &TSINC);

    // Calculate the filter for arbitrary tilt geometry in 2D and apply
    void apply2DFilterArbitraryGeometry() ;
};