track.hpp

// Copyright (C) 2023 Alessandro Fornasier.
// Control of Networked Systems, University of Klagenfurt, Austria.
//
// All rights reserved.
//
// This software is licensed under the terms of the BSD-2-Clause-License with
// no commercial use allowed, the full terms of which are made available
// in the LICENSE file. No license in patents is granted.
//
// You can contact the authors at <alessandro.fornasier@ieee.org>
 
#ifndef TRACK_HPP
#define TRACK_HPP
 
#include <opencv2/opencv.hpp>
 
#include "types/fptypes.hpp"
#include "vision/features.hpp"
 
namespace msceqf
{
struct Track
{
  using Times = std::vector<fp>;  
 
  inline bool empty() const noexcept { return uvs_.empty(); }
 
  inline size_t size() const noexcept
  {
    assert(uvs_.size() == normalized_uvs_.size());
    assert(uvs_.size() == timestamps_.size());
    return uvs_.size();
  }
 
  void removeInvalid(std::vector<bool>& invalid)
  {
    assert(invalid.size() == uvs_.size());
    assert(uvs_.size() == normalized_uvs_.size());
    assert(uvs_.size() == timestamps_.size());
 
    size_t i = 0;
    size_t j = 0;
 
    while (i < uvs_.size())
    {
      if (!invalid[i])
      {
        uvs_[j] = uvs_[i];
        normalized_uvs_[j] = normalized_uvs_[i];
        timestamps_[j] = timestamps_[i];
        ++j;
      }
      ++i;
    }
 
    uvs_.resize(j);
    normalized_uvs_.resize(j);
    timestamps_.resize(j);
  }
 
  void removeTail(const fp& timestamp, const bool& remove_equal = true)
  {
    assert(uvs_.size() == normalized_uvs_.size());
    assert(uvs_.size() == timestamps_.size());
 
    size_t j = 0;
    size_t i = 0;
 
    while (i < uvs_.size())
    {
      if (remove_equal && timestamps_[i] > timestamp)
      {
        uvs_[j] = uvs_[i];
        normalized_uvs_[j] = normalized_uvs_[i];
        timestamps_[j] = timestamps_[i];
        ++j;
      }
      if (!remove_equal && timestamps_[i] >= timestamp)
      {
        uvs_[j] = uvs_[i];
        normalized_uvs_[j] = normalized_uvs_[i];
        timestamps_[j] = timestamps_[i];
        ++j;
      }
      ++i;
    }
 
    uvs_.resize(j);
    normalized_uvs_.resize(j);
    timestamps_.resize(j);
  }
 
  friend bool operator<(const Track& lhs, const Track& rhs) { return lhs.size() < rhs.size(); }
 
  FeaturesCoordinates uvs_;             
  FeaturesCoordinates normalized_uvs_;  
  Times timestamps_;                    
};
 
using Tracks = std::unordered_map<uint, Track>;  
 
}  // namespace msceqf
 
#endif  // TRACK_HPP