MergeOperator.h

#pragma once
// ===========================================================================
// Copyright (c) 2015 Toolchefs Ltd. All rights reserved.
//
// Use of this software is subject to the terms of the Toolchefs license
// agreement provided at the time of installation or download, or which
// otherwise accompanies this software in either electronic or hard copy form.
// ===========================================================================
 
#include <AtomsGraph/Ports.h>
 
#include <Atoms/Globals.h>
#include <Atoms/Graph/Operator.h>
 
namespace Atoms
{
 
    class ATOMS_EXPORT MergeOperator: public Operator
    {
    public:
 
        struct InputPosePortData
        {
            AtomsGraph::PosePort* pp;
            AtomsGraph::DoublePort* wp;
            AtomsGraph::DoubleArrayPort* jwp;
        };
 
        NODE_STANDARD_MEMBERS
        
        MergeOperator();
        
        virtual ~MergeOperator();
        
        bool    compute(const AtomsGraph::ComputeData* computeData);
 
        void reset();
 
        unsigned int addPose();
 
        unsigned int getNumInputs();
 
        inline InputPosePortData& inputPosePort(size_t index) { return m_inputPosePorts[index]; }
        
    private:
        void processMetadata(AtomsCore::MapMetadata &outMetadata, std::vector<AtomsCore::MapMetadata*> &inMetadatas, const std::vector<double> &inputWeightVector);
 
        unsigned int counter;
 
        std::vector<InputPosePortData> m_inputPosePorts;
 
        std::vector<AtomsCore::Pose*> m_posePorts;
        std::vector<double> m_weightPorts;
        std::vector<AtomsGraph::DoubleArrayPort*> m_jointWeightsPorts;
    };
}