OutputManager.hpp

Go to the documentation of this file.

//
//  OutputManager.hpp
//  AI Watch A1
//
//  Created by Denny Caruso on 23/07/22.
//
 
// License: Apache 2.0. See LICENSE file in root directory.
// Copyright(c) 2022. All Rights Reserved.
 
#ifndef OutputManager_hpp
#define OutputManager_hpp
 
#include <iostream>
#include <sstream>
#include <fstream>
#include <string>
#include <vector>
#include <json/json.h>
#include <json/value.h>
#include <opencv2/core.hpp>
#include "../../Points/Point3D.hpp"
#include "../../Points/BodyKeyPoint.hpp"
#include "../../Skeleton.hpp"
 
 
 
class OutputManager {
private:
    std::string stringOutputData;
protected:
    void setStringOutputData                    (std::string stringOutputData);
    std::string getStringOutputData             (void);
    virtual Json::Value makeOutputString        (std::vector <Point3D *> skeletonPoints3D, std::vector <bool> bodyKeyPointsMap,
                                                 unsigned int frameID, unsigned int personID) = 0;
public:
    virtual ~OutputManager() = default;
};
 
 
 
class OutputManagerJSON : public OutputManager {    
protected:
    Json::Value makeOutputString                 (std::vector <Point3D *> skeletonPoints3D, std::vector <bool> bodyKeyPointsMap,
                                                  unsigned int frameID, unsigned int personID) override;
public:
    bool loadJSON                                (std::string filePathJSON, Json::Value & currentJSON);
    void saveJSON                                (std::string filePath);
    
 
 
    Json::Value getValueAt                       (std::string key, Json::Value currentJSON);
    Json::Value getValueAt                       (unsigned int i, Json::Value currentJSON);
    Json::Value getValueAt                       (std::string key, unsigned int i, Json::Value currentJSON);
    void createJSON                              (Json::Value & people, cv::Mat & colorImage, cv::Mat & distanceImage, cv::Mat & skeletonOnlyImage,
                                                  unsigned int nFrame, const char * outputFolder, const float skeletonThreshold);
};
 
#endif /* OutputManager_hpp */