DataSet.h

Go to the documentation of this file.

#pragma once

#include <goofit/GlobalCudaDefines.h>

#include <initializer_list>
#include <set>
#include <string>
#include <vector>

#include <goofit/Error.h>
#include <goofit/Variable.h>

namespace GooFit {

class DataSet {
  public:
    DataSet(const Observable &var, std::string n = "");

    DataSet(const std::vector<Observable> &vars, std::string n = "");
    DataSet(const std::set<Observable> &vars, std::string n = "");
    DataSet(std::initializer_list<Observable> vars, std::string n = "");

    virtual ~DataSet() = default;

    virtual void addEvent() = 0; // Must increment numEventsAdded

    virtual void addWeightedEvent(fptype weight);

    template <typename... Args>
    void addEvent(fptype value, Args... args) {
        std::vector<fptype> values{value, static_cast<fptype>(args)...};

        if(values.size() != observables.size())
            throw GooFit::GeneralError("You must pass the correct number of values ({}) to addEvent",
                                       observables.size());

        for(size_t i = 0; i < values.size(); i++)
            observables[i].setValue(values[i]);
        addEvent();
    }

    const std::vector<Observable> &getObservables() const;

    size_t numVariables() const { return observables.size(); }

    size_t getNumEvents() const { return numEventsAdded; }

    std::string getName() const { return name; }

  protected:
    std::vector<fptype> getCurrentValues() const;
    size_t indexOfVariable(const Observable &var) const;
    size_t numEventsAdded{0};

    void checkAllVars() const;

  private:
    void generateName();

    std::string name;

  protected:
    std::vector<Observable> observables;
};

} // namespace GooFit