distflow.h

/*
 *  Copyright (c) 2002    National Research Council
 *
 *  All rights reserved.
 *
 *  This material is confidential and proprietary information of
 *  National Research Council Canada ("Confidential Information").
 *  This Confidential Information may only be used and reproduced
 *  in accordance with the terms of the license agreement.
 *
 */
 
#ifndef __INCLUDED_SCOPIRA_TOOL_DISTFLOW_H__
#define __INCLUDED_SCOPIRA_TOOL_DISTFLOW_H__

#include <vector>

#include <scopira/tool/flow.h>

namespace scopira
{
  namespace tool
  {
    class teeoflow;
    class distoflow;
  }
}

class scopira::tool::teeoflow : public scopira::tool::oflow_i
{
  protected:
    typedef scopira::tool::count2_ptr< scopira::tool::oflow_i > holder_t;

    holder_t dm_a, dm_b;    

  public:
    explicit teeoflow(bool doref);

    virtual bool failed(void) const;

    virtual size_t write(const byte_t* _buf, size_t _size);

    void set_flow_a(oflow_i* r);
    void set_flow_b(oflow_i* r);
};

class scopira::tool::distoflow : public scopira::tool::oflow_i
{
  protected:
    typedef std::vector< scopira::tool::count2_ptr< scopira::tool::oflow_i > > oflowvec_t;

    oflowvec_t dm_flow;
    bool dm_doref;

  public:
    explicit distoflow(bool doref);

    virtual bool failed(void) const;

    virtual size_t write(const byte_t* _buf, size_t _size);

    void add_flow(oflow_i* r);
    void remove_flow(oflow_i* r);
    void clear_flow(void);
    bool contains_flow(const oflow_i* r) const;
    size_t size_flow(void) const;
};

#endif