#ifndef PythiaFix_h
#define PythiaFix_h

#include "ExRootAnalysis/ExRootModule.h"

class TClonesArray;
class TIterator;

class PythiaFix: public ExRootModule
{
public:

  PythiaFix();
  ~PythiaFix();

  void Init();
  void Process();
  void Finish();

private:

  void FixDaughters(Int_t indexParticle, Int_t indexMother);

  TClonesArray *fBranchParticle; //!
  TIterator *fItBranchParticle; //!

  ClassDef(PythiaFix, 1)
};

#endif