PDG_WorkItemStateTracker.h

Go to the documentation of this file.

/*
* PROPRIETARY INFORMATION.  This software is proprietary to
* Side Effects Software Inc., and is not to be reproduced,
* transmitted, or disclosed in any way without written permission.
*
* COMMENTS:
*/

#ifndef __PDG_WORKITEM_STATETRACKER_H__
#define __PDG_WORKITEM_STATETRACKER_H__

#include "PDG_API.h"
#include "PDG_Types.h"

#include <SYS/SYS_AtomicInt.h>

/*
 * Helper class that maintains a list of work item state counts
 */
class PDG_API PDG_WorkItemStateTracker
{
public:
                            PDG_WorkItemStateTracker()
                                { reset(); }

    inline void             reset()
                                {
                                    myWorkItemCount.store(0);
                                    for (int i = 0;
                                        i < (int)PDG_WorkItemState::eStateCount;
                                        i++)
                                    {
                                        myStateCounts[i].store(0);
                                    }
                                }

    inline int              workItemCount() const
                                { return myWorkItemCount.relaxedLoad(); }

    inline int              stateCount(PDG_WorkItemState state) const
                                { return stateCounter(state).relaxedLoad(); }

    inline void             addWorkItem(PDG_WorkItemState state)
                                {
                                    stateCounter(state).add(1);
                                    myWorkItemCount.add(1);
                                }

    inline void             removeWorkItem(PDG_WorkItemState state)
                                {
                                    stateCounter(state).add(-1);
                                    myWorkItemCount.add(-1);
                                }

    inline void             updateWorkItem(
                                    PDG_WorkItemState prev,
                                    PDG_WorkItemState cur)
                                {
                                    stateCounter(prev).add(-1);
                                    stateCounter(cur).add(1);
                                }

private:
    SYS_AtomicInt32&        stateCounter(PDG_WorkItemState state)
                                { return myStateCounts[(int)state]; }
    const SYS_AtomicInt32&  stateCounter(PDG_WorkItemState state) const
                                { return myStateCounts[(int)state]; }

private:
    SYS_AtomicInt32         myWorkItemCount;
    SYS_AtomicInt32         myStateCounts[(int)PDG_WorkItemState::eStateCount];
};

#endif