AdvancedPhysicalNodeFactory.h

Go to the documentation of this file.

/*****************************************************************************
 *                                                                           *
 *  Copyright 2018 Rice University                                           *
 *                                                                           *
 *  Licensed under the Apache License, Version 2.0 (the "License");          *
 *  you may not use this file except in compliance with the License.         *
 *  You may obtain a copy of the License at                                  *
 *                                                                           *
 *      http://www.apache.org/licenses/LICENSE-2.0                           *
 *                                                                           *
 *  Unless required by applicable law or agreed to in writing, software      *
 *  distributed under the License is distributed on an "AS IS" BASIS,        *
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *
 *  See the License for the specific language governing permissions and      *
 *  limitations under the License.                                           *
 *                                                                           *
 *****************************************************************************/

#ifndef PDB_ADVANCEDPHYSICALNODEFACTORY_H
#define PDB_ADVANCEDPHYSICALNODEFACTORY_H

#include <set>
#include "AbstractPhysicalNodeFactory.h"
#include "AdvancedPhysicalOptimizer/AdvancedPhysicalAbstractPipe.h"

namespace pdb {

class AdvancedPhysicalNodeFactory : public AbstractPhysicalNodeFactory {

 public:

  AdvancedPhysicalNodeFactory(const string &jobId,
                              const Handle<ComputePlan> &computePlan,
                              const ConfigurationPtr &conf);

  vector<AbstractPhysicalNodePtr> generateAnalyzerGraph(std::vector<AtomicComputationPtr> sources) override;

 protected:

  void transverseTCAPGraph(AtomicComputationPtr curNode);

  void setConsumers(shared_ptr<AdvancedPhysicalAbstractPipe> node);

  void connectThePipes();

  template <class T>
  void createPhysicalPipeline() {

    // this must never be empty
    assert(!currentPipe.empty());

    // create the node
    auto node = new T(jobId, computePlan, logicalPlan, conf, currentPipe, currentNodeIndex++);

    // create the node handle
    auto nodeHandle = node->getAdvancedPhysicalNodeHandle();

    // update all the node connections
    setConsumers(nodeHandle);

    // is this a source node
    if(nodeHandle->isSource()) {

      // add the source node
      physicalSourceNodes.push_back(nodeHandle);
    }

    for(auto &c : currentPipe) {
      std::cout << c->getOutputName() << std::endl;
    }

    std::cout << "-----------------------------------------" << std::endl;


    // add the starts with
    startsWith[currentPipe.front()->getOutputName()] = nodeHandle;

    // add the pipe to the physical nodes
    physicalNodes[nodeHandle->getNodeIdentifier()] = nodeHandle;
  }

  std::string jobId;

  size_t currentNodeIndex;

  ConfigurationPtr conf;

  std::set<AtomicComputationPtr> visitedNodes;

  std::vector<AtomicComputationPtr> currentPipe;

  std::map<std::string, AdvancedPhysicalPipelineNodePtr> physicalNodes;

  std::vector<AbstractPhysicalNodePtr> physicalSourceNodes;

  std::map<std::string, AdvancedPhysicalPipelineNodePtr> startsWith;

  std::map<std::string, std::vector<std::string>> consumedBy;

  std::vector<AdvancedPhysicalPipelineNodePtr> sources;
};

}



#endif //PDB_ADVANCEDPHYSICALNODEFACTORY_H