Skeleton.hpp

Go to the documentation of this file.

//
//  Skeleton.hpp
//  AI Watch A1
//
//  Created by Denny Caruso on 17/07/22.
//
 
// License: Apache 2.0. See LICENSE file in root directory.
// Copyright(c) 2022. All Rights Reserved.
 
#ifndef Skeleton_h
#define Skeleton_h
 
#include "FacadeSingleton/FacadeSingleton.hpp"
 
 
 
enum BodyKeypoint { Nose = 0, Neck = 1, RShoulder = 2, RElbow = 3, RWrist = 4,
                    LShoulder = 5, LElbow = 6, LWrist = 7, MidHip = 8, RHip = 9,
                    RKnee = 10, RAnkle = 11, LHip = 12, LKnee = 13, LAnkle = 14,
                    REye = 15, LEye = 16, REar = 17, LEar = 18, LBigToe = 19,
                    LSmallToe = 20, LHeel = 21, RBigToe = 22, RSmallToe = 23, RHeel = 24,
                    Background = 25
                };
 
 
 
class Skeleton {
private:
    cv::Mat                     rgb_Image;
    cv::Mat                     distance_Image;
    cv::Mat                     skeleton_Image; // remove
    std::vector <BodyKeyPoint>  bodyKeyPoints;
    std::vector <bool>          bodyKeyPointsMap;
    Json::Value                 skeletonData;
    std::vector <Point3D *>     skeletonPoints3D_RS;
    std::vector <Point3D *> *   skeletonPoints3D;
    float consistency;
    
 
 
    void setRGB_Image               (cv::Mat & rgbImage);
    void setDistance_Image          (cv::Mat & distanceImage);
    void setSkeleton_Image          (cv::Mat & skeleton_Image);
    void setBodyKeyPoints           (std::vector <BodyKeyPoint> & bodyKeyPoints);
    void setBodyKeyPointsMap        (std::vector <bool> & bodyKeyPointsMap);
    void setSkeletonData            (Json::Value skeletonData);
    void setSkeletonPoints3D_RS     (std::vector <Point3D *> skeletonPoints3D_RS);
    void setSkeletonPoints3D        (std::vector <Point3D *> * skeletonPoints3D);
    void setConsistency             (float consistency);
    
 
 
    cv::Mat getRGB_Image                                    (void);
    cv::Mat getDistance_Image                               (void);
    cv::Mat getSkeleton_Image                               (void);
    std::vector <BodyKeyPoint> getBodyKeyPoints             (void);
    Json::Value getSkeletonData                             (void);
    
 
 
    void calcBodyKeypoints          (void);
    void calcBodyEdges              (void);
    void drawLine                   (unsigned char start, unsigned char end);
    void drawCircle                 (cv::Point center);
    void deprojectSkeletonPoints3D  (void);
    void writeCoordinates           (void);
public:
    Skeleton                                        (cv::Mat & rgbImage, cv::Mat & distanceImage, cv::Mat & skeletonImage, Json::Value skeletonData);
    ~Skeleton                                       (void);
    void generateSkeleton                           (void);
    std::vector <bool> getBodyKeyPointsMap          (void);
    std::vector <Point3D *> getSkeletonPoints3D_RS  (void);
    std::vector <Point3D *> * getSkeletonPoints3D   (void);
    float getConsistency                            (void);
};
 
#endif /* Skeleton_h */