Generator.hpp

Go to the documentation of this file.

#ifndef GENERATOR_H
#define GENERATOR_H

#include <memory>
#include <vector>
#include "Types.hpp"
#include "Tensor.hpp"
#include "InputData.hpp"
#include "SearchGraphNode.hpp"

class Generator {
public:

    Generator(InputData& inputData): inputData(inputData) {};

    std::shared_ptr<SearchGraphNode> generate(std::vector<int>& unassigned,
            Types::CarData& cars,
            std::shared_ptr<SearchGraphNode>& prevNode);

private:

    void createSearchSpace(Tensor<int>& finishTimes,
                           Tensor<int>& finishPoints,
                           Tensor<int>& scores,
                           std::vector<int>& unassigned,
                           Types::CarData& cars);

    std::shared_ptr<SearchGraphNode> selectFromSearchSpace(Types::CarData& cars,
            std::vector<int>& unassigned,
            std::shared_ptr<SearchGraphNode>& prevNode,
            Tensor<int>& finishTimes,
            Tensor<int>& finishPoints,
            Tensor<int>& scores);

    InputData& inputData;
};

#endif // GENERATOR_H