///////////////////////////////////////////////////////////////////////////////
//  Apr 28 2001 P.Murat: simple implementation for DB-type utilities
//  need to know MYRON flag for a given run anyway...
///////////////////////////////////////////////////////////////////////////////
#include "TDirectory.h"

#include "Stntuple/data/TStnBeamPos.hh"
#include "Stntuple/data/TStnBeamPosBlock.hh"
#include "Stntuple/obj/TStnRunSummary.hh"
#include "Stntuple/obj/TStnTriggerTable.hh"
#include "Stntuple/obj/TStnDBManager.hh"
#include "Stntuple/obj/TStnDeadList.hh"

#include <iostream>
#include <cstdio>
ClassImp(TStnDBManager)

TStnDBManager*  TStnDBManager::fgInstance = 0;

//_____________________________________________________________________________
TStnDBManager::TStnDBManager() 
{
  // Create the list of objects which we keep track of
  fListOfDbObjects = new TList();

  // Add some standard run dependent objects
  fListOfDbObjects->Add(new TStnRunSummary());
  fListOfDbObjects->Add(new TStnTriggerTable());
  fListOfDbObjects->Add(new TStnDeadList());

  // Beamline in the new scheme (time dependent)
  fListOfDbObjects->Add(new TStnBeamPosBlock("CotBeamPosBlock"));
  fListOfDbObjects->Add(new TStnBeamPosBlock("SvxBeamPosBlock"));

  //----------------------------------------------------------------------------
  // For backward compatibility
  //----------------------------------------------------------------------------
  TStnBeamPos* bp;
  bp = new TStnBeamPos("CotBeamPos");
  bp->SetBit(kInvalidObject);      // Make sure this friend does not get written
  fListOfDbObjects->Add(bp);
  bp = new TStnBeamPos("SvxBeamPos");
  bp->SetBit(kInvalidObject);      // Make sure this friend does not get written
  fListOfDbObjects->Add(bp);
}

//_____________________________________________________________________________
TStnDBManager::~TStnDBManager() {
  fListOfDbObjects->Delete();
}

//_____________________________________________________________________________
TStnDBManager*  TStnDBManager::Instance() { 
  static Cleaner cleaner;
  return (fgInstance) ? fgInstance : (fgInstance = new TStnDBManager());
}

//------------------------------------------------------------------------------
TStnDBManager::Cleaner::Cleaner() {
}

//------------------------------------------------------------------------------
TStnDBManager::Cleaner::~Cleaner() {
  if (TStnDBManager::fgInstance) {
    delete TStnDBManager::fgInstance;
    TStnDBManager::fgInstance = 0;
  }
}


//_____________________________________________________________________________
TObject* TStnDBManager::GetTable(const char* Name)
{
  // Catch special case
  if (! strcmp("TStnBeamPos",Name)) 
    return fListOfDbObjects->FindObject("CotBeamPos");

  // Standard search
  TObject* obj = fListOfDbObjects->FindObject(Name);
  if (obj)
    return obj;

  // So far the object has not been found. If the object requested was
  // the shorthand name like "RunSummary" instead of "TStnRunSummary",
  // try to look for the latter
  TString TStnName(Name);
  TStnName.Prepend("TStn");
  obj = fListOfDbObjects->FindObject(TStnName.Data());
  if (! obj)
    printf(" Object  %s  does not exist!!!! -> CRASH ??\n",Name);

  return obj;
}


//_____________________________________________________________________________
Int_t TStnDBManager::Write(const char* Name, Int_t Option, Int_t BufSize) 
{
  int rc = 0;
  TIter it(fListOfDbObjects);
  while (TObject* obj = it.Next()) {
    if (! obj->TestBit(kInvalidObject))
      rc += obj->Write();
  }
  return rc;
}

//_____________________________________________________________________________
Int_t TStnDBManager::Read(const char* Name) 
{
  int nb = 0;
  TIter it(fListOfDbObjects);
  while (TObject* obj = it.Next()) {
    if (gDirectory->GetListOfKeys()->FindObject(obj->GetName()) != 0) {
      nb += obj->Read(obj->GetName());
    }
    else
      obj->Clear();
  }

  //----------------------------------------------------------------------------
  // Fill SvxBeamPos and CotBeamPos on the fly, if they are not available
  //----------------------------------------------------------------------------
  MakeAvgBeamline();

  //----------------------------------------------------------------------------
  // For compatibility with old file BeamPos: look for CotBeam only
  //----------------------------------------------------------------------------
  if (gDirectory->GetListOfKeys()->FindObject("TStnBeamPos") != 0) {
    TStnBeamPos* cot = (TStnBeamPos*) GetTable("CotBeamPos");
    nb += cot->Read("TStnBeamPos");
  }

  return nb;
}

//_____________________________________________________________________________
void TStnDBManager::MakeAvgBeamline() 
{
  //----------------------------------------------------------------------------
  // Fill SvxBeamPos and CotBeamPos on the fly, if they are not available
  //----------------------------------------------------------------------------

  // extract run number from default directory
  int run = -1;
  TString srun = gDirectory->GetName();
  // if this is reading from an existing directory
  if(srun(0,4)=="run_") {
    TString rr = srun(4,6);
    sscanf(rr.Data(),"%d",&run);
  }

  TStnBeamPosBlock* cotB = (TStnBeamPosBlock*) GetTable("CotBeamPosBlock");
  if (cotB != 0 && cotB->NBeamPos()>0) {
    TStnBeamPos*      cot  = (TStnBeamPos*)      GetTable("CotBeamPos");
    cotB->AverageBeamPos(cot);    // Now copy the average beamline
    cot->SetBit(kInvalidObject);
    // now also remove extra entries which were made in dev_243
    for(int i=0; i<cotB->NBeamPos(); i++){
      if(run>0 && ((TStnBeamPos*)cotB->BeamPos(i))->RunNumber()!=run) {
	cotB->BeamPositions()->RemoveAt(i);
      }
    }
    cotB->BeamPositions()->Compress();
    cotB->SetNBeamPos(cotB->BeamPositions()->GetEntries());
  }
  else
    GetTable("CotBeamPosBlock")->SetBit(kInvalidObject);

  TStnBeamPosBlock* svxB = (TStnBeamPosBlock*) GetTable("SvxBeamPosBlock");
  if (svxB != 0 && svxB->NBeamPos()>0) {
    TStnBeamPos*      svx  = (TStnBeamPos*)      GetTable("SvxBeamPos");
    svxB->AverageBeamPos(svx);    // Now copy the average beamline
    svx->SetBit(kInvalidObject);
    // now also remove extra entries which were made in dev_243
    for(int i=0; i<svxB->NBeamPos(); i++){
      if(run>0 && ((TStnBeamPos*)svxB->BeamPos(i))->RunNumber()!=run) {
	svxB->BeamPositions()->RemoveAt(i);
      }
    }
    svxB->BeamPositions()->Compress();
    svxB->SetNBeamPos(svxB->BeamPositions()->GetEntries());
  }
  else
    GetTable("SvxBeamPosBlock")->SetBit(kInvalidObject);

  return;
}