ReconstructionFrame.hh

Go to the documentation of this file.

//   RestFrames: particle physics event analysis library
//   --------------------------------------------------------------------
//   Copyright (c) 2014-2016, Christopher Rogan
//   This file is part of RestFrames.
//
//   RestFrames is free software; you can redistribute it and/or modify
//   it under the terms of the GNU General Public License as published by
//   the Free Software Foundation; either version 2 of the License, or
//   (at your option) any later version.
// 
//   RestFrames is distributed in the hope that it will be useful,
//   but WITHOUT ANY WARRANTY; without even the implied warranty of
//   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//   GNU General Public License for more details.
// 
//   You should have received a copy of the GNU General Public License
//   along with RestFrames. If not, see <http://www.gnu.org/licenses/>.

#ifndef ReconstructionFrame_HH
#define ReconstructionFrame_HH

#include "RestFrames/RestFrame.hh"
#include "RestFrames/Group.hh"

namespace RestFrames {
  
  // ReconstructionFrame class
  class ReconstructionFrame : public RestFrame {
  public:
    ReconstructionFrame(const std::string& sname, const std::string& stitle);
    ReconstructionFrame();
    virtual ~ReconstructionFrame();

    virtual void Clear();

    virtual void AddChildFrame(RestFrame& frame);

    virtual void RemoveChildFrame(RestFrame& frame);

    void RemoveChildFrames();
    
    virtual void SetParentFrame(RestFrame& frame = 
                RestFrame::Empty());

    virtual ReconstructionFrame const& GetParentFrame() const;

    virtual ReconstructionFrame& GetChildFrame(int i = 0) const;

    virtual void SetGroup(Group& group = Group::Empty());
    Group& GetGroup() const;
    GroupList GetListGroups() const;

    static ReconstructionFrame& Empty();

  protected:
    bool InitializeAnalysisRecursive();
    bool ClearEventRecursive();
    bool AnalyzeEventRecursive();

    virtual bool ResetRecoFrame();
    virtual bool ReconstructFrame();

    virtual StateList const& GetChildStates(int i = 0) const;
    virtual StateList const& GetChildStates(const RestFrame& child) const;
    
  private:
    Group* m_GroupPtr;
    mutable std::map<const RestFrame*, StateList > m_ChildStates;
    
    bool InitializeVisibleStates();
    bool InitializeGroupStates();
    void FillListGroupsRecursive(GroupList& groups) const;

  };

}

#endif