columnar::PhysliteTestHelpers::ColumnDataSplitLink Struct Reference

an implementation of PhysliteTestHelpers::IColumnData that reads a split ElementLink More...

Inheritance diagram for columnar::PhysliteTestHelpers::ColumnDataSplitLink:

Collaboration diagram for columnar::PhysliteTestHelpers::ColumnDataSplitLink:

Public Member Functions
	ColumnDataSplitLink (const std::string &val_columnName, const std::string &val_branchBaseName)
virtual bool	connect (TTree *tree, std::unordered_map< std::string, const PhysliteTestHelpers::IColumnData * > &sizeColumns, std::unordered_map< std::string, ColumnInfo > &requestedColumns) override
virtual std::size_t	getSize () const override
virtual void	setData (ColumnarToolWrapperData &tool, Long64_t entry) override

Public Attributes
std::string	columnName
std::string	branchBaseName
std::string	offsetName
const PhysliteTestHelpers::IColumnData *	sizeColumn = nullptr
const PhysliteTestHelpers::IColumnData *	targetSizeColumn = nullptr
std::array< ColumnarOffsetType, 2 >	offsets = {0, 0}
std::vector< ColumnarOffsetType >	dataCopy
TBranch *	mainBranch = nullptr
TBranch *	indexBranch = nullptr
TLeaf *	indexLeaf = nullptr
Benchmark	benchmarkUnpack
Benchmark	benchmark

Detailed Description

an implementation of PhysliteTestHelpers::IColumnData that reads a split ElementLink

I'm not sure this is the best way of doing it. Ideally I'd read the branches directly, but that was crashing, so I'm relying on TLeaf instead. This is not ideal, but it works for now.

Definition at line 288 of file ColumnarPhysliteTest.cxx.

Constructor & Destructor Documentation

ColumnDataSplitLink()

columnar::PhysliteTestHelpers::ColumnDataSplitLink::ColumnDataSplitLink ( const std::string &  val_columnName, const std::string &  val_branchBaseName  )

inline

Definition at line 303 of file ColumnarPhysliteTest.cxx.

        : columnName (val_columnName), branchBaseName (val_branchBaseName), benchmarkUnpack (columnName + " (unpack)"), benchmark (columnName) {}

Member Function Documentation

connect()

virtual bool columnar::PhysliteTestHelpers::ColumnDataSplitLink::connect ( TTree *  tree, std::unordered_map< std::string, const PhysliteTestHelpers::IColumnData * > &  sizeColumns, std::unordered_map< std::string, ColumnInfo > &  requestedColumns  )

inlineoverridevirtual

Implements columnar::PhysliteTestHelpers::IColumnData.

Definition at line 306 of file ColumnarPhysliteTest.cxx.

      {
        auto iter = requestedColumns.find (columnName);
        if (iter == requestedColumns.end())
          return false;
 
        mainBranch = tree->GetBranch (branchBaseName.c_str());
        if (!mainBranch)
          throw std::runtime_error ("failed to get branch: " + branchBaseName);
        indexBranch = tree->GetBranch ((branchBaseName + ".m_persIndex").c_str());
        if (!indexBranch)
          throw std::runtime_error ("failed to get branch: " + (branchBaseName + ".m_persIndex"));
        indexLeaf = indexBranch->GetLeaf ((branchBaseName + ".m_persIndex").c_str());
 
        offsetName = iter->second.offsetName;
        if (offsetName.empty())
          throw std::runtime_error ("missing offset column for: " + columnName);
 
        if (!iter->second.linkToName.empty())
        {
          auto linkIter = sizeColumns.find (iter->second.linkToName);
          if (linkIter == sizeColumns.end())
            throw std::runtime_error ("missing link column for: " + columnName);
          targetSizeColumn = linkIter->second;
        }
 
        if (auto sizeIter = sizeColumns.find (offsetName); sizeIter != sizeColumns.end())
          sizeColumn = sizeIter->second;
        else
          sizeColumns.emplace (offsetName, this);
 
        requestedColumns.erase (iter);
 
        iter = requestedColumns.find (offsetName);
        if (iter == requestedColumns.end())
          offsetName.clear();
        else
          requestedColumns.erase (iter);
 
        return true;
      }

getSize()

virtual std::size_t columnar::PhysliteTestHelpers::ColumnDataSplitLink::getSize ( ) const

inlineoverridevirtual

Implements columnar::PhysliteTestHelpers::IColumnData.

Definition at line 348 of file ColumnarPhysliteTest.cxx.

      {
        return offsets[1];
      }

setData()

virtual void columnar::PhysliteTestHelpers::ColumnDataSplitLink::setData ( ColumnarToolWrapperData &  tool, Long64_t  entry  )

inlineoverridevirtual

Implements columnar::PhysliteTestHelpers::IColumnData.

Definition at line 353 of file ColumnarPhysliteTest.cxx.

      {
        benchmark.startTimer ();
        mainBranch->GetEntry (entry);
        indexBranch->GetEntry (entry);
        benchmark.stopTimer ();
 
        benchmarkUnpack.startTimer ();
        offsets[1] = indexLeaf->GetLen();
        dataCopy.resize (offsets[1]);
        for (std::size_t i = 0; i < dataCopy.size(); ++i)
        {
          auto value = indexLeaf->GetTypedValue<UInt_t> (i);
          if (value != std::numeric_limits<UInt_t>::max())
            dataCopy[i] = value;
          else
            dataCopy[i] = invalidObjectIndex;
        }
        tool.setColumn (columnName, dataCopy.size(), dataCopy.data());
        if (!offsetName.empty())
        {
          offsets[1] = dataCopy.size();
          tool.setColumn (offsetName, offsets.size(), offsets.data());
        }
        if (sizeColumn && sizeColumn->getSize() != dataCopy.size())
          throw std::runtime_error ("size mismatch: " + columnName);
        if (targetSizeColumn)
        {
          const auto targetSize = targetSizeColumn->getSize();
          for (auto& item : dataCopy)
          {
            if (item != invalidObjectIndex && item >= targetSize)
              throw std::runtime_error ("link out of range: " + columnName + " " + std::to_string (item) + " " + std::to_string (targetSize) + " " + std::to_string (std::numeric_limits<UInt_t>::max()));
          }
        }
        benchmarkUnpack.stopTimer ();
      }

Member Data Documentation

benchmark

Benchmark columnar::PhysliteTestHelpers::ColumnDataSplitLink::benchmark

Definition at line 301 of file ColumnarPhysliteTest.cxx.

benchmarkUnpack

Benchmark columnar::PhysliteTestHelpers::ColumnDataSplitLink::benchmarkUnpack

Definition at line 300 of file ColumnarPhysliteTest.cxx.

branchBaseName

std::string columnar::PhysliteTestHelpers::ColumnDataSplitLink::branchBaseName

Definition at line 291 of file ColumnarPhysliteTest.cxx.

columnName

std::string columnar::PhysliteTestHelpers::ColumnDataSplitLink::columnName

Definition at line 290 of file ColumnarPhysliteTest.cxx.

dataCopy

std::vector<ColumnarOffsetType> columnar::PhysliteTestHelpers::ColumnDataSplitLink::dataCopy

Definition at line 296 of file ColumnarPhysliteTest.cxx.

indexBranch

TBranch* columnar::PhysliteTestHelpers::ColumnDataSplitLink::indexBranch = nullptr

Definition at line 298 of file ColumnarPhysliteTest.cxx.

indexLeaf

TLeaf* columnar::PhysliteTestHelpers::ColumnDataSplitLink::indexLeaf = nullptr

Definition at line 299 of file ColumnarPhysliteTest.cxx.

mainBranch

TBranch* columnar::PhysliteTestHelpers::ColumnDataSplitLink::mainBranch = nullptr

Definition at line 297 of file ColumnarPhysliteTest.cxx.

offsetName

std::string columnar::PhysliteTestHelpers::ColumnDataSplitLink::offsetName

Definition at line 292 of file ColumnarPhysliteTest.cxx.

offsets

std::array<ColumnarOffsetType, 2> columnar::PhysliteTestHelpers::ColumnDataSplitLink::offsets = {0, 0}

Definition at line 295 of file ColumnarPhysliteTest.cxx.

sizeColumn

const PhysliteTestHelpers::IColumnData* columnar::PhysliteTestHelpers::ColumnDataSplitLink::sizeColumn = nullptr

Definition at line 293 of file ColumnarPhysliteTest.cxx.

targetSizeColumn

const PhysliteTestHelpers::IColumnData* columnar::PhysliteTestHelpers::ColumnDataSplitLink::targetSizeColumn = nullptr

Definition at line 294 of file ColumnarPhysliteTest.cxx.

---

The documentation for this struct was generated from the following file:

• ColumnarPhysliteTest.cxx