SortState.h

Go to the documentation of this file.

#ifndef EXECUTION_SORTSTATE_H
#define EXECUTION_SORTSTATE_H
 
#include "execution/PlanExecState.h"
#include "plan/Sort.h"
#include "extsort/ExternalSort.h"
 
namespace taco {
 
class SortState: public PlanExecState {
public:
    ~SortState() override;
 
    void init() override;
 
    bool next_tuple() override;
 
    void node_properties_to_string(std::string& buf, int indent) const override;
 
    std::vector<NullableDatumRef> get_record() override;
 
    void close() override;
 
    void rewind() override;
 
    Datum save_position() const override;
 
    bool rewind(DatumRef saved_position) override;
 
    const PlanNode*
    get_plan() const override {
        return m_plan;
    }
 
private:
    SortState(const Sort* plan, std::unique_ptr<PlanExecState>&& child);
 
    const Sort* m_plan;
 
    bool m_extracted;
    std::unique_ptr<ExternalSort> m_extsort;
 
    std::unique_ptr<Schema> m_input_schema;
    std::unique_ptr<ItemIterator> m_output_it;
    uint64_t m_output_init_saved_position;
    std::vector<Datum> m_record;
 
    friend class Sort;
};
 
}    // namespace taco
 
#endif