#include <CelltreeSource.h>

Inheritance diagram for WireCell::Root::CelltreeSource:

Public Member Functions
	CelltreeSource ()

virtual	~CelltreeSource ()

virtual bool	operator() (IFrame::pointer &out)
	The calling signature: More...

virtual WireCell::Configuration	default_configuration () const
	Optional, override to return a hard-coded default configuration. More...

virtual void	configure (const WireCell::Configuration &config)
	Accept a configuration. More...

Public Member Functions inherited from WireCell::IFrameSource
virtual	~IFrameSource ()

Public Member Functions inherited from WireCell::ISourceNode< IFrame >
virtual	~ISourceNode ()

virtual NodeCategory	category ()
	Return the behavior category type. More...

virtual std::string	signature ()
	Set the signature for all subclasses. More...

virtual bool	operator() (boost::any &anyout)

virtual std::vector< std::string >	output_types ()

Public Member Functions inherited from WireCell::ISourceNodeBase
virtual	~ISourceNodeBase ()

Public Member Functions inherited from WireCell::INode
virtual	~INode ()

virtual int	concurrency ()

virtual std::vector< std::string >	input_types ()

virtual void	reset ()

Public Member Functions inherited from WireCell::IComponent< INode >
virtual	~IComponent ()

Public Member Functions inherited from WireCell::Interface
virtual	~Interface ()

Public Member Functions inherited from WireCell::IConfigurable
virtual	~IConfigurable ()

Public Member Functions inherited from WireCell::IComponent< IConfigurable >
virtual	~IComponent ()

Private Attributes
Configuration	m_cfg

int	m_calls

Additional Inherited Members
Public Types inherited from WireCell::IFrameSource
typedef std::shared_ptr< IFrameSource >	pointer

Public Types inherited from WireCell::ISourceNode< IFrame >
typedef IFrame	output_type

typedef ISourceNode< IFrame >	signature_type

typedef std::shared_ptr< signature_type >	pointer

typedef std::shared_ptr< const IFrame >	output_pointer

Public Types inherited from WireCell::ISourceNodeBase
typedef std::shared_ptr< ISourceNodeBase >	pointer

Public Types inherited from WireCell::INode
enum	NodeCategory { unknown, sourceNode, sinkNode, functionNode, queuedoutNode, joinNode, splitNode, faninNode, fanoutNode, multioutNode, hydraNode }

Public Types inherited from WireCell::IComponent< INode >
typedef std::shared_ptr< INode >	pointer
	Access subclass facet by pointer. More...

typedef std::vector< pointer >	vector
	Vector of shared pointers. More...

Public Types inherited from WireCell::Interface
typedef std::shared_ptr< Interface >	pointer

Public Types inherited from WireCell::IComponent< IConfigurable >
typedef std::shared_ptr< IConfigurable >	pointer
	Access subclass facet by pointer. More...

typedef std::vector< pointer >	vector
	Vector of shared pointers. More...

Detailed Description

Definition at line 10 of file CelltreeSource.h.

Constructor & Destructor Documentation

Root::CelltreeSource::CelltreeSource ( )

Definition at line 18 of file CelltreeSource.cxx.

     : m_calls(0)
 {
 }

Root::CelltreeSource::~CelltreeSource ( )

virtual

Definition at line 23 of file CelltreeSource.cxx.

24 {

25 }

Member Function Documentation

void Root::CelltreeSource::configure ( const WireCell::Configuration & config )

virtual

Accept a configuration.

Implements WireCell::IConfigurable.

Definition at line 27 of file CelltreeSource.cxx.

 {
     if (cfg["filename"].empty()) {
         THROW(ValueError() << errmsg{"CelltreeSource: must supply input \"filename\" configuration item."});
     }
     m_cfg = cfg;
 }

WireCell::Configuration Root::CelltreeSource::default_configuration ( ) const

virtual

Optional, override to return a hard-coded default configuration.

Reimplemented from WireCell::IConfigurable.

Definition at line 35 of file CelltreeSource.cxx.

 {
     Configuration cfg;
     // Give a URL for the input file.
     cfg["filename"] = "";
 
     // which event in the celltree file to be processed 
     cfg["EventNo"] = "0";
     
     // Give a list of frame/tree tags. 
 
     // just raw waveform and no other choice at present
     // Tree: Sim, Wf: raw_wf
     cfg["frames"][0] = "orig";
 
 
     return cfg;
 }

bool Root::CelltreeSource::operator() ( IFrame::pointer & out )

virtual

The calling signature:

Implements WireCell::ISourceNode< IFrame >.

Definition at line 54 of file CelltreeSource.cxx.

 {
     out = nullptr;
     ++m_calls;
     std::cerr << "CelltreeSource: called " << m_calls << " times\n";
     if (m_calls > 2) {
         std::cerr << "CelltreeSource: past EOS\n";
         return false;
     }
     if (m_calls > 1) {
         std::cerr << "CelltreeSource: EOS\n";
         return true;            // this is to send out EOS
     }
     
 
   std::string url = m_cfg["filename"].asString();
 
   TFile* tfile = TFile::Open(url.c_str());
 
 
   
   TTree *tree = (TTree*)tfile->Get("/Event/Sim");
   if (!tree) {
     THROW(IOError() << errmsg{"No tree: /Event/Sim in input file"});
   }
 
   tree->SetBranchStatus("*",0);
 
   int run_no, subrun_no, event_no;
   tree->SetBranchStatus("eventNo",1);
   tree->SetBranchAddress("eventNo" , &event_no);
   tree->SetBranchStatus("runNo",1);
   tree->SetBranchAddress("runNo"   , &run_no);
   tree->SetBranchStatus("subRunNo",1);
   tree->SetBranchAddress("subRunNo", &subrun_no);
   
 
   std::vector<int> *channelid = new std::vector<int>;
   TClonesArray* esignal = new TClonesArray;
   
   tree->SetBranchStatus("raw_channelId",1);
   tree->SetBranchAddress("raw_channelId", &channelid);
   tree->SetBranchStatus("raw_wf",1);
   tree->SetBranchAddress("raw_wf", &esignal);
 
   int frame_number = std::stoi(m_cfg["EventNo"].asString());
 
   // loop entry 
   int siz = 0;
   unsigned int entries = tree->GetEntries();
   for(unsigned int ent = 0; ent<entries; ent++)
   {
       siz = tree->GetEntry(ent);
       if(event_no == frame_number)
       {
           break;
       }
   }
     
 
   // need run number and subrunnumber
   int frame_ident = event_no;
   double frame_time=0;
 
   // some output using eventNo, runNo, subRunNO, ...
   std::cerr<<"CelltreeSource: frame "<<frame_number<<"\n";
   std::cerr << "CelltreeSource: runNo "<<run_no<<", subrunNo "<<subrun_no<<", eventNo "<<event_no<<"\n";
   
   ITrace::vector all_traces;
   std::unordered_map<IFrame::tag_t, IFrame::trace_list_t> tagged_traces;
   // celltree input now is raw data, no information about any noisy or bad channels
   // leave cmm empty.
   WireCell::Waveform::ChannelMaskMap cmm;
 
   
   if (siz>0 && frame_number == frame_ident){
     int nchannels = channelid->size();
 
     auto frametag = m_cfg["frames"][0].asString();
     int channel_number = 0;
     
     std::cerr<<"CelltreeSource: loading "<<frametag<<" "<<nchannels<<" channels \n";
 
     // fill waveform ...
     TH1S *signal_s;
     TH1F *signal_f;
         
      for (int ind=0; ind < nchannels; ++ind) {
        signal_s = dynamic_cast<TH1S*>(esignal->At(ind));
        signal_f = dynamic_cast<TH1F*>(esignal->At(ind));
        
        bool flag_float = 1;
        if (signal_f==0 && signal_s!=0)
          flag_float = 0;
        
        if (!signal_f && !signal_s) continue;
        channel_number = channelid->at(ind);
        
        ITrace::ChargeSequence charges;
        int nticks;
        if (flag_float==1)
          nticks = signal_f->GetNbinsX();
        else
          nticks = signal_s->GetNbinsX();
        //std::cerr<<"CelltreeSource: tick "<<nticks<<"\n";
        //nticks = 9600,  this could be an issue cause just 9594 have non-ZERO value around baseline
        for (int itickbin = 0; itickbin != nticks; itickbin++){
          if (flag_float==1){
            if(signal_f->GetBinContent(itickbin+1)!=0) {
              charges.push_back(signal_f->GetBinContent(itickbin+1));
            }
          }else{
            if(signal_s->GetBinContent(itickbin+1)!=0) {
              charges.push_back(signal_s->GetBinContent(itickbin+1));
            }
          }
        }
        
        const size_t index = all_traces.size();
          tagged_traces[frametag].push_back(index);
          
          all_traces.push_back(std::make_shared<SimpleTrace>(channel_number, 0, charges));
      }
      auto sframe = new SimpleFrame(frame_ident, frame_time,
                                    all_traces, 0.5*units::microsecond, cmm);
      for (auto const& it : tagged_traces) {
        sframe->tag_traces(it.first, it.second);
        std::cerr << "CelltreeSource: tag " << it.second.size() << " traces as: \"" << it.first << "\"\n";
      }
      
      out = IFrame::pointer(sframe);
      
      return true;
   }else{
     std::cerr << "CelltreeSource: Event Number is out of boundary! " << std::endl;
     return false;
   }
   
 
   
 
   
   
   // // runNo, subRunNo??
   // trun->SetBranchAddress("eventNo", &frame_ident);
   // trun->SetBranchAddress("total_time_bin", &nticks);
   // //trun->SetBranchAddress("time_offset", &frame_time); use this??
   // trun->GetEntry(0);
   
   //  std::cerr << "CelltreeSource: frame ident="<<frame_ident<<", time=0, nticks="<<nticks<<"\n";
   
     
 }

Member Data Documentation

int WireCell::Root::CelltreeSource::m_calls

private

Definition at line 22 of file CelltreeSource.h.

Configuration WireCell::Root::CelltreeSource::m_cfg

private

Definition at line 21 of file CelltreeSource.h.

The documentation for this class was generated from the following files:

wire-cell-build/root/inc/WireCellRoot/CelltreeSource.h
wire-cell-build/root/src/CelltreeSource.cxx

Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation