Hook.cpp

#include "Hook.h"
#include "Snapshot.h"
#include "ResourceHandler.h"
#include "CVs/CVManager.h"
#include <algorithm>

namespace SSAGES
{
    void Hook::PreSimulationHook()
    {
        snapshot_->Changed(false);
    
        // Initialize/evaluate CVs.
        for(auto& cv : cvmanager_->GetCVs())
        {
            cv->Initialize(*snapshot_);
            cv->Evaluate(*snapshot_);
        }

        // Call presimulation method on listeners. 
        for(auto& listener : listeners_)
            listener->PreSimulation(snapshot_, *cvmanager_);

        // Sync snapshot to engine.
        if(snapshot_->HasChanged())
            SyncToEngine();

        snapshot_->Changed(false);      
    }

    void Hook::PostIntegrationHook()
    {
        snapshot_->Changed(false);

        for(auto& cv : cvmanager_->GetCVs())
            cv->Evaluate(*snapshot_);

        for(auto& listener : listeners_)
            if(snapshot_->GetIteration() % listener->GetFrequency() == 0)
                listener->PostIntegration(snapshot_, *cvmanager_);

        if(snapshot_->HasChanged())
            SyncToEngine();

        snapshot_->Changed(false);      
    }

    void Hook::PostSimulationHook()
    {
        snapshot_->Changed(false);

        for(auto& cv : cvmanager_->GetCVs())
            cv->Evaluate(*snapshot_);
        
        for(auto& listener : listeners_)
            listener->PostSimulation(snapshot_, *cvmanager_);

        if(snapshot_->HasChanged())
            SyncToEngine();

        snapshot_->Changed(false);      
    }

    void Hook::SetSnapshot(Snapshot* snapshot)
    {
        snapshot_ = snapshot;
    }

    void Hook::SetCVManager(CVManager* cvmanager)
    {
        cvmanager_ = cvmanager;
    }


    void Hook::AddListener(EventListener* listener)
    {
        if(std::find(listeners_.begin(), listeners_.end(), listener) == listeners_.end())
            listeners_.push_back(listener);
    }
}