TGeoMCBranchArrayContainer.h

Go to the documentation of this file.

// -----------------------------------------------------------------------
// Copyright (C) 2019 CERN and copyright holders of VMC Project.
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "LICENSE".
//
// See https://github.com/vmc-project/vmc for full licensing information.
// -----------------------------------------------------------------------
 
// Authors: Benedikt Volkel 07/03/2019
 
/*************************************************************************
 * Copyright (C) 2019, Rene Brun and Fons Rademakers.                    *
 * Copyright (C) 2019, ALICE Experiment at CERN.                         *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
// Class TGeoMCBranchArrayContainer
// ---------------------
// cache for storing TGeoBranchArray objects
//
 
#ifndef ROOT_TGeoMCBranchArrayContainer
#define ROOT_TGeoMCBranchArrayContainer
 
#include <vector>
#include <memory>
 
#include "TGeoBranchArray.h"
 
class TGeoManager;
 
class TGeoMCBranchArrayContainer {
public:
   TGeoMCBranchArrayContainer() = default;
   ~TGeoMCBranchArrayContainer() = default;

   void Initialize(UInt_t maxlevels = 100, UInt_t size = 8);
   void InitializeFromGeoManager(TGeoManager *man, UInt_t size = 8);
   void ResetCache();

   TGeoBranchArray *GetNewGeoState(UInt_t &userIndex);
   const TGeoBranchArray *GetGeoState(UInt_t userIndex);
   void FreeGeoState(UInt_t userIndex);
   void FreeGeoState(const TGeoBranchArray *geoState);
   void FreeGeoStates();
 
private:
   TGeoMCBranchArrayContainer(const TGeoMCBranchArrayContainer &);
   TGeoMCBranchArrayContainer &operator=(const TGeoMCBranchArrayContainer &);
   void ExtendCache(UInt_t targetSize = 1);
 
private:
   std::vector<std::unique_ptr<TGeoBranchArray>> fCache;
   UInt_t fMaxLevels = 100;
   std::vector<UInt_t> fFreeIndices;
   Bool_t fIsInitialized = kFALSE;
 
   ClassDefNV(TGeoMCBranchArrayContainer, 1)
};
 
#endif /* ROOT_TGeoMCBranchArrayContainer */