ElectricFieldreader: Remove Hack for Plotting (45e81215) · Commits · XRFLab109 / Allpix Squared

src/modules/ElectricFieldReader/ElectricFieldReaderModule.cpp

+26 −32

Original line number	Diff line number	Diff line
		@@ -334,38 +334,38 @@ void ElectricFieldReaderModule::create_output_plots() {
		}

		auto model = detector_->getModel();
		if(config_.get<bool>("output_plots_single_pixel", true)) {
		// If we need to plot a single pixel we change the model to fake the sensor to be a single pixel
		// NOTE: This is a little hacky, but is the easiest approach
		model = std::make_shared<DetectorModel>(*model);
		model->setSensorExcessTop(0);
		model->setSensorExcessBottom(0);
		model->setSensorExcessLeft(0);
		model->setSensorExcessRight(0);
		model->setNPixels(ROOT::Math::DisplacementVector2D<ROOT::Math::Cartesian2D<unsigned int>>(1, 1));
		}

		// If we need to plot a single pixel, we use size and position of the pixel at the origin
		auto single_pixel = config_.get<bool>("output_plots_single_pixel", true);
		auto center = (single_pixel ? ROOT::Math::XYZPoint(model->getPixelCenter(0, 0).x(), model->getPixelCenter(0, 0).y(), 0)
		: model->getSensorCenter());
		auto size =
		(single_pixel
		? ROOT::Math::XYZVector(model->getPixelSize().x(), model->getPixelSize().y(), model->getSensorSize().z())
		: model->getSensorSize());

		// Use either full sensor axis or only depleted region
		double z_min = model->getSensorCenter().z() - model->getSensorSize().z() / 2.0;
		double z_max = model->getSensorCenter().z() + model->getSensorSize().z() / 2.0;
		double z_min = center.z() - size.z() / 2.0;
		double z_max = center.z() + size.z() / 2.0;

		// Determine minimum and maximum index depending on projection axis
		double min1 = NAN, max1 = NAN;
		double min2 = NAN, max2 = NAN;
		if(project == 'x') {
		min1 = model->getSensorCenter().y() - model->getSensorSize().y() / 2.0;
		max1 = model->getSensorCenter().y() + model->getSensorSize().y() / 2.0;
		min1 = center.y() - size.y() / 2.0;
		max1 = center.y() + size.y() / 2.0;
		min2 = z_min;
		max2 = z_max;
		} else if(project == 'y') {
		min1 = model->getSensorCenter().x() - model->getSensorSize().x() / 2.0;
		max1 = model->getSensorCenter().x() + model->getSensorSize().x() / 2.0;
		min1 = center.x() - size.x() / 2.0;
		max1 = center.x() + size.x() / 2.0;
		min2 = z_min;
		max2 = z_max;
		} else {
		min1 = model->getSensorCenter().x() - model->getSensorSize().x() / 2.0;
		max1 = model->getSensorCenter().x() + model->getSensorSize().x() / 2.0;
		min2 = model->getSensorCenter().y() - model->getSensorSize().y() / 2.0;
		max2 = model->getSensorCenter().y() + model->getSensorSize().y() / 2.0;
		min1 = center.x() - size.x() / 2.0;
		max1 = center.x() + size.x() / 2.0;
		min2 = center.y() - size.y() / 2.0;
		max2 = center.y() + size.y() / 2.0;
		}

		// Create 2D histograms
		@@ -397,11 +397,9 @@ void ElectricFieldReaderModule::create_output_plots() {
		// Determine the coordinate to use for projection
		double x = 0, y = 0, z = 0;
		if(project == 'x') {
		x = model->getSensorCenter().x() - model->getSensorSize().x() / 2.0 +
		config_.get<double>("output_plots_projection_percentage", 0.5) * model->getSensorSize().x();
		x = center.x() - size.x() / 2.0 + config_.get<double>("output_plots_projection_percentage", 0.5) * size.x();
		} else if(project == 'y') {
		y = model->getSensorCenter().y() - model->getSensorSize().y() / 2.0 +
		config_.get<double>("output_plots_projection_percentage", 0.5) * model->getSensorSize().y();
		y = center.y() - size.y() / 2.0 + config_.get<double>("output_plots_projection_percentage", 0.5) * size.y();
		} else {
		z = z_min + config_.get<double>("output_plots_projection_percentage", 0.5) * (z_max - z_min);
		}
		@@ -414,22 +412,19 @@ void ElectricFieldReaderModule::create_output_plots() {
		// Find the electric field at every index
		for(size_t j = 0; j < steps; ++j) {
		if(project == 'x') {
		y = model->getSensorCenter().y() - model->getSensorSize().y() / 2.0 +
		((static_cast<double>(j) + 0.5) / static_cast<double>(steps)) * model->getSensorSize().y();
		y = center.y() - size.y() / 2.0 + ((static_cast<double>(j) + 0.5) / static_cast<double>(steps)) * size.y();
		histogram->GetXaxis()->SetTitle("y (mm)");
		histogram_x->GetXaxis()->SetTitle("y (mm)");
		histogram_y->GetXaxis()->SetTitle("y (mm)");
		histogram_z->GetXaxis()->SetTitle("y (mm)");
		} else if(project == 'y') {
		x = model->getSensorCenter().x() - model->getSensorSize().x() / 2.0 +
		((static_cast<double>(j) + 0.5) / static_cast<double>(steps)) * model->getSensorSize().x();
		x = center.x() - size.x() / 2.0 + ((static_cast<double>(j) + 0.5) / static_cast<double>(steps)) * size.x();
		histogram->GetXaxis()->SetTitle("x (mm)");
		histogram_x->GetXaxis()->SetTitle("x (mm)");
		histogram_y->GetXaxis()->SetTitle("x (mm)");
		histogram_z->GetXaxis()->SetTitle("x (mm)");
		} else {
		x = model->getSensorCenter().x() - model->getSensorSize().x() / 2.0 +
		((static_cast<double>(j) + 0.5) / static_cast<double>(steps)) * model->getSensorSize().x();
		x = center.x() - size.x() / 2.0 + ((static_cast<double>(j) + 0.5) / static_cast<double>(steps)) * size.x();
		histogram->GetXaxis()->SetTitle("x (mm)");
		histogram_x->GetXaxis()->SetTitle("x (mm)");
		histogram_y->GetXaxis()->SetTitle("x (mm)");
		@@ -449,8 +444,7 @@ void ElectricFieldReaderModule::create_output_plots() {
		histogram_y->GetYaxis()->SetTitle("z (mm)");
		histogram_z->GetYaxis()->SetTitle("z (mm)");
		} else {
		y = model->getSensorCenter().y() - model->getSensorSize().y() / 2.0 +
		((static_cast<double>(k) + 0.5) / static_cast<double>(steps)) * model->getSensorSize().y();
		y = center.y() - size.y() / 2.0 + ((static_cast<double>(k) + 0.5) / static_cast<double>(steps)) * size.y();
		histogram->GetYaxis()->SetTitle("y (mm)");
		histogram_x->GetYaxis()->SetTitle("y (mm)");
		histogram_y->GetYaxis()->SetTitle("y (mm)");