GGSRunManagerExtensions.h

/*
 * GGSRunManagerExtensions.h
 *
 *  Created on: 5 Nov 2020
 *      Author: Nicola Mori
 */

#ifndef GGSRUNMANAGEREXTENSIONS_H_
#define GGSRUNMANAGEREXTENSIONS_H_

#include "globals.hh"

#include "montecarlo/generators/GGSGeneratorAction.h"

#include "G4GenericMessenger.hh"
#include "G4RunManager.hh"
#include "G4VUserPrimaryGeneratorAction.hh"
#include "Randomize.hh"

#include <array>

class GGSRunManagerExtensions {
public:
  GGSRunManagerExtensions();

  virtual ~GGSRunManagerExtensions() = default;

  int GetNKilledEvents() { return _nKilledEvs; }

  void KillEvent();

  void SimulateAgainKilledEvent() { _simAgainKilledEv = true; };

  bool IsCurrentEventKilled() { return _isCurrEvKilled; };

  const std::string &GetRandomStateAtBeginOfEvent() { return _stateAtBeginOfEvent; }

  void SetGGSGeneratorAction(G4VUserPrimaryGeneratorAction *userAction);

  int GetNDiscardedEvents();

  const std::string &RngStateRestoredFrom() { return _restoredFrom; }

protected:
  void BeginOfEventProcessing();

  void BeginOfEventSimulation();

  void DumpStatus(G4int eventID);

  bool IsKilledAndToBeSimulatedAgain(const G4Event *ev);

  void PrintLogVols();

  void HandleEventRngStateFile(int runID, int eventID);

private:
  int _nDiscardedEvsInKilledEvs; // Total discarded events in killed events
  int _nKilledEvs;
  bool _isCurrEvKilled;
  bool _simAgainKilledEv;

  std::string _stateAtBeginOfEvent;
  G4int _printModulo;
  G4int _lastPrinted;

  GGSGeneratorAction *_ggsGeneratorAction;

  std::string _eventRandomFileBase, _eventRandomFilePath, _restoredFrom;
  bool _readEventRngStateFromFile, _saveEventRngStateToFile;

  G4GenericMessenger _messenger, _randomMessenger;
};

#endif /* GGSRUNMANAGEREXTENSIONS_H_ */