ds_aggregator.h

Go to the documentation of this file.

/*
 *  DeepSight face analysis SDK
 *  Copyright (C) 2013-2023 Raydiant Inc. - All Rights Reserved
 *
 *  This file is subject to the terms and conditions defined in
 *  the 'LICENSE.txt' file, which is part of this source code package.
 *
 */
 
#ifndef DS_AGGREGATOR_H
#define DS_AGGREGATOR_H
 
#include "ds_exports.h"
#include "ds_person.h"
 
#include <memory>
#include <vector>
#include <chrono>
 
namespace ds
{
class DS_API Aggregator
{
public:
  Aggregator(); 
  ~Aggregator() noexcept;
 
  Aggregator( Aggregator && aggregator ) noexcept;
  Aggregator & operator=( Aggregator && aggregator ) noexcept;
 
  static std::vector<ds::AggregatedPerson> aggregateOffline( const std::vector<ds::Person> & people );
 
  void newDetections( const std::vector<ds::Person> & people );
 
  std::vector<ds::AggregatedPerson> getFinalizedAggregations( bool clean = true );
  std::vector<ds::AggregatedPerson> getOngoingAggregations() const;
  std::vector<ds::AggregatedPerson> getAllAggregations()     const;
 
  void cleanFinalizedAggregations() noexcept;
  void cleanAll()                   noexcept;
 
private:
  class AggregatorImpl;
  std::unique_ptr<AggregatorImpl> mAggregatorImpl;
};
 
 
class DS_API SlidingWindowAggregator
{
public:
  SlidingWindowAggregator( std::chrono::milliseconds maxWindowDuration = std::chrono::milliseconds{ 10000 } );
  ~SlidingWindowAggregator() noexcept;
 
  SlidingWindowAggregator( SlidingWindowAggregator && slidingWindowAggregator ) noexcept;
  SlidingWindowAggregator & operator=( SlidingWindowAggregator && slidingWindowAggregator ) noexcept;
 
  void newDetections( const std::vector<ds::Person> & people );
 
  std::vector<ds::AggregatedPerson> getAggregations( std::chrono::milliseconds windowDuration = std::chrono::milliseconds{ 0 } );
 
  std::chrono::milliseconds getMaxWindowDuration() const noexcept;
  void                      setMaxWindowDuration( std::chrono::milliseconds maxWindowDuration ) noexcept;
 
  void clean() noexcept;
 
private:
  class SlidingWindowAggregatorImpl;
  std::unique_ptr<SlidingWindowAggregatorImpl> mSlidingWindowAggregatorImpl;
};
 
 
class DS_API LastDetections
{
public:
  LastDetections( std::chrono::milliseconds maxWindowDuration = std::chrono::milliseconds{ 10000 } );
  ~LastDetections() noexcept;
 
  LastDetections( LastDetections && lastDetections ) noexcept;
  LastDetections & operator=( LastDetections && lastDetections ) noexcept;
 
  void newDetections( const std::vector<ds::Person> & people );
 
  std::vector<ds::Person> getLastDetections( std::chrono::milliseconds windowDuration = std::chrono::milliseconds{ 0 } );
 
  std::chrono::milliseconds getMaxWindowDuration() const noexcept;
  void                      setMaxWindowDuration( std::chrono::milliseconds maxWindowDuration ) noexcept;
 
  void clean() noexcept;
 
private:
  class LastDetectionsImpl;
  std::unique_ptr<LastDetectionsImpl> mLastDetectionsImpl;
};
 
 
class DS_API BodyAggregator
{
public:
  BodyAggregator(); 
  ~BodyAggregator() noexcept;
 
  BodyAggregator( BodyAggregator && bodyAggregator ) noexcept;
  BodyAggregator & operator=( BodyAggregator && bodyAggregator ) noexcept;
 
  static std::vector<ds::AggregatedBody> aggregateOffline( const std::vector<ds::Body> & bodies );
 
  void newDetections( const std::vector<ds::Body> & bodies );
 
  std::vector<ds::AggregatedBody> getFinalizedAggregations( bool clean = true );
  std::vector<ds::AggregatedBody> getOngoingAggregations() const;
  std::vector<ds::AggregatedBody> getAllAggregations()     const;
 
  void cleanFinalizedAggregations() noexcept;
  void cleanAll()                   noexcept;
 
private:
  class BodyAggregatorImpl;
  std::unique_ptr<BodyAggregatorImpl> mBodyAggregatorImpl;
};
 
 
class DS_API SlidingWindowBodyAggregator
{
public:
  SlidingWindowBodyAggregator( std::chrono::milliseconds maxWindowDuration = std::chrono::milliseconds{ 10000 } );
  ~SlidingWindowBodyAggregator() noexcept;
 
  SlidingWindowBodyAggregator( SlidingWindowBodyAggregator && slidingWindowBodyAggregator ) noexcept;
  SlidingWindowBodyAggregator & operator=( SlidingWindowBodyAggregator && slidingWindowBodyAggregator ) noexcept;
 
  void newDetections( const std::vector<ds::Body> & body );
 
  std::vector<ds::AggregatedBody> getAggregations( std::chrono::milliseconds windowDuration = std::chrono::milliseconds{ 0 } );
 
  std::chrono::milliseconds getMaxWindowDuration() const noexcept;
  void                      setMaxWindowDuration( std::chrono::milliseconds maxWindowDuration ) noexcept;
 
  void clean() noexcept;
 
private:
  class SlidingWindowBodyAggregatorImpl;
  std::unique_ptr<SlidingWindowBodyAggregatorImpl> mSlidingWindowBodyAggregatorImpl;
};
 
 
class DS_API LastBodyDetections
{
public:
  LastBodyDetections( std::chrono::milliseconds maxWindowDuration = std::chrono::milliseconds{ 10000 } );
  ~LastBodyDetections() noexcept;
 
  LastBodyDetections( LastBodyDetections && lastBodyDetections ) noexcept;
  LastBodyDetections & operator=( LastBodyDetections && lastBodyDetections ) noexcept;
 
  void newDetections( const std::vector<ds::Body> & bodies );
 
  std::vector<ds::Body> getLastDetections( std::chrono::milliseconds windowDuration = std::chrono::milliseconds{ 0 } );
 
  std::chrono::milliseconds getMaxWindowDuration() const noexcept;
  void                      setMaxWindowDuration( std::chrono::milliseconds maxWindowDuration ) noexcept;
 
  void clean() noexcept;
 
private:
  class LastBodyDetectionsImpl;
  std::unique_ptr<LastBodyDetectionsImpl> mLastBodyDetectionsImpl;
};
} /* namespace ds */
 
#endif /* DS_AGGREGATOR_H */