//==========================================
//
//  Prereq  -- filter module for selecting events by Trigger
//
//  Authors: Kevin McFarland, Kirsten Tollefson
//
//==========================================

#ifdef DEFECT_OLD_IOSTREAM_HEADERS
#include <iostream.h>
#include <iomanip.h>
#else
#include <iostream>
#include <iomanip>
#endif

#ifndef DEFECT_NO_STDLIB_NAMESPACES
using std::dec;
using std::setw;
using std::hex;
using std::cout;
using std::endl;
using std::string;
#endif

#include "Level3Mods/RawTriggerQuery.hh"

#include "TriggerObjects/TFRD_StorableBank.hh"
#include "TriggerObjects/TL2D_StorableBank.hh"
#include "TriggerObjects/TL3D_StorableBank.hh"
#include "Level3Objects/Level3ModuleResults.hh"
#include "BaBar/Cdf.hh"
#include "Edm/EventRecord.hh"
#include "Edm/ConstHandle.hh"
#include "AbsEnv/AbsEnv.hh"

//-------------------
// ConsumerServerLogger headers --  used here for trigger bits
//-------------------
extern "C" {
#include "ConsumerInterface/l3_cs_pointer_space.h"
}

bool RawTriggerQuery::fillL1TriggerMask ( int4* L1Mask, rawTriggerSource theSource ){
  bool foundL1Bits = false;
  AbsEvent* theEvent = AbsEnv::theEvent();

  if ( theSource.L1 == RawTriggerQuery::fromTFRD ||
       theSource.L1 == RawTriggerQuery::fromTFRDunprescaled )
    {
      //
      // Level-1 Trigger Bits From TFRD
      //
      EventRecord::ConstIterator TFRD_iter( theEvent, 
                                            "TFRD_StorableBank" );
      if ( TFRD_iter.is_valid() )
        {
          foundL1Bits = true;
          ConstHandle<TFRD_StorableBank> tfrd( TFRD_iter );
          for (int iw=0; 
               iw<L1_BITMASK_WORDS*BITMASK_BITS/8/sizeof(int4) ; 
               iw++) 
            {
              int ibit = 32*iw;
              if ( theSource.L1 == RawTriggerQuery::fromTFRDunprescaled )
                L1Mask[iw] = tfrd->getL1NoPrescaleTrigDecision(ibit);
              else
                L1Mask[iw] = tfrd->getL1PrescaleTrigDecision(ibit);
            }
          if ( theSource.debug )
            {
              cout << "  TFRD L1 Trig Mask = " << hex << setw(8);
              for (int iw=0; 
                   iw<L1_BITMASK_WORDS*BITMASK_BITS/8/sizeof(int4); 
                   iw++)
                cout << L1Mask[iw] << " ";
              cout << dec << endl;                
            } // debug
        } // valid TFRD
    } // use TFRD
  //
  // Or get Level-1 Trigger Bits from TL2D
  //
  else if ( theSource.L1 == RawTriggerQuery::fromTL2D )
    {
      EventRecord::ConstIterator TL2D_iter( theEvent, 
                                            "TL2D_StorableBank" );
      if ( TL2D_iter.is_valid() )
        {
          foundL1Bits = true;
          ConstHandle<TL2D_StorableBank> tl2d( TL2D_iter );
          for (int iw=0; 
               iw<L1_BITMASK_WORDS*BITMASK_BITS/8/sizeof(int4) ; 
               iw++) 
            {
              int ibit = 32*iw;
              L1Mask[iw] = tl2d->getL1TriggerBit(ibit);
            }
          if ( theSource.debug )
            {
              cout << "  TL2D L1 Trig Mask = " << hex << setw(8);
              for (int iw=0; 
                   iw<L1_BITMASK_WORDS*BITMASK_BITS/8/sizeof(int4); 
                   iw++)
                cout << L1Mask[iw] << " ";
              cout << dec << endl;                
            } // debug
        } // found TL2D
    } // use TL2D
  
  return (foundL1Bits);
}      

bool RawTriggerQuery::fillL2TriggerMask ( int4* L2Mask, rawTriggerSource theSource){
  bool foundL2Bits = false;
  AbsEvent* theEvent = AbsEnv::theEvent();

  if ( theSource.L2 == RawTriggerQuery::fromTL2D )
    {
      EventRecord::ConstIterator TL2D_iter( theEvent, 
                                            "TL2D_StorableBank" );
      if ( TL2D_iter.is_valid() )
        {
          foundL2Bits = true;
          ConstHandle<TL2D_StorableBank> tl2d( TL2D_iter );
          for (int iw=0; 
               iw<L2_BITMASK_WORDS*BITMASK_BITS/8/sizeof(int4) ; 
               iw++) {
            int ibit = 32*iw;
            L2Mask[iw] = tl2d->getL2TriggerBit(ibit);
          } 
          
          if ( theSource.debug )
            {
              cout << "  TL2D L2 Trig Mask = " << hex << setw(8); 
              for (int iw=0; 
                   iw<L2_BITMASK_WORDS*BITMASK_BITS/8/sizeof(int4); 
                   iw++)
                cout << L2Mask[iw] << " ";
              cout << dec << endl;
            } // debug
        } // found TL2D 
    } // TL2D
  
  return (foundL2Bits);
}