test_AdcPedestalFitter.cxx File Reference

#include <string>
#include <iostream>
#include <fstream>
#include "dunecore/DuneInterface/Tool/TpcDataTool.h"
#include "dunecore/ArtSupport/DuneToolManager.h"
#include "TError.h"
#include <cassert>

Go to the source code of this file.

Functions
void	TestErrorHandler (Int_t, Bool_t, const char *, const char *msg)
int	test_AdcPedestalFitter (bool useExistingFcl, bool doUpdate, bool doUpdateMap)
int	main (int argc, char *argv[])

Function Documentation

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 197 of file test_AdcPedestalFitter.cxx.

                                 {
  bool useExistingFcl = false;
  bool doUpdate = true;
  bool doUpdateMap = true;
  if ( argc > 1 ) {
    string sarg(argv[1]);
    if ( sarg == "-h" ) {
      cout << "Usage: " << argv[0] << " [ARG]" << endl;
      cout << "  If ARG = true, existing FCL file is used." << endl;
      return 0;
    }
    useExistingFcl = sarg == "true" || sarg == "1";
  }
  return test_AdcPedestalFitter(useExistingFcl, doUpdate, doUpdateMap);
}

int test_AdcPedestalFitter	(	bool	useExistingFcl,
		bool	doUpdate,
		bool	doUpdateMap
	)

Definition at line 34 of file test_AdcPedestalFitter.cxx.

                                                                                 {
  const string myname = "test_AdcPedestalFitter: ";
#ifdef NDEBUG
  cout << myname << "NDEBUG must be off." << endl;
  abort();
#endif
  string line = "-----------------------------";

  // Set a Root error handler and make sure it is not called when
  // the tool is used.
  SetErrorHandler(TestErrorHandler);

  cout << myname << line << endl;
  string fclfile = "test_AdcPedestalFitter.fcl";
  if ( ! useExistingFcl ) {
    cout << myname << "Creating top-level FCL." << endl;
    ofstream fout(fclfile.c_str());
    fout << "#include \"test_dataprep.fcl\"" << endl;   // For adcNameManipulator
    fout << "tools.mytool: {" << endl;
    fout << "  tool_type: AdcPedestalFitter" << endl;
    fout << "  LogLevel: 1" << endl;
    fout << "  AdcRange: \"pow(2,12)\"" << endl;
    fout << "  FitOpt: 3" << endl;
    fout << "  FitPrecision: 1.0" << endl;
    fout << "  SkipFlags: []" << endl;
    fout << "  AdcFitRange: \"10*int(0.7*[gain]+1)\"" << endl;
    fout << "  FitRmsMin: \"1.0\"" << endl;
    fout << "  FitRmsMax: \"20.0\"" << endl;
    fout << "  RemoveStickyCode: false" << endl;
    fout << "  HistName: \"adcped_%EVENT%_%CHAN%\"" << endl;
    fout << "  HistTitle: \"ADC pedestal for event %EVENT% channel %CHAN%\"" << endl;
    fout << "  PlotFileName: \"adcped_ev%EVENT%_chan%CHAN%.png\"" << endl;
    fout << "  RootFileName: \"adcped.root\"" << endl;
    fout << "  HistManager: \"\"" << endl;
    fout << "  PlotSizeX:  700" << endl;
    fout << "  PlotSizeY:  500" << endl;
    fout << "  PlotShowFit:  2" << endl;
    fout << "  PlotSplitX:  0" << endl;
    fout << "  PlotSplitY:  0" << endl;
    fout << "}" << endl;
    fout << "tools.mymaptool: @local::tools.mytool" << endl;
    fout << "tools.mymaptool.LogLevel: 2" << endl;
    fout << "tools.mymaptool.PlotFileName: \"adcpedmap_ev%EVENT%_chan%CHAN%.png\"" << endl;
    fout << "tools.mymaptool.PlotSplitX: 2" << endl;
    fout << "tools.mymaptool.RootFileName: \"\"" << endl;
    fout << "tools.mymaptool.PlotSizeX: 1400" << endl;
    fout << "tools.mymaptool.PlotSizeY: 1000" << endl;
    fout << "tools.runDataTool: {" << endl;
    fout << "  tool_type: FclRunDataTool" << endl;
    fout << "  LogLevel: 1" << endl;
    fout << "  FileNames: [\"rundata.fcl\"]" << endl;
    fout << "}" << endl;
    fout.close();
    ofstream fout2("rundata.fcl");
    fout2 << "run: 123" << endl;
    fout2 << "gain: 14.0" << endl;
    fout2 << "shaping: 2.0" << endl;
    fout2.close();
  } else {
    cout << myname << "Using existing top-level FCL." << endl;
  }

  cout << myname << line << endl;
  cout << myname << "Fetching tool manager." << endl;
  DuneToolManager* ptm = DuneToolManager::instance(fclfile);
  assert ( ptm != nullptr );
  DuneToolManager& tm = *ptm;
  tm.print();
  assert( tm.toolNames().size() > 1 );

  cout << myname << line << endl;
  cout << myname << "Fetching histogram manaager." << endl;
  auto phm = tm.getShared<TpcDataTool>("mytool");
  assert( phm != nullptr );

  cout << myname << line << endl;
  cout << myname << "Fetching tool." << endl;
  auto padvNotUsed = tm.getPrivate<TpcDataTool>("mytool");
  assert( padvNotUsed != nullptr );
  auto padvsin = tm.getPrivate<TpcDataTool>("mytool");
  assert( padvsin != nullptr );
  auto padvmap = tm.getPrivate<TpcDataTool>("mymaptool");
  assert( padvmap != nullptr );
  if ( ! doUpdate ) padvsin = nullptr;
  if ( ! doUpdateMap ) padvmap = nullptr;

  cout << myname << line << endl;
  cout << myname << "Create data and call tool." << endl;
  AdcIndex nevt = 2;
  string lab = "plane 3u";
  float peds[10] = {701.1, 711.2, 733.3, 690.4, 688.5, 703.6, 720.7, 720.8, 695.9, 702.0};

  for ( AdcIndex ievt=0; ievt<nevt; ++ievt ) {
    cout << myname << "Event " << ievt << endl;
    AdcChannelDataMap datamap;
    AdcIndex ncha = 10;
    for ( AdcIndex icha=0; icha<ncha; ++icha ) {
      std::pair<AdcChannelDataMap::iterator, bool> kdat = datamap.emplace(icha, AdcChannelData());
      assert(kdat.second);
      AdcChannelDataMap::iterator idat = kdat.first;
      AdcChannelData& data = idat->second;
      float ped = peds[icha];
      data.setEventInfo(101, 23);
      data.setChannelInfo(icha);
      data.pedestal = ped;
      for ( AdcIndex itic=0; itic<100; ++itic ) {
        float xadc = ped + rand()%20 - 10.0;
        AdcIndex iticeff = itic - 3*icha;
        if ( iticeff > 20 && iticeff < 40 ) xadc +=600;
        AdcCount iadc = xadc;
        data.raw.push_back(iadc);
        data.flags.push_back(0);
        data.samples.push_back(iadc - ped);
        data.signal.push_back(xadc - ped > 300.0 );
      }
      data.pedestal += 10.0;
      AdcIndex tp = 10*ievt + 60 - 2.3*icha;
      AdcIndex tm = tp - 8;
      data.raw[tp] += 100;
      data.samples[tp] += 100;
      data.raw[tm] -= 100;
      data.samples[tm] -= 100;
      data.flags[tm] = 4;
      data.roisFromSignal();
      if ( padvsin != nullptr ) {
        double ped0 = datamap[icha].pedestal;
        //assert( padv->view(datamap[icha]) == 0 );
        double ped1 = datamap[icha].pedestal;
        assert( ! datamap[icha].hasMetadata("fitPedPeakBinFraction") );
        assert( padvsin->update(datamap[icha]) == 0 );
        double ped2 = datamap[icha].pedestal;
        cout << myname << "Old pedestal: " << ped0 << endl;
        cout << myname << "New pedestal: " << ped2 << endl;
        assert( ped1 == ped0 );
        assert( ped2 != ped0 );
        assert( ped2 != 0.0 );
        assert( datamap[icha].hasMetadata("fitPedPeakBinFraction") );
        assert( datamap[icha].hasMetadata("fitPedFractionLow") );
        assert( datamap[icha].hasMetadata("fitPedFractionHigh") );
        //assert( fabs(ped2-ped) < 0.01 );
      }
    }
    if ( padvmap != nullptr ) {
      for ( AdcIndex icha=0; icha<ncha; ++icha ) datamap[icha].metadata.clear();
      assert( ! datamap[0].hasMetadata("fitPedPeakBinFraction") );
      assert( padvmap->updateMap(datamap) == 0 );
      string mname = "fitPedPeakBinFraction";
      for ( AdcIndex icha=0; icha<ncha; ++icha ) {
        cout << myname << "Checking channel " << icha << endl;
        cout << myname << "  Metadata size: " << datamap[icha].metadata.size() << endl;
        cout << myname << "  " << mname << " = " << datamap[icha].metadata[mname] << endl;
        assert( datamap[icha].hasMetadata(mname) );
      }
    }
  }

  cout << myname << line << endl;
  cout << myname << "Done." << endl;
  return 0;
}

void TestErrorHandler	(	Int_t	,
		Bool_t	,
		const char *	,
		const char *	msg
	)

Definition at line 26 of file test_AdcPedestalFitter.cxx.

                                                                   {
  const string myname = "TestErrorHandler: ";
  cout << myname << msg << endl;
  abort();
}