ClusterDesc.h

Go to the documentation of this file.

// ClusterDesc.h:  Description of a cluster and the nodes in it
//
// Copyright (C) 2020 ASTRON (Netherlands Institute for Radio Astronomy)
// SPDX-License-Identifier: GPL-3.0-or-later
 
 
#ifndef LOFAR_LMWCOMMON_CLUSTERDESC_H
#define LOFAR_LMWCOMMON_CLUSTERDESC_H
 
#include "NodeDesc.h"
 
#include <string>
#include <vector>
#include <iosfwd>
 
namespace dp3 {
namespace common {
class ParameterSet;
}
}  // namespace dp3
 
namespace dp3 {
namespace common {
 
 
 
class ClusterDesc {
 public:
  ClusterDesc() {}
 
  explicit ClusterDesc(const std::string& parsetName);
 
  void setName(const std::string& name) { itsName = name; }
 
  void addNode(const NodeDesc& node, bool canExist = false);
 
  void write(std::ostream& os) const;
 
  const std::string& getName() const { return itsName; }
 
  const NodeDesc& getNode(const std::string& nodeName) const;
 
  const std::vector<NodeDesc>& getNodes() const { return itsNodes; }
 
  const std::map<std::string, std::vector<int>>& getMap() const {
    return itsFS2Nodes;
  }
 
 private:
  void init(const std::string& parsetName);
 
  void getHomCluster(const ParameterSet& parset);
 
  void addNodes(const ParameterSet& parset, NodeDesc::NodeType type);
 
  void getHetCluster(const ParameterSet& parset);
 
  void getSubClusters(const std::vector<std::string>& parsetNames,
                      const std::string& defaultDir);
 
  void add2Map(int nodeIndex);
 
  std::string itsName;
  std::vector<NodeDesc> itsNodes;
  std::map<std::string, int> itsNodeMap;
  std::map<std::string, std::vector<int>> itsFS2Nodes;
};
 
}  // namespace common
}  // namespace dp3
 
#endif