LArSamples::DataTweaker Class Reference

#include <DataTweaker.h>

Collaboration diagram for LArSamples::DataTweaker:

Public Member Functions
	DataTweaker ()=default
bool	set (const TString &tweaks)
bool	findOption (TString &tweaks, const TString &option) const
Data *	tweak (const Data &data, int evtIndex=-1) const
EventData *	tweak (const EventData &eventData, int runIndex=-1) const
void	setNSamplesTrunctation (unsigned int n)
void	addSample (unsigned int n)
void	setRemoveNoiseCorrelations (bool remove=true)
void	setRefit (bool refit=true)
void	setFitParams (Chi2Params params)
void	setAdjust (bool adjust=true)
void	setRemoveRoIs (bool removeRoIs=true)

Private Attributes
unsigned int	m_nSamplesTruncation {0}
bool	m_removeNoiseCorrelations {false}
bool	m_refit {false}
bool	m_adjust {false}
bool	m_removeRoIs {false}
std::map< unsigned int, bool >	m_samples
Chi2Params	m_fitParams {DefaultChi2}

Detailed Description

Definition at line 24 of file DataTweaker.h.

Constructor & Destructor Documentation

DataTweaker()

LArSamples::DataTweaker::DataTweaker ( )

default

Member Function Documentation

addSample()

void LArSamples::DataTweaker::addSample ( unsigned int  n )

inline

Definition at line 37 of file DataTweaker.h.

{ m_samples[n] = true; }

findOption()

bool DataTweaker::findOption	(	TString &	tweaks,
		const TString &	option
	)		const

Definition at line 38 of file DataTweaker.cxx.

{
  int i = tweaks.Index(option);
  if (i < 0) return false;
  tweaks = tweaks(0, i) + tweaks(i + option.Length(), tweaks.Length() - i - option.Length());
  return true;
}

set()

bool DataTweaker::set

(

const TString &

tweaks

)

Definition at line 23 of file DataTweaker.cxx.

{
  TString toMod = tweaks;
  
  if (findOption(toMod, "removeNoiseCorrelations")) { m_removeNoiseCorrelations = true; cout << "--> removeNoiseCorrelations" << endl; }
  if (findOption(toMod, "refit"))                   { m_refit = true;                   cout << "--> refit" << endl; }
  if (findOption(toMod, "adjust"))                  { m_adjust = true;                  cout << "--> adjust" << endl; }
  if (findOption(toMod, "removeRoIs"))              { m_removeRoIs = true;              cout << "--> removeRoIs" << endl; }
 
  if (toMod == "") return true;
  cout << "Invalid tweaker option(s) " << toMod << endl;
  return false;
}

setAdjust()

void LArSamples::DataTweaker::setAdjust ( bool  adjust = true )

inline

Definition at line 41 of file DataTweaker.h.

{ m_adjust = adjust; }

setFitParams()

void LArSamples::DataTweaker::setFitParams ( Chi2Params  params )

inline

Definition at line 40 of file DataTweaker.h.

{ m_fitParams = params; }

setNSamplesTrunctation()

void LArSamples::DataTweaker::setNSamplesTrunctation ( unsigned int  n )

inline

Definition at line 36 of file DataTweaker.h.

{ m_nSamplesTruncation = n; }

setRefit()

void LArSamples::DataTweaker::setRefit ( bool  refit = true )

inline

Definition at line 39 of file DataTweaker.h.

{ m_refit = refit; }

setRemoveNoiseCorrelations()

void LArSamples::DataTweaker::setRemoveNoiseCorrelations ( bool  remove = true )

inline

Definition at line 38 of file DataTweaker.h.

{ m_removeNoiseCorrelations = remove; }

setRemoveRoIs()

void LArSamples::DataTweaker::setRemoveRoIs ( bool  removeRoIs = true )

inline

Definition at line 42 of file DataTweaker.h.

{ m_removeRoIs = removeRoIs; }

tweak() [1/2]

Data * DataTweaker::tweak	(	const Data &	data,
		int	evtIndex = -1
	)		const

Definition at line 47 of file DataTweaker.cxx.

{
  std::vector<short> samples = data.container().samples();
  std::vector<float> corrs = data.container().corrs();
  float time = data.container().ofcTime();
  double k = 1, deltaT = 0;
 
  if (m_removeNoiseCorrelations)
    for (unsigned int i = 1; i < corrs.size(); i++) corrs[i] = 0;
 
  if (m_nSamplesTruncation > 0) {
    unsigned int maxIndex = data.maxPosition();
    
    int i1 = maxIndex - (m_nSamplesTruncation - 1)/2;
    if (i1 < 0) i1 = 0;
    
    int i2 = maxIndex + m_nSamplesTruncation/2 + 1;
    if (i2 > (int)data.nSamples()) i2 = (int)data.nSamples();
    samples.clear();
    for (int i = i1; i < i2; i++) samples.push_back(data.sample(i));
    corrs.resize(i2 - i1);
    time = time - data.time(i1);
  }
  
  if (!m_samples.empty()) {
    std::vector<short> oldSamples = samples;
    samples.clear();
    for (std::map<unsigned int, bool>::const_iterator i = m_samples.begin();
         i != m_samples.end(); ++i) 
    {
      samples.push_back(oldSamples[i->first]);
      if (i == m_samples.begin()) time = time - data.time(i->first);
    }
  }
 
  if (m_refit) {
    SimpleShape* reference = data.referenceShape();
    const ScaledErrorData* sed = data.scaledErrorData();
    ShapeFitter fitter(m_fitParams);
    double chi2{};
    if (!reference){
      delete sed;
      return nullptr;
    } 
    bool result = fitter.fit(data, *reference, k, deltaT, chi2, sed);
    delete sed;
    if (!result) return nullptr;
  }
 
  if (m_adjust) {
    if (!data.history() || data.adcMax() == 0) return nullptr;
    OFC* ofc = data.history()->ofc(data.index());
    if (!ofc) return nullptr;
    k = ofc->A(data)/data.adcMax();
    deltaT = ofc->time(data);
  }
  
  if (evtIndex < 0) evtIndex = data.container().eventIndex();
  
  DataContainer* newContainer = new DataContainer(data.container().gain(), samples, corrs,
                evtIndex,
                data.container().energy(), time + deltaT, data.container().quality(),
                data.container().pedestal(), data.container().pedestalRMS(),
                data.container().status(), k*data.container().adcMax());
  Data* newData = new Data(*newContainer, *data.eventData(), data.history(), data.index(), true);
  return newData;
}

tweak() [2/2]

EventData * DataTweaker::tweak	(	const EventData &	eventData,
		int	runIndex = -1
	)		const

Definition at line 116 of file DataTweaker.cxx.

{
  EventData* newEventData = new EventData(eventData, runIndex);
  if (m_removeRoIs) newEventData->removeRoIs();
  return newEventData;
}

Member Data Documentation

m_adjust

bool LArSamples::DataTweaker::m_adjust {false}

private

Definition at line 47 of file DataTweaker.h.

m_fitParams

Chi2Params LArSamples::DataTweaker::m_fitParams {DefaultChi2}

private

Definition at line 49 of file DataTweaker.h.

m_nSamplesTruncation

unsigned int LArSamples::DataTweaker::m_nSamplesTruncation {0}

private

Definition at line 46 of file DataTweaker.h.

m_refit

bool LArSamples::DataTweaker::m_refit {false}

private

Definition at line 47 of file DataTweaker.h.

m_removeNoiseCorrelations

bool LArSamples::DataTweaker::m_removeNoiseCorrelations {false}

private

Definition at line 47 of file DataTweaker.h.

m_removeRoIs

bool LArSamples::DataTweaker::m_removeRoIs {false}

private

Definition at line 47 of file DataTweaker.h.

m_samples

std::map<unsigned int, bool> LArSamples::DataTweaker::m_samples

private

Definition at line 48 of file DataTweaker.h.

---

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

• DataTweaker.h
• DataTweaker.cxx