---

CompositeSource.cxx

Go to the documentation of this file.

// $Header: /nfs/slac/g/glast/ground/cvs/flux/src/CompositeSource.cxx,v 1.9 2001/07/25 03:54:53 burnett Exp $

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include "../flux/CompositeSource.h"  //TAKE THE /.. OUT!

#include "dom/DOM_Element.hpp"
#include "facilities/Scheduler.h"
#include "facilities/SimpleEvent.h"
#include "CLHEP/Random/RandFlat.h"

// see coment below: #include "control/EventLoop.h"


#include <strstream>
#include <cassert>
#include <numeric> // for accumulate
#include <functional>
#include <iomanip>

CompositeSource::CompositeSource (double aRate)
: EventSource(aRate), m_recent(0),m_numofiters(0)
{
}

CompositeSource::~CompositeSource()
{
    for (std::vector<EventSource*>::iterator it = m_sourceList.begin();
         it != m_sourceList.end(); ++it ) delete (*it);
}


void CompositeSource::addSource (EventSource* aSource)
{
    m_sourceList.push_back(aSource);
    EventSource::flux( flux() );
}

void CompositeSource::rmvSource (EventSource* aSource)
{
    std::vector<EventSource*>::iterator   it = m_sourceList.begin();
    for (;it != m_sourceList.end(); ++it) {
        if ((*it) == aSource)   break;
    }
    if (it != m_sourceList.end()) {
        m_sourceList.erase(it);
        EventSource::flux( flux() );
    }
}

FluxSource* CompositeSource::event ()
{
        m_numofiters=0;
    double mr = rate();

    if( m_sourceList.size()==1 || mr ==0) {
        m_recent = m_sourceList.front();
    }else {

        // more than one:: choose on basis of relative rates
        double  x = RandFlat::shoot(mr), y = 0;
        std::vector<EventSource*>::iterator  it = m_sourceList.begin();
        for (; it != m_sourceList.end(); ++it) {
            y += fabs((*it)->rate());
            if (x <= y) {
                m_recent = (*it);
                break;
            }
            m_numofiters++;
        }
    }
    // now ask the chosen one to generate the event, if there is a rate
    return m_recent->event();
}

std::string CompositeSource::fullTitle () const
{
    std::strstream  s;
    std::vector<EventSource*>::const_iterator   it = m_sourceList.begin();

    while (it != m_sourceList.end()) {
        
        s << (*it)->fullTitle() << " ";
        ++it;
        if (it != m_sourceList.end())    s << "+ ";
    }
    s << '\0';
    std::string t(s.str());
    s.freeze(false);
    return t;
}

std::string CompositeSource::displayTitle () const
{
    return (m_recent == 0) ? "" : m_recent->displayTitle();
}

double CompositeSource::rate() const
{
    std::vector<EventSource*>::const_iterator it = m_sourceList.begin();
    double      total_rate = 0.;
    for(;it != m_sourceList.end();++it) {
        double rr = fabs((*it)->rate());
        total_rate += rr;
    }
    return total_rate;
}

void    CompositeSource::rate ( double value )
{
    double  f = rate();
    if (f == 0.)    return;

    std::vector<float>  fvec;
    std::vector<EventSource*>::iterator it = m_sourceList.begin();

    while (it != m_sourceList.end())    {

        (*it)->rate( value * (*it)->rate()/f );
        ++it;
    }
    EventSource::rate( value );
}

// implement virtual function
void CompositeSource::setupXML (const DOM_Element&) {}

void CompositeSource::printOn(std::ostream& out)const
{
    out << "Source(s), total rate="<< rate() << std::endl;

    for( std::vector<EventSource*>::const_iterator it = m_sourceList.begin();
        it != m_sourceList.end();++it)  {
        out <<  std::setw(8) << std::setprecision(4) << (*it)->rate() <<" Hz, "
            << '#' << std::setw(6) << (*it)->eventNumber() <<' '
            << (*it)->name() << ' '<< (*it)->fullTitle() << std::endl;

    }

}

std::string CompositeSource::findSource()const
{
        return m_recent->fullTitle();
}

int CompositeSource::numSource()const
{
    return m_numofiters;
}
          

---