-
Notifications
You must be signed in to change notification settings - Fork 125
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
PeakShapeDetectorBin added with boost::python bindings
- Loading branch information
1 parent
2c6a98c
commit 409be08
Showing
13 changed files
with
487 additions
and
4 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
47 changes: 47 additions & 0 deletions
47
Framework/DataObjects/inc/MantidDataObjects/PeakShapeDetectorBin.h
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,47 @@ | ||
// Mantid Repository : https://github.com/mantidproject/mantid | ||
// | ||
// Copyright © 2024 ISIS Rutherford Appleton Laboratory UKRI, | ||
// NScD Oak Ridge National Laboratory, European Spallation Source, | ||
// Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS | ||
// SPDX - License - Identifier: GPL - 3.0 + | ||
#pragma once | ||
|
||
#include "MantidDataObjects/PeakShapeBase.h" | ||
#include <vector> | ||
|
||
namespace Mantid { | ||
namespace DataObjects { | ||
|
||
/** PeakShapeDetectorBin : PeakShape representing detector ids and integration limits of a peak | ||
*/ | ||
class MANTID_DATAOBJECTS_DLL PeakShapeDetectorBin : public PeakShapeBase { | ||
public: | ||
PeakShapeDetectorBin(std::vector<std::tuple<int32_t, double, double>> detectorBinList, | ||
Kernel::SpecialCoordinateSystem frame, std::string algorithmName = std::string(), | ||
int algorithmVersion = -1); | ||
/// Equals operator | ||
bool operator==(const PeakShapeDetectorBin &other) const; | ||
|
||
/// PeakShape interface | ||
std::string toJSON() const override; | ||
/// Clone PeakShapeDetectorBin | ||
Mantid::Geometry::PeakShape *clone() const override; | ||
/// Get the peak shape | ||
std::string shapeName() const override; | ||
|
||
/// PeakBase interface | ||
std::optional<double> radius(RadiusType type = RadiusType::Radius) const override; | ||
|
||
static const std::string detectorBinShapeName(); | ||
|
||
const std::vector<std::tuple<int32_t, double, double>> &getDetectorBinList() const { return m_detectorBinList; } | ||
|
||
private: | ||
std::vector<std::tuple<int32_t, double, double>> m_detectorBinList; | ||
}; | ||
|
||
using PeakShapeDetectorTOF_sptr = std::shared_ptr<PeakShapeDetectorBin>; | ||
using PeakShapeDetectorTOF_const_sptr = std::shared_ptr<const PeakShapeDetectorBin>; | ||
|
||
} // namespace DataObjects | ||
} // namespace Mantid |
35 changes: 35 additions & 0 deletions
35
Framework/DataObjects/inc/MantidDataObjects/PeakShapeDetectorBinFactory.h
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,35 @@ | ||
// Mantid Repository : https://github.com/mantidproject/mantid | ||
// | ||
// Copyright © 2024 ISIS Rutherford Appleton Laboratory UKRI, | ||
// NScD Oak Ridge National Laboratory, European Spallation Source, | ||
// Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS | ||
// SPDX - License - Identifier: GPL - 3.0 + | ||
#pragma once | ||
|
||
#include "MantidDataObjects/PeakShapeFactory.h" | ||
|
||
namespace Mantid { | ||
namespace Geometry { | ||
// Forward declare | ||
class PeakShape; | ||
} // namespace Geometry | ||
namespace DataObjects { | ||
|
||
/** PeakShapeDetectorBinFactory : Factory for DetectorBin peak shapes for | ||
de-serializing from JSON. | ||
* | ||
*/ | ||
class MANTID_DATAOBJECTS_DLL PeakShapeDetectorBinFactory : public PeakShapeFactory { | ||
public: | ||
/// Make product | ||
Mantid::Geometry::PeakShape *create(const std::string &source) const override; | ||
/// Set a successor should this factory be unsuitable | ||
void setSuccessor(PeakShapeFactory_const_sptr successorFactory) override; | ||
|
||
private: | ||
/// Successor factory | ||
PeakShapeFactory_const_sptr m_successor; | ||
}; | ||
|
||
} // namespace DataObjects | ||
} // namespace Mantid |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,62 @@ | ||
// Mantid Repository : https://github.com/mantidproject/mantid | ||
// | ||
// Copyright © 2024 ISIS Rutherford Appleton Laboratory UKRI, | ||
// NScD Oak Ridge National Laboratory, European Spallation Source, | ||
// Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS | ||
// SPDX - License - Identifier: GPL - 3.0 + | ||
#include "MantidDataObjects/PeakShapeDetectorBin.h" | ||
#include "MantidJson/Json.h" | ||
#include "MantidKernel/cow_ptr.h" | ||
#include <json/json.h> | ||
|
||
#include <utility> | ||
|
||
namespace Mantid::DataObjects { | ||
|
||
/** | ||
* Create PeakShapeDetectorBin | ||
* @param detectorBinList The list containing each detector Ids associated with its start and end points in TOF or | ||
* dSpacing domains | ||
* @param frame SpecialCoordinateSystem frame, default is None | ||
* @param algorithmName Name of the algorithm using this shape | ||
* @param algorithmVersion Version of the above algorithm | ||
*/ | ||
PeakShapeDetectorBin::PeakShapeDetectorBin(std::vector<std::tuple<int32_t, double, double>> detectorBinList, | ||
Kernel::SpecialCoordinateSystem frame, std::string algorithmName, | ||
int algorithmVersion) | ||
: PeakShapeBase(frame, std::move(algorithmName), algorithmVersion), m_detectorBinList(detectorBinList) { | ||
|
||
if (detectorBinList.size() == 0) { | ||
throw std::invalid_argument("No data provided for detector data"); | ||
} | ||
} | ||
|
||
bool PeakShapeDetectorBin::operator==(const PeakShapeDetectorBin &other) const { | ||
return PeakShapeBase::operator==(other) && (this->m_detectorBinList == other.m_detectorBinList); | ||
} | ||
|
||
std::string PeakShapeDetectorBin::toJSON() const { | ||
Json::Value root; | ||
PeakShapeBase::buildCommon(root); | ||
Json::Value detBinList; | ||
for (auto const &[detectorId, startX, endX] : m_detectorBinList) { | ||
Json::Value detBinVal; | ||
detBinVal["detId"] = Json::Value(detectorId); | ||
detBinVal["startX"] = Json::Value(startX); | ||
detBinVal["endX"] = Json::Value(endX); | ||
detBinList.append(detBinVal); | ||
} | ||
|
||
root["detectors"] = detBinList; | ||
return Mantid::JsonHelpers::jsonToString(root); | ||
} | ||
|
||
Mantid::Geometry::PeakShape *PeakShapeDetectorBin::clone() const { return new PeakShapeDetectorBin(*this); } | ||
|
||
std::string PeakShapeDetectorBin::shapeName() const { return PeakShapeDetectorBin::detectorBinShapeName(); } | ||
|
||
std::optional<double> PeakShapeDetectorBin::radius(RadiusType type) const { return std::nullopt; } | ||
|
||
const std::string PeakShapeDetectorBin::detectorBinShapeName() { return "PeakShapeDetectorBin"; } | ||
|
||
} // namespace Mantid::DataObjects |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,67 @@ | ||
// Mantid Repository : https://github.com/mantidproject/mantid | ||
// | ||
// Copyright © 2024 ISIS Rutherford Appleton Laboratory UKRI, | ||
// NScD Oak Ridge National Laboratory, European Spallation Source, | ||
// Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS | ||
// SPDX - License - Identifier: GPL - 3.0 + | ||
#include "MantidDataObjects/PeakShapeDetectorBinFactory.h" | ||
#include "MantidDataObjects/PeakShapeDetectorBin.h" | ||
#include "MantidJson/Json.h" | ||
#include "MantidKernel/SpecialCoordinateSystem.h" | ||
|
||
#include <json/json.h> | ||
#include <stdexcept> | ||
|
||
using namespace Mantid::Kernel; | ||
|
||
namespace Mantid::DataObjects { | ||
|
||
/** | ||
* @brief Create the PeakShapeDetectorBin | ||
* @param source : source JSON | ||
* @return PeakShape via this factory or it's successors | ||
*/ | ||
Mantid::Geometry::PeakShape *PeakShapeDetectorBinFactory::create(const std::string &source) const { | ||
Json::Value root; | ||
Mantid::Geometry::PeakShape *peakShape = nullptr; | ||
if (Mantid::JsonHelpers::parse(source, &root)) { | ||
const std::string shape = root["shape"].asString(); | ||
if (shape == PeakShapeDetectorBin::detectorBinShapeName()) { | ||
const std::string algorithmName(root["algorithm_name"].asString()); | ||
const int algorithmVersion(root["algorithm_version"].asInt()); | ||
const auto frame(static_cast<SpecialCoordinateSystem>(root["frame"].asInt())); | ||
|
||
std::vector<std::tuple<int32_t, double, double>> detectorBinList; | ||
const Json::Value detectorList = root["detectors"]; | ||
for (int index = 0; index < detectorList.size(); index++) { | ||
const Json::Value detBinVal = detectorList[index]; | ||
detectorBinList.emplace_back(detBinVal["detId"].asInt(), detBinVal["startX"].asDouble(), | ||
detBinVal["endX"].asDouble()); | ||
} | ||
peakShape = new PeakShapeDetectorBin(detectorBinList, frame, algorithmName, algorithmVersion); | ||
} else { | ||
if (m_successor) { | ||
peakShape = m_successor->create(source); | ||
} else { | ||
throw std::invalid_argument("PeakShapeDetectorBinFactory:: No successor " | ||
"factory able to process : " + | ||
source); | ||
} | ||
} | ||
} else { | ||
throw std::invalid_argument("PeakShapeDetectorBinFactory:: Source JSON for " | ||
"the peak shape is not valid: " + | ||
source); | ||
} | ||
return peakShape; | ||
} | ||
|
||
/** | ||
* @brief Set successor | ||
* @param successorFactory : successor | ||
*/ | ||
void PeakShapeDetectorBinFactory::setSuccessor(std::shared_ptr<const PeakShapeFactory> successorFactory) { | ||
this->m_successor = successorFactory; | ||
} | ||
|
||
} // namespace Mantid::DataObjects |
76 changes: 76 additions & 0 deletions
76
Framework/DataObjects/test/PeakShapeDetectorBinFactoryTest.h
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,76 @@ | ||
// Mantid Repository : https://github.com/mantidproject/mantid | ||
// | ||
// Copyright © 2024 ISIS Rutherford Appleton Laboratory UKRI, | ||
// NScD Oak Ridge National Laboratory, European Spallation Source, | ||
// Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS | ||
// SPDX - License - Identifier: GPL - 3.0 + | ||
#pragma once | ||
|
||
#include <cxxtest/TestSuite.h> | ||
#include <gmock/gmock.h> | ||
#include <json/json.h> | ||
|
||
#include "MantidDataObjects/PeakShapeDetectorBin.h" | ||
#include "MantidDataObjects/PeakShapeDetectorBinFactory.h" | ||
#include "MantidJson/Json.h" | ||
#include "MantidKernel/SpecialCoordinateSystem.h" | ||
#include "MantidKernel/cow_ptr.h" | ||
#include "MockObjects.h" | ||
|
||
using namespace Mantid; | ||
using namespace Mantid::DataObjects; | ||
using namespace Mantid::Kernel; | ||
using Mantid::DataObjects::PeakShapeDetectorBin; | ||
using Mantid::Kernel::SpecialCoordinateSystem; | ||
|
||
class PeakShapeDetectorBinFactoryTest : public CxxTest::TestSuite { | ||
public: | ||
static PeakShapeDetectorBinFactoryTest *createSuite() { return new PeakShapeDetectorBinFactoryTest(); } | ||
static void destroySuite(PeakShapeDetectorBinFactoryTest *suite) { delete suite; } | ||
|
||
void test_invalid_json_with_no_successor() { | ||
PeakShapeDetectorBinFactory factory; | ||
TS_ASSERT_THROWS(factory.create(""), std::invalid_argument &); | ||
} | ||
|
||
void test_successor_calling_when_shape_is_unhandled() { | ||
using namespace testing; | ||
MockPeakShapeFactory *delegate = new MockPeakShapeFactory; | ||
EXPECT_CALL(*delegate, create(_)).Times(1); | ||
|
||
PeakShapeDetectorBinFactory factory; | ||
factory.setSuccessor(PeakShapeFactory_const_sptr(delegate)); | ||
|
||
// Minimal valid JSON for describing the shape. | ||
Json::Value root; | ||
root["shape"] = "NotHandled"; | ||
const std::string str_json = Mantid::JsonHelpers::jsonToString(root); | ||
|
||
TS_ASSERT_THROWS_NOTHING(factory.create(str_json)); | ||
TS_ASSERT(Mock::VerifyAndClearExpectations(delegate)); | ||
} | ||
|
||
void test_when_no_successor() { | ||
PeakShapeDetectorBinFactory factory; | ||
|
||
// Minimal valid JSON for describing the shape. | ||
Json::Value root; | ||
root["shape"] = "NotHandled"; | ||
const std::string str_json = Mantid::JsonHelpers::jsonToString(root); | ||
|
||
TS_ASSERT_THROWS(factory.create(str_json), std::invalid_argument &); | ||
} | ||
|
||
void test_factory_create() { | ||
PeakShapeDetectorBin shape({{100, 10, 50}, {200, 34, 55}}, SpecialCoordinateSystem::None, "test", 1); | ||
|
||
PeakShapeDetectorBinFactory factory; | ||
std::shared_ptr<Mantid::Geometry::PeakShape> productShape(factory.create(shape.toJSON())); | ||
|
||
std::shared_ptr<PeakShapeDetectorBin> factoryShape = std::dynamic_pointer_cast<PeakShapeDetectorBin>(productShape); | ||
TS_ASSERT(factoryShape); | ||
|
||
TS_ASSERT_EQUALS(shape, *factoryShape); | ||
TS_ASSERT_EQUALS(factoryShape->getDetectorBinList(), shape.getDetectorBinList()) | ||
} | ||
}; |
Oops, something went wrong.