ball.h

Go to the documentation of this file.

// -*- C++ -*-
// BALL CLASS
#ifndef _BALL_H_
#define _BALL_H_
 
#include "internal.h"


typedef enum {green, blue, black} color_t;
static const char* colores[] = {"green", "blue", "black"};
static const char* colorsymb[] = {"g", "b", "o"};
static const name_t undeme = 0;
 
class node_t;


class ball_t {
 
private:
  node_t *_holder;
  node_t *_owner;
  name_t _deme;
 
public:
  name_t uniq;
  color_t color;
 
public:

  static const size_t bytesize = 2*sizeof(name_t)+sizeof(color_t);
  friend raw_t* operator>> (const ball_t &b, raw_t *o) {
    name_t buf[2] = {b.uniq, b._deme};
    memcpy(o,buf,sizeof(buf)); o += sizeof(buf);
    memcpy(o,&b.color,sizeof(color_t)); o += sizeof(color_t);
    return o;
  };

  friend raw_t* operator>> (raw_t *o, ball_t &b) {
    name_t buf[2];
    memcpy(buf,o,sizeof(buf)); o += sizeof(buf);
    b.uniq = buf[0]; b._deme = buf[1];
    memcpy(&b.color,o,sizeof(color_t)); o += sizeof(color_t);
    b._holder = 0;              // must be set elsewhere
    b._owner = 0;               // must be set elsewhere
    return o;
  };
 
public:

  ball_t (node_t *who = 0, name_t u = 0,
          color_t col = green, name_t d = undeme) {
    _holder = _owner = who;
    uniq = u;
    color = col;
    _deme = d;
  };

  ball_t (const ball_t&) = delete;
  ball_t (ball_t&&) = delete;
  ball_t & operator= (const ball_t&) = delete;
  ball_t & operator= (ball_t&&) = delete;
  ~ball_t (void) = default;
 
public:

  name_t deme (void) const {
    return _deme;
  };

  name_t& deme (void) {
    return _deme;
  };

  node_t* owner (void) const {
    assert(color==green);
    return _owner;
  };

  node_t*& owner (void) {
    assert(color==green);
    return _owner;
  };

  node_t* child (void) const {
    assert(color==green);
    return _owner;
  };

  node_t* holder (void) const {
    return _holder;
  };

  node_t*& holder (void) {
    return _holder;
  };

  bool is (color_t c) const {
    return color==c;
  };

  string_t color_name (void) const {
    return colores[color];
  };

  string_t color_symbol (void) const {
    if (is(green) && _holder==_owner)
      return "m";               // brown balls
    else
      return colorsymb[color];
  };
 
public:

  string_t yaml (string_t tab = "") const;
  SEXP structure (void) const;
  string_t newick (const slate_t &t, bool showdeme) const;
 
};
 
#endif