update sources to ceph Nautilus 14.2.1

[ceph.git] / ceph / src / dmclock / support / test / test_ind_intru_heap.cc

// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab

/*
 * Copyright (C) 2016 Red Hat Inc.
 *
 * Author: J. Eric Ivancich <ivancich@redhat.com>
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License version
 * 2.1, as published by the Free Software Foundation.  See file
 * COPYING.
 */


#include <memory>
#include <string>
#include <iostream>

#include "indirect_intrusive_heap.h"


class TestCompare;


class Test1 {
    friend TestCompare;

    int data;

public:

    crimson::IndIntruHeapData heap_data;

    explicit Test1(int _data) : data(_data) {}

    friend std::ostream& operator<<(std::ostream& out, const Test1& d) {
        out << d.data << " (" << d.heap_data << ")";
        return out;
    }

    int& the_data() { return data; }
};


struct TestCompare {
    bool operator()(const Test1& d1, const Test1& d2) {
        return d1.data < d2.data;
    }
};


int main(int argc, char** argv) {
    Test1 d1(2);
    Test1 d2(3);
    Test1 d3(1);
    Test1 d4(-5);

    crimson::IndIntruHeap<std::shared_ptr<Test1>, Test1, &Test1::heap_data, TestCompare> my_heap;

    const std::shared_ptr<Test1> d99 = std::make_shared<Test1>(99);

    my_heap.push(std::make_shared<Test1>(2));
    my_heap.push(d99);
    my_heap.push(std::make_shared<Test1>(1));
    my_heap.push(std::make_shared<Test1>(-5));
    my_heap.push(std::make_shared<Test1>(12));
    my_heap.push(std::make_shared<Test1>(-12));
    my_heap.push(std::make_shared<Test1>(-7));

    std::cout << my_heap << std::endl;

    auto& t = my_heap.top();
    t.the_data() = 17;
    my_heap.adjust_down(t);

    std::cout << my_heap << std::endl;

    my_heap.display_sorted(std::cout);

    while (!my_heap.empty()) {
        auto& top = my_heap.top();
        std::cout << top << std::endl;
        my_heap.pop();
        std::cout << my_heap << std::endl;
    }

    return 0;
}