nv_aisle_csvparser.hpp

Go to the documentation of this file.

 
#ifndef _NV_AISLE_CSVPARSER_HPP_
#define _NV_AISLE_CSVPARSER_HPP_
 
 
#include <iostream>
#include <fstream>
#include <map>
#include <vector>
#include <string>
#include <cstdint>
 
using namespace std;
 
namespace nvaisle_csv {
typedef struct _NvAisleCsvFields
{
  uint32_t cameraId;        
  uint32_t rowId;           
  string level;             
  string cameraIDString;    
  string aisleIdStr;        
  uint32_t aisleId;         
  double dewarpTopAngle, dewarpBottomAngle, dewarpPitch, dewarpYaw, dewarpRoll;
  uint32_t numROIpoints;
  float ROI_x0, ROI_y0, ROI_x1, ROI_y1, ROI_x2, ROI_y2, ROI_x3, ROI_y3,
    ROI_x4, ROI_y4, ROI_x5, ROI_y5, ROI_x6, ROI_y6, ROI_x7, ROI_y7;
  float gx0, gy0, gx1, gy1, gx2, gy2, gx3, gy3, cx0, cy0, cx1, cy1, cx2, cy2, cx3, cy3;
 
  int entry;    
  int exit;     
  float entry_ROI_x0, entry_ROI_y0, entry_ROI_x1, entry_ROI_y1, entry_ROI_x2,
        entry_ROI_y2, entry_ROI_x3, entry_ROI_y3;
  float exit_ROI_x0, exit_ROI_y0, exit_ROI_x1, exit_ROI_y1, exit_ROI_x2,
        exit_ROI_y2, exit_ROI_x3, exit_ROI_y3;
  string sensorId;          
  float dewarpFocalLength;  
  uint32_t dewarpWidth;     
  uint32_t dewarpHeight;    
  double h0, h1, h2, h3, h4, h5, h6, h7, h8;
 
} NvAisleCsvFields;
 
typedef std::map<uint32_t, NvAisleCsvFields> _AisleIndex_Map;
 
typedef std::map<uint32_t, _AisleIndex_Map> _AisleCSVMap;
 
typedef std::map<string, uint32_t> _AisleCameraMap;
 
typedef std::map<uint32_t, uint32_t> _AisleCameraViews;
 
typedef std::pair<_AisleIndex_Map::iterator, bool>  _AisleIndexMap_Result;
typedef std::pair<_AisleCSVMap::iterator, bool>  _AisleCSVMap_Result;
typedef std::pair<_AisleCameraMap::iterator, bool> _AisleCameraMap_Result;
typedef std::pair<_AisleCameraViews::iterator, bool> _AisleCameraView_Result;
class AisleCSVParser
{
  private:
    _AisleCSVMap CSVMap;
    _AisleCameraViews CameraViews;
 
    string csvFileName;
 
    void LoadCSVData();
    void DestroyCSVParser();
 
    uint32_t prepareAisleCSVMaxViews (uint32_t cam_id, vector<int>* vector_surface_index);
 
  public:
    ~AisleCSVParser();
 
    int getNvAisleCSVFields (uint32_t cam_id, uint32_t aisle_id, NvAisleCsvFields *fields);
    uint32_t getNvAisleCSVMaxViews (uint32_t cam_id, vector<int>* vector_surface_index);
 
    void getNvAisleCSVData(std::vector<NvAisleCsvFields> &csvAisleData);
 
    void printNvAisleCSVData(void);
    void printNvAisleCSVData(uint32_t cam_id, uint32_t aisle_id);
    void printAisleCSVFields(NvAisleCsvFields val);
 
    AisleCSVParser(string CSVFileName);
};
 
}
#endif