ParticleWeightingKeyMap.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Ilya Komarov                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
#include <map>
#include <vector>
#include <string>
#include <analysis/dbobjects/ParticleWeightingAxis.h>
 
 
namespace Belle2 {
  typedef std::pair<std::vector<int>, int> MultiDimBin;
 
  typedef std::map<std::string, ParticleWeightingBinLimits*> NDBin;
 
  class ParticleWeightingKeyMap {
 
    std::map<std::string, ParticleWeightingAxis*> m_axes;
    std::vector<MultiDimBin> m_bins;
 
    const int m_outOfRangeBin = -1;
 
  public:
 
    ParticleWeightingKeyMap() {}
 
    void addAxis(const std::string& name);
 
    int addKey(NDBin bin, int key_ID);
 
    double addKey(NDBin bin);
 
 
    double getKey(std::map<std::string, double> values) const;
 
    std::vector<std::string> getNames() const;
 
    void printKeyMap() const;
 
  };
 
} // Belle2 namespace