TileManager.h

Go to the documentation of this file.

 
/*
 * TileManager.h
 *
 *  Created on: Feb 23, 2018
 *      Author: mschefer
 */
 
#ifndef _SEFRAMEWORK_IMAGE_TILEMANAGER_H_
#define _SEFRAMEWORK_IMAGE_TILEMANAGER_H_
 
#include <iostream>
#include <thread>
#include <list>
#include <unordered_map>
 
#include <boost/thread/shared_mutex.hpp>
 
#include <ElementsKernel/Logging.h>
 
#include "SEFramework/Image/ImageTile.h"
#include "SEFramework/Image/ImageSource.h"
 
namespace SourceXtractor {
 
 
struct TileKey {
  std::shared_ptr<const ImageSource> m_source;
  int m_tile_x, m_tile_y;
 
  bool operator==(const TileKey& other) const;
 
  std::string getRepr() const;
};
 
inline std::ostream& operator<<(std::ostream& out, const TileKey& tk) {
  out << tk.getRepr();
  return out;
}
 
}
 
namespace std {
 
template<>
struct hash<SourceXtractor::TileKey> {
  std::size_t operator()(const SourceXtractor::TileKey& key) const {
    std::size_t local_hash = 0;
    boost::hash_combine(local_hash, key.m_source);
    boost::hash_combine(local_hash, key.m_tile_x);
    boost::hash_combine(local_hash, key.m_tile_y);
    return local_hash;
  }
};
 
}
 
namespace SourceXtractor {
 
class TileManager {
public:
 
  TileManager();
 
  virtual ~TileManager();
 
  // Actually not thread safe, call before starting the multi-threading
  void setOptions(int tile_width, int tile_height, int max_memory);
 
  void flush();
 
  std::shared_ptr<ImageTile>
  getTileForPixel(int x, int y, std::shared_ptr<const ImageSource> source);
 
  static std::shared_ptr<TileManager> getInstance();
 
  void saveAllTiles();
 
  int getTileWidth() const;
 
  int getTileHeight() const;
 
private:
 
  std::shared_ptr<ImageTile> tryTileFromCache(const TileKey& key);
 
  std::shared_ptr<boost::mutex>& getMutexForImageSource(const ImageSource*);
 
  void removeTile(TileKey tile_key);
 
  void removeExtraTiles();
 
  void addTile(TileKey key, std::shared_ptr<ImageTile> tile);
 
  int m_tile_width, m_tile_height;
  long m_max_memory;
  long m_total_memory_used;
 
  std::unordered_map<TileKey, std::shared_ptr<ImageTile>> m_tile_map;
  std::unordered_map<const ImageSource*, std::shared_ptr<boost::mutex>> m_mutex_map;
  std::list<TileKey> m_tile_list;
 
  boost::shared_mutex m_mutex;
};
 
}
 
 
#endif /* _SEFRAMEWORK_IMAGE_TILEMANAGER_H_ */