rusefi-full/firmware/controllers/sensors/sensor.cpp

#include "sensor.h"

// This struct represents one sensor in the registry.
// It stores whether the sensor should use a mock value,
// the value to use, and if not a pointer to the sensor that
// can provide a real value.
struct SensorRegistryEntry {
	bool useMock;
	float mockValue;
	Sensor *sensor;
};

static SensorRegistryEntry s_sensorRegistry[static_cast<size_t>(SensorType::PlaceholderLast)] = {};

bool Sensor::Register() {
	// Get a ref to where we should be
	auto &entry = s_sensorRegistry[getIndex()];

	// If there's somebody already here - a consumer tried to double-register a sensor
	if (entry.sensor) {
		// This sensor has already been registered. Don't re-register it.
		return false;
	} else {
		// put ourselves in the registry
		s_sensorRegistry[getIndex()].sensor = this;
		return true;
	}
}

/*static*/ void Sensor::resetRegistry() {
	constexpr size_t len = sizeof(s_sensorRegistry) / sizeof(s_sensorRegistry[0]);

	// Clear all entries
	for (size_t i = 0; i < len; i++) {
		auto &entry = s_sensorRegistry[i];

		entry.sensor = nullptr;
		entry.useMock = false;
		entry.mockValue = 0.0f;
	}
}

/*static*/ SensorRegistryEntry *Sensor::getEntryForType(SensorType type) {
	size_t index = getIndex(type);
	// Check that we didn't get garbage
	if (index >= getIndex(SensorType::PlaceholderLast)) {
		return nullptr;
	}

	return &s_sensorRegistry[index];
}

/*static*/ const Sensor *Sensor::getSensorOfType(SensorType type) {
	auto entry = getEntryForType(type);
	return entry ? entry->sensor : nullptr;
}

/*static*/ SensorResult Sensor::get(SensorType type) {
	const auto entry = getEntryForType(type);

	// Check if this is a valid sensor entry
	if (!entry) {
		return {false, 0.0f};
	}

	// Next check for mock
	if (entry->useMock) {
		return {true, entry->mockValue};
	}

	// Get the sensor out of the entry
	const Sensor *s = entry->sensor;
	if (s) {
		// If we found the sensor, ask it for a result.
		return s->get();
	}

	// We've exhausted all valid ways to return something - sensor not found.
	return {false, 0};
}

/*static*/ void Sensor::setMockValue(SensorType type, float value) {
	auto entry = getEntryForType(type);

	if (entry) {
		entry->mockValue = value;
		entry->useMock = true;
	}
}

/*static*/ void Sensor::setMockValue(int type, float value) {
	// bounds check
	if (type <= 0 || type >= static_cast<int>(SensorType::PlaceholderLast)) {
		return;
	}

	setMockValue(static_cast<SensorType>(type), value);
}

/*static*/ void Sensor::resetMockValue(SensorType type) {
	auto entry = getEntryForType(type);

	if (entry) {
		entry->useMock = false;
	}
}

/*static*/ void Sensor::resetAllMocks() {
	constexpr size_t len = sizeof(s_sensorRegistry) / sizeof(s_sensorRegistry[0]);

	// Reset all mocks
	for (size_t i = 0; i < len; i++) {
		auto &entry = s_sensorRegistry[i];

		entry.useMock = false;
	}
}