FusedJobGraph.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_FUSEDJOBGRAPH_H
#define MYRAMATH_JOBGRAPH_FUSEDJOBGRAPH_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 <map>
#include <vector>
#include <string>
#include <stdint.h>

namespace myra {

class FusedJobGraph;

class MYRAMATH_EXPORT FusedJobGraph : public JobGraphBase
  {
  public:

    // --------------------------- Graph/container API.

    FusedJobGraph();

    int insert(const JobGraph& g);

    const JobGraph& at(int g) const;

    void add_edge(int g0, JobID j0, int g1, JobID j1);

    // --------------------------- JobGraphBase obligations.

    virtual JobGraphBase* clone() const;

    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 ~FusedJobGraph();
  
  private:

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

    // Useful typedef.
    typedef std::map<JobID,JobIDs> Map;

    // Dependencies, forward view of underlying bimap.
    Map forward;

    // Dependencies, reverse view of underlying bimap.
    Map reverse;

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

        // Which subgraph it came from.
        int g;

        // Host graph.
        FusedJobGraph* host;
        
      public:

        // Constructor, requires underlying Job* j and host graph g.
        FusedJob(::myra::Job* in_j, int in_g, FusedJobGraph* in_host);

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

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

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

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

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

        // Deletes job.
        virtual ~FusedJob();

      };

    // This implementation detail is a friend.
    friend class FusedJob;

  };

} // namespace myra

#endif