// $Id$
// ============================================================================
//
// = LIBRARY
//    TAO/performance-tests/Demux
//
// = FILENAME
//    demux_test_server.cpp
//
// = AUTHOR
//
//    Aniruddha Gokhale
//
// ============================================================================

#include "demux_test_server.h"

ACE_RCSID(CodeGen, demux_test_server, "$Id$")

// Constructor
Demux_Test_Server::Demux_Test_Server (void)
  : argc_ (0),
    argv_ (0),
    num_POAs_ (1),
    // default number of child POAs is 1 and each one will always have 1 object 
    num_objs_ (1),
    poa_fp_ (0),
    ior_fp_ (0)
{
}

// destructor
Demux_Test_Server::~Demux_Test_Server (void)
{
  ACE_OS::fclose (this->poa_fp_);
  ACE_OS::fclose (this->ior_fp_);
}

//
// initialize the Demux_Test_Server
//

int 
Demux_Test_Server::init (int argc, char *argv [], CORBA::Environment &env)
{
  this->argc_ = argc;
  this->argv_ = argv;
  
  // Grab the ORB
  TAO_TRY_EX (GET_ORB)
    {
      char *orb_name = "internet"; // unused by TAO

      // get the underlying ORB
      this->orb_ = CORBA::ORB_init (argc, argv, orb_name, TAO_TRY_ENV);
      TAO_CHECK_ENV_EX (GET_ORB);
    }
  TAO_CATCHANY
    {
      TAO_TRY_ENV.print_exception ("ORB_init");
      env.exception (TAO_TRY_ENV.exception ());
      return -1;
    }
  TAO_ENDTRY;

  // Grab the ROOT POA
  TAO_TRY_EX (GET_ROOT_POA)
    {
      CORBA::Object_var temp;      // holder for the myriad of times we get
                                   // an object which we then have to narrow.
      // Get the Root POA

      temp = this->orb_->resolve_initial_references ("RootPOA");
      if (CORBA::is_nil (temp.in ()))
        ACE_ERROR_RETURN ((LM_ERROR,
                           "(%P|%t) Unable to get root poa reference.\n"),
                          1);

      this->root_poa_ = PortableServer::POA::_narrow (temp.in (), TAO_TRY_ENV);
      TAO_CHECK_ENV_EX (GET_ROOT_POA);
    }
  TAO_CATCHANY
    {
      TAO_TRY_ENV.print_exception ("PortableServer::POA::_narrow");
      env.exception (TAO_TRY_ENV.exception ());
      return -1;
    }
  TAO_ENDTRY;

  // grab the POA Manager
  TAO_TRY_EX (GET_POA_MGR)
    {
      
      this->poa_mgr_ = this->root_poa_->the_POAManager (TAO_TRY_ENV);
      TAO_CHECK_ENV_EX (GET_POA_MGR);
    }
  TAO_CATCHANY
    {
      TAO_TRY_ENV.print_exception ("RootPOA->the_POAManager");
      env.exception (TAO_TRY_ENV.exception ());
      return -1;
    }
  TAO_ENDTRY;

  // now parse the rest of the arguments to determine the POA depth, the number
  // of objects with each POA and other info

  if (this->parse_args () == -1)
    ACE_ERROR_RETURN ((LM_ERROR,
                       "(%N:%l) Demux_Test_Server::init - "
                       "parse_args failed\n"),
                      -1);
  
  // init the Policies used by all the POAs
  CORBA::PolicyList policies (2);

  TAO_TRY_EX (POLICY)
    {
      // The id_uniqueness_policy by default is UNIQUE_ID. So each of our servants
      // will have a unique name

      policies.length (2);

      // let the system assign us the ID. This way we will be able to use
      // active demultiplexing
      policies[0] =
        this->root_poa_->create_id_assignment_policy (PortableServer::SYSTEM_ID,
                                                      TAO_TRY_ENV);

      TAO_CHECK_ENV_EX (POLICY);

      // servants are persistent
      policies[1] =
        this->root_poa_->create_lifespan_policy (PortableServer::PERSISTENT,
                                                 TAO_TRY_ENV);
      TAO_CHECK_ENV_EX (POLICY);
    }
  TAO_CATCHANY
    {
      TAO_TRY_ENV.print_exception ("creating policy");
      env.exception (TAO_TRY_ENV.exception ());
      return -1;
    }
  TAO_ENDTRY;

      
  // now create a POA hierarchy of the desired depth and populate each POA with
  // the specified number of objects. Finally, activate these objects.
  
  // open the file that has all the POA names in it
  char poa_file [128];
  ACE_OS::sprintf (poa_file, "poa_names_%d.dat", this->num_POAs_);
  if ((this->poa_fp_ = ACE_OS::fopen (poa_file, "r")) == 0)
    {
      ACE_ERROR_RETURN ((LM_ERROR,
                         " (%P|%t) Unable to open POA file %s\n", poa_file),
                        -1);
    }

  // loop indices
  CORBA::ULong i, j;

  PortableServer::POA *prev_poa = this->root_poa_.in ();
  for (i = 0; i < this->num_POAs_; i++)
    {
      char poa_name [128];
      
      ACE_OS::memset (poa_name, 0, 128);
      (void) fscanf (this->poa_fp_, "%s", poa_name);

      TAO_TRY_EX (CREATE_POA)
        {
          this->child_poa_[i] = prev_poa->create_POA (poa_name,
                                                      this->poa_mgr_.in (),
                                                      policies,
                                                      TAO_TRY_ENV);
          TAO_CHECK_ENV_EX (CREATE_POA);
        }
      TAO_CATCHANY
        {
          TAO_TRY_ENV.print_exception ("create_POA");
          env.exception (TAO_TRY_ENV.exception ());
          return -1;
        }
      TAO_ENDTRY;


      for (j = 0; j < this->num_objs_; j++)
        {

          // activate the object
          TAO_TRY_EX (ACTIVATE_OBJ)
            {
              PortableServer::ObjectId_var obj_var = 
                this->child_poa_[i]->activate_object (&this->demux_test_[i][j],
                                                      TAO_TRY_ENV);
              TAO_CHECK_ENV_EX (ACTIVATE_OBJ);
            }
          TAO_CATCHANY
            {
              TAO_TRY_ENV.print_exception ("poa->activate_obj");
              env.exception (TAO_TRY_ENV.exception ());
              return -1;
            }
          TAO_ENDTRY;

          // Get the IOR and output it to the file
          TAO_TRY_EX (IOR)
            {
              Demux_Test_var demux_var = this->demux_test_[i][j]._this (TAO_TRY_ENV);
              
              TAO_CHECK_ENV_EX (IOR);
              
              CORBA::String_var ior = this->orb_->object_to_string
                (demux_var.in (), TAO_TRY_ENV);

              TAO_CHECK_ENV_EX (IOR);

              ACE_OS::fprintf (this->ior_fp_, "%s\n", ior.in ());
            }
          TAO_CATCHANY
            {
              TAO_TRY_ENV.print_exception ("object_to_string");
              env.exception (TAO_TRY_ENV.exception ());
              return -1;
            }
          TAO_ENDTRY;
          
        } // j loop

      prev_poa = this->child_poa_[i].in ();
      
    } // i loop

  ACE_OS::fclose (this->ior_fp_);

  // now activate the POAs

  TAO_TRY_EX (ACTIVATE)
    {
      this->poa_mgr_->activate (TAO_TRY_ENV);

      TAO_CHECK_ENV_EX (ACTIVATE);
    }
  TAO_CATCHANY
    {
      TAO_TRY_ENV.print_exception ("poa_mgr->activate");
      env.exception (TAO_TRY_ENV.exception ());
      return -1;
    }
  TAO_ENDTRY;
  
  
  // success
  return 0;
  
}

// parse command line arguments (if any).
int
Demux_Test_Server::parse_args (void)
{

  ACE_Get_Opt get_opts (this->argc_, this->argv_, "df:o:p:");
  int c;
  
  while ((c = get_opts ()) != -1)
    switch (c)
      {
      case 'd':  // debug flag
        TAO_debug_level++;
        break;
      case 'f':
        this->ior_fp_ = ACE_OS::fopen (get_opts.optarg, "w");
        if (this->ior_fp_ == 0)
          ACE_ERROR_RETURN ((LM_ERROR,
                             "Unable to open %s for writing: %p\n",
                             get_opts.optarg), -1);
        break;
      case 'o':
        this->num_objs_ = ACE_OS::atoi (get_opts.optarg);
        if (this->num_objs_ > TAO_DEMUX_TEST_MAX_OBJS)
          {
            ACE_ERROR_RETURN ((LM_ERROR,
                               "%d exceeds the maximum of "
                               "%d objects per POA\n",
                               this->num_objs_,
                               TAO_DEMUX_TEST_MAX_OBJS),
                              -1);
          }
        break;
      case 'p':
        this->num_POAs_ = ACE_OS::atoi (get_opts.optarg);
        if (this->num_POAs_ > TAO_DEMUX_TEST_MAX_POAS)
          {
            ACE_ERROR_RETURN ((LM_ERROR,
                               "%d exceeds the maximum of "
                               "%d POAs\n",
                               this->num_objs_,
                               TAO_DEMUX_TEST_MAX_POAS),
                              -1);
          }
        break;
      case '?':
      default:
        ACE_ERROR_RETURN ((LM_ERROR,
                           "usage:  %s"
                           " [-d]"
                           " [-o <num objects>]"
                           " [-p <num POAs>]"
                           " [-f <IOR file>]"
                           "\n", this->argv_ [0]), 
                          -1);
      }

  if (!this->ior_fp_)
    {
      // open default IOR file
      this->ior_fp_ = ACE_OS::fopen ("ior.dat", "w");
      if (this->ior_fp_ == 0)
        ACE_ERROR_RETURN ((LM_ERROR,
                           "Unable to open file ior.dat for writing\n"), -1);
    }
  return 0;
}

// grab a reference to the naming service so that we can register with it. 
int
Demux_Test_Server::init_naming_service (void)
{
  // Initialize the naming services
  if (this->my_name_client_.init (this->orb_.in ()) == -1)
    ACE_ERROR_RETURN ((LM_ERROR,
                       " (%P|%t) Unable to initialize "
                       "the TAO_Naming_Client. \n"),
                      -1);
  
  // success
  return 0;
}

// The main program for Demux_Test
int
Demux_Test_Server::run (CORBA::Environment &env)
{
  TAO_TRY
    {

      if (this->orb_->run () == -1)
        {
          ACE_ERROR_RETURN ((LM_ERROR,
                             "(%N:%l) Demux_Test_Server::run - "
                             "orb run failed\n"),
                            -1);
        }
      TAO_CHECK_ENV;
    }
  TAO_CATCHANY
    {
      TAO_TRY_ENV.print_exception ("run failed");
      env.exception (TAO_TRY_ENV.exception ());
      ACE_ERROR_RETURN ((LM_ERROR,
                         "(%N:%l) Demux_Test_Server::run - "
                         "Error running the server\n"),
                        -1);
    }
  TAO_ENDTRY;

  return 0;
}