metadata_split_3D.cpp File Reference

#include <algorithm>
#include "metadata_split_3D.h"
#include "core/geometry.h"
#include "core/metadata_vec.h"
#include "data/filters.h"
#include "data/basic_pca.h"

Include dependency graph for metadata_split_3D.cpp:

Go to the source code of this file.

Functions
void	getNeighbours (MetaDataDb &mdIn, const Matrix1D< double > &projectionDir, MetaDataDb &mdNeighbours, double maxDist)
void	analyzeNeighbours (MetaData &mdNeighbours, const FileName &fnRef)

Function Documentation

analyzeNeighbours()

void analyzeNeighbours	(	MetaData &	mdNeighbours,
		const FileName &	fnRef
	)

Definition at line 108 of file metadata_split_3D.cpp.

{
    std::vector<double> cc;
    mdNeighbours.getColumnValues(MDL_MAXCC,cc);
    std::sort(cc.begin(),cc.end());

    double ccMedian=cc[cc.size()/2];
    for (size_t objId : mdNeighbours.ids())
    {
        double cci;
        mdNeighbours.getValue(MDL_MAXCC, cci, objId);
        if (cci>ccMedian)
            mdNeighbours.setValue(MDL_COST, 1.0, objId);
        else
            mdNeighbours.setValue(MDL_COST, -1.0, objId);
    }
}

getNeighbours()

void getNeighbours	(	MetaDataDb &	mdIn,
		const Matrix1D< double > &	projectionDir,
		MetaDataDb &	mdNeighbours,
		double	maxDist
	)

Definition at line 73 of file metadata_split_3D.cpp.

{
    Matrix1D<double> projectionDir2;
    FileName fnImg;
    mdNeighbours.clear();
    MetaDataDb mdAux;
    size_t refno;
    double cc;
    for (size_t objId : mdIn.ids())
    {
        double rot, tilt;
        mdIn.getValue(MDL_ANGLE_ROT,rot, objId);
        mdIn.getValue(MDL_ANGLE_TILT,tilt, objId);
        mdIn.getValue(MDL_IMAGE_IDX,refno, objId);
        mdIn.getValue(MDL_MAXCC,cc, objId);
        Euler_direction(rot,tilt,0,projectionDir2);

        double angle=acos(dotProduct(projectionDir,projectionDir2));
        if (angle<maxDist)
        {
            mdIn.getValue(MDL_IMAGE,fnImg, objId);
            size_t id=mdAux.addObject();
            mdAux.setValue(MDL_IMAGE,fnImg,id);
            mdAux.setValue(MDL_IMAGE_IDX,refno,id);
            mdAux.setValue(MDL_MAXCC,cc,id);
        }
    }
    mdNeighbours.removeDuplicates(mdAux,MDL_IMAGE_IDX);
    std::vector<MDLabel> groupBy;
    groupBy.push_back(MDL_IMAGE_IDX);
    groupBy.push_back(MDL_IMAGE);
    if (mdAux.size()>0)
        mdNeighbours.aggregateGroupBy(mdAux,AGGR_MAX,groupBy,MDL_MAXCC,MDL_MAXCC);
}