SourceDBCasa.h

Go to the documentation of this file.

// SourceDBCasa.h: Class for a Casa table 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_SOURCEDBCASA_H
#define LOFAR_PARMDB_SOURCEDBCASA_H
 
#include <set>
 
#include <casacore/tables/Tables/Table.h>
#include <casacore/casa/version.h>
 
#include "SourceDB.h"
#include "PatchInfo.h"
 
#include "common/Types.h"
 
namespace dp3 {
namespace parmdb {
 
 
class SourceDBCasa : public SourceDBRep {
 public:
  SourceDBCasa(const ParmDBMeta& pdm, bool forceNew);
 
  ~SourceDBCasa() 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 createTables(const std::string& tableName);
 
  void addSrc(const SourceInfo& sourceInfo, unsigned int patchId,
              const ParmMap& defaultParameters, double ra, double dec);
 
  void writePatch(double apparentBrightness, double ra, double dec,
                  unsigned int rownr);
 
  std::vector<std::string> findDuplicates(casacore::Table& table,
                                          const std::string& columnName);
 
  void fillSets();
 
  std::vector<SourceInfo> readSources(const casacore::Table& table);
 
  casacore::Table selectPatches(int category, const std::string& pattern,
                                double minBrightness,
                                double maxBrightness) const;
 
  double getDefaultParmValue(const std::string& name);
 
  casacore::Table itsPatchTable;
  casacore::Table itsSourceTable;
  std::set<std::string> itsPatchSet;
  std::set<std::string> itsSourceSet;
  bool itsSetsFilled;
  casacore::Vector<common::rownr_t> itsRowNr;
};
 
 
}  // namespace parmdb
}  // namespace dp3
 
#endif