HardTree.h

// -*- C++ -*-
#ifndef HERWIG_HardTree_H
#define HERWIG_HardTree_H
//
// This is the declaration of the HardTree class.
//
 
#include "ThePEG/Config/ThePEG.h"
#include "Herwig/Shower/QTilde/Base/ShowerProgenitor.h"
#include "Herwig/Shower/QTilde/Base/ShowerTree.h"
#include "Herwig/Shower/QTilde/SplittingFunctions/SudakovFormFactor.h"
#include "Herwig/Shower/RealEmissionProcess.fh"
#include "HardBranching.h"
#include "HardTree.fh"
 
namespace Herwig {
 
using namespace ThePEG;
class HardTree : public Base {
 
  friend ostream & operator << (ostream &, const HardTree & );
 
public:
 
  HardTree(vector<HardBranchingPtr>, vector<HardBranchingPtr>, ShowerInteraction);
 
  HardTree(RealEmissionProcessPtr real);
 
  bool connect(ShowerParticlePtr particle, HardBranchingPtr branching) {
    if(branchings_.find(branching)==branchings_.end()) return false;
    particles_[particle]=branching;
    return true;
  }
 
  bool connect(ShowerTreePtr);
  
  map<ShowerParticlePtr,tHardBranchingPtr> & particles() 
  {return particles_;}
 
  set<HardBranchingPtr> & branchings() {return branchings_;}
 
  set<HardBranchingPtr> & incoming() {return spacelike_;}
 
  ShowerInteraction interaction() {return interaction_;}
 
  LorentzRotation showerRot() { return showerRot_; }
 
  void showerRot( LorentzRotation r ) { showerRot_ = r; }
 
  bool partnersSet() const {return partnersSet_;}
 
  void partnersSet(bool in) {partnersSet_=in;}
 
private:
 
  ShowerInteraction interaction_;
 
  ShowerTreePtr _tree;
 
  map<ShowerParticlePtr,tHardBranchingPtr> particles_;
 
  set<HardBranchingPtr> branchings_;
 
  set<HardBranchingPtr> spacelike_;
 
  LorentzRotation showerRot_;
 
  bool partnersSet_;
 
};
 
  ostream & operator << (ostream &, const HardTree & );
 
}
 
#endif /* HERWIG_HardTree_H */