open_toontown_panda3d/pandatool/src/pstatserver/pStatClientData.cxx

/**
 * PANDA 3D SOFTWARE
 * Copyright (c) Carnegie Mellon University.  All rights reserved.
 *
 * All use of this software is subject to the terms of the revised BSD
 * license.  You should have received a copy of this license along
 * with this source code in a file named "LICENSE."
 *
 * @file pStatClientData.cxx
 * @author drose
 * @date 2000-07-11
 */

#include "pStatClientData.h"
#include "pStatReader.h"

#include "pStatCollectorDef.h"

using std::string;

PStatCollectorDef PStatClientData::_null_collector(-1, "Unknown");



/**
 *
 */
PStatClientData::
PStatClientData(PStatReader *reader) :
  _reader(reader)
{
  _is_alive = true;
}

/**
 *
 */
PStatClientData::
~PStatClientData() {
  Collectors::const_iterator ci;
  for (ci = _collectors.begin(); ci != _collectors.end(); ++ci) {
    delete (*ci)._def;
  }
}

/**
 * Returns true if the data is actively getting filled by a connected client,
 * or false if the client has terminated.
 */
bool PStatClientData::
is_alive() const {
  return _is_alive;
}

/**
 * Closes the client connection if it is open.
 */
void PStatClientData::
close() {
  if (_is_alive && _reader != nullptr) {
    _reader->close();
    _reader = nullptr;
    _is_alive = false;
  }
}

/**
 * Returns the total number of collectors the Data knows about.
 */
int PStatClientData::
get_num_collectors() const {
  return _collectors.size();
}

/**
 * Returns true if the indicated collector has been defined by the client
 * already, false otherwise.  It is possible for the client to start streaming
 * data before all of the collectors have been defined.
 */
bool PStatClientData::
has_collector(int index) const {
  return (index >= 0 && index < (int)_collectors.size() &&
          _collectors[index]._def != nullptr);
}

/**
 * Returns the nth collector definition.
 */
const PStatCollectorDef &PStatClientData::
get_collector_def(int index) const {
  if (!has_collector(index)) {
    return _null_collector;
  }
  return *_collectors[index]._def;
}

/**
 * Returns the name of the indicated collector.
 */
string PStatClientData::
get_collector_name(int index) const {
  if (!has_collector(index)) {
    return "Unknown";
  }
  const PStatCollectorDef *def = _collectors[index]._def;
  return def->_name;
}

/**
 * Returns the "full name" of the indicated collector.  This will be the
 * concatenation of all of the collector's parents' names (except Frame) and
 * the collector's own name.
 */
string PStatClientData::
get_collector_fullname(int index) const {
  if (!has_collector(index)) {
    return "Unknown";
  }

  const PStatCollectorDef *def = _collectors[index]._def;
  if (def->_parent_index == 0) {
    return def->_name;
  } else {
    return get_collector_fullname(def->_parent_index) + ":" + def->_name;
  }
}

/**
 * Indicates whether the given collector has level data (and consequently,
 * whether it should appear on the Levels menu).
 *
 * The return value is true if anything changed, false otherwise.
 */
bool PStatClientData::
set_collector_has_level(int index, int thread_index, bool flag) {
  bool any_changed = false;
  slot_collector(index);
  nassertr(index >= 0 && index < (int)_collectors.size(), false);

  if (_collectors[index]._is_level.get_bit(thread_index) != flag) {
    any_changed = true;
    _collectors[index]._is_level.set_bit_to(thread_index, flag);
  }

  // Turning this on for a given collector also implicitly turns all of its
  // ancestors.
  if (flag) {
    PStatCollectorDef *def = _collectors[index]._def;
    if (def != nullptr && def->_parent_index != 0) {
      if (set_collector_has_level(def->_parent_index, thread_index, flag)) {
        any_changed = true;
      }
    }
  }

  return any_changed;
}


/**
 * Returns whether the given collector has level data (and consequently,
 * whether it should appear on the Levels menu).
 */
bool PStatClientData::
get_collector_has_level(int index, int thread_index) const {
  return (index >= 0 && index < (int)_collectors.size() &&
          _collectors[index]._is_level.get_bit(thread_index));
}

/**
 * Returns the total number of collectors that are toplevel collectors.  These
 * are the collectors that are the children of "Frame", which is collector 0.
 */
int PStatClientData::
get_num_toplevel_collectors() const {
  return _toplevel_collectors.size();
}

/**
 * Returns the collector index of the nth toplevel collector.  Use this
 * function to iterate through the n toplevel collectors indicated by
 * get_num_toplevel_collectors().
 */
int PStatClientData::
get_toplevel_collector(int n) const {
  nassertr(n >= 0 && n < (int)_toplevel_collectors.size(), 0);
  return _toplevel_collectors[n];
}

/**
 * Returns the total number of threads the Data knows about.
 */
int PStatClientData::
get_num_threads() const {
  return _threads.size();
}

/**
 * Returns true if the indicated thread has been defined by the client
 * already, false otherwise.  It is possible for the client to start streaming
 * data before all of the threads have been defined.
 */
bool PStatClientData::
has_thread(int index) const {
  return (index >= 0 && index < (int)_threads.size() &&
          !_threads[index]._name.empty());
}

/**
 * Returns the name of the indicated thread.
 */
string PStatClientData::
get_thread_name(int index) const {
  if (!has_thread(index)) {
    return "Unknown";
  }
  return _threads[index]._name;
}

/**
 * Returns the data associated with the indicated thread.  This will create a
 * thread definition if it does not already exist.
 */
const PStatThreadData *PStatClientData::
get_thread_data(int index) const {
  ((PStatClientData *)this)->define_thread(index);
  nassertr(index >= 0 && index < (int)_threads.size(), nullptr);
  return _threads[index]._data;
}

/**
 * Returns the number of Collectors between the indicated parent and the child
 * Collector in the relationship graph.  If child is the same as parent,
 * returns zero.  If child is an immediate child of parent, returns 1.  If
 * child is a grandchild of parent, returns 2, and so on.  If child is not a
 * descendant of parent at all, returns -1.
 */
int PStatClientData::
get_child_distance(int parent, int child) const {
  if (parent == child) {
    return 0;
  }
  if (!has_collector(child) || child == 0) {
    return -1;
  }
  int dist = get_child_distance(parent, get_collector_def(child)._parent_index);
  if (dist == -1) {
    return -1;
  } else {
    return dist + 1;
  }
}

/**
 * Adds a new collector definition to the dataset.  Presumably this is
 * information just arrived from the client.
 *
 * The pointer will become owned by the PStatClientData object and will be
 * freed on destruction.
 */
void PStatClientData::
add_collector(PStatCollectorDef *def) {
  slot_collector(def->_index);
  nassertv(def->_index >= 0 && def->_index < (int)_collectors.size());

  if (_collectors[def->_index]._def != nullptr) {
    // Free the old definition, if any.
    delete _collectors[def->_index]._def;
  }

  _collectors[def->_index]._def = def;
  update_toplevel_collectors();

  // If we already had the _is_level flag set, it should be immediately
  // applied to all ancestors.
  const BitArray &is_level = _collectors[def->_index]._is_level;
  int max_threads = is_level.get_num_bits();
  for (int thread_index = 0; thread_index < max_threads; ++thread_index) {
    if (is_level.get_bit(thread_index)) {
      set_collector_has_level(def->_parent_index, thread_index, true);
    }
  }
}

/**
 * Adds a new thread definition to the dataset.  Presumably this is
 * information just arrived from the client.
 */
void PStatClientData::
define_thread(int thread_index, const string &name) {
  // A sanity check on the index number.
  nassertv(thread_index < 1000);

  // Make sure we have enough slots allocated.
  while ((int)_threads.size() <= thread_index) {
    _threads.push_back(Thread());
  }

  if (!name.empty()) {
    _threads[thread_index]._name = name;
  }

  if (_threads[thread_index]._data.is_null()) {
    _threads[thread_index]._data = new PStatThreadData(this);
  }
}


/**
 * Makes room for and stores a new frame's worth of data associated with some
 * particular thread (which may or may not have already been defined).
 *
 * The pointer will become owned by the PStatThreadData object and will be
 * freed on destruction.
 */
void PStatClientData::
record_new_frame(int thread_index, int frame_number,
                 PStatFrameData *frame_data) {
  define_thread(thread_index);
  nassertv(thread_index >= 0 && thread_index < (int)_threads.size());
  _threads[thread_index]._data->record_new_frame(frame_number, frame_data);
}

/**
 * Makes sure there is an entry in the array for a collector with the given
 * index number.
 */
void PStatClientData::
slot_collector(int collector_index) {
  // A sanity check on the index number.
  nassertv(collector_index < 100000);

  while ((int)_collectors.size() <= collector_index) {
    Collector collector;
    collector._def = nullptr;
    _collectors.push_back(collector);
  }
}

/**
 * Rebuilds the list of toplevel collectors.
 */
void PStatClientData::
update_toplevel_collectors() {
  _toplevel_collectors.clear();

  Collectors::const_iterator ci;
  for (ci = _collectors.begin(); ci != _collectors.end(); ++ci) {
    PStatCollectorDef *def = (*ci)._def;
    if (def != nullptr && def->_parent_index == 0) {
      _toplevel_collectors.push_back(def->_index);
    }
  }
}