BlueprintTests.cpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file BlueprintTests.cpp

// This file is part of the Acts project.
//
// Copyright (C) 2023 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#include <boost/test/unit_test.hpp>

#include "Acts/Detector/Blueprint.hpp"

#include <fstream>

namespace Acts {
namespace Experimental {
class IInternalStructureBuilder {};
}  // namespace Experimental
}  // namespace Acts

BOOST_AUTO_TEST_SUITE(Experimental)

BOOST_AUTO_TEST_CASE(BlueprintTest) {
  std::vector<Acts::ActsScalar> bValues = {0., 10., 100.};

  // Create  root node
  std::vector<Acts::BinningValue> binning = {Acts::binR};
  auto root = std::make_unique<Acts::Experimental::Blueprint::Node>(
      "detector", Acts::Transform3::Identity(), Acts::VolumeBounds::eOther,
      bValues, binning);
  // Check the root node
  BOOST_CHECK(root->isRoot());
  BOOST_CHECK(root->parent == nullptr);
  BOOST_CHECK(root->children.empty());
  BOOST_CHECK(root->name == "detector");

  auto leaf0 = std::make_unique<Acts::Experimental::Blueprint::Node>(
      "volume_0", Acts::Transform3::Identity(), Acts::VolumeBounds::eOther,
      bValues);
  BOOST_CHECK(leaf0->isLeaf());

  auto branch = std::make_unique<Acts::Experimental::Blueprint::Node>(
      "container_0", Acts::Transform3::Identity(), Acts::VolumeBounds::eOther,
      bValues, binning);

  auto leaf1 = std::make_unique<Acts::Experimental::Blueprint::Node>(
      "volume_1", Acts::Transform3::Identity(), Acts::VolumeBounds::eOther,
      bValues);

  auto leaf2 = std::make_unique<Acts::Experimental::Blueprint::Node>(
      "volume_2", Acts::Transform3::Identity(), Acts::VolumeBounds::eOther,
      bValues,
      std::make_shared<Acts::Experimental::IInternalStructureBuilder>());

  // Keep around the pointers of the branch & leaves
  auto* leaf0Ptr = leaf0.get();
  auto* leaf1Ptr = leaf1.get();
  auto* leaf2Ptr = leaf2.get();
  auto* branchPtr = branch.get();

  branch->add(std::move(leaf1));
  branch->add(std::move(leaf2));

  // Branch has two children
  BOOST_CHECK(branch->children.size() == 2u);

  // Parent of the leaves is the branch
  BOOST_CHECK(leaf1Ptr->parent == branchPtr);
  BOOST_CHECK(leaf2Ptr->parent == branchPtr);

  root->add(std::move(branch));

  // Root stays root
  BOOST_CHECK(root->isRoot());
  // Parent of the branch is the root
  BOOST_CHECK(branchPtr->parent == root.get());

  root->add(std::move(leaf0));

  // Parent of the leaf is the root
  BOOST_CHECK(leaf0Ptr->parent == root.get());

  std::ofstream fs("blueprint.dot");
  root->dotStream(fs);
  fs.close();
}

BOOST_AUTO_TEST_SUITE_END()