ParticleSubset.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
 
#include <framework/datastore/SelectSubset.h>
#include <analysis/dataobjects/Particle.h>
 
namespace Belle2 {
  class ParticleSubset : public SelectSubset<Particle> {
  public:
    void removeParticlesNotInLists(const std::vector<std::string>& listNames);
 
    void select(std::function<bool (const Particle*)> f)
    {
      const std::map<int, int>& oldToNewMap = copySetWithRelations(f);
 
      if (m_inheritToSelf) {
        copyRelationsToSelf();
      }
 
      if (m_reduceExistingSet) {
        swapSetsAndDestroyOriginal();
      }
 
      fixParticles(oldToNewMap);
 
      fixParticleLists(oldToNewMap);
    }
 
  protected:
    void fixParticles(const std::map<int, int>& oldToNewMap);
 
    void fixParticleLists(const std::map<int, int>& oldToNewMap);
 
    static void fixVector(std::vector<int>& vec, const std::map<int, int>& oldToNewMap);
 
  };
}