uuc3dlib/subjects.hpp

#ifndef SUBJ83SNKRIWON423KHTJEIYU589
#define SUBJ83SNKRIWON423KHTJEIYU589

// vim:set tabstop=4 cindent:

// Author: Geert-Jan Giezeman <geert@cs.uu.nl>
// Copyright: Utrecht University, 2011

#include "uuc3d.hpp"
#include "skeleton_info.hpp"

namespace UuIcsC3d
{
        struct LabeledSubject {
                LabeledSubject(std::string const &name, MarkerNames const &label_info);
                std::string name() const {return m_name;}
                unsigned label_count() const { return m_label_count;}
                std::vector<int> const &point_index() const { return m_point_index;}
                void set_name(std::string const &name) {m_name=name;}
                void set_index(int el, int val) {m_point_index[el]=val;}
        private:
                std::string m_name; // The prefix that is used in the labelling
                unsigned m_label_count;
                std::vector<int> m_point_index; 
        };
        std::vector<unsigned> fill_subjects(std::vector<LabeledSubject> &subjects,
                                        std::vector<SpacePaddedString> const &labels,
                                        MarkerNames const &marker_names);
        struct LabeledSkeleton {
                LabeledSkeleton(std::string const &name, SkeletonInfo const &skel_info);
                std::string name() const {return m_name;}
                unsigned label_count() const { return m_label_count;}
                std::vector< std::vector<int> > const &point_index() const { return m_point_index;}
                void set_name(std::string const &name) {m_name=name;}
                void set_index(int gr, int el, int val) {m_point_index[gr][el]=val;}
        private:
                std::string m_name; // The prefix that is used in the labelling
                unsigned m_label_count;
                std::vector< std::vector<int> > m_point_index; 
        };

        std::vector<unsigned> fill_skeletons(std::vector<LabeledSkeleton> &subjects,
                                        std::vector<SpacePaddedString> const &labels,
                                        SkeletonInfo const &skel_info);

        
}

#endif // SUBJ83SNKRIWON423KHTJEIYU589