example_traits.h

Go to the documentation of this file.

/*
 *  Copyright (C) 2012 Josh Bialkowski (jbialk@mit.edu)
 *
 *  This file is part of mpblocks.
 *
 *  mpblocks is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  mpblocks is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with mpblocks.  If not, see <http://www.gnu.org/licenses/>.
 */
#ifndef MPBLOCKS_BTPS_EXAMPLETRAITS_H_
#define MPBLOCKS_BTPS_EXAMPLETRAITS_H_

#include <mpblocks/btps/node.h>
#include <cstdint>

namespace mpblocks {
namespace     btps {

struct ExampleTraits {
  struct Node;

  typedef Node* NodeRef;

  struct Node : BasicNode<ExampleTraits> {
    Node(double weight = 0) : BasicNode<ExampleTraits>(weight), freq(0) {}

    uint32_t freq;  
  };

  struct NodeOps {
    NodeRef& Parent(NodeRef N) { return N->parent; }

    NodeRef& LeftChild(NodeRef N) { return N->left; }

    NodeRef& RightChild(NodeRef N) { return N->right; }

    double Weight(NodeRef N) { return N->weight; }

    double& CumulativeWeight(NodeRef N) { return N->cumweight; }

    uint32_t& Count(NodeRef N) { return N->count; }
  };
};

} //< namespace btps
} //< namespace mpblocks














#endif // EXAMPLETRAITS_H_