SourceDBBlob.h

Go to the documentation of this file.

// SourceDBBlob.h: Class for a Blob file holding sources and their parameters
//
// Copyright (C) 2020 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
 
#ifndef LOFAR_PARMDB_SOURCEDBBLOB_H
#define LOFAR_PARMDB_SOURCEDBBLOB_H
 
#include "SourceDB.h"
 
#include <fstream>
#include "blob/BlobOBufStream.h"
#include "blob/BlobIBufStream.h"
#include "blob/BlobOStream.h"
#include "blob/BlobIStream.h"
 
namespace dp3 {
namespace parmdb {
 
 
class SourceDBBlob : public SourceDBRep {
 public:
  SourceDBBlob(const ParmDBMeta& pdm, bool forceNew);
 
  ~SourceDBBlob() override;
 
  void lock(bool lockForWrite) override;
  void unlock() override;
 
  void checkDuplicates() override;
 
  std::vector<std::string> findDuplicatePatches() override;
 
  std::vector<std::string> findDuplicateSources() override;
 
  bool patchExists(const std::string& patchName) override;
 
  bool sourceExists(const std::string& sourceName) override;
 
  unsigned int addPatch(const std::string& patchName, int catType,
                        double apparentBrightness, double ra, double dec,
                        bool check) override;
 
  void updatePatch(unsigned int patchId, double apparentBrightness, double ra,
                   double dec) override;
 
  void addSource(const SourceInfo& sourceInfo, const std::string& patchName,
                 const ParmMap& defaultParameters, double ra, double dec,
                 bool check) override;
  void addSource(const SourceData& source, bool check) override;
 
  void addSource(const SourceInfo& sourceInfo, const std::string& patchName,
                 int catType, double apparentBrightness,
                 const ParmMap& defaultParameters, double ra, double dec,
                 bool check) override;
 
  std::vector<std::string> getPatches(int category, const std::string& pattern,
                                      double minBrightness,
                                      double maxBrightness) override;
 
  std::vector<PatchInfo> getPatchInfo(int category, const std::string& pattern,
                                      double minBrightness,
                                      double maxBrightness) override;
 
  std::vector<SourceInfo> getPatchSources(
      const std::string& patchName) override;
 
  std::vector<SourceData> getPatchSourceData(
      const std::string& patchName) override;
 
  SourceInfo getSource(const std::string& sourceName) override;
 
  std::vector<SourceInfo> getSources(const std::string& pattern) override;
 
  void deleteSources(const std::string& sourceNamePattern) override;
 
  void clearTables() override;
 
  void getNextSource(SourceData& src) override;
 
  bool atEnd() override;
 
  void rewind() override;
 
 private:
  void readAll();
 
  std::fstream itsFile;
  std::shared_ptr<blob::BlobIBufStream> itsBufIn;
  std::shared_ptr<blob::BlobOBufStream> itsBufOut;
  std::shared_ptr<blob::BlobIStream> itsBlobIn;
  std::shared_ptr<blob::BlobOStream> itsBlobOut;
  bool itsCanWrite;
  int64_t itsEndPos;
  std::map<std::string, PatchInfo> itsPatches;
  std::map<std::string, std::vector<SourceData>>
      itsSources;  
};
 
 
}  // namespace parmdb
}  // namespace dp3
 
#endif