n_median.cc

Go to the documentation of this file.

/* FEAT
copyright 2017 William La Cava
license: GNU/GPL v3
*/
#include "n_median.h"
#include "../../../util/utils.h"
 
namespace FT{
 
    namespace Pop{
        namespace Op{
            NodeMedian::NodeMedian()
            {
                name = "median";
                otype = 'f';
                arity['z'] = 1;
                complexity = 1;
            }
 
            #ifndef USE_CUDA
            void NodeMedian::evaluate(const Data& data, State& state)
            {
                ArrayXf tmp(state.z.top().first.size());
                
                int x;
                
                for(x = 0; x < state.z.top().first.size(); x++)
                    tmp(x) = median(limited(state.z.top().first[x]));
                    
                state.z.pop();
 
                state.push<float>(tmp);
                
            }
            #else
            void NodeMedian::evaluate(const Data& data, State& state)
            {
                
                ArrayXf tmp(state.z.top().first.size());
                
                int x;
                
                for(x = 0; x < state.z.top().first.size(); x++)
                    tmp(x) = median(limited(state.z.top().first[x]));
                    
                state.z.pop();
 
                GPU_Variable(state.dev_f, tmp.data(), state.idx[otype], state.N);
 
                
            }
            #endif
 
            void NodeMedian::eval_eqn(State& state)
            {
                state.push<float>("median(" + state.zs.pop() + ")");
            }
            
            NodeMedian* NodeMedian::clone_impl() const { return new NodeMedian(*this); }
 
            NodeMedian* NodeMedian::rnd_clone_impl() const { return new NodeMedian(); } 
        }
    }
}