usMHDSequenceReader.h

/****************************************************************************
 *
 * This file is part of the ustk software.
 * Copyright (C) 2016 - 2017 by Inria. All rights reserved.
 *
 * This software is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * ("GPL") version 2 as published by the Free Software Foundation.
 * See the file LICENSE.txt at the root directory of this source
 * distribution for additional information about the GNU GPL.
 *
 * For using ustk with software that can not be combined with the GNU
 * GPL, please contact Inria about acquiring a ViSP Professional
 * Edition License.
 *
 * This software was developed at:
 * Inria Rennes - Bretagne Atlantique
 * Campus Universitaire de Beaulieu
 * 35042 Rennes Cedex
 * France
 *
 * If you have questions regarding the use of this file, please contact
 * Inria at ustk@inria.fr
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 *
 * Authors:
 * Marc Pouliquen
 *
 *****************************************************************************/
 
#ifndef __usMHDSequenceReader_h_
#define __usMHDSequenceReader_h_
 
#include <iostream>
#include <stdint.h>
#include <vector>
 
#include <visp3/ustk_core/usImageIo.h>
#include <visp3/ustk_core/usRawFileParser.h>
 
class VISP_EXPORT usMHDSequenceReader
{
public:
  usMHDSequenceReader();
  ~usMHDSequenceReader();
 
  void acquire(usImageRF2D<short int> &image, uint64_t &timestamp);
  void acquire(usImagePreScan2D<unsigned char> &image, uint64_t &timestamp);
  void acquire(usImagePostScan2D<unsigned char> &image, uint64_t &timestamp);
  void acquire(usImageRF3D<short int> &image, std::vector<uint64_t> &timestamp);
  void acquire(usImagePreScan3D<unsigned char> &image, std::vector<uint64_t> &timestamp);
  void acquire(usImagePostScan3D<unsigned char> &image, uint64_t &timestamp);
 
  bool end();
 
  void getImage(unsigned int imageNumber, usImageRF2D<short int> &image, uint64_t &timestamp);
  void getImage(unsigned int imageNumber, usImagePreScan2D<unsigned char> &image, uint64_t &timestamp);
  void getImage(unsigned int imageNumber, usImagePostScan2D<unsigned char> &image, uint64_t &timestamp);
  void getImage(unsigned int imageNumber, usImageRF3D<short int> &image, std::vector<uint64_t> &timestamp);
  void getImage(unsigned int imageNumber, usImagePreScan3D<unsigned char> &image, std::vector<uint64_t> &timestamp);
  void getImage(unsigned int imageNumber, usImagePostScan3D<unsigned char> &image, uint64_t &timestamp);
 
  us::ImageType getImageType() const;
 
  int getImageNumber() const;
 
  uint64_t getNextTimeStamp();
  std::vector<uint64_t> getNextTimeStamps();
 
  int getTotalImageNumber() const;
 
  void setSequenceDirectory(const std::string sequenceDirectory);
 
private:
  std::string m_sequenceDirectory;
 
  us::ImageType m_sequenceImageType;
 
  std::vector<std::string> m_sequenceFiles;
 
  int m_totalImageNumber;
  int m_imageCounter;
};
 
#endif // __usMHDSequenceReader_h_