panda3d/panda/src/pstatclient/pStatClientImpl.h

// Filename: pStatClientImpl.h
// Created by:  drose (23Dec04)
//
////////////////////////////////////////////////////////////////////
//
// 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."
//
////////////////////////////////////////////////////////////////////

#ifndef PSTATCLIENTIMPL_H
#define PSTATCLIENTIMPL_H

#include "pandabase.h"

// This class doesn't exist at all unless DO_PSTATS is defined.
#ifdef DO_PSTATS

#include "pStatFrameData.h"
#include "connectionManager.h"
#include "queuedConnectionReader.h"
#include "connectionWriter.h"
#include "netAddress.h"

#include "trueClock.h"
#include "pmap.h"

class PStatClient;
class PStatServerControlMessage;
class PStatCollector;
class PStatCollectorDef;
class PStatThread;

////////////////////////////////////////////////////////////////////
//       Class : PStatClientImpl
// Description : This class is the implementation of the actual
//               PStatClient class (which is just for interface).  All
//               of the stuff to manage sending stats up to the server
//               is handled by this class.
//
//               This separation between PStatClient and
//               PStatClientImpl allows the global PStatClient to be
//               constructed at static init time, without having to
//               consult any config variables at that time.  We don't
//               actually do any real work until someone explicitly
//               calls PStatClient::connect().
//
//               This class doesn't exist at all unless DO_PSTATS is
//               defined.
////////////////////////////////////////////////////////////////////
class EXPCL_PANDA_PSTATCLIENT PStatClientImpl : public ConnectionManager {
public:
  PStatClientImpl(PStatClient *client);
  ~PStatClientImpl();

  INLINE void set_client_name(const string &name);
  INLINE string get_client_name() const;
  INLINE void set_max_rate(double rate);
  INLINE double get_max_rate() const;

  INLINE double get_real_time() const;

  INLINE void client_main_tick();
  bool client_connect(string hostname, int port);
  void client_disconnect();
  INLINE bool client_is_connected() const;

  INLINE void client_resume_after_pause();

  void new_frame(int thread_index);

private:
  void transmit_frame_data(int thread_index, int frame_number,
                           const PStatFrameData &frame_data);

  void transmit_control_data();

  TrueClock *_clock;
  double _delta;
  double _last_frame;

  // Networking stuff
  string get_hostname();
  void send_hello();
  void report_new_collectors();
  void report_new_threads();
  void handle_server_control_message(const PStatServerControlMessage &message);

  virtual void connection_reset(const PT(Connection) &connection, 
                                bool okflag);

  PStatClient *_client;

  bool _is_connected;
  bool _got_udp_port;

  NetAddress _server;
  QueuedConnectionReader _reader;
  ConnectionWriter _writer;

  PT(Connection) _tcp_connection;
  PT(Connection) _udp_connection;

  int _collectors_reported;
  int _threads_reported;

  string _hostname;
  string _client_name;
  double _max_rate;

  double _tcp_count_factor;
  double _udp_count_factor;
  unsigned int _tcp_count;
  unsigned int _udp_count;
};

#include "pStatClientImpl.I"

#endif  // DO_PSTATS

#endif