// $Id$
//==========================================================================
//  AIDA Detector description implementation for LCD
//--------------------------------------------------------------------------
// Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
// All rights reserved.
//
// For the licensing terms see $DD4hepINSTALL/LICENSE.
// For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
//
// Author     : M.Frank
//
//==========================================================================

// Framework include files
#include "DD4hep/LCDD.h"
#include "DD4hep/Printout.h"
#include "DD4hep/Conditions.h"
#include "DD4hep/DetFactoryHelper.h"

#include "DDCond/ConditionsManager.h"
#include "DDCond/ConditionsIOVPool.h"
#include "DDCond/ConditionsInterna.h"
#include "DDCond/ConditionsPool.h"
#include "DDCond/ConditionsInterna.h"

using namespace std;
using namespace DD4hep;
using namespace DD4hep::Conditions;
using Geometry::LCDD;
using Geometry::Position;
using Geometry::DetElement;

namespace {

  int ddcond_dump_conditions_functor(lcdd_t& lcdd, bool print_conditions)   {
    typedef std::vector<const IOVType*> _T;
    typedef ConditionsIOVPool::Entries _E;
    typedef RangeConditions _R;
    ConditionsManager manager = ConditionsManager::from(lcdd);

    const _T types = manager.iovTypesUsed();
    for( _T::const_iterator i = types.begin(); i != types.end(); ++i )    {
      const IOVType* type = *i;
      if ( type )   {
	ConditionsIOVPool* pool = manager.iovPool(*type);
	if ( pool )  {
	  const _E& e = pool->entries;
	  printout(INFO,"CondPoolDump","+++ ConditionsIOVPool for type %s  [%d IOV entries]",
		   type->str().c_str(), int(e.size()));
	  for (_E::const_iterator j=e.begin(); j != e.end(); ++j)  {
	    ConditionsPool* cp = (*j).second;
	    cp->print("");
	    if ( print_conditions )   {
	      _R rc;
	      cp->select_all(rc);
	      //print_conditions<void>(rc);
	    }
	  }
	}
      }
    }
    printout(INFO,"Example","SUCCESS: +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++");
    printout(INFO,"Example","SUCCESS: +++ Conditions pools successfully dumped");
    printout(INFO,"Example","SUCCESS: +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++");
    return 1;
  }

  int ddcond_dump_pools(LCDD& lcdd, int /* argc */, char** /* argv */)   {
    return ddcond_dump_conditions_functor(lcdd,false);
  }
  int ddcond_dump_conditions(LCDD& lcdd, int /* argc */, char** /* argv */)   {
    return ddcond_dump_conditions_functor(lcdd,true);
  }
}

DECLARE_APPLY(DD4hepDDCondPoolDump,ddcond_dump_pools)
DECLARE_APPLY(DD4hepDDCondConditionsDump,ddcond_dump_conditions)

namespace {
  /// Plugin entry point.
  static long synchronize_conditions(lcdd_t& lcdd, int argc, char** argv) {
    if ( argc > 0 )   {
      string iov_type = argv[0];
      IOV::Key::first_type iov_key = *(IOV::Key::first_type*)argv[1];
      ConditionsManager    manager = ConditionsManager::from(lcdd);
      const IOVType*       epoch   = manager.iovType(iov_type);
      dd4hep_ptr<ConditionsPool> user_pool;
      IOV  iov(epoch);

      iov.set(iov_key);
      long num_expired = manager.prepare(iov, user_pool);
      if ( iov_type == "epoch" )  {
	char   c_evt[64];
	struct tm evt;
	::gmtime_r(&iov_key, &evt);
	::strftime(c_evt,sizeof(c_evt),"%T %F",&evt);
	printout(INFO,"Conditions",
		 "+++ ConditionsUpdate: Updated %ld conditions... event time: %s",
		 num_expired, c_evt);
      }
      else  {
	printout(INFO,"Conditions",
		 "+++ ConditionsUpdate: Updated %ld conditions... key[%s]: %ld",
		 num_expired, iov_type.c_str(), iov_key);
      }
      manager.enable(iov, user_pool);
      user_pool->print("User pool");
      manager.clean(epoch, 20);
      user_pool->clear();
      return 1;
    }
    except("DDDB","+++ Failed update DDDB conditions. No event time argument given!");
    return 0;
  }
}
DECLARE_APPLY(DD4hep_ConditionsSynchronize,synchronize_conditions)

namespace {
  /// Plugin entry point.
  static long clean_conditions(lcdd_t& lcdd, int argc, char** argv) {
    if ( argc > 0 )   {
      string iov_type = argv[0];
      int max_age = *(int*)argv[1];
      printout(INFO,"Conditions",
	       "+++ ConditionsUpdate: Cleaning conditions... type:%s max age:%d",
	       iov_type.c_str(), max_age);
      ConditionsManager manager = ConditionsManager::from(lcdd);
      const IOVType* epoch = manager.iovType(iov_type);
      manager.clean(epoch, max_age);
      return 1;
    }
    except("DDDB","+++ Failed cleaning conditions. Insufficient arguments!");
    return 0;
  }
}
DECLARE_APPLY(DD4hep_ConditionsClean,clean_conditions)