ParallelJobGraph.h

Go to the documentation of this file.

// ========================================================================= //
// This file is part of MyraMath, copyright (c) 2014-2019 by Ryan A Chilton  //
// and distributed by MyraCore, LLC. See LICENSE.txt for license terms.      //
// ========================================================================= //

#ifndef MYRAMATH_JOBGRAPH_PARALLELJOBGRAPH_H
#define MYRAMATH_JOBGRAPH_PARALLELJOBGRAPH_H

#include <myramath/MYRAMATH_EXPORT.h>

#include <myramath/jobgraph/Job.h>
#include <myramath/jobgraph/JobID.h>
#include <myramath/jobgraph/JobGraph.h>
#include <myramath/jobgraph/Options.h>

#include <string>
#include <stdint.h>

namespace myra {

class MYRAMATH_EXPORT ParallelJobGraph : public JobGraphBase
  {
  public:

    // --------------------------- Container API.

    ParallelJobGraph();

    int insert(const JobGraph& g);
    
    const JobGraph& at(int g) const;

    // --------------------------- Base JobGraph overrides.

    virtual uint64_t n_work() const;
    
    virtual std::string name() const;

    virtual void begins(JobIDs& output) const;

    virtual void ends(JobIDs& output) const;

    virtual ::myra::Job* create(JobID j);

    virtual std::vector<Job*> create();

    uint64_t size() const;

    virtual ::myra::JobGraphBase* clone() const;

    virtual ~ParallelJobGraph();

  private:

    // Internal contents.
    typedef std::vector<JobGraph> Subgraphs;
    Subgraphs subgraphs;

    // Implementation detail, a Job that delegates everything to an underlying Job from one of the contained subgraphs[]
    class ParallelJob : public ::myra::Job
      {
      private:
      
        // Underlying Job.
        ::myra::Job* job;

        // Which subgraph it came from.
        int g;

        // Host graph.
        ParallelJobGraph* host;
        
      public:

        // Constructor, requires underlying Job* and host ParallelJobGraph*.
        ParallelJob(::myra::Job* in_job, int in_g, ParallelJobGraph* in_host);

        // Delegates to job->id(), adds offset.
        virtual JobID id() const;

        // Delegates to job->parents(), adds offsets to each.
        virtual void parents(JobIDs& output) const;

        // Delegates to job->children(), adds offsets to each.
        virtual void children(JobIDs& output) const;

        // Returns job->execute()
        virtual uint64_t execute();

        // Returns job->name()
        virtual std::string name() const;

        // Deletes job.
        virtual ~ParallelJob();

      };

    // This implementation detail is a friend.
    friend class ParallelJob;
    
  };

MYRAMATH_EXPORT JobGraph parallelize(const JobGraph& g0, const JobGraph& g1);
MYRAMATH_EXPORT JobGraph parallelize(const JobGraph& g0, const JobGraph& g1, const JobGraph& g2);
MYRAMATH_EXPORT JobGraph parallelize(const JobGraph& g0, const JobGraph& g1, const JobGraph& g2, const JobGraph& g3);
MYRAMATH_EXPORT JobGraph parallelize(const JobGraph& g0, const JobGraph& g1, const JobGraph& g2, const JobGraph& g3, const JobGraph& g4);

} // namespace myra

#endif