DynamicTable.hpp

Go to the documentation of this file.

#pragma once
 
#include <string>
 
#include "Utils.hpp"
#include "io/BaseIO.hpp"
#include "io/ReadIO.hpp"
#include "nwb/hdmf/base/Container.hpp"
#include "nwb/hdmf/table/ElementIdentifiers.hpp"
#include "nwb/hdmf/table/VectorData.hpp"
#include "spec/hdmf_common.hpp"
 
namespace AQNWB::NWB
{
class DynamicTable : public Container
{
public:
  // Register the TimeSeries as a subclass of Container
  REGISTER_SUBCLASS(DynamicTable, AQNWB::SPEC::HDMF_COMMON::namespaceName)
 
  
  DynamicTable(const std::string& path, std::shared_ptr<IO::BaseIO> io);

  virtual ~DynamicTable();

  Status initialize(const std::string& description);

  Status finalize();

  Status addColumn(std::unique_ptr<VectorData>& vectorData,
                   const std::vector<std::string>& values);

  Status addReferenceColumn(const std::string& name,
                            const std::string& colDescription,
                            const std::vector<std::string>& dataset);

  Status setRowIDs(std::unique_ptr<ElementIdentifiers>& elementIDs,
                   const std::vector<int>& values);

  virtual void setColNames(const std::vector<std::string>& newColNames)
  {
    m_colNames = newColNames;
  }

  template<typename DTYPE = std::any>
  std::shared_ptr<VectorDataTyped<DTYPE>> readColumn(const std::string& colName)
  {
    std::string columnPath = AQNWB::mergePaths(m_path, colName);
    if (m_io->objectExists(columnPath)) {
      if (m_io->getStorageObjectType(columnPath) == StorageObjectType::Dataset)
      {
        return std::make_shared<VectorDataTyped<DTYPE>>(columnPath, m_io);
      }
    }
    return nullptr;
  }
 
  DEFINE_ATTRIBUTE_FIELD(readColNames,
                         std::string,
                         "colnames",
                         The names of the columns in the table)
 
  DEFINE_ATTRIBUTE_FIELD(readDescription,
                         std::string,
                         "description",
                         Description of what is in this dynamic table)
 
  DEFINE_REGISTERED_FIELD(
      readIdColumn,
      ElementIdentifiers,
      "id",
      "unique identifiers for the rows of this dynamic table")
 
protected:
  std::vector<std::string> m_colNames;
};
}  // namespace AQNWB::NWB