File indexing completed on 2024-09-27 07:02:49
0001
0002
0003
0004
0005
0006 #ifndef HEPMC3_READERFACTORY_H
0007 #define HEPMC3_READERFACTORY_H
0008
0009 #include "HepMC3/ReaderAscii.h"
0010 #include "HepMC3/ReaderAsciiHepMC2.h"
0011 #include "HepMC3/ReaderHEPEVT.h"
0012 #include "HepMC3/ReaderLHEF.h"
0013 #include "HepMC3/ReaderPlugin.h"
0014
0015 #include <memory>
0016 #include <string>
0017 #include <sys/stat.h>
0018 #include <string.h>
0019
0020 namespace HepMC3 {
0021
0022
0023 std::shared_ptr<Reader> deduce_reader(std::istream &stream);
0024
0025
0026
0027
0028
0029
0030
0031
0032
0033 std::shared_ptr<Reader> deduce_reader(const std::string &filename)
0034 {
0035 std::string libHepMC3rootIO="libHepMC3rootIO.so.3";
0036 #if defined(__darwin__) || defined(__APPLE__)
0037 libHepMC3rootIO="libHepMC3rootIO.dydl";
0038 #endif
0039 #if (defined(WIN32) || defined(_WIN32) || defined(__WIN32)) && !defined(__CYGWIN__)
0040 libHepMC3rootIO="HepMC3rootIO.dll";
0041 #endif
0042 bool remote=false;
0043 bool pipe=false;
0044 if (filename.find("http://")!=std::string::npos) remote=true;
0045 if (filename.find("https://")!=std::string::npos) remote=true;
0046 if (filename.find("root://")!=std::string::npos) remote=true;
0047 if (filename.find("gsidcap://")!=std::string::npos) remote=true;
0048
0049 std::vector<std::string> head;
0050 if (!remote)
0051 {
0052 struct stat buffer;
0053 #if (defined(WIN32) || defined(_WIN32) || defined(__WIN32)) && !defined(__CYGWIN__)
0054 if (!(stat (filename.c_str(), &buffer)==0))
0055 #else
0056 if (!(stat (filename.c_str(), &buffer)==0 && (S_ISFIFO(buffer.st_mode)|| S_ISREG(buffer.st_mode) || S_ISLNK(buffer.st_mode))))
0057 #endif
0058 {
0059 HEPMC3_ERROR("deduce_reader: file " << filename << " does not exist or is not a regular file/FIFO/link");
0060 return std::shared_ptr<Reader> (nullptr);
0061 }
0062
0063 std::ifstream* file= new std::ifstream(filename);
0064 if (!file)
0065 {
0066 HEPMC3_ERROR("deduce_reader could not open file for testing HepMC version: " << filename);
0067 return std::shared_ptr<Reader>(nullptr);
0068 }
0069 if (!file->is_open()) {
0070 HEPMC3_ERROR("deduce_reader could not open file for testing HepMC version: " << filename);
0071 file->close();
0072 return std::shared_ptr<Reader>(nullptr);
0073 }
0074
0075 #if defined(__linux__) || defined(__darwin__)|| defined(__APPLE__) || defined(__FreeBSD__) || defined(__sun)
0076 pipe = S_ISFIFO(buffer.st_mode);
0077 if (pipe) {
0078 HEPMC3_DEBUG(0, "deduce_reader: the file " << filename << " is a pipe");
0079 return deduce_reader(*file);
0080 }
0081 #endif
0082
0083 std::string line;
0084 size_t nonempty = 0;
0085 while (std::getline(*file, line) && nonempty < 3) {
0086 if (line.empty()) continue;
0087 nonempty++;
0088 head.push_back(line);
0089 }
0090 file->close();
0091 delete file;
0092 }
0093
0094 head.push_back("");
0095 head.push_back("");
0096 HEPMC3_DEBUG(0, "deduce_reader: Attempt ReaderRootTree for " << filename);
0097 if ( strncmp(head.at(0).c_str(),"root",4) == 0||remote)
0098 return std::make_shared<ReaderPlugin>(filename,libHepMC3rootIO,std::string("newReaderRootTreefile"));
0099 if (!remote)
0100 {
0101 HEPMC3_DEBUG(0, "Attempt ReaderAscii for " << filename);
0102 if ( strncmp(head.at(0).c_str(),"HepMC::Version",14) == 0 && strncmp(head.at(1).c_str(),"HepMC::Asciiv3",14)==0 )
0103 return std::shared_ptr<Reader>((Reader*) ( new ReaderAscii(filename)));
0104 HEPMC3_DEBUG(0, "Attempt ReaderAsciiHepMC2 for " << filename);
0105 if ( strncmp(head.at(0).c_str(),"HepMC::Version",14) == 0 && strncmp(head.at(1).c_str(),"HepMC::IO_GenEvent",18)==0 )
0106 return std::shared_ptr<Reader>((Reader*) ( new ReaderAsciiHepMC2(filename)));
0107 HEPMC3_DEBUG(0, "Attempt ReaderLHEF for " << filename);
0108 if ( strncmp(head.at(0).c_str(),"<LesHouchesEvents",17) == 0)
0109 return std::shared_ptr<Reader>((Reader*) ( new ReaderLHEF(filename)));
0110 HEPMC3_DEBUG(0, "Attempt ReaderHEPEVT for " << filename);
0111 std::stringstream st_e(head.at(0).c_str());
0112 char attr=' ';
0113 bool HEPEVT=true;
0114 int m_i,m_p;
0115 while (true)
0116 {
0117 if (!(st_e>>attr)) {
0118 HEPEVT=false;
0119 break;
0120 }
0121 if (attr==' ') continue;
0122 if (attr!='E') {
0123 HEPEVT=false;
0124 break;
0125 }
0126 HEPEVT=static_cast<bool>(st_e>>m_i>>m_p);
0127 break;
0128 }
0129 if (HEPEVT) return std::shared_ptr<Reader>((Reader*) ( new ReaderHEPEVT(filename)));
0130 }
0131 HEPMC3_DEBUG(0, "deduce_reader: all attempts failed for " << filename);
0132 return std::shared_ptr<Reader>(nullptr);
0133 }
0134
0135
0136
0137 std::shared_ptr<Reader> deduce_reader(std::istream &stream)
0138 {
0139 std::vector<std::string> head;
0140 head.push_back("");
0141 size_t back=0;
0142 size_t backnonempty=0;
0143 while ( (back < 200 && backnonempty < 100) && stream) {
0144 char c = stream.get(); back++;
0145 if (c=='\n') {
0146 if (head.back().length()!=0) head.push_back("");
0147 } else {
0148 head.back()+=c; backnonempty++;
0149 }
0150 }
0151 if (!stream)
0152 {
0153 HEPMC3_WARNING("Input stream is too short or invalid.");
0154 return std::shared_ptr<Reader>(nullptr);
0155 }
0156
0157 for (size_t i=0; i<back; i++) stream.unget();
0158
0159 if ( strncmp(head.at(0).c_str(),"HepMC::Version",14) == 0 && strncmp(head.at(1).c_str(),"HepMC::Asciiv3",14)==0 )
0160 {
0161 HEPMC3_DEBUG(0, "Attempt ReaderAscii");
0162 return std::shared_ptr<Reader>((Reader*) ( new ReaderAscii(stream)));
0163 }
0164
0165 if ( strncmp(head.at(0).c_str(),"HepMC::Version",14) == 0 && strncmp(head.at(1).c_str(),"HepMC::IO_GenEvent",18)==0 )
0166 {
0167 HEPMC3_DEBUG(0, "Attempt ReaderAsciiHepMC2");
0168 return std::shared_ptr<Reader>((Reader*) ( new ReaderAsciiHepMC2(stream)));
0169 }
0170
0171 if ( strncmp(head.at(0).c_str(),"<LesHouchesEvents",17) == 0)
0172 {
0173 HEPMC3_DEBUG(0, "Attempt ReaderLHEF");
0174 return std::shared_ptr<Reader>((Reader*) ( new ReaderLHEF(stream)));
0175 }
0176 HEPMC3_DEBUG(0, "Attempt ReaderHEPEVT");
0177 std::stringstream st_e(head.at(0).c_str());
0178 char attr=' ';
0179 bool HEPEVT=true;
0180 int m_i,m_p;
0181 while (true)
0182 {
0183 if (!(st_e>>attr)) {
0184 HEPEVT=false;
0185 break;
0186 }
0187 if (attr==' ') continue;
0188 if (attr!='E') {
0189 HEPEVT=false;
0190 break;
0191 }
0192 HEPEVT=static_cast<bool>(st_e>>m_i>>m_p);
0193 break;
0194 }
0195 if (HEPEVT) return std::shared_ptr<Reader>((Reader*) ( new ReaderHEPEVT(stream)));
0196 HEPMC3_DEBUG(0, "deduce_reader: all attempts failed");
0197 return std::shared_ptr<Reader>(nullptr);
0198 }
0199 }
0200 #endif