GeoAlignmentStore.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef GEOMODELUTILITIES_GEOALIGNMENTSTORE_H
#define GEOMODELUTILITIES_GEOALIGNMENTSTORE_H

#include <stdexcept>
#include "GeoPrimitives/GeoPrimitives.h"
#include "AthenaKernel/CLASS_DEF.h"
#include "AthenaKernel/CondCont.h"
 
#include "GeoModelKernel/GeoDefinitions.h"
#include "GeoModelKernel/GeoVAlignmentStore.h"
#include "GeoModelKernel/GeoVFullPhysVol.h"
#include "GeoModelKernel/GeoAlignableTransform.h"
 
#include "GeoModelUtilities/TransformMap.h"
#include "GeoModelUtilities/TransientConstSharedPtr.h"
 
class GeoAlignmentStore : public GeoVAlignmentStore {
public:
    GeoAlignmentStore() = default;
    virtual ~GeoAlignmentStore() override = default;

    virtual void setDelta(const GeoAlignableTransform* alignableNode, 
                          const GeoTrf::Transform3D& transform) override;

    void setDelta(const GeoAlignableTransform* alignableNode, 
                  std::shared_ptr<const GeoTrf::Transform3D> trans);

    virtual const GeoTrf::Transform3D* getDelta(const GeoAlignableTransform* alignableNode) const override final;

    virtual void setAbsPosition(const GeoNodePositioning* fullPhysVol, 
                                const GeoTrf::Transform3D& physVolTrf) override final;
    virtual const GeoTrf::Transform3D* getAbsPosition(const GeoNodePositioning* fullPhysVol) const override final;
    virtual void setDefAbsPosition(const GeoNodePositioning* fullPhysVol, 
                                  const GeoTrf::Transform3D& unAlignedTrf) override final;
    virtual const GeoTrf::Transform3D* getDefAbsPosition(const GeoNodePositioning* fullPhysVol) const override final;

    bool append(const GeoAlignmentStore& other);
    void clearPosCache();
    void lockDelta();
    void lockPosCache();
 
    using DeltaMap = TransformMap<GeoAlignableTransform, GeoTrf::Transform3D>;
    using DeltaMapPtr = GeoModel::TransientConstSharedPtr<DeltaMap>;
 
    using PositioningMap = TransformMap<GeoNodePositioning, GeoTrf::Transform3D>;
    using PositioningMapPtr = GeoModel::TransientConstSharedPtr<PositioningMap>;
private:    
    DeltaMapPtr m_deltas{std::make_unique<DeltaMap>()};
    PositioningMapPtr m_absPositions{std::make_unique<PositioningMap>()};
    PositioningMapPtr m_defAbsPositions{std::make_unique<PositioningMap>()};
 
public:
    DeltaMapPtr getDeltas() const;
    PositioningMapPtr getAbsPositions() const;
    PositioningMapPtr getDefAbsPositions() const;
};
 
inline const GeoTrf::Transform3D* GeoAlignmentStore::getDelta(const GeoAlignableTransform* axf) const { 
    return m_deltas->getTransform(axf); 
}
inline const GeoTrf::Transform3D* GeoAlignmentStore::getAbsPosition(const GeoNodePositioning* fpv) const {
    return m_absPositions->getTransform(fpv); 
}
inline const GeoTrf::Transform3D* GeoAlignmentStore::getDefAbsPosition(const GeoNodePositioning* fpv) const {
    return m_defAbsPositions->getTransform(fpv);
}
 
CLASS_DEF(GeoAlignmentStore, 135648236, 1)
CONDCONT_DEF(GeoAlignmentStore, 33985726);
 
#endif