Loaders.cxx

Go to the documentation of this file.

#include "CAFAna/Core/Loaders.h"

#include "CAFAna/Core/SpectrumLoader.h"
#include "CAFAna/Core/Utilities.h"

#include <cassert>
#include <iostream>

namespace ana
{
  //----------------------------------------------------------------------
  Loaders::Loaders()
    : fSpillCut(0), fND(false)
  {
  }

  //----------------------------------------------------------------------
  Loaders::~Loaders()
  {
    for(auto it: fLoaders) delete it.second;
    if(fSpillCut) delete fSpillCut;
  }

  //----------------------------------------------------------------------
  void Loaders::SetLoaderPath(const std::string& path,
                              caf::Det_t det,
                              DataMC datamc,
                              DataSource src,
                              SwappingConfig swap)
  {
    assert(datamc == kMC || swap == kNonSwap);
    if (!fND) assert(det == caf::kFARDET || swap == kNonSwap);
    else assert(det == caf::kNEARDET || swap == kNonSwap);
    assert(src == kBeam || swap == kNonSwap);

    const Key_t key(det, datamc, src, swap);

    // Clear out the old one if necessary
    DisableLoader(det, datamc, src, swap);

    fLoaderPaths[key] = path;
  }

  //----------------------------------------------------------------------
  void Loaders::SetLoaderFiles(const std::vector<std::string>& files,
                               caf::Det_t det,
                               DataMC datamc,
                               DataSource src,
                               SwappingConfig swap)
  {
    assert(datamc == kMC || swap == kNonSwap);
    if (!fND) assert(det == caf::kFARDET || swap == kNonSwap);
    else assert(det == caf::kNEARDET || swap == kNonSwap);
    assert(src == kBeam || swap == kNonSwap);

    const Key_t key(det, datamc, src, swap);

    // Clear out the old one if necessary
    DisableLoader(det, datamc, src, swap);

    fLoaderFiles[key] = files;
  }

  //----------------------------------------------------------------------
  void Loaders::DisableLoader(caf::Det_t det,
                              DataMC datamc,
                              DataSource src,
                              SwappingConfig swap)
  {
    assert(datamc == kMC || swap == kNonSwap);
    if (!fND) assert(det == caf::kFARDET || swap == kNonSwap);
    else assert(det == caf::kNEARDET || swap == kNonSwap);
    assert(src == kBeam || swap == kNonSwap);

    const Key_t key(det, datamc, src, swap);

    // Clear out the current one if possible
    auto it = fLoaders.find(key);
    if(it != fLoaders.end()){
      delete it->second;
      fLoaders.erase(it);
    }

    fLoaderPaths.erase(key);
    fLoaderFiles.erase(key);
  }

  //----------------------------------------------------------------------
  std::string Loaders::GetLoaderPath (caf::Det_t det,
                                      DataMC datamc,
                                      DataSource src,
                                      SwappingConfig swap) const 
  {
    assert(datamc == kMC || swap == kNonSwap);
    if (!fND) assert(det == caf::kFARDET || swap == kNonSwap);
    else assert(det == caf::kNEARDET || swap == kNonSwap);
    assert(src == kBeam || swap == kNonSwap);
    
    const Key_t key(det, datamc, src, swap);

    return fLoaderPaths.at(key);
  }

  //----------------------------------------------------------------------
  std::vector<std::string> Loaders::GetLoaderFiles(caf::Det_t det,
                                                   DataMC datamc,
                                                   DataSource src,
                                                   SwappingConfig swap) const
  {
    assert(datamc == kMC || swap == kNonSwap);
    if (!fND) assert(det == caf::kFARDET || swap == kNonSwap);
    else assert(det == caf::kNEARDET || swap == kNonSwap);
    assert(src == kBeam || swap == kNonSwap);
    
    const Key_t key(det, datamc, src, swap);
    
    return fLoaderFiles.at(key);
  }

  //----------------------------------------------------------------------
  void Loaders::SetSpillCut(const SpillCut& cut)
  {
    for(auto it: fLoaders) it.second->SetSpillCut(cut);
    if(fSpillCut) delete fSpillCut;
    fSpillCut = new SpillCut(cut);
  }

  //----------------------------------------------------------------------
  SpectrumLoaderBase& Loaders::GetLoader(caf::Det_t det,
                                         DataMC datamc,
                                         DataSource src,
                                         SwappingConfig swap)
  {
    assert(datamc == kMC || swap == kNonSwap);
    if (!fND) assert(det == caf::kFARDET || swap == kNonSwap);
    else assert(det == caf::kNEARDET || swap == kNonSwap);
    assert(src == kBeam || swap == kNonSwap);

    const Key_t key(det, datamc, src, swap);

    // Look up and return. Use fNull if no loader is set for this config
    auto itLoader = fLoaders.find(key);
    if(itLoader != fLoaders.end()) return *itLoader->second;

    auto itPath = fLoaderPaths.find(key);
    if(itPath != fLoaderPaths.end()){
      fLoaders[key] = new SpectrumLoader(itPath->second);
      if(fSpillCut) fLoaders[key]->SetSpillCut(*fSpillCut);
      return *fLoaders[key];
    }
    auto itFiles = fLoaderFiles.find(key);
    if(itFiles != fLoaderFiles.end()){
      fLoaders[key] = new SpectrumLoader(itFiles->second);
      if(fSpillCut) fLoaders[key]->SetSpillCut(*fSpillCut);
      return *fLoaders[key];
    }

    return fNull;
  }

  //----------------------------------------------------------------------
  void Loaders::Go()
  {
    for(auto it: fLoaders)
    {
      if (!it.second->Gone())
        it.second->Go();
    }
  }
}