Ticket #80: event-selector.patch

SHERPA/Tools/Input_Output_Handler.C

@@ -34 +34 @@
 #ifdef USING__HEPMC2
   p_hepmc2(NULL),
 #endif
-  p_hepevt(NULL)
+  p_hepevt(NULL),
+  p_eventhandler(NULL), p_mehandler(NULL)
 {
   InitialiseOutput(dr);
 }
@@ -167 +168 @@
 bool Input_Output_Handler::OutputToFormat(ATOOLS::Blob_List *const blobs)
 {
   double weight=blobs->Weight();
-  weight/=p_mehandler->TotalXS()*rpa.Picobarn();
+  if (p_mehandler) weight/=p_mehandler->TotalXS()*rpa.Picobarn();
 
-  double xs(p_eventhandler->TotalXS()), xserr(p_eventhandler->TotalErr());
+  double xs(1.0), xserr(0.0);
+  if (p_eventhandler) {
+    xs=p_eventhandler->TotalXS();
+    xserr=p_eventhandler->TotalErr();
+  }
   for (map<string,Output_Base *>::iterator oit=m_outmap.begin();
        oit!=m_outmap.end();oit++) {
     oit->second->SetXS(xs, xserr);
@@ -178 +183 @@
   
   m_evtnumber++;
   m_evtcount++;
-  
+
   if (m_evtcount%m_filesize==0) {
     m_evtcount = 0;
     string number(ToString(int(m_evtnumber/m_filesize)));

AddOns/Analysis/Observables/Event_Output.C

@@ +1 @@
+#include "AddOns/Analysis/Observables/Primitive_Observable_Base.H"
+#include "ATOOLS/Org/MyStrStream.H"
+#include "ATOOLS/Org/Data_Reader.H"
+#include "ATOOLS/Org/Exception.H"
+
+#include "SHERPA/Tools/Input_Output_Handler.H"
+#include "AddOns/Analysis/Main/Primitive_Analysis.H"
+
+using namespace ANALYSIS;
+using namespace ATOOLS;
+using namespace SHERPA;
+
+namespace ANALYSIS {
+
+  class Event_Output : public Primitive_Observable_Base {
+    Input_Output_Handler* p_io;
+    std::string m_inlist;
+  public:
+
+    Event_Output(Input_Output_Handler* io, const std::string& inlist) :
+      Primitive_Observable_Base(), p_io(io), m_inlist(inlist)
+    {
+    }
+
+    ~Event_Output()
+    {
+      if (p_io) delete p_io; p_io=NULL;
+    }
+
+
+    void Evaluate(const ATOOLS::Blob_List & blobs, double value, double ncount)
+    {
+      if (!p_io) return;
+      Particle_List * pl=p_ana->GetParticleList(m_inlist);
+      if (pl->empty()) return;
+      Blob_List* blobsptr=const_cast<Blob_List*>(&blobs);
+      p_io->OutputToFormat(blobsptr);
+    }
+
+
+    Primitive_Observable_Base * Copy() const
+    {
+      // don't duplicate event output
+      return new Event_Output(NULL, "");
+    }
+
+  };// end of class Event_Output
+
+}
+
+
+
+DECLARE_GETTER(Event_Output_Getter,"Event_Output",
+               Primitive_Observable_Base,Argument_Matrix);
+
+Primitive_Observable_Base *
+Event_Output_Getter::operator()(const Argument_Matrix &parameters) const
+{
+  Data_Reader reader(" ",";","!","=");
+  reader.AddComment("#");
+  reader.AddWordSeparator("\t");
+  std::string inlist="";
+  for (size_t i=0; i<parameters.size(); ++i) {
+    std::string line = "";
+    for (size_t j=0; j<parameters[i].size(); ++j) {
+      line += parameters[i][j]+" ";
+    }
+    reader.AddFileContent(line);
+    if (parameters[i][0]=="InList" && parameters[i].size()>1) inlist=parameters[i][1];
+  }
+
+  if (inlist=="") {
+    THROW(fatal_error, "You didn't specify an InList for Event_Output");
+  }
+  return new Event_Output(new Input_Output_Handler(&reader), inlist);
+}
+
+void Event_Output_Getter::PrintInfo(std::ostream &str,const size_t width) const
+{
+  str<<"{\n"
+     <<std::setw(width+7)<<" "<<"InList  <triggeroutlist>\n"
+     <<std::setw(width+7)<<" "<<"# event output settings cf. manual, e.g.:\n"
+     <<std::setw(width+7)<<" "<<"HEPMC2_GENEVENT_OUTPUT <filename>\n"
+     <<std::setw(width+7)<<" "<<"FILE_SIZE <n>\n"
+     <<std::setw(width+7)<<" "<<"EVT_FILE_PATH <path>\n"
+     <<std::setw(width+4)<<" "<<"}";
+}

AddOns/Analysis/Observables/Makefile.am

@@ -5 +5 @@
 endif
 
 ANAOBS_SOURCES        = \
+                        Event_Output.C \
                         $(EXTRAOBS) \
                         Global_Calculator.C \
                         Primitive_Observable_Base.C \