EIC code displayed by LXR master/​jana2/​src/​libraries/​JANA/​Topology/​JEventQueue.h	Source navigation Diff markup Identifier search General search
	Version: master test Revert   Change

File indexing completed on 2025-12-16 10:36:45

 
 // Copyright 2024, Jefferson Science Associates, LLC.
 // Subject to the terms in the LICENSE file found in the top-level directory.
 // Author: Nathan Brei
 
 #pragma once
 #include <JANA/Utils/JCpuInfo.h>
 #include <JANA/JEvent.h>
 
 #include <memory>
 #include <vector>
 #include <cassert>
 
 // JEventQueue is a queue designed for communication between JArrows, with the following features:
 //
 // - Fixed-capacity during processing, since the max number of events in flight is predetermined and small
 // - Ringbuffer-based, to avoid allocations during processing
 // - Adjustable-capacity outside processing, since the max number of events in flight may change when the threadcount does
 // - Locality-aware, so that events that live in one location (e.g. NUMA domain) can stay within that location
 // - NOT thread-safe, because all queue accesses are protected by the JExecutionEngine mutex.
 //
 // - Previous versions of JEventQueue have allowed work stealing (taking events out of a different 
 //   NUMA domain when none are otherwise available). The current version does not, though we may
 //   bring this feature back later.
 //
 /// To handle memory locality at different granularities, we introduce the concept of a location.
 /// Each thread belongs to exactly one location, represented by contiguous unsigned
 /// ints starting at 0. While JArrows are wired to one logical JEventQueue, threads interact with
 /// the physical LocalQueue corresponding to their location. Locations prevent events from crossing 
 /// NUMA domains as they get picked up by different worker threads.
 
 class JEventQueue {
 
 protected:
     struct alignas(JANA2_CACHE_LINE_BYTES) LocalQueue {
         std::vector<JEvent*> ringbuffer;
         size_t size=0;
         size_t capacity=0;
         size_t front=0;
         size_t back=0;
     };
 
     std::vector<std::unique_ptr<LocalQueue>> m_local_queues;
     size_t m_capacity = 0;
 
     // Order-establishing state
     bool m_establishes_ordering = false;
     size_t m_next_event_index = 0;
 
     // Order-enforcing state
     bool m_enforces_ordering = false;
     size_t m_next_slot = 0;
     int m_min_index = 0;
     int m_max_index = 0;
 
 public:
     inline JEventQueue(size_t initial_capacity, size_t locations_count) {
 
         assert(locations_count >= 1);
         for (size_t location=0; location<locations_count; ++location) {
             m_local_queues.push_back(std::make_unique<LocalQueue>());
         }
         Scale(initial_capacity);
     }
 
     virtual ~JEventQueue() = default;
 
 
     void SetEstablishesOrdering(bool establishes_ordering=true) {
         m_establishes_ordering = establishes_ordering;
     }
 
     void SetEnforcesOrdering(bool enforces_ordering=true) {
         m_enforces_ordering = enforces_ordering;
     }
 
     bool GetEstablishesOrdering() const { 
         return m_establishes_ordering;
     }
 
     bool GetEnforcesOrdering() const {
         return m_enforces_ordering;
     }
 
     virtual void Scale(size_t capacity) {
         if (capacity < m_capacity) {
             for (auto& local_queue : m_local_queues) {
                 if (local_queue->size != 0) {
                     throw JException("Attempted to shrink a non-empty JEventQueue. Please drain the topology before attempting to downscale.");
                 }
             }
         }
         m_capacity = capacity;
         m_max_index = capacity - 1; // zero-indexed
         for (auto& local_queue: m_local_queues) {
             local_queue->ringbuffer.resize(capacity, nullptr);
             local_queue->capacity = capacity;
         }
     }
 
     inline size_t GetLocationCount() {
         return m_local_queues.size();
     }
 
     inline size_t GetSize(size_t location) {
         auto& local_queue = m_local_queues[location];
         return local_queue->size;
     }
 
     inline size_t GetCapacity() {
         return m_capacity;
     }
 
     inline void Push(JEvent* event, size_t location) {
 
         if (m_enforces_ordering) {
             // Repurpose m_local_queues[0] as an associative array instead of a queue
             auto index = event->GetEventIndex();
             if (m_max_index < index) {
                 throw JException("Event index=%lu is above max=%lu", index, m_max_index);
             }
             if (m_min_index > index) {
                 throw JException("Event index=%lu is below min=%lu", index, m_min_index);
             }
             size_t slot = index % m_capacity;
             auto& buffer = m_local_queues[0]->ringbuffer;
 
             if (buffer[slot] != nullptr) {
                 throw JException("Collision when pushing to ordered queue. slot=%lu", slot);
             }
             buffer[slot] = event;
         }
         else {
             // Use local_queue as intended
             auto& local_queue = *m_local_queues[location];
             if (local_queue.size == local_queue.capacity) {
                 throw JException("Attempted to push to a full JEventQueue. This probably means there is an error in your topology wiring");
             }
 
             local_queue.ringbuffer[local_queue.front] = event;
             local_queue.front = (local_queue.front + 1) % local_queue.capacity;
             local_queue.size += 1;
         }
     }
 
     inline JEvent* Pop(size_t location) {
         if (m_enforces_ordering) {
             auto& buffer = m_local_queues[0]->ringbuffer;
             auto event = buffer[m_next_slot];
             if (event == nullptr) {
                 return nullptr;
             }
             buffer[m_next_slot] = nullptr;
             m_min_index += 1;
             m_max_index += 1;
             m_next_slot += 1;
             m_next_slot %= m_capacity;
             return event;
         }
         else {
             auto& local_queue= *m_local_queues[location];
             if (local_queue.size == 0) {
                 return nullptr;
             }
             JEvent* result = local_queue.ringbuffer[local_queue.back];
             local_queue.ringbuffer[local_queue.back] = nullptr;
             local_queue.back = (local_queue.back + 1) % local_queue.capacity;
             local_queue.size -= 1;
 
             if (m_establishes_ordering) {
                 result->SetEventIndex(m_next_event_index);
                 m_next_event_index += 1;
             }
             return result;
         }
     };
 
 };
 
 

This page was automatically generated by the 2.3.7 LXR engine. The LXR team