CustomLightInjector.cxx

Go to the documentation of this file.

// $Id: CustomLightInjector.cxx,v 1.12 2006/12/12 17:07:04 rhatcher Exp $
//
// Job control interface to input data streams
//
// messier@huhepl.harvard.edu
#include "TSystem.h"
#include "TRegexp.h"

#include "CustomLightInjector.h"
#include <cassert>
#include <cmath>

#include "MessageService/MsgService.h"
#include "MinosObjectMap/MomNavigator.h"
#include "JobControl/JobCInputModule.h"
#include "JobControl/JobCModuleRegistry.h"
#include "JobControl/JobCEnv.h"
#include "RawData/RawRecord.h"
#include "RawData/RawDaqSnarlHeader.h"
#include "Record/RecRecord.h"
#include "Record/RecPhysicsHeader.h"
#include "Registry/Registry.h"
#include "Validity/VldContext.h"
#include "Util/UtilString.h"
#include <TRandom.h>
#include "Digitization/DigiList.h"
#include "Digitization/DigiPE.h"
#include "Record/SimSnarlRecord.h"
#include "Record/SimSnarlHeader.h"
#include "RawData/RawRecord.h"
#include "RawData/RawDaqSnarlHeader.h"
#include "RawData/RawDigitDataBlock.h"
#include "RawData/RawDigit.h"
#include "DataUtil/Truthifier.h"
#include "TClonesArray.h"
#include "Conventions/Munits.h"

#include <algorithm>

//CVSID("$Id: CustomLightInjector.cxx,v 1.12 2006/12/12 17:07:04 rhatcher Exp $");
JOBMODULE(CustomLightInjector,"CustomLightInjector","Simulated Light Injection");

RawChannelId CustomLightInjector::fRcid1;
RawChannelId CustomLightInjector::fRcid2;


//......................................................................

CustomLightInjector::CustomLightInjector() : 
  fCurrentSnarl(0)
{ }

//......................................................................

CustomLightInjector::~CustomLightInjector() 
{ 
}

//......................................................................

void CustomLightInjector::BeginJob() 
{
}


//......................................................................

const Registry& CustomLightInjector::DefaultConfig() const 
{
//======================================================================
// Get the default configuration for this module
//======================================================================
  static Registry r; // Default configuration for module

  // Set name of config
  std::string name = this->GetName();
  name += ".config.default";
  r.SetName(name.c_str());

  // Set values in configuration
  r.UnLockValues();
  r.Set("detectorType", 2);
  r.Set("rate",         1.0);
  r.Set("date",         20030101);
  r.Set("time",         000000);
  r.Set("photonRate",   0.0);

  r.LockValues();

  return r;
}

//......................................................................

void CustomLightInjector::Config(const Registry& r) 
{
//======================================================================
// Configure the module given the Registry r
//======================================================================
  int    tmpi;
  double tmpd;

  if (r.Get("detectorType",tmpi)) { fDetector = tmpi; }
  if (r.Get("rate",tmpd)) { fRate = tmpd; }
  if (r.Get("date",tmpi)) { fDate = tmpi; }
  if (r.Get("time",tmpi)) { fTime = tmpi; }
  r.Get("photonRate",fPhotonRate);

  fPulseTime = VldTimeStamp(fDate,fTime,0);
  fContext =   VldContext(Detector::CharToEnum((char)fDetector),
                          SimFlag::kReroot,
                          fPulseTime);
}

//......................................................................  

JobCResult CustomLightInjector::Get()
{
  MomNavigator* mom = this->GetMom();
  assert(mom);
  mom -> Clear(); // Moving on so clear contents of Mom


  // Make a SimSnarl.
  Int_t   run      = 1000;
  Int_t   trigbits = 0;
  Short_t subrun = 0;
  Short_t runtype = 0;
  Int_t   errcode = 0;
  Int_t   timeframe = -1;

  // Increment the time by the pulse rate.
  int secs;
  double frac = frexp(fRate*fCurrentSnarl,&secs);
  int nsecs = int(frac*1e9);
  VldTimeStamp offset(secs,nsecs);
  fPulseTime = VldTimeStamp(fDate,fTime,0);
  fPulseTime.Add(offset);

  // Reset the context.
  fContext = VldContext(Detector::CharToEnum((char)fDetector),
                        SimFlag::kMC,
                        fPulseTime);

  VldTimeStamp now;
  std::string  codename = SimSnarlHeader::TrimCodename("$Name:  $");
  std::string hostname(gSystem->HostName());

  SimSnarlHeader simheader(fContext,run,subrun,runtype,
                           errcode,fCurrentSnarl,trigbits,timeframe,-1,
                           now,codename,hostname);
                                                
  SimSnarlRecord* simsnarl = new SimSnarlRecord(simheader);

  simsnarl->GetTempTags().Set("stream","SimSnarl");  // no idea....
    
  // Give it to mom.
  mom -> AdoptFragment(simsnarl);

  // Add the DigiPE list to the simsnarl.
  TObjArray* PeList = new TObjArray(0);
  PeList->SetName("DigiListPe");
  PeList->SetOwner(true);

  // Find out how many of them to make.
  gRandom->SetSeed(fCurrentSnarl);

  // Make the photoelectron objects.
  // i.e: 
  //  DigiPE*  pe = new DigiPE(time, fPsid, NULL);
  // PeList->Add(pe);
  
  
  PlexHandle plex(fContext);
  PlexStripEndId seid1(fContext.GetDetector(),
                       1, 10, StripEnd::kWest);
  
  PlexStripEndId seid2(fContext.GetDetector(),
                       5, 15, StripEnd::kWest);

  PlexPixelSpotId psid1 = plex.GetPixelSpotId(seid1);
  PlexPixelSpotId psid2 = plex.GetPixelSpotId(seid2);

  fRcid1 = plex.GetRawChannelId(psid1);
  fRcid2 = plex.GetRawChannelId(psid2);

  //cout << "Injecting on " << psid1.AsString() << "->" << fRcid1.AsString() << endl;
  //cout << "         and " << psid2.AsString() << "->" << fRcid2.AsString() << endl;


  float npe = 0;
  for(int i=0;i<1*fCurrentSnarl;i++) {
    npe++;
    double time = double(i) * fPhotonRate*Munits::ns;
    DigiPE* pe;
    pe = new DigiPE(time, psid1, DigiSignal::kGenuine);
    PeList->Add(pe);
    pe = new DigiPE(time, psid2, DigiSignal::kGenuine);
    PeList->Add(pe);
  }

  cout << npe;



  simsnarl->AdoptTemporary(PeList);

  TClonesArray *new_arr = new TClonesArray("DigiScintHit",0);
  new_arr->SetName("DigiScintHits");
  simsnarl->AdoptComponent(new_arr);
  

  return JobCResult::kAOK;
}

//......................................................................

JobCResult CustomLightInjector::Next(int n)
{
//======================================================================
// Advance the position in the stream n record sets. Load the records
// at the last position
//======================================================================
  fCurrentSnarl+=n;
  return this->Get();
}

//......................................................................

JobCResult CustomLightInjector::Prev(int n) 
{
  if(fCurrentSnarl>0) fCurrentSnarl-=n;
  return this->Get();
}

//......................................................................

JobCResult CustomLightInjector::GoTo(int /* run */, int snarl, int /* searchDir */) 
{
  fCurrentSnarl = snarl;
  return this->Get();
}

//......................................................................

JobCResult CustomLightInjector::Ana(const MomNavigator* mom)
{
  const Truthifier& truth = Truthifier::Instance(mom);

  std::vector<int> adcs1(0);
  std::vector<int> adcs2(0);
  std::vector<float> q1(0);
  std::vector<float> q2(0);
          
 
 // Get the RawDigitDataBlock.
  TObject* tobj;
  TIter    fragiter = mom->FragmentIter();
  while( ( tobj = fragiter.Next() ) ) {
    RawRecord* rawRec = dynamic_cast<RawRecord*>(tobj);
    if(rawRec) {
      TIter itr = rawRec->GetRawBlockIter();
      RawDataBlock* rawBlk;
      while ( ( rawBlk = dynamic_cast<RawDataBlock*>(itr.Next()) ) ) {
        RawDigitDataBlock* rddb = dynamic_cast<RawDigitDataBlock*>(rawBlk);
        if(rddb) {
          //cout << "Ana context: " << rawRec->GetRawHeader()->GetVldContext().AsString() << endl;
          //PlexHandle plex(rawRec->GetRawHeader()->GetVldContext());

          TIter iter = rddb->GetDatumIter();
          while( (tobj=iter.Next() ) ) {
            
            // Process a digit.     
            
            RawDigit*        rawDigit = dynamic_cast<RawDigit *>(tobj);
            //cout << *rawDigit << endl;

            if(rawDigit) {
              if(rawDigit->GetChannel().IsSameChannel(fRcid1)) {
                q1.push_back(truth.GetSignal(rawDigit)->GetCharge()/Munits::fC);
                adcs1.push_back(rawDigit->GetADC());
              }
              else if(rawDigit->GetChannel().IsSameChannel(fRcid2)) 
                adcs2.push_back(rawDigit->GetADC());            
              else {
                //cout << "Xtalk chan? "  << *rawDigit << endl;
              }
                
            }

          }

        }
      }
    }
  } // It's so ugly!

  for(UInt_t i=0;i<5;i++) {
    if(adcs1.size()>i) cout << "\t" << adcs1[i];
    else               cout << "\t" << 0;         
  }
  for(UInt_t i=0;i<5;i++) {
    if(q1.size()>i) cout << "\t" << q1[i];
    else            cout << "\t" << 0;    
  }

  cout << endl;
          

  return JobCResult::kPassed; // kNoDecision, kFailed, etc.
}

---