FragmentIndex.h

// FragmentIndex.h
//
// Author : Jeromy Tompkins
// Date   : 3/2014
//

#ifndef FRAGMENTINDEX_H
#define FRAGMENTINDEX_H

#include <vector>
#include <stdint.h>
#include <cstdlib>

struct FragmentInfo // A structure containting basic info from the Fragment header --JP
{
  uint64_t s_timestamp;
  uint32_t s_sourceId;
  uint32_t s_size;
  uint32_t s_barrier;
  uint16_t* s_itemhdr;
  uint16_t* s_itembody;

  FragmentInfo() : 
      s_timestamp(0), s_sourceId(0), s_size(0), s_barrier(0), 
      s_itemhdr(0), s_itembody(0) {}
};


class FragmentIndex
{
  private:
  typedef std::vector<FragmentInfo> Container;

  public:
  typedef Container::iterator iterator;
  typedef Container::const_iterator const_iterator;

  private:
  Container m_frags; 

  public:

  FragmentIndex();

  FragmentIndex(uint16_t* data);

  FragmentInfo getFragment(size_t i) 
  {
    if (i<m_frags.size() && i>=0) {
      return m_frags.at(i); //pointer to ith fragment in the m_frags list --JP
    } else {
      FragmentInfo null;
      null.s_itembody = static_cast<uint16_t*>(0);
      return null;
    }
  }
 
  size_t getNumberFragments() const { return m_frags.size(); }

  void indexFragments(uint16_t* begin, uint16_t* end);

  void indexFragments(uint16_t* data, size_t max_bytes) {
    indexFragments(data, data+max_bytes/sizeof(uint16_t) );
  }

  private:
  size_t computeWordsToNextFragment(uint16_t* data);


  public:
  // Implement an iterator interface
  iterator begin() { return m_frags.begin(); }
  const_iterator begin() const { return m_frags.begin(); }

  iterator end() { return m_frags.end(); }
  const_iterator end() const { return m_frags.end(); }
};

#endif