drag_statistics.h

Go to the documentation of this file.

// Copyright (c) 2020, the GRAPHGEN contributors, as
// shown by the AUTHORS file. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
 
#ifndef GRAPHGEN_DRAG_STATISTICS_H_
#define GRAPHGEN_DRAG_STATISTICS_H_
 
#include <iostream>
#include <set>
 
#include "conact_tree.h"
 
class BinaryDragStatistics {
    std::set<const BinaryDrag<conact>::node*> visited_nodes;
    std::set<const BinaryDrag<conact>::node*> visited_leaves;
 
    void PerformStatistics(const BinaryDrag<conact>::node *n) {
        if (n->isleaf()) {
            visited_leaves.insert(n);
            return;
        }
 
        if (visited_nodes.insert(n).second) {
            PerformStatistics(n->left);
            PerformStatistics(n->right);
        }
    }
 
public:
    BinaryDragStatistics(const BinaryDrag<conact>& bd) {
        for (const auto& t : bd.roots_) {
            PerformStatistics(t);
        }
    }
 
    auto Nodes() const { return visited_nodes.size(); }
 
    auto Leaves() const { return visited_leaves.size(); }
 
    void PrintLeaves(std::ostream& os = std::cout) {
        for (const auto& l : visited_leaves) {
            if (l->isleaf()) {
                for (const auto& a : l->data.actions()) {
                    os << a << " ";
                }
                os << "- " << l->data.next << "\n";
            }
        }
        os << "----------------------------------\n";
    }
};
 
void PrintStats(const BinaryDrag<conact>& bd, std::ostream& os = std::cout);
 
#endif // !GRAPHGEN_DRAG_STATISTICS_H_