diff --git a/.gitignore b/.gitignore index c4d3cce..08756f9 100644 --- a/.gitignore +++ b/.gitignore @@ -1,9 +1,12 @@ -*.root +.vscode *.o *.swp -*.root *.log .depend +example/analysis +example/analysis_fullrec +example/data +example/*.root leaf/DataModelRootDict* leaf/HKAstroAnalysis.* leaf/DataModel diff --git a/README.md b/README.md index 0355e3a..b7afc7f 100644 --- a/README.md +++ b/README.md @@ -1,66 +1,81 @@ -# LEAF -Low Energy Algoritm Framework +# :seedling: LEAF :seedling: + +**L**ow **E**nergy **A**lgorithm **F**ramework + + +# DESCRIPTION This algorithm is an alternative and simple LE fitter than can be used for HyperK and SuperK. ~~~~~~~~~~~~~~~~~~~~~~~~~ 2020/02/09: LEAF was convert as a C++ class and can be included in your code. +~~~~~~~~~~~~~~~~~~~~~~~~~ -New compilation method: - source RunAtStart.sh - ./SetupDataModel.sh - cd leaf/ - make clean; make - -In order to use the class in your code look at example/ -~~~~~~~~~~~~~~~~~~~~~~~~~ +# DOWNLOAD AND PRE-REQUISITES + +To download this repository use : + +``` +git clone --single-branch --branch develop_lite https://github.com/hyperk/LEAF.git +``` + +Then make sure the following pre-requisites are installed. +1. ROOT v5r34 or superior (not tested for older versions, but might work). +2. WCSim version compatible with your ROOT version. +3. BONSAI installation (although LEAF can work without it). + +# COMPATIBILITY -# Pre-requisite to use the code: -1. BONSAI installation. -2. ROOT v5r34 or superior (not tested for older versions, but might work). -3. HKAstroAnalysis class is private and can be downloaded by SK collaborators on sukap cluster. +Versions listed below have been tested so far. -# Compatibility tested so far: 1. WCSim-hybrid version: for geometries "HyperK", "HyperK_mPMT", "HyperK_HybridmPMT", "HyperK_HybridmPMT10PC" 2. In general, with all WCSim-hybrid geometries using whether BoxandLine20inchHQE or PMT3inchR14374 PMTs. 3. With official HK WCSim: Ask G. Pronost. -# How to: -1. Source RunAtStart.sh after you updated your ROOT directory. -2. Use the script ./SetupDataModel.sh to define the DataModel (if you have hk-AstroAnalysis, you should setup the global variable) -2. Enter the leaf/ repository and make clean;make -3. Enter the example repository and make clean;make -4. One example of how to run the code is set in example: test_example.sh -5. inputs PDF, input from WCSim can be downloaded on sukap cluster. Please untar them in the LEAF repository. -6. You can use shell scripts in shell/ in order to run the fitter or launch on batch. - -# Useful scripts in ./macros and ./shell -You can compile with GNUMake like following in ./macros: + +# INSTALLATION + +1. Go to libWCSIM/ repository and make softlinks to your WCSim installation. +``` +$ cd libWCSIM/ +$ ln -s /path/to/your/WCSim/include/ include +$ ln -s /path/to/your/WCSim/src/ src +$ ln -s /path/to/your/WCSim/lib/libWCSimRoot.so libWCSimRoot.so +$ ln -s /path/to/your/WCSim/lib/libWCSimRoot.so.1.12.xx libWCSimRoot.so.1.12.xx ``` +2. Go to the cloned LEAF repository and source the script RunAtStart.sh after making sure you are sourcing the proper ROOT directory. +``` +$ cd /path/to/the/cloned/repository/LEAF +$ source ./RunAtStart.sh +``` +3. Enter the leaf/ repository and `make clean; make` +4. Enter the example/ repository and `make clean; make` +5. One example of how to run the code is set in example: test_example.sh + +# TUNING FILES + +Tuning files are provided in the inputs/ repository (e.g. ./inputs/timePDF_Directionality_DRnew.root) and are made from 10 MeV electrons generated isotropically and uniformly in the tank. +Alternatively, you can generate your own PDFs by following the commands below. + +1. Go to macros/ repository and compile AnalyzeWSHierarchy and ProducePDF with GNUMake. +``` +$ cd ./macros +$ make AnalyzeWSHierarchy $ make ProducePDF ``` -## Making tuning file (e.g. ./inputs/timePDF_Directionality_DRnew.root) -1. Produce plots by AnalyzeWSHierarchy: reads out WCSim output and makes plots. -2. Produce time PDF (and angular PDF) by ProducePDF: uses plots made by AnalyzeWSHierarchy and generate PDFs for LEAF. +2. Produce plots using AnalyzeWSHierarchy which reads out WCSim output. Optionally, you can index a range of events to read with -s (start index) and -e (end index). Use option -h if when using a hybrid HK geometry presenting both PMTs and mPMTs. ``` -$ AnalyzeWSHierarchy -f wcrim_hybrid.root -o plots.root -$ ProducePDF -f plots.root -o PDF.root +$ ./AnalyzeWSHierarchy -i wcrim.root -o plots.root -s 0 -e 1000 -h ``` -## Making generic plots -- LEAFOutputAnalysisHybrid_leafclass: read LEAF output to produce generic plots. If one uses the master branch for LEAF, please use LEAFOutputAnalysisHybrid_master +3. Produce time PDF and angular PDF using ProducePDF: uses plots made using AnalyzeWSHierarchy and generate PDFs for LEAF. +``` +$ ./ProducePDF -i wcsim.root -o PDF.root -s 0 -e 1000 -v +``` -## Shell scripts for analysis of large files -You can refere shell scripts in ./shell in order to analyze many files. -They are not working with latest LEAF class and its examples. They are just example how to analyze. +Pathes to PDF files required as inputs for LEAF are hard-coded inside leaf/LeafSplines.cc, inside the LoadSplines() methods. +Make sure these pathes are consistant with the files you generate on your own. -- generateShellXX.c: this is a root macro which generates shell scripts to be submitted to sukap by LauncherXX.sh - - generateShell.c - - generateShell_analyzeWCSim.c -- LauncherXX.sh: this submits jobs to sukap - - Launcher.sh - - Launcher_analyzeWCSim.sh -- Merger_analyzeWCSim.sh: merge generated output by Launcher_analyzeWCSim.sh diff --git a/SetupDataModel.sh b/SetupDataModel.sh deleted file mode 100755 index 55ccbe6..0000000 --- a/SetupDataModel.sh +++ /dev/null @@ -1,16 +0,0 @@ -#!/bin/bash - -if [ ! -z ./leaf/DataModel ]; then - echo "Remove previous link" - rm ./leaf/DataModel -fi - -if [ "$HK_ASTROANALYSIS_DIR" == "" ] || [ -z $HK_ASTROANALYSIS_DIR ]; then - echo "HK_ASTROANALYSIS_DIR is not defined or doesn't exist ($HK_ASTROANALYSIS_DIR). Will use lite DataModel." - ln -s ${PWD}/leaf/DataModel-lite ./leaf/DataModel -else - echo "HK_ASTROANALYSIS_DIR was found to be: $HK_ASTROANALYSIS_DIR its DataModel will be used" - ln -s $HK_ASTROANALYSIS_DIR/DataModelRoot ./leaf/DataModel -fi - - diff --git a/example/GNUmakefile b/example/GNUmakefile index ede518e..28a8b73 100644 --- a/example/GNUmakefile +++ b/example/GNUmakefile @@ -1,52 +1,62 @@ # Makefile by Guillaume Pronost for LEAF example @ 2020/02/09 OSNAME = $(shell uname -s) +LEAFDIR = .. include ../Makefile/Makefile.${OSNAME} - # set compiler options for ROOT CXXFLAGS += $(shell root-config --cflags) -CXXFLAGS += '-fPIC' -std=c++11 -Wall -Wpedantic -Wno-long-long +CXXFLAGS += -fPIC -std=c++17 -Wall -Wpedantic -Wno-long-long +ROOT_VERSION = $(shell root-config --version | cut -d'.' -f1,2) +ifeq ($(shell echo "$(ROOT_VERSION) >= 6.0" | bc -l), 1) + HK_USE_ROOT7 = 1 + CXXFLAGS += -DHK_USE_ROOT7 +endif -INCFLAGS = -I. -I$(shell root-config --incdir) -INCFLAGS += -I$(WCSIM_BUILD_DIR)/include/WCSim -# INCFLAGS += -I$(WCSIMDIR)/include -# INCFLAGS += -I$(BONSAIDIR)/bonsai +INCFLAGS = -I. +#INCFLAGS += -I$(shell root-config --incdir) #included in --cflags +INCFLAGS += -I$(LEAFDIR)/libWCSIM/include INCFLAGS += -I$(LEAFDIR)/leaf -INCFLAGS += -I$(LEAFDIR)/leaf/DataModel +INCFLAGS += -I$(LEAFDIR)/leaf/DataModel-lite -# LIBS += -L${WCSIMDIR} -lWCSimRoot -LIBS += -L${WCSIM_BUILD_DIR}/lib -lWCSimRoot +LIBS += -L$(LEAFDIR)/libWCSIM -lWCSimRoot LIBS += $(shell root-config --libs) -lMinuit # LIBS_BONSAI += -L${BONSAIDIR} -lWCSimBonsai -LIBS_LEAF += -L${LEAFDIR}/lib -lDataModelLite -lHKManager -lLEAF #-lHKAstroAnalysis - +LIBS_LEAF += -L${LEAFDIR}/lib -lLEAF -lHKManager -lDataModelLite -OBJECT = analysis +OBJECT = analysis analysis_fullrec -CXXFLAGS += -std=c++17 +all: ROOT_CHECK $(OBJECT) -all: $(OBJECT) +ROOT_CHECK: +ifeq ($(HK_USE_ROOT7),1) + @echo '<< compile with ROOT7 >>' +else + @echo '<< compile with old ROOT >>' + @echo $(DATAMODEL_SOURCES) +endif analysis: analysis.o - @echo '<< compiling bin analysis >>' - @$(CXX) -g $(CXXFLAGS) -o $@ $^ $(LIBS) $(LIBS_BONSAI) $(LIBS_LEAF) - - + @echo '<< compiling bin $@ >>' + @$(CXX) -g $(CXXFLAGS) $^ $(LIBS) $(LIBS_LEAF) -o $@ + +analysis_fullrec: analysis_fullrec.o + @echo '<< compiling bin $@ >>' + @$(CXX) -g $(CXXFLAGS) $^ $(LIBS) $(LIBS_LEAF) -o $@ + # default rules .cc.o: @echo '<< compiling' $< '>>' @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c $< - -%.o: %.cc %.hh + +.cpp.o: @echo '<< compiling' $< '>>' - @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c -o $@ $< + @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c $< -%.o: %.cpp +%.o: %.cpp %.hpp @echo '<< compiling' $< '>>' @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c -o $@ $< - %.o: %.C %.h @echo '<< compiling' $< '>>' diff --git a/example/GNUmakefile_nobonsai b/example/GNUmakefile_nobonsai deleted file mode 100644 index 888ec1b..0000000 --- a/example/GNUmakefile_nobonsai +++ /dev/null @@ -1,56 +0,0 @@ -# Makefile by Guillaume Pronost for LEAF example @ 2020/02/09 - -OSNAME = $(shell uname -s) - -include ../Makefile/Makefile.${OSNAME} - - -# set compiler options for ROOT -CXXFLAGS += $(shell root-config --cflags) -CXXFLAGS += '-fPIC' -std=c++11 -Wall -Wpedantic -Wno-long-long - -INCFLAGS = -I. -I$(shell root-config --incdir) -INCFLAGS += -I$(WCSIMDIR)/include -INCFLAGS += -I$(LEAFDIR)/leaf -INCFLAGS += -I$(LEAFDIR)/leaf/DataModel - -LIBS += -L${WCSIMDIR} -lWCSimRoot -LIBS += $(shell root-config --libs) -lMinuit -LIBS_LEAF += -L${LEAFDIR}/lib -lDataModelLite -lHKManager -lLEAF #-lHKAstroAnalysis - - -OBJECT = analysis - -all: $(OBJECT) - -analysis: analysis.o - @echo '<< compiling bin analysis >>' - @$(CXX) -g $(CXXFLAGS) -o $@ $^ $(LIBS) $(LIBS_LEAF) - - -# default rules -.cc.o: - @echo '<< compiling' $< '>>' - @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c $< - -%.o: %.cc %.hh - @echo '<< compiling' $< '>>' - @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c -o $@ $< - -%.o: %.cpp - @echo '<< compiling' $< '>>' - @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c -o $@ $< - - -%.o: %.C %.h - @echo '<< compiling' $< '>>' - @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c -o $@ $< - -# cleaner -clean: - @echo '<< cleaning >>' - @rm -f ./*.o - @rm -f ./lib*.so - @rm -f ./bin/* - @rm -f $(OBJECT) - diff --git a/example/analysis.cpp b/example/analysis.cpp index b2db7ef..bd43f8d 100644 --- a/example/analysis.cpp +++ b/example/analysis.cpp @@ -4,341 +4,203 @@ /** Date: Febuary 10th 2020 **/ /** Desc: Example application code for Benjamin's Low-E Fitter for Hyper-K **/ /*********************************************************************************/ -#include "analysis.h" +#include "analysis.hpp" +//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// ///* Some Utility Functions - -double calculateToWall(double R, double h, const std::vector& position, const std::vector& direction) -{ - double x = position[0], y = position[1], z = position[2]; - double dx = direction[0], dy = direction[1], dz = direction[2]; - - double A = dx * dx + dy * dy; - double B = 2 * (x * dx + y * dy); - double C = x * x + y * y - R * R; - - double discriminant = B * B - 4 * A * C; - - double t_cylinder = -1; - if (discriminant >= 0) { - double t1 = (-B - sqrt(discriminant)) / (2 * A); - double t2 = (-B + sqrt(discriminant)) / (2 * A); - - if (t1 > 0) t_cylinder = t1; - if (t2 > 0 && (t2 < t1 || t_cylinder < 0)) t_cylinder = t2; - } - - //Calculate the t values for intersection with top and bottom - double t_top = (h / 2 - z) / dz; - double t_bottom = (-h / 2 - z) / dz; - - //Check if the intersections are on the top or bottom - double x_top = x + t_top * dx; - double y_top = y + t_top * dy; - double x_bottom = x + t_bottom * dx; - double y_bottom = y + t_bottom * dy; - - if (x_top * x_top + y_top * y_top > R * R) t_top = -1; //Invalid if outside radius - if (x_bottom * x_bottom + y_bottom * y_bottom > R * R) t_bottom = -1; //Invalid if outside radius - - //Find the smallest positive t that corresponds to a valid intersection (side or top/bottom) - double t_wall = -1; - if (t_cylinder > 0) t_wall = t_cylinder; - if (t_top > 0 && (t_top < t_wall || t_wall < 0)) t_wall = t_top; - if (t_bottom > 0 && (t_bottom < t_wall || t_wall < 0)) t_wall = t_bottom; - - return t_wall; -} - -double calculateDWall(double R, double h, const std::vector& position) -{ - double x = position[0], y = position[1], z = position[2]; - double distanceToAxis = sqrt(x * x + y * y); - double dSide = fabs(distanceToAxis - R); - double dTop = fabs(z - h / 2); - double dBottom = fabs(z + h / 2); - return std::min({dSide, dTop, dBottom}); -} - -double angleBetween3DVectors(const std::vector& a, const std::vector& b) -{ - if(a.size() < 3 || b.size() < 3) - { - throw std::invalid_argument("Vectors must be of size at least 3."); - } - // double dotProduct = dot(a,b); - - // Magnitudes - double magA = std::sqrt(a[0]*a[0] + a[1]*a[1] + a[2]*a[2]); - double magB = std::sqrt(b[0]*b[0] + b[1]*b[1] + b[2]*b[2]); - - if (magA == 0 || magB == 0) { - throw std::invalid_argument("Zero-length vector."); +//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// + +arguments FetchInput(int argc, char** argv) { + int c = -1; + arguments arglist; + + //Input in c the argument (-f etc...) and in optarg the next argument + //When the above test becomes -1, it means it fails to find a new argument + std::cout << std::endl; + while( (c = getopt(argc, argv, "i:o:d:h:s:e:v")) != -1 ) + { + switch(c) + { + //Input file name + case 'i': + arglist.inputFile = optarg; + std::cout << "Input WCSim file: " << arglist.inputFile << std::endl; + break; + + //Output file name + case 'o': + arglist.outputFile = optarg; + if(arglist.outputFile == ""){ arglist.outputFile = "out.txt";} + std::cout << "Output root file: " << arglist.outputFile << std::endl; + break; + + //Darknoise + case 'd': + arglist.darkNoise = atof(optarg); + std::cout << "Dark noise frequency: " << arglist.darkNoise << " kHz" << std::endl; + break; + + //Darknoise hybrid + case 'h': + arglist.darkNoiseH = atof(optarg); + arglist.hybrid = true; + std::cout << "Dark noise frequency (hybrid geometry): " << arglist.darkNoiseH << " kHz" << std::endl; + break; + + //Starting event + case 's': + arglist.startEvent = atoi(optarg); + if(arglist.startEvent < 0){arglist.startEvent = 0;} + std::cout << "Starting event #" << arglist.startEvent << std::endl; + break; + + //Ending event + case 'e': + arglist.endEvent = atoi(optarg); + if(arglist.endEvent < 0){arglist.endEvent = 0;} + if(arglist.endEvent >= arglist.startEvent){std::cout << "Ending event #" << arglist.endEvent << std::endl;} + if(arglist.endEvent < arglist.startEvent){std::cout << "Ending event = last WCSim event" << std::endl;} + break; + + //Warning + case 'v': + arglist.verbose = true; + std::cout << "VERBOSE option on" << std::endl; + break; + } } + std::cout << std::endl; - // Cosine of the angle - double cosTheta = dot(a,b) / (magA * magB); - - // Clamp to [-1, 1] to avoid domain errors in acos - cosTheta = std::max(-1.0, std::min(1.0, cosTheta)); - - // Return angle in degrees - return std::acos(cosTheta) * 180.0 / M_PI; -} - -bool IsDarkRateHit(WCSimRootTrigger *fRootTrigger, WCSimRootCherenkovDigiHit *wcDigitHit) -{ - bool isDR = false; - std::vector rawhitphotonIDs = wcDigitHit->GetPhotonIds(); - if(rawhitphotonIDs.size() > 0) - { - double DR_Amount = 0; - for(unsigned int i = 0; i < rawhitphotonIDs.size(); i++) - { - TObject *RawHitTimess; - if (rawhitphotonIDs[i] >= 0 && rawhitphotonIDs[i] < (fRootTrigger->GetCherenkovHitTimes())->GetEntries()) - { - RawHitTimess = (fRootTrigger->GetCherenkovHitTimes())->At(rawhitphotonIDs[i]); - } - else - { - std::cerr << "Error: Invalid photon ID index: " << rawhitphotonIDs[i] << std::endl; - continue; - } - WCSimRootCherenkovHitTime *wcRawHitTimee = dynamic_cast(RawHitTimess); - // std::cout << "Photon type : " << wcRawHitTimee->GetPhotonCreatorProcessName() << std::endl; - if(wcRawHitTimee->GetPhotonCreatorProcessName() == "darkNoise") DR_Amount+=1.0; - } - DR_Amount = DR_Amount / rawhitphotonIDs.size(); - if(DR_Amount > 0.1) isDR = true; //? what threshold to consider a hit as darkRate ? - } - else - { - std::cout << "Error: No photon IDs found for this hit : " << wcDigitHit->GetTubeId() << std::endl; - } - return isDR; -} - -double GetResidualTime(const std::vector& origin, double originTime, WCSimRootCherenkovDigiHit *wcDigitHit, TimeDelta fTimeCorrection, double fLfTriggerTime) -{ - float fCVacuum = 3e8 * 1e2 / 1e9; // speed of light, in centimeter per ns. - float fNIndex = 1.373; // 1.385;//1.373;//refraction index of water - double fLightSpeed = fCVacuum / fNIndex; - double HitT = wcDigitHit->GetT() + fLfTriggerTime; - WCSimRootPMT pmt = fLeafGeometry->GetPMT(wcDigitHit->GetTubeId() - 1, false); - double PMTpos[3]; - for (int j = 0; j < 3; j++) PMTpos[j] = pmt.GetPosition(j); - double distance = Astro_GetDistance(PMTpos, origin); - double tof = distance / fLightSpeed; - double hitTime = (fTimeCorrection + HitT) / TimeDelta::ns; - return hitTime - tof - originTime; -} - -double GetDistanceToNeighbors(WCSimRootTrigger *fRootTrigger, WCSimRootCherenkovDigiHit *wcDigitHit, int N_Neighbors) -{ - std::vector distances; - - WCSimRootPMT pmt; - pmt = fLeafGeometry->GetPMT(wcDigitHit->GetTubeId() - 1, false); - - std::vector PMTpos(3); - for (int j = 0; j < 3; j++) PMTpos[j] = pmt.GetPosition(j); - - //loop over each hit, check if it is not the same as the one we are looking at - for (int i = 0; i < fRootTrigger->GetNcherenkovdigihits(); i++) - { - WCSimRootCherenkovDigiHit *neighborHit = dynamic_cast((fRootTrigger->GetCherenkovDigiHits())->At(i)); - - if(wcDigitHit->GetTubeId() == neighborHit->GetTubeId()) - continue; - - WCSimRootPMT neighborPMT; - neighborPMT = fLeafGeometry->GetPMT(neighborHit->GetTubeId() - 1, false); - - std::vector neighborPMTpos(3); - for (int j = 0; j < 3; j++) neighborPMTpos[j] = neighborPMT.GetPosition(j); - - double distance = calculateDistance(PMTpos, neighborPMTpos); - distances.push_back(distance); - } - - //* Sort the distances, keep the N_Neighbors closest ones - std::sort(distances.begin(), distances.end()); - // size = N_Neighbors; - if (distances.size() < (long unsigned int)N_Neighbors ) - { - std::cerr << "Not enough neighbors found!" << std::endl; - return -1; - } - double sum = 0; - for (int i = 0; i < N_Neighbors; i++) sum += distances[i]; - return sum / N_Neighbors; + return arglist; } -///* MAIN EXECUTION - -int main(int argc, char **argv) -{ - std::string sInputFile = ""; - std::string sOutputFile = ""; - int iNeededArgc = 3; - double dDarkNoise = 0.; // kHz - double dDarkNoiseHybrid = 0.; // kHz +//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// +/* MAIN EXECUTION */ +//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// - iNeededArgc += 1; +int main(int argc, char **argv) { + //Get arguments + arguments arglist = FetchInput(argc, argv); + std::string sInputFile = arglist.inputFile; + std::string sOutputFile = arglist.outputFile; + double dDarkNoise = arglist.darkNoise; + double dDarkNoiseHybrid = 0.; #ifdef mPMT - iNeededArgc += 1; + dDarkNoiseHybrid = arglist.darkNoiseH; #endif - - if (argc == iNeededArgc) - { - int iArg = 1; - sInputFile = argv[iArg]; - iArg += 1; - sOutputFile = argv[iArg]; - iArg += 1; - dDarkNoise = atof(argv[iArg]); - iArg += 1; -#ifdef mPMT - dDarkNoiseHybrid = atof(argv[iArg]); - iArg += 1; -#endif - } - else - { - - std::cout << "Synthax: " << argv[0] << " input output"; - std::cout << " DN_in_kHz_B&L"; - -#ifdef mPMT - std::cout << " DN_in_kHz_mPMT"; -#endif - std::cout << std::endl; - return 0; - } + int firstEvents = arglist.startEvent; //We start counting at 0 + int lastEvents = arglist.endEvent; + bool verbose = arglist.verbose; // Read WCSim output TFile *fInputFile = new TFile(sInputFile.c_str(), "READ"); // Get TTrees - TTree *fInputTree = (TTree *)fInputFile->Get("wcsimT"); + TTree *fInputTree = (TTree *)fInputFile->Get("wcsimT"); TTree *fInputGeoTree = (TTree *)fInputFile->Get("wcsimGeoT"); fLeafGeometry = 0; fInputGeoTree->SetBranchAddress("wcsimrootgeom", &fLeafGeometry); + // Set Branch for PMTs WCSimRootEvent *fIDevent = new WCSimRootEvent(); - // Set Branch fInputTree->SetBranchAddress("wcsimrootevent", &fIDevent); - // Set autodelete to avoid memory leak - fInputTree->GetBranch("wcsimrootevent")->SetAutoDelete(kTRUE); + fInputTree->GetBranch("wcsimrootevent")->SetAutoDelete(kTRUE); // Set autodelete to avoid memory leak - // WCSimRootEvent *fHybridevent = new WCSimRootEvent(); + // Set Branch for mPMTs #ifdef mPMT - // Set Branche - // fInputTree->SetBranchAddress("wcsimrootevent2", &fHybridevent); - // Set autodelete to avoid memory leak - fInputTree->GetBranch("wcsimrootevent2")->SetAutoDelete(kTRUE); + WCSimRootEvent *fHybridevent = new WCSimRootEvent(); + fInputTree->SetBranchAddress("wcsimrootevent2", &fHybridevent); + fInputTree->GetBranch("wcsimrootevent2")->SetAutoDelete(kTRUE); // Set autodelete to avoid memory leak #endif + // Set Branch for OD PMTs #ifdef OD_ON WCSimRootEvent *fODevent = new WCSimRootEvent(); - // Set Branche fInputTree->SetBranchAddress("wcsimrootevent_OD", &fODevent); - // Set autodelete to avoid memory leak - fInputTree->GetBranch("wcsimrootevent_OD")->SetAutoDelete(kTRUE); + fInputTree->GetBranch("wcsimrootevent_OD")->SetAutoDelete(kTRUE); // Set autodelete to avoid memory leak #endif - // Read Geo - fInputGeoTree->GetEntry(0); - // Create Output TTree TFile *fOutputFile = new TFile(sOutputFile.c_str(), "RECREATE"); fOutputFile->SetCompressionLevel(2); - TTree *fGeoTree = new TTree("wcsimGeoT", "Geometry TTree"); - TTree *fPrimaryTree = new TTree("Reduced", "Reduced TTree"); - - // Set Branches - SetGeoBranch(fGeoTree); - - FitterOutput leaf_output; - SetCustomBranch(fPrimaryTree, leaf_output); + // Read Geo and set branch + fInputGeoTree->GetEntry(0); + TTree *fGeoTree = new TTree("wcsimGeoT", "Geometry TTree"); + TTree *fPrimaryTree = new TTree("Reduced", "Reduced TTree"); + fGeoTree->Branch("wcsimrootgeom", fLeafGeometry); + SetCustomBranch(fPrimaryTree); fGeoTree->Fill(); // Get PMT Number: int nPMT_ID = fLeafGeometry->GetWCNumPMT(); + std::cout << " ID " << nPMT_ID << std::endl; #ifdef OD_ON int nPMT_OD = fLeafGeometry->GetODWCNumPMT(); + std::cout << " OD " << nPMT_OD << std::endl; #endif int nMultPMT = fLeafGeometry->GetWCNumPMT(true); - - std::cout << " ID " << nPMT_ID << std::endl; std::cout << " mPMT " << nMultPMT << std::endl; + // Initialize HK Manager geometry HKManager::GetME()->SetGeometry(fLeafGeometry, dDarkNoise * 1e3, dDarkNoiseHybrid * 1e3); // Initialize LEAF - LEAF::GetME()->Initialize(HKManager::GetME()->GetGeometry()); // Loads the geometry in leaf and loads the pdfs + LEAF::GetME()->Initialize( HKManager::GetME()->GetGeometry(), + HKManager::GetME()->GetDarkNoise(), + HKManager::GetME()->GetGeometryPMT_ID(), + HKManager::GetME()->GetGeometryPMT_mPMT()); // Loads the geometry in leaf and loads the pdfs //* for each config variable, check if it has been set, take the default variable if not - int maxEvents = -1; // in cm, the step size for coarse grid search - const char* envValue = std::getenv("nbOfEvents"); - if (envValue != nullptr) - { - maxEvents = std::stoi(envValue); - std::cout << "max events is set to " << maxEvents << std::endl; - } else std::cout << "Environment variable max events is not set. Using all events" << std::endl; - - envValue = std::getenv("maxHitsAngle"); - if (envValue != nullptr) - { + const char* envValue = std::getenv("maxHitsAngle"); + if (envValue != nullptr) { maxHitAngle = std::stof(envValue); std::cout << "max hit angle : " << maxHitAngle << "°" << std::endl; - } else std::cout << "Environment variable max hit angle is not set. Using default value : " << maxHitAngle << std::endl; + } + else std::cout << "Environment variable max hit angle is not set. Using default value : " << maxHitAngle << std::endl; envValue = std::getenv("N_Neighbors"); - if (envValue != nullptr) - { + if (envValue != nullptr) { N_Neighbors = std::stof(envValue); std::cout << "N_Neighbors : " << N_Neighbors << std::endl; - } else std::cout << "Environment variable N_Neighbors is not set. Using default value : " << N_Neighbors << std::endl; + } + else std::cout << "Environment variable N_Neighbors is not set. Using default value : " << N_Neighbors << std::endl; envValue = std::getenv("maxDistanceToNeighbors"); - if (envValue != nullptr) - { + if (envValue != nullptr) { maxDistanceToNeighbors = std::stof(envValue); std::cout << "maxDistanceToNeighbors : " << maxDistanceToNeighbors << "cm" << std::endl; - } else std::cout << "Environment variable maxDistanceToNeighbors is not set. Using default value : " << maxDistanceToNeighbors << std::endl; + } + else std::cout << "Environment variable maxDistanceToNeighbors is not set. Using default value : " << maxDistanceToNeighbors << std::endl; // Read Input Tree - int nPrimaryEvents = fInputTree->GetEntries(); - - if(maxEvents > 0) nPrimaryEvents = std::min(nPrimaryEvents, maxEvents); - int iWrite = 0; int failAmount = 0; + int nPrimaryEvents = fInputTree->GetEntries(); + if(lastEvents > 0) nPrimaryEvents = std::min(nPrimaryEvents, lastEvents); TStopwatch timer; timer.Reset(); timer.Start(); // Loop on Primary events - for (int i = 0; i < nPrimaryEvents; i++) - { - // Reset Hit vector + for (int i=firstEvents ; i 10 ) break; + // Reset Hit vector and clock HKManager::GetME()->ResetHitInfo(); // HKManager::GetME()->ResetSecondaryHitInfo(); - - // if ( i%1000==0 ) { timer.Stop(); - std::cout << "Event # = " << i << " / " << nPrimaryEvents << " ( " << timer.RealTime() << " )\n"; timer.Reset(); - timer.Start(); - //} + if(verbose) { + std::cout << "\n===========================================================================================================================================================" << std::endl; + std::cout << "Event # = " << i+1 << " / " << nPrimaryEvents << std::endl; + } + + timer.Start(); fInputTree->GetEntry(i); // Initialize output variables @@ -348,37 +210,25 @@ int main(int argc, char **argv) rawhit_num_noDN = 0; true_particleId.clear(); - true_energy.clear(); - true_origin_X.clear(); - true_origin_Y.clear(); - true_origin_Z.clear(); - true_origin_T.clear(); + true_energies.clear(); + true_vertexes.clear(); + leaf_Vertex.clear(); leaf_Dir.clear(); leaf_MyDir.clear(); leaf_QuickDir.clear(); - + digithit_pmtId.clear(); digithit_T.clear(); correctedDigithit_T.clear(); digithit_Q.clear(); - digithit_Angle.clear(); - digithit_NormAngle.clear(); - digithit_NeighborsDist.clear(); - hit_is_DR.clear(); Charge_PMT.clear(); - relativeAngle.clear(); - lf_relativeAngle.clear(); - hit_residual.clear(); rawhit_num = 0; digithit_num = 0; fLFTime = -9999; - hit_5_15ns = 0; - hit_50ns = 0; - Hit_ID = 0; Hit_ID_50 = 0; Hit_ID_200 = 0; @@ -397,14 +247,6 @@ int main(int argc, char **argv) fLastRawHit = 0; - trueVertex = std::vector(4); - trueDir = std::vector(3); - - lf_spatial_res = 0.; - lf_Dir_res = 0.; - lf_time_res = 0.; - lf_energy_res = 0.; - bestTrigger = 0; fLfTriggerTime = 0.; @@ -412,18 +254,17 @@ int main(int argc, char **argv) /* ID events */ /****************************************************************************************/ - fHit = 0; - fHit_20 = 0; - fHit_50 = 0; + fHit = 0; + fHit_20 = 0; + fHit_50 = 0; fHit_200 = 0; fHit_400 = 0; // this is where we send all the informations of the event to leaf // we return false if we don't want to process this event based on defined filters - bool shouldProcess = AnalyseEvent(fIDevent, ID_EVENT); + bool shouldProcess = AnalyseEvent(fIDevent, PMTType::kID); - if(!shouldProcess) - { + if(!shouldProcess) { failAmount++; std::cout << "Event # = " << i << " cannot be processed " << std::endl; continue; @@ -436,26 +277,27 @@ int main(int argc, char **argv) Hit_ID_400 = fHit_400; /****************************************************************************************/ - /* mPMT events */ + /* mPMT events */ /****************************************************************************************/ +#ifdef mPMT fHit = 0; fHit_20 = 0; fHit_50 = 0; fHit_200 = 0; fHit_400 = 0; -#ifdef mPMT - // /*bool bmPMT =*/AnalyseEvent(fHybridevent, mPMT_EVENT); -#endif + // /*bool bmPMT =*/AnalyseEvent(fHybridevent, PMTType::kmPMT); + Hit_mPMT = fHit; Hit_mPMT_20 = fHit_20; Hit_mPMT_50 = fHit_50; Hit_mPMT_200 = fHit_200; Hit_mPMT_400 = fHit_400; +#endif /****************************************************************************************/ - /* OD events */ + /* OD events */ /****************************************************************************************/ #ifdef OD_ON @@ -467,53 +309,73 @@ int main(int argc, char **argv) // There should be a dedicated OD analyser, as many thing should be different than for ID // Doesn't exist yet - /*bool bOD =*/AnalyseODEvent(fODevent, OD_EVENT); + /*bool bOD =*/ AnalyseODEvent(fODevent, OD_EVENT); Hit_OD = fHit; Hit_OD_50 = fHit_50; Hit_OD_200 = fHit_200; Hit_OD_400 = fHit_400; #endif + /****************************************************************************************/ - /* Benjamin Fitter */ + /* Benjamin Quilain Fitter */ /****************************************************************************************/ TStopwatch timerLF; timerLF.Reset(); timerLF.Start(); - // To be replaced by meaningful TriggerTime - TimeDelta fDummyTrigger(0.); + LEAF::GetME()->LoadHitsCollection(HKManager::GetME()->GetHitCollection_ID(),HKManager::GetME()->GetHitCollection_mPMT()); + fLeafOutput = LEAF::GetME()->MakeSequentialFit(); + //std::cout << "(X, Y, Z, T): (" << fLeafOutput.vtx.X() << ", " << fLeafOutput.vtx.Y() << ", " << fLeafOutput.vtx.Z() << ", " << fLeafOutput.vtx.T() << ") NLL: " << fLeafOutput.vtx_nll << std::endl; + //std::cout << "(dX, dY, dZ, T): (" << fLeafOutput.dir.X() << ", " << fLeafOutput.dir.Y() << ", " << fLeafOutput.dir.Z() << ") NLL: " << fLeafOutput.dir_nll << std::endl; - leaf_output = LEAF::GetME()->MakeSequentialFit(HKManager::GetME()->GetHitCollection(), fDummyTrigger); // the fit method can be replaced by a joint fit (doesn't work better for now) - // leaf_output = LEAF::GetME()->MakeJointFit(HKManager::GetME()->GetHitCollection(), fDummyTrigger); - fOutputProps = LEAF::fOutputProps; - + // fLeafOutput = LEAF::GetME()->MakeJointFit(HKManager::GetME()->GetHitCollection(), fDummyTrigger); + fLeafPerformances = LEAF::GetME()->GetPerformances(); + timerLF.Stop(); fLFTime = timerLF.RealTime(); - std::cout << " LEAF took: " << timerLF.RealTime() << " for " << HKManager::GetME()->GetHitCollection()->Size() << " Hits"; - std::cout << " (Vtx Search : " << fOutputProps.Vtx_Search_ComputeTime << " , Vtx Minimize : " << fOutputProps.Vtx_Minimize_ComputeTime << ", Dir Search : " << fOutputProps.Dir_Search_ComputeTime << " , Dir Minimize : " << fOutputProps.Dir_Minimize_ComputeTime << " , Energy Fit : " << fOutputProps.Energy_Fit_ComputeTime << ")" << std::endl; - bool validEvent = PostLeafAnalysis(fIDevent,ID_EVENT, leaf_output); + if (verbose) { + std::cout << " LEAF took: " << timerLF.RealTime() << " sec for " << HKManager::GetME()->GetHitCollection_ID()->size() + HKManager::GetME()->GetHitCollection_mPMT()->size() << " Hits"; + std::cout << " (Vtx Search : " << fLeafPerformances.vtx_search_ct + << " , Vtx Minimize : " << fLeafPerformances.vtx_minimize_ct + << ", Dir Search : " << fLeafPerformances.dir_search_ct + << " , Dir Minimize : " << fLeafPerformances.dir_minimize_ct + << " , Energy Fit : " << fLeafPerformances.energy_fit_ct << ")" << std::endl; + std::cout << " True vertex: (" << true_vertex.X() + << ", " << true_vertex.Y() + << ", " << true_vertex.Z() + << ", " << true_vertex.T()<< ") [cm/cm/cm/ns] " + << " // direction: (" << true_dir.X() + << ", " << true_dir.Y() + << ", " << true_dir.Z() + << " // energy: " << true_energy << std::endl; + std::cout << " LEAF vertex: (" << fLeafOutput.vtx.X() + << ", " << fLeafOutput.vtx.Y() + << ", " << fLeafOutput.vtx.Z() + << ", " << fLeafOutput.vtx.T()<< ") [cm/cm/cm/ns]" + << " // direction: (" << fLeafOutput.dir.X() + << ", " << fLeafOutput.dir.Y() + << ", " << fLeafOutput.dir.Z() + << " // energy: " << fLeafOutput.energy << std::endl; + } /****************************************************************************************/ /* Fill output tree */ /****************************************************************************************/ - if(validEvent) - { - fPrimaryTree->Fill(); - iWrite += 1; - } + + fPrimaryTree->Fill(); + iWrite += 1; } - std::cout << "Event # = " << nPrimaryEvents << " / " << nPrimaryEvents << std::endl; - - std::cout << ((nPrimaryEvents - failAmount) / nPrimaryEvents) * 100 << "% of the events were processed" << std::endl; + std::cout << "\n==========================================================================================================================================================="; + std::cout << "\n===========================================================================================================================================================\n" << std::endl; + std::cout << ((nPrimaryEvents - failAmount) / nPrimaryEvents) * 100 << "% of the events have been processed" << std::endl; fOutputFile->cd(); - fOutputFile->Write("", TObject::kOverwrite); delete fPrimaryTree; @@ -521,8 +383,9 @@ int main(int argc, char **argv) return 1; } + //* All the variables that will be kept in the output Tree -void SetCustomBranch(TTree *fPrimaryTree, FitterOutput leaf_output) +void SetCustomBranch(TTree *fPrimaryTree) { fPrimaryTree->Branch("eventId", &eventId, "eventId/I"); fPrimaryTree->Branch("nTrigger", &nTrigger, "nTrigger/I"); @@ -531,14 +394,9 @@ void SetCustomBranch(TTree *fPrimaryTree, FitterOutput leaf_output) fPrimaryTree->Branch("digithit_num", &digithit_num, "digithit_num/I"); fPrimaryTree->Branch("true_particleId", &true_particleId); - fPrimaryTree->Branch("true_origin_X", &true_origin_X); - fPrimaryTree->Branch("true_origin_Y", &true_origin_Y); - fPrimaryTree->Branch("true_origin_Z", &true_origin_Z); - fPrimaryTree->Branch("true_origin_T", &true_origin_T); - fPrimaryTree->Branch("true_energy", &true_energy); - fPrimaryTree->Branch("true_dir", &trueDir, "true_dir[3]/D"); - fPrimaryTree->Branch("DWall", &dWall, "DWall/D"); - fPrimaryTree->Branch("ToWall", &toWall, "lf_wall/D"); + fPrimaryTree->Branch("true_vertexes", &true_vertexes); + fPrimaryTree->Branch("true_energies", &true_energies); + fPrimaryTree->Branch("true_dir", &true_dir); fPrimaryTree->Branch("ID_hits", &Hit_ID, "Hit_ID/I"); fPrimaryTree->Branch("ID_hits_50", &Hit_ID_50, "Hit_ID_50/I"); @@ -551,60 +409,33 @@ void SetCustomBranch(TTree *fPrimaryTree, FitterOutput leaf_output) fPrimaryTree->Branch("mPMT_hits_200", &Hit_mPMT_200, "Hit_mPMT_200/I"); fPrimaryTree->Branch("mPMT_hits_400", &Hit_mPMT_400, "Hit_mPMT_400/I"); - fPrimaryTree->Branch("hit_5_15ns", &hit_5_15ns, "hit_5_15ns/I"); - fPrimaryTree->Branch("hit_50ns", &hit_50ns, "hit_50ns/I"); - - fPrimaryTree->Branch("lf_vertex", &leaf_Vertex); - fPrimaryTree->Branch("lf_NLL", &leaf_output.NLL, "lf_NLL/D"); - fPrimaryTree->Branch("lf_good", &leaf_output.NLLR, "lf_good/D"); - fPrimaryTree->Branch("lf_ctime", &fLFTime, "lf_ctime/D"); // Computation time - fPrimaryTree->Branch("lf_energy", &leaf_output.Energy, "lf_energy/D"); - fPrimaryTree->Branch("lf_Dir", &leaf_Dir); - fPrimaryTree->Branch("lf_MyDir", &leaf_MyDir); - fPrimaryTree->Branch("lf_Quick_Dir", &leaf_QuickDir); - fPrimaryTree->Branch("lf_Dir_NLL", &leaf_output.DNLL, "lf_Dir_NLL/D"); - fPrimaryTree->Branch("lf_DWall", &lf_dWall, "DWall/D"); - fPrimaryTree->Branch("lf_ToWall", &lf_ToWall, "lf_ToWall/D"); - - fPrimaryTree->Branch("lf_spatial_res", &lf_spatial_res, "lf_spatial_res/D"); - fPrimaryTree->Branch("lf_time_res", &lf_time_res, "lf_time_res/D"); - fPrimaryTree->Branch("lf_Dir_res", &lf_Dir_res,"lf_Dir_res/D"); - fPrimaryTree->Branch("lf_MyDir_res", &lf_MyDir_res,"lf_MyDir_res/D"); - fPrimaryTree->Branch("lf_Quick_Dir_res", &lf_Quick_Dir_res,"lf_Quick_Dir_res/D"); - fPrimaryTree->Branch("lf_energy_res", &lf_energy_res, "lf_energy_res/D"); - - fPrimaryTree->Branch("Leaf_ComputeTime", &fOutputProps.Leaf_ComputeTime, "Leaf_ComputeTime/D"); - fPrimaryTree->Branch("Vtx_Search_ComputeTime", &fOutputProps.Vtx_Search_ComputeTime, "Vtx_Search_ComputeTime/D"); - fPrimaryTree->Branch("Vtx_Minimize_ComputeTime", &fOutputProps.Vtx_Minimize_ComputeTime, "Vtx_Minimize_ComputeTime/D"); - fPrimaryTree->Branch("Dir_Quick_Search_ComputeTime", &fOutputProps.Dir_Quick_Search_ComputeTime, "Dir_Quick_Search_ComputeTime/D"); - fPrimaryTree->Branch("Dir_Search_ComputeTime", &fOutputProps.Dir_Search_ComputeTime, "Dir_Search_ComputeTime/D"); - fPrimaryTree->Branch("Dir_Minimize_ComputeTime", &fOutputProps.Dir_Minimize_ComputeTime, "Dir_Minimize_ComputeTime/D"); - fPrimaryTree->Branch("Energy_Fit_ComputeTime", &fOutputProps.Energy_Fit_ComputeTime, "Energy_Fit_ComputeTime/D"); - + fPrimaryTree->Branch("lf_vtx", &fLeafOutput.vtx); + fPrimaryTree->Branch("lf_vtx_nll", &fLeafOutput.vtx_nll); + fPrimaryTree->Branch("lf_dir", &fLeafOutput.dir); + fPrimaryTree->Branch("lf_dir_nll", &fLeafOutput.dir_nll); + fPrimaryTree->Branch("lf_energy", &fLeafOutput.energy); + fPrimaryTree->Branch("lf_nll", &fLeafOutput.nll_r); + fPrimaryTree->Branch("lf_ct", &fLeafPerformances.leaf_ct); + fPrimaryTree->Branch("lf_vtx_search_ct", &fLeafPerformances.vtx_search_ct); + fPrimaryTree->Branch("lf_vtx_minization_ct", &fLeafPerformances.vtx_minimize_ct); + fPrimaryTree->Branch("lf_dir_search_ct", &fLeafPerformances.dir_search_ct); + fPrimaryTree->Branch("lf_dir_minization_ct", &fLeafPerformances.dir_minimize_ct); + fPrimaryTree->Branch("rawTriggerTime", &rawTriggerTime, "rawTriggerTime/D"); - fPrimaryTree->Branch("TotalCharge", &leaf_output.TotalCharge, "TotalCharge/D"); - fPrimaryTree->Branch("RelativeAngle", &relativeAngle); - fPrimaryTree->Branch("lf_RelativeAngle", &lf_relativeAngle); + fPrimaryTree->Branch("TotalCharge", &fLeafOutput.total_charge, "TotalCharge/D"); fPrimaryTree->Branch("DigiHitT", &digithit_T); fPrimaryTree->Branch("CorrectedDigiHitT", &correctedDigithit_T); fPrimaryTree->Branch("HitIsDR", &hit_is_DR); fPrimaryTree->Branch("DigiHitQ", &digithit_Q); - fPrimaryTree->Branch("DigiHitAngle", &digithit_Angle); - fPrimaryTree->Branch("DigiHitNormAngle", &digithit_NormAngle); - fPrimaryTree->Branch("DigiHitNeighborsDist", &digithit_NeighborsDist); fPrimaryTree->Branch("DigiHitResidual", &hit_residual); fPrimaryTree->Branch("Charge_PMT", &Charge_PMT); } -void SetGeoBranch(TTree *fGeoTree) -{ - fGeoTree->Branch("wcsimrootgeom", fLeafGeometry); -} -bool AnalyseEvent(WCSimRootEvent *tEvent, int iEventType) -{ - int iHybrid = 0; +bool AnalyseEvent(WCSimRootEvent *tEvent, PMTType pmtType) { + const int iPMTType = static_cast(pmtType); + int nVertex = 0; // Number of Vertex in event int nTrack = 0; // Number of Track in event int nRawCherenkovHits = 0; // Number of Raw Cherenkov hits @@ -615,19 +446,14 @@ bool AnalyseEvent(WCSimRootEvent *tEvent, int iEventType) rawTriggerTime = 0.; WCSimRootTrigger *fRootTrigger; - TimeDelta fDummyTrigger(0.); - TimeDelta fTimeCorrection = HKManager::GetME()->GetHitCollection()->timestamp - fDummyTrigger; //*still have no idea what this is - //* find best trigger (best number of hits) int bestNumbOfHits = 0; bestTrigger = -1; - for (int iTrig = 0; iTrig < nTrigger; iTrig++) - { + for (int iTrig = 0; iTrig < nTrigger; iTrig++) { fRootTrigger = tEvent->GetTrigger(iTrig); nDigitizedCherenkovHits = fRootTrigger->GetNcherenkovdigihits(); - if(nDigitizedCherenkovHits > bestNumbOfHits && fRootTrigger->GetTriggerInfo().size() >= 3) - { + if(nDigitizedCherenkovHits > bestNumbOfHits && fRootTrigger->GetTriggerInfo().size() >= 3) { bestNumbOfHits = nDigitizedCherenkovHits; bestTrigger = iTrig; } @@ -639,8 +465,7 @@ bool AnalyseEvent(WCSimRootEvent *tEvent, int iEventType) //* Retreive true infos (onlyh set by wcsim in the first trigger) - for (int iTrig = 0; iTrig < tEvent->GetNumberOfEvents(); iTrig++) - { + for (int iTrig = 0; iTrig < tEvent->GetNumberOfEvents(); iTrig++) { fRootTrigger = tEvent->GetTrigger(iTrig); #ifdef OLD_WCSIM @@ -655,26 +480,18 @@ bool AnalyseEvent(WCSimRootEvent *tEvent, int iEventType) nDigitizedCherenkovHits = fRootTrigger->GetNcherenkovdigihits(); //* Fist trigger is the one used to compute true Vertex & Dir - if (iTrig == 0) - { + if (iTrig == 0) { // Loop on vertex - for (int iVertex = 0; iVertex < nVertex; iVertex++) - { - + for (int iVertex = 0; iVertex < nVertex; iVertex++) { #ifdef OLD_WCSIM - for(int j=0; j<3; j++) trueVertex[j]=fRootTrigger->GetVtx(j); - true_origin_X.push_back(fRootTrigger->GetVtx(0)); - true_origin_Y.push_back(fRootTrigger->GetVtx(1)); - true_origin_Z.push_back(fRootTrigger->GetVtx(2)); + true_vertex = ROOT::Math::XYZTVector(fRootTrigger->GetVtx(0),fRootTrigger->GetVtx(1),fRootTrigger->GetVtx(2),fRootTrigger->GetVtx(3)); + true_vertexes.push_back(true_vertex); #else - for(int j=0; j<3; j++) trueVertex[j]=fRootTrigger->GetVtxs(iVertex,j); - true_origin_X.push_back(fRootTrigger->GetVtxs(iVertex, 0)); - true_origin_Y.push_back(fRootTrigger->GetVtxs(iVertex, 1)); - true_origin_Z.push_back(fRootTrigger->GetVtxs(iVertex, 2)); + true_vertex = ROOT::Math::XYZTVector(fRootTrigger->GetVtxs(iVertex,0),fRootTrigger->GetVtxs(iVertex,1),fRootTrigger->GetVtxs(iVertex,2),fRootTrigger->GetVtxs(iVertex,3)); + true_vertexes.push_back(true_vertex); #endif - if (iVertex * 2 > nTrack) - { + if (iVertex * 2 > nTrack) { std::cout << "ERROR: Vertex and Track number incompatible (nVertex: " << nVertex << " ; nTrack " << nTrack << ") " << std::endl; continue; } @@ -683,27 +500,19 @@ bool AnalyseEvent(WCSimRootEvent *tEvent, int iEventType) TObject *Track = (fRootTrigger->GetTracks())->At(iVertex * 2); WCSimRootTrack *wcTrack = dynamic_cast(Track); - if (wcTrack->GetParentId() == 0 && wcTrack->GetIpnu()==11) - { - for(int j=0; j<3; j++) trueDir[j] = wcTrack->GetPdir(j); - Normalize(trueDir); + if (wcTrack->GetParentId() == 0 && wcTrack->GetIpnu()==11) { + true_dir = ROOT::Math::XYZVector(wcTrack->GetPdir(0),wcTrack->GetPdir(1),wcTrack->GetPdir(2)).Unit(); } - - toWall = calculateToWall(R, h, trueVertex, trueDir); - dWall = calculateDWall(R, h, trueVertex); - + + true_energy = wcTrack->GetE(); true_particleId.push_back(wcTrack->GetIpnu()); - true_energy.push_back(wcTrack->GetE()); - true_origin_T.push_back(fRootTrigger->GetVtx(3)); + true_energies.push_back(wcTrack->GetE()); } // Send True Position to LEAF to make checks of success in the steps of leaf (should be removed for true data) - std::vector vVtxTrue(3, 0.); - vVtxTrue[0] = true_origin_X[0]; - vVtxTrue[1] = true_origin_Y[0]; - vVtxTrue[2] = true_origin_Z[0]; - SetTrueVertexInfo(vVtxTrue, true_origin_T[0]); - SetTrueDirInfo(trueDir); + + LEAF::GetME()->SetTrueVertexInfo(true_vertexes[0]); + LEAF::GetME()->SetTrueDirInfo(true_dir); rawhit_num = nRawCherenkovHits; } @@ -715,120 +524,66 @@ bool AnalyseEvent(WCSimRootEvent *tEvent, int iEventType) rawTriggerTime = triggerInfo[2]; startingCherenkovHitID = digithit_pmtId.size(); - std::vector times; - - //* check if trigger is too soon or too late compared to the event (trigger happened because of dark rate) - double hitTimesStack = 0; - for (int iDigitHit = 0; iDigitHit < nDigitizedCherenkovHits; iDigitHit++) - { - TObject *Hit = (fRootTrigger->GetCherenkovDigiHits())->At(iDigitHit); - WCSimRootCherenkovDigiHit *wcDigitHit = dynamic_cast(Hit); - double HitT = wcDigitHit->GetT() + fLfTriggerTime; - hitTimesStack+=HitT; - } - double hitTmean = hitTimesStack/nDigitizedCherenkovHits; - if(hitTmean > 2000.0 || hitTmean < -2000.0) return false; //* set these values to define the filter (currently filtering nothing) - - double AllQ = 0; // Loop on Digitized Hits - for (int iDigitHit = 0; iDigitHit < nDigitizedCherenkovHits; iDigitHit++) - { - //* Get and Compute Hit Informations - + double AllQ = 0; + for (int iDigitHit = 0; iDigitHit < nDigitizedCherenkovHits; iDigitHit++) { + //* Get Hit Informations TObject *Hit = (fRootTrigger->GetCherenkovDigiHits())->At(iDigitHit); WCSimRootCherenkovDigiHit *wcDigitHit = dynamic_cast(Hit); int pmtId = wcDigitHit->GetTubeId(); WCSimRootPMT pmt = fLeafGeometry->GetPMT(pmtId - 1, false); - double HitT = wcDigitHit->GetT() + fLfTriggerTime; - double HitQ = wcDigitHit->GetQ(); + double HitT = wcDigitHit->GetT() + fLfTriggerTime; + double HitQ = wcDigitHit->GetQ(); int peForTube = wcDigitHit->GetQ(); - std::vector lfHitDir = std::vector(3); - std::vector PMTOrientation = std::vector(3); - std::vector vDir = std::vector(3); - bool isDR = IsDarkRateHit(fRootTrigger, wcDigitHit); - double hitResidual = GetResidualTime(trueVertex, true_origin_T[0], wcDigitHit, fTimeCorrection, fLfTriggerTime); - for (int j = 0; j < 3; j++) - { - PMTOrientation[j] = pmt.GetOrientation(j); - vDir[j] = pmt.GetPosition(j) - trueVertex[j]; - lfHitDir[j] = pmt.GetPosition(j); - } - - Normalize(vDir); - Normalize(lfHitDir); - Normalize(PMTOrientation); - - std::vector neg_vDir = { -vDir[0], -vDir[1], -vDir[2] }; - double NormAngle = dot(neg_vDir, PMTOrientation); - - double hitRelativeAngle = dot(vDir, trueDir); - - //* Use that inforation to filter some hits - //* we aim at killing dark rate hits - //* ideally without using any information form the true event - - //* Direction Filter - double angle = angleBetween3DVectors(vDir, trueDir); - // if(angle > maxHitAngle) continue; // works best at 90° - - //* Filter base on distance to hit neighbors - double distanceToNeighbor = GetDistanceToNeighbors(fRootTrigger, wcDigitHit, N_Neighbors); - // if(distanceToNeighbor > maxDistanceToNeighbors) continue; - - //*DR FILTER - if(isDR) continue; // ideal filter to check reconstruction without dark rate - ///* fill infos to output root file + //* Fill output root file AllQ += HitQ; digithit_pmtId.push_back(pmtId); digithit_T.push_back(HitT); correctedDigithit_T.push_back(HitT - fLfTriggerTime - triggerInfo[1]); digithit_Q.push_back(HitQ); - hit_is_DR.push_back(isDR); - digithit_Angle.push_back(angle); - digithit_NormAngle.push_back((TMath::ACos(NormAngle))*180./TMath::Pi()); - digithit_NeighborsDist.push_back(distanceToNeighbor); Charge_PMT.push_back(peForTube); - relativeAngle.push_back((TMath::ACos(hitRelativeAngle))*180./TMath::Pi()); - digithit_Type.push_back(iEventType); - hit_residual.push_back(hitResidual); + digithit_Type.push_back(iPMTType); } - for(long unsigned int j=0;j -5) hit_5_15ns++; - if(hit_residual[j] < 50 && hit_residual[j] > -50) hit_50ns++; - } int iIdx_BS = 0; nDigitizedCherenkovHits = digithit_pmtId.size(); digithit_num = nDigitizedCherenkovHits - startingCherenkovHitID; + std::vector times; // Feed fitter - for (int iDigitHit = startingCherenkovHitID; iDigitHit < nDigitizedCherenkovHits; iDigitHit++) - { - times.push_back(digithit_T[iDigitHit]); - - if (digithit_pmtId[iDigitHit] <= 0) std::cout << " Weird PMT ID " << digithit_pmtId[iDigitHit] << std::endl; - HKManager::GetME()->AddHit(digithit_T[iDigitHit], digithit_Q[iDigitHit], iHybrid, digithit_pmtId[iDigitHit]); - - // Bonsai (do not store mPMT hits) - if (iIdx_BS < 2000 && digithit_T[iDigitHit] < 4000. && digithit_T[iDigitHit] > -4000. && iHybrid == 0) - { - // std::cout << pmtId << " " << HitT << std::endl; - bsCAB[iIdx_BS] = digithit_pmtId[iDigitHit]; - bsT[iIdx_BS] = digithit_T[iDigitHit]; // shift BS time is needed if interaction time is < 0, this needs to be considered - bsQ[iIdx_BS] = digithit_Q[iDigitHit]; - - iIdx_BS += 1; + if (pmtType == PMTType::kID) { + for (int iDigitHit = startingCherenkovHitID; iDigitHit < nDigitizedCherenkovHits; iDigitHit++) { + if (digithit_pmtId[iDigitHit] <= 0) std::cout << " Weird PMT ID " << digithit_pmtId[iDigitHit] << std::endl; + + times.push_back(digithit_T[iDigitHit]); + HKManager::GetME()->AddHit_ID(digithit_T[iDigitHit], digithit_Q[iDigitHit], digithit_pmtId[iDigitHit]); + + // Bonsai (do not store mPMT hits) + if (iIdx_BS < 2000 && digithit_T[iDigitHit] < 4000. && digithit_T[iDigitHit] > -4000.) { + // std::cout << pmtId << " " << HitT << std::endl; + bsCAB[iIdx_BS] = digithit_pmtId[iDigitHit]; + bsT[iIdx_BS] = digithit_T[iDigitHit]; // shift BS time is needed if interaction time is < 0, this needs to be considered + bsQ[iIdx_BS] = digithit_Q[iDigitHit]; + + iIdx_BS += 1; + bsnhit[0] = iIdx_BS; + } } } + else { + for (int iDigitHit = startingCherenkovHitID; iDigitHit < nDigitizedCherenkovHits; iDigitHit++) { + if (digithit_pmtId[iDigitHit] <= 0) std::cout << " Weird PMT ID " << digithit_pmtId[iDigitHit] << std::endl; + times.push_back(digithit_T[iDigitHit]); + HKManager::GetME()->AddHit_mPMT(digithit_T[iDigitHit], digithit_Q[iDigitHit], digithit_pmtId[iDigitHit]); + } + + } - // Bonsai - bsnhit[0] = iIdx_BS; fHit = digithit_pmtId.size(); // Compute hits @@ -838,8 +593,7 @@ bool AnalyseEvent(WCSimRootEvent *tEvent, int iEventType) std::vector Hit_time_400; std::sort(times.begin(), times.end()); - for (int iDigitHit = 0; iDigitHit < nDigitizedCherenkovHits; iDigitHit++) - { + for (int iDigitHit = 0; iDigitHit < nDigitizedCherenkovHits; iDigitHit++) { double HitT = times[iDigitHit]; Hit_time_20.push_back(HitT); @@ -853,7 +607,7 @@ bool AnalyseEvent(WCSimRootEvent *tEvent, int iEventType) // Count hit in 50 ns window while (HitT - Hit_time_50[0] > 50.) Hit_time_50.erase(Hit_time_50.begin()); - // Count hit in 200 ns windowAnalyseEvent + // Count hit in 200 ns window while (HitT - Hit_time_200[0] > 200.) Hit_time_200.erase(Hit_time_200.begin()); // Count hit in 400 ns window @@ -865,49 +619,6 @@ bool AnalyseEvent(WCSimRootEvent *tEvent, int iEventType) if ((unsigned int)fHit_400 < Hit_time_400.size()) fHit_400 = Hit_time_400.size(); } } - return true; -} -bool PostLeafAnalysis(WCSimRootEvent * tEvent, int iEventType, FitterOutput leaf_output) -{ - if (!(true_origin_X.size() > 0 && true_origin_Y.size() > 0 && true_origin_Z.size() > 0)) return false; - - for(int j=0;j<4;j++) leaf_Vertex.push_back(leaf_output.Vtx[j]); - for(int j=0;j<3;j++) leaf_Dir.push_back(leaf_output.Dir[j]); - for(int j=0;j<3;j++) leaf_MyDir.push_back(leaf_output.MyDir[j]); - for(int j=0;j<3;j++) leaf_QuickDir.push_back(leaf_output.Quick_Dir[j]); - - //* RESOLUTIONS COMPUTATION - std::vector true_vertex = {true_origin_X[0], true_origin_Y[0], true_origin_Z[0]}; - lf_dWall = calculateDWall(R, h, leaf_output.Vtx); - lf_ToWall = calculateToWall(R, h,leaf_output.Vtx, leaf_output.Dir); - lf_spatial_res = calculateDistance(true_vertex, leaf_output.Vtx); - lf_time_res = abs(leaf_output.Vtx[3]-true_origin_T[0]); - - if(leaf_output.Dir.size() == 3) lf_Dir_res = TMath::ACos(dot(leaf_output.Dir, trueDir))*180./TMath::Pi(); - if(leaf_output.MyDir.size() == 3) lf_MyDir_res = TMath::ACos(dot(leaf_output.MyDir, trueDir))*180./TMath::Pi(); - if(leaf_output.Quick_Dir.size() == 3) lf_Quick_Dir_res = TMath::ACos(dot(leaf_output.Quick_Dir, trueDir))*180./TMath::Pi(); - - lf_energy_res = abs(leaf_output.Energy - true_energy[0]); - - //* ANGLE ANALYSIS - WCSimRootTrigger * fRootTrigger = tEvent->GetTrigger(bestTrigger); - - for(int iDigitHit = startingCherenkovHitID; (long unsigned int)iDigitHit < digithit_pmtId.size(); iDigitHit++) - { - TObject *Hit = (fRootTrigger->GetCherenkovDigiHits())->At(iDigitHit); - WCSimRootCherenkovDigiHit *wcDigitHit = dynamic_cast(Hit); - WCSimRootPMT pmt = fLeafGeometry->GetPMT(wcDigitHit->GetTubeId() - 1, false); - - std::vector lfHitDir = std::vector(3); - for(int j=0;j<3;j++) lfHitDir[j] = pmt.GetPosition(j) - leaf_output.Vtx[j]; - Normalize(lfHitDir); - - std::vector trueHitDir = std::vector(3); - for(int j=0;j<3;j++) trueHitDir[j] = pmt.GetPosition(j) - trueVertex[j]; - Normalize(trueHitDir); - double hitlfRelativeAngle = dot(trueHitDir, leaf_output.Dir); - lf_relativeAngle.push_back((TMath::ACos(hitlfRelativeAngle))*180./TMath::Pi()); - } return true; -} \ No newline at end of file +} diff --git a/example/analysis.h b/example/analysis.hpp similarity index 75% rename from example/analysis.h rename to example/analysis.hpp index 6d22c78..250d7fb 100644 --- a/example/analysis.h +++ b/example/analysis.hpp @@ -24,17 +24,17 @@ #include "WCSimRootGeom.hh" #include "WCSimEnumerations.hh" -#include "LEAF.hh" -#include "LeafDefinitions.hh" -#include "HKManager.hh" +#include "LEAF.hpp" +#include "LEAFUtilities.hpp" +#include "HKManager.hpp" #define OLD_WCSIM // To be used if WCSim version is older than 1.8 (i.e. without multi vertex) // #define mPMT // To be used if you are using mPMT -#define ID_EVENT 1 -#define OD_EVENT 2 -// #define mPMT_EVENT 0 -#define UNDEFINED_EVENT 0 +#define ID_EVENT PMTType::kID +#define OD_EVENT PMTType::kOD +#define mPMT_EVENT PMTType::kmPMT +#define UNDEFINED_EVENT PMTType::kUndefined //-----------------------------------------------------------------------------------------// @@ -48,14 +48,14 @@ int usedTriggerId; // trigger Id used for the fit double fLfTriggerTime; int bestTrigger; double rawTriggerTime; + std::vector true_particleId; // True particle Id (PDGId) -std::vector true_energy; // True energy (MeV) -std::vector true_origin_X; // True origin vertex (cm) -std::vector true_origin_Y; // True origin vertex (cm) -std::vector true_origin_Z; // True origin vertex (cm) -std::vector true_origin_T; // True origin vertex (ns) -std::vector trueDir; -std::vector trueVertex; +std::vector true_energies; // True energy (MeV) +std::vector true_vertexes; // True origin vertex (cm) +ROOT::Math::XYZVector true_dir; +ROOT::Math::XYZTVector true_vertex; +double true_energy; + double dWall =0; double toWall =0; int digithit_num; // Number of digit hit @@ -110,11 +110,11 @@ std::vector Charge_PMT; std::vector hit_residual; // List of residual times //* Computation Times -FitterOutputProps fOutputProps; +LEAFStructures::FitterPerformances fLeafPerformances; +LEAFStructures::FitterOutput fLeafOutput; double fLFTime; //* Other hits infos - int Hit_ID; int Hit_ID_20; int Hit_ID_50; @@ -153,10 +153,38 @@ float bsT[2000]; float bsQ[2000]; int bsnhit[1]; + //-----------------------------------------------------------------------------------------// -void SetCustomBranch(TTree *fPrimaryTree, FitterOutput leaf_output); -void SetCustomBranchInput(TTree *fPrimaryTree); -void SetGeoBranch(TTree *fGeoTree); -bool AnalyseEvent(WCSimRootEvent *tEvent, int iEventType); -bool PostLeafAnalysis(WCSimRootEvent * tEvent, int iEventType, FitterOutput leaf_output); +struct arguments +{ + std::string inputFile = ""; + std::string outputFile = ""; + double darkNoise = 4.2; // Dark noise frequency in Hz + double darkNoiseH = 0.; // Dark noise frequency in Hz when using the hybrid geometry + double timeshift = 0.; // Shift hit time + int startEvent = 0; // First event to analyze + int endEvent = 0; // Last event to analyze + bool hybrid = false; + bool verbose = false; +}; + +struct FitterAnalysis +{ + double Wall; + double Good; + int n50[3]; + double dir[3][3]; + double dir_goodness[3]; + double dirKS[3]; +}; + + +//-----------------------------------------------------------------------------------------// + +void SetCustomBranch(TTree *fPrimaryTree); +bool AnalyseEvent(WCSimRootEvent *tEvent, PMTType pmtType); +arguments FetchInput(int argc, char* argv[]); + + + diff --git a/example/analysis_fullrec.cpp b/example/analysis_fullrec.cpp new file mode 100644 index 0000000..886a1f1 --- /dev/null +++ b/example/analysis_fullrec.cpp @@ -0,0 +1,1196 @@ +/*********************************************************************************/ +/** analysis_fullrec.cpp **/ +/** Author: Benjamin Quilain (benjamin.quilain@llr.in2p3.fr) **/ +/** Date: April 10th 2025 **/ +/** Desc: Perform the vertex, direction and energy reconstruction **/ +/*********************************************************************************/ + + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + + +#include "TFile.h" +#include "TTree.h" +#include "TBranch.h" +#include "TObject.h" +#include "TSystem.h" +#include "TRandom3.h" +#include "TStopwatch.h" + +#include "WCSimRootEvent.hh" +#include "WCSimRootGeom.hh" +#include "WCSimEnumerations.hh" + +#include "LEAF.hpp" +#include "HKManager.hpp" +//#define WITH_BONSAI +#ifdef WITH_BONSAI +#include "WCSimBonsai.hh" +#endif + +//-----------------------------------------------------------------------------------------// + + int eventId; // event Id + int triggerId; // trigger Id (with event Id) + + double TotalQ; + + ROOT::Math::XYZVector particleDir(0,0,0); + struct FitterAnalysis { + double Wall; + double Good; + int n50[3]; + double dir[3][3]; + double dir_goodness[3]; + + double dirKS[3]; + }; + + void Normalize(double a[3]){ + double l; + l=sqrt(a[0]*a[0]+a[1]*a[1]+a[2]*a[2]); + for(int j=0; j<3; j++){ + a[j]/=l; + } + } + + //True information + std::vector true_particleId; // True particle Id (PDGId) + std::vector true_energies; // True energy (MeV) + std::vector true_origins; // True origin vertex (cm) + std::vector relativeAngle; + std::vector lf_relativeAngle; + + ROOT::Math::XYZVector lf_Dir; + ROOT::Math::XYZVector lf_Dir_interp; + double lf_Dir_res; + WCSimRootGeom *geotree = 0; + + //Raw hit + std::vector rawhit_pmtId; // List of pmtId + std::vector rawhit_Type; // List of pmt Type + std::vector rawhit_T; // List of hit times + std::vector rawhit_Q; // List of hit charge' + std::vector rawhit_dark; // List of DarkNoise flag + int rawhit_num; // Number of rawhit + int rawhit_num_noDN; // Number of rawhit without DarkNoise + + + std::vector rawhit_pmtId_tmp; // List of pmtId + std::vector rawhit_Type_tmp; // List of pmt Type + std::vector rawhit_T_tmp; // List of hit times + + // In case of mPMT mixed with B&L we don't want to re apply the Digitization on the other type PMTs + int fLastRawHit; + + + //Digitized hit + std::vector digithit_pmtId; // List of pmtId + std::vector digithit_Type; // List of pmt Type + std::vector digithit_T; // List of hit times + std::vector digithit_Q; // List of hit charge' + std::vector Charge_PMT; + std::vector digithit_dark; // List of DarkNoise flag + int digithit_num; // Number of digit hit + int digithit_num_noDN; // Number of digit hit without DarkNoise + + //Bonsai output + float bs_vertex[4]; // Bonsai reconstructed vertex (cm) + float bs_good[3]; // Bonsai goodness + float bs_energy; + + double fBSTime; + double fLFTime; + double lf_spatial_res; + double lf_jitter; + LEAFStructures::FitterOutput leaf_output; + FitterAnalysis leaf_output_ana; + FitterAnalysis bs_output_ana; + + int Hit_ID; + int Hit_ID_50; + int Hit_ID_200; + int Hit_ID_400; + + int Hit_mPMT; + int Hit_mPMT_50; + int Hit_mPMT_200; + int Hit_mPMT_400; + + int Hit_OD; + int Hit_OD_50; + int Hit_OD_200; + int Hit_OD_400; + + int fHit = 0; + int fHit_50 = 0; + int fHit_200 = 0; + int fHit_400 = 0; + double dWall =0; + double lf_dWall =0; + double lf_ToWall =0; + WCSimRootGeom * fGeometry; + + double R = 3242.96; + double h = 6701.41; + + // Input variable for Bonsai + int bsCAB[2000]; + float bsT[2000]; + float bsQ[2000]; + int bsnhit[1]; + +//-----------------------------------------------------------------------------------------// + + +void SetCustomBranch(TTree* fPrimaryTree); +void SetCustomBranchInput(TTree* fPrimaryTree); +void SetGeoBranch(TTree* fGeoTree); + +// Apply analysis on the event +bool AnalyseEvent(WCSimRootEvent * tEvent, PMTType iEventType, int i); +bool AnalyselfEvent(WCSimRootEvent * tEvent, PMTType iEventType); +TH1D *hRelativeAngle = nullptr; +TH1D *lf_hRelativeAngle = nullptr; +TH1D *hRelativeAngleCos = nullptr; +TH1D *lf_hRelativeAngleCos = nullptr; +TH1D *ToWall_Charge = nullptr; +TH2D *ToWall_Charge_2D = nullptr; +TH1D* Sum_HitQ = nullptr; +TH1D* Count_Hits = nullptr; + +TH2D* hToWall_RelAngle_Charge = new TH2D("ToWall_RelAngle_Charge", "ToWall vs Rel. Angle and Charge", 100, 0, 9000, 360, 0, 180); +TProfile* pAngle_ToWall = new TProfile("pAngle_ToWall", "Angle vs ToWall", 360, 0, 180); + +TH2D* hdWall_TotalCharge = new TH2D("DWall_TotalCharge", "dWall vs Total Charge", 150, 0, 3500, 50, 0, 400); +TH2D* hdWall_TotalCharge_corr = new TH2D("DWall_TotalChargeCorr", "dWall vs Corrected Total Charge", 150, 0, 3500, 50, 0, 400); +TH1D* hdWall_TotalCharge_hist = new TH1D("DWall_TotalCharge_hist", "dWall vs Total Charge (Hist)", 150, 0, 3500); +TH1D* Sum_HitQ_hdWall = new TH1D("Sum_HitQ", "Sum of HitQ per Wall bin", 150, 0, 3500); +TH1D* Count_Hits_hdWall = new TH1D("Count_Hits", "Count of Hits per Wall bin", 150, 0, 3500); + + + +TH1D* hAngle_HitQ = new TH1D("Angle_HitQ", "Angle vs Hit Charge", 360, 0, 90); + +void initializeHistograms() { + hToWall_RelAngle_Charge->GetXaxis()->SetTitle("ToWall [cm]"); + hToWall_RelAngle_Charge->GetYaxis()->SetTitle("Relative Angle [degrees]"); + hToWall_RelAngle_Charge->GetZaxis()->SetTitle("Charge [p.e.]"); + + hdWall_TotalCharge->GetXaxis()->SetTitle("dWall [cm]"); + hdWall_TotalCharge->GetYaxis()->SetTitle("Total Charge [p.e.]"); + + hdWall_TotalCharge_corr->GetXaxis()->SetTitle("dWall [cm]"); + hdWall_TotalCharge_corr->GetYaxis()->SetTitle("Corrected Total Charge [p.e.]"); + + hdWall_TotalCharge_hist->GetXaxis()->SetTitle("dWall [cm]"); + hdWall_TotalCharge_hist->GetYaxis()->SetTitle("Total Charge [p.e.]"); + + hAngle_HitQ->GetXaxis()->SetTitle("Angle [degrees]"); + hAngle_HitQ->GetYaxis()->SetTitle("Hit Charge [p.e.]"); +} + +double calculateToWall(double R, double h, const ROOT::Math::XYZVector& position, const ROOT::Math::XYZVector& direction) { + double x = position.X(), y = position.Y(), z = position.Z(); + double dx = direction.X(), dy = direction.Y(), dz = direction.Z(); + + double A = dx * dx + dy * dy; + double B = 2 * (x * dx + y * dy); + double C = x * x + y * y - R * R; + + double discriminant = B * B - 4 * A * C; + + double t_cylinder = -1; + if (discriminant >= 0) { + double t1 = (-B - sqrt(discriminant)) / (2 * A); + double t2 = (-B + sqrt(discriminant)) / (2 * A); + + if (t1 > 0) t_cylinder = t1; + if (t2 > 0 && (t2 < t1 || t_cylinder < 0)) t_cylinder = t2; + } + + //Calculate the t values for intersection with top and bottom + double t_top = (h / 2 - z) / dz; + double t_bottom = (-h / 2 - z) / dz; + + //Check if the intersections are on the top or bottom + double x_top = x + t_top * dx; + double y_top = y + t_top * dy; + double x_bottom = x + t_bottom * dx; + double y_bottom = y + t_bottom * dy; + + if (x_top * x_top + y_top * y_top > R * R) t_top = -1; //Invalid if outside radius + if (x_bottom * x_bottom + y_bottom * y_bottom > R * R) t_bottom = -1; //Invalid if outside radius + + //Find the smallest positive t that corresponds to a valid intersection (side or top/bottom) + double t_wall = -1; + if (t_cylinder > 0) t_wall = t_cylinder; + if (t_top > 0 && (t_top < t_wall || t_wall < 0)) t_wall = t_top; + if (t_bottom > 0 && (t_bottom < t_wall || t_wall < 0)) t_wall = t_bottom; + + return t_wall; +} + +double calculateDWall(double R, double h, const ROOT::Math::XYZVector& position) { + double x = position.X(), y = position.Y(), z = position.Z(); + double distanceToAxis = sqrt(x * x + y * y); + double dSide = fabs(distanceToAxis - R); + double dTop = fabs(z - h / 2); + double dBottom = fabs(z + h / 2); + return std::min({dSide, dTop, dBottom}); +} + + + + + + +int main(int argc, char** argv){ + + initializeHistograms(); + + + + + std::string sInputFile = ""; + std::string sOutputFile = ""; + + int iNeededArgc = 5; + double dDarkNoise = 0.; //kHz + double dDarkNoiseHybrid = 0.; //kHz + iNeededArgc += 1; + int StartEvent; + int NumberEvents; +#ifdef mPMT + iNeededArgc += 1; +#endif + + if ( argc == iNeededArgc ) { + int iArg = 1; + sInputFile = argv[iArg]; iArg += 1; + sOutputFile = argv[iArg]; iArg += 1; + dDarkNoise = atof(argv[iArg]); iArg += 1; +#ifdef mPMT + dDarkNoiseHybrid = atof(argv[iArg]); iArg += 1; +#endif + StartEvent = atoi(argv[iArg]); iArg += 1; + NumberEvents = atoi(argv[iArg]); iArg += 1; + + } + else { + + std::cout << "Synthax: " << argv[0] << " input output"; + std::cout << " DN_in_kHz_B&L"; + +#ifdef mPMT + std::cout << " DN_in_kHz_mPMT"; +#endif + std::cout << std::endl; + return 0; + } + + + // Read WCSim output + TFile* fInputFile = new TFile(sInputFile.c_str(),"READ"); + + // Get TTrees + TTree* fInputTree = (TTree*) fInputFile->Get("wcsimT"); + TTree *fInputGeoTree = (TTree*) fInputFile->Get("wcsimGeoT"); + + fGeometry = 0; + fInputGeoTree->SetBranchAddress("wcsimrootgeom", &fGeometry); + + WCSimRootEvent * fIDevent = new WCSimRootEvent(); + // Set Branche + fInputTree->SetBranchAddress("wcsimrootevent" ,&fIDevent ); + // Set autodelete to avoid memory leak + fInputTree->GetBranch("wcsimrootevent" )->SetAutoDelete(kTRUE); + +#ifdef mPMT + WCSimRootEvent * fHybridevent = new WCSimRootEvent(); + // Set Branche + fInputTree->SetBranchAddress("wcsimrootevent2" ,&fHybridevent ); + // Set autodelete to avoid memory leak + fInputTree->GetBranch("wcsimrootevent2" )->SetAutoDelete(kTRUE); +#endif + +#ifdef OD_ON + WCSimRootEvent * fODevent = new WCSimRootEvent(); + // Set Branche + fInputTree->SetBranchAddress("wcsimrootevent_OD",&fODevent ); + // Set autodelete to avoid memory leak + fInputTree->GetBranch("wcsimrootevent_OD")->SetAutoDelete(kTRUE); +#endif + + // Read Geo + fInputGeoTree->GetEntry(0); + + + // Create Output TTree + TFile* fOutputFile = new TFile(sOutputFile.c_str(),"RECREATE"); + fOutputFile->SetCompressionLevel(2); + TTree* fGeoTree = new TTree("wcsimGeoT","Geometry TTree"); + TTree* fPrimaryTree = new TTree("Reduced","Reduced TTree"); + + // Set Branches + SetGeoBranch(fGeoTree); + SetCustomBranch(fPrimaryTree); + + fGeoTree->Fill(); + + // Get PMT Number: + int nPMT_ID = fGeometry->GetWCNumPMT(); +#ifdef OD_ON + int nPMT_OD = fGeometry->GetODWCNumPMT(); +#endif + int nMultPMT = fGeometry->GetWCNumPMT(true); + // TODO solve this issue! + + std::cout << " ID " << nPMT_ID << std::endl; + std::cout << " mPMT " << nMultPMT << std::endl; + + HKManager::GetME()->SetGeometry(fGeometry,dDarkNoise * 1e3,dDarkNoiseHybrid * 1e3); + + // Initialize LEAF + LEAF::GetME()->Initialize( HKManager::GetME()->GetGeometry(), + HKManager::GetME()->GetDarkNoise(), + HKManager::GetME()->GetGeometryPMT_ID(), + HKManager::GetME()->GetGeometryPMT_mPMT()); + LEAF::GetME()->SetNThread(); // Set number of Threads, default in the class is 12 + + // Initialize HKAstroAnalysis +#ifdef WITH_HK_ASTROANALYSIS + HKAstroAnalysis::GetME()->Initialize(HKManager::GetME()->GetGeometry()); +#endif + +#ifdef WITH_BONSAI + // Initialize Bonsai + WCSimBonsai* fBonsai = new WCSimBonsai(); + fBonsai->Init(fGeometry); +#endif + // Read Input Tree + int nPrimaryEvents = fInputTree->GetEntries(); + int iWrite = 0; + + TStopwatch timer; + timer.Reset(); + timer.Start(); + + + lf_hRelativeAngle = new TH1D("lf_ChargeProfile", "lf_ChargeProfile", 120, 0, 180); + lf_hRelativeAngleCos = new TH1D("lf_ChargeProfileCos", "lf_ChargeProfileCos", 100, -1, 1); + hRelativeAngle = new TH1D("ChargeProfile", "ChargeProfile", 120, 0, 180); + hRelativeAngleCos = new TH1D("ChargeProfileCos", "ChargeProfileCos", 100, -1, 1); + ToWall_Charge = new TH1D("ToWallCharge_hist", "ToWallCharge (Hist)", 1000, 0, 10000); + ToWall_Charge_2D = new TH2D("ToWallCharge", "ToWallCharge", 1000, 0, 10000, 50, 0, 400); + Sum_HitQ = new TH1D("Sum_HitQ", "Sum of HitQ per Wall bin", 1000, 0, 10000); + Count_Hits = new TH1D("Count_Hits", "Count of Hits per Wall bin", 1000, 0, 10000); + + // Loop on Primary events + for(int i=StartEvent; i < StartEvent + NumberEvents; i++){ + // Reset Hit vector + HKManager::GetME()->ResetHitInfo(); + + //if ( i%1000==0 ) { + timer.Stop(); + std::cout << "Event # = " << i << " / " << nPrimaryEvents << " ( " << timer.RealTime() << " )\n"; + timer.Reset(); + timer.Start(); + //} + + fInputTree->GetEntry(i); + + // Initialize output variables + eventId = iWrite; + triggerId = 0; + rawhit_num_noDN = 0; + + true_particleId.clear(); + true_energies.clear(); + true_origins.clear(); + + digithit_pmtId.clear(); + digithit_T.clear(); + digithit_Q.clear(); + Charge_PMT.clear(); + digithit_dark.clear(); + relativeAngle.clear(); + lf_relativeAngle.clear(); + rawhit_num = 0; + digithit_num = 0; + + bs_vertex [0] = -9999.; + bs_vertex [1] = -9999.; + bs_vertex [2] = -9999.; + bs_vertex [3] = -9999.; + bs_good [0] = -9999.; + bs_good [1] = -9999.; + bs_good [2] = -9999.; + + fBSTime = -9999; + fLFTime = -9999; + + Hit_ID = 0; + Hit_ID_50 = 0; + Hit_ID_200 = 0; + Hit_ID_400 = 0; + + Hit_mPMT = 0; + Hit_mPMT_50 = 0; + Hit_mPMT_200 = 0; + Hit_mPMT_400 = 0; + + Hit_OD = 0; + Hit_OD_50 = 0; + Hit_OD_200 = 0; + Hit_OD_400 = 0; + + + + +#ifdef WITH_BONSAI + // re-initialize Bonsai input + for ( int iHit = 0; iHit < 2000; iHit++ ) { + bsCAB[iHit] = 0; + bsT [iHit] = 0.; + bsQ [iHit] = 0.; + } + bsnhit[0] = 0; +#endif + leaf_output.vtx = ROOT::Math::XYZTVector(0,0,0,0); + leaf_output.vtx_nll = -9999.; + //leaf_output.InTime = 0; + leaf_output.energy = 0.; + leaf_output.dir = ROOT::Math::XYZVector(0,0,0); + //leaf_output.SNRList = std::vector(); + + leaf_output_ana.Wall = 0; + for ( int iType = 0; iType < 3; iType++ ) { + leaf_output_ana.n50 [iType] = 0; + leaf_output_ana.dir [iType][0] = 0; + leaf_output_ana.dir [iType][1] = 0; + leaf_output_ana.dir [iType][2] = 0; + leaf_output_ana.dir_goodness [iType] = 0; + leaf_output_ana.dirKS [iType] = 0; + } + + fLastRawHit = 0; + + /****************************************************************************************/ + /* ID events */ + /****************************************************************************************/ + + //std::cout << " ID Event " << std::endl; + fHit = 0; + fHit_50 = 0; + fHit_200 = 0; + fHit_400 = 0; + + /*bool bID =*/ AnalyseEvent(fIDevent,PMTType::kID,i); + + + + Hit_ID = fHit; + Hit_ID_50 = fHit_50; + Hit_ID_200 = fHit_200; + Hit_ID_400 = fHit_400; + + //std::cout << " Hit: " << Hit_ID << " event " << i << std::endl; + + /****************************************************************************************/ + /* mPMT events */ + /****************************************************************************************/ + + //std::cout << " mPMT Event " << std::endl; + fHit = 0; + fHit_50 = 0; + fHit_200 = 0; + fHit_400 = 0; + +#ifdef mPMT + /*bool bmPMT =*/ AnalyseEvent(fHybridevent,PMTType::kmPMT,i); +#endif + Hit_mPMT = fHit; + Hit_mPMT_50 = fHit_50; + Hit_mPMT_200 = fHit_200; + Hit_mPMT_400 = fHit_400; + + //std::cout << " Hit: " << Hit_mPMT << " event " << i << std::endl; + + /****************************************************************************************/ + /* OD events */ + /****************************************************************************************/ +#ifdef OD_ON + fHit = 0; + fHit_50 = 0; + fHit_200 = 0; + fHit_400 = 0; + + // There should be a dedicated OD analyser, as many thing should be different than for ID + // Doesn't exist yet + /*bool bOD =*/ AnalyseODEvent(fODevent,PMTType::kOD); + + Hit_OD = fHit; + Hit_OD_50 = fHit_50; + Hit_OD_200 = fHit_200; + Hit_OD_400 = fHit_400; +#endif + /****************************************************************************************/ + /* Benjamin Fitter */ + /****************************************************************************************/ + + //std::cout << " Start LEAF " << std::endl; + TStopwatch timerLF; + timerLF.Reset(); + timerLF.Start(); + + LEAF::GetME()->LoadHitsCollection(HKManager::GetME()->GetHitCollection_ID(),HKManager::GetME()->GetHitCollection_mPMT()); + leaf_output = LEAF::GetME()->MakeSequentialFit(); + + timerLF.Stop(); + + fLFTime = timerLF.RealTime(); + + //std::cout << " n50 " << leaf_output_ana.n50[0] << " " << leaf_output_ana.n50[1] << " " << leaf_output_ana.n50[2] << std::endl; + std::cout << " LEAF took: " << timerLF.RealTime() << " for " << HKManager::GetME()->GetHitCollection_ID()->size() << " Hits"<< std::endl; + AnalyselfEvent(fIDevent,PMTType::kID); + +#ifdef WITH_HK_ASTROANALYSIS + HKAstroAnalysis::GetME()->SetVertex(leaf_output.Vtx); + + leaf_output_ana.Wall = HKAstroAnalysis::GetME()->ComputeDistanceFromWall(); + + HKAstroAnalysis::GetME()->MakeAnalysis(HKManager::GetME()->GetHitCollection(),NormalPMT); + + leaf_output_ana.n50 [NormalPMT] = HKAstroAnalysis::GetME()->Getn50(); + leaf_output_ana.dirKS [NormalPMT] = HKAstroAnalysis::GetME()->GetdirKS(); + leaf_output_ana.dir [NormalPMT][0] = HKAstroAnalysis::GetME()->Getdir_Simple()[0]; + leaf_output_ana.dir [NormalPMT][1] = HKAstroAnalysis::GetME()->Getdir_Simple()[1]; + leaf_output_ana.dir [NormalPMT][2] = HKAstroAnalysis::GetME()->Getdir_Simple()[2]; + leaf_output_ana.dir_goodness [NormalPMT] = HKAstroAnalysis::GetME()->Getdir_Simple()[3]; + + HKAstroAnalysis::GetME()->MakeAnalysis(HKManager::GetME()->GetHitCollection(),MiniPMT); + + leaf_output_ana.n50 [MiniPMT] = HKAstroAnalysis::GetME()->Getn50(); + leaf_output_ana.dirKS [MiniPMT] = HKAstroAnalysis::GetME()->GetdirKS(); + leaf_output_ana.dir [MiniPMT][0] = HKAstroAnalysis::GetME()->Getdir_Simple()[0]; + leaf_output_ana.dir [MiniPMT][1] = HKAstroAnalysis::GetME()->Getdir_Simple()[1]; + leaf_output_ana.dir [MiniPMT][2] = HKAstroAnalysis::GetME()->Getdir_Simple()[2]; + leaf_output_ana.dir_goodness [MiniPMT] = HKAstroAnalysis::GetME()->Getdir_Simple()[3]; + + HKAstroAnalysis::GetME()->MakeAnalysis(HKManager::GetME()->GetHitCollection(),AllPMT); + + leaf_output_ana.n50 [AllPMT] = HKAstroAnalysis::GetME()->Getn50(); + leaf_output_ana.dirKS [AllPMT] = HKAstroAnalysis::GetME()->GetdirKS(); + leaf_output_ana.dir [AllPMT][0] = HKAstroAnalysis::GetME()->Getdir_Simple()[0]; + leaf_output_ana.dir [AllPMT][1] = HKAstroAnalysis::GetME()->Getdir_Simple()[1]; + leaf_output_ana.dir [AllPMT][2] = HKAstroAnalysis::GetME()->Getdir_Simple()[2]; + leaf_output_ana.dir_goodness [AllPMT] = HKAstroAnalysis::GetME()->Getdir_Simple()[3]; + + leaf_output_ana.Good = HKAstroAnalysis::GetME()->GoodnessBonsai(); + +#endif + + if (true_origins.size() > 0 ) { + ROOT::Math::XYZVector true_vertex(true_origins[0].X(), true_origins[0].Y(), true_origins[0].Z()); + ROOT::Math::XYZVector reconstructed_vertex(leaf_output.vtx.X(), leaf_output.vtx.Y(), leaf_output.vtx.Z()); + + ROOT::Math::XYZVector lf_vertex(reconstructed_vertex); + ROOT::Math::XYZVector lf_Dire(leaf_output.dir); + lf_dWall = calculateDWall(R, h, lf_vertex); + lf_ToWall = calculateToWall(R, h, lf_vertex, lf_Dire); + lf_spatial_res = LEAFUtilities::GetDistance(true_vertex, reconstructed_vertex); + lf_jitter = abs(leaf_output.vtx.T()-true_origins[0].T()); + + TotalQ+=leaf_output.energy; + lf_Dir_interp = leaf_output.dir; + double angle_lf = lf_Dir_interp.Dot(particleDir); + + + lf_Dir_res = TMath::ACos(angle_lf)*180./TMath::Pi(); + + + } else { + lf_spatial_res = -9999; // Assign invalid value if true vertex is missing + } +#ifdef WITH_BONSAI + /****************************************************************************************/ + /* Bonsai */ + /****************************************************************************************/ + + if ( bsnhit[0] < 2000 && bsnhit[0] > 0 ) { + + TStopwatch timerBS; + timerBS.Reset(); + timerBS.Start(); + + // Some variable for Bonsai: + float bsvertex[4]; + float bsresult[6]; + float bsgood[3]; + int bsnsel[1]; //nsel (SLE) + + // Fit with Bonsai + //std::cout << "Bonsai hit: " << bsnhit[0] << std::endl; + //for ( int iHit = 0; iHit < bsnhit[0]; iHit++ ) { + // std::cout << " hit " << iHit << " cab " << bsCAB[iHit] << " T " << bsT[iHit] << " Q " << bsQ[iHit] << std::endl; + //} + + fBonsai->BonsaiFit( bsvertex, bsresult, bsgood, bsnsel, bsnhit, bsCAB, bsT, bsQ); + +#ifdef WITH_HK_ASTROANALYSIS + HKAstroAnalysis::GetME()->SetVertex(bs_vertex); + + bs_output_ana.Wall = HKAstroAnalysis::GetME()->ComputeDistanceFromWall(); + + HKAstroAnalysis::GetME()->MakeAnalysis(HKManager::GetME()->GetHitCollection()); + + bs_output_ana.n50 [NormalPMT] = HKAstroAnalysis::GetME()->Getn50(); + bs_output_ana.dirKS [NormalPMT] = HKAstroAnalysis::GetME()->GetdirKS(); + bs_output_ana.dir [NormalPMT][0] = HKAstroAnalysis::GetME()->Getdir_Simple()[0]; + bs_output_ana.dir [NormalPMT][1] = HKAstroAnalysis::GetME()->Getdir_Simple()[1]; + bs_output_ana.dir [NormalPMT][2] = HKAstroAnalysis::GetME()->Getdir_Simple()[2]; + bs_output_ana.dir_goodness [NormalPMT] = HKAstroAnalysis::GetME()->Getdir_Simple()[3]; + + bs_output_ana.Good = HKAstroAnalysis::GetME()->GoodnessBonsai(); +#endif + + bs_vertex[0] = bsvertex[0]; + bs_vertex[1] = bsvertex[1]; + bs_vertex[2] = bsvertex[2]; + bs_vertex[3] = bsvertex[3]; + + bs_good[0] = bsgood[0]; + bs_good[1] = bsgood[1]; + bs_good[2] = bsgood[2]; + + //float diff = sqrt(pow(true_origin_X[0]- bsvertex[0],2.)+pow(true_origin_X[0]- bsvertex[0],2.)+pow(true_origin_X[0]- bsvertex[0],2.)); + + //timerBS.Stop(); + //std::cout << " origin " << true_origin_X[0] << " " << true_origin_Y[0] << " " << true_origin_Z[0] << " diff: " << diff << std::endl; + //std::cout << " BS took: " << timerBS.RealTime() << " goodness " << bsgood[0]<< " " << bsgood[1]<< " " << bsgood[2] << " bsvertex: " << bsvertex[0] << " " << bsvertex[1] << " " << bsvertex[2] << std::endl; + + + fBSTime = timerBS.RealTime(); + } +#endif + /****************************************************************************************/ + /* Fill output tree */ + /****************************************************************************************/ + + fPrimaryTree->Fill(); + iWrite += 1; + } + std::cout << "Event # = " << nPrimaryEvents << " / " << nPrimaryEvents << std::endl; +// Convert histograms to graphs + +fOutputFile->cd(); + + + + + + lf_hRelativeAngle->Write(); + lf_hRelativeAngleCos->Write(); + hRelativeAngle->Write(); + hRelativeAngleCos->Write(); + ToWall_Charge->Write(); + hToWall_RelAngle_Charge->Write(); + pAngle_ToWall->Write(); + hdWall_TotalCharge->Write(); + hdWall_TotalCharge_corr->Write(); + hdWall_TotalCharge_hist->Write(); + ToWall_Charge_2D->Write(); + hAngle_HitQ->Write(); + + fOutputFile->Write("",TObject::kOverwrite); + + // Cleaning + delete fPrimaryTree; + delete fOutputFile; + + //std::cout << "Total Charge: " << leaf_output.TotalCharge << std::endl; + return 1; +//TotalQ 1000e10Mev: 207435 -- 100e10Mev: 20944.7 -- 1000e5MeV: 144119 -- 1000e3MeV: 45966.4 +} + +void SetCustomBranch(TTree* fPrimaryTree) { + + fPrimaryTree->Branch("eventId", &eventId, "eventId/I"); + fPrimaryTree->Branch("triggerId", &triggerId, "triggerId/I"); + + fPrimaryTree->Branch("true_particleId", &true_particleId); + fPrimaryTree->Branch("true_origins", &true_origins); + fPrimaryTree->Branch("true_energies", &true_energies); + + fPrimaryTree->Branch("rawhit_num", &rawhit_num, "rawhit_num/I"); + fPrimaryTree->Branch("digithit_num", &digithit_num, "digithit_num/I"); + + fPrimaryTree->Branch("ID_hits", &Hit_ID, "Hit_ID/I"); + fPrimaryTree->Branch("ID_hits_50", &Hit_ID_50, "Hit_ID_50/I"); + fPrimaryTree->Branch("ID_hits_200", &Hit_ID_200, "Hit_ID_200/I"); + fPrimaryTree->Branch("ID_hits_400", &Hit_ID_400, "Hit_ID_400/I"); + + fPrimaryTree->Branch("mPMT_hits", &Hit_mPMT, "Hit_mPMT/I"); + fPrimaryTree->Branch("mPMT_hits_50", &Hit_mPMT_50, "Hit_mPMT_50/I"); + fPrimaryTree->Branch("mPMT_hits_200", &Hit_mPMT_200, "Hit_mPMT_200/I"); + fPrimaryTree->Branch("mPMT_hits_400", &Hit_mPMT_400, "Hit_mPMT_400/I"); + + fPrimaryTree->Branch("bs_vertex", bs_vertex, "bs_vertex[4]/F"); + fPrimaryTree->Branch("bs_good", bs_good, "bs_good[3]/F"); + fPrimaryTree->Branch("bs_ctime", &fBSTime, "bs_ctime/F"); // Computation time + + fPrimaryTree->Branch("lf_spatial_res", &lf_spatial_res, "lf_spatial_res/D"); + fPrimaryTree->Branch("lf_jitter", &lf_jitter, "lf_jitter/D"); + fPrimaryTree->Branch("DWall", &dWall, "DWall/D"); + fPrimaryTree->Branch("lf_DWall", &lf_dWall, "DWall/D"); + fPrimaryTree->Branch("lf_ToWall", &lf_ToWall, "DWall/D"); + + fPrimaryTree->Branch("reconstructed_energy", &leaf_output.energy, "reconstructed_energy/D"); + fPrimaryTree->Branch("TotalCharge", &leaf_output.total_charge, "TotalCharge/D"); + fPrimaryTree->Branch("ParticleDir", &particleDir,"particleDir[3]/D"); + fPrimaryTree->Branch("lf_Dir", &leaf_output.dir); + fPrimaryTree->Branch("lf_Dir_res", &lf_Dir_res,"lf_Dir_res/D"); + //fPrimaryTree->Branch("SNR", &leaf_output.SNRList); + + fPrimaryTree->Branch("RelativeAngle", &relativeAngle); + fPrimaryTree->Branch("lf_RelativeAngle", &lf_relativeAngle); + + fPrimaryTree->Branch("DigiHitQ", &digithit_Q); + fPrimaryTree->Branch("Charge_PMT", &Charge_PMT); + fPrimaryTree->Branch("DigiHitT", &digithit_T); + + + + + fPrimaryTree->Branch("lf_vertex", &leaf_output.vtx); + fPrimaryTree->Branch("lf_NLL", &leaf_output.vtx_nll, "lf_NLL/D"); + //fPrimaryTree->Branch("lf_intime", &leaf_output.InTime, "lf_intime/I"); + fPrimaryTree->Branch("lf_good", &leaf_output_ana.Good, "lf_good/D"); + fPrimaryTree->Branch("lf_wall", &leaf_output_ana.Wall, "lf_wall/D"); + fPrimaryTree->Branch("lf_n50", &leaf_output_ana.n50, "lf_n50[3]/I"); + fPrimaryTree->Branch("lf_dir", &leaf_output_ana.dir, "lf_dir[3][3]/D"); + fPrimaryTree->Branch("lf_dir_goodness", &leaf_output_ana.dir_goodness, "lf_dir_goodness[3]/D"); + fPrimaryTree->Branch("lf_dirKS", &leaf_output_ana.dirKS, "lf_dirKS[3]/D"); + fPrimaryTree->Branch("lf_ctime", &fLFTime, "lf_ctime/D"); // Computation time + +} + +void SetGeoBranch(TTree* fGeoTree) { + + fGeoTree->Branch("wcsimrootgeom", fGeometry); + +} +bool AnalyselfEvent(WCSimRootEvent * tEvent, PMTType pmtType){ + + int nVertex = 0; // Number of Vertex in event + int nTrack = 0; // Number of Track in event + int nRawCherenkovHits = 0; // Number of Raw Cherenkov hits + //int startingCherenkovHitID = 0; // starting ID of Digitized Cherenkov hits. Usually starts at 0, but if there are two types of PMTs, it does not. + int nDigitizedCherenkovHits = 0; // Number of Digitized Cherenkov hits + //double fTriggerTime = 0.; + + // Declare some useful variables + WCSimRootTrigger * fRootTrigger; + //TClonesArray * fTimeArray; + + // Currently only one Trigger is used (to be check) + //int iTrig = 0; + for(int iTrig = 0; iTrig < tEvent->GetNumberOfEvents(); iTrig++){ + + triggerId = iTrig; + fRootTrigger = tEvent->GetTrigger(iTrig); + + // Grab the big arrays of times and parent IDs + //fTimeArray = fRootTrigger->GetCherenkovHitTimes(); + + // Get number of vertex and tracks +#ifdef OLD_WCSIM + nVertex = 1; +#else + nVertex = fRootTrigger->GetNvtxs(); +#endif + + nTrack = fRootTrigger->GetNtrack(); + nRawCherenkovHits = fRootTrigger->GetNumTubesHit(); + + if ( nTrack == 0 || nRawCherenkovHits == 0 ) { + // No track, no hit, nothing to do + return false; + } + + // Get number of hits + //startingCherenkovHitID = digithit_pmtId.size(); + nDigitizedCherenkovHits = fRootTrigger->GetNcherenkovdigihits(); + + if (nDigitizedCherenkovHits == 0){ + continue; + } + //fTriggerTime = fRootTrigger->GetTriggerInfo()[2] - fRootTrigger->GetTriggerInfo()[1]; + ROOT::Math::XYZVector Truepos; + // Loop on vertex + for(int iVertex = 0; iVertex < nVertex; iVertex++){ +#ifdef OLD_WCSIM + Truepos.SetXYZ(fRootTrigger->GetVtx(0),fRootTrigger->GetVtx(1),fRootTrigger->GetVtx(2)); +#else + Truepos.SetXYZ(fRootTrigger->GetVtxs(iVertex,0),fRootTrigger->GetVtxs(iVertex,1),fRootTrigger->GetVtxs(iVertex,2)); +#endif + if ( iVertex*2 > nTrack ) { + std::cout << "ERROR: Vertex and Track number incompatible (nVertex: " << nVertex << " ; nTrack " << nTrack << ") " << std::endl; + return 0; + } + + // Beam info are registered in the Track + TObject * Track = (fRootTrigger->GetTracks())->At(iVertex*2); + WCSimRootTrack *wcTrack = dynamic_cast(Track); + + if (wcTrack->GetParentId() == 0 && wcTrack->GetIpnu()==11) { + particleDir.SetXYZ(wcTrack->GetPdir(0),wcTrack->GetPdir(1),wcTrack->GetPdir(2)); + particleDir = particleDir.Unit(); + } + } + + if ( nRawCherenkovHits < 1 ) return false; + + std::vector times; + + // Get number of hit + rawhit_num = nRawCherenkovHits; + // Loop on Digitized Hit + lf_Dir.SetXYZ(0,0,0); + double AllQ = 0; + for(int iDigitHit = 0; iDigitHit < nDigitizedCherenkovHits; iDigitHit++){ + ROOT::Math::XYZVector lf_particleRelativePMTpos; + ROOT::Math::XYZVector particleRelativePMTpos; + ROOT::Math::XYZVector lf_relativePMTpos; + TObject *Hit = (fRootTrigger->GetCherenkovDigiHits())->At(iDigitHit); + WCSimRootCherenkovDigiHit *wcDigitHit = dynamic_cast(Hit); + + int pmtId = wcDigitHit->GetTubeId(); + + //double HitT = wcDigitHit->GetT() + fTriggerTime; + double HitQ = wcDigitHit->GetQ(); + int peForTube = wcDigitHit->GetQ(); + + + + WCSimRootPMT pmt; + pmt = fGeometry->GetPMT(pmtId - 1, false); + ROOT::Math::XYZVector PMTpos(pmt.GetPosition(0),pmt.GetPosition(1),pmt.GetPosition(2)); + particleRelativePMTpos = (PMTpos - Truepos); + lf_relativePMTpos = (PMTpos - ROOT::Math::XYZVector(leaf_output.vtx.X(), leaf_output.vtx.Y(), leaf_output.vtx.Z())); + ROOT::Math::XYZVector vDir(particleRelativePMTpos.Unit()); + ROOT::Math::XYZVector lfHitDir(lf_relativePMTpos.Unit()); + + lf_Dir += (lfHitDir * HitQ); + + AllQ += HitQ; + + double hitlfRelativeAngle = vDir.Dot(leaf_output.dir); + //std::cout << TMath::ACos(hitlfRelativeAngle)*180./TMath::Pi() << std::endl; + + lf_relativeAngle.push_back((TMath::ACos(hitlfRelativeAngle))*180./TMath::Pi()); + + //std::cout << "Real Relative Angle : " << hitRelativeAngle*180./TMath::Pi() << std::endl; + //std::cout << "Charge of hit : " << HitQ << std::endl; + lf_hRelativeAngle->Fill((TMath::ACos(hitlfRelativeAngle))*180./TMath::Pi(), peForTube); + lf_hRelativeAngleCos->Fill(hitlfRelativeAngle, peForTube); + } + } + return true; + +} +bool AnalyseEvent(WCSimRootEvent * tEvent, PMTType pmtType, int i) { + + int nVertex = 0; // Number of Vertex in event + int nTrack = 0; // Number of Track in event + int nRawCherenkovHits = 0; // Number of Raw Cherenkov hits + int startingCherenkovHitID = 0; // starting ID of Digitized Cherenkov hits. Usually starts at 0, but if there are two types of PMTs, it does not. + int nDigitizedCherenkovHits = 0; // Number of Digitized Cherenkov hits + double fTriggerTime = 0.; + + // Declare some useful variables + WCSimRootTrigger * fRootTrigger; + //TClonesArray * fTimeArray; + + // Currently only one Trigger is used (to be check) + //int iTrig = 0; + for(int iTrig = 0; iTrig < tEvent->GetNumberOfEvents(); iTrig++){ + + triggerId = iTrig; + fRootTrigger = tEvent->GetTrigger(iTrig); + + // Grab the big arrays of times and parent IDs + //fTimeArray = fRootTrigger->GetCherenkovHitTimes(); + + // Get number of vertex and tracks +#ifdef OLD_WCSIM + nVertex = 1; +#else + nVertex = fRootTrigger->GetNvtxs(); +#endif + + nTrack = fRootTrigger->GetNtrack(); + nRawCherenkovHits = fRootTrigger->GetNumTubesHit(); + + if ( nTrack == 0 || nRawCherenkovHits == 0 ) { + // No track, no hit, nothing to do + return false; + } + + // Get number of hits + + startingCherenkovHitID = digithit_pmtId.size(); + nDigitizedCherenkovHits = fRootTrigger->GetNcherenkovdigihits(); + + if (nDigitizedCherenkovHits == 0){ + continue; + } + fTriggerTime = fRootTrigger->GetTriggerInfo()[2] - fRootTrigger->GetTriggerInfo()[1]; + ROOT::Math::XYZVector Truepos; + // Loop on vertex + for(int iVertex = 0; iVertex < nVertex; iVertex++){ + // Beam info are registered in the Track + TObject * Track = (fRootTrigger->GetTracks())->At(iVertex*2); + WCSimRootTrack *wcTrack = dynamic_cast(Track); + +#ifdef OLD_WCSIM + Truepos.SetXYZ(fRootTrigger->GetVtx(0), fRootTrigger->GetVtx(1), fRootTrigger->GetVtx(2)); + ROOT::Math::XYZTVector true_origin(fRootTrigger->GetVtx(0), fRootTrigger->GetVtx(1), fRootTrigger->GetVtx(2), wcTrack->GetTime()); + true_origins.push_back(true_origin); + +#else + Truepos.SetXYZ(fRootTrigger->GetVtxs(iVertex,0), fRootTrigger->GetVtxs(iVertex,1), fRootTrigger->GetVtxs(iVertex,2)); + ROOT::Math::XYZTVector true_origin(fRootTrigger->GetVtxs(iVertex,0), fRootTrigger->GetVtxs(iVertex,1), fRootTrigger->GetVtxs(iVertex,2), wcTrack->GetTime()); + true_origins.push_back(true_origin); +#endif + if ( iVertex*2 > nTrack ) { + std::cout << "ERROR: Vertex and Track number incompatible (nVertex: " << nVertex << " ; nTrack " << nTrack << ") " << std::endl; + return 0; + } + + + if (wcTrack->GetParentId() == 0 && wcTrack->GetIpnu()==11) { + particleDir.SetXYZ(wcTrack->GetPdir(0),wcTrack->GetPdir(1),wcTrack->GetPdir(2)); + particleDir = particleDir.Unit(); + } + + leaf_output_ana.Wall = calculateToWall(R, h, Truepos, particleDir); + dWall = calculateDWall(R, h, Truepos); + std::cout << "true dir: " << particleDir.X() << "," << particleDir.Y() << "," << particleDir.Z() << std::endl; + std::cout << "true pos: " << Truepos.X() << ","<< Truepos.Y() << ","<< Truepos.Z() << std::endl; + //std::cout << "ToWall: " << leaf_output_ana.Wall << std::endl; + true_particleId.push_back( wcTrack->GetIpnu() ); + true_energies.push_back( wcTrack->GetE() ); + + } + + // Send True Position to LEAF for check + LEAF::GetME()->SetTrueVertexInfo(true_origins[0]); + + if ( nRawCherenkovHits < 1 ) return false; + + std::vector times; + + // Get number of hit + rawhit_num = nRawCherenkovHits; + // Loop on Digitized Hit + lf_Dir.SetXYZ(0,0,0); + double AllQ = 0; + + + + for(int iDigitHit = 0; iDigitHit < nDigitizedCherenkovHits; iDigitHit++){ + ROOT::Math::XYZVector lf_particleRelativePMTpos; + ROOT::Math::XYZVector particleRelativePMTpos; + ROOT::Math::XYZVector lf_relativePMTpos; + TObject *Hit = (fRootTrigger->GetCherenkovDigiHits())->At(iDigitHit); + WCSimRootCherenkovDigiHit *wcDigitHit = dynamic_cast(Hit); + + int pmtId = wcDigitHit->GetTubeId(); + + double HitT = wcDigitHit->GetT() + fTriggerTime; + double HitQ = wcDigitHit->GetQ(); + int peForTube = wcDigitHit->GetQ(); + AllQ += HitQ; + digithit_pmtId.push_back(pmtId); + digithit_T.push_back(HitT); + digithit_Q.push_back(HitQ); + Charge_PMT.push_back(peForTube); + + Sum_HitQ->Fill(leaf_output_ana.Wall, HitQ); + Count_Hits->Fill(leaf_output_ana.Wall, 1); + + Sum_HitQ_hdWall->Fill(dWall, HitQ); + Count_Hits_hdWall->Fill(dWall, 1); + + WCSimRootPMT pmt; + pmt = fGeometry->GetPMT(pmtId - 1, false); + + ROOT::Math::XYZVector PMTpos, PMTOrientation, vDir, lfHitDir; + double normPMTOrientation = 0, NormvDir = 0, NormlfHitDir = 0; + + PMTpos.SetXYZ(pmt.GetPosition(0), pmt.GetPosition(1), pmt.GetPosition(2)); + + PMTOrientation.SetXYZ(pmt.GetOrientation(0), pmt.GetOrientation(1), pmt.GetOrientation(2)); + PMTOrientation = PMTOrientation.Unit(); + + vDir = (PMTpos - Truepos).Unit(); + lfHitDir = (PMTpos - ROOT::Math::XYZVector(leaf_output.vtx.X(),leaf_output.vtx.Y(),leaf_output.vtx.Z())).Unit(); + + normPMTOrientation = sqrt(normPMTOrientation); + NormvDir = sqrt(NormvDir); + NormlfHitDir = sqrt(NormlfHitDir); + + double cosPMThitAngle = vDir.Dot(PMTOrientation); + + double PMThitAngle = acos(-cosPMThitAngle) * (180.0 / M_PI); + //double cos2PMThitAngle = fabs(2 * cosPMThitAngle * cosPMThitAngle - 1); + //double Ftheta = 0.205 + 0.524*cos2PMThitAngle + 0.390*(pow(cos2PMThitAngle,2)) - 0.132*(pow(cos2PMThitAngle,3)); + + //std::cout << TMath::ACos(hitlfRelativeAngle)*180./TMath::Pi() << std::endl; + double hitRelativeAngle = vDir.Dot(particleDir); + + relativeAngle.push_back((TMath::ACos(hitRelativeAngle))*180./TMath::Pi()); + digithit_dark.push_back(false); + digithit_Type.push_back(pmtType); + //std::cout << "Real Relative Angle : " << hitRelativeAngle*180./TMath::Pi() << std::endl; + //std::cout << "Charge of hit : " << HitQ << std::endl; + hRelativeAngle->Fill((TMath::ACos(hitRelativeAngle))*180./TMath::Pi(), peForTube); + hRelativeAngleCos->Fill(hitRelativeAngle, peForTube); + + + + /*WallPMTAngle_Charge->Fill(leaf_output_ana.Wall, PMThitAngle*2, HitQ)*/ + + hToWall_RelAngle_Charge->Fill(leaf_output_ana.Wall, (TMath::ACos(hitRelativeAngle))*180./TMath::Pi(), HitQ); + pAngle_ToWall->Fill((TMath::ACos(hitRelativeAngle))*180./TMath::Pi(),leaf_output_ana.Wall); + hAngle_HitQ->Fill(PMThitAngle, HitQ); + + } + + + std::cout << "Hits: " << nDigitizedCherenkovHits << std::endl; + lf_Dir /= AllQ; + lf_Dir = lf_Dir.Unit(); + + for (int bin = 1; bin <= ToWall_Charge->GetNbinsX(); ++bin) { + double sum = Sum_HitQ->GetBinContent(bin); + double count = Count_Hits->GetBinContent(bin); + + if (count > 0) { + ToWall_Charge->SetBinContent(bin, sum / count); + } else { + ToWall_Charge->SetBinContent(bin, 0); + } + } + ToWall_Charge_2D->Fill(leaf_output_ana.Wall, AllQ); + + for (int bin = 1; bin <= hdWall_TotalCharge_hist->GetNbinsX(); ++bin) { + double sum = Sum_HitQ_hdWall->GetBinContent(bin); + double count = Count_Hits_hdWall->GetBinContent(bin); + + if (count > 0) { + hdWall_TotalCharge_hist->SetBinContent(bin, sum / count); + } else { + hdWall_TotalCharge_hist->SetBinContent(bin, 0); + } + } + + hdWall_TotalCharge->Fill(dWall, AllQ); + hdWall_TotalCharge_corr->Fill(dWall, leaf_output.energy); + + /*double lf_rot_dir[3]; + lf_rot_dir[0]=lf_Dir[2]; + lf_rot_dir[1]=-lf_Dir[0]; + lf_rot_dir[2]=-lf_Dir[1]; + for(int j=0;j<3;j++){ + lf_Dir[j] = lf_rot_dir[j]; + } + lf_dir_res = sqrt(pow(lf_Dir[0] - particleDir[0], 2) + pow(lf_Dir[1] - particleDir[1], 2) + pow(lf_Dir[2] - particleDir[2], 2));*/ + + int iIdx_BS = 0; + + nDigitizedCherenkovHits = digithit_pmtId.size(); + digithit_num = nDigitizedCherenkovHits; + + // Feed fitter + if ( pmtType == PMTType::kID ) { + for(int iDigitHit = startingCherenkovHitID; iDigitHit < nDigitizedCherenkovHits; iDigitHit++){ + times.push_back(digithit_T[iDigitHit]); + + //std::cout << " Add Hit with " << digithit_pmtId[iDigitHit] << " pmtType: " << pmtType << std::endl; + if ( digithit_pmtId[iDigitHit] <= 0 ) std::cout << " Weird PMT ID " << digithit_pmtId[iDigitHit] << std::endl; + HKManager::GetME()->AddHit_ID(digithit_T[iDigitHit],digithit_Q[iDigitHit],digithit_pmtId[iDigitHit]); + + // Bonsai (do not store mPMT hits) + if ( iIdx_BS < 2000 && digithit_T[iDigitHit] < 4000. && digithit_T[iDigitHit] > -4000. ) { + //std::cout << pmtId << " " << HitT << std::endl; + bsCAB[iIdx_BS] = digithit_pmtId[iDigitHit]; + bsT [iIdx_BS] = digithit_T[iDigitHit]; // shift BS time is needed if interaction time is < 0, this needs to be considered + bsQ [iIdx_BS] = digithit_Q[iDigitHit]; + + iIdx_BS += 1; + } + } + } + else { + for(int iDigitHit = startingCherenkovHitID; iDigitHit < nDigitizedCherenkovHits; iDigitHit++){ + times.push_back(digithit_T[iDigitHit]); + //std::cout << " Add Hit with " << digithit_pmtId[iDigitHit] << " pmtType: " << pmtType << std::endl; + if ( digithit_pmtId[iDigitHit] <= 0 ) std::cout << " Weird PMT ID " << digithit_pmtId[iDigitHit] << std::endl; + HKManager::GetME()->AddHit_mPMT(digithit_T[iDigitHit],digithit_Q[iDigitHit],digithit_pmtId[iDigitHit]); + } + } + + // Bonsai + bsnhit[0] = iIdx_BS; + fHit = digithit_pmtId.size(); + + // Compute hits + std::vector Hit_time_50; + std::vector Hit_time_200; + std::vector Hit_time_400; + + std::sort(times.begin(),times.end()); + for(int iDigitHit = 0; iDigitHit < nDigitizedCherenkovHits; iDigitHit++){ + + double HitT = times[iDigitHit]; + + Hit_time_50.push_back(HitT); + Hit_time_200.push_back(HitT); + Hit_time_400.push_back(HitT); + + // Count hit in 50 ns window + while ( HitT - Hit_time_50[0] > 50. ) + Hit_time_50.erase(Hit_time_50.begin()); + + // Count hit in 200 ns window + while ( HitT - Hit_time_200[0] > 200. ) + Hit_time_200.erase(Hit_time_200.begin()); + + + // Count hit in 400 ns window + while ( HitT - Hit_time_400[0] > 400. ) + Hit_time_400.erase(Hit_time_400.begin()); + + if ( (unsigned int) fHit_50 < Hit_time_50.size() ) fHit_50 = Hit_time_50.size(); + if ( (unsigned int) fHit_200 < Hit_time_200.size() ) fHit_200 = Hit_time_200.size(); + if ( (unsigned int) fHit_400 < Hit_time_400.size() ) fHit_400 = Hit_time_400.size(); + } + + } + + + return true; +} \ No newline at end of file diff --git a/example/env.sh b/example/env.sh new file mode 100644 index 0000000..d388c56 --- /dev/null +++ b/example/env.sh @@ -0,0 +1,6 @@ +export LEAFDIR=../ +export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$LEAFDIR/lib +export ROOT_INCLUDE_PATH=${LEAFDIR}/leaf/DataModel-lite + +export WCSIMDIR=${LEAFDIR}/libWCSIM +export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$WCSIMDIR \ No newline at end of file diff --git a/example/test_example.sh b/example/test_example.sh index 9a21b8b..63c37b2 100755 --- a/example/test_example.sh +++ b/example/test_example.sh @@ -1,16 +1,38 @@ #!/bin/bash +# Launch single job +# +#SBATCH --job-name=leaf # Job name +#SBATCH --output=/sps/t2k/lperisse/Soft/leaf/logs/analysis_%j.log # Standard output and error log +#SBATCH --partition=htc # Partition choice +#SBATCH --ntasks=1 # Maximum number of parallel processes +#SBATCH --mem=5G # Amount of memory required +#SBATCH --time=0-01:00 # 7 days by default on htc partition + + # Replace $input contents with the path toward your WCSim input file # (Note: make sure the WCSim path in RunAtStart matches the WCSim version you used to produce the file) -input=$WCSIMDIR/wcsim_hybrid.root +input=/sps/t2k/lperisse/Soft/wcsim/results/electron/wcsim1p12p29_electron_HK_test.root +output=/sps/t2k/lperisse/Soft/wcsim/results/electron/wcsim1p12p29_electron_HK_test.leaf.root + -# test.root doesn't exist, this is just a placeholder +# Argument syntaxe to run ./analysis: +#-i Input WCSim ROOT file +#-o Output ROOT file +# d Dark noise frequency of PMTs in kHz +# h Dark noise frequency of mPMTS in kHz +# s First event to analyze +#-e Last event to analyze +# v VERBOSE option if [ ! -f "$input" ]; then echo "Set input file doesn't exist! Add your WCSim input file here. (Note: make sure the WCSim path in RunAtStart matches the WCSim version you used to produce the file)" else echo "Processing ${input}..." - ./analysis $input output.root 0 0 + cd ${LEAFDIR}/example + ./analysis -i $input -o $output -s 0 -e 5 -d 4.2 -h 0.0 -v fi + + diff --git a/inputs/.gitignore b/inputs/.gitignore new file mode 100644 index 0000000..e69de29 diff --git a/inputs/PDF_electron_upto100MeV_uniform_isotropic_10k_withDN.root b/inputs/PDF_electron_upto100MeV_uniform_isotropic_10k_withDN.root new file mode 100644 index 0000000..c834078 Binary files /dev/null and b/inputs/PDF_electron_upto100MeV_uniform_isotropic_10k_withDN.root differ diff --git a/inputs/timePDF_DRnew_Large.root b/inputs/timePDF_DRnew_Large.root deleted file mode 100644 index 0b1b78c..0000000 Binary files a/inputs/timePDF_DRnew_Large.root and /dev/null differ diff --git a/inputs/timePDF_Directionality.root b/inputs/timePDF_Directionality.root deleted file mode 100644 index 2baa242..0000000 Binary files a/inputs/timePDF_Directionality.root and /dev/null differ diff --git a/inputs/timePDF_Directionality_DRnew.root b/inputs/timePDF_Directionality_DRnew.root deleted file mode 100644 index 0289350..0000000 Binary files a/inputs/timePDF_Directionality_DRnew.root and /dev/null differ diff --git a/inputs/timePDF_HE.root b/inputs/timePDF_HE.root deleted file mode 100644 index f81ca2c..0000000 Binary files a/inputs/timePDF_HE.root and /dev/null differ diff --git a/leaf/DataModel-lite/Constants/Mass.hpp b/leaf/DataModel-lite/Constants/Mass.hpp new file mode 100644 index 0000000..35a1878 --- /dev/null +++ b/leaf/DataModel-lite/Constants/Mass.hpp @@ -0,0 +1,13 @@ +#ifndef MASS_HPP +#define MASS_HPP + +#include + +#include "Constants/Units.hpp" + +static const std::map PDGParticleMasses { + { 11, 0.510998910 * MeV}, // electron + {2212, 938.272013 * MeV}, // proton + {2112, 939.56536 * MeV} // neutron +}; +#endif diff --git a/leaf/DataModel-lite/Constants/Suffix.hpp b/leaf/DataModel-lite/Constants/Suffix.hpp new file mode 100644 index 0000000..bf639a0 --- /dev/null +++ b/leaf/DataModel-lite/Constants/Suffix.hpp @@ -0,0 +1,21 @@ +#ifndef SUFFIX_HPP +#define SUFFIX_HPP + +#include + +// PMT collections + +namespace HKSuffix { +#if HK_USE_ROOT7 + inline const std::string ID = "ID"; + inline const std::string OD = "OD"; + inline const std::string mPMT = "mPMT"; +#else + const std::string ID = "ID"; + const std::string OD = "OD"; + const std::string mPMT = "mPMT"; +#endif + +} // namespace HKSuffix + +#endif diff --git a/leaf/DataModel-lite/Constants/Units.hpp b/leaf/DataModel-lite/Constants/Units.hpp new file mode 100644 index 0000000..bd05ee2 --- /dev/null +++ b/leaf/DataModel-lite/Constants/Units.hpp @@ -0,0 +1,23 @@ +#ifndef UNITS_HPP +#define UNITS_HPP + +// ENERGY +static constexpr double MeV = 1.; +static constexpr double eV = MeV * 1E-3; +static constexpr double GeV = MeV * 1E3; +static constexpr double TeV = MeV * 1E6; + +// TIME +static constexpr double ns = 1.; +static constexpr double ps = ns * 1E-3; +static constexpr double us = ns * 1E3; +static constexpr double ms = ns * 1E6; +static constexpr double s = ns * 1E9; + +// DISTANCE +static constexpr double mm = 1.; +static constexpr double cm = mm * 1E1; +static constexpr double m = mm * 1E3; +static constexpr double km = mm * 1E6; + +#endif diff --git a/leaf/DataModel-lite/Constants/mPMTs.hpp b/leaf/DataModel-lite/Constants/mPMTs.hpp new file mode 100644 index 0000000..aaebe97 --- /dev/null +++ b/leaf/DataModel-lite/Constants/mPMTs.hpp @@ -0,0 +1,19 @@ +#ifndef MPMTS_HPP +#define MPMTS_HPP + +/************************************************ + * Constant related to mPMTs enum class definition + ***********************************************/ + +namespace mPMTConstants { +#if HK_USE_ROOT7 + inline const int kNGroups = 3; + inline const int kNPMTs = 19; //Number of 3" PMT per mPMT +#else + const int kNGroups = 3; + const int kNPMTs = 19; //Number of 3" PMT per mPMT +#endif + +} // namespace mPMTConstants + +#endif \ No newline at end of file diff --git a/leaf/DataModel-lite/CreateDataModel-lite.sh b/leaf/DataModel-lite/CreateDataModel-lite.sh deleted file mode 100755 index e645a0b..0000000 --- a/leaf/DataModel-lite/CreateDataModel-lite.sh +++ /dev/null @@ -1,17 +0,0 @@ -#!/bin/bash -#Script is for GitHub management, to make DataModel available for non-HK members - -if [ "$HK_ASTROANALYSIS_DIR" == "" ] || [ -z $HK_ASTROANALYSIS_DIR ]; then - echo "HK_ASTROANALYSIS_DIR is not defined or doesn't exist ($HK_ASTROANALYSIS_DIR). We can't create DataModel-lite." -else - echo "HK_ASTROANALYSIS_DIR was found to be: $HK_ASTROANALYSIS_DIR . Create DataModel-lite" - rsync -av --exclude=*.o $HK_ASTROANALYSIS_DIR/DataModelRoot/Geometry.* . - rsync -av --exclude=*.o $HK_ASTROANALYSIS_DIR/DataModelRoot/GeoTools.* . - rsync -av --exclude=*.o $HK_ASTROANALYSIS_DIR/DataModelRoot/HitInfo.* . - rsync -av --exclude=*.o $HK_ASTROANALYSIS_DIR/DataModelRoot/HitCollection.* . - rsync -av --exclude=*.o $HK_ASTROANALYSIS_DIR/DataModelRoot/PMTInfo.* . - rsync -av --exclude=*.o $HK_ASTROANALYSIS_DIR/DataModelRoot/TimeDelta.* . - rsync -av --exclude=*.o $HK_ASTROANALYSIS_DIR/DataModelRoot/Pos3DT.* . - echo "You should update manually Environments.h" -fi - diff --git a/leaf/DataModel-lite/Enums/DetectorID.hpp b/leaf/DataModel-lite/Enums/DetectorID.hpp new file mode 100644 index 0000000..bb611fa --- /dev/null +++ b/leaf/DataModel-lite/Enums/DetectorID.hpp @@ -0,0 +1,9 @@ +#ifndef DETECTOR_ID_HPP +#define DETECTOR_ID_HPP + +/************************************************ + * DetectorID enum class definition + ***********************************************/ +enum class DetectorID { kUndefined = -1, kFD, kIWCD, kNumDetectorIDs }; + +#endif diff --git a/leaf/DataModel-lite/Enums/PMTType.hpp b/leaf/DataModel-lite/Enums/PMTType.hpp new file mode 100644 index 0000000..9f5d42f --- /dev/null +++ b/leaf/DataModel-lite/Enums/PMTType.hpp @@ -0,0 +1,9 @@ +#ifndef PMT_TYPE_HPP +#define PMT_TYPE_HPP + +/************************************************ + * PMTType enum class definition + ***********************************************/ +enum class PMTType { kUndefined = -1, kID, kOD, kmPMT, kNumPMTTypes }; + +#endif \ No newline at end of file diff --git a/leaf/DataModel-lite/Environments.h b/leaf/DataModel-lite/Environments.h deleted file mode 100644 index 6aa1202..0000000 --- a/leaf/DataModel-lite/Environments.h +++ /dev/null @@ -1,112 +0,0 @@ -#ifndef Environments_H -#define Environments_H - -#include -#include -#include -#include - -/** -* \namespace Environments - * - * This namespace provides constants and enums for HK AstroAnalysis - * - * - * $Author: G.Pronost $ - * $Date: 2020/11/13 15:16:00 $ - * Contact: pronost@km.icrr.u-tokyo.ac.jp - * - */ - -#define mPMT_ID_SHIFT 100000 -#define PMT_OD_SHIFT 50000 - - // Usefull processor functions: -#define Astro_GetPMTType(x) x>=mPMT_ID_SHIFT?1:0 - -#define Astro_GetIfODPMT(x) (x>=PMT_OD_SHIFT &&x=mPMT_ID_SHIFT )?1:0 -#define Astro_GetIfBnLPMT(x) ( x - -Geometry::Geometry(){ - - // Initialize - - detector_length = 0.; - detector_radius = 0.; - - HasOD = false; - - pmt_radius .resize(HKAA::kPMTTypeMax, 0.); - pmt_num .resize(HKAA::kPMTTypeMax, 0 ); - pmt_num_original .resize(HKAA::kPMTTypeMax, 0 ); - pmt_dark_rate .resize(HKAA::kPMTTypeMax, 0.); - pmt_first .resize(HKAA::kPMTTypeMax, HKAA::kMaxPMT); - - PMTList.resize(HKAA::kMaxPMT); -} - - -Geometry::~Geometry(){ - - PMTList.clear(); -} - - -void Geometry::AddPMTInfo(PMTInfo lInfo) { - - lInfo.Id_original = lInfo.Id; - lInfo.Masked = false; - - // Add Id shift: - lInfo.Id += HKAA::kPMTId_Shift[lInfo.Type];; - - if ( lInfo.Type == HKAA::kIDPMT_3inch ) { - // Setup mPMT variables - lInfo.Setup_mPMT(); - } - if ( pmt_first[lInfo.Type] > lInfo.Id ) { - pmt_first[lInfo.Type] = lInfo.Id; - } - - - PMTList[lInfo.Id] = lInfo; -} - -void Geometry::Finalise() { - - this->Setup_mPMTs(); - - std::cout << "[DEBUG] PMT[" << HKAA::kIDPMT_BnL << "] first Id is: " << pmt_first[HKAA::kIDPMT_BnL ] << ", Id shift is: " << HKAA::kPMTId_Shift[HKAA::kIDPMT_BnL ] << std::endl; - std::cout << "[DEBUG] PMT[" << HKAA::kIDPMT_3inch << "] first Id is: " << pmt_first[HKAA::kIDPMT_3inch ] << ", Id shift is: " << HKAA::kPMTId_Shift[HKAA::kIDPMT_3inch] << std::endl; - std::cout << "[DEBUG] PMT[" << HKAA::kODPMT_3inch << "] first Id is: " << pmt_first[HKAA::kODPMT_3inch ] << ", Id shift is: " << HKAA::kPMTId_Shift[HKAA::kODPMT_3inch] << std::endl; -} - -void Geometry::Setup_mPMTs() { - - for ( int i=HKAA::kmPMT_Shift; i < (HKAA::kmPMT_Shift + pmt_num[HKAA::kIDPMT_3inch]); i++ ) { - - PMTInfo lInfo = PMTList[i]; - PMTInfo lInfoTop = PMTList[lInfo.mPMT_RefTube]; - - lInfo.Setup_mPMT_Referencial(lInfoTop); - } -} diff --git a/leaf/DataModel-lite/Geometry.h b/leaf/DataModel-lite/Geometry.h deleted file mode 100644 index 5a0dd6d..0000000 --- a/leaf/DataModel-lite/Geometry.h +++ /dev/null @@ -1,65 +0,0 @@ -#ifndef Geometry_H -#define Geometry_H - -#include -#include -#include - -#include "Environments.h" -#include "PMTInfo.h" - -/** -* \class Geometry - * - * This class holds Tank Geometry informations - * - * $Author: G.Pronost $ - * $Date: 2020/11/16 09:37:00 $ - * Contact: pronost@km.icrr.u-tokyo.ac.jp - * - */ - -class Geometry { - - friend class WCSimReader; - friend class PMTMask; - - public: - Geometry(); ///< Simple constructor - ~Geometry(); - - // WCSim geometry information - - double detector_length; - double detector_radius; - - bool HasOD; - - // PMT Info - std::vector pmt_radius; - std::vector pmt_num; - std::vector pmt_num_original; - std::vector pmt_dark_rate; - std::vector pmt_first; - - std::vector PMTList; - - const std::vector *GetPMTList() const { return &PMTList; } - - PMTInfo GetPMT(int lChannel) const { return PMTList[lChannel]; } - bool GetIfMasked(int lChannel) const { return PMTList[lChannel].Masked; } - - private: - // Function to properly set PMTInfo - void AddPMTInfo(PMTInfo lInfo); - - void Finalise(); - // Start making referencial for mPMTs - void Setup_mPMTs(); - - // Mask PMT - void Mask_PMT(int lChannel) { PMTList[lChannel].Masked = true; } - void UnMask_PMT(int lChannel) { PMTList[lChannel].Masked = false; } -}; - -#endif diff --git a/leaf/DataModel-lite/HKDarkNoise.cpp b/leaf/DataModel-lite/HKDarkNoise.cpp new file mode 100644 index 0000000..a3b20f1 --- /dev/null +++ b/leaf/DataModel-lite/HKDarkNoise.cpp @@ -0,0 +1,8 @@ +#include "HKDarkNoise.hpp" + +HKDarkNoise::HKDarkNoise() {} + +HKDarkNoise::HKDarkNoise(HKDarkNoise* /*in*/) {} + +// pure virtual destructor needs implementation +HKDarkNoise::~HKDarkNoise() {} diff --git a/leaf/DataModel-lite/HKDarkNoise.hpp b/leaf/DataModel-lite/HKDarkNoise.hpp new file mode 100644 index 0000000..03e8cbd --- /dev/null +++ b/leaf/DataModel-lite/HKDarkNoise.hpp @@ -0,0 +1,85 @@ +#ifndef HK_DARKNOISE_HPP +#define HK_DARKNOISE_HPP + +/************************************************* + * HKDarkNoise.hpp + * - Author: + * - Date: 1970/01/01 + ************************************************/ + +#include +#include + +#include "HKObject.hpp" +#include "Enums/PMTType.hpp" + +/************************************************ + * HKDarkNoise class definition + ***********************************************/ +class HKDarkNoise : public HKObject { + public: + + HKDarkNoise(); //!< Constructor + + //! Create HKDarkNoise from an HKDarkNoise pointor + /*! + \param in HKDarkNoise* + */ + HKDarkNoise(HKDarkNoise* in); + + virtual ~HKDarkNoise() = 0; //!< Destructor + virtual void Reset() = 0; //!< Reset function, called after each event + virtual unsigned int GetVersion() const = 0; //!< Return the class version number + + static std::string GetBaseNameStatic() { + return "HKDarkNoise"; + } //!< Static function. Return the class base name + + std::string GetBaseName() const { + return HKDarkNoise::GetBaseNameStatic(); + } //!< Return the class base name + + // v1 functions + + //! Return average total number of dark noise hit in 1ns for a given PMT type + //! Note: the value is ~ N_PMT * darknoise_per_PMT * 1e-9 + /*! + \param in PMTType. PMT type (ID, OD, mPMT) + */ + virtual double GetAverageTotalDarkNoisePerNS(PMTType in) const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0.0); + } + + //! Set average total number of dark noise hit in 1ns for a given PMT type + //! Note: the value is ~ N_PMT * darknoise_per_PMT * 1e-9 + /*! + \param in PMTType. PMT type (ID, OD, mPMT) + \param noise double. Average total dark noise hit + */ + virtual bool SetAverageTotalDarkNoisePerNS(PMTType in, double noise) { + return false; + } + + //! Return dark noise rate for a given PMT in Hz + /*! + \param in PMTType. PMT type (ID, OD, mPMT) + \param num int. Channel number + */ + virtual double GetDarkNoiseHz(PMTType in, int num) const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0.0); + } + + //! Set dark noise value for a given PMT type in Hz + /*! + \param in PMTType. PMT type (ID, OD, mPMT) + \param num int. Channel number + \param noise double. Dark noise value in Hz + */ + virtual bool SetDarkNoiseHz(PMTType in, int num, double noise) { + return false; + } + + ClassDef(HKDarkNoise, 1); //!< ROOT Class definition +}; + +#endif diff --git a/leaf/DataModel-lite/HKDarkNoiseV1.cpp b/leaf/DataModel-lite/HKDarkNoiseV1.cpp new file mode 100644 index 0000000..30393e9 --- /dev/null +++ b/leaf/DataModel-lite/HKDarkNoiseV1.cpp @@ -0,0 +1,25 @@ +#include "HKDarkNoiseV1.hpp" + +HKDarkNoiseV1::HKDarkNoiseV1() { + this->Reset(); +} + +HKDarkNoiseV1::HKDarkNoiseV1(HKDarkNoise* in) { + // Load variable here: + + // v1 copies + // m_a = in->GetA(); +} + +HKDarkNoiseV1::~HKDarkNoiseV1() { + this->Reset(); +} + +void HKDarkNoiseV1::Reset() { + // Initialize variable here: + // Note, reset is called for every new event, so variables needing to be initialized + // once for a full file should be defined in the creator + + // v1 resets + // m_a = 0; +} diff --git a/leaf/DataModel-lite/HKDarkNoiseV1.hpp b/leaf/DataModel-lite/HKDarkNoiseV1.hpp new file mode 100644 index 0000000..1cecd66 --- /dev/null +++ b/leaf/DataModel-lite/HKDarkNoiseV1.hpp @@ -0,0 +1,91 @@ +#ifndef HK_DARKNOISE_V1_HPP +#define HK_DARKNOISE_V1_HPP + +/************************************************* + * HKDarkNoiseV1.hpp + * - Author: + * - Date: 1970/01/01 + ************************************************/ + +#include "HKDarkNoise.hpp" + +/************************************************ + * HKDarkNoiseV1 class definition + * Version 1 of HKDarkNoise + ***********************************************/ +class HKDarkNoiseV1 : public HKDarkNoise { + public: + + HKDarkNoiseV1(); //!< Constructor + + //! Create HKDarkNoiseV1 from an HKDarkNoise pointor + /*! + \param in HKDarkNoise* + */ + HKDarkNoiseV1(HKDarkNoise* in); + + ~HKDarkNoiseV1() override; //!< Destructor + void Reset() override; //!< Reset function, called after each event. Use it to initialize variable + + static constexpr unsigned int GetVersionStatic() { + return 1; + } //!< Static function. Return the class version number + + unsigned int GetVersion() const override { + return HKDarkNoiseV1::GetVersionStatic(); + } //!< Return the class version number + + // v1 functions + + //! Return average total number of dark noise hit in 1ns for a given PMT type + //! Note: the value is ~ N_PMT * darknoise_per_PMT * 1e-9 + /*! + \param in PMTType. PMT type (ID, OD, mPMT) + */ + double GetAverageTotalDarkNoisePerNS(PMTType in) const override { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0.0); + } + + //! Set average total number of dark noise hit in 1ns for a given PMT type + //! Note: the value is ~ N_PMT * darknoise_per_PMT * 1e-9 + /*! + \param in PMTType. PMT type (ID, OD, mPMT) + \param noise double. Average total dark noise hit + */ + bool SetAverageTotalDarkNoisePerNS(PMTType in, double noise) override { + return false; + } + + //! Return dark noise value for a given PMT in Hz + /*! + \param in PMTType. PMT type (ID, OD, mPMT) + \param num int. Channel number + */ + double GetDarkNoiseHz(PMTType in, int num) const override { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0.0); + } + + //! Set dark noise value for a given PMT type in Hz + /*! + \param in PMTType. PMT type (ID, OD, mPMT) + \param num int. Channel number + \param noise double. Dark noise value in Hz + */ + bool SetDarkNoiseHz(PMTType in, int num, double noise) override { + return false; + } + + // define function here: + // int GetA() const override { return m_a; } + // bool SetA(int in) override { m_a = in; return true; } + + private: + + // v1 data members + std::array(PMTType::kNumPMTTypes)> m_avg_dark_noise; // Average dark noise for each PMT type (ID, OD, mPMT) + std::array, static_cast(PMTType::kNumPMTTypes)> m_dark_noise; // Dark noise for each PMT (ID, OD, mPMT). + + ClassDefOverride(HKDarkNoiseV1, 1); //!< ROOT Class definition +}; + +#endif diff --git a/leaf/DataModel-lite/HKGeometry.cpp b/leaf/DataModel-lite/HKGeometry.cpp new file mode 100644 index 0000000..d4ac3b4 --- /dev/null +++ b/leaf/DataModel-lite/HKGeometry.cpp @@ -0,0 +1,8 @@ +#include "HKGeometry.hpp" + +HKGeometry::HKGeometry() {} + +HKGeometry::HKGeometry(HKGeometry* /*in*/) {} + +// pure virtual destructor needs implementation +HKGeometry::~HKGeometry() {} diff --git a/leaf/DataModel-lite/HKGeometry.hpp b/leaf/DataModel-lite/HKGeometry.hpp new file mode 100644 index 0000000..01ac0dc --- /dev/null +++ b/leaf/DataModel-lite/HKGeometry.hpp @@ -0,0 +1,227 @@ +#ifndef HK_GEOMETRY_HPP +#define HK_GEOMETRY_HPP + +/************************************************* + * HKGeometry.hpp + * - Author: + * - Date: 1970/01/01 + ************************************************/ + +#include +#include + +#include "Enums/DetectorID.hpp" +#include "HKGeometryPMT.hpp" +#include "HKObject.hpp" + +/************************************************ + * HKGeometry class definition + ***********************************************/ +class HKGeometry : public HKObject { + public: + + HKGeometry(); //!< Constructor + + //! Create HKGeometry from an HKGeometry pointor + /*! + \param in HKGeometry* + */ + HKGeometry(HKGeometry* in); + + virtual ~HKGeometry() = 0; //!< Destructor + virtual void Reset() = 0; //!< Reset function, called after each event + virtual unsigned int GetVersion() const = 0; //!< Return the class version number + + static std::string GetBaseNameStatic() { + return "HKGeometry"; + } //!< Static function. Return the class base name + + std::string GetBaseName() const { + return HKGeometry::GetBaseNameStatic(); + } //!< Return the class base name + + // v1 functions + //! Get IWCDElevatorHeight + /*! + \return float + */ + virtual float GetIWCDElevatorHeight() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set IWCDElevatorHeight + /*! + \param in float + \return `true` if successful, `false` if not + */ + virtual bool SetIWCDElevatorHeight(UNUSED_PARAM float in) { return false; } + + //! Get IsCavernMaterialSimulated + /*! + \return bool + */ + virtual bool GetIsCavernMaterialSimulated() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set IsCavernMaterialSimulated + /*! + \param in bool + \return `true` if successful, `false` if not + */ + virtual bool SetIsCavernMaterialSimulated(UNUSED_PARAM bool in) { return false; } + + //! Get IsODSimulated + /*! + \return bool + */ + virtual bool GetIsODSimulated() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set IsODSimulated + /*! + \param in bool + \return `true` if successful, `false` if not + */ + virtual bool SetIsODSimulated(UNUSED_PARAM bool in) { return false; } + + //! Get IsDeadSpaceSimulated + /*! + \return bool + */ + virtual bool GetIsDeadSpaceSimulated() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set IsDeadSpaceSimulated + /*! + \param in bool + \return `true` if successful, `false` if not + */ + virtual bool SetIsDeadSpaceSimulated(UNUSED_PARAM bool in) { return false; } + + //! Get AreODPMTsActive + /*! + \return bool + */ + virtual bool GetAreODPMTsActive() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set AreODPMTsActive + /*! + \param in bool + \return `true` if successful, `false` if not + */ + virtual bool SetAreODPMTsActive(UNUSED_PARAM bool in) { return false; } + + //! Get OuterTyvekTotalHeight + /*! + \return float + */ + virtual float GetOuterTyvekTotalHeight() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set OuterTyvekTotalHeight + /*! + \param in float + \return `true` if successful, `false` if not + */ + virtual bool SetOuterTyvekTotalHeight(UNUSED_PARAM float in) { return false; } + + //! Get OuterTyvekRadius + /*! + \return float + */ + virtual float GetOuterTyvekRadius() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set OuterTyvekRadius + /*! + \param in float + \return `true` if successful, `false` if not + */ + virtual bool SetOuterTyvekRadius(UNUSED_PARAM float in) { return false; } + + //! Get InnerTyvekTotalHeight + /*! + \return float + */ + virtual float GetInnerTyvekTotalHeight() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set InnerTyvekTotalHeight + /*! + \param in float + \return `true` if successful, `false` if not + */ + virtual bool SetInnerTyvekTotalHeight(UNUSED_PARAM float in) { return false; } + + //! Get InnerTyvekRadius + /*! + \return float + */ + virtual float GetInnerTyvekRadius() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set InnerTyvekRadius + /*! + \param in float + \return `true` if successful, `false` if not + */ + virtual bool SetInnerTyvekRadius(UNUSED_PARAM float in) { return false; } + + //! Get BlacksheetTotalHeight + /*! + \return float + */ + virtual float GetBlacksheetTotalHeight() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set BlacksheetTotalHeight + /*! + \param in float + \return `true` if successful, `false` if not + */ + virtual bool SetBlacksheetTotalHeight(UNUSED_PARAM float in) { return false; } + + //! Get BlacksheetRadius + /*! + \return float + */ + virtual float GetBlacksheetRadius() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set BlacksheetRadius + /*! + \param in float + \return `true` if successful, `false` if not + */ + virtual bool SetBlacksheetRadius(UNUSED_PARAM float in) { return false; } + + //! Get DetectorID + /*! + \return DetectorID + */ + virtual DetectorID GetDetectorID() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, DetectorID::kUndefined); + } + + //! Set DetectorID + /*! + \param in DetectorID + \return `true` if successful, `false` if not + */ + virtual bool SetDetectorID(UNUSED_PARAM DetectorID in) { return false; } + + ClassDef(HKGeometry, 1); //!< ROOT Class definition +}; + +#endif diff --git a/leaf/DataModel-lite/HKGeometryPMT copy.hpp b/leaf/DataModel-lite/HKGeometryPMT copy.hpp new file mode 100644 index 0000000..1634ed9 --- /dev/null +++ b/leaf/DataModel-lite/HKGeometryPMT copy.hpp @@ -0,0 +1,351 @@ +#ifndef HK_GEOMETRYPMT_HPP +#define HK_GEOMETRYPMT_HPP + +/************************************************* + * HKGeometryPMT.hpp + * - Author: T. Dealtry + * - Date: 2024/11/14 + ************************************************/ + +#include +#include +#include + +#include "Enums/PMTType.hpp" +#include "HKObject.hpp" + +/************************************************ + * HKGeometryPMT class definition + ***********************************************/ +class HKGeometryPMT : public HKObject { + public: + + HKGeometryPMT(); //!< Constructor + + //! Create HKGeometryPMT from an HKGeometryPMT pointor + /*! + \param in HKGeometryPMT* + */ + HKGeometryPMT(HKGeometryPMT* in); + + virtual ~HKGeometryPMT() = 0; //!< Destructor + virtual void Reset() = 0; //!< Reset function, called after each event + virtual unsigned int GetVersion() const = 0; //!< Return the class version number + + static std::string GetBaseNameStatic() { + return "HKGeometryPMT"; + } //!< Static function. Return the class base name + + std::string GetBaseName() const { + return HKGeometryPMT::GetBaseNameStatic(); + } //!< Return the class base name + + // v1 functions + //! Get PMTSoftwareID + /*! + \return unsigned int + */ + virtual unsigned int GetPMTSoftwareID() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set PMTSoftwareID + /*! + \param in unsigned int + \return `true` if successful, `false` if not + */ + virtual bool SetPMTSoftwareID(UNUSED_PARAM unsigned int in) { return false; } + + //! Get PMT Type + /*! + \return PMTType + */ + virtual PMTType GetType() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, PMTType::kUndefined); + } + + //! Set PMT Type + /*! + \param in PMTType + \return `true` if successful, `false` if not + */ + virtual bool SetType(UNUSED_PARAM PMTType in) { return false; } + + //! Get sub ID + /*! + Return the PMT sub ID number. 0 for 20" PMT and OD PMT. 1-19 for mPMT's 3" PMTs + \return unsigned int + */ + virtual unsigned int GetSubID() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set sub ID + /*! + Set the PMT sub ID number. 0 for 20" PMT and OD PMT. 1-19 for mPMT's 3" PMTs + \param in unsigned int + \return `true` if successful, `false` if not + */ + virtual bool SetSubID(UNUSED_PARAM unsigned int in) { return false; } + + //! Get PMT Position in cm + /*! + \return ROOT::Math::XYZVector. position over [x,y,z] + */ + virtual ROOT::Math::XYZVector GetPositionInCm() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, + ROOT::Math::XYZVector()); + } + + //! Set PMT Position in cm + /*! + \param x double + \param y double + \param z double + \return `true` if successful, `false` if not + */ + virtual bool SetPositionInCm(UNUSED_PARAM double x, UNUSED_PARAM double y, UNUSED_PARAM double z) { + return false; + } + + //! Set PMT Position in cm + /*! + \param x ROOT::Math::XYZVector + \return `true` if successful, `false` if not + */ + virtual bool SetPositionInCm(UNUSED_PARAM ROOT::Math::XYZVector& in) { return false; } + + //! Get PMT Orientation + /*! + \return ROOT::Math::XYZVector. orientation over [x,y,z] + */ + virtual ROOT::Math::XYZVector GetOrientation() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, + ROOT::Math::XYZVector()); + } + + //! Set PMT Orientation + /*! + \param x double + \param y double + \param z double + \return `true` if successful, `false` if not + */ + virtual bool SetOrientation(UNUSED_PARAM double x, UNUSED_PARAM double y, UNUSED_PARAM double z) { + return false; + } + + //! Set PMT Orientation + /*! + \param x ROOT::Math::XYZVector + \return `true` if successful, `false` if not + */ + virtual bool SetPoSetOrientationsitionInCm(UNUSED_PARAM ROOT::Math::XYZVector& in) { return false; } + + ClassDef(HKGeometryPMT, 1); //!< ROOT Class definition +}; + +/************************************************ + * HKGeometryPMTCollection base class definition + * + * Collection class wrapping a vector of HKGeometryPMT + ***********************************************/ +class HKGeometryPMTCollection : public HKObjectCollection { + public: + + HKGeometryPMTCollection() {} //!< Creator + + virtual ~HKGeometryPMTCollection() = 0; //!< Destructor + virtual void Reset() = 0; //!< Reset function, called after each event + virtual unsigned int GetVersion() const = 0; //!< Return the contained class version number + + static std::string GetBaseNameStatic() { + return "HKGeometryPMTCollection"; + } //!< Static function. Return the class base name + + std::string GetBaseName() const { + return HKGeometryPMTCollection::GetBaseNameStatic(); + } //!< Return the class base name + + //! operator [] + /*! + \param i integer. Entry index + */ + virtual HKGeometryPMT* operator[](UNUSED_PARAM int i) { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); + } + + //! Wrapper for std::vector at() + /*! + \param i integer. Entry index + */ + virtual HKGeometryPMT* at(UNUSED_PARAM int i) { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); + } + + virtual const HKGeometryPMT* at(UNUSED_PARAM int i) const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); + } + + + //! Wrapper for std::vector back() + virtual HKGeometryPMT* back() { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); + } + + //! Wrapper for std::vector size() + virtual size_t size() const { return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); } + + //! Wrapper for std::vector resize() + /*! + \param entries integer. Number of entries + */ + virtual void resize(UNUSED_PARAM size_t entries) { return; } + + //! Wrapper for std::vector push_back() + /*! + \param in HKGeometryPMT*. Object to be inserted + */ + virtual void push_back(UNUSED_PARAM HKGeometryPMT* in) { return; } + + //! Wrapper for std::vector emplace_back() + /*! + \param in HKGeometryPMT*. Object to be inserted + */ + virtual void emplace_back(UNUSED_PARAM HKGeometryPMT* in) { return; } + + //! Emplace back an HKGeometryPMT instance in the collection + virtual void AddGeometryPMT() { return; } + + private: + +#ifndef HK_USE_ROOT7 + //! Return the pointor to the unique_ptr + virtual std::unique_ptr > >* GetVectorUniquePointor() { + return HKFormatError::ThrowErrorVersion< + std::unique_ptr > >*>(__PRETTY_FUNCTION__, NULL); + } + + //! Return the pointor to the vector + virtual std::vector >* GetVectorPointor() { + return HKFormatError::ThrowErrorVersion >*>( + __PRETTY_FUNCTION__, + NULL); + } + + template friend class HKBranchManagerCollection; +#endif + + ClassDef(HKGeometryPMTCollection, 1); //!< Class definition for ROOT +}; + +/************************************************ + * HKGeometryPMTCollectionTemplate definition + * + * Collection class wrapping a vector of HKGeometryPMT + ***********************************************/ +template class HKGeometryPMTCollectionT : public HKGeometryPMTCollection { + + static_assert(std::is_base_of::value, "T must inherit from HKGeometryPMT"); + + public: + + HKGeometryPMTCollectionT(bool initialize = true) { + if(initialize) { +#ifdef HK_USE_ROOT7 + contents = std::make_shared > >(); +#else + contents = std::make_unique > >(); +#endif + this->Reset(); + } + } //!< Creator + + //! Constructor. Initialize the collection with specified number of entries + /*! + \param entries integer. Number of entries + */ + HKGeometryPMTCollectionT(size_t entries) { + this->Reset(); + this->resize(entries); + } + + ~HKGeometryPMTCollectionT() override { this->Reset(); } //!< Destructor + + void Reset() override { + contents->clear(); + } //!< Reset function, called after each event. Clear the collection. + + static constexpr unsigned int GetVersionStatic() { + return T::GetVersionStatic(); + } //!< Static function. Return the contained class version number + + unsigned int GetVersion() const override { + return T::GetVersionStatic(); + } //!< Return the contained class version number + + //! operator [] + /*! + \param i integer. Entry index + */ + HKGeometryPMT* operator[](int i) override { return (*contents)[i].get(); } // Use at(i) here too? + + //! Wrapper for std::vector at() + /*! + \param i integer. Entry index + */ + HKGeometryPMT* at(int i) override { return contents->at(i).get(); } + + const HKGeometryPMT* at(int i) const { return contents->at(i).get(); } + + //! Wrapper for std::vector back() + HKGeometryPMT* back() override { return contents->back().get(); } + + //! Wrapper for std::vector size() + size_t size() const override { return contents->size(); } + + //! Wrapper for std::vector resize() + /*! + \param entries integer. Number of entries + */ + void resize(size_t entries) override { contents->resize(entries); } + + //! Wrapper for std::vector push_back() + /*! + \param in HKGeometryPMT*. Object to be inserted + */ + void push_back(HKGeometryPMT* in) override { contents->push_back(std::make_unique(in)); } + + //! Wrapper for std::vector emplace_back() + /*! + \param in HKGeometryPMT*. Object to be inserted + */ + void emplace_back(HKGeometryPMT* in) override { contents->emplace_back(std::make_unique(in)); } + + //! Emplace back an empty HKGeometryPMT instance in the collection + void AddGeometryPMT() override { contents->emplace_back(std::make_unique()); } + + private: + +#ifdef HK_USE_ROOT7 + std::shared_ptr > > contents; //!< Vector of HKGeometryPMT + + template friend class HKDataMemberCollectionCreator; +#else + //! Return the pointor to the unique_ptr + std::unique_ptr > >* GetVectorUniquePointor() override { + return &contents; + } + + //! Return the pointor to the vector + std::vector >* GetVectorPointor() override { return contents.get(); } + + std::unique_ptr > > contents; //!< Vector of HKGeometryPMT + + template friend class HKBranchManagerCollection; +#endif + + ClassDefOverride(HKGeometryPMTCollectionT, 1); //!< Class definition for ROOT +}; + +#endif diff --git a/leaf/DataModel-lite/HKGeometryPMT.cpp b/leaf/DataModel-lite/HKGeometryPMT.cpp new file mode 100644 index 0000000..8ba2a16 --- /dev/null +++ b/leaf/DataModel-lite/HKGeometryPMT.cpp @@ -0,0 +1,13 @@ +#include "HKGeometryPMT.hpp" + +HKGeometryPMT::HKGeometryPMT() {} + +HKGeometryPMT::HKGeometryPMT(HKGeometryPMT* /*in*/) {} + +// pure virtual destructor needs implementation +HKGeometryPMT::~HKGeometryPMT() {} + +/**********************************************************/ + +// pure virtual destructor needs implementation +HKGeometryPMTCollection::~HKGeometryPMTCollection() {} diff --git a/leaf/DataModel-lite/HKGeometryPMT.hpp b/leaf/DataModel-lite/HKGeometryPMT.hpp new file mode 100644 index 0000000..79682fb --- /dev/null +++ b/leaf/DataModel-lite/HKGeometryPMT.hpp @@ -0,0 +1,358 @@ +#ifndef HK_GEOMETRYPMT_HPP +#define HK_GEOMETRYPMT_HPP + +/************************************************* + * HKGeometryPMT.hpp + * - Author: T. Dealtry + * - Date: 2024/11/14 + ************************************************/ + +#include +#include +#include + +#include "Enums/PMTType.hpp" +#include "HKObject.hpp" + +/************************************************ + * HKGeometryPMT class definition + ***********************************************/ +class HKGeometryPMT : public HKObject { + public: + + HKGeometryPMT(); //!< Constructor + + //! Create HKGeometryPMT from an HKGeometryPMT pointor + /*! + \param in HKGeometryPMT* + */ + HKGeometryPMT(HKGeometryPMT* in); + + virtual ~HKGeometryPMT() = 0; //!< Destructor + virtual void Reset() = 0; //!< Reset function, called after each event + virtual unsigned int GetVersion() const = 0; //!< Return the class version number + + static std::string GetBaseNameStatic() { + return "HKGeometryPMT"; + } //!< Static function. Return the class base name + + std::string GetBaseName() const { + return HKGeometryPMT::GetBaseNameStatic(); + } //!< Return the class base name + + // v1 functions + //! Get PMTSoftwareID + /*! + \return unsigned int + */ + virtual unsigned int GetPMTSoftwareID() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set PMTSoftwareID + /*! + \param in unsigned int + \return `true` if successful, `false` if not + */ + virtual bool SetPMTSoftwareID(UNUSED_PARAM unsigned int in) { return false; } + + //! Get PMT Type + /*! + \return PMTType + */ + virtual PMTType GetType() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, PMTType::kUndefined); + } + + //! Set PMT Type + /*! + \param in PMTType + \return `true` if successful, `false` if not + */ + virtual bool SetType(UNUSED_PARAM PMTType in) { return false; } + + //! Get sub ID + /*! + Return the PMT sub ID number. 0 for 20" PMT and OD PMT. 1-19 for mPMT's 3" PMTs + \return unsigned int + */ + virtual unsigned int GetSubID() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Set sub ID + /*! + Set the PMT sub ID number. 0 for 20" PMT and OD PMT. 1-19 for mPMT's 3" PMTs + \param in unsigned int + \return `true` if successful, `false` if not + */ + virtual bool SetSubID(UNUSED_PARAM unsigned int in) { return false; } + + //! Get PMT Position in cm + /*! + \return ROOT::Math::XYZVector. position over [x,y,z] + */ + virtual ROOT::Math::XYZVector GetPositionInCm() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, + ROOT::Math::XYZVector()); + } + + //! Set PMT Position in cm + /*! + \param x double + \param y double + \param z double + \return `true` if successful, `false` if not + */ + virtual bool SetPositionInCm(UNUSED_PARAM double x, UNUSED_PARAM double y, UNUSED_PARAM double z) { + return false; + } + + //! Set PMT Position in cm + /*! + \param x ROOT::Math::XYZVector + \return `true` if successful, `false` if not + */ + virtual bool SetPositionInCm(UNUSED_PARAM ROOT::Math::XYZVector& in) { return false; } + + //! Get PMT Orientation + /*! + \return ROOT::Math::XYZVector. orientation over [x,y,z] + */ + virtual ROOT::Math::XYZVector GetOrientation() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, + ROOT::Math::XYZVector()); + } + + //! Set PMT Orientation + /*! + \param x double + \param y double + \param z double + \return `true` if successful, `false` if not + */ + virtual bool SetOrientation(UNUSED_PARAM double x, UNUSED_PARAM double y, UNUSED_PARAM double z) { + return false; + } + + //! Set PMT Orientation + /*! + \param x ROOT::Math::XYZVector + \return `true` if successful, `false` if not + */ + virtual bool SetPoSetOrientationsitionInCm(UNUSED_PARAM ROOT::Math::XYZVector& in) { return false; } + + ClassDef(HKGeometryPMT, 1); //!< ROOT Class definition +}; + +/************************************************ + * HKGeometryPMTCollection base class definition + * + * Collection class wrapping a vector of HKGeometryPMT + ***********************************************/ +class HKGeometryPMTCollection : public HKObjectCollection { + public: + + HKGeometryPMTCollection() {} //!< Creator + + virtual ~HKGeometryPMTCollection() = 0; //!< Destructor + virtual void Reset() = 0; //!< Reset function, called after each event + virtual unsigned int GetVersion() const = 0; //!< Return the contained class version number + + static std::string GetBaseNameStatic() { + return "HKGeometryPMTCollection"; + } //!< Static function. Return the class base name + + std::string GetBaseName() const { + return HKGeometryPMTCollection::GetBaseNameStatic(); + } //!< Return the class base name + + //! operator [] + /*! + \param i integer. Entry index + */ + virtual HKGeometryPMT* operator[](UNUSED_PARAM int i) { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); + } + + //! Wrapper for std::vector at() + /*! + \param i integer. Entry index + */ + virtual HKGeometryPMT* at(UNUSED_PARAM int i) { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); + } + + virtual const HKGeometryPMT* at(UNUSED_PARAM int i) const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); + } + + + //! Wrapper for std::vector back() + virtual HKGeometryPMT* back() { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); + } + + //! Wrapper for std::vector size() + virtual size_t size() const { return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); } + + //! Wrapper for std::vector resize() + /*! + \param entries integer. Number of entries + */ + virtual void resize(UNUSED_PARAM size_t entries) { return; } + + //! Wrapper for std::vector push_back() + /*! + \param in HKGeometryPMT*. Object to be inserted + */ + virtual void push_back(UNUSED_PARAM HKGeometryPMT* in) { return; } + + //! Wrapper for std::vector emplace_back() + /*! + \param in HKGeometryPMT*. Object to be inserted + */ + virtual void emplace_back(UNUSED_PARAM HKGeometryPMT* in) { return; } + + //! Emplace back an HKGeometryPMT instance in the collection + virtual void AddGeometryPMT() { return; } + + private: + +#ifndef HK_USE_ROOT7 + //! Return the pointor to the unique_ptr + virtual std::unique_ptr > >* GetVectorUniquePointor() { + return HKFormatError::ThrowErrorVersion< + std::unique_ptr > >*>(__PRETTY_FUNCTION__, NULL); + } + + //! Return the pointor to the vector + virtual std::vector >* GetVectorPointor() { + return HKFormatError::ThrowErrorVersion >*>( + __PRETTY_FUNCTION__, + NULL); + } + + template friend class HKBranchManagerCollection; +#endif + + ClassDef(HKGeometryPMTCollection, 1); //!< Class definition for ROOT +}; + +/************************************************ + * HKGeometryPMTCollectionTemplate definition + * + * Collection class wrapping a vector of HKGeometryPMT + ***********************************************/ +template class HKGeometryPMTCollectionT : public HKGeometryPMTCollection { + + static_assert(std::is_base_of::value, "T must inherit from HKGeometryPMT"); + + public: + + HKGeometryPMTCollectionT(bool initialize = true) { + if(initialize) { +#ifdef HK_USE_ROOT7 + contents = std::make_shared > >(); +#else + contents = std::make_unique > >(); +#endif + this->Reset(); + } + } //!< Creator + + //! Constructor. Initialize the collection with specified number of entries + /*! + \param entries integer. Number of entries + */ + HKGeometryPMTCollectionT(size_t entries) { + this->Reset(); + this->resize(entries); + } + + ~HKGeometryPMTCollectionT() override { this->Reset(); } //!< Destructor + + void Reset() override { + contents->clear(); + } //!< Reset function, called after each event. Clear the collection. + + static constexpr unsigned int GetVersionStatic() { + return T::GetVersionStatic(); + } //!< Static function. Return the contained class version number + + unsigned int GetVersion() const override { + return T::GetVersionStatic(); + } //!< Return the contained class version number + + //! operator [] + /*! + \param i integer. Entry index + */ + HKGeometryPMT* operator[](int i) override { return (*contents)[i].get(); } // Use at(i) here too? + + //! Wrapper for std::vector at() + /*! + \param i integer. Entry index + */ + HKGeometryPMT* at(int i) override { return contents->at(i).get(); } + + const HKGeometryPMT* at(int i) const { return contents->at(i).get(); } + + //! Wrapper for std::vector back() + HKGeometryPMT* back() override { return contents->back().get(); } + + //! Wrapper for std::vector size() + size_t size() const override { return contents->size(); } + + //! Wrapper for std::vector resize() + /*! + \param entries integer. Number of entries + */ + void resize(size_t entries) override { + contents->resize(entries); +#ifdef HK_USE_ROOT7 + for (auto& ptr : *contents) { ptr = std::make_unique(); } +#else + for (auto& ptr : *contents) { ptr = std::make_unique(); } +#endif + } + + //! Wrapper for std::vector push_back() + /*! + \param in HKGeometryPMT*. Object to be inserted + */ + void push_back(HKGeometryPMT* in) override { contents->push_back(std::make_unique(in)); } + + //! Wrapper for std::vector emplace_back() + /*! + \param in HKGeometryPMT*. Object to be inserted + */ + void emplace_back(HKGeometryPMT* in) override { contents->emplace_back(std::make_unique(in)); } + + //! Emplace back an empty HKGeometryPMT instance in the collection + void AddGeometryPMT() override { contents->emplace_back(std::make_unique()); } + + private: + +#ifdef HK_USE_ROOT7 + std::shared_ptr > > contents; //!< Vector of HKGeometryPMT + + template friend class HKDataMemberCollectionCreator; +#else + //! Return the pointor to the unique_ptr + std::unique_ptr > >* GetVectorUniquePointor() override { + return &contents; + } + + //! Return the pointor to the vector + std::vector >* GetVectorPointor() override { return contents.get(); } + + std::unique_ptr > > contents; //!< Vector of HKGeometryPMT + + template friend class HKBranchManagerCollection; +#endif + + ClassDefOverride(HKGeometryPMTCollectionT, 1); //!< Class definition for ROOT +}; + +#endif diff --git a/leaf/DataModel-lite/HKGeometryPMTV1.cpp b/leaf/DataModel-lite/HKGeometryPMTV1.cpp new file mode 100644 index 0000000..7906b39 --- /dev/null +++ b/leaf/DataModel-lite/HKGeometryPMTV1.cpp @@ -0,0 +1,33 @@ +#include "HKGeometryPMTV1.hpp" + +HKGeometryPMTV1::HKGeometryPMTV1() { + + m_pmt_software_id = 0; + m_pmt_sub_id = 0; + m_pmt_position.SetXYZ(-9999., -9999., -9999.); + m_pmt_orientation.SetXYZ(-9999., -9999., -9999.); + + this->Reset(); +} + +HKGeometryPMTV1::HKGeometryPMTV1(HKGeometryPMT* in) { + // Load variable here: + + // v1 copies + m_pmt_software_id = in->GetPMTSoftwareID(); + m_pmt_sub_id = in->GetSubID(); + m_pmt_position = in->GetPositionInCm(); + m_pmt_orientation = in->GetOrientation(); +} + +HKGeometryPMTV1::~HKGeometryPMTV1() { + this->Reset(); +} + +void HKGeometryPMTV1::Reset() { + // Initialize variable here: + // Note, reset is called for every new event, so variables needing to be initialized + // once for a full file should be defined in the creator + + // v1 resets +} diff --git a/leaf/DataModel-lite/HKGeometryPMTV1.hpp b/leaf/DataModel-lite/HKGeometryPMTV1.hpp new file mode 100644 index 0000000..617454f --- /dev/null +++ b/leaf/DataModel-lite/HKGeometryPMTV1.hpp @@ -0,0 +1,167 @@ +#ifndef HK_GEOMETRYPMT_V1_HPP +#define HK_GEOMETRYPMT_V1_HPP + +#include "HKGeometryPMT.hpp" + +/************************************************ + * HKGeometryPMTV1 class definition + * Version 1 of HKGeometryPMT + * + * Just stores the PMT ID number used in software for this PMT. + * Other information will be able to be looked up using channel mapping functionality (currently undefined). + * This includes + * * Position + * * Orientation + * * PMT type (20"/mPMT/OD PMT) + * * PMT batch (e.g. 20" vX) + * * PMT serial # + * * Whether PMT is on barrel / top cap / bottom cap + * * Electronics board type + * * Electronics board serial # + * * Electronics board channel # + ***********************************************/ +class HKGeometryPMTV1 : public HKGeometryPMT { + public: + + HKGeometryPMTV1(); //!< Constructor + + //! Create HKGeometryPMTV1 from an HKGeometryPMT pointor + /*! + \param in HKGeometryPMT* + */ + HKGeometryPMTV1(HKGeometryPMT* in); + + ~HKGeometryPMTV1() override; //!< Destructor + void Reset() override; //!< Reset function, called after each event. Use it to initialize variable + + static constexpr unsigned int GetVersionStatic() { + return 1; + } //!< Static function. Return the class version number + + unsigned int GetVersion() const override { + return HKGeometryPMTV1::GetVersionStatic(); + } //!< Return the class version number + + // v1 functions + + //! Get PMTSoftwareID + /*! + The PMT ID number used in software for this PMT + \return unsigned int + */ + unsigned int GetPMTSoftwareID() const override { return m_pmt_software_id; } + + //! Set PMTSoftwareID + /*! + The PMT ID number used in software for this PMT + \param in unsigned int + \return `true` if successful, `false` if not + */ + bool SetPMTSoftwareID(unsigned int in) override { + m_pmt_software_id = in; + return true; + } + + //! Get sub ID + /*! + Return the PMT sub ID number. 0 for 20" PMT and OD PMT. 1-19 for mPMT's 3" PMTs + \return unsigned int + */ + unsigned int GetSubID() const override { return m_pmt_sub_id; } + + //! Set sub ID + /*! + Set the PMT sub ID number. 0 for 20" PMT and OD PMT. 1-19 for mPMT's 3" PMTs + \param in unsigned int + \return `true` if successful, `false` if not + */ + bool SetSubID(unsigned int in) override { + m_pmt_sub_id = in; + return true; + } + + //! Get PMT Type + /*! + \return PMTType + */ + PMTType GetType() const override { return m_pmt_type; } + + //! Set PMT Type + /*! + \param in PMTType + \return `true` if successful, `false` if not + */ + bool SetType(PMTType in) override { + m_pmt_type = in; + return true; + } + + //! Get PMT Position in cm + /*! + \return ROOT::Math::XYZVector. position over [x,y,z] + */ + virtual ROOT::Math::XYZVector GetPositionInCm() const override { return m_pmt_position; } + + //! Set PMT Position in cm + /*! + \param x double + \param y double + \param z double + \return `true` if successful, `false` if not + */ + bool SetPositionInCm(double x, double y, double z) override { + m_pmt_position.SetXYZ(x, y, z); + return true; + } + + //! Set PMT Position in cm + /*! + \param x ROOT::Math::XYZVector + \return `true` if successful, `false` if not + */ + bool SetPositionInCm(ROOT::Math::XYZVector& in) override { + m_pmt_position = in; + return true; + } + + //! Get PMT Orientation + /*! + \return ROOT::Math::XYZVector. orientation over [x,y,z] + */ + ROOT::Math::XYZVector GetOrientation() const override { return m_pmt_orientation; } + + //! Set PMT Orientation + /*! + \param x double + \param y double + \param z double + \return `true` if successful, `false` if not + */ + bool SetOrientation(double x, double y, double z) override { + m_pmt_orientation.SetXYZ(x, y, z); + return true; + } + + //! Set PMT Orientation + /*! + \param x ROOT::Math::XYZVector + \return `true` if successful, `false` if not + */ + bool SetPoSetOrientationsitionInCm(ROOT::Math::XYZVector& in) override { + m_pmt_orientation = in; + return true; + } + + private: + + // v1 data members + unsigned int m_pmt_software_id; //!< PMTSoftwareID + PMTType m_pmt_type; //!< PMT type + unsigned int m_pmt_sub_id; //!< PMT sub id (0 for 20" or OD PMT, 1-19 for mPMT's 3" PMTs) + ROOT::Math::XYZVector m_pmt_position; //!< Position in cm + ROOT::Math::XYZVector m_pmt_orientation; //!< Normalized orientation + + ClassDefOverride(HKGeometryPMTV1, 1); //!< ROOT Class definition +}; + +#endif diff --git a/leaf/DataModel-lite/HKGeometryV1.cpp b/leaf/DataModel-lite/HKGeometryV1.cpp new file mode 100644 index 0000000..14c0647 --- /dev/null +++ b/leaf/DataModel-lite/HKGeometryV1.cpp @@ -0,0 +1,47 @@ +#include "HKGeometryV1.hpp" + +HKGeometryV1::HKGeometryV1() { + this->Reset(); +} + +HKGeometryV1::HKGeometryV1(HKGeometry* in) { + // Load variable here: + + // v1 copies + m_iwcd_elevator_height = in->GetIWCDElevatorHeight(); + m_is_cavern_material_simulated = in->GetIsCavernMaterialSimulated(); + m_is_od_simulated = in->GetIsODSimulated(); + m_is_dead_space_simulated = in->GetIsDeadSpaceSimulated(); + m_are_od_pmts_active = in->GetAreODPMTsActive(); + m_outer_tyvek_total_height = in->GetOuterTyvekTotalHeight(); + m_outer_tyvek_radius = in->GetOuterTyvekRadius(); + m_inner_tyvek_total_height = in->GetInnerTyvekTotalHeight(); + m_inner_tyvek_radius = in->GetInnerTyvekRadius(); + m_blacksheet_total_height = in->GetBlacksheetTotalHeight(); + m_blacksheet_radius = in->GetBlacksheetRadius(); + m_detector_id = in->GetDetectorID(); +} + +HKGeometryV1::~HKGeometryV1() { + this->Reset(); +} + +void HKGeometryV1::Reset() { + // Initialize variable here: + // Note, reset is called for every new event, so variables needing to be initialized + // once for a full file should be defined in the creator + + // v1 resets + m_iwcd_elevator_height = 0; + m_is_cavern_material_simulated = false; + m_is_od_simulated = false; + m_is_dead_space_simulated = false; + m_are_od_pmts_active = false; + m_outer_tyvek_total_height = 0; + m_outer_tyvek_radius = 0; + m_inner_tyvek_total_height = 0; + m_inner_tyvek_radius = 0; + m_blacksheet_total_height = 0; + m_blacksheet_radius = 0; + m_detector_id = DetectorID::kUndefined; +} diff --git a/leaf/DataModel-lite/HKGeometryV1.hpp b/leaf/DataModel-lite/HKGeometryV1.hpp new file mode 100644 index 0000000..225b9ce --- /dev/null +++ b/leaf/DataModel-lite/HKGeometryV1.hpp @@ -0,0 +1,257 @@ +#ifndef HK_GEOMETRY_V1_HPP +#define HK_GEOMETRY_V1_HPP + +/************************************************* + * HKGeometryV1.hpp + * - Author: + * - Date: 1970/01/01 + ************************************************/ + +#include "HKGeometry.hpp" + +/************************************************ + * HKGeometryV1 class definition + * Version 1 of HKGeometry + ***********************************************/ +class HKGeometryV1 : public HKGeometry { + public: + + HKGeometryV1(); //!< Constructor + + //! Create HKGeometryV1 from an HKGeometry pointor + /*! + \param in HKGeometry* + */ + HKGeometryV1(HKGeometry* in); + + ~HKGeometryV1() override; //!< Destructor + void Reset() override; //!< Reset function, called after each event. Use it to initialize variable + + static constexpr unsigned int GetVersionStatic() { + return 1; + } //!< Static function. Return the class version number + + unsigned int GetVersion() const override { + return HKGeometryV1::GetVersionStatic(); + } //!< Return the class version number + + // v1 functions + //! Get IWCDElevatorHeight + /*! + This is meaningless for detectors other than IWCD. + \return float + */ + float GetIWCDElevatorHeight() const override { return m_iwcd_elevator_height; } + + //! Set IWCDElevatorHeight + /*! + \param in float + \return `true` if successful, `false` if not + */ + bool SetIWCDElevatorHeight(float in) override { + m_iwcd_elevator_height = in; + return true; + } + + //! Get IsCavernMaterialSimulated + /*! + \return bool + */ + bool GetIsCavernMaterialSimulated() const override { return m_is_cavern_material_simulated; } + + //! Set IsCavernMaterialSimulated + /*! + \param in bool + \return `true` if successful, `false` if not + */ + bool SetIsCavernMaterialSimulated(bool in) override { + m_is_cavern_material_simulated = in; + return true; + } + + //! Get IsODSimulated + /*! + \return bool + */ + bool GetIsODSimulated() const override { return m_is_od_simulated; } + + //! Set IsODSimulated + /*! + \param in bool + \return `true` if successful, `false` if not + */ + bool SetIsODSimulated(bool in) override { + m_is_od_simulated = in; + return true; + } + + //! Get IsDeadSpaceSimulated + /*! + \return bool + */ + bool GetIsDeadSpaceSimulated() const override { return m_is_dead_space_simulated; } + + //! Set IsDeadSpaceSimulated + /*! + \param in bool + \return `true` if successful, `false` if not + */ + bool SetIsDeadSpaceSimulated(bool in) override { + m_is_dead_space_simulated = in; + return true; + } + + //! Get AreODPMTsActive + /*! + \return bool + */ + bool GetAreODPMTsActive() const override { return m_are_od_pmts_active; } + + //! Set AreODPMTsActive + /*! + \param in bool + \return `true` if successful, `false` if not + */ + bool SetAreODPMTsActive(bool in) override { + m_are_od_pmts_active = in; + return true; + } + + //! Get the total height of the outer (cavern bottom/dome bottom side) tyvek + /*! + Returns value in standard HK units: mm + \return float + */ + float GetOuterTyvekTotalHeight() const override { return m_outer_tyvek_total_height; } + + //! Set the total height of the outer (cavern bottom/dome bottom side) tyvek + /*! + \param in float + \return `true` if successful, `false` if not + */ + bool SetOuterTyvekTotalHeight(float in) override { + m_outer_tyvek_total_height = in; + return true; + } + + //! Get the radius of the outer (cavern bottom/dome bottom side) tyvek + /*! + Returns value in standard HK units: mm + \return float + */ + float GetOuterTyvekRadius() const override { return m_outer_tyvek_radius; } + + //! Set the radius of the outer (cavern bottom/dome bottom side) tyvek + /*! + \param in float + \return `true` if successful, `false` if not + */ + bool SetOuterTyvekRadius(float in) override { + m_outer_tyvek_radius = in; + return true; + } + + //! Get the total height of the inner (PMT support structure side) tyvek + /*! + Returns value in standard HK units: mm + \return float + */ + float GetInnerTyvekTotalHeight() const override { return m_inner_tyvek_total_height; } + + //! Set the total height of the inner (PMT support structure side) tyvek + /*! + \param in float + \return `true` if successful, `false` if not + */ + bool SetInnerTyvekTotalHeight(float in) override { + m_inner_tyvek_total_height = in; + return true; + } + + //! Get the radius of the inner (PMT support structure side) tyvek + /*! + Returns value in standard HK units: mm + \return float + */ + float GetInnerTyvekRadius() const override { return m_inner_tyvek_radius; } + + //! Set the radius of the inner (PMT support structure side) tyvek + /*! + \param in float + \return `true` if successful, `false` if not + */ + bool SetInnerTyvekRadius(float in) override { + m_inner_tyvek_radius = in; + return true; + } + + //! Get the total height of the ID blacksheet + /*! + Returns value in standard HK units: mm + \return float + */ + float GetBlacksheetTotalHeight() const override { return m_blacksheet_total_height; } + + //! Set the total height of the ID blacksheet + /*! + \param in float + \return `true` if successful, `false` if not + */ + bool SetBlacksheetTotalHeight(float in) override { + m_blacksheet_total_height = in; + return true; + } + + //! Get the radius of the ID blacksheet + /*! + Returns value in standard HK units: mm + \return float + */ + float GetBlacksheetRadius() const override { return m_blacksheet_radius; } + + //! Set the radius of the ID blacksheet + /*! + \param in float + \return `true` if successful, `false` if not + */ + bool SetBlacksheetRadius(float in) override { + m_blacksheet_radius = in; + return true; + } + + //! Get DetectorID + /*! + \return DetectorID + */ + DetectorID GetDetectorID() const override { return m_detector_id; } + + //! Set DetectorID + /*! + \param in DetectorID + \return `true` if successful, `false` if not + */ + bool SetDetectorID(DetectorID in) override { + m_detector_id = in; + return true; + } + + private: + + // v1 data members + float m_iwcd_elevator_height; //!< IWCDElevatorHeight + bool m_is_cavern_material_simulated; //!< IsCavernMaterialSimulated + bool m_is_od_simulated; //!< IsODSimulated + bool m_is_dead_space_simulated; //!< IsDeadSpaceSimulated + bool m_are_od_pmts_active; //!< AreODPMTsActive + float m_outer_tyvek_total_height; //!< OuterTyvekTotalHeight + float m_outer_tyvek_radius; //!< OuterTyvekRadius + float m_inner_tyvek_total_height; //!< InnerTyvekTotalHeight + float m_inner_tyvek_radius; //!< InnerTyvekRadius + float m_blacksheet_total_height; //!< BlacksheetTotalHeight + float m_blacksheet_radius; //!< BlacksheetRadius + DetectorID m_detector_id; //!< DetectorID + + ClassDefOverride(HKGeometryV1, 1); //!< ROOT Class definition +}; + +#endif diff --git a/leaf/DataModel-lite/HKHit.cpp b/leaf/DataModel-lite/HKHit.cpp new file mode 100644 index 0000000..a0022eb --- /dev/null +++ b/leaf/DataModel-lite/HKHit.cpp @@ -0,0 +1,13 @@ +#include "HKHit.hpp" + +HKHit::HKHit() {} + +HKHit::HKHit(HKHit* /*in*/) {} + +// pure virtual destructor needs implementation +HKHit::~HKHit() {} + +/**********************************************************/ + +// pure virtual destructor needs implementation +HKHitsCollection::~HKHitsCollection() {} diff --git a/leaf/DataModel-lite/HKHit.hpp b/leaf/DataModel-lite/HKHit.hpp new file mode 100644 index 0000000..d3b11e0 --- /dev/null +++ b/leaf/DataModel-lite/HKHit.hpp @@ -0,0 +1,299 @@ +#ifndef HK_HIT_HPP +#define HK_HIT_HPP + +/************************************************ + * HKHit.hpp + * - Author: Guillaume Pronost + * - Date: 2024/07/10 + ***********************************************/ + +#include +#include + +#include "HKObject.hpp" + +/************************************************ + * HKHit base class definition + * + * Store PMT hit information + ***********************************************/ +class HKHit : public HKObject { + public: + + HKHit(); //!< Constructor + + //! Create HKHit from an HKHit pointor + /*! + \param in HKHit* + */ + HKHit(HKHit* in); + + virtual ~HKHit() = 0; //!< Destructor + virtual void Reset() = 0; //!< Reset function, called after each event + virtual unsigned int GetVersion() const = 0; //!< Return the class version number + + static std::string GetBaseNameStatic() { + return "HKHit"; + } //!< Static function. Return the class base name + + std::string GetBaseName() const { + return HKHit::GetBaseNameStatic(); + } //!< Return the class base name + + // v1 functions + //! Return PMT number + /*! + \return unsigned short. PMT number + */ + virtual unsigned short GetPMTNumber() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); + } + + //! Return hit's time + //! TODO: Define unit + /*! + \return double. Hit's time + */ + virtual double GetTime() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, -9999.); + } + + //! Return hit's charge + //! TODO: Define unit + /*! + \return double. Hit's charge + */ + virtual double GetCharge() const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, -9999.); + } + + //! Set PMT number + /*! + \param in unsigned short. PMT number + */ + virtual bool SetPMTNumber(UNUSED_PARAM unsigned short in) { return false; } + + //! Set hit's time + //! TODO: Define unit + /*! + \param in double. Hit's time + */ + virtual bool SetTime(UNUSED_PARAM double in) { return false; } + + //! Set hit's charge + //! TODO: Define unit + /*! + \param in double. Hit's charge + */ + virtual bool SetCharge(UNUSED_PARAM double in) { return false; } + + ClassDef(HKHit, 1); //!< Class definition for ROOT +}; + +/************************************************ + * HKHitsCollection base class definition + * + * Collection class wrapping a vector of HKHit + ***********************************************/ +class HKHitsCollection : public HKObjectCollection { + public: + + HKHitsCollection() {} //!< Creator + + virtual ~HKHitsCollection() = 0; //!< Destructor + virtual void Reset() = 0; //!< Reset function, called after each event + virtual unsigned int GetVersion() const = 0; //!< Return the contained class version number + + static std::string GetBaseNameStatic() { + return "HKHitsCollection"; + } //!< Static function. Return the class base name + + std::string GetBaseName() const { + return HKHitsCollection::GetBaseNameStatic(); + } //!< Return the class base name + + //! operator [] + /*! + \param i integer. Entry index + */ + virtual HKHit* operator[](UNUSED_PARAM int i) { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); + } + + //! Wrapper for std::vector at() + /*! + \param i integer. Entry index + */ + virtual HKHit* at(UNUSED_PARAM int i) { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); + } + + virtual const HKHit* at(UNUSED_PARAM int i) const { + return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); + } + + //! Wrapper for std::vector back() + virtual HKHit* back() { return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, NULL); } + + //! Wrapper for std::vector size() + virtual size_t size() const { return HKFormatError::ThrowErrorVersion(__PRETTY_FUNCTION__, 0); } + + //! Wrapper for std::vector resize() + /*! + \param entries integer. Number of entries + */ + virtual void resize(UNUSED_PARAM size_t entries) { return; } + + //! Wrapper for std::vector push_back() + /*! + \param in HKHit*. Object to be inserted + */ + virtual void push_back(UNUSED_PARAM HKHit* in) { return; } + + //! Wrapper for std::vector emplace_back() + /*! + \param in HKHit*. Object to be inserted + */ + virtual void emplace_back(UNUSED_PARAM HKHit* in) { return; } + + //! Emplace back an HKHit instance in the collection + virtual void AddHit() { return; } + + private: + +#ifndef HK_USE_ROOT7 + //! Return the pointor to the unique_ptr + virtual std::unique_ptr>>* GetVectorUniquePointor() { + return HKFormatError::ThrowErrorVersion>>*>( + __PRETTY_FUNCTION__, + NULL); + } + + //! Return the pointor to the vector + virtual std::vector>* GetVectorPointor() { + return HKFormatError::ThrowErrorVersion>*>(__PRETTY_FUNCTION__, + NULL); + } + + template friend class HKBranchManagerCollection; +#endif + + ClassDef(HKHitsCollection, 1); //!< Class definition for ROOT +}; + +/************************************************ + * HKHitsCollectionTemplate definition + * + * Collection class wrapping a vector of HKHit + ***********************************************/ +template class HKHitsCollectionT : public HKHitsCollection { + + static_assert(std::is_base_of::value, "T must inherit from HKHit"); + + public: + + HKHitsCollectionT(bool initialize = true) { + if(initialize) { +#ifdef HK_USE_ROOT7 + contents = std::make_shared>>(); +#else + contents = std::make_unique>>(); +#endif + this->Reset(); + } + } //!< Creator + + //! Constructor. Initialize the collection with specified number of entries + /*! + \param entries integer. Number of entries + */ + HKHitsCollectionT(size_t entries) { + this->Reset(); + this->resize(entries); + } + + ~HKHitsCollectionT() { this->Reset(); } //!< Destructor + + void Reset() override { + contents->clear(); + } //!< Reset function, called after each event. Clear the collection. + + static constexpr unsigned int GetVersionStatic() { + return T::GetVersionStatic(); + } //!< Static function. Return the contained class version number + + unsigned int GetVersion() const override { + return T::GetVersionStatic(); + } //!< Return the contained class version number + + //! operator [] + /*! + \param i integer. Entry index + */ + HKHit* operator[](int i) override { return (*contents)[i].get(); } // Use at(i) here too? + + //! Wrapper for std::vector at() + /*! + \param i integer. Entry index + */ + HKHit* at(int i) override { return contents->at(i).get(); } + + const HKHit* at(int i) const { return contents->at(i).get(); } + + //! Wrapper for std::vector back() + HKHit* back() override { return contents->back().get(); } + + //! Wrapper for std::vector size() + size_t size() const override { return contents->size(); } + + //! Wrapper for std::vector resize() + /*! + \param entries integer. Number of entries + */ + void resize(size_t entries) override { +#ifdef HK_USE_ROOT7 + for (auto& ptr : *contents) { ptr = std::make_unique(); } +#else + for (auto& ptr : *contents) { ptr = std::make_unique(); } +#endif + } + + //! Wrapper for std::vector push_back() + /*! + \param in HKHit*. Object to be inserted + */ + void push_back(HKHit* in) override { contents->push_back(std::make_unique(in)); } + + //! Wrapper for std::vector emplace_back() + /*! + \param in HKHit*. Object to be inserted + */ + void emplace_back(HKHit* in) override { contents->emplace_back(std::make_unique(in)); } + + //! Emplace back an empty HKHit instance in the collection + void AddHit() override { contents->emplace_back(std::make_unique()); } + + private: + +#ifdef HK_USE_ROOT7 + std::shared_ptr>> contents; //!< Vector of HKHit + + template friend class HKDataMemberCollectionCreator; +#else + //! Return the pointor to the unique_ptr + std::unique_ptr>>* GetVectorUniquePointor() override { + return &contents; + } + + //! Return the pointor to the vector + std::vector>* GetVectorPointor() override { return contents.get(); } + + std::unique_ptr>> contents; //!< Vector of HKHit + + template friend class HKBranchManagerCollection; +#endif + + ClassDefOverride(HKHitsCollectionT, 1); //!< Class definition for ROOT +}; + +#endif diff --git a/leaf/DataModel-lite/HKHitV1.cpp b/leaf/DataModel-lite/HKHitV1.cpp new file mode 100644 index 0000000..60d35d3 --- /dev/null +++ b/leaf/DataModel-lite/HKHitV1.cpp @@ -0,0 +1,21 @@ +#include "HKHitV1.hpp" + +HKHitV1::HKHitV1() { + this->Reset(); +} + +HKHitV1::HKHitV1(HKHit* in) { + m_pmt_number = in->GetPMTNumber(); + m_time = in->GetTime(); + m_charge = in->GetCharge(); +} + +HKHitV1::~HKHitV1() { + this->Reset(); +} + +void HKHitV1::Reset() { + m_pmt_number = 0; + m_time = 0; + m_charge = 0; +} diff --git a/leaf/DataModel-lite/HKHitV1.hpp b/leaf/DataModel-lite/HKHitV1.hpp new file mode 100644 index 0000000..2e98263 --- /dev/null +++ b/leaf/DataModel-lite/HKHitV1.hpp @@ -0,0 +1,100 @@ +#ifndef HK_HIT_V1_HPP +#define HK_HIT_V1_HPP + +/************************************************ + * HKHitV1.hpp + * - Author: Guillaume Pronost + * - Date: 2024/07/10 + ***********************************************/ + +#include "HKHit.hpp" + +/************************************************ + * HKHitV1 class definition + * + * Version 1 of HKHit + * + * Store PMT hit information + ***********************************************/ +class HKHitV1 : public HKHit { + public: + + HKHitV1(); //!< Creator + + //! Create HKHitV1 from an HKHit pointor + /*! + \param in HKHit* + */ + HKHitV1(HKHit* in); + + ~HKHitV1() override; //!< Destructor + void Reset() override; //!< Reset function, called after each event + + static constexpr unsigned int GetVersionStatic() { + return 1; + } //!< Static function. Return the class version number + + unsigned int GetVersion() const override { + return HKHitV1::GetVersionStatic(); + } //!< Return the class version number + + // v1 functions + //! Return PMT number + /*! + \return unsigned short. PMT number + */ + unsigned short GetPMTNumber() const override { return m_pmt_number; } + + //! Return hit's time + //! TODO: Define unit + /*! + \return double. Hit's time + */ + double GetTime() const override { return m_time; } + + //! Return hit's charge + //! TODO: Define unit + /*! + \return double. Hit's charge + */ + double GetCharge() const override { return m_charge; } + + //! Set PMT number + /*! + \param in unsigned short. PMT number + */ + bool SetPMTNumber(unsigned short in) override { + m_pmt_number = in; + return true; + } + + //! Set hit's time + //! TODO: Define unit + /*! + \param in double. Hit's time + */ + bool SetTime(double in) override { + m_time = in; + return true; + } + + //! Set hit's charge + //! TODO: Define unit + /*! + \param in double. Hit's charge + */ + bool SetCharge(double in) override { + m_charge = in; + return true; + } + + private: + + unsigned short m_pmt_number; //!< PMT number + double m_time; //!< Hit's time + double m_charge; //!< Hit's charge + + ClassDefOverride(HKHitV1, 1); //!< Class definition for ROOT +}; + +#endif diff --git a/leaf/DataModel-lite/HKObject.cpp b/leaf/DataModel-lite/HKObject.cpp new file mode 100644 index 0000000..40db7c2 --- /dev/null +++ b/leaf/DataModel-lite/HKObject.cpp @@ -0,0 +1,11 @@ +#include "HKObject.hpp" + +/**********************************************************/ + +// pure virtual destructor need implementation +HKObject::~HKObject() {} + +// pure virtual destructor need implementation +HKObjectCollection::~HKObjectCollection() {} + +/**********************************************************/ diff --git a/leaf/DataModel-lite/HKObject.hpp b/leaf/DataModel-lite/HKObject.hpp new file mode 100644 index 0000000..8f2bf7a --- /dev/null +++ b/leaf/DataModel-lite/HKObject.hpp @@ -0,0 +1,142 @@ +#ifndef HK_OBJECT_HPP +#define HK_OBJECT_HPP + +/************************************************ + * HKObject.hpp + * - Author: Guillaume Pronost + * - Date: 2024/07/10 + ***********************************************/ + +#include +#include +#include +#include + +#include + +#include +#include + +#define UNUSED_PARAM __attribute__((unused)) + +namespace HKFormatError { + + //! Generic function to throw a runtime error when a virtual function is called. + /*! + \param function_name string. Name of the virtual function + \param default_value type T. Default value, in case error is catched. + \return type T default_value defined in parameters + */ + template T ThrowErrorVersion(std::string function_name, T default_value) { + throw std::runtime_error(function_name + " doesn't exist in this format version"); + return default_value; + } +} // namespace HKFormatError + +namespace HKFormatTools { + + //! Generic function to generate branch name + /*! + \param suffix string. Suffix in the branch name + \return Branch name + */ + template std::string GetBranchName(std::string suffix = "") { + + std::stringstream stream; + if(suffix == "") { + stream << T::GetBaseNameStatic() << "V" << T::GetVersionStatic(); + } + else { + stream << T::GetBaseNameStatic() << "_" << suffix << "V" << T::GetVersionStatic(); + } + return stream.str(); + } + + //! Generic function to generate registration name + /*! + \param suffix string. Suffix in the branch name + \return Branch name + */ + template std::string GetRegistrationName(std::string suffix = "") { + + std::stringstream stream; + if(suffix == "") { + stream << T::GetBaseNameStatic(); + } + else { + stream << T::GetBaseNameStatic() << "_" << suffix; + } + return stream.str(); + } +} // namespace HKFormatTools + +/**********************************************************/ + +/************************************************ + * HKObject class definition + * + * Default class from which HK data format class derive. + * + * Define the default functions each class should have + ***********************************************/ +class HKObject { + public: + + HKObject() {} //!< Creator + + virtual ~HKObject() = 0; //!< Destructor + + // Each object needs a reset function + virtual void Reset() = 0; //!< Reset function, called after each event + + // Each object needs a version + virtual unsigned int GetVersion() const = 0; //!< Return the class version number + + // Each object needs a name + virtual std::string GetBaseName() const = 0; //!< Return the class base name + + std::string GetBranchName(std::string suffix = "") const { + std::stringstream stream; + if(suffix == "") { + stream << this->GetBaseName() << "V" << this->GetVersion(); + } + else { + stream << this->GetBaseName() << "_" << suffix << "V" << this->GetVersion(); + } + return stream.str(); + } //!< Return the branch name to be used in input/output + + static constexpr bool isCollection() { + return false; + } //!< Static function. Return if the class is a collection (i.e. a vector of object) + + ClassDef(HKObject, 1); //!< Class definition for ROOT +}; + +/************************************************ + * HKObjectCollection class definition + * + * Collection class wrapping a vector of HKObject + * + * Default class from which HK data format collection class derive. + * + * Define the default functions each class should have + ***********************************************/ +class HKObjectCollection : public HKObject { + + public: + + HKObjectCollection() {} //!< Creator + + virtual ~HKObjectCollection() = 0; //!< Destructor + virtual void Reset() = 0; //!< Reset function, called after each event + virtual unsigned int GetVersion() const = 0; //!< Return the contained class version number + + static constexpr bool isCollection() { + return true; + } //!< Static function. Return if the class is a collection (i.e. a vector of object) + + ClassDef(HKObjectCollection, 1); //!< Class definition for ROOT +}; + +#endif diff --git a/leaf/DataModel-lite/HitCollection.cpp b/leaf/DataModel-lite/HitCollection.cpp deleted file mode 100644 index 1ec2219..0000000 --- a/leaf/DataModel-lite/HitCollection.cpp +++ /dev/null @@ -1,193 +0,0 @@ -#include "HitCollection.h" - -template -HitCollection::HitCollection() { - timestamp = 0; - timestamp_last = 0; - first_unique = 0; -} - -template -HitCollection::~HitCollection() { - this->Clean(); -} - -template -void HitCollection::Clean() { - timestamp = 0; - timestamp_last = 0; - first_unique = 0; - hits.clear(); -} - -template -void HitCollection::SortByTime() { - std::sort(hits.begin(), hits.end(), typename T::SortFunctor_HitTime()); -} - -template -void HitCollection::SortByTimeOfFlight() { - std::sort(hits.begin(), hits.end(), typename T::SortFunctor_HitTimeOfFlight()); -} - -template<> -void HitCollection::SortByTimeOfFlight() { - //do nothing - //std::cout << "FATAL: Call HitCollection::SortByTimeOfFlight() for type: " << typeid(T).name() << std::endl; -} - - -template -void HitCollection::Append(const HitCollection lHC) { - // Adapted from SubSample append by T.Dealtry - - if ( hits.size() == 0 ) { - // First time HitCollection is filled - hits = lHC.hits; - timestamp = lHC.timestamp; - timestamp_last = lHC.timestamp_last; - first_unique = lHC.first_unique; - } - else { - // Need to shift the hit times by the difference of timestamp offsets - TimeDelta::short_time_t time_shift = (lHC.timestamp - timestamp) / TimeDelta::ns; - - for ( unsigned int i = 0; i < lHC.Size(); i++ ) { - T lHit = lHC.hits[i]; - lHit.T += time_shift; - - hits.push_back( lHit ); - } - - this->SortByTime(); - - timestamp_last = timestamp + TimeDelta(hits.back().T); - } -} - -template -void HitCollection::ResetTimes() { - - // Check if there is any hit - if ( hits.size() == 0 ) { - //std::cout << "Empty collection" << std::endl; - return; - } - - this->SortByTime(); - - // Get First Hit time - TimeDelta::short_time_t time_shift = hits[0].T; - - // Check if time already reset - if ( time_shift == 0 ) { - //std::cout << "Sorted collection" << std::endl; - // Reset only timestamp_last: - timestamp_last = timestamp + TimeDelta(hits.back().T); - return; - } - - timestamp += time_shift; - - for ( unsigned int i = 0; i < hits.size(); i++ ) { - hits[i].T -= time_shift; - } - - timestamp_last = timestamp + TimeDelta(hits.back().T); -} - -template -void HitCollection::CopyCollection(const HitCollection lHC) { - - timestamp = lHC.timestamp; - timestamp_last = lHC.timestamp_last; - first_unique = lHC.first_unique; - - hits.clear(); - - hits.assign( lHC.hits.begin(), lHC.hits.end() ); -} - -template -void HitCollection::CopyCollection(const HitCollection lHC) { - - timestamp = lHC.timestamp; - timestamp_last = lHC.timestamp_last; - first_unique = lHC.first_unique; - - hits.clear(); - - hits.assign( lHC.hits.begin(), lHC.hits.end() ); -} - -template -void HitCollection::CopyDAQ(const HitCollection lHC, const HKAA::DAQType lDAQ) { - - timestamp = lHC.timestamp; - timestamp_last = lHC.timestamp_last; - first_unique = lHC.first_unique; - - hits.clear(); - - // copy: - switch ( lDAQ ) { - case HKAA::kDAQNone: - std::cout << "[ERROR] CopyDAQ: No DAQ selected ! (" << lDAQ << ")" << std::endl; - break; - case HKAA::kIDDAQ_BnL: - //std::cout << "[NOTICE] CopyDAQ: kIDDAQ_BnL (" << lDAQ << ")" << std::endl; - std::copy_if ( lHC.hits.begin(), lHC.hits.end(), std::back_inserter(hits), [](T tHit) { return Astro_GetIfBnLPMT(tHit.PMT); } ); - break; - case HKAA::kIDDAQ_3inch: - //std::cout << "[NOTICE] CopyDAQ: kIDDAQ_3inch (" << lDAQ << ")" << std::endl; - std::copy_if ( lHC.hits.begin(), lHC.hits.end(), std::back_inserter(hits), [](T tHit) { return Astro_GetIfmPMT(tHit.PMT); } ); - break; - case HKAA::kIDDAQ_Hybrid: - //std::cout << "[NOTICE] CopyDAQ: kIDDAQ_Hybrid (" << lDAQ << ")" << std::endl; - std::copy_if ( lHC.hits.begin(), lHC.hits.end(), std::back_inserter(hits), [](T tHit) { return !Astro_GetIfODPMT(tHit.PMT); } ); - break; - case HKAA::kODDAQ: - //std::cout << "[NOTICE] CopyDAQ: kODDAQ (" << lDAQ << ")" << std::endl; - std::copy_if ( lHC.hits.begin(), lHC.hits.end(), std::back_inserter(hits), [](T tHit) { return Astro_GetIfODPMT(tHit.PMT); } ); - break; - default: - std::cout << "[FATAL] CopyDAQ: this DAQ level is unknown ! (" << lDAQ << ")" << std::endl; - exit(0); - } -} - - -template -void HitCollection::FilterTime(const HitCollection lHC, const TimeDelta tStart, const TimeDelta tEnd) { - - timestamp = lHC.timestamp; - timestamp_last = lHC.timestamp_last; - first_unique = lHC.first_unique; - - hits.clear(); - - for ( unsigned int i = 0; i < lHC.hits.size(); i++ ) { - - T tHit = lHC.hits[i]; - TimeDelta tTime = timestamp + TimeDelta(tHit.T); - - - if ( tTime >= tStart && tTime <= tEnd ) - hits.push_back(tHit); - } -} - -template -void HitCollection::SetCollection(const HitCollection lHC) { - - timestamp = lHC.timestamp; - timestamp_last = lHC.timestamp_last; - first_unique = lHC.first_unique; - - hits.clear(); - hits = lHC.hits; -} - -template class HitCollection; -template class HitCollection; - diff --git a/leaf/DataModel-lite/HitCollection.h b/leaf/DataModel-lite/HitCollection.h deleted file mode 100644 index 53673a7..0000000 --- a/leaf/DataModel-lite/HitCollection.h +++ /dev/null @@ -1,63 +0,0 @@ -#ifndef HitCollection_H -#define HitCollection_H - -#include "TimeDelta.h" -#include "HitInfo.h" - -#include - -/** -* \class HitCollection - * - * This class holds PMT hit information (PMT id, time, charge, etc.) - * - * - * $Author: G.Pronost $ - * $Date: 2020/12/09 16:15:00 $ - * Contact: pronost@km.icrr.u-tokyo.ac.jp - * - */ - -template class HitCollection { - - public: - HitCollection(); - virtual ~HitCollection(); - - int first_unique; - TimeDelta timestamp; - TimeDelta timestamp_last; - std::vector hits; - - // Data access functions - T &operator[](int n) { return hits[n]; } - - // Information getter - unsigned int Size() const { return hits.size(); } - T At(int n) const { return hits[n]; } - - // Filler - void Add(T lHit) { hits.push_back(lHit); } - void Append(const HitCollection lHC); - void SetCollection(const HitCollection lHC); - void CopyCollection(const HitCollection lHC); - void CopyCollection(const HitCollection lHC); - void ResetTimes(); - - void CopyDAQ(const HitCollection lHC, const HKAA::DAQType iDAQ); - void FilterTime(const HitCollection lHC, const TimeDelta tStart, const TimeDelta tEnd); - - // Cleaner - void Clean(); - void EraseFirstHit() { hits.erase(hits.begin()); } - - - void SortByTime(); - void SortByTimeOfFlight(); -}; - - - - - -#endif diff --git a/leaf/DataModel-lite/HitInfo.cpp b/leaf/DataModel-lite/HitInfo.cpp deleted file mode 100644 index d778ac4..0000000 --- a/leaf/DataModel-lite/HitInfo.cpp +++ /dev/null @@ -1,83 +0,0 @@ -#include "HitInfo.h" - - -/*************************************************************************/ - -Hit::Hit() { - - PMT = 0; - PMT_original = 0; - - T = 0.; - Q = 0.; -} - - -Hit::~Hit() { } - - -Hit::Hit(int lID, TimeDelta::short_time_t lT, double lQ, const HKAA::PMTType lPMT) { - - PMT = lID + HKAA::kPMTId_Shift[lPMT]; - PMT_original = lID; - - T = lT; - Q = lQ; -} - -// Sort functor following Hit Time -bool Hit::SortFunctor_HitTime::operator() ( - const Hit &a, - const Hit &b) const { - - double ta = a.T; - double tb = b.T; - - return ta < tb; -} - -/*************************************************************************/ - -HitExtended::HitExtended() { - - distance = 0; - ToF = 0; - - NormX = 0; - NormY = 0; - NormZ = 0; - - theta = 0; - phi = 0; -} - -HitExtended::HitExtended(const Hit& tHit) { - - - PMT = tHit.PMT; - PMT_original = tHit.PMT_original; - T = tHit.T; - Q = tHit.Q; - - distance = 0; - ToF = 0; - - NormX = 0; - NormY = 0; - NormZ = 0; - - theta = 0; - phi = 0; -} - -// Sort functor following Hit Time -bool HitExtended::SortFunctor_HitTimeOfFlight::operator() ( - const HitExtended &a, - const HitExtended &b) const { - - double ta = a.ToF; - double tb = b.ToF; - - return ta < tb; -} - diff --git a/leaf/DataModel-lite/HitInfo.h b/leaf/DataModel-lite/HitInfo.h deleted file mode 100644 index 0603dde..0000000 --- a/leaf/DataModel-lite/HitInfo.h +++ /dev/null @@ -1,64 +0,0 @@ -#ifndef HitInfo_H -#define HitInfo_H - -#include "Environments.h" -#include "TimeDelta.h" - -#include - -/** -* \class HitInfo - * - * This class holds PMT hit information (PMT id, time, charge, etc.) - * - * - * $Author: G.Pronost $ - * $Date: 2020/11/13 16:15:00 $ - * Contact: pronost@km.icrr.u-tokyo.ac.jp - * - */ - -class Hit { - - public: - Hit(); - Hit(int lID, TimeDelta::short_time_t lT, double lQ, const HKAA::PMTType lPMT=HKAA::kIDPMT_BnL); - ~Hit(); - - int PMT; - int PMT_original; - TimeDelta::short_time_t T; - double Q; - - // Sort functor following hit time after resolution applied - struct SortFunctor_HitTime { - bool operator() (const Hit &a, const Hit &b) const; - }; - -}; - -class HitExtended : public Hit { - - - public: - HitExtended(); - HitExtended(const Hit&); - - double distance; - double ToF; - - double NormX; - double NormY; - double NormZ; - - double theta; - double phi; - - // Sort functor following hit time of flight after resolution applied - struct SortFunctor_HitTimeOfFlight { - bool operator() (const HitExtended &a, const HitExtended &b) const; - }; - -}; - -#endif diff --git a/leaf/DataModel-lite/LinkDef.h b/leaf/DataModel-lite/LinkDef.h new file mode 100644 index 0000000..d9649e9 --- /dev/null +++ b/leaf/DataModel-lite/LinkDef.h @@ -0,0 +1,36 @@ +#ifdef __CINT__ + + #pragma link off all globals; + #pragma link off all classes; + #pragma link off all functions; + + #pragma link C++ namespace HKFormatError + ; + + #pragma link C++ class vector < unsigned int> + ; + #pragma link C++ class map < string, unsigned int> + ; + + #pragma link C++ class HKObject + ; + #pragma link C++ class HKObjectCollection + ; + + #pragma link C++ class HKHit + ; + #pragma link C++ class HKHitV1 + ; + + #pragma link C++ class HKHitsCollection + ; + #pragma link C++ class vector < unique_ptr < HKHit, default_delete < HKHit>>> + ; + #pragma link C++ class vector < shared_ptr < HKHit >> + ; + #pragma link C++ class HKHitsCollectionT < HKHitV1> + ; + + #pragma link C++ class HKGeometryPMT + ; + #pragma link C++ class HKGeometryPMTV1 + ; + #pragma link C++ class HKGeometryPMTCollection + ; + #pragma link C++ class vector < unique_ptr < HKGeometryPMT, default_delete < HKGeometryPMT>>> + ; + #pragma link C++ class vector < shared_ptr < HKGeometryPMT >> + ; + #pragma link C++ class HKGeometryPMTCollectionT < HKGeometryPMTV1> + ; + + #pragma link C++ class HKGeometry + ; + #pragma link C++ class HKGeometryV1 + ; + + #pragma link C++ class HKDarkNoise+; + #pragma link C++ class HKDarkNoiseV1+; + +#endif diff --git a/leaf/DataModel-lite/PMTInfo.cpp b/leaf/DataModel-lite/PMTInfo.cpp deleted file mode 100644 index b5035ba..0000000 --- a/leaf/DataModel-lite/PMTInfo.cpp +++ /dev/null @@ -1,145 +0,0 @@ -#include "PMTInfo.h" - -PMTInfo::PMTInfo() { - - Id_original = 0; - Id = 0; - Type = HKAA::kIDPMT_BnL; - - // PMT Position initialization - Position[0] = 0.; - Position[1] = 0.; - Position[2] = 0.; - - // PMT Direction initialization - Orientation[0] = 0.; - Orientation[1] = 0.; - Orientation[2] = 0.; - - // mPMT related info initialization - mPMT = false; - mPMT_TubeNum = 0; - mPMT_Group = 0; - mPMT_RefTube = 0; - - // Mask information - Masked = true; -} - -PMTInfo::~PMTInfo() { - mPMT_RefX.clear(); - mPMT_RefY.clear(); - mPMT_RefZ.clear(); -} - -void PMTInfo::Setup_mPMT() { - - mPMT_RefTube = Id + ( HKAA::kmPMT_TopID - mPMT_TubeNum ); - - if ( mPMT_TubeNum <= 12 ) mPMT_Group = 0; - else if ( mPMT_TubeNum <= 18) mPMT_Group = 1; - else mPMT_Group = HKAA::kmPMT_Groups - 1; -} - -void PMTInfo::Setup_mPMT_Referencial(const PMTInfo lTopPMT) { - // Function from LEAF - - if ( lTopPMT.Id != mPMT_RefTube ) { - std::cout << " Incorrect reference tube send to setup mPMT Referencial " << std::endl; - return; - } - - mPMT_RefX.resize(3.,0); - mPMT_RefY.resize(3.,0); - mPMT_RefZ.resize(3.,0); - - double dEX[3]; - double dEY[3]; - double dEZ[3]; - - //1. Get the PMT Orientation - dEZ[0] = Orientation[0]; - dEZ[1] = Orientation[1]; - dEZ[2] = Orientation[2]; - - // Save in global - mPMT_RefZ[0] = dEZ[0]; - mPMT_RefZ[1] = dEZ[1]; - mPMT_RefZ[2] = dEZ[2]; - - /* - if ( VERBOSE >= 3 ) { - std::cout << " PMT number = " << Id << ", Tube number in mPMT = " << mPMT_TubeNum << std::endl; - std::cout << " Position of the PMT = " << Position[0] << ", " << Position[1] << ", " << Position[1] << std::endl; - std::cout << " Orientation of the PMT = " << mPMT_RefZ[0] << ", " << mPMT_RefZ[1] << ", " << mPMT_RefZ[2] << std::endl; - } - */ - - //2. Get the mPMT position and its direction -> Provides orgin and first axis of the referential (ez) - //We do not have the mPMT position, so we will take the position of the PMT on top of the mPMT: number: - - /* - if ( VERBOSE >= 3 ) { - std::cout << " Top PMT number = " << lTopPMT.Id << ", Tube number in mPMT = " << lTopPMT.mPMT_TubeNum << std::endl; - std::cout << " Position of the PMT = " << lTopPMT.Position[0] << ", " << lTopPMT.Position[1] << ", " << lTopPMT.Position[2] << std::endl; - std::cout << " Orientation of the PMT = " << lTopPMT.Orientation[0] << ", " << lTopPMT.Orientation[1] << ", " << lTopPMT.Orientation[2] << std::endl; - } - */ - - //3. Define the second vector of the referential in the orthogonal plane to ez and colinear to the center PMT -> Active PMT plane - //a. PMT position in the mPMT referencec plane - - double dPosition_Ref[3]; - - dPosition_Ref[0] = Position[0] - lTopPMT.Position[0]; - dPosition_Ref[1] = Position[1] - lTopPMT.Position[1]; - dPosition_Ref[2] = Position[2] - lTopPMT.Position[2]; - - GeoTools::Normalize(dPosition_Ref); - - /* - if ( VERBOSE >= 3 ) { - std::cout << " PMT position in the mPMT referential = " << dPosition_Ref[0] << ", " << dPosition_Ref[1] << ", " << dPosition_Ref[2] << std::endl; - } - */ - - //b. Define ey, perpendicular to ez and the center PMT -> Active PMT vector - - GeoTools::CrossProduct(dEZ,dPosition_Ref,dEY); - GeoTools::Normalize(dEY); - - // Save in global - mPMT_RefY[0] = dEY[0]; - mPMT_RefY[1] = dEY[1]; - mPMT_RefY[2] = dEY[2]; - /* - if ( VERBOSE >= 3 ) { - std::cout << " Scal product test = " << TMath::ACos(Astro_GetScalarProd(mPMT_RefY,mPMT_RefZ))*180/TMath::Pi() << ", " - << Astro_GetLength(mPMT_RefY) << ", " << Astro_GetLength(mPMT_RefZ) << std::endl; - } - */ - - //4. And then create ex which should be orthogonal to the 2 remaining vectors - GeoTools::CrossProduct(dEY,dEZ,dEX); - - // Save in global - mPMT_RefX[0] = dEX[0]; - mPMT_RefX[1] = dEX[1]; - mPMT_RefX[2] = dEX[2]; - - //As ez and ey should be unitary, so does ex. So, let's check as a debug - - double dLengthX = Astro_GetLength(mPMT_RefX); - // dLengthX is not exactly 1 let's allow +/- 1e-6 - - if ( dLengthX > (1 + 1e-6) || dLengthX < (1 - 1e-6) ) { - std::cout << " There is a referential vector unitarity issue for PMT " << Id << ", length is = "<< dLengthX << std::endl; - - std::cout << " \t Top PMT number = " << lTopPMT.Id << ", Tube number in mPMT = " << lTopPMT.mPMT_TubeNum << std::endl; - std::cout << " \t PMT position in the mPMT referential = " << dPosition_Ref[0] << ", " << dPosition_Ref[1] << ", " << dPosition_Ref[2] << std::endl; - std::cout << " \t Scalar product test = " << TMath::ACos(Astro_GetScalarProd(mPMT_RefY,mPMT_RefZ))*180/TMath::Pi() << ", Y: " - << Astro_GetLength(mPMT_RefY) << ", Z: " << Astro_GetLength(mPMT_RefZ) << std::endl; - } - - //Conclusion: we now have our referential. -} diff --git a/leaf/DataModel-lite/PMTInfo.h b/leaf/DataModel-lite/PMTInfo.h deleted file mode 100644 index 3bd8e07..0000000 --- a/leaf/DataModel-lite/PMTInfo.h +++ /dev/null @@ -1,64 +0,0 @@ -#ifndef PMTInfo_H -#define PMTInfo_H - -#include - -#include "Environments.h" -#include "GeoTools.h" - -//ROOT Headers -#include "TMath.h" - -/** -* \class PMTInfo - * - * This class holds PMT geometry information (PMT type, position, direction, etc.) - * - * - * $Author: G.Pronost $ - * $Date: 2020/11/13 16:15:00 $ - * Contact: pronost@km.icrr.u-tokyo.ac.jp - * - */ - - -class PMTInfo { - - friend class Geometry; - - public: - PMTInfo(); - virtual ~PMTInfo(); - - int Id_original; - int Id; - - int Type; - - double Position[3]; - double Orientation[3]; - - bool mPMT; - - int mPMT_Num; - int mPMT_TubeNum; - int mPMT_Group; - int mPMT_RefTube; - - bool Masked; - - std::vector mPMT_RefX; - std::vector mPMT_RefY; - std::vector mPMT_RefZ; - - int GetTubeNo() { return Id; } - double GetPosition(int i) { return Position[i]; } - double GetOrientation(int i) { return Orientation[i]; } - - protected: - - void Setup_mPMT(); - void Setup_mPMT_Referencial(const PMTInfo lTopPMT); -}; - -#endif diff --git a/leaf/DataModel-lite/Pos3DT.cpp b/leaf/DataModel-lite/Pos3DT.cpp deleted file mode 100644 index 9912c7e..0000000 --- a/leaf/DataModel-lite/Pos3DT.cpp +++ /dev/null @@ -1,10 +0,0 @@ -#include "Pos3DT.h" - -Pos3DT::Pos3DT() { - Vtx.resize(4,0.); - -} - -Pos3DT::~Pos3DT() { - Vtx.clear(); -} diff --git a/leaf/DataModel-lite/Pos3DT.h b/leaf/DataModel-lite/Pos3DT.h deleted file mode 100644 index 45cbef7..0000000 --- a/leaf/DataModel-lite/Pos3DT.h +++ /dev/null @@ -1,41 +0,0 @@ -#ifndef Pos3DT_H -#define Pos3DT_H - - -#include "Environments.h" - -/** -* \class Pos3DT - * - * This class holds Position information (aims to replace Pos3DT) - * - * - * $Author: G.Pronost $ - * $Date: 2020/12/01 17:12:00 $ - * Contact: pronost@km.icrr.u-tokyo.ac.jp - * - */ - - -class Pos3DT { - public: - Pos3DT(); - ~Pos3DT(); - - - double &operator[](int n) { return Vtx[n]; } - double &x() { return Vtx[0]; } - double &y() { return Vtx[1]; } - double &z() { return Vtx[2]; } - double &t() { return Vtx[3]; } - - double R() { return Astro_GetLength(Vtx); } - double Radius() { return Astro_GetR(Vtx); } - - void clear() { Vtx.clear(); Vtx.resize(4,0.); } - - std::vector Vtx; - -}; - -#endif diff --git a/leaf/DataModel-lite/TimeDelta.cpp b/leaf/DataModel-lite/TimeDelta.cpp deleted file mode 100644 index 88ba399..0000000 --- a/leaf/DataModel-lite/TimeDelta.cpp +++ /dev/null @@ -1,121 +0,0 @@ -#include -#include "TimeDelta.h" - -#ifdef ROOT5 -const double TimeDelta::s_long_time_unit = 1.; -#endif - -TimeDelta::TimeDelta(double naive_ns){ - m_long_time = 0; - m_short_time = naive_ns; - Normalize(); - // Avoid loss of precision when possible - double difference = naive_ns - (*this / ns); - m_short_time += difference; - Normalize(); -} - -TimeDelta::~TimeDelta() { -} - -void TimeDelta::Normalize(){ - long_time_t long_diff = std::floor(m_short_time / s_long_time_unit); - m_long_time += long_diff; - m_short_time -= (long_diff * s_long_time_unit); -} -/* -TimeDelta TimeDelta::operator=(const double& time) { - TimeDelta new_delta(time); - return new_delta; -} -*/ -TimeDelta operator*(const TimeDelta& old_delta, double factor){ - TimeDelta new_delta(old_delta); - new_delta.m_short_time *= factor; - new_delta.m_long_time *= factor; - // Make sure no information of m_long_time is lost - if (factor != 0){ - double remainder = old_delta.m_long_time - new_delta.m_long_time * (1. / factor); - new_delta.m_short_time += (remainder * TimeDelta::s_long_time_unit); - } - new_delta.Normalize(); - return new_delta; -} - -TimeDelta operator*(double factor, const TimeDelta& old_delta){ - return old_delta * factor; -} - -double operator/(const TimeDelta& left_delta, const TimeDelta& right_delta){ - double left_ns = (left_delta.m_long_time * (double)TimeDelta::s_long_time_unit); - left_ns += left_delta.m_short_time; - double right_ns = (right_delta.m_long_time * (double)TimeDelta::s_long_time_unit); - right_ns += right_delta.m_short_time; - return left_ns / right_ns; -} - -TimeDelta operator+(const TimeDelta& left_delta, const TimeDelta& right_delta){ - TimeDelta new_delta(left_delta); - new_delta.m_long_time += right_delta.m_long_time; - new_delta.m_short_time += right_delta.m_short_time; - new_delta.Normalize(); - return new_delta; -} - -TimeDelta operator-(const TimeDelta& left_delta, const TimeDelta& right_delta){ - return left_delta + (-1.0 * right_delta); -} - -TimeDelta& operator+=(TimeDelta& left_delta, const TimeDelta& right_delta){ - left_delta = left_delta + right_delta; - return left_delta; -} - -TimeDelta& operator-=(TimeDelta& left_delta, const TimeDelta& right_delta){ - left_delta = left_delta - right_delta; - return left_delta; -} - -bool operator==(const TimeDelta& left_delta, const TimeDelta& right_delta){ - TimeDelta A(left_delta); - TimeDelta B(right_delta); - A.Normalize(); - B.Normalize(); - return (A.m_long_time == B.m_long_time) and (A.m_short_time == B.m_short_time); -} - -bool operator<(const TimeDelta& left_delta, const TimeDelta& right_delta){ - TimeDelta A(left_delta); - TimeDelta B(right_delta); - A.Normalize(); - B.Normalize(); - return (A.m_long_time < B.m_long_time) - or ((A.m_long_time == B.m_long_time) and (A.m_short_time < B.m_short_time)); -} - -bool operator<=(const TimeDelta& left_delta, const TimeDelta& right_delta){ - return (left_delta < right_delta) or (left_delta == right_delta); -} - -bool operator>(const TimeDelta& left_delta, const TimeDelta& right_delta){ - return not (left_delta <= right_delta); -} - -bool operator>=(const TimeDelta& left_delta, const TimeDelta& right_delta){ - return not (left_delta < right_delta); -} - -bool operator!=(const TimeDelta& left_delta, const TimeDelta& right_delta){ - return not (left_delta == right_delta); -} - -std::ostream& operator<<(std::ostream& outs, const TimeDelta& delta){ - return outs << delta.m_short_time + delta.m_long_time*TimeDelta::s_long_time_unit << " ns"; -} - -// Unit constants -const TimeDelta TimeDelta::ps = TimeDelta(0.001); -const TimeDelta TimeDelta::ns = TimeDelta(1.0); -const TimeDelta TimeDelta::us = 1000 * TimeDelta::ns; -const TimeDelta TimeDelta::ms = 1000 * TimeDelta::us; -const TimeDelta TimeDelta::s = 1000 * TimeDelta::ms; diff --git a/leaf/DataModel-lite/TimeDelta.h b/leaf/DataModel-lite/TimeDelta.h deleted file mode 100644 index 0745ffd..0000000 --- a/leaf/DataModel-lite/TimeDelta.h +++ /dev/null @@ -1,98 +0,0 @@ -#ifndef TIMEDELTA_H -#define TIMEDELTA_H - -#include -#include -#include - -/// Universal data type to store both large scale (unix timestamp) and short scale (hit times) time differences -/// -/// The type and unit of the long and short scale times are optimised to allow -/// SubSamples to store many hits relative to a single timestamp efficiently. -/// -/// The base unit for physics simulations (at least in WCSim) is ns, that that -/// is the chosen as the unit for the short time scales. A signed 64 bit -/// integer can hold values up to 2^63 = 9.22337204 x 10^18, which in ns would -/// be more than 290 years. Long enough for the long time deltas. -/// -/// When creating TimeDelta instances, one should use the unit constants to do so: -/// -/// TimeDelta t = 0.7 * TimeDelta::ms; -/// -/// To convert into a naive floating point value, use them with the division operator: -/// -/// double naive_t = t / TimeDelta::us; -/// -class TimeDelta { - - public: - /// Default constructor (sets all times to 0) - TimeDelta() : m_short_time(0), m_long_time(0){}; - #ifndef ROOT5 - /// Copy constructor (just copies all member variables) - TimeDelta(const TimeDelta&) = default; - #else - /// Copy constructor (just copies all member variables) - TimeDelta(const TimeDelta& a) { m_short_time = a.m_short_time; m_long_time = a.m_long_time; } - #endif - /// Constructor from naive floating point value (in ns) - TimeDelta(double naive_ns); - - ~TimeDelta(); - - /// Type for relative hit times within a SubSample. Unit = ns - typedef double short_time_t; - /// Type for absolute timestamps of a SubSample. Unit = ns - typedef int64_t long_time_t; - - /// Member for short time deltas - short_time_t m_short_time; - /// Member for long time delta - long_time_t m_long_time; - - #ifndef ROOT5 - /// Relative unit of long time member, i.e. long_unit / short_unit, both ns so = 1. - static constexpr double s_long_time_unit = 1.; - #else - static const double s_long_time_unit; - #endif - - /// Ensure that the time difference stored in m_short_time is small and positive. - void Normalize(); - - // Unit constants: - - /// TimeDelta of 1 ps - static const TimeDelta ps; - /// TimeDelta of 1 ns - static const TimeDelta ns; - /// TimeDelta of 1 us - static const TimeDelta us; - /// TimeDelta of 1 ms - static const TimeDelta ms; - /// TimeDelta of 1 s - static const TimeDelta s; - - // Operators - //TimeDelta operator=(const double& time); - -}; - - // Operators - TimeDelta operator*(const TimeDelta& old_delta, double factor); - TimeDelta operator*(double factor, const TimeDelta& old_delta); - double operator/(const TimeDelta& left_delta, const TimeDelta& right_delta); - TimeDelta operator+(const TimeDelta& left_delta, const TimeDelta& right_delta); - TimeDelta operator-(const TimeDelta& left_delta, const TimeDelta& right_delta); - bool operator<(const TimeDelta& left_delta, const TimeDelta& right_delta); - bool operator<=(const TimeDelta& left_delta, const TimeDelta& right_delta); - bool operator>(const TimeDelta& left_delta, const TimeDelta& right_delta); - bool operator>=(const TimeDelta& left_delta, const TimeDelta& right_delta); - bool operator==(const TimeDelta& left_delta, const TimeDelta& right_delta); - bool operator!=(const TimeDelta& left_delta, const TimeDelta& right_delta); - TimeDelta& operator+=(TimeDelta& left_delta, const TimeDelta& right_delta); - TimeDelta& operator-=(TimeDelta& left_delta, const TimeDelta& right_delta); - std::ostream& operator<<(std::ostream& outs, const TimeDelta& delta); - - -#endif diff --git a/leaf/GNUmakefile b/leaf/GNUmakefile index e8194e0..94c9c3f 100755 --- a/leaf/GNUmakefile +++ b/leaf/GNUmakefile @@ -3,138 +3,108 @@ OSNAME = $(shell uname -s) HOST = $(shell uname -n) OSNAMEMODE = $(OSNAME) -SOURCES = LEAF.cc -HEADERS = LEAF.hh +SOURCES = LEAF.cpp +HEADERS = LEAF.hpp include ../Makefile/Makefile.${OSNAME} - - # set compiler options for ROOT CXXFLAGS_BASE += $(shell root-config --cflags) CXXFLAGS_BASE += '-fPIC' -Wall -Wpedantic -Wno-long-long -DDATAMODEL_LITE -CXXFLAGS = $(CXXFLAGS_BASE) -#CXXFLAGS_ROOT = $(CXXFLAGS_BASE) - -# #ifeq ($(shell echo "$(ROOT_VERSION) >= 6.0" | bc -l ), 1) -# ROOT_VERSION = $(shell root-config --version | tr -d '/' ) -# ifeq ($(shell [ $(ROOT_VERSION) -ge 6.0 ] && echo 1 || echo 0), 1) -# MAKE_LEAF_WITH_ROOT6 = 1 -# CXXFLAGS += -std=c++17 -DROOT6 # modifies to c++17 (Nicolas) -# #CXXFLAGS_ROOT += -std=c++11 -DROOT6 -# else -# CXXFLAGS += -std=c++11 -DROOT5 -# endif - -# ROOT_VERSION = $(shell root-config --version | tr -d '/') -# ifeq ($(shell echo "$(ROOT_VERSION) >= 6.0" | bc -l), 1) -# MAKE_LEAF_WITH_ROOT6 = 1 -# CXXFLAGS += -std=c++17 -DROOT6 # modifies to c++17 (Nicolas) -# #CXXFLAGS_ROOT += -std=c++11 -DROOT6 -# else -# CXXFLAGS += -std=c++11 -DROOT5 -# endif - +CXXFLAGS = $(CXXFLAGS_BASE) -std=c++17 ROOT_VERSION = $(shell root-config --version | cut -d'.' -f1,2) ifeq ($(shell echo "$(ROOT_VERSION) >= 6.0" | bc -l), 1) -MAKE_LEAF_WITH_ROOT6 = 1 -CXXFLAGS += -std=c++17 -DROOT6 # modifies to c++17 (Nicolas) -#CXXFLAGS_ROOT += -std=c++11 -DROOT6 -else -CXXFLAGS += -std=c++17 -DROOT5 + HK_USE_ROOT7 = 1 + CXXFLAGS += -DHK_USE_ROOT7 endif INCFLAGS = -I. -I$(shell root-config --incdir) -INCFLAGS += -I$(WCSIM_BUILD_DIR)/include/WCSim -INCFLAGS += -I./DataModel +INCFLAGS += -I../libWCSIM/include +INCFLAGS += -I./DataModel-lite -LIBS += -L${WCSIM_BUILD_DIR}/lib -lWCSimRoot +LIBS += -L../libWCSIM -lWCSimRoot LIBS += $(shell root-config --libs) -lMinuit LIB_DATAMODEL = ../lib/libDataModelLite.so LIB_MANAGER = ../lib/libHKManager.so LIB_FITTER = ../lib/libLEAF.so -LIB_ASTROANALYSIS = ../lib/libHKAstroAnalysis.so +PCM_LIB_DATAMODEL = ../lib/libDataModelLite_rdict.pcm +PCM_DATAMODEL = ../lib/DataModelRootDict_rdict.pcm -LIBS_OBJECTS = $(LIB_DATAMODEL) $(LIB_MANAGER) $(LIB_FITTER) #$(LIB_ASTROANALYSIS) +LIBS_OBJECTS = $(LIB_DATAMODEL) $(LIB_MANAGER) $(LIB_FITTER) -DATAMODEL_SOURCES = $(filter-out DataModel/DataModelRootDict.cpp, $(wildcard DataModel/*.cpp)) -DATAMODEL_SRC_NOP = $(DATAMODEL_SOURCES:DataModel/%=./%) +DATAMODEL_SOURCES = $(filter-out DataModel-lite/DataModelRootDict.cpp, $(wildcard DataModel-lite/*.cpp)) +DATAMODEL_SRC_NOP = $(DATAMODEL_SOURCES:DataModel-lite/%=./%) -DATAMODEL_HEADERS = $(DATAMODEL_SOURCES:%.cpp=%.h) -DATAMODEL_OBJECTS = $(DATAMODEL_SRC_NOP:%.cpp=%.o) #DataModelRootDict.o +DATAMODEL_HEADERS = $(DATAMODEL_SOURCES:%.cpp=%.hpp) +DATAMODEL_OBJECTS = $(DATAMODEL_SRC_NOP:%.cpp=%.o) DataModelRootDict.o - -all: .depend ROOT_CHECK $(LIBS_OBJECTS) +all: .depend ROOT_CHECK DIRECTORY_CHECK $(LIBS_OBJECTS) $(PCM_DATAMODEL) ROOT_CHECK: -ifeq ($(MAKE_LEAF_WITH_ROOT6),1) - @echo '<< compile with ROOT6 >>' +ifeq ($(HK_USE_ROOT7),1) + @echo '<< compile with ROOT7 >>' else - @echo '<< compile with ROOT5 >>' - @echo $(DATAMODEL_SOURCES) - + @echo '<< compile with old ROOT >>' + @echo $(DATAMODEL_SOURCES) endif -# library -$(LIB_FITTER): LeafConfig.o LeafInputs.o LeafUtility.o LeafSplines.o Likelihoods.o LEAF.o +DIRECTORY_CHECK: @if [ ! -d ../lib ]; then \ mkdir ../lib; \ fi + +# library +$(LIB_FITTER): LEAF.o LEAFUtilities.o SplineUtilities.o @echo '<< generating ' $@ ' >>' @$(SOMAKER) $(SOFLAGS) -o $@ $^ $(LIBS) $(LIB_MANAGER): HKManager.o @echo '<< generating ' $@ ' >>' - @if [ ! -d ../lib ]; then \ - mkdir ../lib; \ - fi @$(SOMAKER) $(SOFLAGS) -o $@ $^ $(LIBS) - -$(LIB_ASTROANALYSIS): HKAstroAnalysis.o - @echo '<< generating ' $@ ' >>' - @if [ ! -d ../lib ]; then \ - mkdir ../lib; \ - fi - @$(SOMAKER) $(SOFLAGS) -o $@ $^ $(LIBS) - + #$(LIB_DATAMODEL): DataModelRootDict.cpp $(DATAMODEL_OBJECTS) $(LIB_DATAMODEL): $(DATAMODEL_OBJECTS) @echo '<< generating ' $@ ' >>' - @if [ ! -d ../lib ]; then \ - mkdir ../lib; \ - fi @$(SOMAKER) $(SOFLAGS) -o $@ $(DATAMODEL_OBJECTS) $(LIBS) -# Not needed anymore -#DataModelRootDict.cpp: -# @echo '<< generating ' $@ ' >>' -# @rootcint -f DataModelRootDict.cpp -c $(CXXFLAGS_ROOT) -p $(DATAMODEL_HEADERS) $(PWD)/DataModel/LinkDef.h -#ifeq ($(MAKE_LEAF_WITH_ROOT6),1) -# @echo '<< install DataModelRootDict_rdict.pcm >>' -# @cp DataModelRootDict_rdict.pcm ../lib/. -#endif - # DataModel objects -%.o: DataModel/%.cpp DataModel/%.h +%.o: DataModel-lite/%.cpp DataModel-lite/%.hpp @echo '<< compiling DataModel object: ' $@ ' >>' @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c -o $@ $< +# Generate ROOT dictionary +DataModelRootDict.cpp: $(DATAMODEL_HEADERS) DataModel-lite/LinkDef.h + @echo '<< generating ROOT dictionary >>' + @rootcling -f $@ -rml $(LIB_DATAMODEL) -rmf $(PCM_LIB_DATAMODEL) \ + $(INCFLAGS) $(DATAMODEL_HEADERS) DataModel-lite/LinkDef.h + +DataModelRootDict.o: DataModelRootDict.cpp + @echo '<< compiling ROOT dictionary >>' + @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c -o $@ $< + +$(PCM_DATAMODEL): DataModelRootDict.o + @echo '<< installing ROOT dictionary >>' + @mv DataModelRootDict*.pcm ../lib/. + # default rules .cc.o: @echo '<< compiling' $< '>>' @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c $< +.cpp.o: + @echo '<< compiling' $< '>>' + @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c $< + %.o: %.cc %.hh @echo '<< compiling' $< '>>' @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c -o $@ $< -%.o: %.cpp +%.o: %.cpp %.hpp @echo '<< compiling' $< '>>' @$(CXX) $(CXXFLAGS) $(INCFLAGS) -c -o $@ $< - %.o: %.C %.h @echo '<< compiling' $< '>>' @@ -157,5 +127,7 @@ clean: @rm -f ./*.o @rm -f ./../lib/lib*.so @rm -f ./../lib/*Dict* + @rm -f ./../lib/*.pcm @rm -f ./*Dict* + @rm -f ./*.pcm diff --git a/leaf/HKManager.cc b/leaf/HKManager.cc deleted file mode 100644 index 1795c4c..0000000 --- a/leaf/HKManager.cc +++ /dev/null @@ -1,206 +0,0 @@ -/*********************************************************************************/ -/** HKManager.cc **/ -/** Author: Guillaume Pronost (pronost@km.icrr.u-tokyo.ac.jp) **/ -/** Date: June 17th 2020 **/ -/** Desc: Manage fitter and PMT hits for HK analysis **/ -/*********************************************************************************/ - -#include "HKManager.hh" - -HKManager* HKManager::myManager=NULL; - - -/************************************************************************************************************************/ -WCSimReader::WCSimReader() { - fGeometry = new Geometry(); -} - -WCSimReader::~WCSimReader() { - delete fGeometry; -} -/************************************************************************************************************************/ - -HKManager::HKManager() { - myManager = this; -} - -HKManager::~HKManager() { - - myManager = NULL; - - /* - fPMTDiagDistance = 0; - fPMTBarrelExtremZ_Lo = 0; - fPMTBarrelExtremZ_Hi = 0; - */ -} - -HKManager* HKManager::GetME() { - - if ( myManager ) return myManager; - - myManager = new HKManager(); - return myManager; -} - -/************************************************************************************************************************/ - -void WCSimReader::SetGeometry( WCSimRootGeom * wGeo, double dDarkRate_Normal, double dDarkRate_mPMT ) { - - // Keep RootGeometry event: - fWCGeo = wGeo; - - // Fill Geometry class - fGeometry = new Geometry(); - - fGeometry->detector_radius = fWCGeo->GetWCCylRadius(); - fGeometry->detector_length = fWCGeo->GetWCCylLength(); - - fDarkRate_Normal = dDarkRate_Normal; - fDarkRate_mPMT = dDarkRate_mPMT; - - if ( fDarkRate_Normal == 0 ) { -#ifdef WCSIM_single_PMT_type - fDarkRate_ns[0] = 4200. * fWCGeo->GetWCNumPMT() * 1e-9; - fDarkRate_ns[1] = 0; -#else - fDarkRate_ns[0] = 8400. * fWCGeo->GetWCNumPMT(false) * 1e-9; - fDarkRate_ns[1] = 100. * fWCGeo->GetWCNumPMT(true ) * 1e-9 * HKAA::kmPMT_TopID; -#endif - } - else { -#ifdef WCSIM_single_PMT_type - fDarkRate_ns[0] = fDarkRate_Normal * fWCGeo->GetWCNumPMT() * 1e-9; - fDarkRate_ns[1] = 0; -#else - - int iNbr_Norm = fWCGeo->GetWCNumPMT(false); - int iNbr_mPMT = fWCGeo->GetWCNumPMT(true ); - - fDarkRate_ns[0] = fDarkRate_Normal * iNbr_Norm * 1e-9; - fDarkRate_ns[1] = fDarkRate_mPMT * iNbr_mPMT * 1e-9 * HKAA::kmPMT_TopID; -#endif - } - - fGeometry->pmt_dark_rate[HKAA::kIDPMT_BnL] = fDarkRate_ns[0]; - fGeometry->pmt_dark_rate[HKAA::kIDPMT_3inch] = fDarkRate_ns[1]; - - - this->LoadPMTInfo(); -} - -/************************************************************************************************************************/ -// Geometry functions: - -void WCSimReader::LoadPMTInfo() { - - // Get number of PMTs - -#ifdef WCSIM_single_PMT_type - int iNbr_Norm = fWCGeo->GetWCNumPMT(); - int iNbr_mPMT = 0; -#else - int iNbr_Norm = fWCGeo->GetWCNumPMT(false); - int iNbr_mPMT = fWCGeo->GetWCNumPMT(true); -#endif - - // Fill geometry object - fGeometry->pmt_num[HKAA::kIDPMT_BnL] = iNbr_Norm; - fGeometry->pmt_num[HKAA::kIDPMT_3inch] = iNbr_mPMT; - - - std::cout << " LEAF setting # PMT = " << iNbr_Norm << std::endl; - std::cout << " LEAF setting # mPMT = " << iNbr_mPMT << std::endl; - - WCSimRootPMT wPMT; - /* - double dMin1 = 0; - double dMin2 = 0; - */ - - // Normal PMTs - for ( int iPMT=0; iPMT < iNbr_Norm; iPMT++ ) { - -#ifdef WCSIM_single_PMT_type - wPMT = fWCGeo->GetPMT(iPMT); -#else - wPMT = fWCGeo->GetPMT(iPMT,false); -#endif - - PMTInfo lPMTInfo; - - lPMTInfo.Id = wPMT.GetTubeNo(); - - lPMTInfo.Type = HKAA::kIDPMT_BnL; - lPMTInfo.mPMT = false; - - for ( int j=0; j < 3; j++ ) { - lPMTInfo.Position[j] = wPMT.GetPosition(j); - lPMTInfo.Orientation[j] = wPMT.GetOrientation(j); - } - /* - if ( lPMTInfo.Position[2] < dMin1 ) { - dMin2 = dMin1; - dMin1 = lPMTInfo.Position[2]; - } - - if ( lPMTInfo.Position[2] > dMin1 && lPMTInfo.Position[2] < dMin2 ) { - dMin2 = lPMTInfo.Position[2]; - } - */ - - fGeometry->AddPMTInfo(lPMTInfo); - } - - - // Compute minimal diagonal distance: - // PMT 4 is the closest diag PMT from PMT 0 - /* - PMTInfo lInfoIdxZero = fGeometry->PMTList.at(HKAA::kID)[0]; - PMTInfo lInfoIdxFour = fGeometry->PMTList.at(HKAA::kID)[4]; - fPMTDiagDistance = std::ceil( sqrt( pow(lInfoIdxZero.Position[0]-lInfoIdxFour.Position[0],2.) - + pow(lInfoIdxZero.Position[1]-lInfoIdxFour.Position[1],2.) - + pow(lInfoIdxZero.Position[2]-lInfoIdxFour.Position[2],2.) ) ); - fPMTDiagDistance += 2.; // Add 2cm for safety - - fPMTBarrelExtremZ_Lo = std::abs( std::ceil(dMin2) ) + 1; - fPMTBarrelExtremZ_Hi = std::abs( std::ceil(dMin1) ) + 1; - */ - -#ifndef WCSIM_single_PMT_type - // mPMTs - - for ( int iPMT=0; iPMT < iNbr_mPMT; iPMT++ ) { - - wPMT = fWCGeo->GetPMT(iPMT,true); - - PMTInfo lPMTInfo; - - lPMTInfo.Id = wPMT.GetTubeNo(); - lPMTInfo.Type = HKAA::kIDPMT_3inch; - lPMTInfo.mPMT = true; - - for ( int j=0; j < 3; j++ ) { - lPMTInfo.Position[j] = wPMT.GetPosition(j); - lPMTInfo.Orientation[j] = wPMT.GetOrientation(j); - } - - lPMTInfo.mPMT_TubeNum = wPMT.GetmPMT_PMTNo(); - - if ( lPMTInfo.mPMT_TubeNum < 0 || lPMTInfo.mPMT_TubeNum > 30 ) { - if ( iPMT > 0 ) { - std::cout << " ERROR mPMT " << iPMT << "("<< lPMTInfo.Id << ") is not defined " - << "(" << lPMTInfo.Position[0] << ", " << lPMTInfo.Position[1] << ", " << lPMTInfo.Position[2] << ") " - << "(" << lPMTInfo.Orientation[0] << ", " << lPMTInfo.Orientation[1] << ", " << lPMTInfo.Orientation[2] << ") " - << lPMTInfo.mPMT_TubeNum << std::endl; - } - lPMTInfo.mPMT_TubeNum = 0; - - } - fGeometry->AddPMTInfo(lPMTInfo); - } - - fGeometry->Setup_mPMTs(); - -#endif -} diff --git a/leaf/HKManager.cpp b/leaf/HKManager.cpp new file mode 100644 index 0000000..76ed0e2 --- /dev/null +++ b/leaf/HKManager.cpp @@ -0,0 +1,194 @@ +/*********************************************************************************/ +/** HKManager.cc **/ +/** Author: Guillaume Pronost (pronost@km.icrr.u-tokyo.ac.jp) **/ +/** Date: June 17th 2020 **/ +/** Desc: Manage fitter and PMT hits for HK analysis **/ +/*********************************************************************************/ + +#include "HKManager.hpp" + +HKManager* HKManager::myManager=NULL; + +#include "HKGeometryV1.hpp" +#include "HKGeometryPMTV1.hpp" +#include "HKDarkNoiseV1.hpp" +#include "HKHitV1.hpp" + +/************************************************************************************************************************/ +WCSimReader::WCSimReader() { +#ifndef HK_USE_ROOT7 + fGeometry = std::make_unique(); + fDarkNoise = std::make_unique(); + fGeometryPMT_ID = std::make_unique>(); + fGeometryPMT_mPMT = std::make_unique>(); + + fHitCollection_ID = std::make_unique>(); + fHitCollection_mPMT = std::make_unique>(); +#else + fGeometry = std::static_pointer_cast(std::make_shared()); + fDarkNoise = std::static_pointer_cast(std::make_shared()); + + fGeometryPMT_ID = std::static_pointer_cast(std::make_shared>()); + fGeometryPMT_mPMT = std::static_pointer_cast(std::make_shared>()); + + fHitCollection_ID = std::static_pointer_cast(std::make_shared>()); + fHitCollection_mPMT = std::static_pointer_cast(std::make_shared>()); +#endif +} + +WCSimReader::~WCSimReader() { +} +/************************************************************************************************************************/ + +HKManager::HKManager() { + myManager = this; +} + +HKManager::~HKManager() { + + myManager = NULL; + + /* + fPMTDiagDistance = 0; + fPMTBarrelExtremZ_Lo = 0; + fPMTBarrelExtremZ_Hi = 0; + */ +} + +HKManager* HKManager::GetME() { + + if ( myManager ) return myManager; + + myManager = new HKManager(); + return myManager; +} + +/************************************************************************************************************************/ + +void WCSimReader::SetGeometry( WCSimRootGeom * wGeo, double dDarkRate_Normal, double dDarkRate_mPMT ) { + + // Keep RootGeometry event: + fWCGeo = wGeo; + + fGeometry->SetInnerTyvekRadius(fWCGeo->GetWCCylRadius()); + fGeometry->SetInnerTyvekTotalHeight(fWCGeo->GetWCCylLength()); + + if ( dDarkRate_Normal == 0 ) { +#ifdef WCSIM_single_PMT_type + fDarkNoise->SetAverageTotalDarkNoisePerNS(PMTType::kID, 4200. * fWCGeo->GetWCNumPMT() * 1e-9); + fDarkNoise->SetAverageTotalDarkNoisePerNS(PMTType::kmPMT, 0.0); +#else + fDarkNoise->SetAverageTotalDarkNoisePerNS(PMTType::kID, 8400. * fWCGeo->GetWCNumPMT(false) * 1e-9); + fDarkNoise->SetAverageTotalDarkNoisePerNS(PMTType::kmPMT, 100. * fWCGeo->GetWCNumPMT(true ) * 1e-9 * 19); // 19 is the number of 3" PMTs per mPMT +#endif + } + else { +#ifdef WCSIM_single_PMT_type + fDarkNoise->SetAverageTotalDarkNoisePerNS(PMTType::kID, dDarkRate_Normal * fWCGeo->GetWCNumPMT() * 1e-9); + fDarkNoise->SetAverageTotalDarkNoisePerNS(PMTType::kmPMT, 0.0); +#else + + int iNbr_Norm = fWCGeo->GetWCNumPMT(false); + int iNbr_mPMT = fWCGeo->GetWCNumPMT(true ); + + fDarkNoise->SetAverageTotalDarkNoisePerNS(PMTType::kID, dDarkRate_Normal * iNbr_Norm * 1e-9); + fDarkNoise->SetAverageTotalDarkNoisePerNS(PMTType::kmPMT, dDarkRate_mPMT * iNbr_mPMT * 1e-9 * 19); // 19 is the number of 3" PMTs per mPMT +#endif + } + + this->LoadPMTInfo(); +} + +/************************************************************************************************************************/ +// Geometry functions: + +void WCSimReader::LoadPMTInfo() { + + // Get number of PMTs + +#ifdef WCSIM_single_PMT_type + int iNbr_Norm = fWCGeo->GetWCNumPMT(); + int iNbr_mPMT = 0; +#else + int iNbr_Norm = fWCGeo->GetWCNumPMT(false); + int iNbr_mPMT = fWCGeo->GetWCNumPMT(true); +#endif + + // Fill geometry object + fGeometryPMT_ID->resize(iNbr_Norm ? iNbr_Norm+1 : 0); + fGeometryPMT_mPMT->resize(iNbr_mPMT? iNbr_mPMT+1 : 0); + + std::cout << " LEAF setting # PMT = " << iNbr_Norm << std::endl; + std::cout << " LEAF setting # mPMT = " << iNbr_mPMT << std::endl; + + WCSimRootPMT wPMT; + /* + double dMin1 = 0; + double dMin2 = 0; + */ + + // Normal PMTs + std::cout << "Filling ID PMTs" << std::endl; + for ( int iPMT=0; iPMT < iNbr_Norm; iPMT++ ) { +#ifdef WCSIM_single_PMT_type + wPMT = fWCGeo->GetPMT(iPMT); +#else + wPMT = fWCGeo->GetPMT(iPMT,false); +#endif + + int tubeNo = wPMT.GetTubeNo(); + /* + std::cout << iPMT << " " + << "TubeNo: " << wPMT.GetTubeNo() << " " + << "Pos: (" << wPMT.GetPosition(0) << ", " << wPMT.GetPosition(1) << ", " << wPMT.GetPosition(2) << ") " + << "Dir: (" << wPMT.GetOrientation(0) << ", " << wPMT.GetOrientation(1) << ", " << wPMT.GetOrientation(2) << ") " + << std::endl; + + std::cout << " check " << fGeometryPMT_ID->at(iPMT)->GetPMTSoftwareID() << std::endl; + */ + fGeometryPMT_ID->at(tubeNo)->SetPMTSoftwareID(wPMT.GetTubeNo()); + fGeometryPMT_ID->at(tubeNo)->SetType(PMTType::kID); + fGeometryPMT_ID->at(tubeNo)->SetPositionInCm(wPMT.GetPosition(0), wPMT.GetPosition(1), wPMT.GetPosition(2)); + fGeometryPMT_ID->at(tubeNo)->SetOrientation(wPMT.GetOrientation(0), wPMT.GetOrientation(1), wPMT.GetOrientation(2)); + + /* + if ( wPMT.GetPosition(2) < dMin1 ) { + dMin2 = dMin1; + dMin1 = wPMT.GetPosition(2); + } + + if ( wPMT.GetPosition(2) > dMin1 && wPMT.GetPosition(2) < dMin2 ) { + dMin2 = wPMT.GetPosition(2); + } + */ + } + + // Compute minimal diagonal distance: + // PMT 4 is the closest diag PMT from PMT 0 + /* + PMTInfo lInfoIdxZero = fGeometry->PMTList.at(PMTType::kID)[0]; + PMTInfo lInfoIdxFour = fGeometry->PMTList.at(PMTType::kID)[4]; + fPMTDiagDistance = std::ceil( sqrt( pow(lInfoIdxZero.Position[0]-lInfoIdxFour.Position[0],2.) + + pow(lInfoIdxZero.Position[1]-lInfoIdxFour.Position[1],2.) + + pow(lInfoIdxZero.Position[2]-lInfoIdxFour.Position[2],2.) ) ); + fPMTDiagDistance += 2.; // Add 2cm for safety + + fPMTBarrelExtremZ_Lo = std::abs( std::ceil(dMin2) ) + 1; + fPMTBarrelExtremZ_Hi = std::abs( std::ceil(dMin1) ) + 1; + */ + +#ifndef WCSIM_single_PMT_type + // mPMTs + std::cout << "Filling mPMT" << std::endl; + for ( int iPMT=0; iPMT < iNbr_mPMT; iPMT++ ) { + wPMT = fWCGeo->GetPMT(iPMT,true); + int tubeNo = wPMT.GetTubeNo(); + + fGeometryPMT_mPMT->at(tubeNo)->SetPMTSoftwareID(wPMT.GetTubeNo()); + fGeometryPMT_mPMT->at(tubeNo)->SetType(PMTType::kmPMT); + fGeometryPMT_mPMT->at(tubeNo)->SetPositionInCm(wPMT.GetPosition(0), wPMT.GetPosition(1), wPMT.GetPosition(2)); + fGeometryPMT_mPMT->at(tubeNo)->SetOrientation(wPMT.GetOrientation(0), wPMT.GetOrientation(1), wPMT.GetOrientation(2)); + fGeometryPMT_mPMT->at(tubeNo)->SetSubID(wPMT.GetmPMT_PMTNo()); + } +#endif +} diff --git a/leaf/HKManager.hh b/leaf/HKManager.hh deleted file mode 100644 index 4703f54..0000000 --- a/leaf/HKManager.hh +++ /dev/null @@ -1,128 +0,0 @@ -/*********************************************************************************/ -/** HKManager.hh **/ -/** Author: Guillaume Pronost (pronost@km.icrr.u-tokyo.ac.jp) **/ -/** Date: June 17th 2020 **/ -/** Desc: Manage fitter and PMT hits for HK analysis **/ -/*********************************************************************************/ - - -#ifndef HKManager_hh -#define HKManager_hh - -//#include -//#include - -#include -#include -#include -#include -#include - -//WCSim Headers -#include "WCSimRootGeom.hh" - -//ROOT Headers -#include "TMath.h" - -//DataModel informations -#include "Geometry.h" -#include "HitCollection.h" - -//WCSim parameters: -// If using a WCSim version without mPMT implementation -//#define WCSIM_single_PMT_type - -// Number of PMT configuration: -#define NPMT_CONFIGURATION 1 - -class WCSimReader { - - public: - WCSimReader(); - ~WCSimReader(); - - const Geometry* GetGeometry() { return fGeometry; } - void SetGeometry( WCSimRootGeom * wGeo, double dDarkRate_Normal=4200., double dDarkRate_mPMT=100. ); - - const HitCollection* GetHitCollection() { return &fHitCollection; } - // const HitCollection* GetSecondaryHitCollection() { return &fSecondaryHitCollection; } - - // Manage hit info - void ResetHitInfo() { fHitCollection.Clean(); } - // void ResetSecondaryHitInfo() { fSecondaryHitCollection.Clean(); } - void AddHit(double time, double charge, int pmtType, int tubeNumber) { - - // tubeNumber is from 1 to xxx in Hit array - if ( tubeNumber < 1 ) { - std::cout << "ERROR: tubeNumber is below 0 (" << tubeNumber << ")" << std::endl; - } - - Hit hHit (tubeNumber, time, charge, (HKAA::PMTType) pmtType); - - fHitCollection.Add(hHit); - } - - // void AddSecondaryHit(double time, double charge, int pmtType, int tubeNumber) { - - // // tubeNumber is from 1 to xxx in Hit array - // if ( tubeNumber < 1 ) { - // std::cout << "ERROR: tubeNumber is below 0 (" << tubeNumber << ")" << std::endl; - // } - - // Hit hHit (tubeNumber, time, charge, (HKAA::PMTType) pmtType); - - // fSecondaryHitCollection.Add(hHit); - // } - - /* - // Commented on 2020/11/19 by Guillaume: I don't remember their purpose. To be removed? - double GetMinimumDiagonal() { return fPMTDiagDistance; } - double GetBarrelExtremum_Lo() { return fPMTBarrelExtremZ_Lo; } - double GetBarrelExtremum_Hi() { return fPMTBarrelExtremZ_Hi; } - */ - - private: - void LoadPMTInfo(); - - // Geometry - WCSimRootGeom * fWCGeo; - Geometry* fGeometry; - - // HitCollection - HitCollection fHitCollection; - HitCollection fSecondaryHitCollection; - - // Darknoise - double fDarkRate_Normal; - double fDarkRate_mPMT; - - // Darknoise - double fDarkRate_ns[NPMT_CONFIGURATION]; - - // PMT informations - - - - /* - double fPMTDiagDistance; - double fPMTBarrelExtremZ_Hi; - double fPMTBarrelExtremZ_Lo; - */ - -}; - -class HKManager : public WCSimReader { - - public: - HKManager(); - ~HKManager(); - static HKManager* GetME(); - - private: - // WCSim objects: - static HKManager* myManager; - -}; - -#endif - diff --git a/leaf/HKManager.hpp b/leaf/HKManager.hpp new file mode 100644 index 0000000..f4d64e6 --- /dev/null +++ b/leaf/HKManager.hpp @@ -0,0 +1,168 @@ +/*********************************************************************************/ +/** HKManager.hh **/ +/** Author: Guillaume Pronost (pronost@km.icrr.u-tokyo.ac.jp) **/ +/** Date: June 17th 2020 **/ +/** Desc: Manage fitter and PMT hits for HK analysis **/ +/*********************************************************************************/ + + +#ifndef HKManager_hh +#define HKManager_hh + +//#include +//#include + +#include +#include +#include +#include +#include + +//WCSim Headers +#include "WCSimRootGeom.hh" + +//ROOT Headers +#include "TMath.h" + +//DataModel informations +#include "HKGeometry.hpp" +#include "HKDarkNoise.hpp" +#include "HKGeometryPMT.hpp" +#include "HKHit.hpp" +#include "Constants/Suffix.hpp" +#include "Enums/PMTType.hpp" + +//WCSim parameters: +// If using a WCSim version without mPMT implementation +//#define WCSIM_single_PMT_type + +// Number of PMT configuration: +#define NPMT_CONFIGURATION 1 + +class WCSimReader { + + public: + WCSimReader(); + ~WCSimReader(); + +#ifndef HK_USE_ROOT7 + const HKGeometry* GetGeometry() { return fGeometry.get(); } + const HKDarkNoise* GetDarkNoise() { return fDarkNoise.get(); } + const HKGeometryPMTCollection* GetGeometryPMT_ID() { return fGeometryPMT_ID.get(); } + const HKHitsCollection* GetHitCollection_ID() { return fHitCollection_ID.get(); } + const HKGeometryPMTCollection* GetGeometryPMT_mPMT() { return fGeometryPMT_mPMT.get(); } + const HKHitsCollection* GetHitCollection_mPMT() { return fHitCollection_mPMT.get(); } +#else + std::shared_ptr GetGeometry() { return fGeometry; } + std::shared_ptr GetDarkNoise() { return fDarkNoise; } + std::shared_ptr GetGeometryPMT_ID() { return fGeometryPMT_ID; } + std::shared_ptr GetHitCollection_ID() { return fHitCollection_ID; } + std::shared_ptr GetGeometryPMT_mPMT() { return fGeometryPMT_mPMT; } + std::shared_ptr GetHitCollection_mPMT() { return fHitCollection_mPMT; } +#endif + void SetGeometry( WCSimRootGeom * wGeo, double dDarkRate_Normal=4200., double dDarkRate_mPMT=100. ); + + // const HitCollection* GetSecondaryHitCollection() { return &fSecondaryHitCollection; } + + // Manage hit info + void ResetHitInfo() { fHitCollection_ID->Reset(); fHitCollection_mPMT->Reset(); } + // void ResetSecondaryHitInfo() { fSecondaryHitCollection.Clean(); } + + void AddHit_ID(double time, double charge, int tubeNumber) { + // tubeNumber is from 1 to xxx in Hit array + if ( tubeNumber < 1 ) { + std::cout << "ERROR: tubeNumber is below 0 (" << tubeNumber << ")" << std::endl; + } + + fHitCollection_ID->AddHit(); + fHitCollection_ID->back()->SetPMTNumber(tubeNumber); + fHitCollection_ID->back()->SetTime(time); + fHitCollection_ID->back()->SetCharge(charge); + } + + void AddHit_mPMT(double time, double charge, int tubeNumber) { + // tubeNumber is from 1 to xxx in Hit array + if ( tubeNumber < 1 ) { + std::cout << "ERROR: tubeNumber is below 0 (" << tubeNumber << ")" << std::endl; + } + + fHitCollection_mPMT->AddHit(); + fHitCollection_mPMT->back()->SetPMTNumber(tubeNumber); + fHitCollection_mPMT->back()->SetTime(time); + fHitCollection_mPMT->back()->SetCharge(charge); + } + + // void AddSecondaryHit(double time, double charge, int pmtType, int tubeNumber) { + + // // tubeNumber is from 1 to xxx in Hit array + // if ( tubeNumber < 1 ) { + // std::cout << "ERROR: tubeNumber is below 0 (" << tubeNumber << ")" << std::endl; + // } + + // Hit hHit (tubeNumber, time, charge, (HKAA::PMTType) pmtType); + + // fSecondaryHitCollection.Add(hHit); + // } + + /* + // Commented on 2020/11/19 by Guillaume: I don't remember their purpose. To be removed? + double GetMinimumDiagonal() { return fPMTDiagDistance; } + double GetBarrelExtremum_Lo() { return fPMTBarrelExtremZ_Lo; } + double GetBarrelExtremum_Hi() { return fPMTBarrelExtremZ_Hi; } + */ + + private: + void LoadPMTInfo(); + + // WCSim Geometry + WCSimRootGeom * fWCGeo; + +#ifndef HK_USE_ROOT7 + // Geometry + std::unique_ptr fGeometry; + std::unique_ptr fDarkNoise; + std::unique_ptr fGeometryPMT_ID; + std::unique_ptr fGeometryPMT_mPMT; + + // HitCollection + std::unique_ptr fHitCollection_ID; + std::unique_ptr fHitCollection_mPMT; +#else + // Geometry + std::shared_ptr fGeometry; + std::shared_ptr fDarkNoise; + std::shared_ptr fGeometryPMT_ID; + std::shared_ptr fGeometryPMT_mPMT; + + // HitCollection + std::shared_ptr fHitCollection_ID; + std::shared_ptr fHitCollection_mPMT; +#endif + + // PMT informations + + + + /* + double fPMTDiagDistance; + double fPMTBarrelExtremZ_Hi; + double fPMTBarrelExtremZ_Lo; + */ + +}; + +class HKManager : public WCSimReader { + + public: + HKManager(); + ~HKManager(); + static HKManager* GetME(); + + private: + // WCSim objects: + static HKManager* myManager; + +}; + +#endif + diff --git a/leaf/LEAF.cc b/leaf/LEAF.cc deleted file mode 100644 index 9a86454..0000000 --- a/leaf/LEAF.cc +++ /dev/null @@ -1,1215 +0,0 @@ -/*****************************************************************************************************/ -/** LEAF.cc **/ -/** Author: Guillaume Pronost (pronost@km.icrr.u-tokyo.ac.jp) **/ -/** Original author: Benjamin Quilain **/ -/** Date: December 18th 2019 **/ -/** Desc: Low-E Fitter for Hyper-K **/ -/*****************************************************************************************************/ - -#include "LEAF.hh" -#include "TStopwatch.h" - -LEAF *LEAF::myFitter = NULL; -std::mutex mtx; - -/*****************************************************************************************************/ - -LEAF::LEAF() -{ - myFitter = this; -} - -LEAF::~LEAF() -{ - delete fRand; -} - -LEAF *LEAF::GetME() -{ - if (myFitter) - return myFitter; - - myFitter = new LEAF(); - return myFitter; -} - -void LEAF::DeleteME() -{ - if (myFitter) delete myFitter; -} - -FitterOutput LEAF::fOutput; -FitterOutputProps LEAF::fOutputProps; - -/*****************************************************************************************************/ -void LEAF::Initialize(const Geometry *lGeometry) -{ - InitConfig(lGeometry); - InitInputs(lGeometry); - LoadSplines(); - fOutput = NewOutuput(); -} - -//Creates the candidates list after Coarse grid search, refines the fit, then outputs the best candidate -std::vector LEAF::FitDirection(const std::vector& fixedVertexPosition, int nhits, bool searchPrior) -{ - //* Init - TStopwatch timer; - timer.Reset(); - timer.Start(); - double *tLimits = new double[4]; - int verbose = VERBOSE; - for (int i = 0; i < 4; i++) tLimits[i] = 2 * fSearchVtxStep; - - //* Fit first step : find candidates for the direction - std::vector candidates; - if(searchPrior) - { - //* Only one candidate, chosen wisely by computing a mean hit direction for hits within a specified residual time window - std::vector dirPriorCart = FitDirectionQuick(fixedVertexPosition); - std::vector dirPrior = CartesianToPolarNorm(dirPriorCart); - DirectionCandidate priorCandidate; - priorCandidate.theta = dirPrior[0]; - priorCandidate.phi = dirPrior[1]; - candidates.push_back(priorCandidate); - } - else candidates = SearchDirection(fixedVertexPosition, DirTolerance); //* multiple candidates, spread uniformly - - // std::cout << "nb direction candidates : " << candidates.size() << std::endl; - - fOutputProps.Dir_Search_ComputeTime = timer.RealTime(); - TStopwatch timer2; - timer2.Reset(); - timer2.Start(); - - //? To be removed for real data where true dir is not known, no impact on the fit - // fOutput.stepOneContainsTrueDir = (int)ContainsTrueDir(&candidates, fTrueDir); //* a quick check to see if the step one works (one of the candidate is near the true direction) - - if (verbose >= 2) - { - for (unsigned int j = 0; j < candidates.size(); j++) std::cout << "Candidate theta = " << candidates[j].theta * TMath::RadToDeg() << " deg, phi = " << candidates[j].phi * TMath::RadToDeg() << " deg, NLL = " << candidates[j].DNLL << std::endl; - } - - DirectionCandidate finalCandidate = MinimizeDirection(fixedVertexPosition, candidates, nhits, tLimits, verbose); - - fOutput.Dir = PolarToCartesianNorm({finalCandidate.theta, finalCandidate.phi}); - fOutput.DNLL = finalCandidate.DNLL; - - fOutputProps.Dir_Minimize_ComputeTime = timer2.RealTime(); - - return fOutput.Dir; -} - -std::vector LEAF::FitDirectionQuick(const std::vector& fixedVertexPosition) -{ - // Quick fit implementation - - TStopwatch timer; - timer.Reset(); - timer.Start(); - - const std::vector& UsedVertex = fixedVertexPosition; - - std::vector> toHitVectors; - std::vector residuals; - - for(unsigned int i = 0; i < fHitCollection->Size(); i++) - { - std::vector pmtPosition; - std::vector toHit; - - Hit lHit = fHitCollection->At(i); - - int iPMT = lHit.PMT; - PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - - for(int j = 0; j < 3 ; j++) pmtPosition.push_back(lPMTInfo.Position[j]); - for(int j = 0; j < 3 ; j++) toHit.push_back(pmtPosition[j] - UsedVertex[j]); - Normalize(toHit); - for(int j = 0; j < 3 ; j++) toHit[j] *= lHit.Q; // Scale by charge - toHitVectors.push_back(toHit); - - residuals.push_back(ComputeResidualTime(UsedVertex, UsedVertex[3], lHit)); - } - std::vector lf_Dir(3, 0.0); - for(long unsigned int i = 0; i < toHitVectors.size(); i++) - { - if(residuals[i] < fDirectionPDF_maxResidual && residuals[i] > fDirectionPDF_minResidual) for(int j = 0; j < 3; j++) lf_Dir[j] += toHitVectors[i][j]; - } - - fOutput.Dir = Normalize(lf_Dir); - fOutput.Quick_Dir = Normalize(lf_Dir); - fOutputProps.Dir_Quick_Search_ComputeTime = timer.RealTime(); - return fOutput.Dir; -} - -std::vector LEAF::SearchDirection(const std::vector& fixedVertexPosition, int tolerance) -{ - std::vector candidates; - - // Grid search over theta and phi - for (double theta = 0.0; theta <= TMath::Pi(); theta += theta_step) { - for (double phi = -TMath::Pi(); phi < TMath::Pi(); phi += phi_step) { - // Compute NLL for this direction - double DNLL = Likelihoods::Dir_NLL(fHitCollection ,fixedVertexPosition, theta, phi, fHitCollection->Size()); - - // Store the candidate - DirectionCandidate candidate; - candidate.theta = theta; - candidate.phi = phi; - candidate.DNLL = DNLL; - candidates.push_back(candidate); - } - } - - std::sort(candidates.begin(), candidates.end(), [](const DirectionCandidate& a, const DirectionCandidate& b) - { - return a.DNLL < b.DNLL; - }); - - // Select the best candidates based on tolerance - while((int)candidates.size() > tolerance) candidates.pop_back(); - - return candidates; -} - -//MIGRAD Optimization -DirectionCandidate LEAF::MinimizeDirection(const std::vector& fixedVertexPosition, std::vector& candidates, int nhits, double *limits, int verbose) -{ - // Create a minimizer - TFitter* minimizer = new TFitter(7); // 2 parameters: theta, phi - TMinuit* minuit = minimizer->GetMinuit(); - - double arglist[10]; - int err = 0; - double p1 = verbose - 1; - minimizer->ExecuteCommand("SET PRINTOUT", &p1, 1); - minuit->SetErrorDef(1); - - if (verbose < 2) minuit->mnexcm("SET NOWarnings", 0, 0, err); - - arglist[0] = 2; - minuit->mnexcm("SET STR", arglist, 1, err); - minimizer->SetFCN(MinuitDirNLL); - - for (int icand = 0; icand < (int)candidates.size(); icand++) - { - // Set initial parameters and bounds - minimizer->SetParameter(0, "theta", candidates[icand].theta, 2*M_PI/180, 0.0, TMath::Pi()); - minimizer->SetParameter(1, "phi", candidates[icand].phi, 2*M_PI/180, -TMath::Pi(), TMath::Pi()); - minimizer->SetParameter(2, "nhits", nhits, nhits, nhits - 1, nhits + 1); - minimizer->SetParameter(3, "vertex0", fixedVertexPosition[0], 5e-1, -limits[0], limits[0]); - minimizer->SetParameter(4, "vertex1", fixedVertexPosition[1], 5e-1, -limits[1], limits[1]); - minimizer->SetParameter(5, "vertex2", fixedVertexPosition[2], 5e-1, -limits[2], limits[2]); - minimizer->SetParameter(6, "vertex3", fixedVertexPosition[3], 5e-1 / fLightSpeed, -limits[3] / fLightSpeed, limits[3] / fLightSpeed); - - minimizer->FixParameter(2); - minimizer->FixParameter(3); - minimizer->FixParameter(4); - minimizer->FixParameter(5); - minimizer->FixParameter(6); - - // Minimize - arglist[0] = 1e9; // Maximum number of function evaluations - arglist[1] = 1e-1; // Tolerance - minimizer->ExecuteCommand("MIGRAD", arglist, 2 ); - - // Retrieve results - std::vector optimizedParameters(2); - candidates[icand].theta = minimizer->GetParameter(0); // theta - candidates[icand].phi = minimizer->GetParameter(1); // phi - - minuit = minimizer->GetMinuit(); - double fmin, fedm, errdef; - int npar, nparx, istat; - minuit->mnstat(fmin, fedm, errdef, npar, nparx, istat); - - candidates[icand].DNLL = fmin; - } - - std::sort(candidates.begin(), candidates.end(), [](const DirectionCandidate& a, const DirectionCandidate& b) - { - return a.DNLL < b.DNLL; - }); - // Optional: Print the refined parameters - // if (verbose >= 2) std::cout << "After refinement: theta = " << optimizedParameters[0] * TMath::RadToDeg() << " deg, phi = " << optimizedParameters[1] * TMath::RadToDeg() << " deg" << std::endl; - - delete minimizer; - return candidates[0]; // Returns [theta, phi] -} - -/*****************************************************************************************************/ - -// Coarse search of the vertex. It will search in a cylinder having the radius = tankRadius and a height tankHeight. -// The step between each grid points is given by stepSize (in centimeters). -// We give the list of hit information through the 2D array fHitInfo to make the code faster, instead of using the whole code information -// The code provide as an output not only one vertex, but a list of vertices whose likelihood is higher (Negative Log Likelihood is lower). -// The number of output vertices can be chosen using "tolerance". -std::vector> LEAF::SearchVertex(int nhits, int tolerance, bool likelihood, double lowerLimit, double upperLimit, int directionality) -{ - // 2. How to set the search? - // double stepSize = 0.5;//in m - - std::vector tBestReconstructedVertexPosition; - - // double ** reconstructedVertexPosition = new double[4]; - - std::vector tVtxContainer; - - clock_t timeStart = clock(); - - if (fPositionList.size() == 0) - std::cout << " Position list not filled " << std::endl; - if (VERBOSE >= 3) - std::cout << "Number of hits for coarse grid search = " << nhits << std::endl; - for (unsigned int iPos = 0; iPos < fPositionList.size(); iPos++) - { - struct FitPosition hPos; - hPos.Vtx = fPositionList[iPos]; - hPos.NLL = 0; - - if (likelihood) - hPos.NLL = Likelihoods::Vertex_Time_NLL(fHitCollection, hPos.Vtx, nhits, lowerLimit, upperLimit, false, false, directionality); - else - hPos.NLL = Likelihoods::Vertex_Score(fHitCollection, hPos.Vtx, nhits, lowerLimit, upperLimit, false, false, directionality); - - hPos.NLL += fRand->Uniform(0, 1e-5); // Here only not to have exactly the same value for 2NLL, and be removed from map - - tVtxContainer.push_back(hPos); - - if (VERBOSE >= 3) - std::cout << "Test Pos: " << hPos.Vtx[3] << " " << hPos.Vtx[0] << " " << hPos.Vtx[1] << " " << hPos.Vtx[2] << " NLL = " << hPos.NLL << " " << likelihood << std::endl; - -#ifdef VERBOSE_VTX - // if(verbose && distanceToTrue < 2) std::cout << "Distance to true vertex = " << distanceToTrue << ", Probability = " << NLL < tolerance) - { -#ifdef VERBOSE_VTX - std::cout << " Remove end: " << tVtxContainer.back().NLL << " (first " << tVtxContainer[0].NLL << ") " << tVtxContainer.size() << std::endl; -#endif - tVtxContainer.pop_back(); - } - - clock_t timeEndLoop = clock(); - - std::vector> tReconstructedVertexPosition; - for (int iPos = 0; iPos < tolerance; iPos++) - { - std::vector vPos(5, 0.); - struct FitPosition hPos = tVtxContainer[iPos]; - - vPos[0] = hPos.Vtx[0]; - vPos[1] = hPos.Vtx[1]; - vPos[2] = hPos.Vtx[2]; - vPos[3] = hPos.Vtx[3]; - vPos[4] = hPos.NLL; - - tReconstructedVertexPosition.push_back(vPos); - -#ifdef VERBOSE_VTX - std::cout << "NLL = " << hPos.NLL << ", vertex = " << vPos[0] << ", " << vPos[1] << ", " << vPos[2] << ", " << vPos[3] << std::endl; -#endif - } - - clock_t timeEnd = clock(); - - std::cout << "SearchVertex: Time it took for the loop = " << (timeEndLoop - timeStart) / 1e6 << ", time total = " << (timeEnd - timeStart) / 1e6 << " " << fPositionList.size() << std::endl; - - return tReconstructedVertexPosition; -} - -std::vector LEAF::SearchVertex_Main(int nhits, int tolerance, bool likelihood, double lowerLimit, double upperLimit, int directionality) -{ -// 1) Launch threads just like in SearchVertex_Main. - std::vector lThreadList; - std::vector> lOutputFinal; - TStopwatch timer; - timer.Reset(); - timer.Start(); - - int iCand_Step = (int)fPositionList.size()/fThread; - if(iCand_Step < 1) iCand_Step = 1; - - for(int iStart=0; iStart<(int)fPositionList.size(); iStart+=iCand_Step) - { - std::thread tThrd( - &LEAF::SearchVertex_thread, - LEAF::GetME(), - iStart, iCand_Step, - nhits, tolerance, - likelihood, lowerLimit, upperLimit, directionality - ); - lThreadList.push_back(std::move(tThrd)); - } - - // std::cout << "time after vtx search threads in SearchVertex_Main : " << timer.RealTime() << std::endl; - - for(auto &th: lThreadList) if(th.joinable()) th.join(); - - // 2) collect partial results from fThreadOutput. - mtx.lock(); - lOutputFinal = fThreadOutput; // each row: [x, y, z, t, NLL] - fThreadOutput.clear(); - mtx.unlock(); - - // 3) Sort by NLL and keep 'tolerance' number of candidates - std::sort(lOutputFinal.begin(), lOutputFinal.end(), SortOutputVector); - while((int)lOutputFinal.size() > tolerance) lOutputFinal.pop_back(); - - //4) Now build a vector of VtxCandidate. - std::vector finalList; - finalList.reserve(lOutputFinal.size()); - - for(auto &cand : lOutputFinal) - { - VtxCandidate cOut = CreateCandidate(cand, lowerLimit, upperLimit, false); - cOut.NLL = cand[4]; // store the NLL from the coarse search. - finalList.push_back(cOut); - } - - return finalList; -} - -// Coarse search of the vertex. It will search in a cylinder having the radius = tankRadius and a height tankHeight. -// The step between each grid points is given by stepSize (in centimeters). -// We give the list of hit information through the 2D array fHitInfo to make the code faster, instead of using the whole code information -// The code provide as an output not only one vertex, but a list of vertices whose likelihood is higher (Negative Log Likelihood is lower). -// The number of output vertices can be chosen using "tolerance". -void LEAF::SearchVertex_thread(int iStart, int iIte, int nhits, int tolerance, bool likelihood, double lowerLimit, double upperLimit, int directionality) -{ - std::vector tVtxContainer; - - TStopwatch timer; - timer.Reset(); - timer.Start(); - - if (fPositionList.size() == 0) - std::cout << " Position list not filled " << std::endl; - if (VERBOSE >= 3) - std::cout << "Number of hits for coarse grid search = " << nhits << std::endl; - int iEnd = (iIte + iStart); - if (iEnd > (int)fPositionList.size()) - iEnd = fPositionList.size(); - - for (int iPos = iStart; iPos < iEnd; iPos++) - { - - struct FitPosition hPos; - hPos.Vtx = fPositionList[iPos]; - hPos.NLL = 0; - - if (likelihood) hPos.NLL = Likelihoods::Vertex_Time_NLL(fHitCollection, hPos.Vtx, nhits, lowerLimit, upperLimit, false, false, directionality); - else hPos.NLL = Likelihoods::Vertex_Score(fHitCollection, hPos.Vtx, nhits, lowerLimit, upperLimit, false, false, directionality); - - tVtxContainer.push_back(hPos); - } - - std::sort(tVtxContainer.begin(), tVtxContainer.end(), Likelihoods::SortingNLL()); - while ((int)tVtxContainer.size() > tolerance) tVtxContainer.pop_back(); - - // Check if mutex is already lock, if not -> lock - // We don't want simultaneous modification of the common output - mtx.lock(); - for (unsigned int iPos = 0; iPos < tVtxContainer.size(); iPos++) - { - std::vector vPos(5, 0.); - struct FitPosition hPos = tVtxContainer[iPos]; - - vPos[0] = hPos.Vtx[0]; - vPos[1] = hPos.Vtx[1]; - vPos[2] = hPos.Vtx[2]; - vPos[3] = hPos.Vtx[3]; - vPos[4] = hPos.NLL; - fThreadOutput.push_back(vPos); - } - mtx.unlock(); -} - -// Fine grid search of the vertex. It will search in a box this time, and not a cylinder as searchVertex is doing. It is useful when we have one or several first candidate vertices. Therefore, this code is generally run after serachVertex. -// The box is centered around the initial vertex provided. -// The box size is 2 x limits in each direction, centered on the initial vertex provided. -// nCandidate provide the size of the list of inital vertices, while tolerance provide the size of the output list. -// Other informations are the same as searchVertex. -std::vector> LEAF::SearchVertexFine(std::vector> initialVertex, double *limits, double stepSize, int nhits, int nCandidates, int tolerance, int verbose, bool likelihood, bool average, double lowerLimit, double upperLimit, int directionality) -{ - if (verbose) std::cout << "Start fine search, use directionality? " << directionality << std::endl; - std::vector tBestReconstructedVertexPosition; - double minNLL = 1e15; - clock_t timeStart = clock(); - - std::map> vertexContainer; - - for (int icand = 0; icand < nCandidates; icand++) - { - if (verbose) - std::cout << "Candidate #" << icand << std::endl; - for (double time = initialVertex[icand][VTX_T] - limits[0] / fLightSpeed; time <= initialVertex[icand][VTX_T] + limits[0] / fLightSpeed; time += (stepSize / fLightSpeed)) - { - if (VERBOSE >= 2) - std::cout << "Time = " << time << "ns" << std::endl; - for (double x = initialVertex[icand][VTX_X] - limits[1]; x <= initialVertex[icand][VTX_X] + limits[1]; x += stepSize) - { - if (VERBOSE >= 2) - std::cout << "x = " << x << "m" << std::endl; - for (double y = initialVertex[icand][VTX_Y] - limits[2]; y <= initialVertex[icand][VTX_Y] + limits[2]; y += stepSize) - { - if (VERBOSE >= 2) - std::cout << "y = " << y << "m" << std::endl; - for (double z = initialVertex[icand][VTX_Z] - limits[3]; z <= initialVertex[icand][VTX_Z] + limits[3]; z += stepSize) - { - if (VERBOSE >= 2) - std::cout << "z = " << z << "m" << std::endl; - - std::vector tVtxPosition(4, 0.); // In centimeters - tVtxPosition[0] = x; // radius*TMath::Cos(angle); - tVtxPosition[1] = y; // radius*TMath::Sin(angle); - tVtxPosition[2] = z; // height; - tVtxPosition[3] = time; - - double NLL = 0; - if (average) - { - // double vNLL[fAveraging]; - for (int irand = 0; irand < fAveraging; irand++) - { - // Determine the position around the vertex - double radius = 1.5 * fRand->Uniform(0, stepSize); - double phi = fRand->Uniform(0, 2 * TMath::Pi()); - double theta = fRand->Uniform(0, TMath::Pi()); - double timeRand = 1.5 * fRand->Uniform(-stepSize / fLightSpeed, stepSize / fLightSpeed); - std::vector randVertexPosition(4, 0.); // In centimeters - randVertexPosition[0] = tVtxPosition[0] + radius * TMath::Sin(theta) * TMath::Cos(phi); - randVertexPosition[1] = tVtxPosition[1] + radius * TMath::Sin(theta) * TMath::Sin(phi); - randVertexPosition[2] = tVtxPosition[2] + radius * TMath::Cos(theta); - randVertexPosition[3] = tVtxPosition[3] + timeRand; // rand->Uniform(-stepSize/fLightSpeed/2,stepSize/fLightSpeed/2); - // Evaluate its NLL - double nll = Likelihoods::FindNLL(fHitCollection, randVertexPosition, nhits, likelihood, verbose, lowerLimit, upperLimit, false, false, directionality); - NLL += nll; - // vNLL[irand]=nll; - } - - if (fAveraging != 0) - NLL /= fAveraging; - } - else - NLL = Likelihoods::FindNLL(fHitCollection, tVtxPosition, nhits, likelihood, verbose, lowerLimit, upperLimit, false, false, directionality); - - std::vector candidateReconstructedVertexPosition(4, 0.); - for (int i = 0; i < 4; i++) - candidateReconstructedVertexPosition[i] = tVtxPosition[i]; - - vertexContainer[NLL] = candidateReconstructedVertexPosition; - if ((int)vertexContainer.size() > tolerance) - { - vertexContainer.erase(--vertexContainer.rbegin().base()); - } - - if (NLL < minNLL) - { - minNLL = NLL; - // std::cout << "minNLL = "<> tReconstructedVertexPosition; - for (int count = 0; count < tolerance; count++) - { - std::vector vTmp(5, 0.); - tReconstructedVertexPosition.push_back(vTmp); - } - int counter = 0; - for (std::map>::iterator it = vertexContainer.begin(); it != vertexContainer.end(); it++) - { - // for(std::map< double, std::vector >::iterator it = vertexContainer.begin();it!=vertexContainer.end();it++){ - if (counter < tolerance) - { - // tReconstructedVertexPosition[counter] = new double[4]; - for (int i = 0; i < 4; i++) - tReconstructedVertexPosition[counter][i] = (it->second)[i]; - tReconstructedVertexPosition[counter][4] = it->first; - } - counter++; - } - - clock_t timeEnd = clock(); - if (verbose) std::cout << "SearchVertexFine: Time it took for the loop = " << (timeEndLoop - timeStart) / 1e6 << ", time total = " << (timeEnd - timeStart) / 1e6 << std::endl << std::endl; - - return tReconstructedVertexPosition; - // return tBestReconstructedVertexPosition; -} - -// here is the function where minimization gonna take place. The loop is inside this function -std::vector> LEAF::MinimizeVertex(std::vector> initialVertex, double *limits, double stepSize, int nhits, int nCandidates, int tolerance, int verbose, bool /*likelihood*/, bool /*average*/, double lowerLimit, double upperLimit, int directionality) -{ - if (VERBOSE >= 2) std::cout << "Minimizer" << std::endl; - // 2. How to set the search? - // double stepSize = 0.5;//in m - // double * reconstructedVertexPosition = new double[4]; - std::vector tBestReconstructedVertexPosition; - clock_t timeStart = clock(); - - std::vector tVtxContainer; - - if (VERBOSE >= 2) - std::cout << "Minimizer" << std::endl; - TFitter *minimizer = new TFitter(8); // 4=nb de params? - TMinuit *minuit = minimizer->GetMinuit(); - - double arglist[20]; - int err = 0; - double p1 = verbose - 1; - minimizer->ExecuteCommand("SET PRINTOUT", &p1, 1); // quiet mode - minuit->SetErrorDef(1); - // minuit->SetPrintLevel(-1); // quiet mode - if (VERBOSE < 2) minuit->mnexcm("SET NOWarnings", 0, 0, err); - arglist[0] = 2; - minuit->mnexcm("SET STR", arglist, 1, err); // set strategy sets the number of derivative estimated. in the present case (2), this is the highest number of it with a counter-balance: very slow! - minimizer->SetFCN(MinuitLikelihood); // here is function to minimize - - double Mig[2] = {1e6, 1e0}; // maxcalls and tolerance - - minimizer->SetParameter(0, "vertex0", 0, stepSize, -limits[0], limits[0]); - minimizer->SetParameter(1, "vertex1", 0, stepSize, -limits[1], limits[1]); - minimizer->SetParameter(2, "vertex2", 0, stepSize, -limits[2], limits[2]); - minimizer->SetParameter(3, "vertex3", 0, stepSize / fLightSpeed, -limits[3] / fLightSpeed, limits[3] / fLightSpeed); - - minimizer->SetParameter(4, "PMTConfiguration", 0, 0, 0, 0); // Useless now - minimizer->SetParameter(5, "nhits", nhits, nhits, nhits - 1, nhits + 1); - minimizer->SetParameter(6, "lowerLimit", lowerLimit, 5e-1, -7, -2); - minimizer->SetParameter(7, "upperLimit", upperLimit, 5e-1, 2, 10); - minimizer->SetParameter(8, "expoSigma", 100, 5, 0, 1e3); - minimizer->SetParameter(9, "directionality", directionality, 1, 0, 2); - minimizer->SetParameter(10, "thread", 0, 0, 0, 0); - minimizer->FixParameter(4); - minimizer->FixParameter(5); - minimizer->FixParameter(6); - minimizer->FixParameter(7); - minimizer->FixParameter(8); - minimizer->FixParameter(9); - minimizer->FixParameter(10); - - for (int icand = 0; icand < nCandidates; icand++) - { -#ifdef VERBOSE_MIN - if (verbose >= 1) - { - std::cout << "Candidate #" << icand << std::endl; - std::cout << "Initial (x,y,z,t): " << initialVertex[icand][0] - << ", " << initialVertex[icand][1] - << ", " << initialVertex[icand][2] - << ", " << initialVertex[icand][3] << std::endl; - } -#endif - - minimizer->SetParameter(0, "vertex0", initialVertex[icand][0], stepSize, initialVertex[icand][0] - limits[0], initialVertex[icand][0] + limits[0]); - minimizer->SetParameter(1, "vertex1", initialVertex[icand][1], stepSize, initialVertex[icand][1] - limits[1], initialVertex[icand][1] + limits[1]); - minimizer->SetParameter(2, "vertex2", initialVertex[icand][2], stepSize, initialVertex[icand][2] - limits[2], initialVertex[icand][2] + limits[2]); - minimizer->SetParameter(3, "vertex3", initialVertex[icand][3], stepSize / fLightSpeed, initialVertex[icand][3] - limits[3] / fLightSpeed, initialVertex[icand][3] + limits[3] / fLightSpeed); - - minimizer->SetParameter(4, "PMTConfiguration", 0, 0, 0, 0); // Useless now - minimizer->SetParameter(5, "nhits", nhits, nhits, nhits - 1, nhits + 1); - minimizer->SetParameter(6, "lowerLimit", lowerLimit, 5e-1, -7, -2); - minimizer->SetParameter(7, "upperLimit", upperLimit, 5e-1, 2, 10); - minimizer->SetParameter(8, "expoSigma", 100, 5, 0, 1e3); - // minimizer->SetParameter(8,"scaling factor",1,1,0,1e3); - // minimizer->SetParameter(8,"signal pe",1,1,0,1e3); - minimizer->SetParameter(9, "directionality", directionality, 1, 0, 2); - minimizer->SetParameter(10, "thread", 0, 0, 0, 0); - minimizer->FixParameter(4); - minimizer->FixParameter(5); - minimizer->FixParameter(6); - minimizer->FixParameter(7); - minimizer->FixParameter(8); - minimizer->FixParameter(9); - minimizer->FixParameter(10); - - minimizer->ExecuteCommand("MIGRAD", Mig, 2); - - std::vector tCandRecoVtxPos(4, 0.); - - tCandRecoVtxPos[0] = minimizer->GetParameter(0); - tCandRecoVtxPos[1] = minimizer->GetParameter(1); - tCandRecoVtxPos[2] = minimizer->GetParameter(2); - tCandRecoVtxPos[3] = minimizer->GetParameter(3); - - struct FitPosition hPos; - hPos.Vtx = tCandRecoVtxPos; - hPos.NLL = 0; - - // Get minuit - minuit = minimizer->GetMinuit(); - double fmin, fedm, errdef; - int npar, nparx, istat; - minuit->mnstat(fmin, fedm, errdef, npar, nparx, istat); - - hPos.NLL = fmin; - -#ifdef VERBOSE_MIN - if (verbose >= 1) - std::cout << "NLL=" << hPos.NLL << std::endl; -#endif - tVtxContainer.push_back(hPos); - - if (VERBOSE >= 2) - std::cout << "Initial vertex position = " << initialVertex[icand][0] << "," << initialVertex[icand][1] << "," << initialVertex[icand][2] << "," << initialVertex[icand][3] << ", Candidate vertex position = (" << tCandRecoVtxPos[0] << "," << tCandRecoVtxPos[1] << "," << tCandRecoVtxPos[2] << "," << tCandRecoVtxPos[3] << ", NLL = " << fmin << std::endl; - } - - std::sort(tVtxContainer.begin(), tVtxContainer.end(), Likelihoods::SortingNLL()); - while ((int)tVtxContainer.size() > tolerance) tVtxContainer.pop_back(); - - clock_t timeEndLoop = clock(); - - std::vector> tReconstructedVertexPosition; - for (int iPos = 0; iPos < tolerance; iPos++) - { - std::vector vPos(5, 0.); - struct FitPosition hPos = tVtxContainer[iPos]; - - vPos[0] = hPos.Vtx[0]; - vPos[1] = hPos.Vtx[1]; - vPos[2] = hPos.Vtx[2]; - vPos[3] = hPos.Vtx[3]; - vPos[4] = hPos.NLL; - - tReconstructedVertexPosition.push_back(vPos); - } - - delete minimizer; - - tBestReconstructedVertexPosition = tReconstructedVertexPosition[0]; - - clock_t timeEnd = clock(); - - if (verbose) std::cout << "MinimizeVertex: Time it took for the loop = " << (timeEndLoop - timeStart) / 1e6 << ", time total = " << (timeEnd - timeStart) / 1e6 << std::endl << std::endl; - - return tReconstructedVertexPosition; - // return tBestReconstructedVertexPosition; -} - -std::vector> LEAF::MinimizeVertex_Main(std::vector> initialVertex, double *limits, double stepSize, int nhits, int nCandidates, int tolerance, int verbose, bool likelihood, bool average, double lowerLimit, double upperLimit, int directionality, std::vector* fDirection_Filter, float FilterThreshold) -{ - std::vector lThreadList; - std::vector> lOutputFinal; - - int iCand_Step = nCandidates / fThread; - if (iCand_Step < 1) iCand_Step = 1; - - for (int iStart = 0; iStart < nCandidates; iStart += iCand_Step) - { - std::thread tThrd(&LEAF::MinimizeVertex_thread, LEAF::GetME(), iStart, iCand_Step, - initialVertex, limits, stepSize, nhits, nCandidates, - tolerance, verbose, likelihood, average, lowerLimit, upperLimit, directionality, fDirection_Filter, FilterThreshold); - - lThreadList.push_back(std::move(tThrd)); - } - - for (unsigned int iIdx = 0; iIdx < lThreadList.size(); iIdx++) - { - // Wait thread - if (lThreadList[iIdx].joinable()) - lThreadList[iIdx].join(); - } - - // Check if mutex is already lock, if not -> lock - // We don't want simultaneous modification of the common output - mtx.lock(); - lOutputFinal = fThreadOutput; - fThreadOutput.clear(); - mtx.unlock(); - - std::sort(lOutputFinal.begin(), lOutputFinal.end(), SortOutputVector); - - while ((int)lOutputFinal.size() > tolerance) lOutputFinal.pop_back(); - - return lOutputFinal; -} - -// here is the function where minimization gonna take place. The loop is inside this function -void LEAF::MinimizeVertex_thread( - int iStart, int iIte, - std::vector> initialVertex, double *limits, double stepSize, int nhits, - int nCandidates, int tolerance, int verbose, bool /*likelihood*/, bool /*average*/, double lowerLimit, double upperLimit, int directionality, std::vector* fDirection_Filter, float FilterThreshold) -{ - int iEnd = (iIte + iStart); - if (iEnd > nCandidates) iEnd = nCandidates; - - std::vector tVtxContainer; - - mtx.lock(); - TFitter *minimizer = new TFitter(14); // 4=nb de params? - mtx.unlock(); - TMinuit *minuit = minimizer->GetMinuit(); - - double arglist[20]; - int err = 0; - // arglist[0]=0; - double p1 = verbose - 1; - minimizer->ExecuteCommand("SET PRINTOUT", &p1, 1); // quiet mode - minuit->SetErrorDef(1); - // minuit->SetPrintLevel(verbose-1); // quiet mode - if (VERBOSE < 2) minuit->mnexcm("SET NOWarnings", 0, 0, err); - arglist[0] = 2; - minuit->mnexcm("SET STR", arglist, 1, err); // set strategy sets the number of derivative estimated. in the present case (2), this is the highest number of it with a counter-balance: very slow! - minimizer->SetFCN(MinuitLikelihood); // here is function to minimize - - double Mig[2] = {1e6, 1e0}; // maxcalls and tolerance - - minimizer->SetParameter(0, "vertex0", 0, stepSize, -limits[0], limits[0]); - minimizer->SetParameter(1, "vertex1", 0, stepSize, -limits[1], limits[1]); - minimizer->SetParameter(2, "vertex2", 0, stepSize, -limits[2], limits[2]); - minimizer->SetParameter(3, "vertex3", 0, stepSize / fLightSpeed, -limits[3] / fLightSpeed, limits[3] / fLightSpeed); - - minimizer->SetParameter(4, "PMTConfiguration", 0, 0, 0, 0); // Useless now - minimizer->SetParameter(5, "nhits", nhits, nhits, nhits - 1, nhits + 1); - minimizer->SetParameter(6, "lowerLimit", lowerLimit, 5e-1, -7, -2); - minimizer->SetParameter(7, "upperLimit", upperLimit, 5e-1, 2, 10); - minimizer->SetParameter(8, "expoSigma", 100, 5, 0, 1e3); - minimizer->SetParameter(9, "directionality", directionality, 1, 0, 2); - minimizer->SetParameter(10, "dir_x", 0, 0, -1, 1); - minimizer->SetParameter(11, "dir_y", 0, 0, -1, 1); - minimizer->SetParameter(12, "dir_z", 0, 0, -1, 1); - minimizer->SetParameter(13, "dir_threshold", 0, 0, 0, 0); - minimizer->FixParameter(4); - minimizer->FixParameter(5); - minimizer->FixParameter(6); - minimizer->FixParameter(7); - minimizer->FixParameter(8); - minimizer->FixParameter(9); - - for (int icand = iStart; icand < iEnd; icand++) - { - minimizer->SetParameter(0, "vertex0", initialVertex[icand][0], stepSize, initialVertex[icand][0] - limits[0], initialVertex[icand][0] + limits[0]); - minimizer->SetParameter(1, "vertex1", initialVertex[icand][1], stepSize, initialVertex[icand][1] - limits[1], initialVertex[icand][1] + limits[1]); - minimizer->SetParameter(2, "vertex2", initialVertex[icand][2], stepSize, initialVertex[icand][2] - limits[2], initialVertex[icand][2] + limits[2]); - minimizer->SetParameter(3, "vertex3", initialVertex[icand][3], stepSize / fLightSpeed, initialVertex[icand][3] - limits[3] / fLightSpeed, initialVertex[icand][3] + limits[3] / fLightSpeed); - - minimizer->SetParameter(4, "PMTConfiguration", 0, 0, 0, 0); // Useless now - minimizer->SetParameter(5, "nhits", nhits, nhits, nhits - 1, nhits + 1); - minimizer->SetParameter(6, "lowerLimit", lowerLimit, 5e-1, -7, -2); - minimizer->SetParameter(7, "upperLimit", upperLimit, 5e-1, 2, 10); - minimizer->SetParameter(8, "expoSigma", 100, 5, 0, 1e3); - minimizer->SetParameter(9, "directionality", directionality, 1, 0, 2); - - if(fDirection_Filter != nullptr && fDirection_Filter->size() > 2) - { - minimizer->SetParameter(10, "dir_x", (*fDirection_Filter)[0], 5e-1, (*fDirection_Filter)[0] - 1, (*fDirection_Filter)[0] + 1); - minimizer->SetParameter(11, "dir_y", (*fDirection_Filter)[1], 5e-1, (*fDirection_Filter)[1] - 1, (*fDirection_Filter)[1] + 1); - minimizer->SetParameter(12, "dir_z", (*fDirection_Filter)[2], 5e-1, (*fDirection_Filter)[2] - 1, (*fDirection_Filter)[2] + 1); - } - minimizer->SetParameter(13, "dir_threshold", FilterThreshold, 5e-1, FilterThreshold - 1, FilterThreshold + 1); - minimizer->FixParameter(4); - minimizer->FixParameter(5); - minimizer->FixParameter(6); - minimizer->FixParameter(7); - minimizer->FixParameter(8); - minimizer->FixParameter(9); - minimizer->FixParameter(10); - minimizer->FixParameter(11); - minimizer->FixParameter(12); - minimizer->FixParameter(13); - - minimizer->ExecuteCommand("MIGRAD", Mig, 2); - - std::vector tCandRecoVtxPos(4, 0.); - - tCandRecoVtxPos[0] = minimizer->GetParameter(0); - tCandRecoVtxPos[1] = minimizer->GetParameter(1); - tCandRecoVtxPos[2] = minimizer->GetParameter(2); - tCandRecoVtxPos[3] = minimizer->GetParameter(3); - - struct FitPosition hPos; - hPos.Vtx = tCandRecoVtxPos; - hPos.NLL = 0; - - // Get minuit - minuit = minimizer->GetMinuit(); - double fmin, fedm, errdef; - int npar, nparx, istat; - minuit->mnstat(fmin, fedm, errdef, npar, nparx, istat); - - hPos.NLL = fmin; - - tVtxContainer.push_back(hPos); - if (VERBOSE >= 2) - std::cout << "Initial vertex position = " << initialVertex[icand][0] << "," << initialVertex[icand][1] << "," << initialVertex[icand][2] << "," << initialVertex[icand][3] << ", Candidate vertex position = (" << tCandRecoVtxPos[0] << "," << tCandRecoVtxPos[1] << "," << tCandRecoVtxPos[2] << "," << tCandRecoVtxPos[3] << ", NLL = " << fmin << std::endl; - } - - std::sort(tVtxContainer.begin(), tVtxContainer.end(), Likelihoods::SortingNLL()); - while ((int)tVtxContainer.size() > tolerance) tVtxContainer.pop_back(); - - // Check if mutex is already lock, if not -> lock - // We don't want simultaneous modification of the common output - mtx.lock(); - for (unsigned int iPos = 0; iPos < tVtxContainer.size(); iPos++) - { - std::vector vPos(5, 0.); - struct FitPosition hPos = tVtxContainer[iPos]; - - vPos[0] = hPos.Vtx[0]; - vPos[1] = hPos.Vtx[1]; - vPos[2] = hPos.Vtx[2]; - vPos[3] = hPos.Vtx[3]; - vPos[4] = hPos.NLL; - - fThreadOutput.push_back(vPos); - } - delete minimizer; - mtx.unlock(); -} - -void LEAF::FitVertex(std::vector* fDirection_Filter, float FilterThreshold) -{ - double *tLimits = new double[4]; - int iHitsTotal = fHitCollection->Size(); - std::vector< std::vector > fRecoVtxPosFinal; - - TStopwatch timer; - timer.Reset(); - timer.Start(); - - std::vector tRecoVtxPosCand = this->SearchVertex_Main(iHitsTotal,fSearchVtxTolerance,false,fSTimePDFLimitsQueueNegative,fSTimePDFLimitsQueuePositive,false); - std::vector> tRecoVtxPos; - - // Extract only the vertex positions from the VtxCandidate structure - for (const auto& cand : tRecoVtxPosCand) - { - tRecoVtxPos.push_back({cand.X, cand.Y, cand.Z, cand.T, cand.NLL}); - } - - if (VERBOSE >= 2) - { - for (unsigned int a = 0; a < tRecoVtxPos.size(); a++) - std::cout << "Candidate vertex [ " << a << " ] = " << tRecoVtxPos[a][0] << " , " << tRecoVtxPos[a][1] << " , " << tRecoVtxPos[a][2] << " , " << tRecoVtxPos[a][3] << ", NLL = " << tRecoVtxPos[a][4] << std::endl; - } - - timer.Stop(); - fOutputProps.Vtx_Search_ComputeTime = timer.RealTime(); - // std::cout << "SearchVertex took: " << timer.RealTime() << std::endl; - - double dStepSizeFinal = 10; - int iToleranceFinal = 1; - if (fStepByStep) //* Not used and not tested - { - // 2.b. Refined vertex search around candidate found previously - // -> Commented since ages, removed on 2020/11/20 - - // 2.c. More refined vertex search around candidate found previously - double dStepSizeFine2 = 100; - int iToleranceFine2 = 20; - for (int i = 0; i < 4; i++) tLimits[i] = fSearchVtxStep; // particleStart[i]*1e-2;//onversion to meter - std::vector> tRecoVtxPosFine2 = this->SearchVertexFine(tRecoVtxPos, tLimits, dStepSizeFine2, iHitsTotal, fSearchVtxTolerance, iToleranceFine2, VERBOSE, true, false, fSTimePDFLimitsQueueNegative, fSTimePDFLimitsQueuePositive, fUseDirectionality); - ////////////////////////////////////////// - - // 2.d. Final refined vertex search around candidate found previously - double dStepSizeFine3 = 20; - int iToleranceFine3 = 1; - for (int i = 0; i < 4; i++) - tLimits[i] = dStepSizeFine2; // particleStart[i]*1e-2;//onversion to meter - std::vector> tRecoVtxPosFine3 = this->SearchVertexFine(tRecoVtxPosFine2, tLimits, dStepSizeFine3, iHitsTotal, iToleranceFine2, iToleranceFine3, VERBOSE, true, false, fSTimePDFLimitsQueueNegative, fSTimePDFLimitsQueuePositive, fUseDirectionality); - if (VERBOSE == 1) - { - for (int a = 0; a < iToleranceFine3; a++) - std::cout << "Final candidate vertex time = " << tRecoVtxPosFine3[a][0] << ", x = " << tRecoVtxPosFine3[a][1] << ", y = " << tRecoVtxPosFine3[a][2] << ", z=" << tRecoVtxPosFine3[a][3] << std::endl; - } - ////////////////////////////////////////// - - for (int i = 0; i < 4; i++) - tLimits[i] = 100; // dStepSizeFine2;//particleStart[i]*1e-2;//onversion to meter - fRecoVtxPosFinal = this->MinimizeVertex(tRecoVtxPosFine3, tLimits, dStepSizeFinal, iHitsTotal, iToleranceFine3, iToleranceFinal, VERBOSE, true, false, fMinimizeLimitsNegative, fMinimizeLimitsPositive, fUseDirectionality); - if (VERBOSE >= 1) - { - for (int a = 0; a < iToleranceFinal; a++) - { - std::cout << "Final candidate vertex time = " << fRecoVtxPosFinal[a][0] << ", x = " << fRecoVtxPosFinal[a][1] << ", y = " << fRecoVtxPosFinal[a][2] << ", z=" << fRecoVtxPosFinal[a][3] << std::endl; - } - } - } - else - { - timer.Reset(); - timer.Start(); - - for (int i = 0; i < 4; i++) tLimits[i] = 2 * fSearchVtxStep; - - fRecoVtxPosFinal = this->MinimizeVertex_Main(tRecoVtxPos, tLimits, dStepSizeFinal, iHitsTotal, fSearchVtxTolerance, iToleranceFinal, VERBOSE, true, false, fMinimizeLimitsNegative, fMinimizeLimitsPositive, fUseDirectionality, fDirection_Filter, FilterThreshold); - - timer.Stop(); - fOutputProps.Vtx_Minimize_ComputeTime = timer.RealTime(); - } - - fOutput.Vtx[0] = fRecoVtxPosFinal[0][0]; - fOutput.Vtx[1] = fRecoVtxPosFinal[0][1]; - fOutput.Vtx[2] = fRecoVtxPosFinal[0][2]; - fOutput.Vtx[3] = fRecoVtxPosFinal[0][3]; - fOutput.NLL = fRecoVtxPosFinal[0][4]; -} - -void LEAF::FitEnergy() -{ - double totalPE = 0; - double totalPECorr = 0; - - TStopwatch timer; - timer.Reset(); - timer.Start(); - - // Taha's Polynomial Correction - // double a = -0.000440808; - // double b = 1.71313; - // double c = -2.01675; - // double d = 1.81755; - - for (unsigned int ihit = 0; ihit < fHitCollection->Size(); ihit++) - { - Hit lHit = fHitCollection->At(ihit); - - // int iPMT = lHit.PMT; - // PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - // double PMTOrientation[3]; - // double dirVector[3]; - // for(int j = 0; j<3; j++) PMTOrientation[j] = lPMTInfo.Orientation[j]; - // for(int j = 0; j<3; j++) dirVector[j] = -(fTrueVtxPos[j] - lPMTInfo.Position[j]); - // GeoTools::Normalize(PMTOrientation); - // GeoTools::Normalize(dirVector); - // double cosPMThitAngle = abs(dirVector[0] * PMTOrientation[0] + dirVector[1] * PMTOrientation[1] + dirVector[2] * PMTOrientation[2]); - // double Ftheta = a + b*cosPMThitAngle + c*(pow(cosPMThitAngle,2)) + d*(pow(cosPMThitAngle,3)); // Taha's Polynomial Correction - - totalPECorr += lHit.Q; - totalPE += lHit.Q; - } - - double energyEstimate = totalPECorr * 0.111247 - 12.970173; //* Affine correction (affine because of dark rate and systematic errors), convert from charge to energy - fOutput.TotalCharge = totalPE; - fOutput.Energy = energyEstimate; - - timer.Stop(); - fOutputProps.Energy_Fit_ComputeTime = timer.RealTime(); -} - -std::vector LEAF::SearchVertexAndDir() -{ - TStopwatch timer; - timer.Reset(); - timer.Start(); - - std::vector Candidates = std::vector(); - - std::vector tVtxCandidates = this->SearchVertex_Main(fHitCollection->Size(),fSearchVtxTolerance,false,fSTimePDFLimitsQueueNegative,fSTimePDFLimitsQueuePositive,false); - - for(int i=0; i<(int)tVtxCandidates.size(); i++) - { - std::vector vtx = {tVtxCandidates[i].X, tVtxCandidates[i].Y, tVtxCandidates[i].Z, tVtxCandidates[i].T}; - std::vector tDirCandidates = this->SearchDirection(vtx, DirTolerance); - for(int j=0; j<(int)tDirCandidates.size(); j++) - { - JointFitCandidate candidate; - candidate.VtxPart = tVtxCandidates[i]; - candidate.DirPart = tDirCandidates[j]; - candidate.NLL = candidate.DirPart.DNLL * candidate.VtxPart.NLL; - Candidates.push_back(candidate); - } - } - - std::sort(Candidates.begin(), Candidates.end(), Likelihoods::CompareByNLL{}); - - while ((int)Candidates.size() > fSearchVtxTolerance) Candidates.pop_back(); - - std::vector dirCandidates = std::vector(); - for(int i=0; i<(int)Candidates.size(); i++) dirCandidates.push_back(Candidates[i].DirPart); - // fOutput.stepOneContainsTrueDir = (int)ContainsTrueDir(&dirCandidates, fTrueDir); - - timer.Stop(); - fOutputProps.Vtx_Search_ComputeTime = timer.RealTime(); - - return Candidates; -} - -JointFitCandidate LEAF::MinimizeVertexAndDir(std::vector initialCandidates) -{ - TStopwatch timer; - timer.Reset(); - timer.Start(); - - double stepSize = 10; - int nhits = fHitCollection->Size(); - double *limits = new double[4]; - for (int i = 0; i < 4; i++) limits[i] = 2 * fSearchVtxStep; - - std::vector tContainer; - - TFitter *minimizer = new TFitter(12); - TMinuit *minuit = minimizer->GetMinuit(); - - double arglist[20]; - int err = 0; - double p1 = VERBOSE - 1; - minimizer->ExecuteCommand("SET PRINTOUT", &p1, 1); // quiet mode - minuit->SetErrorDef(1); - if (VERBOSE < 2) minuit->mnexcm("SET NOWarnings", 0, 0, err); - arglist[0] = 2; - minuit->mnexcm("SET STR", arglist, 1, err); // set strategy sets the number of derivative estimated. in the present case (2), this is the highest number of it with a counter-balance: very slow! - minimizer->SetFCN(MinuitJointNLL); // here is function to minimize - - double Mig[2] = {1e6, 1e0}; // maxcalls and tolerance - - minimizer->SetParameter(0, "vertex0", 0, stepSize, -limits[0], limits[0]); - minimizer->SetParameter(1, "vertex1", 0, stepSize, -limits[1], limits[1]); - minimizer->SetParameter(2, "vertex2", 0, stepSize, -limits[2], limits[2]); - minimizer->SetParameter(3, "vertex3", 0, stepSize / fLightSpeed, -limits[3] / fLightSpeed, limits[3] / fLightSpeed); - minimizer->SetParameter(4, "theta", 0, 2*M_PI/180, 0.0, TMath::Pi()); - minimizer->SetParameter(5, "phi", 0, 2*M_PI/180, -TMath::Pi(), TMath::Pi()); - - minimizer->SetParameter(6, "PMTConfiguration", 0, 0, 0, 0); // Useless now - minimizer->SetParameter(7, "nhits", nhits, nhits, nhits - 1, nhits + 1); - minimizer->SetParameter(8, "lowerLimit", fSTimePDFLimitsQueueNegative, 5e-1, -7, -2); - minimizer->SetParameter(9, "upperLimit", fSTimePDFLimitsQueuePositive, 5e-1, 2, 10); - minimizer->SetParameter(10, "expoSigma", 100, 5, 0, 1e3); - minimizer->SetParameter(11, "thread", 0, 0, 0, 0); - minimizer->FixParameter(6); - minimizer->FixParameter(7); - minimizer->FixParameter(8); - minimizer->FixParameter(9); - minimizer->FixParameter(10); - minimizer->FixParameter(11); - - for (int icand = 0; icand < (int)initialCandidates.size(); icand++) - { - minimizer->SetParameter(0, "vertex0", initialCandidates[icand].VtxPart.X, stepSize, initialCandidates[icand].VtxPart.X - limits[0], initialCandidates[icand].VtxPart.X + limits[0]); - minimizer->SetParameter(1, "vertex1", initialCandidates[icand].VtxPart.Y, stepSize, initialCandidates[icand].VtxPart.Y - limits[1], initialCandidates[icand].VtxPart.Y + limits[1]); - minimizer->SetParameter(2, "vertex2", initialCandidates[icand].VtxPart.Z, stepSize, initialCandidates[icand].VtxPart.Z - limits[2], initialCandidates[icand].VtxPart.Z + limits[2]); - minimizer->SetParameter(3, "vertex3", initialCandidates[icand].VtxPart.T, stepSize / fLightSpeed, initialCandidates[icand].VtxPart.T - limits[3] / fLightSpeed, initialCandidates[icand].VtxPart.T + limits[3] / fLightSpeed); - minimizer->SetParameter(4, "theta", initialCandidates[icand].DirPart.theta, 2*M_PI/180, 0.0, TMath::Pi()); - minimizer->SetParameter(5, "phi", initialCandidates[icand].DirPart.phi, 2*M_PI/180, -TMath::Pi(), TMath::Pi()); - - minimizer->ExecuteCommand("MIGRAD", Mig, 2); - - JointFitCandidate tCandReco; - - tCandReco.VtxPart.X = minimizer->GetParameter(0); - tCandReco.VtxPart.Y = minimizer->GetParameter(1); - tCandReco.VtxPart.Z = minimizer->GetParameter(2); - tCandReco.VtxPart.T = minimizer->GetParameter(3); - tCandReco.DirPart.theta = minimizer->GetParameter(4); - tCandReco.DirPart.phi = minimizer->GetParameter(5); - - // Get minuit - double fmin, fedm, errdef; - int npar, nparx, istat; - minuit->mnstat(fmin, fedm, errdef, npar, nparx, istat); - tCandReco.NLL = fmin; - - tContainer.push_back(tCandReco); - } - - std::sort(tContainer.begin(), tContainer.end(), Likelihoods::CompareByNLL{}); - - delete minimizer; - - timer.Stop(); - fOutputProps.Vtx_Minimize_ComputeTime = timer.RealTime(); - - return tContainer[0]; -} - -void LEAF::LoadHitCollection(const HitCollection *lHitCol, const TimeDelta lTriggerTime, bool bMultiPMT) -{ - fHitCollection = lHitCol; - fTriggerTime = lTriggerTime; - fTimeCorrection = fHitCollection->timestamp - fTriggerTime; - if (fPositionList.size() == 0) MakePositionList(); - fUseDirectionality = bMultiPMT ? fUseDirectionality : false; -} - -struct FitterOutput LEAF::NewOutuput() -{ - struct FitterOutput fOutput; - fOutput.Vtx = std::vector(4); - fOutput.Dir = std::vector(3); - fOutput.NLL = 0.; - fOutput.DNLL = 0.; - fOutput.Energy = 0.; - fOutput.TotalCharge = 0.; - - return fOutput; -} - -struct FitterOutput LEAF::MakeSequentialFit(const HitCollection *lHitCol, const TimeDelta lTriggerTime, bool bMultiPMT) -{ - TStopwatch timer; - timer.Reset(); - timer.Start(); - - LoadHitCollection(lHitCol, lTriggerTime, bMultiPMT); - - if(fHitCollection->Size() <= 0) return NewOutuput(); - - //* Vertex - FitVertex(); - - //* Goodness of fit - fOutput.NLLR = Likelihoods::GoodnessOfFit(fHitCollection, fOutput.Vtx, fHitCollection->Size(), fMinimizeLimitsNegative, fMinimizeLimitsPositive, true, false, fUseDirectionality); //* Goodness of Fit - - //* Direction - - //? three direction methods are stored, one can use only the Fit Direction method - // FitDirectionQuick(fOutput.Vtx); // prefit - - // // the prefit is stored in fOutput.Dir - // FitDirection(fOutput.Vtx, fHitCollection->Size(), true); // optimization with one candidate that is the prefit - // fOutput.MyDir = fOutput.Dir; - // FitDirection(fOutput.Vtx, fHitCollection->Size(), false); // optimization without prefit, 30 candidates used - - FitDirectionQuick(fTrueVtxPos); // prefit - - // the prefit is stored in fOutput.Dir - FitDirection(fTrueVtxPos, fHitCollection->Size(), true); // optimization with one candidate that is the prefit - fOutput.MyDir = fOutput.Dir; - FitDirection(fTrueVtxPos, fHitCollection->Size(), false); // optimization without prefit, 30 candidates used - - - //* Energy - FitEnergy(); - - timer.Stop(); - fOutputProps.Leaf_ComputeTime = timer.RealTime(); - - return fOutput; -} - -struct FitterOutput LEAF::MakeJointFit(const HitCollection *lHitCol, const TimeDelta lTriggerTime, bool bMultiPMT) -{ - LoadHitCollection(lHitCol, lTriggerTime, bMultiPMT); - - TStopwatch timer; - timer.Reset(); - timer.Start(); - - if(fHitCollection->Size() <= 0) return NewOutuput(); - - std::vector Candidates = SearchVertexAndDir(); - JointFitCandidate finalCandidate = MinimizeVertexAndDir(Candidates); - - fOutput.Vtx = {finalCandidate.VtxPart.X, finalCandidate.VtxPart.Y, finalCandidate.VtxPart.Z, finalCandidate.VtxPart.T}; - fOutput.Dir = PolarToCartesianNorm({finalCandidate.DirPart.theta, finalCandidate.DirPart.phi}); - fOutput.NLL = finalCandidate.NLL; - fOutput.DNLL = Likelihoods::Dir_NLL(fHitCollection, fOutput.Vtx, finalCandidate.DirPart.theta, finalCandidate.DirPart.phi , fHitCollection->Size()); - - fOutput.NLLR = Likelihoods::GoodnessOfFit(fHitCollection, fOutput.Vtx, fHitCollection->Size(), fMinimizeLimitsNegative, fMinimizeLimitsPositive, true, false, fUseDirectionality); //* Goodness of Fit - - //* Energy, alone - FitEnergy(); - - timer.Stop(); - fOutputProps.Leaf_ComputeTime = timer.RealTime(); - - return fOutput; -} diff --git a/leaf/LEAF.cpp b/leaf/LEAF.cpp new file mode 100644 index 0000000..542bfc2 --- /dev/null +++ b/leaf/LEAF.cpp @@ -0,0 +1,111 @@ +/*****************************************************************************************************/ +/** LEAF.cc **/ +/** Author: Guillaume Pronost (pronost@km.icrr.u-tokyo.ac.jp) **/ +/** Original author: Benjamin Quilain **/ +/** Date: December 18th 2019 **/ +/** Desc: Low-E Fitter for Hyper-K **/ +/*****************************************************************************************************/ + +#include "LEAF.hpp" +#include + +LEAF *LEAF::myFitter = NULL; + +/*****************************************************************************************************/ +/* GENERAL LEAF METHODS AND STRUCTURES FOR FITS */ +/*****************************************************************************************************/ + +LEAF::LEAF() { + myFitter = this; + fDirectionFilterSet = false; +} + +LEAF::~LEAF() { + delete fRand; + myFitter = NULL; +} + +LEAF *LEAF::GetME() { + if (myFitter) + return myFitter; + + myFitter = new LEAF(); + return myFitter; +} + +void LEAF::DeleteME() { + if (myFitter) delete myFitter; +} + +LEAFStructures::FitterOutput LEAF::MakeSequentialFit() { + TStopwatch timer; + timer.Reset(); + timer.Start(); + + fOutput.Initialize(); + + if(fHitsCollection[int(PMTType::kID)]->size() == 0 && fHitsCollection[int(PMTType::kmPMT)] == 0) return fOutput; + + //* Vertex + this->FitVertex(); + + //* Goodness of fit + fOutput.nll_r = this->GoodnessOfFit(fOutput.vtx, fOutput.vtx_nll, LEAFConfig::fMinimizeLimitsNegative, LEAFConfig::fMinimizeLimitsPositive, true, false, LEAFConfig::fUseDirectionality); //* Goodness of Fit + + //* Direction + + //? three direction methods are stored, one can use only the Fit Direction method + // FitDirectionQuick(fOutput.Vtx); // prefit + + // // the prefit is stored in fOutput.Dir + // FitDirection(fOutput.Vtx, fHitCollection->Size(), true); // optimization with one candidate that is the prefit + // fOutput.MyDir = fOutput.Dir; + // FitDirection(fOutput.Vtx, fHitCollection->Size(), false); // optimization without prefit, 30 candidates used + + this->FitDirectionQuick(fTrueVtxPos); // prefit + + // the prefit is stored in fOutput.Dir + this->FitDirection(fTrueVtxPos, true); // optimization with one candidate that is the prefit + fOutput.dir_pre_fit = fOutput.dir; + this->FitDirection(fTrueVtxPos, false); // optimization without prefit, 30 candidates used + + + //* Energy + FitEnergy(); + + timer.Stop(); + fPerformances.leaf_ct = timer.RealTime(); + + return fOutput; +} + +LEAFStructures::FitterOutput LEAF::MakeJointFit() { + TStopwatch timer; + timer.Reset(); + timer.Start(); + + fOutput.Initialize(); + + std::vector Candidates = SearchVertexAndDir(); + JointFitCandidate finalCandidate = MinimizeVertexAndDir(Candidates); + + fOutput.vtx = finalCandidate.vtx; + fOutput.dir = ROOT::Math::XYZVector(ROOT::Math::Polar3DVector(1.,finalCandidate.theta, finalCandidate.phi).Unit() ); + + fOutput.vtx_nll = finalCandidate.nll; + fOutput.dir_nll = this->Dir_NLL(fOutput.vtx, finalCandidate.theta, finalCandidate.phi); + fOutput.nll_r = this->GoodnessOfFit(fOutput.vtx, fOutput.vtx_nll, LEAFConfig::fMinimizeLimitsNegative, LEAFConfig::fMinimizeLimitsPositive, true, false, LEAFConfig::fUseDirectionality); + + //* Energy, alone + this->FitEnergy(); + + timer.Stop(); + fPerformances.leaf_ct = timer.RealTime(); + + return fOutput; +} + +// Functions implementation was divided among the following files to make the code more readable +#include "LEAFSrcInputs.cpp" +#include "LEAFSrcFitter.cpp" +#include "LEAFSrcLikelihoods.cpp" \ No newline at end of file diff --git a/leaf/LEAF.hh b/leaf/LEAF.hh deleted file mode 100644 index 5c64fb9..0000000 --- a/leaf/LEAF.hh +++ /dev/null @@ -1,153 +0,0 @@ -/*****************************************************************************************************/ -/** LEAF.hh **/ -/** Author: Guillaume Pronost (pronost@km.icrr.u-tokyo.ac.jp) **/ -/** Original author: Benjamin Quilain **/ -/** Date: December 18th 2019 **/ -/** Desc: Low-E Fitter for Hyper-K **/ -/** + Add Low-E functions from Super-K **/ -/*****************************************************************************************************/ - -/*****************************************************************************************************/ -// This class contains a vertex finder for Low-E WCSim-based detectors through different methods -// The vertex finder relies on a likelihood of timing residual for each PMT hit. -// Time residuals are defined as: -// time - time-of-flight-assuming-straight-line-from-vertex-to PMT - time-of-vertex -/*****************************************************************************************************/ -/*****************************************************************************************************/ -// The standard method relies on two parts which are runned successively: -// -// 1. SearchVertex: A coarse grid search in the tank, where steps in space/time can be set by users. -// In default mode, this class does not use the full likelihood but just number of hits in time to -// be faster -// -// 2. MinimizeVertex: A MINUIT-based search within a 4D sphere around a specific position and time -// Note that in default mode, the sphere radius should be set to be the same or similar to the step -// size in SearchVertex. In default mode, the best candidates from SearchVertex are fed to -// MinimizeVertex. The number of best candidates can be selected. // Both MinimizeVertex and -// SearchVertex provides an output containing the best fits vertices, ordered from lower to higher -// NLL. -// -// Most of default parameters of method 1 and 2 can be set in LEAF.cc Init() method -/*****************************************************************************************************/ - -#ifndef LEAF_hh -#define LEAF_hh - -#include -#include -#include -#include -#include -#include -#include - -//* WCSim Headers -#include "WCSimRootGeom.hh" - -//* ROOT Headers -#include "TFile.h" -#include "TFitter.h" -#include "TH1D.h" -#include "TH2D.h" -#include "TGraph2D.h" -#include "TF1.h" -#include "TMath.h" -#include "TMinuit.h" -#include "TObject.h" -#include "TRandom3.h" -#include "TSpline.h" -#include "TPaletteAxis.h" - -//* DataModel informations -#include "Geometry.h" -#include "HitCollection.h" - -//* LEAF Headers -#include "Likelihoods.hh" -#include "LeafInputs.hh" -#include "LeafConfig.hh" -#include "LeafSplines.hh" -#include "LeafUtility.hh" -#include "LeafDefinitions.hh" - -class LEAF -{ - public: - static LEAF* GetME(); - void DeleteME(); - - static FitterOutput fOutput; - static FitterOutputProps fOutputProps; // additional infos about the fit - - void Initialize(const Geometry* lGeometry); - - //* Fitter Main Method. Process the whole fit. - struct FitterOutput MakeSequentialFit(const HitCollection* lHitCol, const TimeDelta lTriggerTime, bool bMultiPMT=true); - - //* slower and less performant for now - struct FitterOutput MakeJointFit(const HitCollection* lHitCol, const TimeDelta lTriggerTime, bool bMultiPMT=true); - - void SearchVertex_thread(int iStart, int iIte, int nhits, int tolerance=1, bool likelihood=false, double lowerLimit=fSTimePDFLimitsQueueNegative, double upperLimit=fSTimePDFLimitsQueuePositive, int directionality=false); - - void MinimizeVertex_thread(int iStart, int iIte, std::vector< std::vector > initialVertex, double * limits, double stepSize, int nhits, int nCandidates = 1, int tolerance = 1, int verbose=0, bool likelihood=false, bool average=false, double lowerLimit=fSTimePDFLimitsQueueNegative, double upperLimit=fSTimePDFLimitsQueuePositive, int directionality = true, std::vector* fDirection_Filter = nullptr, float FilterThreshold = 180); - - private: - - LEAF(); - ~LEAF(); - - struct FitterOutput NewOutuput(); - - void LoadHitCollection(const HitCollection *lHitCol, const TimeDelta lTriggerTime, bool bMultiPMT); - - //##### VERTEX FITTER ##### - - void FitVertex(std::vector* fDirection_Filter = nullptr, float FilterThreshold = 180); - - //? Coarse GRID Search Raw version, not parallelized, slow - std::vector< std::vector > SearchVertex(int nhits, int tolerance=1, bool likelihood=false, double lowerLimit=fSTimePDFLimitsQueueNegative, double upperLimit=fSTimePDFLimitsQueuePositive, int directionality=false); - - //? Coarse GRID Search Parallelized version - std::vector SearchVertex_Main(int nhits, int tolerance, bool likelihood, double lowerLimit, double upperLimit, int directionality); - - //? Coarse GRID Search used for step by step vertex fit, written in 2023, not used and not tested - std::vector< std::vector > SearchVertexFine(std::vector< std::vector > initialVertex, double * limits, double stepSize, int nhits, int nCandidates = 1, int tolerance = 1, int verbose=0, bool likelihood=false, bool average=false, double lowerLimit=fSTimePDFLimitsQueueNegative, double upperLimit=fSTimePDFLimitsQueuePositive, int directionality=false); - - //? LLH Minimization Raw version, not parallelized, slow - std::vector< std::vector > MinimizeVertex(std::vector< std::vector > initialVertex, double * limits, double stepSize, int nhits, int nCandidates = 1, int tolerance = 1, int verbose=0, bool likelihood=false, bool average=false, double lowerLimit=fSTimePDFLimitsQueueNegative, double upperLimit=fSTimePDFLimitsQueuePositive, int directionality = true); - - //? Second Step Parallelized version - std::vector< std::vector > MinimizeVertex_Main(std::vector< std::vector > initialVertex, double * limits, double stepSize, int nhits, int nCandidates = 1, int tolerance = 1, int verbose=0, bool likelihood=false, bool average=false, double lowerLimit=fSTimePDFLimitsQueueNegative, double upperLimit=fSTimePDFLimitsQueuePositive, int directionality = true, std::vector* fDirection_Filter = nullptr, float FilterThreshold = 180); - - //##### Direction FITTER ##### - - // Global fitter function that selects canidates and call the minimize method - std::vector FitDirection(const std::vector& fixedVertexPosition, int nhits, bool searchPrior = false); - - // Determine direction using the sum of all unit vectors from vertex to PMTs. Can be used as a prefit - std::vector FitDirectionQuick(const std::vector& fixedVertexPosition); - - // Coarse GRID Search - std::vector SearchDirection(const std::vector& fixedVertexPosition, int tolerance); - - // LLH Minimization Search - DirectionCandidate MinimizeDirection(const std::vector& fixedVertexPosition, std::vector& candidates, int nhits, double *limits, int verbose); - - //##### ENERGY FITTER ##### - - void FitEnergy(); - - //##### JOINT FITTER ##### - - // Search combined candidates - std::vector SearchVertexAndDir(); - - // Minimize these candidates - JointFitCandidate MinimizeVertexAndDir(std::vector initialCandidates); - - static LEAF* myFitter; - - std::vector< std::vector > fThreadOutput; -}; - -#endif diff --git a/leaf/LEAF.hpp b/leaf/LEAF.hpp new file mode 100644 index 0000000..898e7c1 --- /dev/null +++ b/leaf/LEAF.hpp @@ -0,0 +1,379 @@ +/*****************************************************************************************************/ +// LEAF.hpp +// Authors: Guillaume Pronost (2019 - active) (pronost@km.icrr.u-tokyo.ac.jp) +// Benjamin Quilain (2019 - active) +// Lorenzo Perisse (2025 - active) +// Nicolas Moreau (???? - 2025) +// Shota Izumiyama (2020 - 2020) +// Original author: Benjamin Quilain +// Date: December 18th 2019 +// Desc: Low-E Fitter for Hyper-K +/*****************************************************************************************************/ + +/*****************************************************************************************************/ +// This class contains a vertex finder for Low-E WCSim-based detectors through different methods +// The vertex finder relies on a likelihood of timing residual for each PMT hit. +// Time residuals are defined as: +// time - time-of-flight-assuming-straight-line-from-vertex-to PMT - time-of-vertex +/*****************************************************************************************************/ + +/*****************************************************************************************************/ +// The standard method relies on two parts which are runned successively: +// +// 1. SearchVertex: A coarse grid search in the tank, where steps in space/time can be set by users. +// In default mode, this class does not use the full likelihood but just number of hits in time to +// be faster +// +// 2. MinimizeVertex: A MINUIT-based search within a 4D sphere around a specific position and time +// Note that in default mode, the sphere radius should be set to be the same or similar to the step +// size in SearchVertex. In default mode, the best candidates from SearchVertex are fed to +// MinimizeVertex. The number of best candidates can be selected. // Both MinimizeVertex and +// SearchVertex provides an output containing the best fits vertices, ordered from lower to higher +// NLL. +// +// Most of default parameters of method 1 and 2 can be set in LEAF.cc Init() method +/*****************************************************************************************************/ + +#ifndef LEAF_HPP +#define LEAF_HPP + +#include +#include +#include +#include +#include +#include +#include + +//* ROOT Headers +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + + +#include "Constants/mPMTs.hpp" +#include "Enums/PMTType.hpp" +#include "HKGeometry.hpp" +#include "HKDarkNoise.hpp" +#include "HKHit.hpp" + +#include "LEAFConfig.hpp" +#include "LEAFUtilities.hpp" +#include "SplineUtilities.hpp" + +std::mutex mtx; + +namespace LEAFStructures { + + struct FitterOutput { + + void Initialize() { + vtx = ROOT::Math::XYZTVector(); + vtx_nll = 0.0; + nll_r = 0.0; + dir = ROOT::Math::XYZVector(); + dir_pre_fit = ROOT::Math::XYZVector(); + dir_nll = 0.0; + energy = 0.0; + total_charge = 0.0; + } + + ROOT::Math::XYZTVector vtx; // leaf vertex + double vtx_nll; // Vertex Likelihood + double nll_r; // Goodness of fit + + ROOT::Math::XYZVector dir; // leaf Direction + ROOT::Math::XYZVector dir_pre_fit; // leaf Direction pre-fit + ROOT::Math::XYZVector dir_quick; // leaf Direction quick fit + double dir_nll; // Direction Likelihood + + double energy; // leaf energy + double total_charge; // raw total charge before affine correction + }; + + struct FitterPerformances { + // Computations times + double leaf_ct = 0.0; //Total leaf computational time + double vtx_search_ct = 0.0; //Vertex coarse grid search computational time + double vtx_minimize_ct = 0.0; //Vertex MINUIT minimisation computational time + double dir_quick_search_ct = 0.0; //Direction using unit vectors from vector to PMTT - computational time + double dir_search_ct = 0.0; //Direction coarse grid search computational time + double dir_minimize_ct = 0.0; //Direction MINUIT minimisation computational time + double energy_fit_ct = 0.0; //Energy finder computational time + }; + + struct VtxCandidate { + ROOT::Math::XYZTVector vtx; + double nll = 0.0; + }; +} + +namespace LEAFThreads { + void SearchVertex_thread(int iStart, int iIte, int tolerance=1, bool likelihood=false, double lowerLimit=LEAFConfig::fSTimePDFLimitsQueueNegative, double upperLimit=LEAFConfig::fSTimePDFLimitsQueuePositive, int directionality=false); + void MinimizeVertex_thread(int iStart, int iIte, std::vector initialVertex, ROOT::Math::XYZTVector limits, double stepSize, int nCandidates = 1, int tolerance = 1, int verbose=0, bool likelihood=false, bool average=false, double lowerLimit=LEAFConfig::fSTimePDFLimitsQueueNegative, double upperLimit=LEAFConfig::fSTimePDFLimitsQueuePositive, int directionality = true, float FilterThreshold = 180); +}; + +namespace LEAFLikelihoods { + void MinuitDirNLL(int& nDim, double* gout, double& NLL, double* par, int flg); + void MinuitLikelihood(int& nDim, double * gout, double & NLL, double par[], int flg); + void MinuitJointNLL(int& nDim, double * gout, double & NLL, double par[], int flg); +}; + +class LEAF +{ + public: + static LEAF* GetME(); + void DeleteME(); + + /*****************/ + /* CONFIGURATION */ + /*****************/ + +#ifdef HK_USE_ROOT7 + void Initialize(std::shared_ptr lGeometry, std::shared_ptr lDarkNoise, std::shared_ptr lGeoPMT_ID, std::shared_ptr lGeoPMT_mPMT); +#else + void Initialize(const HKGeometry *lGeometry, const HKDarkNoise *lDarkNoise, const HKGeometryPMTCollection *lGeoPMT_ID, const HKGeometryPMTCollection *lGeoPMT_mPMT); +#endif + void SetNThread(int iThread=N_THREAD); + + //* True Vertex isn't used for the fit, just to get feedback on how well the fit is at different steps + void SetTrueVertexInfo(ROOT::Math::XYZTVector vtx); + void SetTrueDirInfo(ROOT::Math::XYZVector dir); + + /****************/ + /* MAIN PROCESS */ + /****************/ + +#ifdef HK_USE_ROOT7 + void LoadHitsCollection(std::shared_ptr lHitCol_ID, std::shared_ptr lHitCol_mPMT = nullptr); +#else + void LoadHitsCollection(const HKHitsCollection* lHitCol_ID, const HKHitsCollection* lHitCol_mPMT = nullptr); +#endif + + //* Fitter Main Method. Process the whole fit. + LEAFStructures::FitterOutput MakeSequentialFit(); + + //* slower and less performant for now + LEAFStructures::FitterOutput MakeJointFit(); + + //* Return performance information + LEAFStructures::FitterPerformances GetPerformances() { return fPerformances; } + + /*****************/ + /* VERTEX FITTER */ + /*****************/ + + void SearchVertex_thread(int iStart, int iIte, int tolerance=1, bool likelihood=false, double lowerLimit=LEAFConfig::fSTimePDFLimitsQueueNegative, double upperLimit=LEAFConfig::fSTimePDFLimitsQueuePositive, int directionality=false); + void MinimizeVertex_thread(int iStart, int iIte, std::vector initialVertex, ROOT::Math::XYZTVector limits, double stepSize, int nCandidates = 1, int tolerance = 1, int verbose=0, bool likelihood=false, bool average=false, double lowerLimit=LEAFConfig::fSTimePDFLimitsQueueNegative, double upperLimit=LEAFConfig::fSTimePDFLimitsQueuePositive, int directionality = true, float FilterThreshold = 180); + + /***************/ + /* LIKELIHOODS */ + /***************/ + + //* Calculate likelihood without using PDF, but just using hits within a given timing window (hits "in-time"). + double Vertex_Score(ROOT::Math::XYZTVector vertexPosition, double lowerLimit, double upperLimit, bool killEdges, bool scaleDR, int directionality); + + //* Contain the two functions above in one function, where usage of PDF or not can be set through the flag: likelihood = true/false + double FindNLL(ROOT::Math::XYZTVector vertexPosition, bool likelihood, int verbose, double lowerLimit, double upperLimit, bool killEdges=false, bool scaleDR=false, int directionality=false); + + double Vertex_Time_NLL(ROOT::Math::XYZTVector vertexPosition, double lowerLimit, double upperLimit, bool killEdges, bool scaleDR, int directionality); + + double Dir_NLL(const ROOT::Math::XYZTVector& vertexPosition, const double theta_track, const double phi_track); + + double ComputeDirNLL_NoPDF(const ROOT::Math::XYZTVector& vertexPosition, const ROOT::Math::XYZVector& vertexDirection); + + double AngleNLL(ROOT::Math::XYZTVector vertexPosition, ROOT::Math::XYZVector vertexDirection); + + double FindNLLDirectionality(ROOT::Math::XYZTVector vertexPosition, int verbose, double lowerLimit, double upperLimit); + + double GoodnessOfFit(ROOT::Math::XYZTVector vertexPosition, double NLL_theory, double lowerLimit, double upperLimit, bool killEdges, bool scaleDR, int directionality); + + private: + + /*******************/ + /* INTERNAL STRUCT */ + /*******************/ + + struct EventInfo { + int nHits = 0; + double SignalOverNoise = 0.0; + double NoiseIntegral = 0.0; + double SignalIntegral = 0.0; + }; + + struct DirectionCandidate { + double theta = 0.0; + double phi = 0.0; + double nll = 0.0; + }; + + struct JointFitCandidate { + ROOT::Math::XYZTVector vtx; + double theta = 0.0; + double phi = 0.0; + double nll = 0.0; + }; + + template + struct SortingNLL { + bool operator()(T const &a, T const &b) const + { + return a.nll < b.nll; + } + }; + + private: + + LEAF(); + ~LEAF(); + + static LEAF* myFitter; + std::vector fVtxThreadOutput; + std::vector fDirThreadOutput; + + /******************/ + /* INITIALIZATION */ + /******************/ + + void InitInputs(); + void MakePositionList(); + void LoadSplines(); + + /****************/ + /* DATA PROCESS */ + /****************/ + + void GenerateEventInfo(double lowerLimit, double upperLimit); + double CalculateSNR(ROOT::Math::XYZTVector vertex, double lowerLimit, double upperLimit); + + /*****************/ + /* VERTEX FITTER */ + /*****************/ + + void FitVertex(float FilterThreshold = 180); + + //? Coarse GRID Search Parallelized version + std::vector SearchVertex_Main(int tolerance, bool likelihood, double lowerLimit, double upperLimit, int directionality); + + //? Second Step Parallelized version + std::vector MinimizeVertex_Main(std::vector initialVertex, ROOT::Math::XYZTVector limits, double stepSize, int nCandidates = 1, int tolerance = 1, int verbose=0, bool likelihood=false, bool average=false, double lowerLimit=LEAFConfig::fSTimePDFLimitsQueueNegative, double upperLimit=LEAFConfig::fSTimePDFLimitsQueuePositive, int directionality = true, float FilterThreshold = 180); + + /********************/ + /* DIRECTION FITTER */ + /********************/ + + // Global fitter function that selects canidates and call the minimize method + ROOT::Math::XYZVector FitDirection(const ROOT::Math::XYZTVector& vertex, bool searchPrior = false); + + // Determine direction using the sum of all unit vectors from vertex to PMTs. Can be used as a prefit + ROOT::Math::XYZVector FitDirectionQuick(const ROOT::Math::XYZTVector& vertex); + + // Coarse GRID Search + std::vector SearchDirection(const ROOT::Math::XYZTVector& vertex, int tolerance); + + // LLH Minimization Search + DirectionCandidate MinimizeDirection(const ROOT::Math::XYZTVector& vertex, std::vector& candidates, ROOT::Math::XYZTVector limits, int verbose); + + + /*****************/ + /* ENERGY FITTER */ + /*****************/ + + void FitEnergy(); + + /****************/ + /* JOINT FITTER */ + /****************/ + + // Search combined candidates + std::vector SearchVertexAndDir(); + + // Minimize these candidates + JointFitCandidate MinimizeVertexAndDir(std::vector initialCandidates); + + private: + + /*********************/ + /* Private variables */ + /*********************/ + + LEAFStructures::FitterOutput fOutput; + LEAFStructures::FitterPerformances fPerformances; // additional infos about the fit + ROOT::Math::XYZVector fDirectionFilter; + bool fDirectionFilterSet; + + //! Number of threads to use for parallelization + int fThread; + + //! Random number generator + TRandom3 * fRand; + + //! Dark rate for each PMT type, in Hz + std::array(PMTType::kNumPMTTypes)> fDarkRate_ns; + + //! List of position for grid coarse search + std::vector fPositionList; + + //! True vertex position, use for debug + ROOT::Math::XYZTVector fTrueVtxPos; + + //! True vertex direction, use for debug + ROOT::Math::XYZVector fTrueDir; + + //! Splines + std::array(PMTType::kNumPMTTypes)> fSplineTimePDFConv; + std::array(PMTType::kNumPMTTypes)> fSplineTimePDFQueue; + std::array(PMTType::kNumPMTTypes)> fSplineTimePDFDarkRate; + std::array(PMTType::kNumPMTTypes)> fDirectionPDF; + std::array(mPMTConstants::kNGroups)>, static_cast(PMTType::kNumPMTTypes)> gPMTDirectionality_2D; + std::array(PMTType::kNumPMTTypes)> fDistResponsePMT; + + std::array(mPMTConstants::kNGroups)>, static_cast(PMTType::kNumPMTTypes)> fDarkRate_dir_proba; + double fLastLowerLimit; + double fLastUpperLimit; + + //! + std::array(PMTType::kNumPMTTypes)> fEventInfo; + + //! DataModel +#ifdef HK_USE_ROOT7 + //! Geometry object + std::shared_ptr fGeometry; + + //! Dark noise object + std::shared_ptr fDarkNoise; + + //! PMT Informations + std::array, static_cast(PMTType::kNumPMTTypes)> fGeoPMTs; + + //! Hit Collections + std::array, static_cast(PMTType::kNumPMTTypes)> fHitsCollection; + +#else + //! Geometry object + const HKGeometry* fGeometry; + + //! Dark noise object + const HKDarkNoise* fDarkNoise; + + //! PMT Informations + std::array(PMTType::kNumPMTTypes)> fGeoPMTs; + + //! Hit Collections + std::array(PMTType::kNumPMTTypes)> fHitsCollection; +#endif +}; + +#endif diff --git a/leaf/LEAFConfig.hpp b/leaf/LEAFConfig.hpp new file mode 100644 index 0000000..f08dcb5 --- /dev/null +++ b/leaf/LEAFConfig.hpp @@ -0,0 +1,200 @@ +#ifndef LEAFCONFIG_HPP +#define LEAFCONFIG_HPP + +#include +#include +#include +#include +#include "Enums/PMTType.hpp" + +#define VERBOSE 0 + +#undef VERBOSE_VTX // In SearchVertex +#undef VERBOSE_NLL // In Likelihood Computation + +#define VTX_X 0 +#define VTX_Y 1 +#define VTX_Z 2 +#define VTX_T 3 + +#define N_THREAD 2 // Default for sukap +#define C_WATER_CM_NS 21.849964 // light celerity in water, in centimeter per ns. + +namespace LEAFConfig { + + //* Geometry & Constants + double fTankRadius; + double fTankHeight; + double fTankHalfHeight; + double fLightSpeed = C_WATER_CM_NS; + + //* PDFs + double fSTimePDFLimitsQueueNegative = -3.; + double fSTimePDFLimitsQueuePositive = 4.; + double fSTimePDFLimitsQueueNegative_fullTimeWindow = 0.; + double fSTimePDFLimitsQueuePositive_fullTimeWindow = 0.; + double fPDFNorm_fullTimeWindow = 0.; + + double fDirectionPDF_maxResidual = 15.0; + double fDirectionPDF_minResidual = -5.0; + + //* Overall Fit Parameters + double fTimeWindowSizeFull = 1500.; + double fIntegrationTimeWindow = 50.; + bool fStepByStep = false; + bool fUseDirectionality = false; + bool fDoubleNLL = false; + bool fLimit_mPMT = true; + int fAveraging = 20; + bool fHighEnergy = false; + double fSearchVtxStep = 300.; + double fSearchVtxTolerance = 60; + + //* Step 1 Specific Parameters + double fHitTimeLimitsNegative = -5; + double fHitTimeLimitsPositive = -7; + + //* Step 2 Specific Parameters + double fMinimizeLimitsNegative = -700; + double fMinimizeLimitsPositive = 1000; + + //* Direction Fit Parameters + bool fDirTakeAll = false; + double fThetaStep = 5 * TMath::DegToRad(); + double fPhiStep = 5 * TMath::DegToRad(); + int fDirTolerance = 50; + + //! List of active PMT types + std::vector fActivePMTTypes; + + void Initialize(double lTankRadius, double lTankHeight) { + fTankRadius = lTankRadius; + fTankHeight = lTankHeight; + fTankHalfHeight = fTankHeight / 2.; + + fStepByStep = false; // Step by step mode was not test and is not supported by multithreading + fUseDirectionality = false; + fHighEnergy = false; + fDoubleNLL = true; + + // If true, we apply same cuts for timing as B&L PMT when searching vertex. + // Otherwise, we do not apply them and use all the hits. + // The latter is particularly useful when directionality is added, as it can kill DR hits. + fLimit_mPMT = true; + + fSTimePDFLimitsQueueNegative = -3.; //-5;//-3;//-1.5;//-3;//-10; + fSTimePDFLimitsQueuePositive = 4.; // 10;//4;//3;//4;//500; + fSTimePDFLimitsQueueNegative_fullTimeWindow = 0.; //-1.5;//-3;//-10; + fSTimePDFLimitsQueuePositive_fullTimeWindow = 0.; // 3;//4;//500; + fTimeWindowSizeFull = 1500; + fAveraging = 20; // Number of points we used to average the vertex on. + + fMinimizeLimitsNegative = -700; + fMinimizeLimitsPositive = 1000; + + fHitTimeLimitsNegative = -5; //-10;//-5;//-5 //-5 + fHitTimeLimitsPositive = 7; // 15;//7;//7 //7 + + fSearchVtxStep = 300; // in cm, the step size for coarse grid search + + fSearchVtxTolerance = 60; // Number of candidate vertex that are kept after coarse grid search + + fIntegrationTimeWindow = 50; // in ns //? never Used ??? + fDirTakeAll = false; + fThetaStep = 5 * TMath::DegToRad(); + fPhiStep = 5 * TMath::DegToRad(); + fDirTolerance = 50; + + //* Compute light speed: + fLightSpeed = C_WATER_CM_NS; + + fPDFNorm_fullTimeWindow = 0.; + + fActivePMTTypes = {PMTType::kID}; + //fActivePMTTypes = {PMTType::kID, PMTType::kmPMT}; + + //*** These maps describe what will be loaded from the config file given */ + + std::map envVariables = + { + {"SearchVtxStep", &fSearchVtxStep}, + {"SearchVtxTolerance", &fSearchVtxTolerance}, + {"HitTimeLimitsNegative", &fHitTimeLimitsNegative}, + {"HitTimeLimitsPositive", &fHitTimeLimitsPositive}, + {"MinimizeLimitsNegative", &fMinimizeLimitsNegative}, + {"MinimizeLimitsPositive", &fMinimizeLimitsPositive}, + {"IntegrationTimeWindow", &fIntegrationTimeWindow}, + {"TimeWindowSizeFull", &fTimeWindowSizeFull}, + {"STimePDFLimitsQueueNegative", &fSTimePDFLimitsQueueNegative}, + {"STimePDFLimitsQueuePositive", &fSTimePDFLimitsQueuePositive}, + {"STimePDFLimitsQueueNegative_fullTimeWindow", &fSTimePDFLimitsQueueNegative_fullTimeWindow}, + {"STimePDFLimitsQueuePositive_fullTimeWindow", &fSTimePDFLimitsQueuePositive_fullTimeWindow}, + {"DirectionPDF_maxResidual", &fDirectionPDF_maxResidual}, + {"DirectionPDF_minResidual", &fDirectionPDF_minResidual}, + {"ThetaStep", &fThetaStep}, + {"PhiStep", &fPhiStep} + }; + + std::map envIntVariables = + { + {"Averaging", &fAveraging}, + {"DirTolerance", &fDirTolerance} + }; + + std::map envBoolVariables = + { + {"Limit_mPMT", &fLimit_mPMT}, + {"UseDirectionality", &fUseDirectionality}, + {"StepByStep", &fStepByStep}, + {"HighEnergy", &fHighEnergy}, + {"DoubleNLL", &fDoubleNLL}, + {"DirTakeAll", &fDirTakeAll}, + }; + + for (const auto& [envName, variable] : envVariables) + { + const char* envValue = std::getenv(envName.c_str()); + if (envValue != nullptr) + { + *variable = std::stod(envValue); + std::cout << envName << " is set to " << *variable << std::endl; + } + else std::cout << "Environment variable " << envName << " is not set. Using default value : "<< *variable << std::endl; + } + + for (const auto& [envName, variable] : envIntVariables) + { + const char* envValue = std::getenv(envName.c_str()); + if (envValue != nullptr) + { + *variable = std::stoi(envValue); + std::cout << envName << " is set to " << *variable << std::endl; + } + else std::cout << "Environment variable " << envName << " is not set. Using default value : "<< *variable << std::endl; + } + + for (const auto& [envName, variable] : envBoolVariables) + { + const char* envValue = std::getenv(envName.c_str()); + if (envValue != nullptr) + { + std::string valueStr(envValue); + if (valueStr == "true" || valueStr == "1") + { + *variable = true; + std::cout << envName << " is set to true" << std::endl; + } + else if (valueStr == "false" || valueStr == "0") + { + *variable = false; + std::cout << envName << " is set to false" << std::endl; + } + else std::cout << "Environment variable " << envName << " has an invalid value. Using default value : "<< *variable << std::endl; + } + else std::cout << "Environment variable " << envName << " is not set. Using default value : "<< *variable << std::endl; + } + + } +}; // namespace LEAFConfig + +#endif \ No newline at end of file diff --git a/leaf/LEAFSrcFitter.cpp b/leaf/LEAFSrcFitter.cpp new file mode 100644 index 0000000..b5a6358 --- /dev/null +++ b/leaf/LEAFSrcFitter.cpp @@ -0,0 +1,674 @@ +/*****************************************************************************************************/ +/** LEAFSrcFitter.cc **/ +/** Date: March 27th 2026 **/ +/** Desc: Implementation of the Fitter functions **/ +/*****************************************************************************************************/ + +#ifndef LEAF_HPP +#include "LEAF.hpp" +#endif + +/*****************************************************************************************************/ +/* DIRECTION FIT */ +/*****************************************************************************************************/ + +//Creates the candidates list after Coarse grid search, refines the fit, then outputs the best candidate +ROOT::Math::XYZVector LEAF::FitDirection(const ROOT::Math::XYZTVector& vertex, bool searchPrior) { + //* Init + TStopwatch timer; + timer.Reset(); + timer.Start(); + + ROOT::Math::XYZTVector limits(1.,1.,1.,1.); + limits *= 2.*LEAFConfig::fSearchVtxStep; + + int verbose = VERBOSE; + + //* Fit first step : find candidates for the direction + std::vector candidates; + if(searchPrior) { + //* Only one candidate, chosen wisely by computing a mean hit direction for hits within a specified residual time window + ROOT::Math::XYZVector dirPriorCart = this->FitDirectionQuick(vertex); + ROOT::Math::Polar3DVector dirPriorPol(dirPriorCart); + + DirectionCandidate priorCandidate; + priorCandidate.theta = dirPriorPol.Theta(); + priorCandidate.phi = dirPriorPol.Phi(); + candidates.push_back(priorCandidate); + } + else { + candidates = this->SearchDirection(vertex, LEAFConfig::fDirTolerance); //* multiple candidates, spread uniformly + } + + // std::cout << "nb direction candidates : " << candidates.size() << std::endl; + + fPerformances.dir_search_ct = timer.RealTime(); + TStopwatch timer2; + timer2.Reset(); + timer2.Start(); + + //? To be removed for real data where true dir is not known, no impact on the fit + // fOutput.stepOneContainsTrueDir = (int)ContainsTrueDir(&candidates, fTrueDir); //* a quick check to see if the step one works (one of the candidate is near the true direction) + + if (verbose >= 2) { + for (unsigned int j = 0; j < candidates.size(); j++) { + std::cout << "Candidate " + << "theta = " << candidates[j].theta * TMath::RadToDeg() << " deg, " + << "phi = " << candidates[j].phi * TMath::RadToDeg() << " deg, " + << "NLL = " << candidates[j].nll + << std::endl; + } + } + + DirectionCandidate finalCandidate = this->MinimizeDirection(vertex, candidates, limits, verbose); + + ROOT::Math::Polar3DVector dirPol(1, finalCandidate.theta, finalCandidate.phi); + fOutput.dir = ROOT::Math::XYZVector(dirPol.Unit()); // Unit is probably not needed? + fOutput.dir_nll = finalCandidate.nll; + + fPerformances.dir_minimize_ct = timer2.RealTime(); + + return fOutput.dir; +} + +ROOT::Math::XYZVector LEAF::FitDirectionQuick(const ROOT::Math::XYZTVector& vertex) { + // Quick fit implementation + + TStopwatch timer; + timer.Reset(); + timer.Start(); + + ROOT::Math::XYZVector vertexPos(vertex.X(), vertex.Y(), vertex.Z()); + + std::vector scaledDistToPMTs; + std::vector residuals; + + double dir_x = 0.0; + double dir_y = 0.0; + double dir_z = 0.0; + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + const int iPMTType = static_cast(pmtType); + for(unsigned int i = 0; i < fHitsCollection[iPMTType]->size(); i++) { + std::vector toHit; + + const HKHit* lHit = fHitsCollection[iPMTType]->at(i); + int iPMT = lHit->GetPMTNumber(); + const HKGeometryPMT* lPMTInfo = fGeoPMTs[iPMTType]->at(iPMT); + + ROOT::Math::XYZVector distToPMT = (lPMTInfo->GetPositionInCm() - vertexPos); + double residual = LEAFUtilities::GetResidual(distToPMT, lHit->GetTime(), vertex.T()); + + if ( residual < LEAFConfig::fDirectionPDF_maxResidual && residual > LEAFConfig::fDirectionPDF_minResidual) { + ROOT::Math::XYZVector scaledDist = (distToPMT.Unit() * lHit->GetCharge()); + dir_x += scaledDist.X(); + dir_y += scaledDist.Y(); + dir_z += scaledDist.Z(); + } + } + } + + fPerformances.dir_quick_search_ct = timer.RealTime(); + fOutput.dir_quick = ROOT::Math::XYZVector(dir_x, dir_y, dir_z).Unit(); + return ROOT::Math::XYZVector(dir_x, dir_y, dir_z).Unit(); +} + +std::vector LEAF::SearchDirection(const ROOT::Math::XYZTVector& vertex, int tolerance) +{ + std::vector candidates; + + // Grid search over theta and phi + for (double theta = 0.0; theta <= TMath::Pi(); theta += LEAFConfig::fThetaStep) { + for (double phi = -TMath::Pi(); phi < TMath::Pi(); phi += LEAFConfig::fPhiStep) { + // Store the candidate + DirectionCandidate candidate; + candidate.theta = theta; + candidate.phi = phi; + candidate.nll = this->Dir_NLL(vertex, theta, phi); + candidates.push_back(candidate); + } + } + + + std::sort(candidates.begin(), candidates.end(), LEAF::SortingNLL()); + + // Select the best candidates based on tolerance + while((int)candidates.size() > tolerance) candidates.pop_back(); + + return candidates; +} + +//MIGRAD Optimization +LEAF::DirectionCandidate LEAF::MinimizeDirection(const ROOT::Math::XYZTVector& vertex, std::vector& candidates, ROOT::Math::XYZTVector limits, int verbose) { + // Create a minimizer + TFitter* minimizer = new TFitter(7); // 2 parameters: theta, phi + TMinuit* minuit = minimizer->GetMinuit(); + + double arglist[10]; + int err = 0; + double p1 = verbose - 1; + minimizer->ExecuteCommand("SET PRINTOUT", &p1, 1); + minuit->SetErrorDef(1); + + if (verbose < 2) minuit->mnexcm("SET NOWarnings", 0, 0, err); + + arglist[0] = 2; + minuit->mnexcm("SET STR", arglist, 1, err); + minimizer->SetFCN(LEAFLikelihoods::MinuitDirNLL); + + for (int icand = 0; icand < (int)candidates.size(); icand++) + { + // Set initial parameters and bounds + minimizer->SetParameter(0, "theta", candidates[icand].theta, 2*TMath::Pi()/180, 0.0, TMath::Pi()); + minimizer->SetParameter(1, "phi", candidates[icand].phi, 2*TMath::Pi()/180, -TMath::Pi(), TMath::Pi()); + minimizer->SetParameter(2, "nhits", 0, 0, -1, +1); + minimizer->SetParameter(3, "vertex0", vertex.X(), 5e-1, -limits.X(), limits.X()); + minimizer->SetParameter(4, "vertex1", vertex.Y(), 5e-1, -limits.Y(), limits.Y()); + minimizer->SetParameter(5, "vertex2", vertex.Z(), 5e-1, -limits.Z(), limits.Z()); + minimizer->SetParameter(6, "vertex3", vertex.T(), 5e-1 / LEAFConfig::fLightSpeed, -limits.T() / LEAFConfig::fLightSpeed, limits.T() / LEAFConfig::fLightSpeed); + + minimizer->FixParameter(2); + minimizer->FixParameter(3); + minimizer->FixParameter(4); + minimizer->FixParameter(5); + minimizer->FixParameter(6); + + // Minimize + arglist[0] = 1e9; // Maximum number of function evaluations + arglist[1] = 1e-1; // Tolerance + minimizer->ExecuteCommand("MIGRAD", arglist, 2 ); + + // Retrieve results + candidates[icand].theta = minimizer->GetParameter(0); // theta + candidates[icand].phi = minimizer->GetParameter(1); // phi + + minuit = minimizer->GetMinuit(); + double fmin, fedm, errdef; + int npar, nparx, istat; + minuit->mnstat(fmin, fedm, errdef, npar, nparx, istat); + + candidates[icand].nll = fmin; + } + + std::sort(candidates.begin(), candidates.end(), LEAF::SortingNLL()); + // Optional: Print the refined parameters + // if (verbose >= 2) std::cout << "After refinement: theta = " << optimizedParameters[0] * TMath::RadToDeg() << " deg, phi = " << optimizedParameters[1] * TMath::RadToDeg() << " deg" << std::endl; + + delete minimizer; + return candidates[0]; // Returns [theta, phi] +} + + +/*****************************************************************************************************/ +/* VERTEX FIT */ +/*****************************************************************************************************/ + + +void LEAF::FitVertex(float FilterThreshold) { + std::vector fRecoVtxPosFinal; + + TStopwatch timer; + timer.Reset(); + timer.Start(); + + std::vector tRecoVtxPos = this->SearchVertex_Main(LEAFConfig::fSearchVtxTolerance,false,LEAFConfig::fSTimePDFLimitsQueueNegative,LEAFConfig::fSTimePDFLimitsQueuePositive,false); + + if (VERBOSE >= 2) { + for (unsigned int i=0; iMinimizeVertex_Main(tRecoVtxPos, limits, dStepSizeFinal, LEAFConfig::fSearchVtxTolerance, iToleranceFinal, VERBOSE, true, false, LEAFConfig::fMinimizeLimitsNegative, LEAFConfig::fMinimizeLimitsPositive, LEAFConfig::fUseDirectionality, FilterThreshold); + + timer.Stop(); + fPerformances.vtx_minimize_ct = timer.RealTime(); + + fOutput.vtx = fRecoVtxPosFinal[0].vtx; + fOutput.vtx_nll = fRecoVtxPosFinal[0].nll; +} + +std::vector LEAF::SearchVertex_Main(int tolerance, bool likelihood, double lowerLimit, double upperLimit, int directionality) { +// 1) Launch threads just like in SearchVertex_Main. + std::vector lThreadList; + std::vector lOutputFinal; + TStopwatch timer; + timer.Reset(); + timer.Start(); + + int iCand_Step = (int)fPositionList.size()/fThread; + if(iCand_Step < 1) iCand_Step = 1; + + mtx.lock(); + fVtxThreadOutput.clear(); + mtx.unlock(); + + for(int iStart=0; iStart<(int)fPositionList.size(); iStart+=iCand_Step) + { + std::thread tThrd( + LEAFThreads::SearchVertex_thread, + iStart, iCand_Step, + tolerance, + likelihood, lowerLimit, upperLimit, directionality + ); + lThreadList.push_back(std::move(tThrd)); + } + + // std::cout << "time after vtx search threads in SearchVertex_Main : " << timer.RealTime() << std::endl; + + for(auto &th: lThreadList) if(th.joinable()) th.join(); + + // 2) collect partial results from fThreadOutput. + mtx.lock(); + lOutputFinal = fVtxThreadOutput; + fVtxThreadOutput.clear(); + mtx.unlock(); + + // 3) Sort by NLL and keep 'tolerance' number of candidates + std::sort(lOutputFinal.begin(), lOutputFinal.end(), LEAF::SortingNLL()); + while((int)lOutputFinal.size() > tolerance) lOutputFinal.pop_back(); + + //4) Now build a vector of VtxCandidate. + std::vector finalList; + finalList.reserve(lOutputFinal.size()); + + for(auto &cand : lOutputFinal) { + finalList.push_back(cand); + } + + return finalList; +} + +// Coarse search of the vertex. It will search in a cylinder having the radius = tankRadius and a height tankHeight. +// The step between each grid points is given by stepSize (in centimeters). +// We give the list of hit information through the 2D array fHitInfo to make the code faster, instead of using the whole code information +// The code provide as an output not only one vertex, but a list of vertices whose likelihood is higher (Negative Log Likelihood is lower). +// The number of output vertices can be chosen using "tolerance". +void LEAFThreads::SearchVertex_thread(int iStart, int iIte, int tolerance, bool likelihood, double lowerLimit, double upperLimit, int directionality) { + LEAF::GetME()->SearchVertex_thread(iStart, iIte, tolerance, likelihood, lowerLimit, upperLimit, directionality); +} + +void LEAF::SearchVertex_thread(int iStart, int iIte, int tolerance, bool likelihood, double lowerLimit, double upperLimit, int directionality) { + std::vector tVtxContainer; + + TStopwatch timer; + timer.Reset(); + timer.Start(); + + if (fPositionList.size() == 0) + std::cout << " Position list not filled " << std::endl; + + int iEnd = (iIte + iStart); + if (iEnd > (int)fPositionList.size()) iEnd = fPositionList.size(); + + for (int iPos = iStart; iPos < iEnd; iPos++) { + + LEAFStructures::VtxCandidate hPos; + hPos.vtx = fPositionList[iPos]; + hPos.nll = 0; + + if (likelihood) hPos.nll = this->Vertex_Time_NLL(hPos.vtx, lowerLimit, upperLimit, false, false, directionality); + else hPos.nll = this->Vertex_Score(hPos.vtx, lowerLimit, upperLimit, false, false, directionality); + + //if (iPos < 10) + // std::cout << iPos << " -> " << hPos.vtx.X() << ", " << hPos.vtx.Y() << ", " << hPos.vtx.Z() << ", " << hPos.vtx.T() << " = " << hPos.nll << " " << likelihood << std::endl; + + tVtxContainer.push_back(hPos); + } + + std::sort(tVtxContainer.begin(), tVtxContainer.end(), LEAF::SortingNLL()); + while ((int)tVtxContainer.size() > tolerance) tVtxContainer.pop_back(); + + // Check if mutex is already lock, if not -> lock + // We don't want simultaneous modification of the common output + mtx.lock(); + for (unsigned int iPos = 0; iPos < tVtxContainer.size(); iPos++) { + fVtxThreadOutput.push_back(tVtxContainer[iPos]); + } + mtx.unlock(); +} + +std::vector LEAF::MinimizeVertex_Main(std::vector initialVertex, ROOT::Math::XYZTVector limits, double stepSize, int nCandidates, int tolerance, int verbose, bool likelihood, bool average, double lowerLimit, double upperLimit, int directionality, float FilterThreshold) +{ + std::vector lThreadList; + std::vector lOutputFinal; + + int iCand_Step = nCandidates / fThread; + if (iCand_Step < 1) iCand_Step = 1; + + mtx.lock(); + fVtxThreadOutput.clear(); + mtx.unlock(); + + for (int iStart = 0; iStart < nCandidates; iStart += iCand_Step) { + std::thread tThrd(LEAFThreads::MinimizeVertex_thread, iStart, iCand_Step, + initialVertex, limits, stepSize, nCandidates, + tolerance, verbose, likelihood, average, lowerLimit, upperLimit, directionality, FilterThreshold); + + lThreadList.push_back(std::move(tThrd)); + } + + for(auto &th: lThreadList) if(th.joinable()) th.join(); + + // Check if mutex is already lock, if not -> lock + // We don't want simultaneous modification of the common output + mtx.lock(); + lOutputFinal = fVtxThreadOutput; + fVtxThreadOutput.clear(); + mtx.unlock(); + + std::sort(lOutputFinal.begin(), lOutputFinal.end(), LEAF::SortingNLL()); + + while ((int)lOutputFinal.size() > tolerance) lOutputFinal.pop_back(); + + return lOutputFinal; +} + +// here is the function where minimization gonna take place. The loop is inside this function +void LEAFThreads::MinimizeVertex_thread(int iStart, int iIte, std::vector initialVertex, ROOT::Math::XYZTVector limits, double stepSize, int nCandidates, int tolerance, int verbose, bool likelihood, bool average, double lowerLimit, double upperLimit, int directionality, float FilterThreshold) { + LEAF::GetME()->MinimizeVertex_thread(iStart, iIte, initialVertex, limits, stepSize, nCandidates, tolerance, verbose, likelihood, average, lowerLimit, upperLimit, directionality, FilterThreshold); +} + +void LEAF::MinimizeVertex_thread(int iStart, int iIte, std::vector initialVertex, ROOT::Math::XYZTVector limits, double stepSize, int nCandidates, int tolerance, int verbose, bool /*likelihood*/, bool /*average*/, double lowerLimit, double upperLimit, int directionality, float FilterThreshold) { + int iEnd = (iIte + iStart); + if (iEnd > nCandidates) iEnd = nCandidates; + + std::vector tVtxContainer; + + mtx.lock(); + TFitter *minimizer = new TFitter(14); // 4=nb de params? + mtx.unlock(); + TMinuit *minuit = minimizer->GetMinuit(); + + double arglist[20]; + int err = 0; + // arglist[0]=0; + double p1 = verbose - 1; + minimizer->ExecuteCommand("SET PRINTOUT", &p1, 1); // quiet mode + minuit->SetErrorDef(1); + // minuit->SetPrintLevel(verbose-1); // quiet mode + if (VERBOSE < 2) minuit->mnexcm("SET NOWarnings", 0, 0, err); + arglist[0] = 2; + minuit->mnexcm("SET STR", arglist, 1, err); // set strategy sets the number of derivative estimated. in the present case (2), this is the highest number of it with a counter-balance: very slow! + minimizer->SetFCN(LEAFLikelihoods::MinuitLikelihood); // here is function to minimize + + double Mig[2] = {1e6, 1e0}; // maxcalls and tolerance + + minimizer->SetParameter(0, "vertex0", 0, stepSize, -limits.X(), limits.X()); + minimizer->SetParameter(1, "vertex1", 0, stepSize, -limits.Y(), limits.Y()); + minimizer->SetParameter(2, "vertex2", 0, stepSize, -limits.Z(), limits.Z()); + minimizer->SetParameter(3, "vertex3", 0, stepSize / LEAFConfig::fLightSpeed, -limits.T() / LEAFConfig::fLightSpeed, limits.T() / LEAFConfig::fLightSpeed); + + minimizer->SetParameter(4, "PMTConfiguration", 0, 0, 0, 0); // Useless now + minimizer->SetParameter(5, "nhits", 0, 0, -1, 1); + minimizer->SetParameter(6, "lowerLimit", lowerLimit, 5e-1, -7, -2); + minimizer->SetParameter(7, "upperLimit", upperLimit, 5e-1, 2, 10); + minimizer->SetParameter(8, "expoSigma", 100, 5, 0, 1e3); + minimizer->SetParameter(9, "directionality", directionality, 1, 0, 2); + minimizer->SetParameter(10, "dir_x", 0, 0, -1, 1); + minimizer->SetParameter(11, "dir_y", 0, 0, -1, 1); + minimizer->SetParameter(12, "dir_z", 0, 0, -1, 1); + minimizer->SetParameter(13, "dir_threshold", 0, 0, 0, 0); + minimizer->FixParameter(4); + minimizer->FixParameter(5); + minimizer->FixParameter(6); + minimizer->FixParameter(7); + minimizer->FixParameter(8); + minimizer->FixParameter(9); + + for (int icand = iStart; icand < iEnd; icand++) { + minimizer->SetParameter(0, "vertex0", initialVertex[icand].vtx.X(), stepSize, initialVertex[icand].vtx.X() - limits.X(), initialVertex[icand].vtx.X() + limits.X()); + minimizer->SetParameter(1, "vertex1", initialVertex[icand].vtx.Y(), stepSize, initialVertex[icand].vtx.Y() - limits.Y(), initialVertex[icand].vtx.Y() + limits.Y()); + minimizer->SetParameter(2, "vertex2", initialVertex[icand].vtx.Z(), stepSize, initialVertex[icand].vtx.Z() - limits.Z(), initialVertex[icand].vtx.Z() + limits.Z()); + minimizer->SetParameter(3, "vertex3", initialVertex[icand].vtx.Y(), stepSize / LEAFConfig::fLightSpeed, initialVertex[icand].vtx.T() - limits.T() / LEAFConfig::fLightSpeed, initialVertex[icand].vtx.T() + limits.T() / LEAFConfig::fLightSpeed); + + minimizer->SetParameter(4, "PMTConfiguration", 0, 0, 0, 0); // Useless now + minimizer->SetParameter(5, "nhits", 0, 0, -1, 1); + minimizer->SetParameter(6, "lowerLimit", lowerLimit, 5e-1, -7, -2); + minimizer->SetParameter(7, "upperLimit", upperLimit, 5e-1, 2, 10); + minimizer->SetParameter(8, "expoSigma", 100, 5, 0, 1e3); + minimizer->SetParameter(9, "directionality", directionality, 1, 0, 2); + + if(fDirectionFilterSet) { + minimizer->SetParameter(10, "dir_x", fDirectionFilter.X(), 5e-1, fDirectionFilter.X() - 1, fDirectionFilter.X() + 1); + minimizer->SetParameter(11, "dir_y", fDirectionFilter.Y(), 5e-1, fDirectionFilter.Y() - 1, fDirectionFilter.Y() + 1); + minimizer->SetParameter(12, "dir_z", fDirectionFilter.Z(), 5e-1, fDirectionFilter.Z() - 1, fDirectionFilter.Z() + 1); + } + minimizer->SetParameter(13, "dir_threshold", FilterThreshold, 5e-1, FilterThreshold - 1, FilterThreshold + 1); + minimizer->FixParameter(4); + minimizer->FixParameter(5); + minimizer->FixParameter(6); + minimizer->FixParameter(7); + minimizer->FixParameter(8); + minimizer->FixParameter(9); + minimizer->FixParameter(10); + minimizer->FixParameter(11); + minimizer->FixParameter(12); + minimizer->FixParameter(13); + + minimizer->ExecuteCommand("MIGRAD", Mig, 2); + + LEAFStructures::VtxCandidate hPos; + hPos.vtx = ROOT::Math::XYZTVector(minimizer->GetParameter(0), minimizer->GetParameter(1), minimizer->GetParameter(2), minimizer->GetParameter(3)); + + // Get minuit + minuit = minimizer->GetMinuit(); + double fmin, fedm, errdef; + int npar, nparx, istat; + minuit->mnstat(fmin, fedm, errdef, npar, nparx, istat); + + hPos.nll = fmin; + + tVtxContainer.push_back(hPos); + + if (VERBOSE >= 2) { + std::cout << "Initial vertex position = (" + << initialVertex[icand].vtx.X() << "," + << initialVertex[icand].vtx.Y() << "," + << initialVertex[icand].vtx.Z() << "," + << initialVertex[icand].vtx.T() << "), " + << "Candidate vertex position = (" + << hPos.vtx.X() << "," + << hPos.vtx.Y() << "," + << hPos.vtx.Z() << "," + << hPos.vtx.T() << "), " + << "NLL = " << hPos.nll << std::endl; + } + } + + std::sort(tVtxContainer.begin(), tVtxContainer.end(), LEAF::SortingNLL()); + while ((int)tVtxContainer.size() > tolerance) tVtxContainer.pop_back(); + + // Check if mutex is already lock, if not -> lock + // We don't want simultaneous modification of the common output + mtx.lock(); + for (unsigned int iPos = 0; iPos < tVtxContainer.size(); iPos++) { + fVtxThreadOutput.push_back(tVtxContainer[iPos]); + } + delete minimizer; + mtx.unlock(); +} + + +/*****************************************************************************************************/ +/* ENERGY FIT */ +/*****************************************************************************************************/ + +void LEAF::FitEnergy() +{ + double totalPE = 0; + double totalPECorr = 0; + + TStopwatch timer; + timer.Reset(); + timer.Start(); + + // Taha's Polynomial Correction + // double a = -0.000440808; + // double b = 1.71313; + // double c = -2.01675; + // double d = 1.81755; + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + const int iPMTType = static_cast(pmtType); + for(unsigned int i = 0; i < fHitsCollection[iPMTType]->size(); i++) { + const HKHit* lHit = fHitsCollection[iPMTType]->at(i); + + // int iPMT = lHit.PMT; + // PMTInfo lPMTInfo = (*fPMTList)[iPMT]; + // double PMTOrientation[3]; + // double dirVector[3]; + // for(int j = 0; j<3; j++) PMTOrientation[j] = lPMTInfo.Orientation[j]; + // for(int j = 0; j<3; j++) dirVector[j] = -(fTrueVtxPos[j] - lPMTInfo.Position[j]); + // GeoTools::Normalize(PMTOrientation); + // GeoTools::Normalize(dirVector); + // double cosPMThitAngle = abs(dirVector[0] * PMTOrientation[0] + dirVector[1] * PMTOrientation[1] + dirVector[2] * PMTOrientation[2]); + // double Ftheta = a + b*cosPMThitAngle + c*(pow(cosPMThitAngle,2)) + d*(pow(cosPMThitAngle,3)); // Taha's Polynomial Correction + + totalPECorr += lHit->GetCharge(); + totalPE += lHit->GetCharge(); + } + } + + double energyEstimate = totalPECorr * 0.111247 - 12.970173; //* Affine correction (affine because of dark rate and systematic errors), convert from charge to energy + fOutput.total_charge = totalPE; + fOutput.energy = energyEstimate; + + timer.Stop(); + fPerformances.energy_fit_ct = timer.RealTime(); +} + + +/*****************************************************************************************************/ +/* DIRECTION AND VERTEX FIT */ +/*****************************************************************************************************/ + +std::vector LEAF::SearchVertexAndDir() { + TStopwatch timer; + timer.Reset(); + timer.Start(); + + std::vector tVtxCandidates = this->SearchVertex_Main(LEAFConfig::fSearchVtxTolerance,false,LEAFConfig::fSTimePDFLimitsQueueNegative,LEAFConfig::fSTimePDFLimitsQueuePositive,false); + + std::vector Candidates; + for(int i=0; i<(int)tVtxCandidates.size(); i++) { + std::vector tDirCandidates = this->SearchDirection(tVtxCandidates[i].vtx, LEAFConfig::fDirTolerance); + for(int j=0; j<(int) tDirCandidates.size(); j++) { + JointFitCandidate candidate; + candidate.vtx = tVtxCandidates[i].vtx; + candidate.theta = tDirCandidates[j].theta; + candidate.phi = tDirCandidates[j].phi; + candidate.nll = tDirCandidates[j].nll * tVtxCandidates[i].nll; + Candidates.push_back(candidate); + } + } + + std::sort(Candidates.begin(), Candidates.end(), LEAF::SortingNLL{}); + + while ((int)Candidates.size() > LEAFConfig::fSearchVtxTolerance) Candidates.pop_back(); + + timer.Stop(); + fPerformances.vtx_search_ct = timer.RealTime(); + + return Candidates; +} + +LEAF::JointFitCandidate LEAF::MinimizeVertexAndDir(std::vector initialCandidates) { + TStopwatch timer; + timer.Reset(); + timer.Start(); + + double stepSize = 10; + ROOT::Math::XYZTVector limits(1.,1.,1.,1.); + limits *= 2.*LEAFConfig::fSearchVtxStep; + + std::vector tContainer; + + TFitter *minimizer = new TFitter(12); + TMinuit *minuit = minimizer->GetMinuit(); + + double arglist[20]; + int err = 0; + double p1 = VERBOSE - 1; + minimizer->ExecuteCommand("SET PRINTOUT", &p1, 1); // quiet mode + minuit->SetErrorDef(1); + if (VERBOSE < 2) minuit->mnexcm("SET NOWarnings", 0, 0, err); + arglist[0] = 2; + minuit->mnexcm("SET STR", arglist, 1, err); // set strategy sets the number of derivative estimated. in the present case (2), this is the highest number of it with a counter-balance: very slow! + minimizer->SetFCN(LEAFLikelihoods::MinuitJointNLL); // here is function to minimize + + double Mig[2] = {1e6, 1e0}; // maxcalls and tolerance + + minimizer->SetParameter(0, "vertex0", 0, stepSize, -limits.X(), limits.X()); + minimizer->SetParameter(1, "vertex1", 0, stepSize, -limits.Y(), limits.Y()); + minimizer->SetParameter(2, "vertex2", 0, stepSize, -limits.Z(), limits.Z()); + minimizer->SetParameter(3, "vertex3", 0, stepSize / LEAFConfig::fLightSpeed, -limits.T() / LEAFConfig::fLightSpeed, limits.T() / LEAFConfig::fLightSpeed); + minimizer->SetParameter(4, "theta", 0, 2*M_PI/180, 0.0, TMath::Pi()); + minimizer->SetParameter(5, "phi", 0, 2*M_PI/180, -TMath::Pi(), TMath::Pi()); + + minimizer->SetParameter(6, "PMTConfiguration", 0, 0, 0, 0); // Useless now + minimizer->SetParameter(7, "nhits", 0, 0, -1, 1); + minimizer->SetParameter(8, "lowerLimit", LEAFConfig::fSTimePDFLimitsQueueNegative, 5e-1, -7, -2); + minimizer->SetParameter(9, "upperLimit", LEAFConfig::fSTimePDFLimitsQueuePositive, 5e-1, 2, 10); + minimizer->SetParameter(10, "expoSigma", 100, 5, 0, 1e3); + minimizer->SetParameter(11, "thread", 0, 0, 0, 0); + minimizer->FixParameter(6); + minimizer->FixParameter(7); + minimizer->FixParameter(8); + minimizer->FixParameter(9); + minimizer->FixParameter(10); + minimizer->FixParameter(11); + + for (int icand = 0; icand < (int)initialCandidates.size(); icand++) + { + minimizer->SetParameter(0, "vertex0", initialCandidates[icand].vtx.X(), stepSize, initialCandidates[icand].vtx.X() - limits.X(), initialCandidates[icand].vtx.X() + limits.X()); + minimizer->SetParameter(1, "vertex1", initialCandidates[icand].vtx.Y(), stepSize, initialCandidates[icand].vtx.Y() - limits.Y(), initialCandidates[icand].vtx.Y() + limits.Y()); + minimizer->SetParameter(2, "vertex2", initialCandidates[icand].vtx.Z(), stepSize, initialCandidates[icand].vtx.Z() - limits.Z(), initialCandidates[icand].vtx.Z() + limits.Z()); + minimizer->SetParameter(3, "vertex3", initialCandidates[icand].vtx.T(), stepSize / LEAFConfig::fLightSpeed, initialCandidates[icand].vtx.T() - limits.T() / LEAFConfig::fLightSpeed, initialCandidates[icand].vtx.T() + limits.T() / LEAFConfig::fLightSpeed); + minimizer->SetParameter(4, "theta", initialCandidates[icand].theta, 2*M_PI/180, 0.0, TMath::Pi()); + minimizer->SetParameter(5, "phi", initialCandidates[icand].phi, 2*M_PI/180, -TMath::Pi(), TMath::Pi()); + + minimizer->ExecuteCommand("MIGRAD", Mig, 2); + + JointFitCandidate tCandReco; + + tCandReco.vtx.SetXYZT(minimizer->GetParameter(0), minimizer->GetParameter(1), minimizer->GetParameter(2), minimizer->GetParameter(3)); + tCandReco.theta = minimizer->GetParameter(4); + tCandReco.phi = minimizer->GetParameter(5); + + // Get minuit + double fmin, fedm, errdef; + int npar, nparx, istat; + minuit->mnstat(fmin, fedm, errdef, npar, nparx, istat); + tCandReco.nll = fmin; + + tContainer.push_back(tCandReco); + } + + std::sort(tContainer.begin(), tContainer.end(), LEAF::SortingNLL{}); + + delete minimizer; + + timer.Stop(); + fPerformances.vtx_minimize_ct = timer.RealTime(); + + return tContainer[0]; +} + diff --git a/leaf/LEAFSrcInputs.cpp b/leaf/LEAFSrcInputs.cpp new file mode 100644 index 0000000..5a514d1 --- /dev/null +++ b/leaf/LEAFSrcInputs.cpp @@ -0,0 +1,217 @@ +/*****************************************************************************************************/ +/** LEAFSrcInputs.cc **/ +/** Date: March 27th 2026 **/ +/** Desc: Implementation of the I/O functions **/ +/*****************************************************************************************************/ + +#ifndef LEAF_HPP +#include "LEAF.hpp" +#endif + +#ifdef HK_USE_ROOT7 +void LEAF::Initialize(std::shared_ptr lGeometry, std::shared_ptr lDarkNoise, std::shared_ptr lGeoPMT_ID, std::shared_ptr lGeoPMT_mPMT) +#else +void LEAF::Initialize(const HKGeometry *lGeometry, const HKDarkNoise *lDarkNoise, const HKGeometryPMTCollection *lGeoPMT_ID, const HKGeometryPMTCollection *lGeoPMT_mPMT) +#endif +{ + fThread = N_THREAD; + fRand = new TRandom3(); + + fGeometry = lGeometry; + fDarkNoise = lDarkNoise; + fGeoPMTs[int(PMTType::kID)] = lGeoPMT_ID; + fGeoPMTs[int(PMTType::kmPMT)] = lGeoPMT_mPMT; + + LEAFConfig::Initialize(fGeometry->GetInnerTyvekRadius(),fGeometry->GetInnerTyvekTotalHeight()); + this->InitInputs(); + this->LoadSplines(); +} + +void LEAF::InitInputs() { + this->MakePositionList(); +} + +void LEAF::LoadSplines() { + std::cout << "Loading splines..." << std::endl; + TFile *fSplines; + if (LEAFConfig::fHighEnergy) { + fSplines = new TFile("${LEAFDIR}/inputs/PDF_electron_upto100MeV_uniform_isotropic_10k_withDN.root", "read"); + } + else { + fSplines = new TFile("${LEAFDIR}/inputs/PDF_electron_upto100MeV_uniform_isotropic_10k_withDN.root","read"); + } + + // Prevent TGraph2D to be append to TFile (this is needed as we are doing multiple copy of TGraph2D) + TH1::AddDirectory(false); + + //Check if the files are opened + if (!fSplines || !fSplines->IsOpen()) { + std::cerr << "Error: Splines file is not opened." << std::endl; + exit(1); + } + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + int iPMTType = static_cast(pmtType); + + std::cout << "Process spline" << std::endl; + + //Load TOF residuals splines + fSplineTimePDFConv[iPMTType] = (TSpline3*) fSplines->Get(Form("splineExpoConv_%d", iPMTType)); + fSplineTimePDFQueue[iPMTType] = (TSpline3*) fSplines->Get(Form("splineExpoQueue_%d", iPMTType)); + fSplineTimePDFDarkRate[iPMTType] = (TSpline3*) fSplines->Get(Form("splineDR_%d", iPMTType)); + fDirectionPDF[iPMTType] = (TSpline3*) fSplines->Get(Form("splineChargeProfile_%d", iPMTType)); + + // std::cout << "ok ?" << std::endl; + LEAFConfig::fSTimePDFLimitsQueueNegative_fullTimeWindow = fSplineTimePDFQueue[iPMTType]->GetXmin(); + LEAFConfig::fSTimePDFLimitsQueuePositive_fullTimeWindow = fSplineTimePDFQueue[iPMTType]->GetXmax(); + //std::cout<<"Min="< lHitCol_ID, std::shared_ptr lHitCol_mPMT) +#else +void LEAF::LoadHitsCollection(const HKHitsCollection *lHitCol_ID, const HKHitsCollection * lHitCol_mPMT) +#endif +{ + fHitsCollection[int(PMTType::kID)] = lHitCol_ID; + fHitsCollection[int(PMTType::kmPMT)] = lHitCol_mPMT; + + if (fPositionList.size() == 0) MakePositionList(); + if (fHitsCollection[int(PMTType::kmPMT)] == nullptr) { + LEAFConfig::fUseDirectionality = false; + } +} + +void LEAF::SetTrueVertexInfo(ROOT::Math::XYZTVector vtx) { + fTrueVtxPos = vtx; +} + +void LEAF::SetTrueDirInfo(ROOT::Math::XYZVector trueDir) { + fTrueDir = trueDir; +} + +void LEAF::SetNThread(int iThread) { + fThread = iThread; +} + +void LEAF::GenerateEventInfo(double lowerLimit, double upperLimit) { + + // Check if process was already completed for these limits + if (fLastLowerLimit == lowerLimit && fLastUpperLimit == upperLimit) return; + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + int iPMTType = static_cast(pmtType); + + fLastLowerLimit = lowerLimit; + fLastUpperLimit = upperLimit; + + // Initialize + fEventInfo[iPMTType].nHits = fHitsCollection[iPMTType]->size(); + fEventInfo[iPMTType].SignalOverNoise = 0.; + fEventInfo[iPMTType].NoiseIntegral = 0.; + fEventInfo[iPMTType].SignalIntegral = 0.; + + double signalDR, signalPE; + signalDR = LEAFConfig::fTimeWindowSizeFull * fDarkRate_ns[iPMTType]; // Over the whole time window + signalPE = std::max(fEventInfo[iPMTType].nHits - signalDR, 0.); // Over the whole time window. + + double signalPETime = signalPE; // I assume that all signal is here. + double signalDRTime = (upperLimit - lowerLimit) * signalDR / LEAFConfig::fTimeWindowSizeFull; + + fEventInfo[iPMTType].SignalOverNoise = signalPETime / signalDRTime; // Over the + fEventInfo[iPMTType].NoiseIntegral = SplineUtilities::SplineIntegral(fSplineTimePDFDarkRate[iPMTType], lowerLimit, upperLimit); + fEventInfo[iPMTType].SignalIntegral = SplineUtilities::SplineIntegralAndSubstract(fSplineTimePDFQueue[iPMTType], fSplineTimePDFDarkRate[iPMTType], lowerLimit, upperLimit); + + if (VERBOSE >= 3) { + std::cout << "nhits=" << fEventInfo[iPMTType].nHits + << ", DR average=" << signalDR + << ", signal over noise=" << fEventInfo[iPMTType].SignalOverNoise + << ", signal integral=" << fEventInfo[iPMTType].SignalIntegral + << ", DR integral=" << fEventInfo[iPMTType].NoiseIntegral + << ", in integral=" << fEventInfo[iPMTType].SignalIntegral / fEventInfo[iPMTType].NoiseIntegral + << std::endl; + + } + } +} + +double LEAF::CalculateSNR(ROOT::Math::XYZTVector vertex, double lowerLimit, double upperLimit) { + + // We'll do a simple approach: count how many hits are in [lower,upper] + // for each PMT type, then subtract the dark rate. + + int nHitsTotal = 0; + std::array(PMTType::kNumPMTTypes)> inTimes; + + ROOT::Math::XYZVector vertexPos(vertex.X(),vertex.Y(),vertex.Z()); + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + const int iPMTType = static_cast(pmtType); + nHitsTotal += fHitsCollection[iPMTType]->size(); + + int nHitsInTime = 0; + for(unsigned int i = 0; i < fHitsCollection[iPMTType]->size(); i++) { + std::vector toHit; + + const HKHit* lHit = fHitsCollection[iPMTType]->at(i); + int iPMT = lHit->GetPMTNumber(); + const HKGeometryPMT* lPMTInfo = fGeoPMTs[iPMTType]->at(iPMT); + + double hitTime = lHit->GetTime(); + double residual = LEAFUtilities::GetResidual(lPMTInfo->GetPositionInCm(), vertexPos, hitTime, vertex.T()); + + //std::cout << "residual: " << residual << ", " << "lowerlimit and upperlimit " << lowerLimit << ", " << upperLimit << std::endl; + if(residual >= lowerLimit && residual <= upperLimit) { + nHitsInTime += 1; + } + } + inTimes[iPMTType] = nHitsInTime; + } + + double dt = (upperLimit - lowerLimit); + + // BnL: + double darkRateBnL = fDarkNoise->GetAverageTotalDarkNoisePerNS(PMTType::kID); + // average dark over dt: + double darkInWindowBnL = dt * darkRateBnL; + double signalBnL = std::max(double(inTimes[int(PMTType::kID)]) - darkInWindowBnL, 0.0); + //std::cout << "inTimeBnL - darkInWindowBnL : " << inTimeBnL << "-" << darkInWindowBnL << std::endl; + + double snr = (darkInWindowBnL > 1e-9) ? (signalBnL / darkInWindowBnL) : 999.0; + return snr; +} \ No newline at end of file diff --git a/leaf/LEAFSrcLikelihoods.cpp b/leaf/LEAFSrcLikelihoods.cpp new file mode 100644 index 0000000..dba0d2c --- /dev/null +++ b/leaf/LEAFSrcLikelihoods.cpp @@ -0,0 +1,639 @@ +/*****************************************************************************************************/ +/** LEAFSrcInputs.cc **/ +/** Date: March 27th 2026 **/ +/** Desc: Implementation of the likelihood functions **/ +/*****************************************************************************************************/ + +#ifndef LEAF_HPP +#include "LEAF.hpp" +#endif + +//Function to optimize, for MIGRAD +void LEAFLikelihoods::MinuitLikelihood(int & /*nDim*/, double * /*gout*/, double &NLL, double par[], int /*flg*/) +{ + ROOT::Math::XYZTVector vertex(par[0], par[1], par[2], par[3]); // In centimeters + ROOT::Math::XYZVector direction(par[10], par[11], par[12]); + + // double dirThreshold = par[13]; + double lowerLimit = par[6]; + double upperLimit = par[7]; + double directionality = par[9]; + // std::vector vertexDirection(3,0.); + double timeNLL = LEAF::GetME()->Vertex_Time_NLL(vertex, lowerLimit, upperLimit, true, false, directionality); + // double angleNLL = LEAF::GetME()->AngleNLL(vertex, fTrueDir); + NLL = timeNLL; +} + +//Function to optimize, for MIGRAD +void LEAFLikelihoods::MinuitDirNLL(int& /*nDim*/, double* /*gout*/, double& DNLL, double par[], int /*flg*/) +{ + // Extract theta and phi from parameters + double theta = par[0]; + double phi = par[1]; + + // Ensure the vertex is properly initialized + ROOT::Math::XYZTVector vertex(par[3], par[4], par[5], par[6]); + + // Ensure theta and phi are within valid ranges + if (theta < 0 || theta > TMath::Pi()) { + DNLL = 1e10; // Assign a large NLL to invalid directions + return; + } + if (phi < -TMath::Pi() || phi > TMath::Pi()) { + DNLL = 1e10; + return; + } + + // Calculate NLL using theta and phi + DNLL = LEAF::GetME()->Dir_NLL(vertex, theta, phi); +} + +void LEAFLikelihoods::MinuitJointNLL(int& nDim, double * gout, double & NLL, double par[], int flg) +{ + ROOT::Math::XYZTVector vertex(par[0], par[1], par[2], par[3]); + double theta = par[4]; + double phi = par[5]; + double lowerLimit = par[8]; + double upperLimit = par[9]; + // double directionality = par[9]; + + // Ensure theta and phi are within valid ranges + if (theta < 0 || theta > TMath::Pi()) { + NLL = 1e10; // Assign a large NLL to invalid directions + return; + } + if (phi < -TMath::Pi() || phi > TMath::Pi()) { + NLL = 1e10; + return; + } + + double VtxNLL = LEAF::GetME()->Vertex_Time_NLL(vertex, lowerLimit, upperLimit, true, false, 0); + double DirNLL = LEAF::GetME()->Dir_NLL(vertex, theta, phi); + + NLL = VtxNLL * DirNLL; +} + +double LEAF::Vertex_Time_NLL(ROOT::Math::XYZTVector vertex, double lowerLimit, double upperLimit, bool killEdges, bool scaleDR, int directionality) +{ + double NLL = 0; + + ROOT::Math::XYZVector vertexPos(vertex.X(), vertex.Y(), vertex.Z()); + + this->GenerateEventInfo(lowerLimit, upperLimit); + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + const int iPMTType = static_cast(pmtType); + + // Loop over ID PMT hits + for (unsigned int ihit =0; ihit < fHitsCollection[iPMTType]->size(); ihit++) { + const HKHit* lHit = fHitsCollection[iPMTType]->at(ihit); + + int iPMT = lHit->GetPMTNumber(); + // double hitTime = (fTimeCorrection + lHit.T) / TimeDelta::ns; + const HKGeometryPMT* lPMTInfo = fGeoPMTs[iPMTType]->at(iPMT); + + // double distance = Astro_GetDistance(lPMTInfo.Position, vertexPosition); + + // double tof = distance / fLightSpeed; + // double residual = hitTime - tof - vertexPosition[3]; + double residual = LEAFUtilities::GetResidual(lPMTInfo->GetPositionInCm(), vertexPos, lHit->GetTime(), vertex.T()); + + // std::vector toHit = VectorToHitNorm(vertexPosition, lHit); + // if(dot(fTrueDir, toHit) < cos(90 * TMath::DegToRad())) + // if(dot(fTrueDir, toHit) < 0) + // { + // continue; //! Skip this hit if it's not in the same hemisphere + // } + + double proba = 0; + + bool condition = residual > lowerLimit && residual < upperLimit; + + if (condition) + { + proba = fSplineTimePDFQueue[iPMTType]->Eval(residual); + +#ifdef VERBOSE_NLL + if (VERBOSE >= 3) + { + std::cout << "hit#" << ihit << ", hit time =" << hitTime << ", vertex time = " << vertexPosition[3] << std::endl; + std::cout << "Residual=" << residual << ", proba=" << proba << ", pmt type=" << pmtType << std::endl; + } +#endif + + if (scaleDR) + { + std::cout << "scaleDR with Lower Limit: " << lowerLimit << " Upper Limit: " << upperLimit << std::endl; + + // First, substract the DR from the PDF to keep only the signal: + double DR = fSplineTimePDFDarkRate[iPMTType]->Eval(residual); + proba -= DR; + // Then, scale signal so that signal integral / DR integral = signalOverNoise + // To do so, we should scale signal so that integral = DR integral * signalOverNoise + // And of course, to rescale, we should divide by the signal integral + double Factor = fEventInfo[iPMTType].NoiseIntegral * fEventInfo[iPMTType].SignalOverNoise / fEventInfo[iPMTType].SignalIntegral; + proba *= Factor; + // And add again the DR + proba += DR; +#ifdef VERBOSE_NLL + if (VERBOSE >= 3) std::cout << "proba after scaling=" << proba << std::endl; +#endif + } + } + else if (killEdges) + { + if (residual >= upperLimit) + { + proba = fSplineTimePDFQueue[iPMTType]->Eval(upperLimit); +#ifdef VERBOSE_NLL + if (VERBOSE >= 3 && pmtType == PMTType::kmPMT) + std::cout << "PMT type = " << pmtType << ", Upper limit = " << upperLimit << ", residual = " << residual << ", proba = " << proba << std::endl; +#endif + } + else if (residual <= lowerLimit) + { + // continue; + proba = fSplineTimePDFQueue[iPMTType]->Eval(lowerLimit); +#ifdef VERBOSE_NLL + if (VERBOSE >= 3 && pmtType == PMTType::kmPMT) + std::cout << "PMT type = " << pmtType << ", Lower limit = " << lowerLimit << ", residual = " << residual << ", proba = " << proba << std::endl; +#endif + } + } + else proba = 0; + + if (proba < 0) + { + std::cout << "Error in PDF" << std::endl; + if (proba > -1e-1) proba = 0; // Since spline sometimes slightly goes below 0 due to interpolation + else + { + std::cout << "Error in " << residual << "ns where proba = " << proba << std::endl; + return 0; + } + } + if (proba == 0) proba = 1e-20; + NLL += -TMath::Log(proba); + } + + //* Directionality, not used RN + + // if (directionality != 0) + // { + // double NLLdir = 0; + + // NLLdir = this->FindNLLDirectionality(vertexPosition, VERBOSE, lowerLimit, upperLimit); + + // #ifdef VERBOSE_NLL + // if (VERBOSE >= 2) + // std::cout << "NLL = " << NLL << ", dir (L) = " << TMath::Exp(-NLLdir) << ", dir 2 = " << NLLdir << std::endl; + // #endif + // if (directionality == 1) + // NLL += NLLdir; + // else if (directionality == 2) + // NLL = NLLdir; + // } + + } + //timer.Stop(); + //std::cout << "NLL = " << NLL << std::endl; + return NLL; +} + +double LEAF::Vertex_Score(ROOT::Math::XYZTVector vertex, double /*lowerLimit*/, double /*upperLimit*/, bool /*killEdges*/, bool /*scaleDR*/, int /*directionality*/) +{ + double NLL = 0; + ROOT::Math::XYZVector vertexPos(vertex.X(), vertex.Y(), vertex.Z()); + + // std::cout << " First Hit " << fHitCollection->At(0).PMT << " " << vertexPosition.size() << std::endl; + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + const int iPMTType = static_cast(pmtType); + + for (unsigned int ihit =0; ihit < fHitsCollection[iPMTType]->size(); ihit++) { + // Hit lHit = fHitInfo[ihit]; + const HKHit* lHit = fHitsCollection[iPMTType]->at(ihit); + + int iPMT = lHit->GetPMTNumber(); + // double hitTime = (fTimeCorrection + lHit.T) / TimeDelta::ns; + const HKGeometryPMT* lPMTInfo = fGeoPMTs[iPMTType]->at(iPMT); + + // double distance = Astro_GetDistance(lPMTInfo.Position, vertexPosition); + + // double tof = distance / fLightSpeed; + // double residual = hitTime - tof - vertexPosition[3]; + double residual = LEAFUtilities::GetResidual(lPMTInfo->GetPositionInCm(), vertexPos, lHit->GetTime(), vertex.T()); + + /* + if (ihit < 5) { + double distance = LEAFUtilities::GetDistance(vertexPos,lPMTInfo->GetPositionInCm()); + std::cout << "PMT: " << iPMT << " \t X: " << lPMTInfo->GetPositionInCm().X() << " Y: " << lPMTInfo->GetPositionInCm().Y() << " Z: " << lPMTInfo->GetPositionInCm().Z() << " T " << lHit->GetTime() << std::endl; + std::cout << "Vertex: \t X:" << vertexPos.X() << " Y: " << vertexPos.Y() << " Z: " << vertexPos.Z() << " T " << vertex.T() << std::endl; + ROOT::Math::XYZVector distVector = (vertexPos - lPMTInfo->GetPositionInCm()); + std::cout << "Diff: \t X:" << distVector.X() << " Y: " << distVector.Y() << " Z: " << distVector.Z() << std::endl; + std::cout << " Distance " << distance << " -> " << residual << std::endl; + } + */ +#ifdef KILLHALF + if (pmtType == PMTType::kmPMT) { + double t = fRand->Uniform(0, 1); + if (t > 0.5) + continue; + } +#endif + + bool bCondition = (residual > LEAFConfig::fHitTimeLimitsNegative && residual < LEAFConfig::fHitTimeLimitsPositive) || (pmtType == PMTType::kmPMT && !LEAFConfig::fLimit_mPMT); + + if (bCondition) { + NLL++; +#ifdef VERBOSE_NLL + if (VERBOSE >= 3) { + std::cout << "Residual=" << residual << ", pmt type=" << pmtType << std::endl; + std::cout << "hit#" << ihit << ", hit time =" << hitTime << ", vertex time = " << vertexPosition[3] << std::endl; + std::cout << "residual=" << residual << std::endl; + } +#endif + } + } + } + + return NLL != 0 ? -TMath::Log(NLL) : 1e15; +} + +double LEAF::FindNLL(ROOT::Math::XYZTVector vertex, bool likelihood, int verbose, double lowerLimit, double upperLimit, bool killEdges, bool scaleDR, int directionality) +{ + double NLL = 0; + ROOT::Math::XYZVector vertexPos(vertex.X(), vertex.Y(), vertex.Z()); + + this->GenerateEventInfo(lowerLimit, upperLimit); + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + const int iPMTType = static_cast(pmtType); + + for (unsigned int ihit =0; ihit < fHitsCollection[iPMTType]->size(); ihit++) { + const HKHit* lHit = fHitsCollection[iPMTType]->at(ihit); + int iPMT = lHit->GetPMTNumber(); + const HKGeometryPMT* lPMTInfo = fGeoPMTs[iPMTType]->at(iPMT); + + // double hitTime = (fTimeCorrection + lHit.T) / TimeDelta::ns; + // double distance = Astro_GetDistance(lPMTInfo.Position, vertexPosition); + + // double tof = distance / fLightSpeed; + // double residual = hitTime - tof - vertexPosition[3]; + double residual = LEAFUtilities::GetResidual(lPMTInfo->GetPositionInCm(), vertexPos, lHit->GetTime(), vertex.T()); + double proba; + #ifdef KILLHALF + if (pmtType == PMTType::kmPMT) + { + double t = fRand->Uniform(0, 1); + if (t > 0.5) + continue; + } + #endif + + if (likelihood) { + bool condition; + + condition = residual > lowerLimit && residual < upperLimit; + if (condition) { + proba = fSplineTimePDFQueue[iPMTType]->Eval(residual); +#ifdef VERBOSE_NLL + if (VERBOSE >= 3) { + std::cout << "hit#" << ihit << ", hit time =" << hitTime << ", vertex time = " << vertexPosition[3] << std::endl; + std::cout << "Residual=" << residual << ", proba=" << proba << ", pmt type=" << pmtType << std::endl; + } +#endif + if (scaleDR) { + std::cout << "scaleDR with Lower Limit: " << lowerLimit << " Upper Limit: " << upperLimit << std::endl; + + // First, substract the DR from the PDF to keep only the signal: + double DR = fSplineTimePDFDarkRate[iPMTType]->Eval(residual); + proba -= DR; + // Then, scale signal so that signal integral / DR integral = signalOverNoise + // To do so, we should scale signal so that integral = DR integral * signalOverNoise + // And of course, to rescale, we should divide by the signal integral + double Factor = fEventInfo[iPMTType].NoiseIntegral * fEventInfo[iPMTType].SignalOverNoise / fEventInfo[iPMTType].SignalIntegral; + proba *= Factor; + // And add again the DR + proba += DR; + if (VERBOSE >= 3) + std::cout << "proba after scaling=" << proba << std::endl; + } + } + else if (killEdges && residual >= upperLimit) { + proba = fSplineTimePDFQueue[iPMTType]->Eval(upperLimit); +#ifdef VERBOSE_NLL + if (VERBOSE >= 3 && pmtType == PMTType::kmPMT) + std::cout << "PMT type = " << pmtType << ", Upper limit = " << upperLimit << ", residual = " << residual << ", proba = " << proba << std::endl; +#endif + } + else if (killEdges && residual <= lowerLimit) { + proba = fSplineTimePDFQueue[iPMTType]->Eval(lowerLimit); + #ifdef VERBOSE_NLL + if (VERBOSE >= 3 && pmtType == PMTType::kmPMT) + std::cout << "PMT type = " << pmtType << ", Lower limit = " << lowerLimit << ", residual = " << residual << ", proba = " << proba << std::endl; + #endif + } + else { + proba = 0; + } + } + else { + bool condition; + if (pmtType == PMTType::kID || (pmtType == PMTType::kmPMT && LEAFConfig::fLimit_mPMT)) { + condition = (residual > LEAFConfig::fHitTimeLimitsNegative && residual < LEAFConfig::fHitTimeLimitsPositive); + } + else { + condition = true; + } + if (condition) { + NLL++; //= fSplineTimePDFQueue[pmtType]->Eval(residual); + #ifdef VERBOSE_NLL + if (VERBOSE >= 3) { + std::cout << "Residual=" << residual << ", pmt type=" << pmtType << std::endl; + std::cout << "hit#" << ihit << ", hit time =" << hitTime << ", vertex time = " << vertexPosition[3] << ", distance PMT vs vertex = " << distance << ", tof=" << tof << std::endl; + std::cout << "residual=" << residual << ", proba=" << proba << std::endl; + } + #endif + } + else if (killEdges) { + continue; + } + else { + proba = 0; + } + } + + if (likelihood) { + if (proba < 0) { + std::cout << "Error in PDF" << std::endl; + if (proba > -1e-1) proba = 0; // Since spline sometimes slightly goes below 0 due to interpolation + else { + std::cout << "Error in " << residual << "ns where proba = " << proba << std::endl; + return 0; + } + } + if (proba == 0) proba = 1e-20; + NLL += -TMath::Log(proba); + } + } + // std::cout << "FindNLL: Time it took for the loop = " << timer.RealTime() << std::endl; + // cout<<"NLL="<=2) cout<<"NLL="<FindNLLDirectionality(vertex, verbose, lowerLimit, upperLimit); + + // double test = FindNLLDirectionality(lHitCol, fHitCollection, vertexPosition, verbose, lowerLimit, upperLimit); + // std::cout << "test = " << test << std::endl; + } + else { + NLLdir2 = this->FindNLLDirectionality(vertex, verbose, LEAFConfig::fHitTimeLimitsNegative, LEAFConfig::fHitTimeLimitsPositive); + } + if (VERBOSE >= 2) { + std::cout << "NLL = " << NLL << ", dir (L) = " << TMath::Exp(-NLLdir) << ", dir 2 = " << NLLdir2 << std::endl; + } + + if (directionality == 1) { + NLL += NLLdir2; + } + else if (directionality == 2) { + NLL = NLLdir2; + } + } + } + + return NLL; +} + +double LEAF::ComputeDirNLL_NoPDF(const ROOT::Math::XYZTVector& vertex, const ROOT::Math::XYZVector& direction) { + double DNLL = 0.; + + ROOT::Math::XYZVector vertexPos(vertex.X(),vertex.Y(),vertex.Z()); + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + const int iPMTType = static_cast(pmtType); + for (unsigned int ihit =0; ihit < fHitsCollection[iPMTType]->size(); ihit++) { + const HKHit* lHit = fHitsCollection[iPMTType]->at(ihit); + int iPMT = lHit->GetPMTNumber(); + const HKGeometryPMT* lPMTInfo = fGeoPMTs[iPMTType]->at(iPMT); + + // Calculate vector from PMT to vertex + ROOT::Math::XYZVector dirVectorNormalized = (vertexPos - lPMTInfo->GetPositionInCm()).Unit(); + ROOT::Math::XYZVector directionNormalized = direction.Unit(); + + // Calculate RelativeAngle (cosine of angle between vectors) + double relativeAngle = dirVectorNormalized.Dot(directionNormalized); + double theta = acos(relativeAngle)*180./TMath::Pi(); + //std::cout << "theta: " << theta << std::endl; + if (theta<46 && theta>38) { + std::cout << "Got hit! " << theta << std::endl; + DNLL += -1; + } + + if (VERBOSE >= 2) std::cout << "Direction NLL = " << DNLL << std::endl; + } + } + + return DNLL; + +} + +//Uses the PDF to find the DirNLL, used mainly for the MIGRAD optimization +double LEAF::Dir_NLL(const ROOT::Math::XYZTVector& vertex, double theta_track, double phi_track) +{ + double DNLL = 0; + ROOT::Math::XYZVector vertexPos(vertex.X(), vertex.Y(), vertex.Z()); + + // Ensure theta and phi are within valid ranges + // theta in [0, pi], phi in [-pi, pi] + if (theta_track < 0 || theta_track > TMath::Pi()) + { + std::cerr << "Error: theta_track is out of range [0, π]." << std::endl; + return std::numeric_limits::infinity(); + } + if (phi_track < -TMath::Pi() || phi_track > TMath::Pi()) + { + std::cerr << "Error: phi_track is out of range [-π, π]." << std::endl; + return std::numeric_limits::infinity(); + } + + // Precompute sine and cosine of theta_track and phi_track + // double sin_theta_track = sin(theta_track); + // double cos_theta_track = cos(theta_track); + // double sin_phi_track = sin(phi_track); + // double cos_phi_track = cos(phi_track); + /* 1 + double a = 0.556805; + double b = 1.06697; + double c = -0.902084; + double d = -0.0707118;*/ + + // 2 + // double a = -0.000440808; + // double b = 1.71313; + // double c = -2.01675; + // double d = 1.81755; + + // Direction vector components from theta and phi + // double vertexDirNormalized[3]; + // vertexDirNormalized[0] = sin_theta_track * cos_phi_track; + // vertexDirNormalized[1] = sin_theta_track * sin_phi_track; + // vertexDirNormalized[2] = cos_theta_track; + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + const int iPMTType = static_cast(pmtType); + for (unsigned int ihit =0; ihit < fHitsCollection[iPMTType]->size(); ihit++) { + + const HKHit* lHit = fHitsCollection[iPMTType]->at(ihit); + int iPMT = lHit->GetPMTNumber(); + const HKGeometryPMT* lPMTInfo = fGeoPMTs[iPMTType]->at(iPMT); + + // double hitTime = (fTimeCorrection + lHit.T) / TimeDelta::ns; + + // double distance = Astro_GetDistance(lPMTInfo.Position,vertexPosition); + + // double tof = distance / fLightSpeed; + // double residual = hitTime - tof - vertexPosition[3]; + + double residual = LEAFUtilities::GetResidual(lPMTInfo->GetPositionInCm(), vertexPos, lHit->GetTime(), vertex.T()); + + bool DirCondition = (residual >= LEAFConfig::fDirectionPDF_minResidual && residual <= LEAFConfig::fDirectionPDF_maxResidual) || (pmtType == PMTType::kmPMT && !LEAFConfig::fLimit_mPMT); + + + if(DirCondition || LEAFConfig::fDirTakeAll) { + + ROOT::Math::XYZVector dirVector = (lPMTInfo->GetPositionInCm() - vertexPos).Unit(); + + // Convert dirVector to spherical coordinates (theta_pmt, phi_pmt)fDirectionPDF + double theta_pmt = acos(dirVector.Z()); // theta in [0, π] + double phi_pmt = atan2(dirVector.Y(), dirVector.X()); // phi in [-π, π] + + // Calculate the angular difference between the two directions + // double delta_theta = theta_pmt - theta_track; //* could be useful + double delta_phi = phi_pmt - phi_track; + + // Ensure delta_phi is within [-π, π] + if (delta_phi > TMath::Pi()) delta_phi -= 2 * TMath::Pi(); + else if (delta_phi < -TMath::Pi()) delta_phi += 2 * TMath::Pi(); + + // Calculate the angle between the two directions using the spherical law of cosines + double cos_relative_angle = sin(theta_track) * sin(theta_pmt) * cos(delta_phi) + + cos(theta_track) * cos(theta_pmt); + + //cos_relative_angle = std::max(-1.0, std::min(1.0, cos_relative_angle)); + + double relative_angle = acos(cos_relative_angle); // Angle in radians + + // Convert angle to degrees if your PDF is defined in degrees + double theta = relative_angle * 180.0 / TMath::Pi(); + + // Evaluate your PDF at theta + double proba = fDirectionPDF[iPMTType]->Eval(theta); //* Nicolas + + // Correct for PMT orientation if necessary + ROOT::Math::XYZVector pmtOrientation = lPMTInfo->GetOrientation().Unit(); + + double cosPMThitAngle = dirVector.Dot(pmtOrientation); + // double PMThitAngle = acos(-cosPMThitAngle); + cosPMThitAngle *= -1; + // double cos2PMThitAngle = fabs(-2 * cosPMThitAngle * cosPMThitAngle + 1); + // double Ftheta = a + b*cosPMThitAngle + c*(pow(cosPMThitAngle,2)) + d*(pow(cosPMThitAngle,3)); + + //proba *= hitCharge; + + // Avoid log(0) by ensuring proba is positive + if (proba <= 0) { + proba = 1e-20; + if (VERBOSE >= 1) std::cout << "Warning: Probability is zero or negative at hit #" << ihit << std::endl; + } + //std::cout << "Proba Corrected by " << PMThitAngle * 180 / M_PI << " degrees with a sincos " << sin(PMThitAngle) << ", " << cosPMThitAngle << std::endl; + // Accumulate Negative Log-Likelihood + DNLL += -TMath::Log(proba);//*(cosPMThitAngle/Ftheta)*(sin(PMThitAngle)); + + if (VERBOSE >= 3) std::cout << "Hit #" << ihit << ", Theta = " << theta << " degrees, Probability = " << proba << std::endl; + } + + if (VERBOSE >= 2) std::cout << "Total Direction NLL = " << DNLL << std::endl; + } + } + return DNLL; +} + +double LEAF::AngleNLL(ROOT::Math::XYZTVector vertex, ROOT::Math::XYZVector direction) +{ + double NLL = 0; + ROOT::Math::XYZVector vertexPos(vertex.X(), vertex.Y(), vertex.Z()); + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + const int iPMTType = static_cast(pmtType); + for (unsigned int ihit =0; ihit < fHitsCollection[iPMTType]->size(); ihit++) { + //* compute angle + const HKHit* lHit = fHitsCollection[iPMTType]->at(ihit); + int iPMT = lHit->GetPMTNumber(); + const HKGeometryPMT* lPMTInfo = fGeoPMTs[iPMTType]->at(iPMT); + + ROOT::Math::XYZVector toPMT = (lPMTInfo->GetPositionInCm() - vertexPos).Unit(); + double dotP = toPMT.Dot(direction.Unit()); + + dotP = std::max(-1.0, std::min(1.0, dotP)); // Ensure dot product is within valid range for acos + double angle = std::acos(dotP) * 180 / TMath::Pi(); + NLL += -TMath::Log(std::max(1e-20, fDirectionPDF[iPMTType]->Eval(angle))); //? Nicolas + } + } + + return NLL; +} + + +double LEAF::FindNLLDirectionality(ROOT::Math::XYZTVector vVtxPos, int verbose, double /*lowerLimit*/, double /*upperLimit*/) { + + double NLL = 0; + return NLL; +} + + +double LEAF::GoodnessOfFit(ROOT::Math::XYZTVector vertex, double NLL_theory, double lowerLimit, double upperLimit, bool killEdges, bool scaleDR, int directionality) { + // double NLL_theory = Vertex_Time_NLL(lHitCol, vertexPosition, lowerLimit, upperLimit, killEdges, scaleDR, directionality); + + ROOT::Math::XYZVector vertexPos(vertex.X(), vertex.Y(), vertex.Z()); + std::vector residuals; + + for (auto pmtType : LEAFConfig::fActivePMTTypes) { + const int iPMTType = static_cast(pmtType); + for (unsigned int ihit =0; ihit < fHitsCollection[iPMTType]->size(); ihit++) { + const HKHit* lHit = fHitsCollection[iPMTType]->at(ihit); + int iPMT = lHit->GetPMTNumber(); + const HKGeometryPMT* lPMTInfo = fGeoPMTs[iPMTType]->at(iPMT); + + double residual = LEAFUtilities::GetResidual(lPMTInfo->GetPositionInCm(), vertexPos, lHit->GetTime(), vertex.T()); + residuals.push_back(residual); + } + } + + double bandwidth = LEAFUtilities::EstimateBandeWidth(residuals); + + //* Compute KDE and total log likelihood of data + double log_p_data = 0.0; + for (const auto& t_i : residuals) { + double p = LEAFUtilities::KDE_Estimate(residuals, t_i, bandwidth); + log_p_data += std::log(p + 1e-12); // avoid log(0) + } + + //* Likelihood ratio + double NLL_data = -log_p_data; + double NLLR = NLL_data - NLL_theory; + + return NLLR; +} \ No newline at end of file diff --git a/leaf/LEAFUtilities.cpp b/leaf/LEAFUtilities.cpp new file mode 100644 index 0000000..5923b62 --- /dev/null +++ b/leaf/LEAFUtilities.cpp @@ -0,0 +1,54 @@ +#include "LEAFUtilities.hpp" + +double LEAFUtilities::GetDistance(ROOT::Math::XYZVector a, ROOT::Math::XYZVector b) { + ROOT::Math::XYZVector distVector = (a - b); + return sqrt(distVector.Dot(distVector)); +} + +double LEAFUtilities::GetLength(ROOT::Math::XYZVector v) { + return sqrt(v.Dot(v)); +} + +double LEAFUtilities::GetTimeOfFlight(ROOT::Math::XYZVector a, ROOT::Math::XYZVector b) { + return LEAFUtilities::GetDistance(a,b) / C_WATER_CM_NS; +} + +double LEAFUtilities::GetTimeOfFlight(ROOT::Math::XYZVector v) { + return LEAFUtilities::GetLength(v) / C_WATER_CM_NS; +} + +double LEAFUtilities::GetTimeOfFlight(double distance) { + return distance / C_WATER_CM_NS; +} + +double LEAFUtilities::GetResidual(ROOT::Math::XYZVector pmtPos, ROOT::Math::XYZVector vertexPos, double hitTime, double vertexTime) { + return hitTime - LEAFUtilities::GetTimeOfFlight(vertexPos, pmtPos) - vertexTime; +} + +double LEAFUtilities::GetResidual(ROOT::Math::XYZVector distToPMT, double hitTime, double vertexTime) { + return hitTime - LEAFUtilities::GetTimeOfFlight(distToPMT) - vertexTime; +} + +double LEAFUtilities::GetResidual(double tof, double hitTime, double vertexTime) { + return hitTime - tof - vertexTime; +} + +double LEAFUtilities::EstimateBandeWidth(const std::vector& residuals) { + double sum = std::accumulate(residuals.begin(), residuals.end(), 0.0); + double mean = sum / residuals.size(); + + double sq_sum = std::inner_product(residuals.begin(), residuals.end(), residuals.begin(), 0.0); + double variance = sq_sum / residuals.size() - mean * mean; + double stddev = std::sqrt(variance); + return 1.06 * stddev * std::pow(residuals.size(), -1.0 / 5.0); +} + +double LEAFUtilities::GaussianKernel(double x, double bandwidth) { + return std::exp(-0.5 * x * x / (bandwidth * bandwidth)) / (bandwidth * std::sqrt(2.0 * M_PI)); +} + +double LEAFUtilities::KDE_Estimate(const std::vector& residuals, double t_i, double bandwidth) { + double sum = 0.0; + for (const double& t_j : residuals) sum += GaussianKernel(t_i - t_j, bandwidth); + return sum / residuals.size(); +} \ No newline at end of file diff --git a/leaf/LEAFUtilities.hpp b/leaf/LEAFUtilities.hpp new file mode 100644 index 0000000..4e6fa4b --- /dev/null +++ b/leaf/LEAFUtilities.hpp @@ -0,0 +1,36 @@ +#ifndef LEAFUTILITIES_HPP +#define LEAFUTILITIES_HPP + +#include +#include +#include + +#include +#include +#include + +#ifndef C_WATER_CM_NS +#define C_WATER_CM_NS 21.849964 // light celerity in water, in centimeter per ns. +#endif + +namespace LEAFUtilities { + + double GetDistance(ROOT::Math::XYZVector a, ROOT::Math::XYZVector b); + double GetLength(ROOT::Math::XYZVector v); + + double GetTimeOfFlight(ROOT::Math::XYZVector a, ROOT::Math::XYZVector b); + double GetTimeOfFlight(ROOT::Math::XYZVector v); + double GetTimeOfFlight(double distance); + + double GetResidual(ROOT::Math::XYZVector pmtPos, ROOT::Math::XYZVector vertexPos, double hitTime, double vertexTime); + double GetResidual(ROOT::Math::XYZVector distToPMT, double hitTime, double vertexTime); + double GetResidual(double tof, double hitTime, double vertexTime); + + double EstimateBandeWidth(const std::vector& residuals); + + double GaussianKernel(double x, double bandwidth); + double KDE_Estimate(const std::vector& residuals, double t_i, double bandwidth); + +}; // namespace LEAFUtilities + +#endif \ No newline at end of file diff --git a/leaf/LeafConfig.cc b/leaf/LeafConfig.cc deleted file mode 100644 index d749b07..0000000 --- a/leaf/LeafConfig.cc +++ /dev/null @@ -1,173 +0,0 @@ -#include "LeafConfig.hh" - -//* Geometry & Constants -double fTankRadius; -double fTankHeight; -double fTankHalfHeight; -double fLightSpeed; - -//* PDFs -double fSTimePDFLimitsQueueNegative = -3.; -double fSTimePDFLimitsQueuePositive = 4.; -double fSTimePDFLimitsQueueNegative_fullTimeWindow = 0.; -double fSTimePDFLimitsQueuePositive_fullTimeWindow = 0.; -double fPDFNorm_fullTimeWindow = 0.; - -double fDirectionPDF_maxResidual = 15.0; -double fDirectionPDF_minResidual = -5.0; - -//* Overall Fit Parameters -double fTimeWindowSizeFull = 1500.; -double fIntegrationTimeWindow = 50.; -bool fStepByStep = false; -bool fUseDirectionality = false; -bool DoubleNLL = false; -bool fLimit_mPMT = true; -int fAveraging = 20; -bool fHighEnergy = false; -double fSearchVtxStep = 300.; -double fSearchVtxTolerance = 60; -bool bDirectionnality = false; - -//* Step 1 Specific Parameters -double fHitTimeLimitsNegative = -5; -double fHitTimeLimitsPositive = -7; - -//* Step 2 Specific Parameters -double fMinimizeLimitsNegative = -700; -double fMinimizeLimitsPositive = 1000; - -//* Direction Fit Parameters -bool DirTakeAll = false; -double theta_step = 5 * TMath::DegToRad(); -double phi_step = 5 * TMath::DegToRad(); -int DirTolerance = 50; - -void InitConfig(const Geometry *lGeometry) -{ - fTankRadius = lGeometry->detector_radius; - fTankHeight = lGeometry->detector_length; - fTankHalfHeight = fTankHeight / 2.; - - fStepByStep = false; // Step by step mode was not test and is not supported by multithreading - fUseDirectionality = false; - fHighEnergy = false; - DoubleNLL = true; - - // If true, we apply same cuts for timing as B&L PMT when searching vertex. - // Otherwise, we do not apply them and use all the hits. - // The latter is particularly useful when directionality is added, as it can kill DR hits. - fLimit_mPMT = true; - - fSTimePDFLimitsQueueNegative = -3.; //-5;//-3;//-1.5;//-3;//-10; - fSTimePDFLimitsQueuePositive = 4.; // 10;//4;//3;//4;//500; - fSTimePDFLimitsQueueNegative_fullTimeWindow = 0.; //-1.5;//-3;//-10; - fSTimePDFLimitsQueuePositive_fullTimeWindow = 0.; // 3;//4;//500; - fTimeWindowSizeFull = 1500; - fAveraging = 20; // Number of points we used to average the vertex on. - - fMinimizeLimitsNegative = -700; - fMinimizeLimitsPositive = 1000; - - fHitTimeLimitsNegative = -5; //-10;//-5;//-5 //-5 - fHitTimeLimitsPositive = 7; // 15;//7;//7 //7 - - fSearchVtxStep = 300; // in cm, the step size for coarse grid search - - fSearchVtxTolerance = 60; // Number of candidate vertex that are kept after coarse grid search - - fIntegrationTimeWindow = 50; // in ns //? never Used ??? - DirTakeAll = false; - theta_step = 5 * TMath::DegToRad(); - phi_step = 5 * TMath::DegToRad(); - DirTolerance = 50; - - //* Compute light speed: - float fCVacuum = 3e8 * 1e2 / 1e9; // speed of light, in centimeter per ns. - float fNIndex = 1.373; // 1.385;//1.373;//refraction index of water - fLightSpeed = fCVacuum / fNIndex; - - fPDFNorm_fullTimeWindow = 0.; - - //*** These maps describe what will be loaded from the config file given */ - - std::map envVariables = - { - {"SearchVtxStep", &fSearchVtxStep}, - {"SearchVtxTolerance", &fSearchVtxTolerance}, - {"HitTimeLimitsNegative", &fHitTimeLimitsNegative}, - {"HitTimeLimitsPositive", &fHitTimeLimitsPositive}, - {"MinimizeLimitsNegative", &fMinimizeLimitsNegative}, - {"MinimizeLimitsPositive", &fMinimizeLimitsPositive}, - {"IntegrationTimeWindow", &fIntegrationTimeWindow}, - {"TimeWindowSizeFull", &fTimeWindowSizeFull}, - {"STimePDFLimitsQueueNegative", &fSTimePDFLimitsQueueNegative}, - {"STimePDFLimitsQueuePositive", &fSTimePDFLimitsQueuePositive}, - {"STimePDFLimitsQueueNegative_fullTimeWindow", &fSTimePDFLimitsQueueNegative_fullTimeWindow}, - {"STimePDFLimitsQueuePositive_fullTimeWindow", &fSTimePDFLimitsQueuePositive_fullTimeWindow}, - {"DirectionPDF_maxResidual", &fDirectionPDF_maxResidual}, - {"DirectionPDF_minResidual", &fDirectionPDF_minResidual}, - {"ThetaStep", &theta_step}, - {"PhiStep", &phi_step} - }; - - std::map envIntVariables = - { - {"Averaging", &fAveraging}, - {"DirTolerance", &DirTolerance} - }; - - std::map envBoolVariables = - { - {"Limit_mPMT", &fLimit_mPMT}, - {"UseDirectionality", &fUseDirectionality}, - {"StepByStep", &fStepByStep}, - {"HighEnergy", &fHighEnergy}, - {"DoubleNLL", &DoubleNLL}, - {"DirTakeAll", &DirTakeAll}, - }; - - for (const auto& [envName, variable] : envVariables) - { - const char* envValue = std::getenv(envName.c_str()); - if (envValue != nullptr) - { - *variable = std::stod(envValue); - std::cout << envName << " is set to " << *variable << std::endl; - } - else std::cout << "Environment variable " << envName << " is not set. Using default value : "<< *variable << std::endl; - } - - for (const auto& [envName, variable] : envIntVariables) - { - const char* envValue = std::getenv(envName.c_str()); - if (envValue != nullptr) - { - *variable = std::stoi(envValue); - std::cout << envName << " is set to " << *variable << std::endl; - } - else std::cout << "Environment variable " << envName << " is not set. Using default value : "<< *variable << std::endl; - } - - for (const auto& [envName, variable] : envBoolVariables) - { - const char* envValue = std::getenv(envName.c_str()); - if (envValue != nullptr) - { - std::string valueStr(envValue); - if (valueStr == "true" || valueStr == "1") - { - *variable = true; - std::cout << envName << " is set to true" << std::endl; - } - else if (valueStr == "false" || valueStr == "0") - { - *variable = false; - std::cout << envName << " is set to false" << std::endl; - } - else std::cout << "Environment variable " << envName << " has an invalid value. Using default value : "<< *variable << std::endl; - } - else std::cout << "Environment variable " << envName << " is not set. Using default value : "<< *variable << std::endl; - } - -} \ No newline at end of file diff --git a/leaf/LeafConfig.hh b/leaf/LeafConfig.hh deleted file mode 100644 index 983f226..0000000 --- a/leaf/LeafConfig.hh +++ /dev/null @@ -1,71 +0,0 @@ -#pragma once - -#define NPMT_CONFIGURATION 1 - -// Hit definition: -#define NormalPMT 0 // B&L hit -#define MiniPMT 1 // 3" PMT hit -#define AllPMT 2 - -#define VERBOSE 0 - -#undef VERBOSE_VTX // In SearchVertex -#undef VERBOSE_NLL // In Likelihood Computation - -#define VTX_X 0 -#define VTX_Y 1 -#define VTX_Z 2 -#define VTX_T 3 - -#define N_THREAD 12 // Default for sukap - -#include "WCSimRootGeom.hh" -#include "Geometry.h" - -//**** all these parameters will be loaded automatically from the config if they are defined here AND in the tables of the .cc */ - -//* Geometry & Constants -extern double fTankRadius; -extern double fTankHeight; -extern double fTankHalfHeight; -extern double fLightSpeed; - -//* PDFs -extern double fPDFNorm_fullTimeWindow; -extern double fSTimePDFLimitsQueueNegative; -extern double fSTimePDFLimitsQueuePositive; -extern double fSTimePDFLimitsQueueNegative_fullTimeWindow; -extern double fSTimePDFLimitsQueuePositive_fullTimeWindow; - -extern double fDirectionPDF_maxResidual; -extern double fDirectionPDF_minResidual; - -//* Overall Fit Parameters -extern double fTimeWindowSizeFull; -extern double fIntegrationTimeWindow; -extern bool fStepByStep; -extern bool fUseDirectionality; -extern bool DoubleNLL; -extern bool fLimit_mPMT; -// extern double KELimitsPos; -// extern double KELimitsNeg; -extern int fAveraging; -extern bool fHighEnergy; -extern double fSearchVtxStep; -extern double fSearchVtxTolerance; - -//* Step 1 Specific Parameters -extern double fHitTimeLimitsNegative; -extern double fHitTimeLimitsPositive; - -//* Step 2 Specific Parameters -extern double fMinimizeLimitsNegative; -extern double fMinimizeLimitsPositive; - -//* Direction Fit Parameters -extern bool DirTakeAll; -extern double theta_step; -extern double phi_step; -extern int DirTolerance; - -void InitConfig(const Geometry *lGeometry); \ No newline at end of file diff --git a/leaf/LeafDefinitions.hh b/leaf/LeafDefinitions.hh deleted file mode 100644 index bd2454c..0000000 --- a/leaf/LeafDefinitions.hh +++ /dev/null @@ -1,88 +0,0 @@ -#pragma once - -#include -#include -#include -#include -#include -#include -#include - -//* WCSim Headers -#include "WCSimRootGeom.hh" - -//* ROOT Headers -#include "TFile.h" -#include "TFitter.h" -#include "TH1D.h" -#include "TH2D.h" -#include "TGraph2D.h" -#include "TF1.h" -#include "TMath.h" -#include "TMinuit.h" -#include "TObject.h" -#include "TRandom3.h" -#include "TSpline.h" -#include "TPaletteAxis.h" - -//* DataModel informations -#include "Geometry.h" -#include "HitCollection.h" - -struct FitterOutput -{ - std::vector Vtx; // leaf vertex - double NLL; // Vertex Likelihood - double NLLR; // Goodness of fit - - std::vector Dir; // leaf Direction - double DNLL; // Direction Likelihood - - double Energy; // leaf energy - double TotalCharge; // raw total charge before affine correction - - std::vector MyDir; - std::vector Quick_Dir; -}; - -struct FitterOutputProps -{ - // Computations times - double Leaf_ComputeTime;//Total leaf computational time - double Vtx_Search_ComputeTime;//Vertex coarse grid search computational time - double Vtx_Minimize_ComputeTime;//Vertex MINUIT minimisation computational time - double Dir_Quick_Search_ComputeTime;//Direction using unit vectors from vector to PMTT - computational time - double Dir_Search_ComputeTime;//Direction coarse grid search computational time - double Dir_Minimize_ComputeTime;//Direction MINUIT minimisation computational time - double Energy_Fit_ComputeTime;//Energy finder computational time -}; - -struct DirectionCandidate -{ - double theta; - double phi; - double DNLL; -}; - -struct VtxCandidate -{ - double X; - double Y; - double Z; - double T; - double NLL; - double SNR; // SNR value -}; - -struct JointFitCandidate -{ - VtxCandidate VtxPart; - DirectionCandidate DirPart; - double NLL; -}; - -struct FitPosition -{ - std::vector Vtx; - double NLL; -}; diff --git a/leaf/LeafInputs.cc b/leaf/LeafInputs.cc deleted file mode 100644 index 74b9beb..0000000 --- a/leaf/LeafInputs.cc +++ /dev/null @@ -1,82 +0,0 @@ -#include "LeafInputs.hh" - -const Geometry* fGeometry; - -const HitCollection* fHitCollection; -TimeDelta fTriggerTime; -TimeDelta fTimeCorrection; - -const std::vector *fPMTList; -double fDarkRate_ns[NPMT_CONFIGURATION]; - -int fThread = N_THREAD; - -TRandom3 * fRand; - -std::vector fTrueVtxPos; -std::vector fTrueDir; - -std::vector< std::vector > fPositionList; - -void InitInputs(const Geometry *lGeometry) -{ - fGeometry = lGeometry; - fPMTList = fGeometry->GetPMTList(); - fThread = N_THREAD; - fRand = new TRandom3(); - - fDarkRate_ns[NormalPMT] = fGeometry->pmt_dark_rate[HKAA::kIDPMT_BnL]; - fDarkRate_ns[MiniPMT] = fGeometry->pmt_dark_rate[HKAA::kIDPMT_3inch]; - MakePositionList(); -} - -void MakePositionList() -{ - // Make position list for a given step size - double dStep = fSearchVtxStep; - fPositionList = std::vector< std::vector >(); - fPositionList.clear(); - - for (double time = -50; time < 50; time += (dStep / fLightSpeed)) - { - for (double radius = 0; radius <= fTankRadius; radius += dStep) - { - double perimeter = 2 * TMath::Pi() * radius; - int numberOfStepsOnCircle = floor(perimeter / dStep); - if (numberOfStepsOnCircle == 0) numberOfStepsOnCircle = 1; - double angleStepOnCircle = 2 * TMath::Pi() / numberOfStepsOnCircle; - - for (double angle = 0; angle <= 2 * TMath::Pi(); angle += angleStepOnCircle) - { - for (double height = -fTankHalfHeight; height <= fTankHalfHeight; height += dStep) - { - std::vector tVtxPosition(4, 0.); // In centimeters - tVtxPosition[0] = radius * TMath::Cos(angle); - tVtxPosition[1] = radius * TMath::Sin(angle); - tVtxPosition[2] = height; - tVtxPosition[3] = time; - fPositionList.push_back(tVtxPosition); - } - } - } - } -} - -void SetTrueVertexInfo(std::vector vtx, double time) -{ - fTrueVtxPos = std::vector(4, 0.); - fTrueVtxPos[0] = vtx[0]; - fTrueVtxPos[1] = vtx[1]; - fTrueVtxPos[2] = vtx[2]; - fTrueVtxPos[3] = time; -} - -void SetTrueDirInfo(std::vector trueDir) -{ - fTrueDir = std::vector(3, 0.); - fTrueDir[0] = trueDir[0]; - fTrueDir[1] = trueDir[1]; - fTrueDir[2] = trueDir[2]; -} - -void SetNThread(int iThread){ fThread=iThread; } \ No newline at end of file diff --git a/leaf/LeafInputs.hh b/leaf/LeafInputs.hh deleted file mode 100644 index a9643fd..0000000 --- a/leaf/LeafInputs.hh +++ /dev/null @@ -1,47 +0,0 @@ -#pragma once - -#include -#include -#include -#include -#include -#include -#include - -//* Root Headers -#include "TRandom3.h" - -//* WCSim Headers -#include "WCSimRootGeom.hh" -#include "Geometry.h" -#include "HitCollection.h" -#include "LeafConfig.hh" - -extern const Geometry* fGeometry; -extern const HitCollection* fHitCollection; -extern TimeDelta fTriggerTime; -extern TimeDelta fTimeCorrection; - -extern const std::vector *fPMTList; -extern double fDarkRate_ns[NPMT_CONFIGURATION]; - -extern std::vector< std::vector > fPositionList; - -extern int fThread; - -extern TRandom3 * fRand; - -extern std::vector fTrueVtxPos; -extern std::vector fTrueDir; - -void InitInputs(const Geometry *lGeometry); - -void MakePositionList(); - -void SetNThread(int iThread=N_THREAD); - -//* True Vertex isn't used for the fit, just to get feedback on how well the fit is at different steps -void SetTrueVertexInfo(std::vector vtx, double time); -void SetTrueDirInfo(std::vector trueDir); - - diff --git a/leaf/LeafSplines.cc b/leaf/LeafSplines.cc deleted file mode 100644 index 1abb100..0000000 --- a/leaf/LeafSplines.cc +++ /dev/null @@ -1,208 +0,0 @@ -#include "LeafSplines.hh" - -TSpline3 * fSplineTimePDFQueue[NPMT_CONFIGURATION]; -TSpline3 * fSplineTimePDFDarkRate[NPMT_CONFIGURATION]; -TSpline3* fDirectionPDF; -TSpline3* fHitAnglePDF; -TGraph2D * gPMTDirectionality_2D[NPMT_CONFIGURATION][HKAA::kmPMT_Groups]; -TF1 * fDistResponsePMT[NPMT_CONFIGURATION]; - -double fDarkRate_dir_proba[NPMT_CONFIGURATION][HKAA::kmPMT_Groups]; -double fLastLowerLimit; -double fLastUpperLimit; - -double SplineIntegral(TSpline3 *s, double start, double end, double stepSize) -{ - double integral = 0; - for (double i = start; i < end; i += stepSize) - { - integral += stepSize * s->Eval(i + stepSize / 2); - } - return integral; -} -double SplineIntegralAndSubstract(TSpline3 *s0, TSpline3 *s1, double start, double end, double stepSize) -{ - double integral = 0; - for (double i = start; i < end; i += stepSize) - { - integral += stepSize * (s0->Eval(i + stepSize / 2) - s1->Eval(i + stepSize / 2)); - } - return integral; -} - -double SplineIntegralExpo(TSpline3 *s, double start, double end, double sigma, double stepSize) -{ - double integral = 0; - for (double i = start; i < end; i += stepSize) - { - integral += stepSize * s->Eval(i + stepSize / 2) * TMath::Gaus(i + stepSize / 2, 0, sigma); - } - return integral; -} - -void LoadSplines() -{ - std::cout << "Loading splines..." << std::endl; - TFile *fSplines, *fSplines2, *fSplines1, *fSplineAngleFile; - if (fHighEnergy) - { - fSplines = new TFile("${LEAFDIR}/inputs/timePDF_HE.root", "read"); // To generate with my code ProduceWSPlots.c - fSplines2 = fSplines; - } - else - { - std::cout << "spline 1" << std::endl; - fSplines1 = new TFile("${LEAFDIR}/inputs/timePDFNoDR_50000_e10MeV_Hit_720.root","read");//To generate with my code ProduceWSPlots.c - // fSplines = new TFile("${LEAFDIR}/inputs/timePDF_DRnew_Large.root", "read"); // To generate with my code ProduceWSPlots.c - std::cout << "spline 2" << std::endl; - fSplines = new TFile("${LEAFDIR}/inputs/timePDFDR_5000_e10MeV_Hit_fiducial.root","read"); //*TAHA - // fSplines = new TFile("${LEAFDIR}/inputs/timePDF_3M_10T_500000.root","read"); //*NICOLAS - // fSplines = new TFile("${LEAFDIR}/inputs/timePDF_3M_10T.root","read"); - std::cout << "spline 3" << std::endl; - fSplines2 = new TFile("${LEAFDIR}/inputs/timePDF_Directionality_DRnew.root", "read"); // To generate with my code ProduceWSPlots.c - // fSplines2 = new TFile("${LEAFDIR}/inputs/timePDF_Directionality_DRnew.root","read");//To generate with my code ProduceWSPlots.c - std::cout<< "angle spline" << std::endl; - fSplineAngleFile = new TFile("${LEAFDIR}/inputs/Hit_Angle_PDF.root", "read"); - } - - std::cout << "spline file read" << std::endl; - - // Prevent TGraph2D to be append to TFile (this is needed as we are doing multiple copy of TGraph2D) - // For a strange reason sometimes the TGraph2D is see as a TH1 and stored in an open TFile - TH1::AddDirectory(false); - - //Check if the files are opened - if (!fSplines || !fSplines->IsOpen()) - { - std::cerr << "Error: Splines file is not opened." << std::endl; - exit(1); - } - if (!fSplines2 || !fSplines2->IsOpen()) - { - std::cerr << "Error: Splines file is not opened." << std::endl; - exit(1); - } - if (!fSplines1 || !fSplines1->IsOpen()) - { - std::cerr << "Error: Splines file is not opened." << std::endl; - exit(1); - } - if (!fSplineAngleFile || !fSplineAngleFile->IsOpen()) - { - std::cerr << "Error: Angle spline file is not opened." << std::endl; - exit(1); - } - - // std::cout << "splines files loaded" << std::endl; - - int configs = NPMT_CONFIGURATION; - - for(int pmtType=0;pmtTypeGet(Form("splineExpoQueue%d_%d",0,pmtType)); - fSplineTimePDFDarkRate[pmtType] = (TSpline3*) fSplines->Get(Form("splineDR%d_%d",0,pmtType)); - fDirectionPDF = (TSpline3*) fSplines1->Get(Form("ChargeProfileCos_PDF_spline_pmtType0")); - // fDirectionPDF = (TSpline3*) fSplineAngleFile->Get("Hit_Angle_Spline"); - if (!fDirectionPDF) - { - std::cerr << "Error: fDirectionPDF is not loaded properly." << std::endl; - exit(1); - } - fHitAnglePDF = (TSpline3*) fSplineAngleFile->Get("Hit_Angle_Spline"); - // fHitAnglePDF = (TSpline3*) fSplines1->Get(Form("ChargeProfileCos_PDF_spline_pmtType0")); - - std::cout << "process spline" << std::endl; - // std::cout << "ok ?" << std::endl; - fSTimePDFLimitsQueueNegative_fullTimeWindow = fSplineTimePDFQueue[pmtType]->GetXmin(); - fSTimePDFLimitsQueuePositive_fullTimeWindow = fSplineTimePDFQueue[pmtType]->GetXmax(); - //std::cout<<"Min="<IsOpen()) { - std::cerr << "Error: fSplines2 is not opened or is null." << std::endl; - exit(1); - } - - int grpNB = 0; - if (HKAA::kmPMT_Groups > 0) { - grpNB = HKAA::kmPMT_Groups; - } else { - std::cerr << "Error: HKAA::kmPMT_Groups is not properly defined or initialized." << std::endl; - exit(1); - } - - std::cout << "got grnNB" << std::endl; - - for (int pmtGroup = 0; pmtGroup < grpNB; pmtGroup++) { - gPMTDirectionality_2D[pmtType][pmtGroup] = (TGraph2D*) fSplines2->Get(Form("gPMTDirectionality_2D_%d_%d_%d", 0, pmtType, pmtGroup)); - if (!gPMTDirectionality_2D[pmtType][pmtGroup]) { - std::cerr << "Error: Failed to load gPMTDirectionality_2D for pmtType " << pmtType << " and pmtGroup " << pmtGroup << std::endl; - exit(1); - } - } - - std::cout << "dist response" << std::endl; - - fDistResponsePMT[pmtType] = (TF1*) fSplines2->Get(Form("fDistResponsePMT_pmtType%d", pmtType)); - if (!fDistResponsePMT[pmtType]) - { - std::cerr << "Error: Failed to load fDistResponsePMT for pmtType " << pmtType << std::endl; - exit(1); - } - } - std::cout << "Splines loaded." << std::endl; -} - -EventInfo MakeEventInfo(double lowerLimit, double upperLimit, int pmtType) -{ - struct EventInfo fEventInfo; - - if (fLastLowerLimit == lowerLimit && fLastUpperLimit == upperLimit) return fEventInfo; - fLastLowerLimit = lowerLimit; - fLastUpperLimit = upperLimit; - - fEventInfo.hits = 0; - fEventInfo.SignaloverNoise = 0.; - fEventInfo.NoiseIntegral = 0.; - fEventInfo.SignalIntegral = 0.; - - int iHitTotal = fHitCollection->Size(); - - for (int iHit = 0; iHit < iHitTotal; iHit++) - { - // Hit lHit = fHitInfo[iHit]; - Hit lHit = fHitCollection->At(iHit); - - int iPMT = lHit.PMT; - int iType = Astro_GetPMTType(iPMT); - if(pmtType == iType) fEventInfo.hits += 1; - } - - double signalDR, signalPE; - signalDR = fTimeWindowSizeFull * fDarkRate_ns[pmtType]; // Over the whole time window - signalPE = std::max(fEventInfo.hits - signalDR, 0.); // Over the whole time window. - - double signalPETime = signalPE; // I assume that all signal is here. - double signalDRTime = (upperLimit - lowerLimit) * signalDR / fTimeWindowSizeFull; - - fEventInfo.SignaloverNoise = signalPETime / signalDRTime; // Over the - fEventInfo.NoiseIntegral = SplineIntegral(fSplineTimePDFDarkRate[pmtType], lowerLimit, upperLimit); - fEventInfo.SignalIntegral = SplineIntegralAndSubstract(fSplineTimePDFQueue[pmtType], fSplineTimePDFDarkRate[pmtType], lowerLimit, upperLimit); - - if (VERBOSE >= 3) std::cout - << "nhits=" << fEventInfo.hits - << ", DR average=" << signalDR - << ", signal over noise=" << fEventInfo.SignaloverNoise - << ", signal integral=" << fEventInfo.SignalIntegral - << ", DR integral=" << fEventInfo.NoiseIntegral - << ",in integral=" << fEventInfo.SignalIntegral / fEventInfo.NoiseIntegral << std::endl; - - return fEventInfo; -} \ No newline at end of file diff --git a/leaf/LeafSplines.hh b/leaf/LeafSplines.hh deleted file mode 100644 index 6d2b783..0000000 --- a/leaf/LeafSplines.hh +++ /dev/null @@ -1,64 +0,0 @@ -#pragma once - -#include -#include -#include -#include -#include -#include -#include - -#include "TFile.h" -#include "TFitter.h" -#include "TH1D.h" -#include "TH2D.h" -#include "TGraph2D.h" -#include "TF1.h" -#include "TMath.h" -#include "TMinuit.h" -#include "TObject.h" -#include "TRandom3.h" -#include "TSpline.h" -#include "TPaletteAxis.h" - -//WCSim Headers -#include "WCSimRootGeom.hh" -#include "Geometry.h" -#include "HitCollection.h" -#include "LeafInputs.hh" - - -struct EventInfo -{ - int hits; - double SignaloverNoise; - double NoiseIntegral; - double SignalIntegral; -}; - -// extern bool fHighEnergySplines; - -extern TSpline3 * fSplineTimePDFQueue[NPMT_CONFIGURATION]; -extern TSpline3 * fSplineTimePDFDarkRate[NPMT_CONFIGURATION]; -extern TSpline3* fDirectionPDF; -extern TSpline3* fHitAnglePDF; - -// Histo -extern TGraph2D * gPMTDirectionality_2D[NPMT_CONFIGURATION][HKAA::kmPMT_Groups]; - -// TF1 -extern TF1 * fDistResponsePMT[NPMT_CONFIGURATION]; - -extern double fDarkRate_dir_proba[NPMT_CONFIGURATION][HKAA::kmPMT_Groups]; - -extern double fLastLowerLimit; -extern double fLastUpperLimit; -// double fTimeWindowSizeFull; - -void LoadSplines(); - -double SplineIntegral (TSpline3 * s, double start, double end, double stepSize=5e-1); -double SplineIntegralAndSubstract (TSpline3 * s0, TSpline3 * s1, double start, double end, double stepSize=5e-1); -double SplineIntegralExpo (TSpline3 * s, double start, double end, double sigma, double stepSize=5e-1); - -EventInfo MakeEventInfo(double lowerLimit, double upperLimit, int pmtType); \ No newline at end of file diff --git a/leaf/LeafUtility.cc b/leaf/LeafUtility.cc deleted file mode 100644 index cc2be55..0000000 --- a/leaf/LeafUtility.cc +++ /dev/null @@ -1,351 +0,0 @@ -#include "LeafUtility.hh" - -bool ContainsTrueVtx(std::vector* candidates, std::vector fTrueVtxPos) -{ - for (unsigned int i = 0; i < candidates->size(); i++) - { - std::vector vtx = {(*candidates)[i].X, (*candidates)[i].Y, (*candidates)[i].Z, (*candidates)[i].T}; - double distance = Distance3D(vtx, fTrueVtxPos); - if (distance <= fSearchVtxStep) //? should we take step / 2 ? - { - return true; - } - } - return false; -} - -bool ContainsTrueDir(std::vector* candidates, std::vector fTrueDir) -{ - for (unsigned int i = 0; i < candidates->size(); i++) - { - std::vector truePolarDir = CartesianToPolarNorm(fTrueDir); - if(abs((*candidates)[i].theta - truePolarDir[0]) < theta_step && abs((*candidates)[i].phi - truePolarDir[1]) < phi_step) - { - return true; - } - } - return false; -} - - -//* to know if the predicted vertex and the true vertex belong to the same candidate -//? doesn't work because the radius overlap between candidates -bool CanidatesMatch(std::vector>* candidates, std::vector point, std::vector fTrueVtxPos) -{ - for(unsigned int i = 0; i < candidates->size(); i++) - { - if (Distance3D((*candidates)[i], fTrueVtxPos) <= fSearchVtxStep && Distance3D((*candidates)[i], point) <= fSearchVtxStep) return true; - } - return false; -} - -double Distance3D(std::vector point1, std::vector point2) -{ - return sqrt(pow(point1[0] - point2[0], 2) + - pow(point1[1] - point2[1], 2) + - pow(point1[2] - point2[2], 2)); -} - -void Normalize(double a[3]) -{ - double l; - l=sqrt(a[0]*a[0]+a[1]*a[1]+a[2]*a[2]); - for(int j=0; j<3; j++){ - a[j]/=l; - } -} - -std::vector& Normalize(std::vector& vector) -{ - if (vector.size() < 3) throw std::invalid_argument("Vector must be of size at least 3 to be normalized"); - double length = sqrt(vector[0] * vector[0] + vector[1] * vector[1] + vector[2] * vector[2]); - if (length == 0) throw std::invalid_argument("Cannot normalize a zero-length vector."); - for (int i = 0; i < 3; ++i) vector[i] /= length; - return vector; -} - -double calculateDistance(const std::vector& A, const std::vector& B) -{ - return sqrt(pow(A[0] - B[0], 2) + pow(A[1] - B[1], 2) + pow(A[2] - B[2], 2)); -} - -double dot(const std::vector& A, const std::vector& B) -{ - if (A.size() != B.size()) throw std::invalid_argument("Vectors must be of the same size for dot product."); - double result = 0; - for (size_t i = 0; i < A.size(); ++i) result += A[i] * B[i]; - return result; -} - -std::vector PolarToCartesianNorm(std::vector polarVector) -{ - double theta = polarVector[0]; - double phi = polarVector[1]; - std::vector cartVector = std::vector(3); - cartVector[0] = sin(theta) * cos(phi); - cartVector[1] = sin(theta) * sin(phi); - cartVector[2] = cos(theta); - Normalize(cartVector); - return cartVector; -} - -std::vector CartesianToPolarNorm(const std::vector& cartVector) -{ - std::vector polarVector(2); - double x = cartVector[0]; - double y = cartVector[1]; - double z = cartVector[2]; - - polarVector[0] = acos(z); // theta - polarVector[1] = atan2(y, x); // phi - - return polarVector; -} - -double EstimateBandeWidth(const std::vector& residuals) -{ - double sum = std::accumulate(residuals.begin(), residuals.end(), 0.0); - double mean = sum / residuals.size(); - - double sq_sum = std::inner_product(residuals.begin(), residuals.end(), residuals.begin(), 0.0); - double variance = sq_sum / residuals.size() - mean * mean; - double stddev = std::sqrt(variance); - return 1.06 * stddev * std::pow(residuals.size(), -1.0 / 5.0); -} - -double GaussianKernel(double x, double bandwidth) -{ - return std::exp(-0.5 * x * x / (bandwidth * bandwidth)) / (bandwidth * std::sqrt(2.0 * M_PI)); -} - -double KDE_Estimate(const std::vector& residuals, double t_i, double bandwidth) -{ - double sum = 0.0; - for (const double& t_j : residuals) sum += GaussianKernel(t_i - t_j, bandwidth); - return sum / residuals.size(); -} - -double ComputeResidualTime(std::vector vertexPos, double originTime, Hit lHit) -{ - int iPMT = lHit.PMT; - double HitT = (fTimeCorrection + lHit.T) / TimeDelta::ns; - PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - double distance = Astro_GetDistance(lPMTInfo.Position, vertexPos); - - double tof = distance / fLightSpeed; - return HitT - tof - originTime; -} - -VtxCandidate CreateCandidate(const std::vector& vertex, double lowerLimit, double upperLimit, bool computeSNR) -{ - VtxCandidate out; - - // Copy the vertex info - out.X = vertex[0]; - out.Y = vertex[1]; - out.Z = vertex[2]; - out.T = vertex[3]; - out.NLL = 0.0; - - if(computeSNR) - { - // We'll do a simple approach: count how many hits are in [lower,upper] - // for each PMT type, then subtract the dark rate. - - int nHitsTotal = fHitCollection->Size(); - int inTimeBnL = 0; - int inTimemPMT = 0; - - for(int iHit = 0; iHit < nHitsTotal; iHit++) { - const Hit& hit = fHitCollection->At(iHit); - int iPMT = hit.PMT; - int pmtType = Astro_GetPMTType(iPMT); // 0=BnL, 1=mPMT - - double hitTime = (fTimeCorrection + hit.T) / TimeDelta::ns; - PMTInfo pInfo = (*fPMTList)[iPMT]; - double distance = Astro_GetDistance(pInfo.Position, vertex); - double tof = distance / fLightSpeed; - double residual = hitTime - tof - vertex[3]; - //std::cout << "residual: " << residual << ", " << "lowerlimit and upperlimit " << lowerLimit << ", " << upperLimit << std::endl; - if(residual >= lowerLimit && residual <= upperLimit) - { - if(pmtType == 0) { - // BnL - inTimeBnL++; - } else { - // mPMT - inTimemPMT++; - } - } - } - - double dt = (upperLimit - lowerLimit); - - // BnL: - double darkRateBnL = fTimeWindowSizeFull * fDarkRate_ns[NormalPMT]; - // average dark over dt: - double darkInWindowBnL = (dt / double(fTimeWindowSizeFull)) * darkRateBnL; - double signalBnL = std::max(double(inTimeBnL) - darkInWindowBnL, 0.0); - //std::cout << "inTimeBnL - darkInWindowBnL : " << inTimeBnL << "-" << darkInWindowBnL << std::endl; - out.SNR = (darkInWindowBnL > 1e-9) ? (signalBnL / darkInWindowBnL) : 999.0; - } - return out; -} - -void VectorVertexPMT(std::vector vertex, int iPMT, double *dAngles) -{ - // Guillaume 2020/05/20: - // mPMT referencial is computed once for all PMT in LoadPMTInfo() and MakeMPMTReferencial(iPMT) - // Guillaume 2020/11/20: - // Reference is moved to outside LEAF (HKManager or hk-AstroAnalysis) - - PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - int iPMTTop = lPMTInfo.mPMT_RefTube; - PMTInfo lPMTInfoTop = (*fPMTList)[iPMTTop]; - - // 5. Now we have our referential, we should just calculate the angles of the PMT to vertex position vector in this referential. - // a. calculate the PMT to vertex position vector. - - double dVtx_PMTRef[3]; - - dVtx_PMTRef[0] = vertex[0] - lPMTInfoTop.Position[0]; - dVtx_PMTRef[1] = vertex[1] - lPMTInfoTop.Position[1]; - dVtx_PMTRef[2] = vertex[2] - lPMTInfoTop.Position[2]; - - double dLengthVtx = Astro_GetLength(dVtx_PMTRef); - GeoTools::Normalize(dVtx_PMTRef); - - if (VERBOSE >= 3) - { - std::cout << "Vertex position = " << vertex[0] << ", " << vertex[1] << ", " << vertex[2] << std::endl; - std::cout << "Vertex position from PMT = " << dVtx_PMTRef[0] << ", " << dVtx_PMTRef[1] << ", " << dVtx_PMTRef[2] << std::endl; - } - - // b. Then extract Theta and Phi: - double dCosTheta = Astro_GetScalarProd(dVtx_PMTRef, lPMTInfo.mPMT_RefZ); - double dTheta = TMath::ACos(dCosTheta); - - double dPhi = 0.; - - if ((*fPMTList)[iPMT].mPMT_TubeNum == HKAA::kmPMT_TopID) - { - // Phi is not defined in that case.. - } - else - { - // We know x=cosPhi x sinTheta and y=sinPhi x sinTheta - double dX = Astro_GetScalarProd(dVtx_PMTRef, lPMTInfo.mPMT_RefX); - double dY = Astro_GetScalarProd(dVtx_PMTRef, lPMTInfo.mPMT_RefY); - double dTanPhi = dY / dX; - dPhi = TMath::ATan(dTanPhi); - - // tan is symetric from -pi/2 to +pi/2 - if (dX == 0) - { - if (dY < 0) - dPhi = -TMath::Pi() / 2.; - else - dPhi = TMath::Pi() / 2.; - } - if (dX < 0) - dPhi += TMath::Pi(); // With this, angle become defines between -pi/2 to -pi/2 +2pi. - - // We wish to bring this from 0 to 2pi: - if (dPhi < 0) - dPhi += 2 * TMath::Pi(); - - // Actually, we have a symmetry in Phi between [0,pi] and [pi,2pi]. So, we will just define Phi in [0,pi] modulo pi - if (dPhi > TMath::Pi()) - dPhi = TMath::Pi() - (dPhi - TMath::Pi()); - } - - dAngles[0] = dPhi * 180. / TMath::Pi(); - dAngles[1] = dTheta * 180. / TMath::Pi(); - dAngles[2] = dLengthVtx; - dAngles[3] = 1; // calculateWeight(dLengthVtx,PMTradius[pmtType],Theta,verbose); - - if (VERBOSE >= 3) - { - std::cout << "Angles Phi = " << dAngles[0] << ", Theta = " << dAngles[1] << std::endl; - } -} - -// std::vector ProjectPointToCylinder(std::vector point, double R, double H) -// { -// std::vector dummy; -// return dummy; -// } - -std::vector ProjectPointToCylinder(std::vector point, double R, double H) -{ - double x = point[0]; - double y = point[1]; - double z = point[2]; - double halfHeight = H / 2.0; - - // Distance from the z-axis - double r_xy = std::sqrt(x * x + y * y); - - // --- Project to side surface --- - double x_side, y_side; - if (r_xy == 0.0) - { - x_side = R; - y_side = 0.0; // Arbitrary direction - } - else - { - x_side = x * R / r_xy; - y_side = y * R / r_xy; - } - double z_side = std::clamp(z, -halfHeight, halfHeight); - - // --- Project to top cap (z = +halfHeight) --- - double z_top = halfHeight; - double x_top = x, y_top = y; - double r_top = std::sqrt(x * x + y * y); - if (r_top > R && r_top != 0.0) - { - x_top = x * R / r_top; - y_top = y * R / r_top; - } - - // --- Project to bottom cap (z = -halfHeight) --- - double z_bottom = -halfHeight; - double x_bottom = x, y_bottom = y; - double r_bottom = std::sqrt(x * x + y * y); - if (r_bottom > R && r_bottom != 0.0) - { - x_bottom = x * R / r_bottom; - y_bottom = y * R / r_bottom; - } - - // --- Compute squared distances to each projection --- - auto dist2 = [](double dx, double dy, double dz) - { - return dx * dx + dy * dy + dz * dz; - }; - - double d2_side = dist2(x - x_side, y - y_side, z - z_side); - double d2_top = dist2(x - x_top, y - y_top, z - z_top); - double d2_bottom = dist2(x - x_bottom, y - y_bottom, z - z_bottom); - - // --- Return closest projection --- - if (d2_side <= d2_top && d2_side <= d2_bottom) return {x_side, y_side, z_side}; - else if (d2_top <= d2_bottom) return {x_top, y_top, z_top}; - else return {x_bottom, y_bottom, z_bottom}; -} - -std::vector VectorToHitNorm(const std::vector& from, Hit lHitt) -{ - std::vector toHit(3, 0.0); - std::vector hitPos(3, 0.0); - int iPMT = lHitt.PMT; - PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - for (int j = 0; j < 3; j++) hitPos[j] = lPMTInfo.Position[j]; - for (int j = 0; j < 3; j++) toHit[j] = hitPos[j] - from[j]; - - Normalize(toHit); - - return toHit; -} \ No newline at end of file diff --git a/leaf/LeafUtility.hh b/leaf/LeafUtility.hh deleted file mode 100644 index 444de27..0000000 --- a/leaf/LeafUtility.hh +++ /dev/null @@ -1,59 +0,0 @@ -#pragma once - -#include -#include -#include -#include -#include -#include - -//* DataModel informations -#include "Geometry.h" -#include "HitCollection.h" -#include "LeafConfig.hh" -#include "LeafInputs.hh" -#include "LeafDefinitions.hh" - -inline bool SortOutputVector ( const std::vector& v1, const std::vector& v2 ) -{ - return v1[4] < v2[4]; -} - -double Distance3D(std::vector point1, std::vector point2); - -void Normalize(double a[3]); - -std::vector& Normalize(std::vector& vector); - -double dot(const std::vector& A, const std::vector& B); - -double calculateDistance(const std::vector& A, const std::vector& B); - -std::vector VectorToHitNorm(const std::vector& from, Hit lHitt); - -std::vector PolarToCartesianNorm(std::vector polarVector); - -std::vector CartesianToPolarNorm(const std::vector& cartVector); - -bool CanidatesMatch(std::vector>* candidates, std::vector point, std::vector fTrueVtxPos); - -bool ContainsTrueVtx(std::vector* candidates, std::vector fTrueVtxPos); - -bool ContainsTrueDir(std::vector* candidates, std::vector fTrueDir); - -//* Estimate P_data(t_i) using KDE -double KDE_Estimate(const std::vector& residuals, double t_i, double bandwidth); - -double EstimateBandeWidth(const std::vector& residuals); - -double ComputeResidualTime(std::vector vertexPos, double originTime, Hit lHitt); - -//* Gaussian kernel function -double GaussianKernel(double x, double bandwidth); - -//* Signal noise ratio -VtxCandidate CreateCandidate(const std::vector& vertex, double lowerLimit, double upperLimit, bool computeSNR); - -void VectorVertexPMT(std::vector vertex, int iPMT, double* dAngles ); - -std::vector ProjectPointToCylinder(std::vector point, double R, double H); \ No newline at end of file diff --git a/leaf/Likelihoods.cc b/leaf/Likelihoods.cc deleted file mode 100644 index 5cc79a2..0000000 --- a/leaf/Likelihoods.cc +++ /dev/null @@ -1,743 +0,0 @@ -#include "Likelihoods.hh" -#include "LeafInputs.hh" - -//Function to optimize, for MIGRAD -void MinuitLikelihood(int & /*nDim*/, double * /*gout*/, double &NLL, double par[], int /*flg*/) -{ - std::vector vertexPosition(4, 0.); // In centimeters - for (int i = 0; i < 4; i++) vertexPosition[i] = par[i]; - std::vector eventDirection(3, 0.); - for (int i = 0; i < 3; i++) eventDirection[i] = par[i + 10]; - // double dirThreshold = par[13]; - int nhits = par[5]; - double lowerLimit = par[6]; - double upperLimit = par[7]; - double directionality = par[9]; - // std::vector vertexDirection(3,0.); - double timeNLL = Likelihoods::Vertex_Time_NLL(fHitCollection, vertexPosition, nhits, lowerLimit, upperLimit, true, false, directionality); - // double angleNLL = Likelihoods::AngleNLL(fHitCollection, vertexPosition, fTrueDir); - NLL = timeNLL; -} - -//Function to optimize, for MIGRAD -void MinuitDirNLL(int& /*nDim*/, double* /*gout*/, double& DNLL, double par[], int /*flg*/) -{ - // Extract theta and phi from parameters - double theta = par[0]; - double phi = par[1]; - int nhits = static_cast(par[2]); - - // Ensure the vertexPosition vector is properly initialized - std::vector vertexPosition(4, 0.0); - for (int i = 0; i < 4; i++) vertexPosition[i] = par[i + 3]; - - // Ensure theta and phi are within valid ranges - if (theta < 0 || theta > TMath::Pi()) - { - DNLL = 1e10; // Assign a large NLL to invalid directions - return; - } - if (phi < -TMath::Pi() || phi > TMath::Pi()) - { - DNLL = 1e10; - return; - } - - // Calculate NLL using theta and phi - DNLL = Likelihoods::Dir_NLL(fHitCollection, vertexPosition, theta, phi, nhits); -} - -void MinuitJointNLL(int& nDim, double * gout, double & NLL, double par[], int flg) -{ - std::vector vertexPosition(4, 0.); - for (int i = 0; i < 4; i++) vertexPosition[i] = par[i]; - double theta = par[4]; - double phi = par[5]; - int nhits = par[7]; - double lowerLimit = par[8]; - double upperLimit = par[9]; - // double directionality = par[9]; - - // Ensure theta and phi are within valid ranges - if (theta < 0 || theta > TMath::Pi()) - { - NLL = 1e10; // Assign a large NLL to invalid directions - return; - } - if (phi < -TMath::Pi() || phi > TMath::Pi()) - { - NLL = 1e10; - return; - } - - double VtxNLL = Likelihoods::Vertex_Time_NLL(fHitCollection, vertexPosition, nhits, lowerLimit, upperLimit, true, false, 0); - double DirNLL = Likelihoods::Dir_NLL(fHitCollection, vertexPosition, theta, phi, nhits); - - NLL = VtxNLL * DirNLL; -} - -double Likelihoods::Vertex_Time_NLL(const HitCollection* lHitCol, std::vector vertexPosition, int nhits, double lowerLimit, double upperLimit, bool killEdges, bool scaleDR, int directionality) -{ - double NLL = 0; - - for (int ihit = 0; ihit < nhits; ihit++) - { - Hit lHit = lHitCol->At(ihit); - - int iPMT = lHit.PMT; - // double hitTime = (fTimeCorrection + lHit.T) / TimeDelta::ns; - PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - - int pmtType = Astro_GetPMTType(iPMT); - // double distance = Astro_GetDistance(lPMTInfo.Position, vertexPosition); - - // double tof = distance / fLightSpeed; - // double residual = hitTime - tof - vertexPosition[3]; - double residual = ComputeResidualTime(vertexPosition, vertexPosition[3], lHitCol->At(ihit)); - - residual = ComputeResidualTime(vertexPosition, vertexPosition[3], lHit); - - // std::vector toHit = VectorToHitNorm(vertexPosition, lHit); - // if(dot(fTrueDir, toHit) < cos(90 * TMath::DegToRad())) - // if(dot(fTrueDir, toHit) < 0) - // { - // continue; //! Skip this hit if it's not in the same hemisphere - // } - - double proba = 0; - - bool condition = residual > lowerLimit && residual < upperLimit; - - if (condition) - { - proba = fSplineTimePDFQueue[pmtType]->Eval(residual); - -#ifdef VERBOSE_NLL - if (VERBOSE >= 3) - { - std::cout << "hit#" << ihit << ", hit time =" << hitTime << ", vertex time = " << vertexPosition[3] << std::endl; - std::cout << "Residual=" << residual << ", proba=" << proba << ", pmt type=" << pmtType << std::endl; - } -#endif - - if (scaleDR) - { - std::cout << "scaleDR with Lower Limit: " << lowerLimit << " Upper Limit: " << upperLimit << std::endl; - EventInfo fEventInfo = MakeEventInfo(lowerLimit, upperLimit, pmtType); - - // First, substract the DR from the PDF to keep only the signal: - double DR = fSplineTimePDFDarkRate[pmtType]->Eval(residual); - proba -= DR; - // Then, scale signal so that signal integral / DR integral = signalOverNoise - // To do so, we should scale signal so that integral = DR integral * signalOverNoise - // And of course, to rescale, we should divide by the signal integral - double Factor = fEventInfo.NoiseIntegral * fEventInfo.SignaloverNoise / fEventInfo.SignalIntegral; - proba *= Factor; - // And add again the DR - proba += DR; -#ifdef VERBOSE_NLL - if (VERBOSE >= 3) std::cout << "proba after scaling=" << proba << std::endl; -#endif - } - } - else if (killEdges) - { - if (residual >= upperLimit) - { - proba = fSplineTimePDFQueue[pmtType]->Eval(upperLimit); -#ifdef VERBOSE_NLL - if (VERBOSE >= 3 && pmtType == 1) - std::cout << "PMT type = " << pmtType << ", Upper limit = " << upperLimit << ", residual = " << residual << ", proba = " << proba << std::endl; -#endif - } - else if (residual <= lowerLimit) - { - // continue; - proba = fSplineTimePDFQueue[pmtType]->Eval(lowerLimit); -#ifdef VERBOSE_NLL - if (VERBOSE >= 3 && pmtType == 1) - std::cout << "PMT type = " << pmtType << ", Lower limit = " << lowerLimit << ", residual = " << residual << ", proba = " << proba << std::endl; -#endif - } - } - else proba = 0; - - if (proba < 0) - { - std::cout << "Error in PDF" << std::endl; - if (proba > -1e-1) proba = 0; // Since spline sometimes slightly goes below 0 due to interpolation - else - { - std::cout << "Error in " << residual << "ns where proba = " << proba << std::endl; - return 0; - } - } - if (proba == 0) proba = 1e-20; - NLL += -TMath::Log(proba); - } - - //* Directionality, not used RN - -// if (directionality != 0) -// { -// double NLLdir = 0; - -// NLLdir = this->FindNLLDirectionality(vertexPosition, nhits, VERBOSE, lowerLimit, upperLimit); - -// #ifdef VERBOSE_NLL -// if (VERBOSE >= 2) -// std::cout << "NLL = " << NLL << ", dir (L) = " << TMath::Exp(-NLLdir) << ", dir 2 = " << NLLdir << std::endl; -// #endif -// if (directionality == 1) -// NLL += NLLdir; -// else if (directionality == 2) -// NLL = NLLdir; -// } - - // timer.Stop(); - // std::cout << "NLL = " << NLL << std::endl; - return NLL; -} - -double Likelihoods::Vertex_Score(const HitCollection* lHitCol, std::vector vertexPosition, int nhits, double /*lowerLimit*/, double /*upperLimit*/, bool /*killEdges*/, bool /*scaleDR*/, int /*directionality*/) -{ - double NLL = 0; - - // std::cout << " Find NLL " << nhits << std::endl; - // std::cout << " First Hit " << fHitCollection->At(0).PMT << " " << vertexPosition.size() << std::endl; - for (int ihit = 0; ihit < nhits; ihit++) - { - // Hit lHit = fHitInfo[ihit]; - Hit lHit = lHitCol->At(ihit); - - // std::cout << " NLL Hit " << ihit << " " << lHit.PMT << std::endl; - int iPMT = lHit.PMT; - - // double hitTime = (fTimeCorrection + lHit.T) / TimeDelta::ns; - PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - - int pmtType = Astro_GetPMTType(iPMT); - // double distance = Astro_GetDistance(lPMTInfo.Position, vertexPosition); - - // double tof = distance / fLightSpeed; - // double residual = hitTime - tof - vertexPosition[3]; - double residual = ComputeResidualTime(vertexPosition, vertexPosition[3], lHitCol->At(ihit)); -#ifdef KILLHALF - if (pmtType == 1) - { - double t = fRand->Uniform(0, 1); - if (t > 0.5) - continue; - } -#endif - - bool bCondition = (residual > fHitTimeLimitsNegative && residual < fHitTimeLimitsPositive) || (pmtType == 1 && !fLimit_mPMT); - - if (bCondition) - { - NLL++; -#ifdef VERBOSE_NLL - if (VERBOSE >= 3) - { - std::cout << "Residual=" << residual << ", pmt type=" << pmtType << std::endl; - std::cout << "hit#" << ihit << ", hit time =" << hitTime << ", vertex time = " << vertexPosition[3] << std::endl; - std::cout << "residual=" << residual << std::endl; - } -#endif - } - } - - return NLL != 0 ? -TMath::Log(NLL) : 1e15; -} - -double Likelihoods::FindNLL(const HitCollection* lHitCol, std::vector vertexPosition, int nhits, bool likelihood, int verbose, double lowerLimit, double upperLimit, bool killEdges, bool scaleDR, int directionality) -{ - double NLL = 0; - - for (int ihit = 0; ihit < nhits; ihit++) - { - Hit lHit = lHitCol->At(ihit); - - int iPMT = lHit.PMT; - // double hitTime = (fTimeCorrection + lHit.T) / TimeDelta::ns; - PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - - int pmtType = Astro_GetPMTType(iPMT); - // double distance = Astro_GetDistance(lPMTInfo.Position, vertexPosition); - - // double tof = distance / fLightSpeed; - // double residual = hitTime - tof - vertexPosition[3]; - double residual = ComputeResidualTime(vertexPosition, vertexPosition[3], lHitCol->At(ihit)); - double proba; -#ifdef KILLHALF - if (pmtType == 1) - { - double t = fRand->Uniform(0, 1); - if (t > 0.5) - continue; - } -#endif - - if (likelihood) - { - bool condition; - - condition = residual > lowerLimit && residual < upperLimit; - if (condition) - { - proba = fSplineTimePDFQueue[pmtType]->Eval(residual); -#ifdef VERBOSE_NLL - if (VERBOSE >= 3) - { - std::cout << "hit#" << ihit << ", hit time =" << hitTime << ", vertex time = " << vertexPosition[3] << std::endl; - std::cout << "Residual=" << residual << ", proba=" << proba << ", pmt type=" << pmtType << std::endl; - } -#endif - if (scaleDR) - { - std::cout << "scaleDR with Lower Limit: " << lowerLimit << " Upper Limit: " << upperLimit << std::endl; - EventInfo fEventInfo = MakeEventInfo(lowerLimit, upperLimit, pmtType); - - // First, substract the DR from the PDF to keep only the signal: - double DR = fSplineTimePDFDarkRate[pmtType]->Eval(residual); - proba -= DR; - // Then, scale signal so that signal integral / DR integral = signalOverNoise - // To do so, we should scale signal so that integral = DR integral * signalOverNoise - // And of course, to rescale, we should divide by the signal integral - double Factor = fEventInfo.NoiseIntegral * fEventInfo.SignaloverNoise / fEventInfo.SignalIntegral; - proba *= Factor; - // And add again the DR - proba += DR; - if (VERBOSE >= 3) - std::cout << "proba after scaling=" << proba << std::endl; - } - } - else if (killEdges && residual >= upperLimit) - { - proba = fSplineTimePDFQueue[pmtType]->Eval(upperLimit); -#ifdef VERBOSE_NLL - if (VERBOSE >= 3 && pmtType == 1) - std::cout << "PMT type = " << pmtType << ", Upper limit = " << upperLimit << ", residual = " << residual << ", proba = " << proba << std::endl; -#endif - } - else if (killEdges && residual <= lowerLimit) - { - proba = fSplineTimePDFQueue[pmtType]->Eval(lowerLimit); -#ifdef VERBOSE_NLL - if (VERBOSE >= 3 && pmtType == 1) - std::cout << "PMT type = " << pmtType << ", Lower limit = " << lowerLimit << ", residual = " << residual << ", proba = " << proba << std::endl; -#endif - } - else - { - proba = 0; - } - } - else - { - bool condition; - if (pmtType == 0 || (pmtType == 1 && fLimit_mPMT)) condition = residual > fHitTimeLimitsNegative && residual < fHitTimeLimitsPositive; - else condition = true; - if (condition) - { - NLL++; //= fSplineTimePDFQueue[pmtType]->Eval(residual); -#ifdef VERBOSE_NLL - if (VERBOSE >= 3) - { - std::cout << "Residual=" << residual << ", pmt type=" << pmtType << std::endl; - std::cout << "hit#" << ihit << ", hit time =" << hitTime << ", vertex time = " << vertexPosition[3] << ", distance PMT vs vertex = " << distance << ", tof=" << tof << std::endl; - std::cout << "residual=" << residual << ", proba=" << proba << std::endl; - } -#endif - } - else if (killEdges) continue; - else proba = 0; - } - - if (likelihood) - { - if (proba < 0) - { - std::cout << "Error in PDF" << std::endl; - if (proba > -1e-1) proba = 0; // Since spline sometimes slightly goes below 0 due to interpolation - else - { - std::cout << "Error in " << residual << "ns where proba = " << proba << std::endl; - return 0; - } - } - if (proba == 0) proba = 1e-20; - NLL += -TMath::Log(proba); - } - } - // std::cout << "FindNLL: Time it took for the loop = " << timer.RealTime() << std::endl; - // cout<<"NLL="<=2) cout<<"NLL="<38) - { - std::cout << "Got hit! " << theta << std::endl; - DNLL += -1; - } - - - if (VERBOSE >= 2) std::cout << "Direction NLL = " << DNLL << std::endl; - } - - return DNLL; - -} - -//Uses the PDF to find the DirNLL, used mainly for the MIGRAD optimization -double Likelihoods::Dir_NLL(const HitCollection* lHitCol, const std::vector& vertexPosition, double theta_track, double phi_track, int nhits) { - double DNLL = 0; - - // Ensure theta and phi are within valid ranges - // theta in [0, pi], phi in [-pi, pi] - if (theta_track < 0 || theta_track > TMath::Pi()) - { - std::cerr << "Error: theta_track is out of range [0, π]." << std::endl; - return std::numeric_limits::infinity(); - } - if (phi_track < -TMath::Pi() || phi_track > TMath::Pi()) - { - std::cerr << "Error: phi_track is out of range [-π, π]." << std::endl; - return std::numeric_limits::infinity(); - } - - // Precompute sine and cosine of theta_track and phi_track - // double sin_theta_track = sin(theta_track); - // double cos_theta_track = cos(theta_track); - // double sin_phi_track = sin(phi_track); - // double cos_phi_track = cos(phi_track); - /* 1 - double a = 0.556805; - double b = 1.06697; - double c = -0.902084; - double d = -0.0707118;*/ - - // 2 - // double a = -0.000440808; - // double b = 1.71313; - // double c = -2.01675; - // double d = 1.81755; - - // Direction vector components from theta and phi - // double vertexDirNormalized[3]; - // vertexDirNormalized[0] = sin_theta_track * cos_phi_track; - // vertexDirNormalized[1] = sin_theta_track * sin_phi_track; - // vertexDirNormalized[2] = cos_theta_track; - - for (int ihit = 0; ihit < nhits; ihit++) { - int trueNHits = lHitCol->Size(); - if(ihit >= trueNHits) - { - std::cout << "Error: ihit is out of range in direction NLL computation" << std::endl; - return std::numeric_limits::infinity(); - } - if(vertexPosition.size() != 4) - { - std::cout << "Error: vertexPosition is not of size 4 in direction NLL computation" << std::endl; - return std::numeric_limits::infinity(); - } - - Hit lHit = lHitCol->At(ihit); - int iPMT = lHit.PMT; - - // double hitTime = (fTimeCorrection + lHit.T) / TimeDelta::ns; - PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - - int pmtType = Astro_GetPMTType(iPMT); - // double distance = Astro_GetDistance(lPMTInfo.Position,vertexPosition); - - // double tof = distance / fLightSpeed; - // double residual = hitTime - tof - vertexPosition[3]; - - double residual = ComputeResidualTime(vertexPosition, vertexPosition[3], lHitCol->At(ihit)); - - bool DirCondition = (residual >= fDirectionPDF_minResidual && residual <= fDirectionPDF_maxResidual) || (pmtType == 1 && !fLimit_mPMT); - - - if(DirCondition || DirTakeAll) - { - int iPMT = lHit.PMT; - PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - - // Ensure lPMTInfo.Position and lPMTInfo.Orientation have the correct size - if (sizeof(lPMTInfo.Position) / sizeof(lPMTInfo.Position[0]) < 3 || - sizeof(lPMTInfo.Orientation) / sizeof(lPMTInfo.Orientation[0]) < 3) { - std::cerr << "Error: lPMTInfo.Position or lPMTInfo.Orientation is not of size 3." << std::endl; - continue; - } - - double dirVector[3]; - dirVector[0] = lPMTInfo.Position[0] - vertexPosition[0]; - dirVector[1] = lPMTInfo.Position[1] - vertexPosition[1]; - dirVector[2] = lPMTInfo.Position[2] - vertexPosition[2]; - - double dirNorm = sqrt(dirVector[0]*dirVector[0] + dirVector[1]*dirVector[1] + dirVector[2]*dirVector[2]); - if (dirNorm == 0) - { - std::cerr << "Error: dirNorm is zero, cannot normalize direction vector." << std::endl; - continue; - } - dirVector[0] /= dirNorm; - dirVector[1] /= dirNorm; - dirVector[2] /= dirNorm; - - // Convert dirVector to spherical coordinates (theta_pmt, phi_pmt)fDirectionPDF - double theta_pmt = acos(dirVector[2]); // theta in [0, π] - double phi_pmt = atan2(dirVector[1], dirVector[0]); // phi in [-π, π] - - // Calculate the angular difference between the two directions - // double delta_theta = theta_pmt - theta_track; //* could be useful - double delta_phi = phi_pmt - phi_track; - - // Ensure delta_phi is within [-π, π] - if (delta_phi > TMath::Pi()) delta_phi -= 2 * TMath::Pi(); - else if (delta_phi < -TMath::Pi()) delta_phi += 2 * TMath::Pi(); - - // Calculate the angle between the two directions using the spherical law of cosines - double cos_relative_angle = sin(theta_track) * sin(theta_pmt) * cos(delta_phi) + - cos(theta_track) * cos(theta_pmt); - - //cos_relative_angle = std::max(-1.0, std::min(1.0, cos_relative_angle)); - - double relative_angle = acos(cos_relative_angle); // Angle in radians - - // Convert angle to degrees if your PDF is defined in degrees - double theta = relative_angle * 180.0 / TMath::Pi(); - - // Evaluate your PDF at theta - // std::cout << "before pdf : " << ihit << std::endl; - // double proba = fDirectionPDF->Eval(cos_relative_angle)/*sin(theta)*/; //* if cos spline - // double proba = fDirectionPDF->Eval(cos_relative_angle); //* Taha - double proba = fHitAnglePDF->Eval(theta); //* Nicolas - // std::cout << "after pdf : " << ihit << std::endl; - - // Correct for PMT orientation if necessary - double PMTOrientation[3]; - PMTOrientation[0] = lPMTInfo.Orientation[0]; - PMTOrientation[1] = lPMTInfo.Orientation[1]; - PMTOrientation[2] = lPMTInfo.Orientation[2]; - - // Normalize PMT orientation vector - double pmtOrientNorm = sqrt(PMTOrientation[0]*PMTOrientation[0] + - PMTOrientation[1]*PMTOrientation[1] + - PMTOrientation[2]*PMTOrientation[2]); - - PMTOrientation[0] /= pmtOrientNorm; - PMTOrientation[1] /= pmtOrientNorm; - PMTOrientation[2] /= pmtOrientNorm; - - double cosPMThitAngle = dirVector[0]*PMTOrientation[0] + - dirVector[1]*PMTOrientation[1] + - dirVector[2]*PMTOrientation[2]; - // double PMThitAngle = acos(-cosPMThitAngle); - cosPMThitAngle *= -1; - // double cos2PMThitAngle = fabs(-2 * cosPMThitAngle * cosPMThitAngle + 1); - // double Ftheta = a + b*cosPMThitAngle + c*(pow(cosPMThitAngle,2)) + d*(pow(cosPMThitAngle,3)); - - //proba *= hitCharge; - - // Avoid log(0) by ensuring proba is positive - if (proba <= 0) - { - proba = 1e-20; - if (VERBOSE >= 1) std::cout << "Warning: Probability is zero or negative at hit #" << ihit << std::endl; - } - //std::cout << "Proba Corrected by " << PMThitAngle * 180 / M_PI << " degrees with a sincos " << sin(PMThitAngle) << ", " << cosPMThitAngle << std::endl; - // Accumulate Negative Log-Likelihood - DNLL += -TMath::Log(proba);//*(cosPMThitAngle/Ftheta)*(sin(PMThitAngle)); - - if (VERBOSE >= 3) std::cout << "Hit #" << ihit << ", Theta = " << theta << " degrees, Probability = " << proba << std::endl; - } - - if (VERBOSE >= 2) std::cout << "Total Direction NLL = " << DNLL << std::endl; - } - return DNLL; -} - -double Likelihoods::AngleNLL(const HitCollection* lHitCol, std::vector vertexPosition, std::vector vertexDirection) -{ - double NLL = 0; - - int nhits = lHitCol->Size(); - - for (int ihit = 0; ihit < nhits; ihit++) - { - //* compute angle - Hit lHit = lHitCol->At(ihit); - int iPMT = lHit.PMT; - PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - std::vector toPMT = std::vector(3); - for(int j = 0; j < 3; j++) toPMT[j] = lPMTInfo.Position[j] - vertexPosition[j]; - Normalize(toPMT); - Normalize(vertexDirection); - double dotP = dot(toPMT, vertexDirection); - dotP = std::max(-1.0, std::min(1.0, dotP)); // Ensure dot product is within valid range for acos - double angle = std::acos(dotP) * 180 / TMath::Pi(); - // std::cout<<"angle : " << angle << " on pdf : " << fHitAnglePDF->Eval(angle) << std::endl; - // NLL += -TMath::Log(std::max(1e-20, fDirectionPDF->Eval(dotP))); //? Taha - NLL += -TMath::Log(std::max(1e-20, fHitAnglePDF->Eval(angle))); //? Nicolas - } - - return NLL; -} - - -double Likelihoods::FindNLLDirectionality(const HitCollection* lHitCol, std::vector vVtxPos, int nhits, int verbose, double /*lowerLimit*/, double /*upperLimit*/) -{ - - double NLL = 0; - // std::vector vDirection(2, 0.); // Return phi and theta. - - // // Interpolate isn't compatible with multi-thread, need to use mutex which lead to long deadtime. - // // Copy TGraph2D - // mtx.lock(); - // static thread_local TGraph2D gPMTDirectionality_2D_local_0 = TGraph2D(*gPMTDirectionality_2D[MiniPMT][0]); - // static thread_local TGraph2D gPMTDirectionality_2D_local_1 = TGraph2D(*gPMTDirectionality_2D[MiniPMT][1]); - // static thread_local TGraph2D gPMTDirectionality_2D_local_2 = TGraph2D(*gPMTDirectionality_2D[MiniPMT][2]); - // mtx.unlock(); - - // TGraph2D *tDirectionality[3]; - // tDirectionality[0] = &gPMTDirectionality_2D_local_0; - // tDirectionality[1] = &gPMTDirectionality_2D_local_1; - // tDirectionality[2] = &gPMTDirectionality_2D_local_2; - - // for (int ihit = 0; ihit < nhits; ihit++) - // { - // // Hit lHit = fHitInfo[ihit]; - // Hit lHit = LeafInputs::fHitCollection->At(ihit); - - // int iPMT = lHit.PMT; - - // PMTInfo lPMTInfo = (*fPMTList)[iPMT]; - // int pmtType = Astro_GetPMTType(iPMT); - - // if (pmtType == 0) - // continue; - // bool condition = true; - - // if (condition) - // { - // double vPMTVtx[4]; - // this->VectorVertexPMT(vVtxPos, iPMT, vPMTVtx); - - // double dPhi = vPMTVtx[0]; - // double dTheta = vPMTVtx[1]; - // double dDist = vPMTVtx[2]; - - // int pmtGroup = lPMTInfo.mPMT_Group; - - // double proba = 0; - // proba = tDirectionality[pmtGroup]->Interpolate(dPhi, dTheta); - - // double dDistCorr = fDistResponsePMT[pmtType]->Eval(dDist); - - // proba *= dDistCorr; - - // if (proba == 0) - // { - // proba = 1e-20; - // if (verbose) - // std::cout << "We are at proba = 0, theta = " << dTheta << ", proba used = " << proba << std::endl; - // } - // NLL += -TMath::Log(proba); - - // if (VERBOSE >= 3) - // { - // // std::cout<<"PMT type="< - struct CompareByNLL { - bool operator()(T const &a, T const &b) const { - return a.NLL < b.NLL; - } - }; - - //* Calculate likelihood without using PDF, but just using hits within a given timing window (hits "in-time"). - static double Vertex_Score(const HitCollection* lHitCol, std::vector vertexPosition, int nhits, double lowerLimit, double upperLimit, bool killEdges, bool scaleDR, int directionality); - - //* Contain the two functions above in one function, where usage of PDF or not can be set through the flag: likelihood = true/false - static double FindNLL(const HitCollection* lHitCol, std::vector vertexPosition,int nhits, bool likelihood, int verbose, double lowerLimit, double upperLimit, bool killEdges=false, bool scaleDR=false, int directionality=false); - - static double Vertex_Time_NLL(const HitCollection* lHitCol, std::vector vertexPosition, int nhits, double lowerLimit, double upperLimit, bool killEdges, bool scaleDR, int directionality); - - static double Dir_NLL(const HitCollection* lHitCol, const std::vector& vertexPosition, const double theta_track, const double phi_track, int nhits); - - static double ComputeDirNLL_NoPDF(const HitCollection* lHitCol, const std::vector& vertexPosition, const std::vector& vertexDirection, int nhits); - - static double AngleNLL(const HitCollection* lHitCol, std::vector vertexPosition, std::vector vertexDirection); - - static double FindNLLDirectionality(const HitCollection* lHitCol, std::vector vertexPosition, int nhits, int verbose, double lowerLimit, double upperLimit); - - static double GoodnessOfFit(const HitCollection* lHitCol, std::vector vertexPosition, int nhits, double lowerLimit, double upperLimit, bool killEdges, bool scaleDR, int directionality); - -}; \ No newline at end of file diff --git a/leaf/SplineUtilities.cpp b/leaf/SplineUtilities.cpp new file mode 100644 index 0000000..0595516 --- /dev/null +++ b/leaf/SplineUtilities.cpp @@ -0,0 +1,28 @@ +#include "SplineUtilities.hpp" + +double SplineUtilities::SplineIntegral(TSpline3 *s, double start, double end, double stepSize) { + double integral = 0; + for (double i = start; i < end; i += stepSize) + { + integral += stepSize * s->Eval(i + stepSize / 2); + } + return integral; +} + +double SplineUtilities::SplineIntegralAndSubstract(TSpline3 *s0, TSpline3 *s1, double start, double end, double stepSize) { + double integral = 0; + for (double i = start; i < end; i += stepSize) + { + integral += stepSize * (s0->Eval(i + stepSize / 2) - s1->Eval(i + stepSize / 2)); + } + return integral; +} + +double SplineUtilities::SplineIntegralExpo(TSpline3 *s, double start, double end, double sigma, double stepSize) { + double integral = 0; + for (double i = start; i < end; i += stepSize) + { + integral += stepSize * s->Eval(i + stepSize / 2) * TMath::Gaus(i + stepSize / 2, 0, sigma); + } + return integral; +} \ No newline at end of file diff --git a/leaf/SplineUtilities.hpp b/leaf/SplineUtilities.hpp new file mode 100644 index 0000000..d7540bc --- /dev/null +++ b/leaf/SplineUtilities.hpp @@ -0,0 +1,13 @@ +#ifndef SPLINE_UTILITIES_HPP +#define SPLINE_UTILITIES_HPP + +#include +#include + +namespace SplineUtilities { + double SplineIntegral(TSpline3 *s, double start, double end, double stepSize=5e-1); + double SplineIntegralAndSubstract(TSpline3 *s0, TSpline3 *s1, double start, double end, double stepSize=5e-1); + double SplineIntegralExpo(TSpline3 *s, double start, double end, double sigma, double stepSize=5e-1); +}; // namespace SplineUtilities + +#endif \ No newline at end of file diff --git a/libWCSIM/.gitignore b/libWCSIM/.gitignore new file mode 100644 index 0000000..63f5566 --- /dev/null +++ b/libWCSIM/.gitignore @@ -0,0 +1,4 @@ +*/ +include +src +libWCSim* diff --git a/libWCSIM/include/#WCSimPmtInfo.hh# b/libWCSIM/include/#WCSimPmtInfo.hh# deleted file mode 100644 index 90a147a..0000000 --- a/libWCSIM/include/#WCSimPmtInfo.hh# +++ /dev/null @@ -1,52 +0,0 @@ -#ifndef WCSim_PmtInfo -#define WCSim_PmtInfo - -///////////////////////////////////////////////////////////////// -// -// Save the Pmt info -// -///////////////////////////////////////////////////////////////// - -#include "TObject.h" -#include "G4String.hh" - -class TDirectory; - -class WCSimPmtInfo : public TObject { -private: - Int_t cylocation; - Double_t trans_x; - Double_t trans_y; - Double_t trans_z; - Double_t orien_x; - Double_t orien_y; - Double_t orien_z; - Int_t tube; - Int_t mPMT; - Int_t mPMT_pmt; - -public: - WCSimPmtInfo(); - - WCSimPmtInfo(Int_t i, Double_t t1, Double_t t2, Double_t t3, Double_t o1, Double_t o2, Double_t o3, Int_t tubeid); - - WCSimPmtInfo(Int_t i, Double_t t1, Double_t t2, Double_t t3, Double_t o1, Double_t o2, Double_t o3, Int_t tubeid, Int_t mPMTid, Int_t mPMT_pmtid, G4String pmt_type = ""); - - virtual ~WCSimPmtInfo(); - - Int_t Get_cylocation(){return cylocation;} - Double_t Get_transx() {return trans_x;} - Double_t Get_transy() {return trans_y;} - Double_t Get_transz() {return trans_z;} - Double_t Get_orienx() {return orien_x;} - Double_t Get_orieny() {return orien_y;} - Double_t Get_orienz() {return orien_z;} - Int_t Get_tubeid(){return tube;} - Int_t Get_mPMTid(){return mPMT;} - Int_t Get_mPMT_pmtid(){return mPMT_pmt;} - G4String Get_pmt_type(){return pmt_type;}//Added by B.Quilain - - ClassDef(WCSimPmtInfo,2); -}; - -#endif diff --git a/libWCSIM/include/GdCaptureGammas.hh b/libWCSIM/include/GdCaptureGammas.hh deleted file mode 100755 index 40c9db6..0000000 --- a/libWCSIM/include/GdCaptureGammas.hh +++ /dev/null @@ -1,51 +0,0 @@ - -/////////////////////////////////////////////////////////////////////////////// -// Spectrum of radiative neutron capture by Gadolinium -// version 1.0.0 -// (Sep.09.2005) - -// Author : karim.zbiri@subatech.in2p3.fr - -//This file contains the gammas spectrum produced in radiative capture of -//neutrons by gadolinium. -//This work is adapted from earlier work in geant3 for chooz 1. - -//First version by Karim Zbiri, April, 2005 -/////////////////////////////////////////////////////////////////////////////// - - - -#ifndef GdCaptureGammas_hh -#define GdCaptureGammas_hh - -#include "G4ReactionProductVector.hh" -#include - -using namespace std; - -class GdCaptureGammas -{ - public: - - GdCaptureGammas(); - ~GdCaptureGammas(); - G4ReactionProductVector *GetGammas(G4int isotopeA); - vector Initialize(G4int isotopeA); - vector CapGad155(); - vector CapGad157(); - vector casca1Gd155(); - vector casca2Gd155(); - vector casca1Gd157(); - vector casca2Gd157(); - vector casca3Gd157(); - vector continuum(); - double cumule(int ); - void intc(); - - - public: - double Elevel; - double xint[4][750]; - -}; -#endif diff --git a/libWCSIM/include/GdNeutronHPCapture.hh b/libWCSIM/include/GdNeutronHPCapture.hh deleted file mode 100755 index 8d3d3ba..0000000 --- a/libWCSIM/include/GdNeutronHPCapture.hh +++ /dev/null @@ -1,48 +0,0 @@ - - -/////////////////////////////////////////////////////////////////////////////// -// Spectrum of radiative neutron capture by Gadolinium -// version 1.0.0 -// (Sep.09.2005) - -//Modified class from original G4NeutronHPCapture class to include -//the gammas spectrum of radiative neutron capture by Gadolinium. - -// Karim Zbiri, April, 2005 -/////////////////////////////////////////////////////////////////////////////// - - - -#ifndef GdNeutronHPCapture_h -#define GdNeutronHPCapture_h 1 - -#include "globals.hh" -#include "G4NeutronHPChannel.hh" -#include "G4HadronicInteraction.hh" - -class GdNeutronHPCapture : public G4HadronicInteraction -{ - public: - - GdNeutronHPCapture(); - - ~GdNeutronHPCapture(); - - G4HadFinalState * ApplyYourself(const G4HadProjectile& aTrack, G4Nucleus& aTargetNucleus); - - void BuildPhysicsTable(const G4ParticleDefinition&); - - private: - - G4double * xSec; - //G4NeutronHPChannel * theCapture; - std::vector* theCapture; - G4String dirName; - G4int numEle; - G4int it; - // G4String nomel; - - G4HadFinalState theResult; -}; - -#endif diff --git a/libWCSIM/include/GdNeutronHPCaptureFS.hh b/libWCSIM/include/GdNeutronHPCaptureFS.hh deleted file mode 100644 index 8efc03d..0000000 --- a/libWCSIM/include/GdNeutronHPCaptureFS.hh +++ /dev/null @@ -1,74 +0,0 @@ - - -/////////////////////////////////////////////////////////////////////////////// -// Spectrum of radiative neutron capture by Gadolinium -// version 1.0.0 -// (Sep.09.2005) - -// Author : karim.zbiri@subatech.in2p3.fr - -//Modified class from original G4NeutronHPCaptureFS class to deexcite and -//add correctly the secondary to the hadronic final state - -// Karim Zbiri, Aug, 2005 -/////////////////////////////////////////////////////////////////////////////// - - - -#ifndef GdNeutronHPCaptureFS_h -#define GdNeutronHPCaptureFS_h 1 - -#include "globals.hh" -#include "G4HadProjectile.hh" -#include "G4HadFinalState.hh" -#include "G4NeutronHPFinalState.hh" -#include "G4ReactionProductVector.hh" -#include "G4NeutronHPNames.hh" -#include "G4NeutronHPPhotonDist.hh" -#include "G4Nucleus.hh" -#include "G4Fragment.hh" - -#include "GdCaptureGammas.hh" - -class GdNeutronHPCaptureFS : public G4NeutronHPFinalState -{ - public: - - GdNeutronHPCaptureFS() - { - hasXsec = false; - targetMass = 0; - } - - ~GdNeutronHPCaptureFS() - { - } - - void UpdateNucleus( const G4Fragment* , G4double ); - void Init (G4double A, G4double Z, G4int M, G4String & dirName, G4String & aFSType); - G4HadFinalState * ApplyYourself(const G4HadProjectile & theTrack); - G4NeutronHPFinalState * New() - { - GdNeutronHPCaptureFS * theNew = new GdNeutronHPCaptureFS; - return theNew; - } - - private: - - G4Fragment * nucleus; - - G4DynamicParticle * theTwo ; - G4ReactionProduct theTarget; - G4Nucleus aNucleus; - G4ReactionProduct theNeutron; - - G4double targetMass; - - G4NeutronHPPhotonDist theFinalStatePhotons; - GdCaptureGammas theFinalgammas; - G4NeutronHPNames theNames; - - G4double theCurrentA; - G4double theCurrentZ; -}; -#endif diff --git a/libWCSIM/include/TJNuBeamFlux.hh b/libWCSIM/include/TJNuBeamFlux.hh deleted file mode 100644 index 15996de..0000000 --- a/libWCSIM/include/TJNuBeamFlux.hh +++ /dev/null @@ -1,107 +0,0 @@ -#ifndef JNuBeamFlux_hh_seen -#define JNuBeamFlux_hh_seen - -#include - -#include "TObject.h" -#include "TObjString.h" - -/// -/// JNuBeam flux pass-through info. Both GRooTracker and NRooTracker inherit -/// from this so as to avoid duplication of code. Currently up to date with -/// 11a version of pass-through info. -/// - -const int kNgmax = 12; - -// This is the base class for all RooTracker type objects. For now -// it is empty and just so that an pass round pointers to -// all derived classes (will be usefull for reweighting applications etc...) -class RooTrackerVtxBase : public TObject { - public: - RooTrackerVtxBase(); - virtual ~RooTrackerVtxBase(){} - ClassDef(RooTrackerVtxBase, 1); -}; - -class JNuBeamFlux : public RooTrackerVtxBase { - - public: - JNuBeamFlux(); - virtual ~JNuBeamFlux(); - - void Reset (void); - // Using methods from TObject to remove 'hidden' compiler warnings - using TObject::Copy; - void Copy (const JNuBeamFlux * flux); - - // - // Meta information - // - long NuFluxEntry; ///< corresponding entry in orig flux file - TObjString* NuFileName; ///< orig flux file name - // - // neutrino parent info - // - int NuParentPdg; ///< parent hadron pdg code - int NuParentDecMode; ///< parent hadron decay mode - double NuParentDecP4 [4]; ///< parent hadron 4-momentum at decay - double NuParentDecX4 [4]; ///< parent hadron 4-position at decay - float NuCospibm; ///< parent hadron cos(theta) relative to beam at decay - float NuNorm; ///< Normalization weight - double NuParentProP4 [4]; ///< parent hadron 4-momentum at production - double NuParentProX4 [4]; ///< parent hadron 4-position at production - float NuCospi0bm; ///< parent hadron cos(theta) relative to beam at production - float NuRnu; ///< neutrino r position at ND5/6 plane - float NuXnu [2]; ///< neutrino (x,y) position at ND5/6 plane - int NuIdfd; ///< Detector ID - // - // primary particle information - // - int NuGipart; ///< primary particle ID - float NuGpos0[3]; ///< primary particle starting point - float NuGvec0[3]; ///< primary particle direction at starting point - float NuGamom0; ///< momentum of the primary particle at the starting point - // - // Interation History Information - // - int NuNg; ///< Number of interaction steps - float NuGp[kNgmax][3]; ///< Momentum of ancestor particle - float NuGcosbm[kNgmax]; ///< Ancestor particle cos(theta) relative to beam - float NuGv[kNgmax][3]; ///< Position of ancestor particle - int NuGpid[kNgmax]; ///< PDG ancestor particle ID - int NuGmec[kNgmax]; ///< particle production mechanism of ancestor particle - // - // Normalization and Transfer Matrix Information - // - float NuEnusk; ///< Neutrino Energy at SK - float NuNormsk; ///< Normalization weight at SK - float NuAnorm; ///< ND Acceptance Weight - // - // Out-of-target Secondary Interactions - // - int NuGmat[kNgmax]; ///< material in which the particle originates - float NuGdistc[kNgmax]; ///< distance traveled through carbon - float NuGdistal[kNgmax]; ///< distance traveled through aluminum - float NuGdistti[kNgmax]; ///< distance traveled through titanium - float NuGdistfe[kNgmax]; ///< distance traveled through iron - // - // Beam parameter information - // - float NuVersion; ///< jnubeam flux version - int NuTuneid; ///< beam tune ID # - int NuNtrig; ///< Number of triggers - int NuPint; ///< Interaction model ID - float NuBpos[2]; ///< Beam center position - float NuBtilt[2]; ///< Beam angle - float NuBrms[2]; ///< Beam RMS width - float NuEmit[2]; ///< Beam Emittance - float NuAlpha[2]; ///< Beam Alpha - float NuHcur[3]; ///< Horn currents - int NuRand; ///< Random seed - // int NuRseed[2]; // Random seed - - ClassDef(JNuBeamFlux, 1); -}; - -#endif diff --git a/libWCSIM/include/TNRooTrackerVtx.hh b/libWCSIM/include/TNRooTrackerVtx.hh deleted file mode 100644 index 245d7e0..0000000 --- a/libWCSIM/include/TNRooTrackerVtx.hh +++ /dev/null @@ -1,195 +0,0 @@ -// -// -#ifndef NRooTrackerVtx_hh_seen -#define NRooTrackerVtx_hh_seen - -#include -#include - -#include "TObject.h" -#include "TBits.h" -#include "TObjString.h" - -#include "TJNuBeamFlux.hh" - -using std::ostream; - -/// -/// This is a simple event class which is essentially an objectified version -/// of the NEUT nRooTracker output format. Because the nRooTracker is forcing -/// NEUT event into a GENIE defined storage care needs to be taken when -/// interpreting the meaning of various data members. For example the StdHepStatus -/// does not have a one to one mapping to GENIE StdHep status. Using this class -/// with GENIE based utils or the GENIE ReWeighting tools can (and most likely -/// will result in false results! -/// -/// Therefore, the native NEUT particle information common blocks (VCWORK, FSIHIST) -/// are included. Analysis tools under development... - -/// DBrailsford 10-2013: All 1D arrays are now in a dynamic format (rather than fixed length). -/// Due to the class structure of the oaAnalysis NRooTracker object being used to extract the -/// information from the numc file, two versions of the 1D arrays are initialised. -/// Each array has a dynamic version, and a 'Temp' fixed length version. The 'Temp' version -/// extracts the info from the numc object and feeds the information in the dynamic version. -/// The 'Temp' version of each data member is NOT saved to the rooTracker tree, only the dynamic version is. -/// Note that for each new event, only the 'Temp' version of the data member is reset to a default value. -/// This current implementation reduces the NRooTracker tree by ~10Mb. - -const int kNStdHepNPmax = 100; -const int kNStdHepIdxPx = 0; -const int kNStdHepIdxPy = 1; -const int kNStdHepIdxPz = 2; -const int kNStdHepIdxE = 3; -const int kNStdHepIdxX = 0; -const int kNStdHepIdxY = 1; -const int kNStdHepIdxZ = 2; -const int kNStdHepIdxT = 3; - -const int kNEmaxvc = 100; -const int kNEmaxvert = 100; -const int kNEmaxvertp = 300; - -class NRooTrackerVtx : public JNuBeamFlux { - - public: - NRooTrackerVtx(); - ~NRooTrackerVtx(); - - void Reset (void); - void Init (void); - - // Using methods from TObject to remove 'hidden' compiler warnings - using TObject::Copy; - using TObject::Print; - - void Copy (const NRooTrackerVtx * event); - void Print (const Option_t* = "") const; - - // Define the output rootracker tree branches - TObjString* EvtCode; ///< generator-specific string with 'event code' - int EvtNum; ///< event num. - double EvtXSec; ///< cross section for selected event (1E-38 cm2) CORRECT - double EvtDXSec; ///< cross section for selected event kinematics (1E-38 cm2 /{K^n}) CORRECT - double EvtWght; ///< weight for that event CORRECT - double EvtProb; ///< probability for that event (given cross section, path lengths, etc) CORRECT - double EvtVtx[4]; ///< event vertex position in detector coord syst (SI) CORRECT - int StdHepN; ///< number of particles in particle array - // - // stdhep-like particle array - // - int* StdHepPdg; //[StdHepN] ///< pdg codes (& generator specific codes for pseudoparticles) - int StdHepPdgTemp [kNStdHepNPmax]; //! - - int* StdHepStatus; //[StdHepN] ///< generator-specific status code - int StdHepStatusTemp[kNStdHepNPmax]; //! - - double StdHepX4 [kNStdHepNPmax][4]; ///< 4-x (x, y, z, t) of particle in hit nucleus frame (fm) CORRECT - - double StdHepP4 [kNStdHepNPmax][4]; ///< 4-p (px,py,pz,E) of particle in LAB frame (GeV) CORRECT - - double StdHepPolz [kNStdHepNPmax][3]; ///< polarization vector CORRECT - - int StdHepFdTemp [kNStdHepNPmax]; //! - int StdHepLdTemp [kNStdHepNPmax]; //! - int StdHepFmTemp [kNStdHepNPmax]; //! - int StdHepLmTemp [kNStdHepNPmax]; //! - int* StdHepFd; //[StdHepN] ///< first daughter - int* StdHepLd; //[StdHepN] ///< last daughter - int* StdHepFm; //[StdHepN] ///< first mother - int* StdHepLm; //[StdHepN] ///< last mother - - // NEUT native VCWORK information - int NEnvc; ///< Number of particles - - int NEipvcTemp[kNEmaxvc]; //! PDG particle code - int* NEipvc; //[NEnvc] ///< PDG particle code - - float NEpvc[kNEmaxvc][3]; ///< 3-momentum (MeV/c) CORRECT - - int NEiorgvcTemp[kNEmaxvc]; //! - int* NEiorgvc; //[NEnvc] ///< Index of parent (Fortran convention: starting at 1) - - - /// Flag of final state: - /// 0 : DETERMINED LATER PROCEDURE - /// 1 : DECAY TO OTHER PARTICLE - /// 2 : ESCAPE FROM DETECTOR - /// 3 : ABSORPTION - /// 4 : CHARGE EXCHANGE - /// 5 : STOP A NOT CONSIDER IN M.C. - /// 6 : E.M. SHOWER - /// 7 : HADRON PRODUCTION - /// 8 : QUASI-ELASTIC SCATTER - /// 9 : FORWARD (ELASTIC-LIKE) SCATTER - - int NEiflgvcTemp[kNEmaxvc]; //! - int NEicrnvcTemp[kNEmaxvc]; //! - int* NEiflgvc; //[NEnvc] ///< Flag of final state - int* NEicrnvc; //[NEnvc] ///< Escaped nucleus (1) or not (0) - - - // Rest of the NEUT variables below are mainly for internal reweighting routines - - float NEcrsx; ///< Cross section calculation variables (currently used for coherent interactions) CORRECT - - float NEcrsy; ///< Cross section calculation variables (currently used for coherent interactions) CORRECT - - float NEcrsz; ///< Cross section calculation variables (currently used for coherent interactions) CORRECT - - float NEcrsphi; ///< Cross section calculation variables (currently used for coherent interactions) CORRECT - - - // NEUT FSIHIST pion interaction history - int NEnvert; ///< Number of vertices (including production and exit points) - float NEposvert[kNEmaxvert][3]; ///< Position of vertex within nucleus (fm) CORRECT - - ///< Interaction type - ///< (*10 FOR HI-NRG interaction, >~400 MeV/c) - ///< -1 : ESCAPE - ///< 0 : INITIAL (or unmatched parent vertex if I>1) - ///< 3 : ABSORPTION - ///< 4 : CHARGE EXCHANGE - ///< 7 : HADRON PRODUCTION (hi-nrg only, i.e. 70) - ///< 8 : QUASI-ELASTIC SCATTER - ///< 9 : FORWARD (ELASTIC-LIKE) SCATTER - - int NEiflgvertTemp[kNEmaxvert]; //! - int* NEiflgvert; //[NEnvert] ///< Interaction type - - - - int NEnvcvert; ///< Number of intermediate particles (including initial and final) - float NEdirvert[kNEmaxvertp][3]; ///< Direction of particle CORRECT - - float NEabspvertTemp[kNEmaxvertp]; //! - float NEabstpvertTemp[kNEmaxvertp]; //! - int NEipvertTemp[kNEmaxvertp]; //! - int NEivertiTemp[kNEmaxvertp]; //! - int NEivertfTemp[kNEmaxvertp]; //! - - float* NEabspvert; //[NEnvcvert] ///< Absolute momentum in the lab frame (MeV/c) CORRECT - float* NEabstpvert; //[NEnvcvert] ///< Absolute momentum in the nucleon rest frame (MeV/c) CORRECT - int* NEipvert; //[NEnvcvert] ///< PDG particle code - int* NEiverti; //[NEnvcvert] ///< Index of initial vertex (pointing to nvert array above) - int* NEivertf; //[NEnvcvert] ///< Index of final vertex (pointing to nvert array above) - - // - // etc - // - TObjString* GeomPath; // - - // Some pass through info - TObjString* GeneratorName; // - TObjString* OrigFileName; // - TObjString* OrigTreeName; // - int OrigEvtNum; - int OrigTreeEntries; - double OrigTreePOT; - double TimeInSpill; - - int TruthVertexID; - - ClassDef(NRooTrackerVtx, 1); -}; - -#endif diff --git a/libWCSIM/include/WCSimDarkRateMessenger.hh b/libWCSIM/include/WCSimDarkRateMessenger.hh deleted file mode 100644 index 928ca3d..0000000 --- a/libWCSIM/include/WCSimDarkRateMessenger.hh +++ /dev/null @@ -1,48 +0,0 @@ -#ifndef WCSimDarkRateMessenger_h -#define WCSimDarkRateMessenger_h 1 - - -#include "G4UImessenger.hh" -#include -#include "G4UIcmdWithAString.hh" - -class G4UIcommand; -class G4UIdirectory; -class G4UIcmdWithADoubleAndUnit; -class G4UIcmdWithADouble; -class G4UIcmdWithAnInteger; -class WCSimWCAddDarkNoise; - -class WCSimDarkRateMessenger: public G4UImessenger -{ -public: - static WCSimDarkRateMessenger* iInstance; - void Initialize(); - - static WCSimDarkRateMessenger* GetInstance(); - void AddDarkRateInstance(WCSimWCAddDarkNoise* darkratepoint, G4String detectorElement); - void RemoveDarkRateInstance(G4String detectorElement); - ~WCSimDarkRateMessenger(); - - void SetNewValue(G4UIcommand* command, G4String newValue); - -private: - WCSimDarkRateMessenger(WCSimWCAddDarkNoise*); - WCSimWCAddDarkNoise* WCSimAddDarkNoise; - std::map darknoiseptrs; - - G4UIdirectory* WCSimDir; - G4UIcmdWithADoubleAndUnit* SetFrequency; - G4UIcmdWithADouble* SetConversionRate; - G4UIcmdWithAnInteger* SetDarkMode; - G4UIcmdWithADoubleAndUnit* SetDarkLow; - G4UIcmdWithADoubleAndUnit* SetDarkHigh; - G4UIcmdWithADoubleAndUnit* SetDarkWindow; - G4UIcmdWithAString* SetDetectorElement; - - G4String initaliseString; - G4String detectorElement; - -}; - -#endif diff --git a/libWCSIM/include/WCSimDetectorConstruction.hh b/libWCSIM/include/WCSimDetectorConstruction.hh deleted file mode 100644 index 22c854e..0000000 --- a/libWCSIM/include/WCSimDetectorConstruction.hh +++ /dev/null @@ -1,699 +0,0 @@ -#ifndef WCSimDetectorConstruction_H -#define WCSimDetectorConstruction_H 1 - -#include "WCSimPmtInfo.hh" -#include "WCSimPMTObject.hh" -#include "WCSimRootOptions.hh" - -#include "G4Transform3D.hh" -#include "G4VUserDetectorConstruction.hh" -#include "G4LogicalVolume.hh" -#include "G4VPhysicalVolume.hh" -#include "G4OpticalSurface.hh" -#include "globals.hh" - -#include -#include -#include -//#include -// warning : hash_map is not part of the standard -//#include //TF: deprecated, but need new C++ features, probably from gcc4.2 onwards -#include //--> need to fix the "using" and namespace statements - -//instead of using forward declaration, just include: -#include "G4Material.hh" -// TF: ToDo: Are these required? -#include "TFile.h" -#include "TTree.h" -#include "TMath.h" - - - -//using __gnu_cxx::hash; //deprecated -//using __gnu_cxx::hash_map; - -// (JF) We don't need this distinction for DUSEL -//enum cyl_location {endcap1,wall,endcap2}; - -class G4Box; -class G4Tubs; -class G4Material; -class G4LogicalVolume; -class G4AssemblyVolume; -class G4VPhysicalVolume; -class WCSimTuningParameters; -class WCSimDetectorMessenger; -class WCSimWCSD; - -/* Deprecated -namespace __gnu_cxx { - template<> struct std::hash< std::string > - { - size_t operator()( const std::string& x ) const - { - return std::hash< const char* >()( x.c_str() ); - } - }; - }*/ - -//Move to G4Enumerations -enum mPMT_orientation{ - VERTICAL, - HORIZONTAL, - PERPENDICULAR}; - -class WCSimDetectorConstruction : public G4VUserDetectorConstruction -{ -public: - - WCSimDetectorConstruction(G4int DetConfig,WCSimTuningParameters* WCSimTuningPars); - ~WCSimDetectorConstruction(); - - void SaveOptionsToOutput(WCSimRootOptions * wcopt); - - G4VPhysicalVolume* Construct(); - - // Related to the WC geometry - void SetSuperKGeometry(); - void InitSinglePMT(); - void SetTestSinglemPMTGeometry(); - void Cylinder_60x74_3inchmPMT_14perCent(); - void Cylinder_60x74_3inchmPMT_40perCent(); - void Cylinder_60x74_3inch_14perCent(); - void Cylinder_60x74_3inch_40perCent(); - void SuperK_20inchPMT_20perCent(); - void SuperK_20inchBandL_20perCent(); - void SuperK_12inchBandL_15perCent(); - void SuperK_20inchBandL_14perCent(); - void Cylinder_60x74_20inchBandL_14perCent(); - void Cylinder_60x74_20inchBandL_40perCent(); - void Cylinder_12inchHPD_15perCent(); - void SetHyperKGeometry(); - void SetHyperK_3inchGeometry();//B.Q, 2018/05/03 - void SetHyperK_8inchGeometry();//B.Q, 2018/05/03 - void SetHyperK_10inchGeometry();//B.Q, 2018/05/09 - void SetHyperK_SKPMTGeometry();//B.Q, 2018/05/09 - void SetHyperK_mPMTGeometry();//B.Q, 2017/12/08 - void SetHyperK_HybridmPMTGeometry();//B.Q, 2018/12/10 - void SetHyperK_HybridmPMT10PCGeometry();//B.Q, 2019/01/26 - void SetHyperK_HybridFakeGeometry();//B.Q, 2019/01/26 - void SetNuPrismGeometry(G4String PMTType, G4double PMTCoverage, G4double detectorHeight, G4double detectorDiameter, G4double verticalPosition); - void SetNuPrism_mPMTGeometry(); - void SetNuPrismShort_mPMTGeometry(); - void SetDefaultNuPrismGeometry(); - void UpdateGeometry(); - - G4String GetDetectorName() {return WCDetectorName;} - G4double GetWaterTubeLength() {return WCLength;} - G4double GetWaterTubePosition() {return WCPosition;} - G4double GetPMTSize() {return WCPMTRadius;} - G4String GetPMTName() {return WCPMTName;} - G4int GetMyConfiguration() {return myConfiguration;} - G4double GetGeo_Dm(G4int); - G4int GetTotalNumPmts() {return totalNumPMTs;} - G4int GetTotalNum_mPmts() {return totalNum_mPMTs;} - G4int GetTotalNumPmts2() {return totalNumPMTs2;}//For the hybrid config - G4int GetTotalNum_mPmts2() {return totalNum_mPMTs2;}//For the hybrid config - - G4int GetPMT_QE_Method(){return PMT_QE_Method;} - G4double GetwaterTank_Length() {return waterTank_Length;} - G4int UsePMT_Coll_Eff(){return PMT_Coll_Eff;} - - G4double GetPMTSize1() {return WCPMTSize;} - - G4float GetPMTQE(G4String,G4float, G4int, G4float, G4float, G4float); - G4float GetPMTCollectionEfficiency(G4float theta_angle, G4String CollectionName) { return GetPMTPointer(CollectionName)->GetCollectionEfficiency(theta_angle); }; - - WCSimPMTObject *CreatePMTObject(G4String, G4String); - - std::map CollectionNameMap; - WCSimPMTObject * PMTptr; - - void SetPMTPointer(WCSimPMTObject* PMT, G4String CollectionName){ - CollectionNameMap[CollectionName] = PMT; - } - - WCSimPMTObject* GetPMTPointer(G4String CollectionName){ - PMTptr = CollectionNameMap[CollectionName]; - if (PMTptr == NULL) {G4cout << CollectionName << " is not a recognized hit collection. Exiting WCSim." << G4endl; exit(1);} - return PMTptr; - } - - G4ThreeVector GetWCOffset(){return WCOffset;} - G4ThreeVector GetWCXRotation(){return WCXRotation;} - G4ThreeVector GetWCYRotation(){return WCYRotation;} - G4ThreeVector GetWCZRotation(){return WCZRotation;} - G4ThreeVector GetWCDetCentre(){return WCDetCentre;} - - void SetWCDetCentre(double x, double y, double z){ - WCDetCentre[0] = x; - WCDetCentre[1] = y; - WCDetCentre[2] = z; - } - - // Related to the WC tube IDs - static G4int GetTubeID(std::string tubeTag){return tubeLocationMap[tubeTag];} - static G4Transform3D GetTubeTransform(int tubeNo){return tubeIDMap[tubeNo];} - //For the hybrid configuration - static G4int GetTubeID2(std::string tubeTag){return tubeLocationMap2[tubeTag];} - static G4Transform3D GetTubeTransform2(int tubeNo){return tubeIDMap2[tubeNo];} - - // Related to Pi0 analysis - G4bool SavePi0Info() {return pi0Info_isSaved;} - void SavePi0Info(G4bool choice) {pi0Info_isSaved=choice;} - - // Related to neutron capture analysis - G4bool SaveCaptureInfo() {return captureInfo_isSaved;} - void SaveCaptureInfo(G4bool choice) {captureInfo_isSaved=choice;} - - void SetPMT_QE_Method(G4int choice){PMT_QE_Method = choice;} - void SetPMT_Coll_Eff(G4int choice){PMT_Coll_Eff = choice;} - void SetVis_Choice(G4String choice){Vis_Choice = choice;} - G4String GetVis_Choice() {return Vis_Choice;} - - //Partition Length - void SetwaterTank_Length(G4double length){waterTank_Length = length;} - void SetWaterTubeLength(G4double length){WCLength = length;} - // Geometry options - void SetIsUpright(G4bool choice) {isUpright = choice;} - - // *** Begin Egg-Shaped HyperK Geometry *** - - void SetIsEggShapedHyperK(G4bool choice) {isEggShapedHyperK = choice;} - G4bool GetIsEggShapedHyperK() {return isEggShapedHyperK;} - - void SetEggShapedHyperKGeometry(); - void SetEggShapedHyperKGeometry_withHPD(); - - - // *** End Egg-Shaped HyperK Geometry *** - - // Getters and Setters for MultiPMT details from macro. - void SetmPMT_VesselRadius(G4double radius){ - mPMT_vessel_radius = radius; - - // WCBarrelPMTOffset is affected, so need to be updated!! - G4double mPMT_vessel_tot_height = mPMT_vessel_radius + mPMT_vessel_cyl_height; - // If no acrylic cover/pressure vessel - if(mPMT_vessel_tot_height < WCPMTRadius) - mPMT_vessel_tot_height = WCPMTRadius; - WCBarrelPMTOffset = mPMT_vessel_tot_height; // BarrelPMTOffset needs PMT/mPMT height - if(!fix_nModules){ - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(CLHEP::pi*WCPMTPercentCoverage)/(10.0*mPMT_vessel_radius)); - if(WCBarrelNumPMTHorizontal < 1) - G4cerr << "Bug in dimensions as less than 1 BarrelPMT specified" << G4endl; - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(CLHEP::pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (CLHEP::pi*WCIDDiameter/WCBarrelNumPMTHorizontal); - WCCapEdgeLimit = WCIDDiameter/2.0 - mPMT_vessel_tot_height; // CapEdgeLimit needs PMT/mPMT height - } - } - - void SetmPMT_VesselRadiusCurv(G4double radius){ - mPMT_vessel_radius_curv = radius; - } - - void SetmPMT_VesselCylHeight(G4double height){ - mPMT_vessel_cyl_height = height; - - G4double mPMT_vessel_tot_height = mPMT_vessel_radius + mPMT_vessel_cyl_height; - // If no acrylic cover/pressure mPMT_vessel - if(mPMT_vessel_tot_height < WCPMTRadius) - mPMT_vessel_tot_height = WCPMTRadius; - WCBarrelPMTOffset = mPMT_vessel_tot_height; - if(!fix_nModules) - WCCapEdgeLimit = WCIDDiameter/2.0 - mPMT_vessel_tot_height; - - } - void SetmPMT_DistPMTVessel(G4double dist){dist_pmt_vessel = dist;} - void SetmPMT_Orientation(mPMT_orientation orient){orientation = orient;} - void SetmPMT_ReflectorHeight(G4double ref_height){id_reflector_height = ref_height;} - void SetmPMT_ReflectorZoffset(G4double ref_zoffset){id_reflector_z_offset = ref_zoffset;} - void SetmPMT_ReflectorAngle(G4double ref_angle){id_reflector_angle = ref_angle;} - void SetmPMT_MaterialOuter(G4String outer_material){ - mPMT_outer_material = outer_material; - if(outer_material == "Acrylic") - mPMT_outer_material = "G4_PLEXIGLASS"; - } - void SetmPMT_MaterialInner(G4String inner_material){mPMT_inner_material = inner_material;} - void SetmPMT_PMTtype_inner(G4String type){ - mPMT_ID_PMT = type; - - //ToDo: replace by common function - WCSimPMTObject * PMT = CreatePMTObject(mPMT_ID_PMT, WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - } - void SetmPMT_PMTtype_outer(G4String type){mPMT_OD_PMT = type;} - void SetmPMT_MaterialOuterThickness(G4double thickness){ - mPMT_outer_material_d = thickness; - } - void SetmPMT_nID(G4int nPMTs){nID_PMTs = nPMTs;} - void SetmPMT_nID2(G4int nPMTs){nID_PMTs2 = nPMTs;}//Added by B.Quilain for the second type of photosensor in the hybrid version - void SetmPMT_Config(G4String inputfile){config_file = inputfile;} - - int GetmPMT_nID(void){return nID_PMTs;}; - int GetmPMT_nID2(void){return nID_PMTs2;};//Added by B.Quilain for the second type of photosensor in the hybrid version - G4String GetPMTtype_ID(void){return mPMT_ID_PMT;}; - G4String GetPMTtype_OD(void){return mPMT_OD_PMT;}; //might want to replace the name mPMT by general var name - - void SetmPMT_MaterialPMTassembly(G4String material){mPMT_material_pmtAssembly = material;} - void SetmPMT_FixModules(G4bool fix){fix_nModules = fix;} - void SetmPMT_OpeningAngle(G4double angle){mPMT_pmt_openingAngle = angle;} - - //Filling mPMT - G4int FillCircles(void); - // DEPRECATED: - G4int CountPMT(G4int NoPmt); - G4double ComputeEta (G4int NoPmt); - G4double ComputeAlpha (G4double alphaOfPrevC, G4double Eta); - G4int ComputeNiC (G4double alphaOfCircle, G4double Eta); - - - - - // Set if nuPRISM - void SetIsNuPrism(G4bool choice) {isNuPrism = choice;} - G4bool GetIsNuPrism() {return isNuPrism;} - void SetHybridPMT(G4bool choice) {hybrid = choice;} - - void SetPMTType(G4String type) { - WCPMTType = type; - //And update everything that is affected by a new PMT - WCSimPMTObject * PMT = CreatePMTObject(WCPMTType, WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - - WCBarrelPMTOffset = WCPMTRadius; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(CLHEP::pi*WCPMTPercentCoverage/100.0)/WCPMTRadius); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(CLHEP::pi*WCIDDiameter)))/WCPMTperCellVertical)); - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - - } - bool GetHybridPMT() {return hybrid;} - G4String GetPMTType() {return WCPMTType;} - - void SetPMTCoverage(G4double cover) { - // TF: do the same as for mPMTs to make "Update" deprecated: - // change the variables from WCDetectorConstruction that - // are affected. - WCPMTPercentCoverage = cover; - if(WCDetectorName == "NuPRISM_mPMT" || WCDetectorName == "HyperK_mPMT"){ - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(CLHEP::pi*WCPMTPercentCoverage/100.0)/mPMT_vessel_radius); - } else - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(CLHEP::pi*WCPMTPercentCoverage/100.0)/WCPMTRadius); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(CLHEP::pi*WCIDDiameter)))/WCPMTperCellVertical)); - WCCapPMTSpacing = (CLHEP::pi*WCIDDiameter/WCBarrelNumPMTHorizontal); - } - G4double GetPMTCoverage() {return WCPMTPercentCoverage;} - - void SetDopedWater(G4bool dopedWater){WCAddGd = dopedWater; } - void AddDopedWater(G4double percentGd = 0.1); - - std::vector* Get_Pmts() {return &fpmts;} - std::vector* Get_Pmts2() {return &fpmts2;}//For the hybrid config - - void SetDetectorHeight(G4double height) { - WCIDHeight = height; - // Affects Number of Barrel rings: - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(CLHEP::pi*WCIDDiameter)))/WCPMTperCellVertical)); - } - G4double GetWCIDHeight(){ return WCIDHeight; } - - void SetDetectorVerticalPosition(G4double position) {WCIDVerticalPosition = position;} - G4double GetWCIDVerticalPosition(){ return WCIDVerticalPosition; } - - void SetDetectorDiameter(G4double diameter) { - WCIDDiameter = diameter; - // Affects several cylinder parameters: - if(WCDetectorName == "NuPRISM_mPMT"){ - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(CLHEP::pi*WCPMTPercentCoverage/100.0)/mPMT_vessel_radius); - WCCapEdgeLimit = WCIDDiameter/2.0 - mPMT_vessel_radius; - } else{ - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(CLHEP::pi*WCPMTPercentCoverage/100.0)/WCPMTRadius); - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - } - - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(CLHEP::pi*WCIDDiameter)))/WCPMTperCellVertical)); - WCCapPMTSpacing = (CLHEP::pi*WCIDDiameter/WCBarrelNumPMTHorizontal); - - -} - G4double GetWCIDDiameter(){ return WCIDDiameter; } - - G4String GetIDCollectionName(){return WCIDCollectionName;} - G4String GetIDCollectionName2(){return WCIDCollectionName2;}//Added by B.Quilain for hybrid PMT configuration - WCSimTuningParameters* GetParameters(){return WCSimTuningParams;}//Added by TD to set up a parameter on PMT TTS uncertainty and use it in WCSimWCPMT -private: - - - // Tuning parameters - - WCSimTuningParameters* WCSimTuningParams; - - // Sensitive Detectors. We declare the pointers here because we need - // to check their state if we change the geometry, otherwise will segfault - // between events! - WCSimWCSD* aWCPMT; - - //Water, Blacksheet surface - G4OpticalSurface * OpWaterBSSurface; - - //Glass, Cathode surface in PMTs - - G4OpticalSurface * OpGlassCathodeSurface; - - //Tyvek surface - jl145 - G4OpticalSurface * OpWaterTySurface; - - //Reflector skin surface -tf - G4OpticalSurface * ReflectorSkinSurface; - - //Foam for mPMT: Gel - absorbing support structure -tf - G4OpticalSurface * OpGelFoamSurface; - - //TF fix for blacksheet errors "missing refractive index" - // or "photon travelling faster than c_light" when trying to refract - // through blacksheet due to undefined border crossing (after setting n_BS) - G4OpticalSurface * BSSkinSurface; - - - // The messenger we use to change the geometry. - - WCSimDetectorMessenger* messenger; - - // The Construction routines - G4LogicalVolume* ConstructCylinder(); - - G4LogicalVolume* ConstructPMT(G4String,G4String,G4String detectorElement="tank",G4int nIDPMTs=1);//Modified by B.Quilain 2018/12 to implement hybrid detector - G4LogicalVolume* ConstructMultiPMT(G4String,G4String, G4String detectorElement="tank",G4int nIDPMTs=1); - - - G4LogicalVolume* ConstructCaps(G4int zflip); - - void ConstructMaterials(); - - G4LogicalVolume* logicWCBarrelCellBlackSheet; - G4LogicalVolume* logicWCTowerBlackSheet; - G4double capAssemblyHeight; - - G4bool WCAddGd; - - // Code for traversing the geometry and assigning tubeIDs. - - // First make a typedef for the pointer to the member fcn. The - // syntax is too wacked out to be using all over. - - typedef void (WCSimDetectorConstruction::*DescriptionFcnPtr) - (G4VPhysicalVolume*, int, int, const G4Transform3D&); - - // Now Funcs for traversing the geometry - void TraverseReplicas(G4VPhysicalVolume*, int, const G4Transform3D&, - DescriptionFcnPtr); - - void DescribeAndDescendGeometry(G4VPhysicalVolume*, int, int, - const G4Transform3D&, DescriptionFcnPtr); - - // Functions that the traversal routines call or we use to manipulate the - // data we accumulate. - void DumpGeometryTableToFile(); - - void PrintGeometryTree(G4VPhysicalVolume*, int, int, const G4Transform3D&); - void DescribeAndRegisterPMT(G4VPhysicalVolume*, int, int, - const G4Transform3D&); - void DescribeAndRegisterPMT_1KT(G4VPhysicalVolume*, int, int, - const G4Transform3D&); - void GetWCGeom(G4VPhysicalVolume*, int, int, - const G4Transform3D&); - - //---Volume lengths - - // These are shared between the different member functions - // constructWC, constructFGD, constructlArD, constuctMRD - // toggle between FGD(0) and lArD(1) - // toggle between lArD readout types - // toggle between MRDScint and MRDk2k - - G4bool pi0Info_isSaved; - G4bool captureInfo_isSaved; - - - // XQ 08/17/10 - // PMT_QE_Method == 1 - // Only use it in the stacking function (no WLS) - // PMT_QE_Method == 2 - // Use Part of it in the stacking function (MAX QE) - // Put the rest of it in the sensitive detector according to QE/Max_QE - // PMT_QE_Method == 3 - // Put all of it in the sensitive detector according to QE - // Good for low energy photons - G4int PMT_QE_Method; - - //XQ 03/31/11 - // 0 to not use collection efficiency - // 1 to use - G4int PMT_Coll_Eff; - - //NP 06/17/15 - // "OGLSX" for classic visualization - // "RayTracer" for RayTracer visualization - G4String Vis_Choice; - - - G4double WCLength; - - G4double WCPosition; - - // Hit collection name parameters - G4String WCDetectorName; - G4String WCIDCollectionName; - G4String WCIDCollectionName2; - G4String WCODCollectionName; - G4bool hybrid = false; - - - // WC PMT parameters - G4String WCPMTName; - G4String WCPMTName2;//B. Quilain: for Hybrid configuration - typedef std::pair PMTKey_t; - typedef std::map PMTMap_t; - - static PMTMap_t PMTLogicalVolumes; - - // WC geometry parameters - G4double WCPMTRadius; - G4double WCPMTExposeHeight; - G4double WCBarrelPMTOffset; - - G4double WCPMTRadius2;//B. Quilain: for Hybrid configuration - G4double WCPMTExposeHeight2;//B. Quilain: for Hybrid configuration - G4double WCBarrelPMTOffset2;//B. Quilain: for Hybrid configuration - - G4double WCIDDiameter; - - G4double WCCapLength; - G4double WCBackODLength; - G4double WCFrontODLength; - G4double WCIDHeight; - G4double WCIDVerticalPosition; - - G4double WCBarrelRingRadius; - - G4double WCBarrelRingNPhi; - G4double WCBarrelNRings; - G4double WCPMTperCellHorizontal; - G4double WCPMTperCellVertical; - - G4double WCPMTPercentCoverage; - G4double WCPMTPercentCoverage2;//Added by B.Q for hybrid - - G4double WCBarrelNumPMTHorizontal; - G4double WCCapPMTSpacing; - G4double WCCapEdgeWidth;//jh ToDo: not used - - G4double WCCapEdgeLimit; - G4double WCBlackSheetThickness; - -// raise scope of derived parameters - G4double WCIDRadius; - G4double totalAngle; - G4double dPhi; - G4double barrelCellHeight; - G4double mainAnnulusHeight; - G4double innerAnnulusRadius; - G4double outerAnnulusRadius; - G4String water; - - - //for 1kt - G4double WCDiameter; - G4double WCRadius; - G4double WCBarrelPMTRadius; - G4double WCBarrelRingdPhi; - G4double WCBarrelCellLength; - G4double WCCapNCell; - G4double WCBarrelLength; - - // amb79: to universally make changes in structure and geometry - bool isUpright; - - - // Add bool to indicate whether we load nuPRISM geometry - G4bool isNuPrism; - G4String WCPMTType; - // G4double WCPMTCoverage; //TF: already using this variable "WCPMTPercentCoverage - - // *** Begin egg-shaped HyperK Geometry *** - - void MatchWCSimAndEggShapedHyperK(); - G4LogicalVolume* ConstructEggShapedHyperK(); - - G4Material* FindMaterial(G4String); - - G4VSolid* ConstructHalf(G4double, G4double); - - G4LogicalVolume* ConstructRadialPMT(G4bool, - G4double, G4double, - G4double, G4double, - G4double, G4double); - - G4LogicalVolume* ConstructEndWallPMT(); - - G4LogicalVolume* ConstructCeilingPMT(G4bool, - G4double, G4double, - G4double, G4double); - - G4bool isEggShapedHyperK; - - G4double waterTank_TopR; - G4double waterTank_BotR; - G4double waterTank_Height; - G4double waterTank_UpperA; - G4double waterTank_LowerB; - G4double waterTank_Length; - - G4double innerPMT_TopR; - G4double innerPMT_BotR; - G4double innerPMT_TopW; - G4double innerPMT_BotW; - G4double innerPMT_Height; - G4double innerPMT_Radius; - G4double innerPMT_Expose; - G4double innerPMT_Rpitch; - G4double innerPMT_Apitch; - - G4double outerPMT_TopR; - G4double outerPMT_BotR; - G4double outerPMT_TopW; - G4double outerPMT_BotW; - G4double outerPMT_Height; - G4double outerPMT_Radius; - G4double outerPMT_Expose; - G4String outerPMT_Name; - G4double outerPMT_TopRpitch; - G4double outerPMT_BotRpitch; - G4double outerPMT_Apitch; - - G4double blackSheetThickness; - - G4int innerPMT_TopN; - G4int innerPMT_BotN; - - G4bool checkOverlaps; - G4bool checkOverlapsPMT; - - G4bool placeBarrelPMTs; - G4bool placeCapPMTs; - G4bool placeBorderPMTs; - G4LogicalVolume* waterTankLV; - - G4int PMTCopyNo; - G4int wallSlabCopyNo; - - // *** End egg-shaped HyperK Geometry *** - - // amb79: debug to display all parts - bool debugMode; - - // Variables related to the geometry - - std::ofstream geoFile; // File for text output - - G4int totalNumPMTs=0; // The number of PMTs for this configuration - G4int totalNum_mPMTs=0; // The number of mPMTs (+1 for single PMT, +1 for mPMT) - G4int totalNumPMTs2=0; // The number of PMTs for this configuration, hybrid config - G4int totalNum_mPMTs2=0; // The number of mPMTs (+1 for single PMT, +1 for mPMT), hybrid config - G4double WCCylInfo[3]; // Info for the geometry tree: radius & length or mail box, length, width and depth - G4double WCPMTSize; // Info for the geometry tree: pmt size - G4double WCPMTSize2; // Info for the geometry tree: pmt size - G4ThreeVector WCOffset; // Info for the geometry tree: WC center offset - G4ThreeVector WCXRotation; // Info for the geometry tree: WC detector local X axis in the global coordinate system - G4ThreeVector WCYRotation; // Info for the geometry tree: WC detector local Y axis in the global coordinate system - G4ThreeVector WCZRotation; // Info for the geometry tree: WC detector local Z axis in the global coordinate system - G4ThreeVector WCDetCentre; - - // Tube map information - - //static std::map tubeIDMap; - //nding mPMT and mPMT_pmt ID -// static std::map tubeCylLocation; - //static hash_map > tubeLocationMap_old; //Deprecated - - static std::map tubeIDMap; - // static std::map tubeCylLocation; - static std::unordered_map > tubeLocationMap; - static std::map > mPMTIDMap; //maps tubeID to correspo - - // ID PMTs hybrid version - static std::map tubeIDMap2; - static std::unordered_map > tubeLocationMap2; - static std::map > mPMTIDMap2; //maps tubeID to correspo - //static std::unordered_map > tubeLocationMap; - - // Variables related to configuration - - G4int myConfiguration; // Detector Config Parameter - G4double innerradius; - - // Variables related to MultiPMTs - G4double vessel_cyl_height; - G4double vessel_radius_curv; // radius of the sphere to determine curvature of cap of pressure vessel - G4double vessel_radius; // radius of the pressure vessel (spherical cap) - - G4double mPMT_vessel_cyl_height; - G4double mPMT_vessel_radius_curv; // radius of the sphere to determine curvature of cap of pressure vessel - G4double mPMT_vessel_radius; // radius of the pressure vessel (spherical cap) - G4double dist_pmt_vessel; // distance between glass of pmt and inner radius of pressure vessel (region where water/gel lives) - mPMT_orientation orientation; - G4String mPMT_outer_material; - G4String mPMT_inner_material; - G4double mPMT_outer_material_d; - G4double id_reflector_height; - G4double id_reflector_z_offset; - G4double id_reflector_angle; - G4int nID_PMTs; - G4int nID_PMTs2; - G4String wcsimdir_path; - G4String config_file; - G4String mPMT_ID_PMT; //or ToDo: ideally ENUM - G4String mPMT_OD_PMT; - G4bool fix_nModules; - G4double mPMT_pmt_openingAngle; - G4String mPMT_material_pmtAssembly; - - //Filling mPMT - std::vector vNiC; // Nb of Chambers in each circle - std::vector vAlpha; // Tilt angle for each circle - std::vector vCircle; // Circle numbers - std::vector vAzimOffset; // Azimuthal offset of first PMT in each circle. - - - std::vector fpmts; - std::vector fpmts2;//For the hybrid config - -}; - -#endif - diff --git a/libWCSIM/include/WCSimDetectorMessenger.hh b/libWCSIM/include/WCSimDetectorMessenger.hh deleted file mode 100644 index f30a6bc..0000000 --- a/libWCSIM/include/WCSimDetectorMessenger.hh +++ /dev/null @@ -1,88 +0,0 @@ -#ifndef WCSimDetectorMessenger_h -#define WCSimDetectorMessenger_h 1 - -class WCSimDetectorConstruction; -class G4UIdirectory; -class G4UIcmdWithAString; - -#include "G4UImessenger.hh" -#include "globals.hh" -#include "G4ios.hh" -#include "G4UImessenger.hh" -#include "G4UIdirectory.hh" -#include "G4UIcmdWithADouble.hh" -#include "G4UIcmdWithADoubleAndUnit.hh" -#include "G4UIcmdWithoutParameter.hh" -#include "G4UIcmdWithABool.hh" -#include "G4UIcmdWithAString.hh" -#include "G4UIcmdWithAnInteger.hh" - - -class WCSimDetectorMessenger: public G4UImessenger -{ - public: - WCSimDetectorMessenger(WCSimDetectorConstruction* mpga); - ~WCSimDetectorMessenger(); - - public: - void SetNewValue(G4UIcommand* command, G4String newValues); - - private: - WCSimDetectorConstruction* WCSimDetector; - - private: //commands - - G4UIdirectory* WCSimDir; - G4UIcmdWithAString* PMTConfig; - G4UIcmdWithAString* WCVisChoice; - G4UIcmdWithAString* PMTGeomDetails; - G4UIcmdWithAString* PMTSize; - G4UIcmdWithAString* PMTSize2; - G4UIcmdWithAString* SavePi0; - G4UIcmdWithAString* SaveCapture; - G4UIcmdWithAString* PMTQEMethod; - G4UIcmdWithAString* PMTCollEff; - G4UIcmdWithABool* DopedWater; - G4UIcmdWithADouble* DopingConcentration; - G4UIcmdWithADoubleAndUnit* waterTank_Length; - - G4UIdirectory* mPMTDir; - G4UIcmdWithADoubleAndUnit* mPMT_CylHeight; - G4UIcmdWithADoubleAndUnit* mPMT_CylRadius; - G4UIcmdWithADoubleAndUnit* mPMT_CylRadiusCurv; - G4UIcmdWithADoubleAndUnit* mPMT_DistPMTVessel; - G4UIcmdWithAString* mPMT_orientation; - G4UIcmdWithAString* mPMT_material_inner; - G4UIcmdWithAString* mPMT_material_outer; - G4UIcmdWithAString* mPMT_PMTtype_inner; - G4UIcmdWithAString* mPMT_PMTtype_outer; - G4UIcmdWithADoubleAndUnit* mPMT_material_outer_thickness; - G4UIcmdWithADoubleAndUnit* mPMT_ID_reflector_height; - G4UIcmdWithADoubleAndUnit* mPMT_ID_reflector_z_offset; - G4UIcmdWithADoubleAndUnit* mPMT_ID_reflector_angle; - G4UIcmdWithAnInteger* mPMT_nID_PMTs; - G4UIcmdWithAString* mPMT_config; - G4UIcmdWithABool* mPMT_numModulesFixed; - G4UIcmdWithADoubleAndUnit* mPMT_pmtOpeningAngle; - G4UIcmdWithAString* mPMT_material_pmtAssembly; - //G4UIcmdWithAString* mPMT_PMTtype_inner; //enums internally - //G4UIcmdWithAString* mPMT_PMTtype_outer; - - G4UIcmdWithAString* tubeCmd; - G4UIcmdWithAString* distortionCmd; - G4UIcmdWithoutParameter* WCConstruct; - - //nuPRISM commands - G4UIcmdWithoutParameter* UpdateNuPrism; - G4UIcmdWithAString* SetPMTType; - G4UIcmdWithABool* SetHybridPMT; - G4UIcmdWithAString* SetPMTCoverage; - G4UIcmdWithADoubleAndUnit* SetDetectorHeight; - G4UIcmdWithADoubleAndUnit* SetDetectorVerticalPosition; - G4UIcmdWithADoubleAndUnit* SetDetectorDiameter; - -}; - -#endif - - diff --git a/libWCSIM/include/WCSimEnumerations.hh b/libWCSIM/include/WCSimEnumerations.hh deleted file mode 100644 index 4f25abf..0000000 --- a/libWCSIM/include/WCSimEnumerations.hh +++ /dev/null @@ -1,43 +0,0 @@ -#ifndef WCSimEnumerations_h -#define WCSimEnumerations_h 1 - -#include - -typedef enum ETriggerType { - kTriggerUndefined = -1, - kTriggerNDigits, - kTriggerNDigitsTest, - kTriggerFailure // this should always be the last entry (for looping) -} TriggerType_t; - -typedef enum EDigitizerType { - kDigitizerUndefined = -1, - kDigitizerSKI -} DigitizerType_t; - -typedef enum EInteractionMode { - UNKNOWN = -1, - BEAM = 0, - COSMICS = 10, - PARTICLEGUN = 20, - LASER = 30 -} InteractionType_t; - -typedef enum ERandomGeneratorType { - RANDOM_E_RANLUX=1, - RANDOM_E_RANECU=2, - RANDOM_E_HEPJAMES=3 -} WCSimRandomGenerator_t; - -class WCSimEnumerations -{ -public: - - static std::string EnumAsString(DigitizerType_t d); - static std::string EnumAsString(TriggerType_t t); - static std::string EnumAsString(WCSimRandomGenerator_t r); - static TriggerType_t TriggerTypeFromString(std::string s); - -}; - -#endif diff --git a/libWCSIM/include/WCSimEventAction.hh b/libWCSIM/include/WCSimEventAction.hh deleted file mode 100644 index e74a855..0000000 --- a/libWCSIM/include/WCSimEventAction.hh +++ /dev/null @@ -1,86 +0,0 @@ -#ifndef WCSimEventAction_h -#define WCSimEventAction_h 1 - - -#include "G4UserEventAction.hh" -#include "G4ThreeVector.hh" -#include "G4Types.hh" -#include "G4ios.hh" - -#include "WCSimDetectorConstruction.hh" -#include "G4TrajectoryContainer.hh" -#include "WCSimWCHit.hh" -#include "WCSimWCDigi.hh" -#include "WCSimWCTrigger.hh" -#include "WCSimWCDAQMessenger.hh" -#include "WCSimRootEvent.hh" -#include "WCSimRootGeom.hh" - -#include "TRandom3.h" - -class WCSimRunAction; -class WCSimPrimaryGeneratorAction; -class G4Event; - -class WCSimEventAction : public G4UserEventAction -{ -private: - WCSimRunAction* runAction; - WCSimPrimaryGeneratorAction* generatorAction; - WCSimDetectorConstruction* detectorConstructor; - - TRandom3 * randGen; - WCSimWCDAQMessenger* DAQMessenger; - -public: - WCSimEventAction(WCSimRunAction*, WCSimDetectorConstruction*, - WCSimPrimaryGeneratorAction*); - ~WCSimEventAction(); - -public: - void BeginOfEventAction(const G4Event*); - void EndOfEventAction(const G4Event*); - void FillRootEvent(G4int, - const struct ntupleStruct&, - G4TrajectoryContainer*, - WCSimWCDigitsCollection*, - WCSimWCTriggeredDigitsCollection*, - G4String detectorElement="tank"); - void FillRootEventHybrid(G4int, - const struct ntupleStruct&, - G4TrajectoryContainer*, - WCSimWCDigitsCollection*, - WCSimWCTriggeredDigitsCollection*, - G4String, - WCSimRootEvent*, - WCSimRootTrigger*); - WCSimRunAction* GetRunAction(){return runAction;} - void SetDigitizerChoice(G4String digitizer) { DigitizerChoice = digitizer; } - void SetTriggerChoice (G4String trigger) { TriggerChoice = trigger; } - - void FillFlatTree(G4int, - const struct ntupleStruct&, - G4TrajectoryContainer*, - WCSimWCDigitsCollection*, - WCSimWCTriggeredDigitsCollection*, - G4String detectorElement); - - -private: - G4int WCSimEventFindStartingVolume( G4ThreeVector vtx); - G4int WCSimEventFindStoppingVolume( G4String stopVolumeName); - - G4String vtxVolumeName; //TF new - - ///Create instances of the user-chosen digitizer and trigger classes - void CreateDAQInstances(); - - G4String DigitizerChoice; - G4String TriggerChoice; - bool ConstructedDAQClasses; - bool SavedOptions; -}; - - -#endif - diff --git a/libWCSIM/include/WCSimMultiPMTParameterisation.hh b/libWCSIM/include/WCSimMultiPMTParameterisation.hh deleted file mode 100644 index 6a81231..0000000 --- a/libWCSIM/include/WCSimMultiPMTParameterisation.hh +++ /dev/null @@ -1,75 +0,0 @@ -#ifndef WCSimMultiPMTParameterisation_h -#define WCSimMultiPMTParameterisation_h 1 - -#include "globals.hh" -#include "G4VPVParameterisation.hh" -#include -#include "G4RotationMatrix.hh" -#include "G4ThreeVector.hh" - - -class G4VPhysicalVolume; -class G4Tubs; - -// Dummy declarations to get rid of warnings ... -class G4Trd; -class G4Trap; -class G4Cons; -class G4Orb; -class G4Sphere; -class G4Ellipsoid; -class G4Torus; -class G4Para; -class G4Hype; -class G4Box; -class G4Polycone; -class G4Polyhedra; - -/// A parameterisation that describes a series of boxes along Z. -/// -/// The boxes have equal width, & their lengths are a linear equation. -/// They are spaced an equal distance apart, starting from given location. - -class WCSimMultiPMTParameterisation : public G4VPVParameterisation -{ - public: - - WCSimMultiPMTParameterisation(G4int noPmt, - G4double apoth, - std::vector& vNiC, - std::vector& vAlpha, - std::vector& vCircle, - std::vector& vAzimOffset, - G4double height); - - virtual ~WCSimMultiPMTParameterisation(); - void ComputeTransformation (const G4int copyNo, - G4VPhysicalVolume* physVol) const; - - -private: - - unsigned int fNoPmt; - G4double fApothema; // The distance from the Z axis - - // Local NiC, Alpha and Circle vectors - std::vector vNiCLocal; - std::vector vAlphaLocal; - std::vector vCircleLocal; - std::vector vAzimOffsetLocal; - - G4double fHeight; - std::vector vPMTpos; - std::vector vPMTorient; - - //preCalc position and rotation once to reduce CPU time and mem allocation in ComputeTransformation. - // The latter is called a lot throughout G4 execution (ie. during tracking for each event). - void PreCalculateTransform(); - - - -}; - -//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... - -#endif diff --git a/libWCSIM/include/WCSimPMTObject.hh b/libWCSIM/include/WCSimPMTObject.hh deleted file mode 100644 index 41838aa..0000000 --- a/libWCSIM/include/WCSimPMTObject.hh +++ /dev/null @@ -1,427 +0,0 @@ -#ifndef WCSimWCPMTObject_h -#define WCSimWCPMTObject_h 1 - -#include "WCSimTuningParameters.hh" -#include "WCSimDetectorConstruction.hh" -#include "globals.hh" -#include "Randomize.hh" -#include -#include - - -class WCSimPMTObject -{ - -public: - virtual G4String GetPMTName()=0; - virtual G4double GetExposeHeight()=0; - virtual G4double GetRadius()=0; - virtual G4float* Getqpe()=0; - virtual G4double* GetQE()=0; - virtual G4float* GetQEWavelength()=0; - virtual G4float GetmaxQE()=0; - virtual G4float GetCollectionEfficiency(float); - virtual float HitTimeSmearing(float, float /*, float*/)=0; - virtual float SaturFactor(float, float)=0; //TD 2019.07.16 - virtual G4double GetPMTGlassThickness()=0; - virtual G4float GetDarkRate()=0; - virtual G4float GetDarkRateConversionFactor()=0; -protected: - virtual G4float* GetCollectionEfficiencyArray(); - virtual G4float* GetCollectionEfficiencyAngle(); - G4float Interpolate_func(G4float, G4int, G4float*, G4float*); -}; - -class PMT20inch : public WCSimPMTObject -{ - -public: - - PMT20inch(); - ~PMT20inch(); - -public: - G4String GetPMTName() ; - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); - - -}; - -class PMT8inch : public WCSimPMTObject -{ - -public: - - PMT8inch(); - ~PMT8inch(); - -public: - G4String GetPMTName(); - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); -}; - -class PMT5inch : public WCSimPMTObject -{ - -public: - - PMT5inch(); - ~PMT5inch(); - -public: - G4String GetPMTName(); - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); -}; - -class PMT3inch : public WCSimPMTObject -{ - -public: - - PMT3inch(); - ~PMT3inch(); - -public: - G4String GetPMTName(); - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); -}; - -class PMT3inchGT : public WCSimPMTObject -{ - -public: - - PMT3inchGT(); - ~PMT3inchGT(); - -public: - G4String GetPMTName(); - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); -}; - - - class PMT10inch : public WCSimPMTObject -{ - -public: - PMT10inch(); - ~PMT10inch(); - -public: - G4String GetPMTName(); - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); - }; - - class PMT10inchHQE : public WCSimPMTObject -{ - -public: - PMT10inchHQE(); - ~PMT10inchHQE(); - -public: - G4String GetPMTName(); - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); - }; - - class PMT12inchHQE : public WCSimPMTObject -{ - -public: - PMT12inchHQE(); - ~PMT12inchHQE(); - -public: - G4String GetPMTName(); - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); - }; - -class HPD20inchHQE : public WCSimPMTObject -{ - -public: - - HPD20inchHQE(); - ~HPD20inchHQE(); - -public: - G4String GetPMTName() ; - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); -protected: - G4float* GetCollectionEfficiencyArray(); -}; - -class HPD12inchHQE : public WCSimPMTObject -{ - -public: - - HPD12inchHQE(); - ~HPD12inchHQE(); - -public: - G4String GetPMTName() ; - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); -protected: - G4float* GetCollectionEfficiencyArray(); -}; - -class BoxandLine20inchHQE : public WCSimPMTObject -{ - -public: - - BoxandLine20inchHQE(); - ~BoxandLine20inchHQE(); - -public: - G4String GetPMTName() ; - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); //has to be double due to double math inside array ("narrowing conversion" error in C++11) - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); -protected: - G4float* GetCollectionEfficiencyArray(); -}; - -class BoxandLine12inchHQE : public WCSimPMTObject -{ - -public: - - BoxandLine12inchHQE(); - ~BoxandLine12inchHQE(); - -public: - G4String GetPMTName() ; - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); - -protected: - G4float* GetCollectionEfficiencyArray(); -}; - - -// ToDo: make object of classes instead of infinite cp-pastes -class PMT3inchR12199_02 : public WCSimPMTObject -{ - -public: - - PMT3inchR12199_02(); - ~PMT3inchR12199_02(); - -public: - G4String GetPMTName() ; - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); -protected: - G4float* GetCollectionEfficiencyArray(); -}; - -//B.Q, 2019/01/18 -// ToDo: make object of classes instead of infinite cp-pastes -class PMT3inchR14374 : public WCSimPMTObject -{ - -public: - - PMT3inchR14374(); - ~PMT3inchR14374(); - -public: - G4String GetPMTName() ; - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); -protected: - G4float* GetCollectionEfficiencyArray(); -}; - -// ToDo: make object of classes instead of infinite cp-pastes -class PMT4inchR12199_02 : public WCSimPMTObject -{ - -public: - - PMT4inchR12199_02(); - ~PMT4inchR12199_02(); - -public: - G4String GetPMTName() ; - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); -protected: - G4float* GetCollectionEfficiencyArray(); -}; - -// ToDo: make object of classes instead of infinite cp-pastes -class PMT5inchR12199_02 : public WCSimPMTObject -{ - -public: - - PMT5inchR12199_02(); - ~PMT5inchR12199_02(); - -public: - G4String GetPMTName() ; - G4double GetExposeHeight(); - G4double GetRadius(); - G4float* Getqpe(); - G4double* GetQE(); - G4float* GetQEWavelength(); - G4float GetmaxQE(); - float HitTimeSmearing(float, float); - float SaturFactor(float, float); //TD 2019.07.16 - G4double GetPMTGlassThickness(); - G4float GetDarkRate(); - G4float GetDarkRateConversionFactor(); -protected: - G4float* GetCollectionEfficiencyArray(); -}; - - -#endif diff --git a/libWCSIM/include/WCSimPhysicsListFactory.hh b/libWCSIM/include/WCSimPhysicsListFactory.hh deleted file mode 100644 index bce7397..0000000 --- a/libWCSIM/include/WCSimPhysicsListFactory.hh +++ /dev/null @@ -1,45 +0,0 @@ -#ifndef WCSimPhysicsListFactory_h -#define WCSimPhysicsListFactory_h 1 - -#include "globals.hh" -#include "G4VModularPhysicsList.hh" -#include "G4PhysListFactory.hh" -#include "G4UnitsTable.hh" -#include "G4OpticalPhysics.hh" - -#include "WCSimPhysicsListFactoryMessenger.hh" -#include "WCSimRootOptions.hh" - -//class WCSimPhysicsList; - -class WCSimPhysicsListFactory : public G4VModularPhysicsList -{ - public: - WCSimPhysicsListFactory(); - ~WCSimPhysicsListFactory(); - - void SetList(G4String newvalue); // called by messenger - void SetnCaptModel(G4String newvalue); // called by messenger - void InitializeList(); - - //G4String GetPhysicsListName() {return PhysicsListName;} - - void ConstructParticle(); - void ConstructProcess(); - void SetCuts(); - - void SaveOptionsToOutput(WCSimRootOptions * wcopt); - - private: - - G4String PhysicsListName; - G4String ValidListsString; - - G4String nCaptModelChoice; - - WCSimPhysicsListFactoryMessenger* PhysicsMessenger; - G4PhysListFactory* factory; - -}; - -#endif diff --git a/libWCSIM/include/WCSimPhysicsListFactoryMessenger.hh b/libWCSIM/include/WCSimPhysicsListFactoryMessenger.hh deleted file mode 100644 index 83de751..0000000 --- a/libWCSIM/include/WCSimPhysicsListFactoryMessenger.hh +++ /dev/null @@ -1,30 +0,0 @@ -#ifndef WCSimPhysicsListFactoryMessenger_h -#define WCSimPhysicsListFactoryMessenger_h 1 - -#include "globals.hh" -#include "G4UImessenger.hh" - -class WCSimPhysicsListFactory; -class G4UIdirectory; -class G4UIcommand; -class G4UIcmdWithAString; - -class WCSimPhysicsListFactoryMessenger: public G4UImessenger -{ -public: - WCSimPhysicsListFactoryMessenger(WCSimPhysicsListFactory*, G4String); - ~WCSimPhysicsListFactoryMessenger(); - - void SetNewValue(G4UIcommand* command, G4String newValue); - -private: - WCSimPhysicsListFactory* thisWCSimPhysicsListFactory; - G4String ValidListsString; - - G4UIdirectory* WCSimDir; - G4UIcmdWithAString* physListCmd; - G4UIcmdWithAString* nCaptureModelCmd; - -}; - -#endif diff --git a/libWCSIM/include/WCSimPmtInfo.hh b/libWCSIM/include/WCSimPmtInfo.hh deleted file mode 100644 index 7347094..0000000 --- a/libWCSIM/include/WCSimPmtInfo.hh +++ /dev/null @@ -1,50 +0,0 @@ -#ifndef WCSim_PmtInfo -#define WCSim_PmtInfo - -///////////////////////////////////////////////////////////////// -// -// Save the Pmt info -// -///////////////////////////////////////////////////////////////// - -#include "TObject.h" - -class TDirectory; - -class WCSimPmtInfo : public TObject { -private: - Int_t cylocation; - Double_t trans_x; - Double_t trans_y; - Double_t trans_z; - Double_t orien_x; - Double_t orien_y; - Double_t orien_z; - Int_t tube; - Int_t mPMT; - Int_t mPMT_pmt; - -public: - WCSimPmtInfo(); - - WCSimPmtInfo(Int_t i, Double_t t1, Double_t t2, Double_t t3, Double_t o1, Double_t o2, Double_t o3, Int_t tubeid); - - WCSimPmtInfo(Int_t i, Double_t t1, Double_t t2, Double_t t3, Double_t o1, Double_t o2, Double_t o3, Int_t tubeid, Int_t mPMTid, Int_t mPMT_pmtid); - - virtual ~WCSimPmtInfo(); - - Int_t Get_cylocation(){return cylocation;} - Double_t Get_transx() {return trans_x;} - Double_t Get_transy() {return trans_y;} - Double_t Get_transz() {return trans_z;} - Double_t Get_orienx() {return orien_x;} - Double_t Get_orieny() {return orien_y;} - Double_t Get_orienz() {return orien_z;} - Int_t Get_tubeid(){return tube;} - Int_t Get_mPMTid(){return mPMT;} - Int_t Get_mPMT_pmtid(){return mPMT_pmt;} - - ClassDef(WCSimPmtInfo,2); -}; - -#endif diff --git a/libWCSIM/include/WCSimPrimaryGeneratorAction.hh b/libWCSIM/include/WCSimPrimaryGeneratorAction.hh deleted file mode 100644 index 8f49b1b..0000000 --- a/libWCSIM/include/WCSimPrimaryGeneratorAction.hh +++ /dev/null @@ -1,167 +0,0 @@ -#ifndef WCSimPrimaryGeneratorAction_h -#define WCSimPrimaryGeneratorAction_h - -#include "G4VUserPrimaryGeneratorAction.hh" -#include "G4ThreeVector.hh" -#include "globals.hh" - -#include "WCSimEnumerations.hh" - -#include - -#include "WCSimRootOptions.hh" -#include "TFile.h" -#include "TTree.h" -#include "TNRooTrackerVtx.hh" -#include "TClonesArray.h" - -class WCSimDetectorConstruction; -class G4ParticleGun; -class G4GeneralParticleSource; -class G4Event; -class WCSimPrimaryGeneratorMessenger; - -class WCSimPrimaryGeneratorAction : public G4VUserPrimaryGeneratorAction -{ - public: - WCSimPrimaryGeneratorAction(WCSimDetectorConstruction*); - ~WCSimPrimaryGeneratorAction(); - - public: - void GeneratePrimaries(G4Event* anEvent); - void SetupBranchAddresses(NRooTrackerVtx* nrootrackervtx); - void OpenRootrackerFile(G4String fileName); - void CopyRootrackerVertex(NRooTrackerVtx* nrootrackervtx); - bool GetIsRooTrackerFileFinished(){return (fEvNum==fNEntries);} - - // Gun, laser & gps setting calls these functions to fill jhfNtuple and Root tree - void SetVtx(G4ThreeVector i) { vtx = i; }; - void SetBeamEnergy(G4double i) { beamenergy = i; }; - void SetBeamDir(G4ThreeVector i) { beamdir = i; }; - void SetBeamPDG(G4int i) { beampdg = i; }; - - // These go with jhfNtuple - G4int GetVecRecNumber(){return vecRecNumber;} - //G4int GetMode() {return mode;}; - InteractionType_t GetMode() {return mode;}; - G4int GetVtxVol() {return vtxvol;}; - G4ThreeVector GetVtx() {return vtx;} - G4int GetNpar() {return npar;}; - G4int GetBeamPDG() {return beampdg;}; - G4double GetBeamEnergy() {return beamenergy;}; - G4ThreeVector GetBeamDir() {return beamdir;}; - G4int GetTargetPDG() {return targetpdg;}; - G4double GetTargetEnergy() {return targetenergy;}; - G4ThreeVector GetTargetDir() {return targetdir;}; - - // older ... - G4double GetNuEnergy() {return nuEnergy;}; - G4double GetEnergy() {return energy;}; - G4double GetXPos() {return xPos;}; - G4double GetYPos() {return yPos;}; - G4double GetZPos() {return zPos;}; - G4double GetXDir() {return xDir;}; - G4double GetYDir() {return yDir;}; - G4double GetZDir() {return zDir;}; - - G4String GetGeneratorTypeString(); - - void SaveOptionsToOutput(WCSimRootOptions * wcopt); - - private: - WCSimDetectorConstruction* myDetector; - G4ParticleGun* particleGun; - G4GeneralParticleSource* MyGPS; //T. Akiri: GPS to run Laser - WCSimPrimaryGeneratorMessenger* messenger; - - // Variables set by the messenger - G4bool useMulineEvt; - G4bool useRootrackerEvt; - G4bool useGunEvt; - G4bool useLaserEvt; //T. Akiri: Laser flag - G4bool useGPSEvt; - std::fstream inputFile; - G4String vectorFileName; - G4bool GenerateVertexInRock; - G4bool usePoissonPMT; - G4double poissonPMTMean; - - // These go with jhfNtuple - //G4int mode; - InteractionType_t mode; - G4int vtxvol; - G4ThreeVector vtx; - G4int npar; - G4int beampdg, targetpdg; - G4ThreeVector beamdir, targetdir; - G4double beamenergy, targetenergy; - G4int vecRecNumber; - - G4double nuEnergy; - G4double energy; - G4double xPos, yPos, zPos; - G4double xDir, yDir, zDir; - - G4int _counterRock; - G4int _counterCublic; - - // Counters to read Rootracker event file - int fEvNum; - int fNEntries; - TFile* fInputRootrackerFile; - - // Pointers to Rootracker vertex objects - // Temporary vertex that is saved if desired, according to WCSimIO macro option - TTree* fRooTrackerTree; - TTree* fSettingsTree; - NRooTrackerVtx* fTmpRootrackerVtx; - float fNuPrismRadius; - float fNuBeamAng; - float fNuPlanePos[3]; - - public: - - inline TFile* GetInputRootrackerFile(){ return fInputRootrackerFile;} - - inline void SetMulineEvtGenerator(G4bool choice) { useMulineEvt = choice; } - inline G4bool IsUsingMulineEvtGenerator() { return useMulineEvt; } - - inline void SetRootrackerEvtGenerator(G4bool choice) { useRootrackerEvt = choice; } - inline G4bool IsUsingRootrackerEvtGenerator() { return useRootrackerEvt; } - - inline void SetGunEvtGenerator(G4bool choice) { useGunEvt = choice; } - inline G4bool IsUsingGunEvtGenerator() { return useGunEvt; } - - //T. Akiri: Addition of function for the laser flag - inline void SetLaserEvtGenerator(G4bool choice) { useLaserEvt = choice; } - inline G4bool IsUsingLaserEvtGenerator() { return useLaserEvt; } - - inline void SetGPSEvtGenerator(G4bool choice) { useGPSEvt = choice; } - inline G4bool IsUsingGPSEvtGenerator() { return useGPSEvt; } - - inline void OpenVectorFile(G4String fileName) - { - if ( inputFile.is_open() ) - inputFile.close(); - - vectorFileName = fileName; - inputFile.open(vectorFileName, std::fstream::in); - if ( !inputFile.is_open() ) { - G4cout << "Vector file " << vectorFileName << " not found" << G4endl; - exit(-1); - } - - } - inline G4bool IsGeneratingVertexInRock() { return GenerateVertexInRock; } - inline void SetGenerateVertexInRock(G4bool choice) { GenerateVertexInRock = choice; } - - inline void SetPoissonPMT(G4bool choice) { usePoissonPMT = choice; } - inline G4bool IsUsingPoissonPMT(){ return usePoissonPMT; } - - inline void SetPoissonPMTMean(G4double val){ poissonPMTMean = val; } - inline G4double GetPoissonPMTMean(){ return poissonPMTMean; } -}; - -#endif - - diff --git a/libWCSIM/include/WCSimPrimaryGeneratorMessenger.hh b/libWCSIM/include/WCSimPrimaryGeneratorMessenger.hh deleted file mode 100644 index 1c38bba..0000000 --- a/libWCSIM/include/WCSimPrimaryGeneratorMessenger.hh +++ /dev/null @@ -1,40 +0,0 @@ -#ifndef WCSimPrimaryGeneratorMessenger_h -#define WCSimPrimaryGeneratorMessenger_h 1 - -class WCSimPrimaryGeneratorAction; -class G4UIdirectory; -class G4UIcmdWithAString; -class G4UIcmdWithABool; -class G4UIcmdWithADouble; - -#include "G4UImessenger.hh" -#include "globals.hh" - -class WCSimPrimaryGeneratorMessenger: public G4UImessenger -{ - public: - WCSimPrimaryGeneratorMessenger(WCSimPrimaryGeneratorAction* mpga); - ~WCSimPrimaryGeneratorMessenger(); - - public: - void SetNewValue(G4UIcommand* command, G4String newValues); - G4String GetCurrentValue(G4UIcommand* command); - - private: - WCSimPrimaryGeneratorAction* myAction; - // Boolean to determine whether the neutrino generator has been set yet - // Need to know whether we expect to open a ROOT file or a text file - bool genSet; - - private: //commands - G4UIdirectory* mydetDirectory; - G4UIcmdWithAString* genCmd; - G4UIcmdWithAString* fileNameCmd; - G4UIcmdWithABool* poisCmd; - G4UIcmdWithADouble* poisMeanCmd; - -}; - -#endif - - diff --git a/libWCSIM/include/WCSimRandomMessenger.hh b/libWCSIM/include/WCSimRandomMessenger.hh deleted file mode 100644 index 804c0a2..0000000 --- a/libWCSIM/include/WCSimRandomMessenger.hh +++ /dev/null @@ -1,28 +0,0 @@ -#ifndef WCSimRandomMessenger_h -#define WCSimRandomMessenger_h 1 - -#include "G4UImessenger.hh" -class WCSimRandomParameters; -class G4UIdirectory; -class G4UIcommand; -class G4UIcmdWithAString; -class G4UIcmdWithAnInteger; - -class WCSimRandomMessenger: public G4UImessenger -{ -public: - WCSimRandomMessenger(WCSimRandomParameters*); - ~WCSimRandomMessenger(); - - void SetNewValue(G4UIcommand* command, G4String newValue); - -private: - - WCSimRandomParameters* WCSimRandomParams; - - G4UIdirectory* WCSimDir; - G4UIcmdWithAString* Rangen; - G4UIcmdWithAnInteger* Ranseed; -}; - -#endif diff --git a/libWCSIM/include/WCSimRandomParameters.hh b/libWCSIM/include/WCSimRandomParameters.hh deleted file mode 100644 index 8d0ed62..0000000 --- a/libWCSIM/include/WCSimRandomParameters.hh +++ /dev/null @@ -1,76 +0,0 @@ -#ifndef WCSimRandomParameters_h -#define WCSimRandomParameters_h 1 -#include "WCSimRandomMessenger.hh" -#include "WCSimRootOptions.hh" -#include "CLHEP/Random/Random.h" -#include "CLHEP/Random/RanluxEngine.h" -#include "CLHEP/Random/JamesRandom.h" -#include "CLHEP/Random/RanecuEngine.h" - -class WCSimRandomParameters -{ - -public: - WCSimRandomParameters() - { - seed=31415; - generator=RANDOM_E_HEPJAMES; - RandomMessenger = new WCSimRandomMessenger(this); - } - ~WCSimRandomParameters() {delete RandomMessenger;} - - WCSimRandomGenerator_t GetGenerator() {return generator; } - void SetGenerator(WCSimRandomGenerator_t rng) - { - switch (rng) - { - case RANDOM_E_RANLUX: - { - printf("Setting the random number generator to RANLUX\n"); - CLHEP::RanluxEngine *newluxengine = new CLHEP::RanluxEngine(31415,4); // highest luxury level - CLHEP::HepRandom::setTheEngine(newluxengine); - } - break; - case RANDOM_E_RANECU: - { - printf("Setting the random number generator to RANECU\n"); - CLHEP::RanecuEngine *newecuengine = new CLHEP::RanecuEngine(); - CLHEP::HepRandom::setTheEngine(newecuengine); - } - break; - - case RANDOM_E_HEPJAMES: - { - printf("Setting the random number generator to HEPJAMES\n"); - CLHEP::HepJamesRandom *newjamesengine = new CLHEP::HepJamesRandom(); - CLHEP::HepRandom::setTheEngine(newjamesengine); - } - break; - default: - { - printf("Random number generator type not understood: %d\n",rng); - exit(0); - } - } - }; - int GetSeed() {return CLHEP::HepRandom::getTheSeed();} - void SetSeed(int iseed) - { - CLHEP::HepRandom::setTheSeed(iseed); - printf("Setting the Random Seed to: %d\n",iseed); - seed = iseed; - } - - void SaveOptionsToOutput(WCSimRootOptions * wcopt) - { - wcopt->SetRandomSeed(seed); - wcopt->SetRandomGenerator(generator); - } - -private: - WCSimRandomGenerator_t generator; - int seed; - WCSimRandomMessenger *RandomMessenger; -}; - -#endif diff --git a/libWCSIM/include/WCSimRootEvent.hh b/libWCSIM/include/WCSimRootEvent.hh deleted file mode 100644 index b7df904..0000000 --- a/libWCSIM/include/WCSimRootEvent.hh +++ /dev/null @@ -1,485 +0,0 @@ -#ifndef WCSim_RootEvent -#define WCSim_RootEvent - -////////////////////////////////////////////////////////////////////////// -// // -// WCSim_RootEvent // -// // -////////////////////////////////////////////////////////////////////////// - -#include "TObject.h" -#include "TClonesArray.h" -#include -#include -//#include -//#include "G4Transform3D.hh" - -// #include "WCSimDetectorConstruction.hh" -#include "WCSimEnumerations.hh" - -class TDirectory; - - -class WCSimRootTrack : public TObject { - -private: - - // See jhfNtuple.h for the meaning of these data members: - Int_t fIpnu; - Int_t fFlag; - Float_t fM; - Float_t fP; - Float_t fE; - Int_t fStartvol; - Int_t fStopvol; - Float_t fDir[3]; - Float_t fPdir[3]; - Float_t fStop[3]; - Float_t fStart[3]; - Int_t fParenttype; - Float_t fTime; - Int_t fId; - -public: - WCSimRootTrack() {} - WCSimRootTrack(Int_t ipnu, - Int_t flag, - Float_t m, - Float_t p, - Float_t E, - Int_t startvol, - Int_t stopvol, - Float_t dir[3], - Float_t pdir[3], - Float_t stop[3], - Float_t start[3], - Int_t parenttype, - Float_t time,Int_t id); - - virtual ~WCSimRootTrack() { } - - Int_t GetIpnu() const { return fIpnu;} - Int_t GetFlag() const { return fFlag;} - Float_t GetM() const { return fM;} - Float_t GetP() const { return fP;} - Float_t GetE() const { return fE;} - Int_t GetStartvol() { return fStartvol;} - Int_t GetStopvol() { return fStopvol;} - Float_t GetDir(Int_t i=0) {return (i<3) ? fDir[i] : 0;} - Float_t GetPdir(Int_t i=0) {return (i<3) ? fPdir[i] : 0;} - Float_t GetStop(Int_t i=0) {return (i<3) ? fStop[i] : 0;} - Float_t GetStart(Int_t i=0) {return (i<3) ? fStart[i] : 0;} - Int_t GetParenttype(/*Int_t i=0*/) {return fParenttype;} - Float_t GetTime() { return fTime;} - Int_t GetId(){return fId;} - - ClassDef(WCSimRootTrack,1) -}; - - -////////////////////////////////////////////////////////////////////////// - - -class WCSimRootCherenkovHit : public TObject { - -private: - Int_t fTubeID; - Int_t fmPMTID; - Int_t fmPMT_PMTID; - Int_t fTotalPe[2]; - -public: - WCSimRootCherenkovHit() {} - WCSimRootCherenkovHit(Int_t tubeID, - Int_t totalPe[2]); - - WCSimRootCherenkovHit(Int_t tubeID, - Int_t mPMTID, - Int_t mPMT_PMTID, - Int_t totalPe[2]); - - virtual ~WCSimRootCherenkovHit() { } - - Int_t GetTubeID() const { return fTubeID;} - Int_t GetmPMTID() const { return fmPMTID;} - Int_t GetmPMT_PMTID() const { return fmPMT_PMTID;} - Int_t GetTotalPe(int i) const { return (i<2) ? fTotalPe[i]: 0;} - - ClassDef(WCSimRootCherenkovHit,2) -}; - -class WCSimRootCherenkovHitTime : public TObject { - -private: - // See jhfNtuple.h for the meaning of these data members: - Float_t fTruetime; - Int_t fPrimaryParentID; - -public: - WCSimRootCherenkovHitTime() {} - WCSimRootCherenkovHitTime(Float_t truetime, - Int_t primaryParentID); - virtual ~WCSimRootCherenkovHitTime() { } - - Float_t GetTruetime() { return fTruetime;} - Int_t GetParentID() { return fPrimaryParentID;} - - ClassDef(WCSimRootCherenkovHitTime,1) -}; - - -////////////////////////////////////////////////////////////////////////// - - -class WCSimRootCherenkovDigiHit : public TObject { - -private: - // See jhfNtuple.h for the meaning of these data members: - Float_t fQ; - Float_t fT; - Int_t fTubeId; - Int_t fmPMTId; - Int_t fmPMT_PMTId; - std::vector fPhotonIds; - -public: - WCSimRootCherenkovDigiHit() {} - WCSimRootCherenkovDigiHit(Float_t q, Float_t t, Int_t tubeid, std::vector photon_ids); - WCSimRootCherenkovDigiHit(Float_t q, Float_t t, Int_t tubeid, Int_t mpmtid, Int_t mpmt_pmtid, std::vector photon_ids); - - virtual ~WCSimRootCherenkovDigiHit() { } - - Float_t GetQ() const { return fQ;} - Float_t GetT() const { return fT;} - Int_t GetTubeId() const { return fTubeId;} - Int_t GetmPMTId() const { return fmPMTId;} - Int_t GetmPMT_PMTId() const { return fmPMT_PMTId;} - std::vector GetPhotonIds() const { return fPhotonIds; } - - ClassDef(WCSimRootCherenkovDigiHit,3) -}; - - -////////////////////////////////////////////////////////////////////////// - -class WCSimRootEventHeader { - -private: - Int_t fEvtNum; - Int_t fRun; - Int_t fDate; - Int_t fSubEvtNumber; - -public: - WCSimRootEventHeader() : fEvtNum(0), fRun(0), fDate(0), fSubEvtNumber(1) { } - virtual ~WCSimRootEventHeader() { } - void Set(Int_t i, Int_t r, Int_t d, Int_t s=1) { fEvtNum = i; fRun = r; fDate = d; fSubEvtNumber = s;} - void SetDate(Int_t d) { fDate=d; } - Int_t GetEvtNum() const { return fEvtNum; } - Int_t GetRun() const { return fRun; } - Int_t GetDate() const { return fDate; } - Int_t GetSubEvtNumber() const { return fSubEvtNumber;} - - - ClassDef(WCSimRootEventHeader,2) //WCSimRootEvent Header -}; - -////////////////////////////////////////////////////////////////////////// - -class WCSimRootPi0 : public TObject { - // this is a class used specifically for Pi0 events - -private: - Float_t fPi0Vtx[3]; - Int_t fGammaID[2]; - Float_t fGammaE[2]; - Float_t fGammaVtx[2][3]; - -public: - WCSimRootPi0() {} - - virtual ~WCSimRootPi0() {} - - void Set(Float_t pi0Vtx[3], - Int_t gammaID[2], - Float_t gammaE[2], - Float_t gammaVtx[2][3]); - - Float_t GetPi0Vtx(int i) const { return (i<3) ? fPi0Vtx[i]: 0;} - Int_t GetGammaID(int i) const { return (i<2) ? fGammaID[i]: 0;} - Float_t GetGammaE(int i) const { return (i<2) ? fGammaE[i]: 0;} - Float_t GetGammaVtx(int i, int j) const { return fGammaVtx[i][j];} - -ClassDef(WCSimRootPi0,1) -}; - -////////////////////////////////////////////////////////////////////////// - -class WCSimRootCaptureGamma : public TObject { - -private: - Int_t fID; - Float_t fEnergy; - Float_t fDir[3]; - -public: - WCSimRootCaptureGamma() {} - WCSimRootCaptureGamma(Int_t id, - Float_t energy, - Float_t dir[3] - ); - - virtual ~WCSimRootCaptureGamma() {} - - Int_t GetID() const { return fID;} - Float_t GetE() const { return fEnergy;} - Float_t GetDir(int i) const { return (i<3) ? fDir[i]: 0;} - -ClassDef(WCSimRootCaptureGamma,1) -}; - -////////////////////////////////////////////////////////////////////////// - -class WCSimRootCapture : public TObject { - // this is a class used specifically for neutron capture events - -private: - Int_t fCaptureParent; - Float_t fCaptureVtx[3]; - Int_t fNGamma; - Float_t fTotalGammaE; - Float_t fCaptureT; - Int_t fCaptureNucleus; - TClonesArray * fGammas; - bool IsZombie; - -public: - WCSimRootCapture() { - fGammas = 0; - IsZombie = true; - } - WCSimRootCapture(Int_t captureParent); - - virtual ~WCSimRootCapture(); - - void SetInfo(Float_t captureVtx[3], - Float_t captureT, - Int_t captureNucleus - ); - - void AddGamma(Int_t gammaID, - Float_t gammaE, - Float_t gammaDir[3] - ); - - Int_t GetCaptureParent() const { return fCaptureParent;} - Float_t GetCaptureVtx(int i) const { return (i<3) ? fCaptureVtx[i]: 0;} - Int_t GetNGamma() const { return fNGamma;} - Float_t GetTotalGammaE() const { return fTotalGammaE;} - Float_t GetCaptureT() const { return fCaptureT;} - Int_t GetCaptureNucleus() const { return fCaptureNucleus;} - TClonesArray *GetGammas() const { return fGammas;} - -ClassDef(WCSimRootCapture,1) -}; - -////////////////////////////////////////////////////////////////////////// - -class WCSimRootTrigger : public TObject { - -private: - WCSimRootEventHeader fEvtHdr; // The header - // See jhfNtuple.h for the meaning of these data members: - Int_t fMode; - Int_t fVtxvol; - Float_t fVtx[3]; - Int_t fVecRecNumber; // "info event" number in inputvectorfile - Int_t fJmu; - Int_t fJp; - - WCSimRootPi0 fPi0; // Pi0 info (default = not used) - - TClonesArray *fCaptures; // Neutron capture info (default = not used) - Int_t fNcaptures; // Number of tracks in the array - - Int_t fNpar; // Number of particles - Int_t fNtrack; // Number of tracks in the array - TClonesArray *fTracks; //-> Array of WCSimRootTracks - - Int_t fNumTubesHit; // Number of tubes hit - Int_t fNcherenkovhits; // Number of hits in the array - TClonesArray *fCherenkovHits; //-> Array of WCSimRootCherenkovHits - - Int_t fCherenkovHitCounter; - Int_t fNcherenkovhittimes; // Number of hits in the array - TClonesArray *fCherenkovHitTimes; //-> Array of WCSimRootCherenkovHits - - Int_t fNumDigitizedTubes; // Number of digitized tubes - Int_t fNcherenkovdigihits; // Number of digihits in the array - Float_t fSumQ; - TClonesArray *fCherenkovDigiHits; //-> Array of WCSimRootCherenkovDigiHit's - - TriggerType_t fTriggerType; // Trigger algorithm that created this trigger - std::vector fTriggerInfo; // Information about how it passed the trigger (e.g. how many hits in the NDigits window) - - bool IsZombie; - -public: - WCSimRootTrigger(); - WCSimRootTrigger(int, int); - virtual ~WCSimRootTrigger(); - - void Initialize(); - - void Clear(Option_t *option =""); - static void Reset(Option_t *option =""); - - void SetHeader(Int_t i, Int_t run, Int_t date,Int_t subevtn=1); - void SetTriggerInfo(TriggerType_t trigger_type, std::vector trigger_info); - bool IsASubEvent() { return (fEvtHdr.GetSubEvtNumber()>=1); } - void SetMode(Int_t i) {fMode = i;} - void SetVtxvol(Int_t i) {fVtxvol = i;} - void SetVtx(Int_t i, Float_t f) {fVtx[i]= ( (i<3) ? f : 0);} - void SetVecRecNumber(Int_t i) {fVecRecNumber = i;} - void SetJmu(Int_t i) {fJmu = i;} - void SetJp(Int_t i) {fJp = i;} - void SetNpar(Int_t i) {fNpar = i;} - void SetNumTubesHit(Int_t i) {fNumTubesHit = i;} - void SetSumQ(Float_t x){fSumQ = x;} - void SetNumDigitizedTubes(Int_t i) {fNumDigitizedTubes = i;} - void SetPi0Info(Float_t pi0Vtx[3], - Int_t gammaID[2], - Float_t gammaE[2], - Float_t gammaVtx[2][3]); - void SetCaptureParticle(Int_t parent, - Int_t ipnu, - Float_t time, - Float_t vtx[3], - Float_t dir[3], - Float_t energy, - Int_t id); - - - WCSimRootEventHeader *GetHeader() {return &fEvtHdr; } - WCSimRootPi0 *GetPi0Info() {return &fPi0; } - Int_t GetMode() const {return fMode;} - Int_t GetVtxvol() const {return fVtxvol;} - Float_t GetVtx(Int_t i=0) {return (i<3) ? fVtx[i]: 0;} - Int_t GetVecRecNumber() const {return fVecRecNumber;} - Int_t GetJmu() const {return fJmu;} - Int_t GetJp() const {return fJp;} - Int_t GetNpar() const {return fNpar;} - Int_t GetNumTubesHit() const {return fNumTubesHit;} - Int_t GetNumDigiTubesHit() const {return fNumDigitizedTubes;} - Int_t GetNtrack() const {return fNtrack; } - Int_t GetNcaptures() const {return fNcaptures; } - Int_t GetNcherenkovhits() const {return fNcherenkovhits; } - Int_t GetNcherenkovhittimes() const {return fNcherenkovhittimes;} - Int_t GetNcherenkovdigihits() const {return fNcherenkovdigihits;} - Float_t GetSumQ() const { return fSumQ;} - TriggerType_t GetTriggerType() const { return fTriggerType;} - std::vector GetTriggerInfo() const { return fTriggerInfo;} - - WCSimRootTrack *AddTrack(Int_t ipnu, - Int_t flag, - Float_t m, - Float_t p, - Float_t E, - Int_t startvol, - Int_t stopvol, - Float_t dir[3], - Float_t pdir[3], - Float_t stop[3], - Float_t start[3], - Int_t parenttype, - Float_t time, - Int_t id); - - TClonesArray *GetTracks() const {return fTracks;} - - WCSimRootCherenkovHit *AddCherenkovHit(Int_t tubeID, - Int_t mPMTID, - Int_t mPMT_PMTID, - std::vector truetime, - std::vector primParID); - TClonesArray *GetCherenkovHits() const {return fCherenkovHits;} - TClonesArray *GetCherenkovHitTimes() const {return fCherenkovHitTimes;} - - WCSimRootCherenkovDigiHit *AddCherenkovDigiHit(Float_t q, - Float_t t, - Int_t tubeid, - Int_t mpmtid, - Int_t mpmt_pmtid, - std::vector photon_ids); -// WCSimRootCherenkovDigiHit *AddCherenkovDigiHit(Float_t q, -// Float_t t, -// Int_t tubeid, - // Float_t sumq); - - TClonesArray *GetCherenkovDigiHits() const {return fCherenkovDigiHits;} - - TClonesArray *GetCaptures() const {return fCaptures;} - - -ClassDef(WCSimRootTrigger,2) //WCSimRootEvent structure -}; - - -class WCSimRootEvent : public TObject { -public: - WCSimRootEvent(); - virtual ~WCSimRootEvent(); - - void Clear(Option_t *option =""); - static void Reset(Option_t *option =""); - Int_t GetCurrentIndex() { return Current;} - - // WCSimRootTrigger* GetTrigger(int number) { return fEventList[number];} - WCSimRootTrigger* GetTrigger(int number) { return (WCSimRootTrigger*) (*fEventList)[number];} - - Int_t GetNumberOfEvents() const { return fEventList->GetEntriesFast();} - Int_t GetNumberOfSubEvents() const { return (fEventList->GetEntriesFast()-1);} - bool HasSubEvents() { return (fEventList->GetEntriesFast() > 1); } - - //Int_t GetNumberOfEvents() const { return fEventList.size();} - //Int_t GetNumberOfSubEvents() const { return (fEventList.size()-1);} - - //void AddSubEvent() { fEventList.push_back(new WCSimRootTrigger()); } - void AddSubEvent() { - // be sure not to call the default constructor BUT the actual one - WCSimRootTrigger* tmp = dynamic_cast( (*fEventList)[0] ); - int num = tmp->GetHeader()->GetEvtNum(); - ++Current; - if ( Current > 9 ) fEventList->Expand(150); - fEventList->AddAt(new WCSimRootTrigger(num,Current),Current); - } - - /* void ReInitialize() { // need to remove all subevents at the end, or they just get added anyway... - std::vector::iterator iter = fEventList.begin(); - ++iter; // do not delete the first event --> regular beaviour for this program ? - */ - void Initialize(); - - void ReInitialize() { // need to remove all subevents at the end, or they just get added anyway... - for ( int i = fEventList->GetLast() ; i>=1 ; i--) { - // G4cout << "removing element # " << i << "..."; - WCSimRootTrigger* tmp = - dynamic_cast(fEventList->RemoveAt(i)); - delete tmp; - //G4cout <<"done !\n"; - } - Current = 0; - WCSimRootTrigger* tmp = dynamic_cast( (*fEventList)[0]); - tmp->Clear(); - } - -private: - //std::vector fEventList; - TObjArray* fEventList; - Int_t Current; //! means transient, not writable to file - ClassDef(WCSimRootEvent,2) - -}; - - -#endif diff --git a/libWCSIM/include/WCSimRootGeom.hh b/libWCSIM/include/WCSimRootGeom.hh deleted file mode 100644 index 3a9902a..0000000 --- a/libWCSIM/include/WCSimRootGeom.hh +++ /dev/null @@ -1,132 +0,0 @@ -#ifndef WCSim_RootGeom -#define WCSim_RootGeom - -////////////////////////////////////////////////////////////////////////// -// -// WCSim_RootGeom -// -// This class contains information needed to be passed to reconstruction -// routines. It's just simple right now-- only the bare-bones -// WC info -////////////////////////////////////////////////////////////////////////// - -#include "TObject.h" -#include "TClonesArray.h" - -class TDirectory; - -////////////////////////////////////////////////////////////////////////// - -class WCSimRootPMT : public TObject { - -private: - Int_t fTubeNo; - Int_t fmPMTNo; - Int_t fmPMT_PMTNo; - Int_t fCylLoc; // endcap1, wall, endcap2 - Float_t fOrientation[3]; - Float_t fPosition[3]; - -public: - WCSimRootPMT(); - WCSimRootPMT(Int_t tubeNo, Int_t cylLoc, Float_t orientation[3], Float_t position[3]); - WCSimRootPMT(Int_t tubeNo, Int_t mPMTNo, Int_t mPMT_PMTno, Int_t cylLoc, Float_t orientation[3], Float_t position[3]); - virtual ~WCSimRootPMT(); - - void SetTubeNo(Int_t i) {fTubeNo=i;} - void SetmPMTNo(Int_t i) {fmPMTNo=i;} - void SetmPMT_PMTNo(Int_t i) {fmPMT_PMTNo=i;} - void SetCylLoc(Int_t i) {fCylLoc=i;} - void SetOrientation(Int_t i, Float_t f) {fOrientation[i]= ( (i<3) ? f : 0);} - void SetPosition(Int_t i, Float_t f) {fPosition[i]= ( (i<3) ? f : 0);} - - Int_t GetTubeNo() const {return fTubeNo;} - Int_t GetmPMTNo() const {return fmPMTNo;} - Int_t GetmPMT_PMTNo() const {return fmPMT_PMTNo;} - Int_t GetCylLoc() const {return fCylLoc;} - Float_t GetOrientation(Int_t i=0) {return (i<3) ? fOrientation[i] : 0;} - Float_t GetPosition(Int_t i=0) {return (i<3) ? fPosition[i] : 0;} - - ClassDef(WCSimRootPMT,2) //WCSimPMT structure -}; - - -////////////////////////////////////////////////////////////////////////// - -class WCSimRootGeom : public TObject { - -private: - - static const Int_t maxNumPMT = 1000000; - //B.Qstatic const Int_t maxNumPMT = 40000; - Float_t fWCCylRadius; // Radius of WC tank - Float_t fWCCylLength; // Length of WC tank - - Int_t fgeo_type; // mailbox or cylinder? - - Float_t fWCPMTRadius; // Radius of PMT - Float_t fWCPMTRadius2; // Radius of PMT, hybrid case - Int_t fWCNumPMT; // Number of PMTs - Int_t fWCNumPMT2; // Number of PMTs, hybrid case - Float_t fWCOffset[3]; // Offset of barrel center in global coords - - Int_t fOrientation; //Orientation o detector, 0 is 2km horizontal, 1 is Upright - - // Could make a TClonesArray of PMTs but let's keep it simple - // since the arrays just won't be that large - //WCSimRootPMT fPMTArray[maxNumPMT]; // Array of PMTs - TClonesArray *fPMTArray; - TClonesArray *fPMTArray2; - -public: - - WCSimRootGeom(); - virtual ~WCSimRootGeom(); - - // Sets and gets - - void SetWCCylRadius(Float_t f) {fWCCylRadius=f;} - void SetWCCylLength(Float_t f) {fWCCylLength=f;} - - void SetGeo_Type(Int_t f){fgeo_type = f;} - - void SetWCNumPMT(Int_t i,bool hybridsecondtype=false){ - if(hybridsecondtype) fWCNumPMT2 = i; - else fWCNumPMT= i; - } - void SetWCPMTRadius(Float_t f,int hybridsecondtype=false) {(hybridsecondtype?fWCPMTRadius2=f:fWCPMTRadius=f);} - void SetWCOffset(Float_t x, Float_t y, Float_t z) - {fWCOffset[0]=x; fWCOffset[1]=y; fWCOffset[2] = z;} - - void SetPMT(Int_t i, Int_t tubeno, Int_t cyl_loc, Float_t rot[3], Float_t pos[3], bool expand=true, bool hybridsecondtype=false); - void SetPMT(Int_t i, Int_t tubeno, Int_t mPMTNo, Int_t mPMT_PMTno, Int_t cyl_loc, Float_t rot[3], Float_t pos[3], bool expand=true, bool hybridsecondtype=false); - void SetOrientation(Int_t o) {fOrientation = o;} - - Float_t GetWCCylRadius() const {return fWCCylRadius;} - Float_t GetWCCylLength() const {return fWCCylLength;} - - Int_t GetGeo_Type() const {return fgeo_type;} - - - Int_t GetWCNumPMT(bool hybridsecondtype=false){ - if(hybridsecondtype) return fWCNumPMT2; - else return fWCNumPMT; - } - Float_t GetWCPMTRadius(bool hybridsecondtype=false) const { - if(hybridsecondtype) return fWCPMTRadius2; - else return fWCPMTRadius; - } - Float_t GetWCOffset(Int_t i) const {return (i<3) ? fWCOffset[i] : 0.;} - - Int_t GetOrientation() { return fOrientation; } - //WCSimRootPMT GetPMT(Int_t i){return *(new WCSimRootPMT());} - WCSimRootPMT GetPMT(Int_t i,bool hybridsecondtype=false){ - if(hybridsecondtype) return *(WCSimRootPMT*)(*fPMTArray2)[i]; - else return *(WCSimRootPMT*)(*fPMTArray)[i]; - } - - ClassDef(WCSimRootGeom,1) //WCSimRootEvent structure -}; - - -#endif diff --git a/libWCSIM/include/WCSimRootLinkDef.hh b/libWCSIM/include/WCSimRootLinkDef.hh deleted file mode 100644 index 0e4f774..0000000 --- a/libWCSIM/include/WCSimRootLinkDef.hh +++ /dev/null @@ -1,29 +0,0 @@ -#ifdef __CINT__ - -#pragma link off all globals; -#pragma link off all classes; -#pragma link off all functions; - -#pragma link C++ enum TriggerType_t; -#pragma link C++ enum DigitizerType_t; - -#pragma link C++ class WCSimRootCherenkovDigiHit+; -#pragma link C++ class WCSimRootCherenkovHit+; -#pragma link C++ class WCSimRootCherenkovHitTime+; -#pragma link C++ class WCSimRootTrack+; -#pragma link C++ class WCSimRootEventHeader+; -#pragma link C++ class WCSimRootTrigger+; -#pragma link C++ class WCSimRootEvent+; -#pragma link C++ class WCSimRootPi0+; -#pragma link C++ class WCSimRootCapture+; -#pragma link C++ class WCSimRootCaptureGamma+; -#pragma link C++ class WCSimRootGeom+; -#pragma link C++ class WCSimRootPMT+; -#pragma link C++ class WCSimPmtInfo+; -#pragma link C++ class WCSimEnumerations+; -#pragma link C++ class WCSimRootOptions+; -#pragma link C++ class RooTrackerVtxBase+; -#pragma link C++ class JNuBeamFlux+; -#pragma link C++ class NRooTrackerVtx+; - -#endif diff --git a/libWCSIM/include/WCSimRootOptions.hh b/libWCSIM/include/WCSimRootOptions.hh deleted file mode 100644 index ff0e9e1..0000000 --- a/libWCSIM/include/WCSimRootOptions.hh +++ /dev/null @@ -1,219 +0,0 @@ -#ifndef WCSim_RootOptions -#define WCSim_RootOptions - -////////////////////////////////////////////////////////////////////////// -// -// WCSim_RootOptions -// -// This class contains run option information -////////////////////////////////////////////////////////////////////////// - -#include "TObject.h" -#include "TClonesArray.h" -#include -#include -#include - -#include "WCSimEnumerations.hh" - -class TDirectory; -using std::string; -using std::map; - -////////////////////////////////////////////////////////////////////////// - -struct WCSimDarkNoiseOptions { - double PMTDarkRate; // kHz - double ConvRate; // kHz - double DarkHigh; // ns - double DarkLow; // ns - double DarkWindow; // ns - int DarkMode; - WCSimDarkNoiseOptions() : - PMTDarkRate(-999), ConvRate(-999), DarkHigh(-999), DarkLow(-999), - DarkWindow(-999), DarkMode(-999) - {} -}; - -////////////////////////////////////////////////////////////////////////// - -class WCSimRootOptions : public TObject { - -public: - - WCSimRootOptions(); - virtual ~WCSimRootOptions(); - void Print(Option_t *option = "") const; - - //WCSimDetector* gets - void SetDetectorName(string iDetectorName) {DetectorName = iDetectorName;} - void SetGeomHasOD(bool iGeomHasOD) {GeomHasOD = iGeomHasOD;} - void SetSavePi0(bool iSavePi0) {SavePi0 = iSavePi0;} - void SetPMTQEMethod(bool iPMTQEMethod) {PMTQEMethod = iPMTQEMethod;} - void SetPMTCollEff(bool iPMTCollEff) {PMTCollEff = iPMTCollEff;} - //WCSimDetector* sets - string GetDetectorName() {return DetectorName;} - bool GetGeomHasOD() {return GeomHasOD;} - bool GetSavePi0() {return SavePi0;} - int GetPMTQEMethod() {return PMTQEMethod;} - int GetPMTCollEff() {return PMTCollEff;} - //WCSimWCAddDarkNoise sets - void SetPMTDarkRate(string tag, double iPMTDarkRate) {DarkOptMap[tag].PMTDarkRate = iPMTDarkRate;} - void SetConvRate(string tag, double iConvRate) {DarkOptMap[tag].ConvRate = iConvRate;} - void SetDarkHigh(string tag, double iDarkHigh) {DarkOptMap[tag].DarkHigh = iDarkHigh;} - void SetDarkLow(string tag, double iDarkLow) {DarkOptMap[tag].DarkLow = iDarkLow;} - void SetDarkWindow(string tag, double iDarkWindow) {DarkOptMap[tag].DarkWindow = iDarkWindow;} - void SetDarkMode(string tag, int iDarkMode) {DarkOptMap[tag].DarkMode = iDarkMode;} - //WCSimWCAddDarkNoise gets - bool IsValidDarkTag(string tag) const; - double GetPMTDarkRate(string tag); - double GetConvRate(string tag); - double GetDarkHigh(string tag); - double GetDarkLow(string tag); - double GetDarkWindow(string tag); - int GetDarkMode(string tag); - //WCSimWCDigitizer* sets - void SetDigitizerClassName(string iDigitizerClassName) {DigitizerClassName = iDigitizerClassName;} - void SetDigitizerDeadTime(int iDigitizerDeadTime) {DigitizerDeadTime = iDigitizerDeadTime;} - void SetDigitizerIntegrationWindow(int iDigitizerIntegrationWindow) {DigitizerIntegrationWindow = iDigitizerIntegrationWindow;} - void SetDigitizerTimingPrecision(double iDigitizerTimingPrecision) {DigitizerTimingPrecision = iDigitizerTimingPrecision;} - void SetDigitizerPEPrecision(double iDigitizerPEPrecision) {DigitizerPEPrecision = iDigitizerPEPrecision;} - //WCSimWCDigitizer* gets - string GetDigitizerClassName() {return DigitizerClassName;} - int GetDigitizerDeadTime() {return DigitizerDeadTime;} - int GetDigitizerIntegrationWindow() {return DigitizerIntegrationWindow;} - int GetDigitizerTimingPrecision() {return DigitizerTimingPrecision;} - int GetDigitizerPEPrecision() {return DigitizerPEPrecision;} - //WCSimWCTrigger* sets - void SetTriggerClassName(string itriggerClassName) {TriggerClassName = itriggerClassName;}; - void SetMultiDigitsPerTrigger(bool imultiDigitsPerTrigger) {MultiDigitsPerTrigger = imultiDigitsPerTrigger;}; - //ndigits - void SetNDigitsThreshold(int indigitsThreshold) {NDigitsThreshold = indigitsThreshold;}; - void SetNDigitsWindow(int indigitsWindow) {NDigitsWindow = indigitsWindow;}; - void SetNDigitsAdjustForNoise(bool indigitsAdjustForNoise) {NDigitsAdjustForNoise = indigitsAdjustForNoise;}; - void SetNDigitsPreTriggerWindow(int indigitsPreTriggerWindow) {NDigitsPreTriggerWindow = indigitsPreTriggerWindow;}; - void SetNDigitsPostTriggerWindow(int indigitsPostTriggerWindow) {NDigitsPostTriggerWindow = indigitsPostTriggerWindow;}; - //savefailures - void SetSaveFailuresMode(int isaveFailuresMode) {SaveFailuresMode = isaveFailuresMode;}; - void SetSaveFailuresTime(double isaveFailuresTime) {SaveFailuresTime = isaveFailuresTime;}; - void SetSaveFailuresPreTriggerWindow(int isaveFailuresPreTriggerWindow) {SaveFailuresPreTriggerWindow = isaveFailuresPreTriggerWindow;}; - void SetSaveFailuresPostTriggerWindow(int isaveFailuresPostTriggerWindow) {SaveFailuresPostTriggerWindow = isaveFailuresPostTriggerWindow;}; - //WCSimWCTrigger* gets - string GetTriggerClassName() {return TriggerClassName;} - bool GetMultiDigitsPerTrigger() {return MultiDigitsPerTrigger;} - //ndigits - int GetNDigitsThreshold() {return NDigitsThreshold;} - int GetNDigitsWindow() {return NDigitsWindow;} - bool GetNDigitsAdjustForNoise() {return NDigitsAdjustForNoise;} - int GetNDigitsPreTriggerWindow() {return NDigitsPreTriggerWindow;} - int GetNDigitsPostTriggerWindow() {return NDigitsPostTriggerWindow;} - //savefailures - int GetSaveFailuresMode() {return SaveFailuresMode;} - double GetSaveFailuresTime() {return SaveFailuresTime;} - int GetSaveFailuresPreTriggerWindow() {return SaveFailuresPreTriggerWindow;} - int GetSaveFailuresPostTriggerWindow() {return SaveFailuresPostTriggerWindow;} - //WCSimTuningParameters sets - void SetRayff(double iRayff) {Rayff = iRayff;} - void SetBsrff(double iBsrff) {Bsrff = iBsrff;} - void SetAbwff(double iAbwff) {Abwff = iAbwff;} - void SetRgcff(double iRgcff) {Rgcff = iRgcff;} - void SetMieff(double iMieff) {Mieff = iMieff;} - void SetQeff(double iQeff) {Qeff = iQeff;}//B.Q 2018/07/25 - void SetTtsff(double iTtsff) {Ttsff = iTtsff;}//TD 2019.06.22 - void SetPMTSatur(double iPmtSatur) {PmtSatur = iPmtSatur;}//TD 2019.07.16 - // void SetQoiff(double iQoiff) {Qoiff = iQoiff;}//TD 2019.06.26 - void SetTvspacing(double iTvspacing) {Tvspacing = iTvspacing;} - void SetTopveto(bool iTopveto) {Topveto = iTopveto;} - //WCSimTuningParameters gets - double GetRayff() {return Rayff;} - double GetBsrff() {return Bsrff;} - double GetAbwff() {return Abwff;} - double GetRgcff() {return Rgcff;} - double GetMieff() {return Mieff;} - double GetQeff() {return Qeff;} - double GetTtsff() {return Ttsff;} - double GetPMTSatur() {return PmtSatur;}//TD 2019.07.16 - // double GetQoiff() {return Qoiff;}//TD 2019.06.26 - double GetTvspacing() {return Tvspacing;} - bool GetTopveto() {return Topveto;} - //WCSimPhysicsListFactory sets - void SetPhysicsListName(string iPhysicsListName) {PhysicsListName = iPhysicsListName;} - //WCSimPhysicsListFactory gets - string GetPhysicsListName() {return PhysicsListName;} - //WCSimPrimaryGeneratorAction sets - void SetVectorFileName(string iVectorFileName) {VectorFileName = iVectorFileName;} - void SetGeneratorType(string iGeneratorType) {GeneratorType = iGeneratorType;} - //WCSimPrimaryGeneratorAction gets - string GetVectorFileName() {return VectorFileName;} - string GetGeneratorType() {return GeneratorType;} - //WCSimRandomParameters sets - void SetRandomSeed(int iRandomSeed) {RandomSeed = iRandomSeed;} - void SetRandomGenerator(WCSimRandomGenerator_t iRandomGenerator) {RandomGenerator = iRandomGenerator;} - //WCSimRandomParameters gets - int GetRandomSeed() {return RandomSeed;} - WCSimRandomGenerator_t GetRandomGenerator() {return RandomGenerator;} - -private: - //WCSimDetector* - string DetectorName; - bool GeomHasOD; - bool SavePi0; - int PMTQEMethod; - int PMTCollEff; - - //WCSimWCAddDarkNoise - map DarkOptMap; - - //WCSimWCDigitizer* - string DigitizerClassName; - int DigitizerDeadTime; // ns - int DigitizerIntegrationWindow; // ns - int DigitizerTimingPrecision; // - int DigitizerPEPrecision; // - - //WCSimWCTrigger* - string TriggerClassName; - bool MultiDigitsPerTrigger; - //ndigits - int NDigitsThreshold; // digitized hits - int NDigitsWindow; // ns - bool NDigitsAdjustForNoise; - int NDigitsPreTriggerWindow; // ns - int NDigitsPostTriggerWindow; // ns - //savefailures - int SaveFailuresMode; - double SaveFailuresTime; // ns - int SaveFailuresPreTriggerWindow; // ns - int SaveFailuresPostTriggerWindow; // ns - - //WCSimTuningParameters - double Rayff; - double Bsrff; - double Abwff; - double Rgcff; - double Qeff; - double Mieff; - double Ttsff; - // double Qoiff; //TD 2019.6.26 - double PmtSatur; //TD 2019.07.16 - double Tvspacing; - bool Topveto; - - //WCSimPhysicsListFactory - string PhysicsListName; - - //WCSimPrimaryGeneratorAction - string VectorFileName; - string GeneratorType; - - //WCSimRandomParameters - int RandomSeed; - WCSimRandomGenerator_t RandomGenerator; - - ClassDef(WCSimRootOptions,4) -}; - - -////////////////////////////////////////////////////////////////////////// - -#endif //WCSim_RootOptions diff --git a/libWCSIM/include/WCSimRunAction.hh b/libWCSIM/include/WCSimRunAction.hh deleted file mode 100644 index 3b8362d..0000000 --- a/libWCSIM/include/WCSimRunAction.hh +++ /dev/null @@ -1,232 +0,0 @@ -#ifndef WCSimRunAction_h -#define WCSimRunAction_h 1 - -#include "G4UserRunAction.hh" -#include "globals.hh" -#include "G4String.hh" - -#include "TFile.h" -#include "TTree.h" -#include "TClonesArray.h" -#include "WCSimRootEvent.hh" -#include "WCSimRootGeom.hh" -#include "WCSimRootOptions.hh" -#include "WCSimDetectorConstruction.hh" -#include "WCSimEnumerations.hh" //ToDo: move more Enums over there! -#include "evNtuple.h" -#include "WCSimRandomParameters.hh" - - -#include "TNRooTrackerVtx.hh" - -class G4Run; -class WCSimRunActionMessenger; - -class WCSimRunAction : public G4UserRunAction -{ -public: - WCSimRunAction(WCSimDetectorConstruction*, WCSimRandomParameters*); - ~WCSimRunAction(); - -public: - void BeginOfRunAction(const G4Run*); - void EndOfRunAction(const G4Run*); - void SetRootFileName(G4String fname) { RootFileName = fname; } - void SetSaveRooTracker(G4bool fsave) { SaveRooTracker = fsave; } - G4String GetRootFileName() { return RootFileName; } - void SetOptionalRootFile(G4bool choice) { useDefaultROOTout = choice; } - G4bool GetRootFileOption() { return useDefaultROOTout; } - bool GetSaveRooTracker() { return SaveRooTracker; } - void FillGeoTree(); - TTree* GetTree(){return WCSimTree;} - TBranch* GetBranch(G4String detectorElement = "tank"){ - if(detectorElement=="tank") return wcsimrooteventbranch; - else if(detectorElement=="tankPMT2") return wcsimrooteventbranch2; - else G4cout << "Unkown detector element" << G4endl; - } - TTree* GetFlatMasterTree(){return masterTree;} - TTree* GetGeoTree(){return geoTree;} - TTree* GetOptionsTree(){return optionsTree;} - WCSimRootGeom* GetRootGeom(){return wcsimrootgeom;} - WCSimRootEvent* GetRootEvent(G4String detectorElement = "tank"){ - if(detectorElement=="tank") return wcsimrootsuperevent; - if(detectorElement=="tankPMT2") return wcsimrootsuperevent2; - } - WCSimRootOptions* GetRootOptions(){return wcsimrootoptions;} - - //Need to share with EventAction: - TTree *GetCherenkovHitsTree(){return cherenkovHitsTree;} - TTree *GetCherenkovDigiHitsTree(){return cherenkovDigiHitsTree;} - TTree *GetTracksTree(){return tracksTree;} - TTree *GetTriggerTree(){return triggerTree;} - TTree *GetEventInfoTree(){return eventInfoTree;} - TTree *GetFlatRooTrackerTree(){return flatRooTrackerTree;} - - void SetEventHeaderNew(G4int run_id, G4int event_id, G4int subevent_id){ - event = event_id; - run = run_id; - subevent = subevent_id;} - - void SetTriggerInfoNew(TriggerType_t trigger_type,int trigger_info, /* std::vector trigger_info, */ - G4double trigger_length, G4double trigger_start){ - trig_type = trigger_type; - /* - for(size_t j = 0; j < trigger_info.size(); j++){ - if(j < 10) - trig_info[j] = trigger_info[j]; - else{ - G4cerr << "Trigger Info too large for array" << G4endl; - break; - } - }*/ - trig_info = trigger_info; - trig_length = trigger_length; - trig_start = trigger_start; - return; - } - - eventNtuple * GetMyStruct(){return evNtup;} - NRooTrackerVtx * GetMyRooTracker(){return evNRooTracker;} - - void SetTree(TTree* tree){WCSimTree=tree;} - void SetBranch(TBranch* branchin, G4String detectorElement = "tank"){ - if(detectorElement=="tank") wcsimrooteventbranch=branchin; - if(detectorElement=="tankPMT2") wcsimrooteventbranch2=branchin; - } - void SetGeoTree(TTree* tree){geoTree=tree;} - void SetRootEvent(WCSimRootEvent* revent, G4String detectorElement = "tank"){ - if(detectorElement=="tank") wcsimrootsuperevent=revent; - if(detectorElement=="tankPMT2") wcsimrootsuperevent2=revent; - } - void SetRootGeom(WCSimRootGeom* rgeom){wcsimrootgeom=rgeom;} - - int GetNumberOfEventsGenerated() { return numberOfEventsGenerated;} - int GetNtuples(){return ntuples;} - - void incrementEventsGenerated() { numberOfEventsGenerated++;} - void incrementWaterTubeHits() { numberOfTimesWaterTubeHit++;} - void incrementFVWaterTubeHits() { numberOfTimesFVWaterTubeHit++;} - void incrementCatcherHits() { numberOfTimesCatcherHit++;} - void SetNtuples(int ntup) {ntuples=ntup;} - - //New: - void FillFlatGeoTree(); - - NRooTrackerVtx* GetRootrackerVertex(); - void FillRootrackerVertexTree() { - fRooTrackerOutputTree->Fill(); - } - void ClearRootrackerVertexArray() { - fVertices->Clear(); - fNVtx = 0; - } - -private: - // MFechner : set by the messenger - std::string RootFileName; - // Only required for verification scripts and current fiTQun tuning - // But making initialization very slow due to large TCloneArray init. - G4bool useDefaultROOTout; - - // - TTree* WCSimTree; - TBranch* wcsimrooteventbranch; - TBranch* wcsimrooteventbranch2; - TTree* geoTree; - TTree* optionsTree; - WCSimRootEvent* wcsimrootsuperevent; - WCSimRootEvent* wcsimrootsuperevent2; - WCSimRootGeom* wcsimrootgeom; - WCSimRootOptions* wcsimrootoptions; - WCSimDetectorConstruction* wcsimdetector; - WCSimRandomParameters* wcsimrandomparameters; - - int numberOfEventsGenerated; - int numberOfTimesWaterTubeHit; - int numberOfTimesFVWaterTubeHit; - int numberOfTimesCatcherHit; - - TClonesArray* fVertices; - TTree* fRooTrackerOutputTree; - int fNVtx; - bool SaveRooTracker; - TTree* fSettingsOutputTree; - TTree* fSettingsInputTree; - - float WCXRotation[3]; - float WCYRotation[3]; - float WCZRotation[3]; - float WCDetCentre[3]; - float WCDetRadius; - float WCDetHeight; - float fNuPlanePos[3]; - float fNuPrismRadius; - - WCSimRunActionMessenger* messenger; - int ntuples; // 1 for ntuples to be written - - //new: vars for FLAT Trees - TTree* masterTree; - TTree* geomTree; - TTree* cherenkovHitsTree; //cfr. RecoHitSeriesMap - TTree* cherenkovDigiHitsTree; //cfr. RecoHitSeriesMap - TTree* tracksTree; - TTree* triggerTree; - TTree* eventInfoTree; - TTree* flatRooTrackerTree; - - // GeoTree - Char_t geo_type_string[20]; - double cyl_radius; - double cyl_length; - Char_t pmt_id_string[50]; - double pmt_radius_id; - Char_t pmt_od_string[50]; - double pmt_radius_od; - int numPMT_id; - int numPMT_od; - int orient; - double offset_x; - double offset_y; - double offset_z; - int num_mPMT; - int tube_id[5000000]; - int mPMT_id[5000000]; - int mPMT_pmt_id[5000000]; - double tube_x[5000000]; - double tube_y[5000000]; - double tube_z[5000000]; - double dir_x[5000000]; - double dir_y[5000000]; - double dir_z[5000000]; - double theta[5000000]; - double phi[5000000]; - - enum CylinderLocation{ - UNDEFINED = -1, - WALL = 0, - TOPCAP = 10, - BOTTOMCAP = 20 - } cylLocation[5000000]; - - //EventHeader - int run; - int event; - int subevent; - - //TriggerTree - TriggerType_t trig_type; - int trig_info; //double trig_info[100]; - double trig_length; - double trig_start; - - //Event info: General, Tracks and Hits - eventNtuple *evNtup; - - //NRooTracker - NRooTrackerVtx *evNRooTracker; - - const G4Run* fG4Run; -}; - -#endif diff --git a/libWCSIM/include/WCSimRunActionMessenger.hh b/libWCSIM/include/WCSimRunActionMessenger.hh deleted file mode 100644 index 5e679e9..0000000 --- a/libWCSIM/include/WCSimRunActionMessenger.hh +++ /dev/null @@ -1,35 +0,0 @@ -#ifndef WCSimRunActionMessenger_h -#define WCSimRunActionMessenger_h 1 - -class WCSimRunAction; -class G4UIdirectory; -class G4UIcmdWithAString; -class G4UIcmdWithABool; - -#include "G4UImessenger.hh" -#include "globals.hh" -#include "G4UIcmdWithABool.hh" - -class WCSimRunActionMessenger: public G4UImessenger -{ - public: - WCSimRunActionMessenger(WCSimRunAction* mpga); - ~WCSimRunActionMessenger(); - - public: - void SetNewValue(G4UIcommand* command, G4String newValues); - - private: - WCSimRunAction* WCSimRun; - - private: //commands - G4UIdirectory* WCSimIODir; - G4UIcmdWithAString* RootFile; - G4UIcmdWithABool* WriteDefaultRootFile; - G4UIcmdWithABool* RooTracker; - -}; - -#endif - - diff --git a/libWCSIM/include/WCSimStackingAction.hh b/libWCSIM/include/WCSimStackingAction.hh deleted file mode 100644 index 9c1b67d..0000000 --- a/libWCSIM/include/WCSimStackingAction.hh +++ /dev/null @@ -1,27 +0,0 @@ -#ifndef WCSimStackingAction_H -#define WCSimStackingAction_H 1 - -#include "globals.hh" -#include "G4UserStackingAction.hh" -#include "WCSimDetectorConstruction.hh" - -class G4Track; - -class WCSimStackingAction : public G4UserStackingAction { - - public: - WCSimStackingAction(WCSimDetectorConstruction*); - virtual ~WCSimStackingAction(); - - public: - virtual G4ClassificationOfNewTrack ClassifyNewTrack(const G4Track* aTrack); - virtual void NewStage(); - virtual void PrepareNewEvent(); - - private: - WCSimDetectorConstruction* DetConstruct; - -}; - -#endif - diff --git a/libWCSIM/include/WCSimSteppingAction.hh b/libWCSIM/include/WCSimSteppingAction.hh deleted file mode 100644 index 04e5475..0000000 --- a/libWCSIM/include/WCSimSteppingAction.hh +++ /dev/null @@ -1,49 +0,0 @@ -#ifndef WCSimSteppingAction_h -#define WCSimSteppingAction_h 1 - -#include "G4Event.hh" -#include "G4UserSteppingAction.hh" -#include "G4ThreeVector.hh" - -class G4HCofThisEvent; -class G4Event; - -class WCSimSteppingAction : public G4UserSteppingAction -{ - -public: - WCSimSteppingAction() - {}; - - ~WCSimSteppingAction() - { }; - - void UserSteppingAction(const G4Step*); - - G4int G4ThreeVectorToWireTime(G4ThreeVector *pos3d, - G4ThreeVector lArPos, - G4ThreeVector start, - G4int i); - - void Distortion(G4double x, - G4double y); - - G4double FieldLines(G4double x, - G4double y, - G4int xy); - - - static G4int n_photons_through_mPMTLV; - static G4int n_photons_through_acrylic; - static G4int n_photons_through_gel; - static G4int n_photons_on_blacksheet; - static G4int n_photons_on_smallPMT; - -private: - - G4double ret[2]; - -}; - - -#endif diff --git a/libWCSIM/include/WCSimTrackInformation.hh b/libWCSIM/include/WCSimTrackInformation.hh deleted file mode 100644 index 0a218bc..0000000 --- a/libWCSIM/include/WCSimTrackInformation.hh +++ /dev/null @@ -1,55 +0,0 @@ -#ifndef WCSimTrackInformation_h -#define WCSimTrackInformation_h 1 - - -#include "globals.hh" -#include "G4Track.hh" -#include "G4Allocator.hh" -#include "G4VUserTrackInformation.hh" - -// Maximilien Fechner, december 2004 -// Information class for flagging the secondaries -// I'm interested in (namely gammas from pi0s and secondaries -// from muon decay -// TF: Also gamma's from neutron capture and oxygen de-excitation and -// electrons from pion decay are very relevant!! -class WCSimTrackInformation : public G4VUserTrackInformation { -private: - G4bool saveit; - G4int primaryParentID; - -public: - WCSimTrackInformation() : saveit(false), primaryParentID(-99) {} //TF: initialize to value with NO meaning instead of DN - WCSimTrackInformation(const WCSimTrackInformation* aninfo) - { saveit = aninfo->saveit; primaryParentID = aninfo->primaryParentID;} - virtual ~WCSimTrackInformation() {} - WCSimTrackInformation(const G4Track* ); - - G4bool isSaved() { return saveit;} - void WillBeSaved(G4bool choice) { saveit = choice;} - - void SetPrimaryParentID(G4int i) { primaryParentID = i;} - G4int GetPrimaryParentID() {return primaryParentID;} - - inline void *operator new(size_t); - inline void operator delete(void *aTrackInfo); - inline int operator ==(const WCSimTrackInformation& right) const - {return (this==&right);} - - void Print() const; - -}; - -extern G4Allocator aWCSimTrackInfoAllocator; - -inline void* WCSimTrackInformation::operator new(size_t) -{ void* aTrackInfo; - aTrackInfo = (void*)aWCSimTrackInfoAllocator.MallocSingle(); - return aTrackInfo; -} - -inline void WCSimTrackInformation::operator delete(void *aTrackInfo) -{ aWCSimTrackInfoAllocator.FreeSingle((WCSimTrackInformation*)aTrackInfo);} - - -#endif diff --git a/libWCSIM/include/WCSimTrackingAction.hh b/libWCSIM/include/WCSimTrackingAction.hh deleted file mode 100644 index 0afab53..0000000 --- a/libWCSIM/include/WCSimTrackingAction.hh +++ /dev/null @@ -1,38 +0,0 @@ -#ifndef WCSimTrackingAction_h -#define WCSimTrackingAction_h - -#include -#include "G4UserTrackingAction.hh" -#include "globals.hh" - -//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... - -class G4Track; -class WCSimTrackingMessenger; - -class WCSimTrackingAction : public G4UserTrackingAction -{ - public: - WCSimTrackingAction(); - ~WCSimTrackingAction(); - - void PreUserTrackingAction (const G4Track* aTrack); - void PostUserTrackingAction(const G4Track*); - - void SetFractionChPhotons(G4double fraction){percentageOfCherenkovPhotonsToDraw = fraction;} - -private: - std::set ProcessList; - std::set ParticleList; - std::set pi0List; - - // TF: define in macro now - G4float percentageOfCherenkovPhotonsToDraw; - - WCSimTrackingMessenger* messenger; -}; - - -#endif - - diff --git a/libWCSIM/include/WCSimTrackingMessenger.hh b/libWCSIM/include/WCSimTrackingMessenger.hh deleted file mode 100644 index 672469b..0000000 --- a/libWCSIM/include/WCSimTrackingMessenger.hh +++ /dev/null @@ -1,28 +0,0 @@ -#ifndef WCSimTrackingMessenger_h -#define WCSimTrackingMessenger_h 1 - - -#include "G4UImessenger.hh" - -class G4UIcommand; -class G4UIdirectory; -class G4UIcmdWithADouble; -class WCSimTrackingAction; - -class WCSimTrackingMessenger: public G4UImessenger -{ -public: - WCSimTrackingMessenger(WCSimTrackingAction*); - - ~WCSimTrackingMessenger(); - - void SetNewValue(G4UIcommand* command, G4String newValue); - -private: - WCSimTrackingAction* myTracking; - - G4UIdirectory* WCSimDir; - G4UIcmdWithADouble* fractionPhotonsToDraw; -}; - -#endif diff --git a/libWCSIM/include/WCSimTrajectory.hh b/libWCSIM/include/WCSimTrajectory.hh deleted file mode 100644 index 597a19b..0000000 --- a/libWCSIM/include/WCSimTrajectory.hh +++ /dev/null @@ -1,137 +0,0 @@ -class WCSimTrajectory; - -#ifndef WCSimTrajectory_h -#define WCSimTrajectory_h 1 - -#include "G4VTrajectory.hh" -#include "G4Allocator.hh" -#include // Include from 'system' -#include "G4ios.hh" // Include from 'system' -#include // G4RWTValOrderedVector -#include "globals.hh" // Include from 'global' -#include "G4ParticleDefinition.hh" // Include from 'particle+matter' -#include "G4TrajectoryPoint.hh" // Include from 'tracking' -#include "G4Track.hh" -#include "G4Step.hh" - -class G4Polyline; // Forward declaration. - -typedef std::vector TrajectoryPointContainer; -/////////////////// -class WCSimTrajectory : public G4VTrajectory -/////////////////// -{ - -//-------- -public: // with description -//-------- - -// Constructor/Destrcutor - - WCSimTrajectory(); - - WCSimTrajectory(const G4Track* aTrack); - WCSimTrajectory(WCSimTrajectory &); - virtual ~WCSimTrajectory(); - -// Operators - inline void* operator new(size_t); - inline void operator delete(void*); - inline int operator == (const WCSimTrajectory& right) const - {return (this==&right);} - -// Get/Set functions - inline G4int GetTrackID() const - { return fTrackID; } - inline G4int GetParentID() const - { return fParentID; } - inline G4String GetParticleName() const - { return ParticleName; } - inline G4double GetCharge() const - { return PDGCharge; } - inline G4int GetPDGEncoding() const - { return PDGEncoding; } - inline G4ThreeVector GetInitialMomentum() const - { return initialMomentum; } - inline G4String GetCreatorProcessName() const { - return creatorProcess; - } - - inline G4double GetGlobalTime() const - { return globalTime; } - inline G4bool GetSaveFlag() const { return SaveIt; } - inline void SetSaveFlag(G4bool value) { SaveIt = value; } - -// New function we have added - inline G4ThreeVector GetStoppingPoint() const - { return stoppingPoint; } - inline G4VPhysicalVolume* GetStoppingVolume() const - { return stoppingVolume;} - inline void SetStoppingPoint(G4ThreeVector& currentPosition) - { stoppingPoint = currentPosition;} - inline void SetStoppingVolume(G4VPhysicalVolume* currentVolume) - { stoppingVolume = currentVolume;} - - -// Other member functions - virtual void ShowTrajectory(std::ostream& os=G4cout) const; - virtual void DrawTrajectory(G4int i_mode=0) const; - virtual void AppendStep(const G4Step* aStep); - virtual int GetPointEntries() const { return positionRecord->size(); } - virtual G4VTrajectoryPoint* GetPoint(G4int i) const - { return (*positionRecord)[i]; } - virtual void MergeTrajectory(G4VTrajectory* secondTrajectory); - - G4ParticleDefinition* GetParticleDefinition(); - - virtual const std::map* GetAttDefs() const; - virtual std::vector* CreateAttValues() const; - -//--------- - private: -//--------- - - TrajectoryPointContainer* positionRecord; - G4int fTrackID; - G4int fParentID; - G4int PDGEncoding; - G4double PDGCharge; - G4String ParticleName; - G4ThreeVector initialMomentum; - - // These are new variables - G4ThreeVector stoppingPoint; - G4VPhysicalVolume *stoppingVolume; - - // M Fechner : new saving mechanism - G4bool SaveIt; - G4String creatorProcess; - G4double globalTime; -}; - -/*** TEMP : M FECHNER *********** -** modification by Chris Walter that works for geant4 >= 4.6.2p01 -** does not compile with 4.6.1 -#if defined G4TRACKING_ALLOC_EXPORT - extern G4DLLEXPORT G4Allocator myTrajectoryAllocator; -#else - extern G4DLLIMPORT G4Allocator myTrajectoryAllocator; -#endif -*/ - -extern G4Allocator myTrajectoryAllocator; - -inline void* WCSimTrajectory::operator new(size_t) -{ - void* aTrajectory; - aTrajectory = (void*)myTrajectoryAllocator.MallocSingle(); - return aTrajectory; -} - -inline void WCSimTrajectory::operator delete(void* aTrajectory) -{ - myTrajectoryAllocator.FreeSingle((WCSimTrajectory*)aTrajectory); -} - -#endif - diff --git a/libWCSIM/include/WCSimTuningMessenger.hh b/libWCSIM/include/WCSimTuningMessenger.hh deleted file mode 100644 index c1c1a2c..0000000 --- a/libWCSIM/include/WCSimTuningMessenger.hh +++ /dev/null @@ -1,47 +0,0 @@ -#ifndef WCSimTuningMessenger_h -#define WCSimTuningMessenger_h 1 - -// Contains detector parameters that need to be set up before detector is constructed -// KS Mar 2010 - -#include "globals.hh" -#include "G4UImessenger.hh" - -class WCSimTuningParameters; -class G4UIdirectory; -class G4UIcommand; -class G4UIcmdWithADouble; -class G4UIcmdWithABool; //jl145 - - -class WCSimTuningMessenger: public G4UImessenger -{ -public: - WCSimTuningMessenger(WCSimTuningParameters*); - ~WCSimTuningMessenger(); - - void SetNewValue(G4UIcommand* command, G4String newValue); - -private: - - WCSimTuningParameters* WCSimTuningParams; - - G4UIdirectory* WCSimDir; - G4UIcmdWithADouble* Rayff; - G4UIcmdWithADouble* Bsrff; - G4UIcmdWithADouble* Abwff; - G4UIcmdWithADouble* Rgcff; - G4UIcmdWithADouble* Qeff; - G4UIcmdWithADouble* Mieff; - G4UIcmdWithADouble* Ttsff; - G4UIcmdWithADouble* PMTSatur; //TD 2019.7.16 - //G4UIcmdWithADouble* Qoiff; //TD 2019.6.26 - //G4UIcmdWithADouble* NLTinfo; //TD 2019.6.26 - - //For Top Veto - jl145 - G4UIcmdWithADouble* TVSpacing; - G4UIcmdWithABool* TopVeto; - -}; - -#endif diff --git a/libWCSIM/include/WCSimTuningParameters.hh b/libWCSIM/include/WCSimTuningParameters.hh deleted file mode 100644 index b65723e..0000000 --- a/libWCSIM/include/WCSimTuningParameters.hh +++ /dev/null @@ -1,90 +0,0 @@ -#ifndef WCSimTuningParameters_h -#define WCSimTuningParameters_h 1 -#include "WCSimTuningMessenger.hh" -#include "WCSimRootOptions.hh" -#include "globals.hh" - -class WCSimTuningParameters -{ -public: - WCSimTuningParameters(); - ~WCSimTuningParameters(); - - - // Setters and getters - G4double GetRayff() {return rayff;} - void SetRayff(G4double rparam) {rayff=rparam;} - - G4double GetBsrff() {return bsrff;} - void SetBsrff(G4double rparam) {bsrff=rparam;} - - G4double GetAbwff() {return abwff;} - void SetAbwff(G4double rparam) {abwff=rparam;} - - G4double GetRgcff() {return rgcff;} - void SetRgcff(G4double rparam) {rgcff=rparam;} - - //Added by B.Q 2018/07/25 - G4double GetQeff() {return qeff;} - void SetQeff(G4double rparam) {qeff=rparam;} - - G4double GetMieff() {return mieff;} - void SetMieff(G4double rparam) {mieff=rparam;} - - //Added by TD 2019/06/22 - G4double GetTtsff() {return ttsff;} - void SetTtsff(G4double rparam) {ttsff=rparam;} - - //TD 2019.7.16 - G4double GetPMTSatur() {return pmtsatur;} - void SetPMTSatur(G4double rparam) {pmtsatur=rparam;} - -// TD 2019.6.26 - // G4double GetQoiff() {return qoiff;} - // void SetQoiff(G4double rparam) {qoiff=rparam;} - - // G4double GetNLTinfo() {return nltinfo;} - // void SetNLTinfo(G4double rparam) {nltinfo=rparam;} - - //For Top Veto - jl145 - G4double GetTVSpacing() {return tvspacing;} - void SetTVSpacing(G4double tparam) {tvspacing=tparam;} - - G4bool GetTopVeto() {return topveto;} - void SetTopVeto(G4double tparam) {topveto=tparam;} - - void SaveOptionsToOutput(WCSimRootOptions * wcopt); - -private: - - // The messenger - WCSimTuningMessenger* TuningMessenger; - - - // The parameters that need to be set before WCSimDetectorConstruction - // is created - - G4double rayff; - G4double bsrff; - G4double abwff; - G4double rgcff; - G4double qeff; - G4double mieff; - G4double ttsff; - // G4double qoiff; // TD 2019.6.26 - G4double pmtsatur; - - //For Top Veto - jl145 - G4double tvspacing; - G4bool topveto; - -}; - -#endif - - - - - - - diff --git a/libWCSIM/include/WCSimVisManager.hh b/libWCSIM/include/WCSimVisManager.hh deleted file mode 100644 index 5d8b438..0000000 --- a/libWCSIM/include/WCSimVisManager.hh +++ /dev/null @@ -1,27 +0,0 @@ - - - -#ifndef WCSimVisManager_h -#define WCSimVisManager_h 1 - -#ifdef G4VIS_USE - -#include "G4VisManager.hh" - - -class WCSimVisManager: public G4VisManager { - -public: - - WCSimVisManager (); - -private: - - void RegisterGraphicsSystems (); - -}; - - -#endif - -#endif diff --git a/libWCSIM/include/WCSimWCAddDarkNoise.hh b/libWCSIM/include/WCSimWCAddDarkNoise.hh deleted file mode 100644 index eb55ce7..0000000 --- a/libWCSIM/include/WCSimWCAddDarkNoise.hh +++ /dev/null @@ -1,72 +0,0 @@ -#ifndef WCSimWCAddDarkNoise_h -#define WCSimWCAddDarkNoise_h 1 - -#include "WCSimDarkRateMessenger.hh" -#include "WCSimDetectorConstruction.hh" -#include "G4VDigitizerModule.hh" -#include "WCSimWCDigi.hh" -#include "WCSimWCHit.hh" -#include "WCSimRootOptions.hh" -#include "globals.hh" -#include "Randomize.hh" -#include -#include -#include - -class WCSimWCAddDarkNoise : public G4VDigitizerModule -{ -public: - - WCSimWCAddDarkNoise(G4String name, WCSimDetectorConstruction*, G4String); - //WCSimWCAddDarkNoise(G4String name, WCSimDetectorConstruction*, G4String); - ~WCSimWCAddDarkNoise(); - -public: - void AddDarkNoise(); - void AddDarkNoiseBeforeDigi(WCSimWCDigitsCollection* WCHCPMT, float num1 ,float num2); - void FindDarkNoiseRanges(WCSimWCDigitsCollection* WCHCPMT, float width); - //As it inherits from G4VDigitizerModule it needs a digitize class. Not used - void Digitize() { } - void SetDarkRate(double idarkrate){ PMTDarkRate = idarkrate; } - double GetDarkRate() { return PMTDarkRate; } - void SetConversion(double iconvrate){ ConvRate = iconvrate; } - void SetDarkMode(int imode){DarkMode = imode;} - int GetDarkMode(){return DarkMode;} - void SetDarkHigh(int idarkhigh){DarkHigh = idarkhigh;} - double GetDarkHigh(){return DarkHigh;} - void SetDarkLow(int idarklow){DarkLow = idarklow;} - double GetDarkLow(){return DarkLow;} - void SetDarkWindow(int idarkwindow){DarkWindow = idarkwindow;} - int GetDarkWindow(){return (int)(DarkWindow);} - void SaveOptionsToOutput(WCSimRootOptions * wcopt, string tag); - -private: - void ReInitialize() { ranges.clear(); result.clear();} - void SetPMTDarkDefaults(); - - WCSimDarkRateMessenger *DarkRateMessenger; - double PMTDarkRate; // kHz - double ConvRate; // kHz - double DarkHigh; //ns - double DarkLow; //ns - double DarkWindow; //ns - int DarkMode; - bool fCalledAddDarkNoise; - - WCSimDetectorConstruction* myDetector; - G4String detectorElement; - - std::vector > ranges; - std::vector > result; - -}; - -#endif - - - - - - - - diff --git a/libWCSIM/include/WCSimWCDAQMessenger.hh b/libWCSIM/include/WCSimWCDAQMessenger.hh deleted file mode 100644 index ad96009..0000000 --- a/libWCSIM/include/WCSimWCDAQMessenger.hh +++ /dev/null @@ -1,78 +0,0 @@ -#ifndef WCSimWCDAQMessenger_h -#define WCSimWCDAQMessenger_h 1 - -#include "G4UImessenger.hh" -#include "G4UIcmdWithAString.hh" -#include "G4UIcmdWithABool.hh" -#include "G4UIcmdWithAnInteger.hh" -#include "G4UIcmdWithADouble.hh" -#include "G4UIdirectory.hh" -#include "G4UIcmdWithoutParameter.hh" - -class G4UIcommand; -class WCSimEventAction; -class WCSimWCDigitizerBase; -class WCSimWCTriggerBase; - -class WCSimWCDAQMessenger: public G4UImessenger -{ -public: - WCSimWCDAQMessenger(WCSimEventAction*); - - ~WCSimWCDAQMessenger(); - - void SetNewValue(G4UIcommand* command, G4String newValue); - - void SetTriggerOptions(); - void SetDigitizerOptions(); - - void TellMeAboutTheDigitizer (WCSimWCDigitizerBase* digitizer) { WCSimDigitize = digitizer; } - void TellMeAboutTheTrigger (WCSimWCTriggerBase* trigger) { WCSimTrigger = trigger; } - -private: - WCSimEventAction* WCSimEvent; - WCSimWCDigitizerBase* WCSimDigitize; - WCSimWCTriggerBase* WCSimTrigger; - - G4UIdirectory* WCSimDAQDir; - G4UIcmdWithAString* DigitizerChoice; - G4String StoreDigitizerChoice; - G4UIcmdWithAString* TriggerChoice; - G4String StoreTriggerChoice; - G4UIcmdWithABool* MultiDigitsPerTrigger; - G4bool StoreMultiDigitsPerTrigger; - G4bool MultiDigitsPerTriggerSet; - - G4UIdirectory* DigitizerDir; - G4UIcmdWithAnInteger* DigitizerDeadTime; - G4int StoreDigitizerDeadTime; - G4UIcmdWithAnInteger* DigitizerIntegrationWindow; - G4int StoreDigitizerIntegrationWindow; - - G4UIdirectory* SaveFailuresTriggerDir; - G4UIcmdWithAnInteger* SaveFailuresTriggerMode; - G4int StoreSaveFailuresMode; - G4UIcmdWithADouble* SaveFailuresTriggerTime; - G4double StoreSaveFailuresTime; - G4UIcmdWithAnInteger* SaveFailuresPreTriggerWindow; - G4int StoreSaveFailuresPreWindow; - G4UIcmdWithAnInteger* SaveFailuresPostTriggerWindow; - G4int StoreSaveFailuresPostWindow; - - G4UIdirectory* NDigitsTriggerDir; - G4UIcmdWithAnInteger* NDigitsTriggerThreshold; - G4int StoreNDigitsThreshold; - G4UIcmdWithAnInteger* NDigitsTriggerWindow; - G4int StoreNDigitsWindow; - G4UIcmdWithABool* NDigitsTriggerAdjustForNoise; - G4bool StoreNDigitsAdjustForNoise; - G4UIcmdWithAnInteger* NDigitsPreTriggerWindow; - G4int StoreNDigitsPreWindow; - G4UIcmdWithAnInteger* NDigitsPostTriggerWindow; - G4int StoreNDigitsPostWindow; - - G4String initialiseString; - G4bool initialised; -}; - -#endif diff --git a/libWCSIM/include/WCSimWCDigi.hh b/libWCSIM/include/WCSimWCDigi.hh deleted file mode 100644 index 9e374f5..0000000 --- a/libWCSIM/include/WCSimWCDigi.hh +++ /dev/null @@ -1,232 +0,0 @@ -#ifndef WCSimWCDigi_h -#define WCSimWCDigi_h 1 - -#include "G4VDigi.hh" -#include "G4TDigiCollection.hh" -#include "G4Allocator.hh" -#include "G4ThreeVector.hh" -#include "G4LogicalVolume.hh" -#include "G4ios.hh" - -#include -#include -#include - -// for sort, find, count_if -#include -//for less_equal, bind2nd,... -#include - -// compose2 is not part of the C++ standard -#include -using __gnu_cxx::compose2; - - - -class WCSimWCDigi : public G4VDigi -{ - -public: - - WCSimWCDigi(); - ~WCSimWCDigi(); - WCSimWCDigi(const WCSimWCDigi&); - const WCSimWCDigi& operator=(const WCSimWCDigi&); - int operator==(const WCSimWCDigi&) const; - - inline void* operator new(size_t); - inline void operator delete(void*); - - void Draw(); - void Print(); - -private: - - //PMT parameters - G4int tubeID; - G4RotationMatrix rot; - G4ThreeVector pos; - G4ThreeVector orient; - G4LogicalVolume* pLogV; - G4String tubeType;//Added by B.Q for hybrid PMT configuration - - //'Gates' is a digit counter or specifies subevent - //'TriggerTimes' specifies e.g. the subevent trigger time - std::set Gates; // list of gates that were hit - std::vector TriggerTimes; - - //lists (meaning vector/map) of information for each hit/digit created on the PMT - std::map pe; ///< Charge of each Digi - std::map time_presmear; ///< Time of each Digi, before smearing - std::map time; ///< Time of each Digi - std::vector time_float; ///< Same information as "time" but stored in a vector for quick time sorting - /** \brief IDs of the hits that make up this Digit (do not use for Hits) - * - * Stores the unique IDs of each photon making up a digit - * Each digit can be made from multiple photons, therefore a vector is used - * For example: 0:[3,4,6]; 1:[10,11,13,14] - * The first digit in the event is made of of photons 3,4,6; - * The second digit is made up of photons: 10,11,13,14 - */ - std::map > fDigiComp; - std::map primaryParentID; ///< Primary parent ID of the Hit (do not use for Digits) - - - //integrated hit/digit parameters - G4int totalPe; - - //parameters not actually used? - G4int totalPeInGate; - G4double edep; - static G4int maxPe; - G4int trackID; - -public: - void RemoveDigitizedGate(G4int gate); - - inline void SetTubeID(G4int tube) {tubeID = tube;}; - inline void SetTubeType(G4String tube_type) {tubeType = tube_type;}; - inline void AddGate(int g,float t) { Gates.insert(g); TriggerTimes.push_back(t);} - inline void SetPe(G4int gate, G4float Q) {pe[gate] = Q;}; - inline void SetTime(G4int gate, G4float T) {time[gate] = T;}; - inline void SetPreSmearTime(G4int gate, G4float T) {time_presmear[gate] = T;}; - inline void SetParentID(G4int gate, G4int parent) { primaryParentID[gate] = parent; }; - - // Add a digit number and unique photon number to fDigiComp - inline void AddPhotonToDigiComposition(int digi_number, int photon_number){ - fDigiComp[digi_number].push_back(photon_number); - } - // Add a whole vector for one digit to fDigiComp. Clear input vector once added. - void AddDigiCompositionInfo(std::vector & digi_comp){ - const size_t size = fDigiComp.size(); - fDigiComp[size] = digi_comp; - digi_comp.clear(); - } - - - inline G4int GetParentID(int gate) { return primaryParentID[gate];}; - inline G4int GetTrackID() { return trackID;}; - inline G4float GetGateTime(int gate) { return TriggerTimes[gate];} - inline G4int GetTubeID() {return tubeID;}; - inline G4String GetTubeType() {return tubeType;}; - inline G4ThreeVector GetPos(){ return pos;}; - inline G4ThreeVector GetOrientation(){ return orient;}; - inline G4RotationMatrix GetRot(){ return rot;}; - inline G4float GetPe(int gate) {return pe[gate];}; - inline G4float GetTime(int gate) {return time[gate];}; - inline G4float GetPreSmearTime(int gate) {return time_presmear[gate];}; - std::vector GetDigiCompositionInfo(int gate); - inline std::map< int, std::vector > GetDigiCompositionInfo(){return fDigiComp;} - - inline int NumberOfGates() { return Gates.size();} - inline int NumberOfSubEvents() { return (Gates.size()-1);} - inline bool HasHitsInGate(int number) { return (Gates.count(number)>0); } - - G4LogicalVolume* GetLogicalVolume() {return pLogV;}; - - - void SetEdep (G4double de) { edep = de; }; - void SetPos (G4ThreeVector xyz) { pos = xyz; }; - void SetOrientation (G4ThreeVector xyz) { orient = xyz; }; - void SetLogicalVolume(G4LogicalVolume* logV) { pLogV = logV;} - void SetTrackID (G4int track) { trackID = track; }; - void SetRot (G4RotationMatrix rotMatrix) { rot = rotMatrix; }; - G4int GetTotalPe() { return totalPe;}; - - void SetMaxPe(G4int number = 0) {maxPe = number;}; - - void AddPe(G4float hitTime) - { - // Increment the totalPe number - totalPe++; - - time_float.push_back(hitTime); - } - - void SortHitTimes() { sort(time_float.begin(),time_float.end()); } - - - void SortArrayByHitTime() { - int i, j; - float index_time,index_timepresmear,index_pe; - std::vector index_digicomp; - int index_primaryparentid; - for (i = 1; i < (int) time.size(); ++i) - { - index_time = time[i]; - index_timepresmear = time_presmear[i]; - index_pe = pe[i]; - index_digicomp = fDigiComp[i]; - index_primaryparentid = primaryParentID[i]; - for (j = i; j > 0 && time[j-1] > index_time; j--) { - time[j] = time[j-1]; - pe[j] = pe[j-1]; - fDigiComp[j] = fDigiComp[j-1]; - primaryParentID[j] = primaryParentID[j-1]; - //G4cout <<"swapping "< 0 && a[j-1] > index; j--) - a[j] = a[j-1]; - - a[j] = index; - } - } - - - G4float GetFirstHitTimeInGate(G4float low,G4float upevent) - { - G4float firsttime; - std::vector::iterator tfirst = time_float.begin(); - std::vector::iterator tlast = time_float.end(); - - std::vector::iterator found = - std::find_if(tfirst,tlast, - compose2(std::logical_and(), - std::bind2nd(std::greater_equal(),low), - std::bind2nd(std::less_equal(),upevent) - ) - ); - if ( found != tlast ) { - firsttime = *found; // first hit time - } - else { - firsttime = -10000.; //error code. - } - //G4cout << "firsthit time " << firsttime << "\n"; - return firsttime; - } - - -}; - -typedef G4TDigiCollection WCSimWCDigitsCollection; -extern G4Allocator WCSimWCDigiAllocator; - -inline void* WCSimWCDigi::operator new(size_t) -{ - void* aDigi; - aDigi = (void*) WCSimWCDigiAllocator.MallocSingle(); - return aDigi; -} - -inline void WCSimWCDigi::operator delete(void* aDigi) -{ - WCSimWCDigiAllocator.FreeSingle((WCSimWCDigi*) aDigi); -} - -#endif diff --git a/libWCSIM/include/WCSimWCDigitizer.hh b/libWCSIM/include/WCSimWCDigitizer.hh deleted file mode 100644 index 0e3b6fd..0000000 --- a/libWCSIM/include/WCSimWCDigitizer.hh +++ /dev/null @@ -1,112 +0,0 @@ -#ifndef WCSimWCDigitizer_h -#define WCSimWCDigitizer_h 1 - -#include "WCSimRootOptions.hh" -#include "WCSimDarkRateMessenger.hh" -#include "WCSimWCDAQMessenger.hh" -#include "WCSimDetectorConstruction.hh" -#include "WCSimEnumerations.hh" -#include "G4VDigitizerModule.hh" -#include "WCSimWCDigi.hh" -#include "WCSimWCHit.hh" -#include "globals.hh" -#include "Randomize.hh" -#include -#include - - -// ******************************************* -// BASE CLASS -// ******************************************* - -class WCSimWCDigitizerBase : public G4VDigitizerModule -{ -public: - - WCSimWCDigitizerBase(G4String name, WCSimDetectorConstruction*, WCSimWCDAQMessenger*, DigitizerType_t, G4String detectorElement); - virtual ~WCSimWCDigitizerBase(); - - bool AddNewDigit(int tube, int gate, float digihittime, float peSmeared, std::vector digi_comp); - virtual void DigitizeHits(WCSimWCDigitsCollection* WCHCPMT) = 0; - void DigitizeGate(WCSimWCDigitsCollection* WCHC,G4int G); - void Digitize(); - - //.mac file option setting methods - void SetDigitizerDeadTime (int deadtime) { DigitizerDeadTime = deadtime; }; ///< Override the default digitizer deadtime (ns) - void SetDigitizerIntegrationWindow(int inttime ) { DigitizerIntegrationWindow = inttime; }; ///< Override the default digitizer integration window (ns) - - ///Save current values of options - void SaveOptionsToOutput(WCSimRootOptions * wcopt); - -protected: - void ReInitialize() { DigiStoreHitMap.clear(); } - - G4double peSmeared; - - WCSimDetectorConstruction* myDetector; ///< Get the geometry information - WCSimWCDAQMessenger* DAQMessenger; ///< Get the /DAQ/ .mac options - G4String detectorElement; //to identify which hit collection the trigger is associated: OD, ID, ID PMT type 2(for hybrid) - - WCSimWCDigitsCollection* DigiStore; - std::map DigiStoreHitMap; ///< Used to check if a digit has already been created on a PMT - - //generic digitizer properties. Defaults set with the GetDefault*() methods. Overidden by .mac options - G4String DigitizerClassName; ///< Name of the digitizer class being run - int DigitizerDeadTime; ///< Digitizer deadtime (ns) - int DigitizerIntegrationWindow; ///< Digitizer integration window (ns) - - DigitizerType_t DigitizerType; ///< Enumeration to say which digitizer we've constructed - - virtual int GetDefaultDeadTime() = 0; ///< Set the default digitizer-specific deadtime (in ns) (overridden by .mac) - virtual int GetDefaultIntegrationWindow() = 0; ///< Set the default digitizer-specific integration window (in ns) (overridden by .mac) - - void GetVariables(); ///< Get the default deadtime, etc. from the derived class, and override with read from the .mac file -}; - - - -// ******************************************* -// DERIVED CLASSES -// ******************************************* - - -// SKI-based digitizer class -class WCSimWCDigitizerSKI : public WCSimWCDigitizerBase -{ -public: - - WCSimWCDigitizerSKI(G4String name, WCSimDetectorConstruction*, WCSimWCDAQMessenger*, G4String); - ~WCSimWCDigitizerSKI(); - - void DigitizeHits(WCSimWCDigitsCollection* WCHCPMT); - -private: - int GetDefaultDeadTime() { return 0; } ///< SKI digitizer deadtime is 0 ns - int GetDefaultIntegrationWindow() { return 200; } ///< - - static void Threshold(double& pe,int& iflag){ - // CLHEP::HepRandom::setTheSeed(pe+2014); - double x = pe+0.1; iflag=0; - double thr; double RDUMMY,err; - if ( x<1.1) { - thr = std::min(1.0, - -0.06374+x*(3.748+x*(-63.23+x*(452.0+x*(-1449.0+x*(2513.0 - +x*(-2529.+x*(1472.0+x*(-452.2+x*(51.34+x*2.370)))))))))); - } else { - thr = 1.0; - } - RDUMMY = G4UniformRand(); - if (thr < RDUMMY) { - pe = 0.0; - iflag = 1; - } - else { - err = G4RandGauss::shoot(0.0,0.03); - ///// call rngaus(0.0, 0.03, err); - pe = pe+err; - } - } -}; - - -#endif //WCSimWCDigitizer_h diff --git a/libWCSIM/include/WCSimWCHit.hh b/libWCSIM/include/WCSimWCHit.hh deleted file mode 100644 index 03a8c3b..0000000 --- a/libWCSIM/include/WCSimWCHit.hh +++ /dev/null @@ -1,188 +0,0 @@ -#ifndef WCSimWCHit_h -#define WCSimWCHit_h 1 - -#include "G4VHit.hh" -#include "G4THitsCollection.hh" -#include "G4Allocator.hh" -#include "G4ThreeVector.hh" -#include "G4LogicalVolume.hh" -#include "G4ios.hh" -// for sort, find, count_if -#include -//for less_equal, bind2nd,... -#include - -// compose2 is not part of the C++ standard -// use this kludgy technique to use it -#include -using __gnu_cxx::identity_element; -using __gnu_cxx::unary_compose; -using __gnu_cxx::binary_compose; -using __gnu_cxx::compose1; -using __gnu_cxx::compose2; -using __gnu_cxx::identity; -using __gnu_cxx::select1st; -using __gnu_cxx::select2nd; -using __gnu_cxx::project1st; -using __gnu_cxx::project2nd; -using __gnu_cxx::constant_void_fun; -using __gnu_cxx::constant_unary_fun; -using __gnu_cxx::constant_binary_fun; -using __gnu_cxx::constant0; -using __gnu_cxx::constant1; -using __gnu_cxx::constant2; -using __gnu_cxx::subtractive_rng; -using __gnu_cxx::mem_fun1; -using __gnu_cxx::mem_fun1_ref; - - - -class WCSimWCHit : public G4VHit -{ - public: - - WCSimWCHit(); - ~WCSimWCHit(); - WCSimWCHit(const WCSimWCHit&); - const WCSimWCHit& operator=(const WCSimWCHit&); - G4int operator==(const WCSimWCHit&) const; - - inline void* operator new(size_t); - inline void operator delete(void*); - - void Draw(); - void Print(); - - public: - - void SetTubeID (G4int tube) { tubeID = tube; }; - void SetTrackID (G4int track) { trackID = track; }; - void SetEdep (G4double de) { edep = de; }; - void SetPos (G4ThreeVector xyz) { pos = xyz; }; - void SetOrientation (G4ThreeVector xyz) { orient = xyz; }; - void SetRot (G4RotationMatrix rotMatrix) { rot = rotMatrix; }; - void SetLogicalVolume(G4LogicalVolume* logV) { pLogV = logV;} - void AddParentID (G4int primParentID) - { primaryParentID.push_back(primParentID); } - void SetTubeType (G4String tube_type) { tubeType = tube_type; }; //Added by B.Quilain to transmit on which PMT type the hit happened. For detectors with several PMT types in ID. - - - // This is temporarily used for the drawing scale - void SetMaxPe(G4int number = 0) {maxPe = number;}; - - void AddPe(G4float hitTime) - { - // First increment the totalPe number - totalPe++; - - if (totalPe > maxPe) - maxPe = totalPe; - - time.push_back(hitTime); - } - - G4int GetTubeID() { return tubeID; }; - G4int GetTrackID() { return trackID; }; - G4ThreeVector GetPos() { return pos; }; - G4ThreeVector GetOrientation() { return orient; }; - G4int GetTotalPe() { return totalPe;}; - G4float GetTime(int i) { return time[i];}; - G4int GetParentID(int i) { return primaryParentID[i];}; - G4String GetTubeType() { return tubeType; }; - - G4LogicalVolume* GetLogicalVolume() {return pLogV;}; - - void SortHitTimes() { sort(time.begin(),time.end()); } - - - // low is the trigger time, up is trigger+950ns (end of event) - G4float GetFirstHitTimeInGate(G4float low,G4float upevent) - { - G4float firsttime; - std::vector::iterator tfirst = time.begin(); - std::vector::iterator tlast = time.end(); - - std::vector::iterator found = - std::find_if(tfirst,tlast, - compose2(std::logical_and(), - std::bind2nd(std::greater_equal(),low), - std::bind2nd(std::less_equal(),upevent) - ) - ); - if ( found != tlast ) { - firsttime = *found; // first hit time - } - else { - firsttime = -10000.; //error code. - } - //G4cout << "firsthit time " << firsttime << "\n"; - return firsttime; - } - - - // pmtgate and evgate are durations, ie not absolute times - - G4int GetPeInGate(double low, double pmtgate,double evgate) { - // M Fechner; april 2005 - // assumes that time has already been sorted - std::vector::iterator tfirst = time.begin(); - std::vector::iterator tlast = time.end(); - // select min time - G4float mintime = (pmtgate < evgate) ? pmtgate : evgate; - - // return number of hits in the time window... - - G4int number = std::count_if(tfirst,tlast, - compose2(std::logical_and(), - std::bind2nd(std::greater_equal(),low), - std::bind2nd(std::less_equal(),mintime) - ) - ); - - totalPeInGate = number; - // G4cout << "numer = " << number <<"\n"; - return number; - } - - - private: - - void HSVtoRGB(float& fR, float& fG, float& fB, float& fH, float& fS, float& fV); - - G4int tubeID; - G4int trackID; - G4double edep; - G4ThreeVector pos; - G4ThreeVector orient; - G4RotationMatrix rot; - G4LogicalVolume* pLogV; - G4String tubeType;//Added by B.Quilain to transmit on which PMT type the hit happened. For detectors with several PMT types in ID. - - // This is temporarily used for the drawing scale - // Since its static *every* WChit sees the same value for this. - - static G4int maxPe; - - G4int totalPe; - std::vector time; - std::vector primaryParentID; - G4int totalPeInGate; -}; - -typedef G4THitsCollection WCSimWCHitsCollection; - -extern G4Allocator WCSimWCHitAllocator; - -inline void* WCSimWCHit::operator new(size_t) -{ - void *aHit; - aHit = (void *) WCSimWCHitAllocator.MallocSingle(); - return aHit; -} - -inline void WCSimWCHit::operator delete(void *aHit) -{ - WCSimWCHitAllocator.FreeSingle((WCSimWCHit*) aHit); -} - -#endif diff --git a/libWCSIM/include/WCSimWCPMT.hh b/libWCSIM/include/WCSimWCPMT.hh deleted file mode 100644 index eb136ec..0000000 --- a/libWCSIM/include/WCSimWCPMT.hh +++ /dev/null @@ -1,54 +0,0 @@ -#ifndef WCSimWCPMT_h -#define WCSimWCPMT_h 1 - -#include "WCSimDarkRateMessenger.hh" -#include "WCSimDetectorConstruction.hh" -#include "G4VDigitizerModule.hh" -#include "WCSimWCDigi.hh" -#include "WCSimWCHit.hh" -#include "globals.hh" -#include "Randomize.hh" -#include -#include - - -class WCSimWCPMT : public G4VDigitizerModule -{ -public: - - WCSimWCPMT(G4String name, WCSimDetectorConstruction*, G4String detectorElement); - ~WCSimWCPMT(); - - void ReInitialize() { DigiHitMapPMT.clear(); TriggerTimes.clear(); } - - -public: - - void AddPMTDarkRate(WCSimWCDigitsCollection*); - void MakePeCorrection(WCSimWCHitsCollection*); - void Digitize(); - G4double GetTriggerTime(int i) { return TriggerTimes[i];} - // void SetConversion(double iconvrate){ ConvRate = iconvrate; } - // static G4double GetLongTime() { return LongTime;} - - G4double rn1pe(); - G4double peSmeared; - // double ConvRate; // kHz - std::vector TriggerTimes; - std::map DigiHitMapPMT; // need to check if a hit already exists.. - - WCSimWCDigitsCollection* DigitsCollection; - WCSimDetectorConstruction* myDetector; - G4String detectorElement; - -}; - -#endif - - - - - - - - diff --git a/libWCSIM/include/WCSimWCSD.hh b/libWCSIM/include/WCSimWCSD.hh deleted file mode 100644 index 0851138..0000000 --- a/libWCSIM/include/WCSimWCSD.hh +++ /dev/null @@ -1,34 +0,0 @@ -#ifndef WCSimWCSD_h -#define WCSimWCSD_h 1 - -#include "G4VSensitiveDetector.hh" -#include "WCSimWCHit.hh" -#include "WCSimDetectorConstruction.hh" - -class G4Step; -class G4HCofThisEvent; - -class WCSimWCSD : public G4VSensitiveDetector -{ - public: - WCSimWCSD(G4String,G4String,WCSimDetectorConstruction*, G4String); - ~WCSimWCSD(); - - void Initialize(G4HCofThisEvent*); - - G4bool ProcessHits(G4Step*, G4TouchableHistory*); - void EndOfEvent(G4HCofThisEvent*); - - private: - - G4int HCID; - WCSimDetectorConstruction* fdet; - WCSimWCHitsCollection* hitsCollection; - WCSimWCHitsCollection* hitsCollection2; - std::map PMTHitMap; // Whether a PMT was hit already - - G4String detectorElement; -}; - -#endif - diff --git a/libWCSIM/include/WCSimWCTrigger.hh b/libWCSIM/include/WCSimWCTrigger.hh deleted file mode 100644 index 4874bb0..0000000 --- a/libWCSIM/include/WCSimWCTrigger.hh +++ /dev/null @@ -1,362 +0,0 @@ -#ifndef WCSimWCTrigger_h -#define WCSimWCTrigger_h 1 - -#include "WCSimRootOptions.hh" -#include "WCSimEnumerations.hh" -#include "WCSimWCDAQMessenger.hh" -#include "WCSimDetectorConstruction.hh" -#include "G4VDigitizerModule.hh" -#include "WCSimWCDigi.hh" -#include "WCSimWCHit.hh" -#include "globals.hh" -#include "Randomize.hh" -#include -#include - -class WCSimWCDigiTrigger; -typedef G4TDigiCollection WCSimWCTriggeredDigitsCollection; - -// ******************************************* -// BASE CLASS -// ******************************************* - -/** - * \class WCSimWCTriggerBase - * - * \brief The base class for WCSim triggering algorithms - * - * Concrete implementations of a trigger class should inherit from this class. - * Minimally, only DoTheWork() needs to be implemented in the implementation class. - * - */ - -class WCSimWCTriggerBase : public G4VDigitizerModule -{ - -public: - - ///Create WCSimWCTriggerBase instance with knowledge of the detector and DAQ options - WCSimWCTriggerBase(G4String name, WCSimDetectorConstruction*, WCSimWCDAQMessenger*, G4String); - - virtual ~WCSimWCTriggerBase(); - - ///The main user-callable routine of the class. Gets the input & creates the output WCSimWCTriggeredDigitsCollection's, then calls DoTheWork() - void Digitize(); - - ///Returns the number of trigger gates in the event (i.e. the number of triggers passed) - int NumberOfGatesInThisEvent() { return TriggerTimes.size(); } - ///Get the time of the ith trigger - Float_t GetTriggerTime(int i) { return TriggerTimes[i];} - ///Get the trigger type of the ith trigger - TriggerType_t GetTriggerType(int i) { return TriggerTypes[i];} - ///Get the additional trigger information associated with the ith trigger - std::vector GetTriggerInfo(int i) { return TriggerInfos[i];} - ///Get the trigger class name - G4String GetTriggerClassName(){ return triggerClassName; } - - // - // Trigger algorithm option set methods - // - - ///Set whether to allow the number of digits per PMT per trigger to go > 1 - void SetMultiDigitsPerTrigger(G4bool allow_multi) { multiDigitsPerTrigger = allow_multi; } - - // NDigits options - ///Set the threshold for the NDigits trigger - void SetNDigitsThreshold(G4int threshold) { ndigitsThreshold = threshold; } - ///Set the time window for the NDigits trigger - void SetNDigitsWindow(G4int window) { ndigitsWindow = window; } - ///Automatically adjust the NDigits threshold based on the average noise occupancy? - void SetNDigitsAdjustForNoise (G4bool adjust) { ndigitsAdjustForNoise = adjust; } - ///Set the pretrigger window for the NDigits trigger (value will be forced negative) - void SetNDigitsPreTriggerWindow(G4int window) { ndigitsPreTriggerWindow = - abs(window); } - ///Set the posttrigger window for the NDigits trigger (value will be forced positive) - void SetNDigitsPostTriggerWindow(G4int window) { ndigitsPostTriggerWindow = + abs(window); } - - // Save trigger failures options - ///Set the mode for saving failed triggers (0:save only triggered events, 1:save both triggered events & failed events, 2:save only failed events) - void SetSaveFailuresMode (G4int mode ) { saveFailuresMode = mode; } - ///Set the dummy trigger time for the failed triggers - void SetSaveFailuresTime (G4double time ) { saveFailuresTime = time; } - ///Set the pretrigger window for the SaveFailures trigger (value will be forced negative) - void SetSaveFailuresPreTriggerWindow(G4int window) { saveFailuresPreTriggerWindow = - abs(window); } - ///Set the posttrigger window for the SaveFailures trigger (value will be forced positive) - void SetSaveFailuresPostTriggerWindow(G4int window) { saveFailuresPostTriggerWindow = + abs(window); } - - ///Knowledge of the dark rate (use for automatically adjusting for noise) - void SetDarkRate(double idarkrate){ PMTDarkRate = idarkrate; } - - ///Save current values of options - void SaveOptionsToOutput(WCSimRootOptions * wcopt); - -protected: - - ///This should call the trigger algorithms, and handle any temporary DigitsCollection's - virtual void DoTheWork(WCSimWCDigitsCollection* WCDCPMT) = 0; - - /// Get the default threshold, etc. from the derived class, and override with read from the .mac file - void GetVariables(); - - ///Set the default trigger class specific decision of whether to save multiple digits per PMT per trigger (overridden by .mac) - virtual bool GetDefaultMultiDigitsPerTrigger() { return true; } - ///Set the default trigger class specific NDigits window (in ns) (overridden by .mac) - virtual int GetDefaultNDigitsWindow() { return 200; } - ///Set the default trigger class specific NDigits threshold (in ns) (overridden by .mac) - virtual int GetDefaultNDigitsThreshold() { return 25; } //Default - //virtual int GetDefaultNDigitsThreshold() { return 20; } //B.Q - ///Set the default trigger class specific NDigits pretrigger window (in ns) (overridden by .mac) - virtual int GetDefaultNDigitsPreTriggerWindow() { return -400; } - ///Set the default trigger class specific NDigits posttrigger window (in ns) (overridden by .mac) - virtual int GetDefaultNDigitsPostTriggerWindow() { return 950; } - - ///Get the pretrigger window for a given trigger algorithm - int GetPreTriggerWindow(TriggerType_t t); - ///Get the posttrigger window for a given trigger algorithm - int GetPostTriggerWindow(TriggerType_t t); - - //these are the algorithms that perform triggering - //they are stored here so that different trigger classes can use the same algorithms without copying code - /** - * \brief An NDigits trigger algorithm - * - * Looks through the input WCSimWCDigitsCollection and integrates the number of hits in a (specified) time window - * If the integral passes above a (specified) threshold, a trigger is issued - * - * The trigger type is kTriggerNDigits - * - * The trigger time is the time of the first digit above threshold - * - * The trigger information is the number of hits in the time window (i.e. the number of hits that caused the trigger to fire) - * - * Currently setup with the optional 'test' argument which runs the algorithm with half the hit threshold - * for testing purposes. Triggers issued in this mode have type kTriggerNDigitsTest - */ - void AlgNDigits(WCSimWCDigitsCollection* WCDCPMT, bool remove_hits, bool test=false); - - WCSimWCTriggeredDigitsCollection* DigitsCollection; ///< The main output of the class - collection of digits in the trigger window - std::map DigiHitMap; ///< Keeps track of the PMTs that have been added to the output WCSimWCTriggeredDigitsCollection - - std::vector TriggerTimes; ///< The times of the triggers - std::vector TriggerTypes; ///< The type of the triggers - std::vector< std::vector > TriggerInfos; ///< Additional information associated with each trigger - - WCSimWCDAQMessenger* DAQMessenger; ///< Get the options from the .mac file - WCSimDetectorConstruction* myDetector; ///< Know about the detector, so can add appropriate PMT time smearing - - G4String detectorElement; - - /// Clear the Trigger* vectors and DigiHitMap - void ReInitialize() { - TriggerTimes.clear(); - TriggerTypes.clear(); - TriggerInfos.clear(); - DigiHitMap.clear(); - } - - double PMTDarkRate; ///< Dark noise rate of the PMTs - - // Trigger algorithm options - G4bool multiDigitsPerTrigger; ///< Allow the number of digits per PMT saved in each trigger window to go > 1? - //NDigits - G4int ndigitsThreshold; ///< The threshold for the NDigits trigger - G4int ndigitsWindow; ///< The time window for the NDigits trigger - G4bool ndigitsAdjustForNoise; ///< Automatically adjust the NDigits trigger threshold based on the average dark noise rate? - G4int ndigitsPreTriggerWindow; ///< The pretrigger window to save before an NDigits trigger - G4int ndigitsPostTriggerWindow; ///< The posttrigger window to save after an NDigits trigger - //Save failures - G4int saveFailuresMode; ///< The mode for saving events which don't pass triggers - G4double saveFailuresTime; ///< The dummy trigger time for failed events - G4int saveFailuresPreTriggerWindow; ///< The pretrigger window to save before an SaveFailures trigger - G4int saveFailuresPostTriggerWindow; ///< The posttrigger window to save after an SaveFailures trigger - - G4String triggerClassName; ///< Save the name of the trigger class - -private: - ///modify the NDigits threshold based on the average dark noise rate - void AdjustNDigitsThresholdForNoise(); - - ///takes all trigger times, then loops over all Digits & fills the output DigitsCollection - void FillDigitsCollection(WCSimWCDigitsCollection* WCDCPMT, bool remove_hits, TriggerType_t save_triggerType); - - ///sort the Trigger vectors (Time, Type, Info) by Trigger Time - void SortTriggersByTime() { - int i, j; - TriggerType_t index_type; - float index_time; - std::vector index_info; - for (i = 1; i < (int) TriggerTimes.size(); ++i) { - index_time = TriggerTimes[i]; - index_type = TriggerTypes[i]; - index_info = TriggerInfos[i]; - for (j = i; j > 0 && TriggerTimes[j-1] > index_time; j--) { - TriggerTimes[j] = TriggerTimes[j-1]; - TriggerTypes[j] = TriggerTypes[j-1]; - TriggerInfos[j] = TriggerInfos[j-1]; - }//j - TriggerTimes[j] = index_time; - TriggerTypes[j] = index_type; - TriggerInfos[j] = index_info; - }//i - } - - static const double offset; ///< Hit time offset (ns) - static const double LongTime; ///< An arbitrary long time to use in loops (ns) - - bool digitizeCalled; ///< Has Digitize() been called yet? -}; - -// ******************************************* -// CONTAINER CLASS -// ******************************************* - -class WCSimWCDigiTrigger : public G4VDigi -{ -public: - - WCSimWCDigiTrigger(); - ~WCSimWCDigiTrigger(); - WCSimWCDigiTrigger(const WCSimWCDigiTrigger&); - const WCSimWCDigiTrigger& operator=(const WCSimWCDigiTrigger&); - int operator==(const WCSimWCDigiTrigger&) const; - - inline void* operator new(size_t); - inline void operator delete(void*); - - void Draw() {} - void Print(); - - inline void SetTubeID(G4int tube) { tubeID = tube; } - inline void AddGate (G4int gate) { Gates.insert(gate); } - inline void AddPe () { totalPe++; } - inline void SetPe (G4int gate, G4float Q) { pe.insert(std::pair(gate,Q)); } - inline void SetTime (G4int gate, G4float T) { time.insert(std::pair(gate,T)); } - - /// Add a whole vector for one digit to fDigiComp. Clear input vector once added. - void AddDigiCompositionInfo(G4int gate, std::vector &digi_comp){ - fDigiComp.insert(std::pair >(gate, digi_comp)); - digi_comp.clear(); - } - - inline G4int GetTubeID() {return tubeID;} - inline std::vector GetPe (int gate) { return FindInMultimap(gate, pe); } - inline std::vector GetTime (int gate) { return FindInMultimap(gate, time); } - std::vector > GetDigiCompositionInfo(int gate) - { - std::vector > v; - std::multimap >::iterator it = fDigiComp.begin(); - for (; it != fDigiComp.end(); ++it) { - if((it->first) == gate) - v.push_back(it->second); - } - return v; - } - - inline int NumberOfGates() { return Gates.size(); } - inline int NumberOfSubEvents() { return Gates.size() - 1; } - inline bool HasHitsInGate(int gate) { return Gates.count(gate) > 0; } - -private: - G4int tubeID; ///< PMT id of the digit - - std::set Gates; ///< 'Gates' specifies subevent - - //lists (meaning multimap) of information for each digit created on the PMT - std::multimap pe; ///< Digit charge - std::multimap time; ///< Digit time - std::multimap > fDigiComp; ///< Stores the unique IDs of each photon making up a digit - - //integrated hit/digit parameters - G4int totalPe; ///< Total charge on digit - - template std::vector FindInMultimap(const int compare, typename std::multimap &map) - { - typename std::vector v; - typename std::multimap::iterator it = map.begin(); - for (; it != map.end(); ++it) { - if((it->first) == compare) - v.push_back(it->second); - } - return v; - } - -}; - -extern G4Allocator WCSimWCDigiTriggerAllocator; - -inline void* WCSimWCDigiTrigger::operator new(size_t) -{ - void* aDigi; - aDigi = (void*) WCSimWCDigiTriggerAllocator.MallocSingle(); - return aDigi; -} - -inline void WCSimWCDigiTrigger::operator delete(void* aDigi) -{ - WCSimWCDigiTriggerAllocator.FreeSingle((WCSimWCDigiTrigger*) aDigi); -} - - - -// ******************************************* -// DERIVED CLASSES -// ******************************************* - - -/** - * \class WCSimWCTriggerNDigits - * - * \brief A simple NDigits trigger class - * - */ - -class WCSimWCTriggerNDigits : public WCSimWCTriggerBase -{ -public: - - ///Create WCSimWCTriggerNDigits instance with knowledge of the detector and DAQ options - WCSimWCTriggerNDigits(G4String name, WCSimDetectorConstruction*, WCSimWCDAQMessenger*, G4String detectorElement); - - ~WCSimWCTriggerNDigits(); - -private: - ///Calls the workhorse of this class: AlgNDigits - void DoTheWork(WCSimWCDigitsCollection* WCDCPMT); - - bool GetDefaultMultiDigitsPerTrigger() { return false; } ///< SKI saves only earliest digit on a PMT in the trigger window - int GetDefaultNDigitsWindow() { return 200; } ///< SK max light travel time ~200 ns - // int GetDefaultNDigitsThreshold() { return 25; } ///< SK NDigits threshold ~25 - int GetDefaultNDigitsThreshold() { return 20; } ///< SK NDigits threshold ~25 B.Q - int GetDefaultNDigitsPreTriggerWindow() { return -400; } ///< SK SLE trigger window ~-400 - int GetDefaultNDigitsPostTriggerWindow() { return 950; } ///< SK SLE trigger window ~+950 -}; - - -/** - * \class WCSimWCTriggerNDigits2 - * - * \brief An (incomplete) example of running two trigger algorithms, one after the other - * - */ - -class WCSimWCTriggerNDigits2 : public WCSimWCTriggerBase -{ -public: - - //not recommended to override these methods - WCSimWCTriggerNDigits2(G4String name, WCSimDetectorConstruction*, WCSimWCDAQMessenger*, G4String detectorElements); - ~WCSimWCTriggerNDigits2(); - -private: - void DoTheWork(WCSimWCDigitsCollection* WCDCPMT); - - bool GetDefaultMultiDigitsPerTrigger() { return false; } ///< SKI saves only earliest digit on a PMT in the trigger window - int GetDefaultNDigitsWindow() { return 200; } ///< SK max light travel time ~200 ns - // int GetDefaultNDigitsThreshold() { return 50; } ///< 2 * SK NDigits threshold ~25 - int GetDefaultNDigitsThreshold() { return 40; } ///< 2 * SK NDigits threshold ~25 B.Q - int GetDefaultNDigitsPreTriggerWindow() { return -400; } ///< SK SLE trigger window ~-400 - int GetDefaultNDigitsPostTriggerWindow() { return 950; } ///< SK SLE trigger window ~+950 -}; - - - -#endif //WCSimWCTrigger_h diff --git a/libWCSIM/include/evNtuple.h b/libWCSIM/include/evNtuple.h deleted file mode 100644 index b81c597..0000000 --- a/libWCSIM/include/evNtuple.h +++ /dev/null @@ -1,106 +0,0 @@ -// Should at some point replace jhfNtuple, but that is now a global, extern, -// while it should be a member variable. I don't want to break WCSim too much yet. - -// Makes filling branches from EventAction to RunAction very easy, while keeping flat structure. -#include "WCSimEnumerations.hh" - -struct eventNtuple{ - - //event info: - InteractionType_t interaction_mode; // interaction mode - Char_t vtxVolume[100]; // volume of vertex - //already flatten array because have to initialize index by index anyway - double vtx_x; // interaction vertex - double vtx_y; - double vtx_z; - - // Beam and target move to eventInfo from Tracks? - // Both don't have start and stop, only vertex and PiD and direction and flag - // While pGun is one primary with start and stop if generated inside the detector - - - - - //tracks: - int nTracks; // number of final state particles - int pid[100]; // id of ith final state particle - int flag[100]; // flag: -1 = beam particle (incoming neutrino) - // -2 = target - // 0 = pGun or Cosmic primary - // 1 = outgoing lepton - // 2 = most energetic outgoing nucleon - float mass[100]; // mass of ith final state particle - float p[100]; // momentum of ith final state particle - float energy[100]; // energy of ith final state particle - int startvol[100]; // starting volume of ith final state particle - int stopvol[100]; // stopping volume of ith final state particle - float dir_x[100]; // direction of ith final state particle - float dir_y[100]; // direction of ith final state particle - float dir_z[100]; // direction of ith final state particle - float pdir_x[100]; // momentum-vector of ith final state particle - float pdir_y[100]; // momentum-vector of ith final state particle - float pdir_z[100]; // momentum-vector of ith final state particle - float start_x[100]; // starting point of ith final state particle - float start_y[100]; // starting point of ith final state particle - float start_z[100]; // starting point of ith final state particle - - float stop_x[100]; // stopping point of ith final state particle - float stop_y[100]; // stopping point of ith final state particle - float stop_z[100]; // stopping point of ith final state particle - float length[100]; // TrackLength - - int parent[100]; // ID of parent of ith particle (0 if primary) - int trackID[100]; // trackID of ith particle (1 if primary) - float time[100]; // creation time of the ith particle - //ToDo: add distance start to wall and stop to wall - - - - //hits: from WCDC_hits (= WCHC + tagged noise) - // ToDo: extend to OD/ID when OD is implemented - int numTubesHit; // Total number of tubes with hits - int numTubesHit_noNoise; // Total number of tubes with physics hits - int totalNumHits; // hits per event = hits per Tube x numTubesHit - int totalNumHits_noNoise; // hits per event = hits per Tube x numTubesHit - - - int totalPe[10000000]; // The totalPE recorded at each tube - int totalPe_noNoise[10000000];// The totalPE recorded at each tube, without DN - float truetime[10000000]; // The true time of each hit - int vector_index[10000000]; // Index in vector of hits - int tubeid[10000000]; // Readout tube ID - int mPMTid[10000000]; // Readout tube mPMT ID - int mPMT_pmtid[10000000]; // Readout tube mPMT-PMT ID - int parentid[10000000]; // Track ID of originating parent, 1 = initial primary particle ( NOT parentID of parent!!) - int trackid[10000000]; // Track ID - // Optional? - float tube_x[10000000]; // Tube position - float tube_y[10000000]; - float tube_z[10000000]; - float tube_dirx[10000000]; // Hit tube orientation - float tube_diry[10000000]; - float tube_dirz[10000000]; - - - //digits: from WCDC (= WCDC_hits: but triggered and digitized) - // ToDo: extend to OD/ID when OD is implemented - int numDigiTubesHit; // Number of PMTs with digitized hits - int totalNumDigiHits; // Digitized hits per event = hits per Tube x numTubesHit - float sumq; // Sum of q(readout digitized pe) in event - - float q[10000000]; // The readout digitized pe - float t[10000000]; // The readout digitized time - int digivector_index[10000000]; // Index in vector of digits - int digitubeid[10000000]; // Readout tube ID - int digimPMTid[10000000]; // Readout tube mPMT ID - int digimPMT_pmtid[10000000]; // Readout tube mPMT-PMT ID - // Optional? - float digitube_x[10000000]; // Tube position - float digitube_y[10000000]; - float digitube_z[10000000]; - float digitube_dirx[10000000]; // Hit tube orientation - float digitube_diry[10000000]; - float digitube_dirz[10000000]; - // Save DigiCompositionInfo? Not now: ToDo, needs to be a 2D array, even in flat_root - -}; diff --git a/libWCSIM/include/jhfNtuple.h b/libWCSIM/include/jhfNtuple.h deleted file mode 100644 index 332b31f..0000000 --- a/libWCSIM/include/jhfNtuple.h +++ /dev/null @@ -1,89 +0,0 @@ -#include "WCSimEnumerations.hh" -struct ntupleStruct -{ - //int mode; // interaction mode - InteractionType_t mode; // interaction mode - int vtxvol; // volume of vertex - float vtx[3]; // interaction vertex - int vecRecNumber; // info event number in inputvetcotfile - int jmu; // index to muon - int jp; // index to proton - int npar; // number of final state particles - int ipnu[50]; // id of ith final state particle - int flag[50]; // flag: -1 = incoming neutrino - // -2 = target - // 1 = outgoing lepton - // 2 = most energetic outgoing nucleon - float m[50]; // mass of ith final state particle - float p[50]; // momentum of ith final state particle - float E[50]; // energy of ith final state particle - int startvol[50]; // starting volume of ith final state particle - int stopvol[50]; // stopping volume of ith final state particle - float dir[50][3]; // direction of ith final state particle - float pdir[50][3]; // momentum-vector of ith final state particle - float stop[50][3]; // stopping point of ith final state particle - float start[50][3]; // starting point of ith final state particle - int parent[50]; // ID of parent of ith particle (0 if primary) - float time[50]; // creation time of the ith particle - - int numTubesHit; // Total number of tubes with hits - int totalPe[500]; // The totalPE recorded at each tube - float truetime[500][90]; // The true time of each hit - - int numDigitizedTubes; // Number of PMTs with digitized hits - float q[500]; // The readout digitized pe - float t[500]; // The readout digitized time - int tubeid[500]; // The readout tube ID - float sumq; // sum of q(readout digitized pe) in event - - int fvnumTubesHit; // Total number of inner tubes with hits - int fvtotalPe[500]; // The totalPE recorded at each innertube - float fvtruetime[500][90]; // The true time of each inner hit - - int fvnumDigitizedTubes; // Number of PMTs with digitized hits - float fvq[500]; // The readout digitized pe - float fvt[500]; // The readout digitized time - int fvtubeid[500]; // The readout tube ID - float fvsumq; // sum of q(readout digitized pe) in event -}; - -extern struct ntupleStruct jhfNtuple; - -/* Not used: -static const char* ntDesc = -"mode:I," -"vtxvol:I," -"vtx(3):R," -"vecRecNumber:I," -"jmu:I," -"jp:I," -"npar[0,50]:I," -"ipnu(npar):I," -"flag(npar):I," -"m(npar):R," -"p(npar):R," -"E(npar):R," -"startvol(npar):I," -"stopvol(npar):I," -"dir(3,npar):R," -"pdir(3,npar):R," -"stoppos(3,npar):R," -"startpos(3,npar):R," -"parent(npar):I," -"time(npar):R," -"numTubesHit[0,500]:I," -"totalPe(numTubesHit):I," -"truetime(90,numTubesHit):R," -"numDigTubes[0,500]:I," -"q(numDigTubes):R," -"t(numDigTubes):R," -"tubeid(numDigTubes):I" -"fvnumTubesHit[0,500]:I," -"fvtotalPe(fvnumTubesHit):I," -"fvtruetime(90,fvnumTubesHit):R," -"fvnumDigTubes[0,500]:I," -"fvq(fvnumDigTubes):R," -"fvt(fvnumDigTubes):R," -"fvtubeid(fvnumDigTubes):I"; - -*/ diff --git a/libWCSIM/libWCSimRoot.so b/libWCSIM/libWCSimRoot.so deleted file mode 100755 index b90a46d..0000000 Binary files a/libWCSIM/libWCSimRoot.so and /dev/null differ diff --git a/libWCSIM/src/#WCSimConstructGeometryTables.cc# b/libWCSIM/src/#WCSimConstructGeometryTables.cc# deleted file mode 100644 index 7015dce..0000000 --- a/libWCSIM/src/#WCSimConstructGeometryTables.cc# +++ /dev/null @@ -1,494 +0,0 @@ -#include "WCSimDetectorConstruction.hh" -#include "WCSimPmtInfo.hh" - -#include "G4Material.hh" -#include "G4Element.hh" -#include "G4Box.hh" -#include "G4LogicalVolume.hh" -#include "G4VPhysicalVolume.hh" -#include "G4PVPlacement.hh" -#include "G4ThreeVector.hh" -#include "G4Vector3D.hh" -#include "globals.hh" -#include "G4VisAttributes.hh" -#include "G4Tubs.hh" -#include "G4Sphere.hh" -#include "G4VPVParameterisation.hh" - -#include "G4PhysicalConstants.hh" -#include "G4SystemOfUnits.hh" - -#include -#include - - -using std::setw; -// These routines are object registration routines that you can pass -// to the traversal code. - -void WCSimDetectorConstruction::PrintGeometryTree -(G4VPhysicalVolume* aPV ,int aDepth, int /*replicaNo*/, - const G4Transform3D& aTransform) -{ - for (int levels = 0; levels < aDepth; levels++) G4cout << " "; - G4cout << aPV->GetName() << " Level:" << aDepth - << " Pos:" << aTransform.getTranslation() - << " Rot:" << aTransform.getRotation().getTheta()/deg - << "," << aTransform.getRotation().getPhi()/deg - << "," << aTransform.getRotation().getPsi()/deg - << G4endl; -} - -void WCSimDetectorConstruction::GetWCGeom -(G4VPhysicalVolume* aPV ,int aDepth, int /*replicaNo*/, - const G4Transform3D& aTransform) -{ - - // Grab mishmash of useful information from the tree while traversing it - // This information will later be written to the geometry file - // (Alternatively one might define accessible constants) - - if ((aPV->GetName() == "WCBox")) { // last condition is the HyperK Envelope name. - // Stash info in data member - WCOffset = G4ThreeVector(aTransform.getTranslation().getX()/cm, - aTransform.getTranslation().getY()/cm, - aTransform.getTranslation().getZ()/cm); - WCXRotation = G4ThreeVector(aTransform.getRotation().xx(), aTransform.getRotation().xy(), aTransform.getRotation().xz()); - WCYRotation = G4ThreeVector(aTransform.getRotation().yx(), aTransform.getRotation().yy(), aTransform.getRotation().yz()); - WCZRotation = G4ThreeVector(aTransform.getRotation().zx(), aTransform.getRotation().zy(), aTransform.getRotation().zz()); - } - - // Stash info in data member - // AH Need to store this in CM for it to be understood by SK code - WCPMTSize = WCPMTRadius/cm;// I think this is just a variable no if needed - - // Note WC can be off-center... get both extremities - static G4float zmin=100000,zmax=-100000.; - static G4float xmin=100000,xmax=-100000.; - static G4float ymin=100000,ymax=-100000.; - if (aDepth == 0) { // Reset for this traversal - xmin=100000,xmax=-100000.; - ymin=100000,ymax=-100000.; - zmin=100000,zmax=-100000.; - } - - if ((aPV->GetName() == "WCCapBlackSheet") || (aPV->GetName().find("glassFaceWCPMT") != std::string::npos)){ - G4float x = aTransform.getTranslation().getX()/cm; - G4float y = aTransform.getTranslation().getY()/cm; - G4float z = aTransform.getTranslation().getZ()/cm; - - if (xxmax){xmax=x;} - - if (yymax){ymax=y;} - - if (zzmax){zmax=z;} - - WCCylInfo[0] = xmax-xmin; - WCCylInfo[1] = ymax-ymin; - WCCylInfo[2] = zmax-zmin; - // G4cout << "determine height: " << zmin << " " << zmax << " " << aPV->GetName()<<" " << z << G4endl; - } -} - -void WCSimDetectorConstruction::DescribeAndRegisterPMT(G4VPhysicalVolume* aPV ,int aDepth, int replicaNo, - const G4Transform3D& aTransform) -{ - static std::string replicaNoString[20]; - - std::stringstream depth; - std::stringstream pvname; - - depth << replicaNo; - pvname << aPV->GetName(); - - replicaNoString[aDepth] = pvname.str() + "-" + depth.str(); - - - //TF: To Consider: add a separate table for mPMT positions? Need to use its orientation anyway - // Could be useful for the near future. Need to add an == WCMultiPMT here then. - if (aPV->GetName()== WCIDCollectionName || aPV->GetName()== WCIDCollectionName2 ||aPV->GetName()== WCODCollectionName ) - { - - // First increment the number of PMTs in the tank. - //totalNumPMTs++; - if(aPV->GetName()== WCIDCollectionName) totalNumPMTs++; - else if(aPV->GetName()== WCIDCollectionName2) totalNumPMTs2++; - - // Put the location of this tube into the location map so we can find - // its ID later. It is coded by its tubeTag string. - // This scheme must match that used in WCSimWCSD::ProcessHits() - - std::string tubeTag; - int mPMT_pmtno = -1; - bool foundString = false; - // - - for (int i=0; i <= aDepth; i++){ - tubeTag += ":" + replicaNoString[i]; - std::string::size_type position = replicaNoString[i].find("pmt-"); - if( position == 0) { - foundString = true; - mPMT_pmtno = atoi(replicaNoString[i].substr(position+4).c_str())+1; - if(mPMT_pmtno == 1) - if(aPV->GetName()== WCIDCollectionName) totalNum_mPMTs++; - else if(aPV->GetName()== WCIDCollectionName2) totalNum_mPMTs2++; - } - } - if(!foundString){ - // to distinguish mPMT PMTs from single PMTs: - mPMT_pmtno = 0; - if(aPV->GetName()== WCIDCollectionName) totalNum_mPMTs++; - else if(aPV->GetName()== WCIDCollectionName2) totalNum_mPMTs2++; - } - - // G4cout << tubeTag << G4endl; - if(aPV->GetName()== WCIDCollectionName){ - if ( tubeLocationMap.find(tubeTag) != tubeLocationMap.end() ) { - G4cerr << "Repeated tube tag: " << tubeTag << G4endl; - G4cerr << "Assigned to both tube #" << tubeLocationMap[tubeTag] << " and #" << totalNumPMTs << G4endl; - G4cerr << "Cannot continue -- hits will not be recorded correctly." << G4endl; - G4cerr << "Please make sure that logical volumes with multiple placements are each given a unique copy number" << G4endl; - assert(false); - } - G4cout << "First PMT type, the tube is " << tubeTag << ", number of tubes = " << totalNumPMTs << ", Num mPMT/mPMT = " << totalNum_mPMTs << ", PMT number in mPMT = " << mPMT_pmtno << G4endl; - tubeLocationMap[tubeTag] = totalNumPMTs; - // Put the transform for this tube into the map keyed by its ID - tubeIDMap[totalNumPMTs] = aTransform; - - - mPMTIDMap[totalNumPMTs] = std::make_pair(totalNum_mPMTs,mPMT_pmtno); - //G4cout << "depth " << depth.str() << G4endl; - //G4cout << "tubeLocationmap[" << tubeTag << "]= " << tubeLocationMap[tubeTag] << "\n"; - - // Print - // G4cout << "Tube: "<GetName()== WCIDCollectionName2){ - if ( tubeLocationMap2.find(tubeTag) != tubeLocationMap2.end() ) { - G4cerr << "Repeated tube tag: " << tubeTag << G4endl; - G4cerr << "Assigned to both tube #" << tubeLocationMap2[tubeTag] << " and #" << totalNumPMTs2 << G4endl; - G4cerr << "Cannot continue -- hits will not be recorded correctly." << G4endl; - G4cerr << "Please make sure that logical volumes with multiple placements are each given a unique copy number" << G4endl; - assert(false); - } - tubeLocationMap2[tubeTag] = totalNumPMTs2; - // Put the transform for this tube into the map keyed by its ID - tubeIDMap2[totalNumPMTs2] = aTransform; - - - G4cout << "Second PMT type, the tube is " << tubeTag << ", number of tubes = " << totalNumPMTs2 << ", Num mPMT/mPMT = " << totalNum_mPMTs2 << ", PMT number in mPMT = " << mPMT_pmtno << G4endl; - mPMTIDMap2[totalNumPMTs2] = std::make_pair(totalNum_mPMTs2,mPMT_pmtno); - //G4cout << "depth " << depth.str() << G4endl; - //G4cout << "tubeLocationmap[" << tubeTag << "]= " << tubeLocationMap[tubeTag] << "\n"; - - // Print - // G4cout << "Tube: "< 0)//veto pmt - {cylLocation=3;} - else if (pmtOrientation.z()==1.0)//bottom - {cylLocation=2;} - else if (pmtOrientation.z()==-1.0)//top - {cylLocation=0;} - else // barrel - {cylLocation=1;} - - G4cout << "Debug B.Q: First type of PMT implemented, number = " << totalNumPMTs << ", Tube ID = " << tubeID << ",Number of the PMT/mPMT is = " << mPMTIDMap[tubeID].first << " and numbering of the PMT in the mPMT is = " << mPMTIDMap[tubeID].second << G4endl; - geoFile.precision(9); - geoFile << setw(4) << tubeID - << " " << setw(4) << mPMTIDMap[tubeID].first - << " " << setw(4) << mPMTIDMap[tubeID].second - << " " << setw(8) << newTransform.getTranslation().getX()/cm - << " " << setw(8) << newTransform.getTranslation().getY()/cm - << " " << setw(8) << newTransform.getTranslation().getZ()/cm - << " " << setw(7) << pmtOrientation.x() - << " " << setw(7) << pmtOrientation.y() - << " " << setw(7) << pmtOrientation.z() - << " " << setw(3) << cylLocation - << G4endl; - - WCSimPmtInfo *new_pmt = new WCSimPmtInfo(cylLocation, - newTransform.getTranslation().getX()/cm, - newTransform.getTranslation().getY()/cm, - newTransform.getTranslation().getZ()/cm, - pmtOrientation.x(), - pmtOrientation.y(), - pmtOrientation.z(), - tubeID, - mPMTIDMap[tubeID].first, - mPMTIDMap[tubeID].second); - - fpmts.push_back(new_pmt); - - } - - //Record location of the second PMT type for the hybrid configuration - for (unsigned int i=0;i 0)//veto pmt - {cylLocation=3;} - else if (pmtOrientation.z()==1.0)//bottom - {cylLocation=2;} - else if (pmtOrientation.z()==-1.0)//top - {cylLocation=0;} - else // barrel - {cylLocation=1;} - - G4cout << "Debug B.Q: Second type of PMT implemented, number = " << totalNumPMTs2 << ", Tube ID = " << tubeID << ",Number of the PMT/mPMT is = " << mPMTIDMap2[tubeID].first << " and numbering of the PMT in the mPMT is = " << mPMTIDMap2[tubeID].second << G4endl; - - geoFile.precision(9); - geoFile << setw(4) << tubeID - << " " << setw(4) << mPMTIDMap2[tubeID].first - << " " << setw(4) << mPMTIDMap2[tubeID].second - << " " << setw(8) << newTransform.getTranslation().getX()/cm - << " " << setw(8) << newTransform.getTranslation().getY()/cm - << " " << setw(8) << newTransform.getTranslation().getZ()/cm - << " " << setw(7) << pmtOrientation.x() - << " " << setw(7) << pmtOrientation.y() - << " " << setw(7) << pmtOrientation.z() - << " " << setw(3) << cylLocation - << G4endl; - - WCSimPmtInfo *new_pmt = new WCSimPmtInfo(cylLocation, - newTransform.getTranslation().getX()/cm, - newTransform.getTranslation().getY()/cm, - newTransform.getTranslation().getZ()/cm, - pmtOrientation.x(), - pmtOrientation.y(), - pmtOrientation.z(), - tubeID, - mPMTIDMap2[tubeID].first, - mPMTIDMap2[tubeID].second); - - fpmts2.push_back(new_pmt); - - } - - geoFile.close(); - - std::cout << "Geofile written" << std::endl; -} - - -// Code for traversing the geometry tree. This code is very general you pass -// it a function and it will call the function with the information on each -// object it finds. -// -// The traversal code comes from a combination of me/G4Lab project & -// from source/visualization/modeling/src/G4PhysicalVolumeModel.cc -// -// If you are trying to understand how passing the function works you need -// to understand pointers to member functions... -// -// Also notice that DescriptionFcnPtr is a (complicated) typedef. -// - -void WCSimDetectorConstruction::TraverseReplicas -(G4VPhysicalVolume* aPV, int aDepth, const G4Transform3D& aTransform, - DescriptionFcnPtr registrationRoutine) -{ - // Recursively visit all of the geometry below the physical volume - // pointed to by aPV including replicas. - - G4ThreeVector originalTranslation = aPV->GetTranslation(); - G4RotationMatrix* pOriginalRotation = aPV->GetRotation(); - - if (aPV->IsReplicated() ) - { - EAxis axis; - G4int nReplicas; - G4double width, offset; - G4bool consuming; - - aPV->GetReplicationData(axis,nReplicas,width,offset,consuming); - - for (int n = 0; n < nReplicas; n++) - { - //TF: for mPMT parameterization - // because kZAxis is a dummy value altered by Parameterised volume in MultiPMTParam - if(aPV->IsParameterised() ){ - //G4cout << "Replica No. " << n << G4endl; - //use the actual translation and rotation of the replica. - (aPV->GetParameterisation())->ComputeTransformation(n,aPV); - } - - else{ - - switch(axis) { - default: - case kXAxis: - aPV->SetTranslation(G4ThreeVector - (-width*(nReplicas-1)*0.5+n*width,0,0)); - aPV->SetRotation(0); - break; - case kYAxis: - aPV->SetTranslation(G4ThreeVector - (0,-width*(nReplicas-1)*0.5+n*width,0)); - aPV->SetRotation(0); - break; - case kZAxis: - aPV->SetTranslation(G4ThreeVector - (0,0,-width*(nReplicas-1)*0.5+n*width)); - aPV->SetRotation(0); - break; - case kRho: - //Lib::Out::putL("GeometryVisitor::visit: WARNING:"); - //Lib::Out::putL(" built-in replicated volumes replicated"); - //Lib::Out::putL(" in radius are not yet properly visualizable."); - aPV->SetTranslation(G4ThreeVector(0,0,0)); - aPV->SetRotation(0); - break; - case kPhi: - { - G4RotationMatrix rotation; - rotation.rotateZ(-(offset+(n+0.5)*width)); - // Minus Sign because for the physical volume we need the - // coordinate system rotation. - aPV->SetTranslation(G4ThreeVector(0,0,0)); - aPV->SetRotation(&rotation); - } - break; - - } // axis switch - } - DescribeAndDescendGeometry(aPV, aDepth, n, aTransform, - registrationRoutine); - - } // num replicas for loop - } // if replicated - else - DescribeAndDescendGeometry(aPV, aDepth, aPV->GetCopyNo(), aTransform, - registrationRoutine); - - // Restore original transformation... - aPV->SetTranslation(originalTranslation); - aPV->SetRotation(pOriginalRotation); -} - -void WCSimDetectorConstruction::DescribeAndDescendGeometry -(G4VPhysicalVolume* aPV ,int aDepth, int replicaNo, - const G4Transform3D& aTransform, DescriptionFcnPtr registrationRoutine) -{ - - // Calculate the new transform relative to the old transform - G4Transform3D* transform = - new G4Transform3D(*(aPV->GetObjectRotation()), aPV->GetTranslation()); - - G4Transform3D newTransform = aTransform * (*transform); - delete transform; - - /* - G4cout << aPV->GetObjectRotation()->getPhi() << " " << aPV->GetObjectRotation()->getTheta() << " " - << aPV->GetObjectRotation()->getPsi() << G4endl; - G4cout << aPV->GetTranslation().x() << " " << aPV->GetTranslation().y() << " " - << aPV->GetTranslation().z() << G4endl; - */ - - // Call the routine we use to print out geometry descriptions, make - // tables, etc. The routine was passed here as a parameter. It needs to - // be a member function of the class - - (this->*registrationRoutine)(aPV, aDepth, replicaNo, newTransform); - - int nDaughters = aPV->GetLogicalVolume()->GetNoDaughters(); - - for (int iDaughter = 0; iDaughter < nDaughters; iDaughter++) - TraverseReplicas(aPV->GetLogicalVolume()->GetDaughter(iDaughter), - aDepth+1, newTransform, registrationRoutine); -} - - - -G4double WCSimDetectorConstruction::GetGeo_Dm(G4int i){ - if (i>=0&&i<=2){ - return WCCylInfo[i]; - }else if(i==3){ - return innerradius; - }else{ - return 0; - } -} diff --git a/libWCSIM/src/#WCSimWCDigitizer.cc# b/libWCSIM/src/#WCSimWCDigitizer.cc# deleted file mode 100644 index 28ddac4..0000000 --- a/libWCSIM/src/#WCSimWCDigitizer.cc# +++ /dev/null @@ -1,384 +0,0 @@ -#include "WCSimWCDigitizer.hh" -#include "WCSimWCPMT.hh" -#include "WCSimWCDigi.hh" -#include "WCSimWCHit.hh" -#include "G4EventManager.hh" -#include "G4Event.hh" -#include "G4SDManager.hh" -#include "G4DigiManager.hh" -#include "G4ios.hh" -#include "G4RotationMatrix.hh" -#include "G4ThreeVector.hh" - -#include "WCSimDetectorConstruction.hh" -#include "WCSimPmtInfo.hh" -#include "WCSimDarkRateMessenger.hh" - -#include -// for memset -#include - -#ifndef NPMTS_VERBOSE -#define NPMTS_VERBOSE 10 -#endif - - -// ******************************************* -// BASE CLASS -// ******************************************* - -#ifndef WCSIMWCDIGITIZER_VERBOSE -#define WCSIMWCDIGITIZER_VERBOSE -#endif - -WCSimWCDigitizerBase::WCSimWCDigitizerBase(G4String name, - WCSimDetectorConstruction* inDetector, - WCSimWCDAQMessenger* myMessenger, - DigitizerType_t digitype, - G4String detectorElement) - :G4VDigitizerModule(name), myDetector(inDetector), DAQMessenger(myMessenger), DigitizerType(digitype), DigitizerClassName(""), detectorElement(detectorElement) -{ - G4String colName; - if(detectorElement=="tank") colName = "WCDigitizedStoreCollection"; - else if(detectorElement=="tankPMT2") colName = "WCDigitizedStoreCollection2"; - collectionName.push_back(colName); - ReInitialize(); - - if(DAQMessenger == NULL) { - G4cerr << "WCSimWCDAQMessenger pointer is NULL when passed to WCSimWCDigitizerBase constructor. Exiting..." - << G4endl; - exit(-1); - } -} - -WCSimWCDigitizerBase::~WCSimWCDigitizerBase(){ -} - -void WCSimWCDigitizerBase::GetVariables() -{ - //set the options to digitizer-specific defaults - DigitizerDeadTime = GetDefaultDeadTime(); - DigitizerIntegrationWindow = GetDefaultIntegrationWindow(); - - //read the .mac file to override them - if(DAQMessenger != NULL) { - DAQMessenger->TellMeAboutTheDigitizer(this); - DAQMessenger->SetDigitizerOptions(); - } - else { - G4cerr << "WCSimWCDAQMessenger pointer is NULL when used in WCSimWCDigitizerBase::GetVariables(). Exiting..." - << G4endl; - exit(-1); - } - - G4cout << "Using digitizer deadtime " << DigitizerDeadTime << " ns" << G4endl; - G4cout << "Using digitizer integration window " << DigitizerIntegrationWindow << " ns" << G4endl; -} - -void WCSimWCDigitizerBase::Digitize() -{ - //Input is WCSimWCDigitsCollection with raw PMT hits (photon + dark noise) - //Output is WCSimWCDigitsCollection with digitied PMT hits - - //Clear the DigiStoreHitMap - ReInitialize(); - - //Temporary Storage of Digitized hits which is passed to the trigger - DigiStore = new WCSimWCDigitsCollection(collectionName[0],collectionName[0]); - - G4DigiManager* DigiMan = G4DigiManager::GetDMpointer(); - - - // Get the PMT collection ID - G4String rawcollectionName; - if(detectorElement=="tank") rawcollectionName = "WCRawPMTSignalCollection"; - else if(detectorElement=="tankPMT2") rawcollectionName = "WCRawPMTSignalCollection2"; - G4int WCHCID = DigiMan->GetDigiCollectionID(rawcollectionName); - - // Get the PMT Digits collection - WCSimWCDigitsCollection* WCHCPMT = - (WCSimWCDigitsCollection*)(DigiMan->GetDigiCollection(WCHCID)); - - if (WCHCPMT) { - DigitizeHits(WCHCPMT); - } - - StoreDigiCollection(DigiStore); - -} - -bool WCSimWCDigitizerBase::AddNewDigit(int tube, int gate, float digihittime, float peSmeared, std::vector digi_comp) -{ - //gate is not a trigger, but just the position of the digit in the array - //inside the WCSimWCDigi object -#ifdef WCSIMWCDIGITIZER_VERBOSE - if(tube < NPMTS_VERBOSE) { - G4cout<<"Adding hit "< 0.0) { - if ( DigiStoreHitMap[tube] == 0) { - WCSimWCDigi* Digi = new WCSimWCDigi(); - Digi->SetTubeID(tube); - Digi->SetPe(gate,peSmeared); - Digi->AddPe(digihittime); - Digi->SetTime(gate,digihittime); - Digi->AddDigiCompositionInfo(digi_comp); - DigiStoreHitMap[tube] = DigiStore->insert(Digi); -#ifdef WCSIMWCDIGITIZER_VERBOSE - if(tube < NPMTS_VERBOSE) - G4cout << " NEW HIT" << G4endl; -#endif - } - else { - (*DigiStore)[DigiStoreHitMap[tube]-1]->SetPe(gate,peSmeared); - (*DigiStore)[DigiStoreHitMap[tube]-1]->SetTime(gate,digihittime); - (*DigiStore)[DigiStoreHitMap[tube]-1]->AddPe(digihittime); - (*DigiStore)[DigiStoreHitMap[tube]-1]->AddDigiCompositionInfo(digi_comp); -#ifdef WCSIMWCDIGITIZER_VERBOSE - if(tube < NPMTS_VERBOSE) - G4cout << " DEJA VU" << G4endl; -#endif - } - return true; - }//peSmeared > 0 - else { -#ifdef WCSIMWCDIGITIZER_VERBOSE - if(tube < NPMTS_VERBOSE) - G4cout << "DIGIT REJECTED with charge " << peSmeared - << " time " << digihittime << G4endl; -#endif - return false; - } -} - -void WCSimWCDigitizerBase::SaveOptionsToOutput(WCSimRootOptions * wcopt) -{ - wcopt->SetDigitizerClassName(DigitizerClassName); - wcopt->SetDigitizerDeadTime(DigitizerDeadTime); - wcopt->SetDigitizerIntegrationWindow(DigitizerIntegrationWindow); -} - - -// ******************************************* -// DERIVED CLASS -// ******************************************* - -WCSimWCDigitizerSKI::WCSimWCDigitizerSKI(G4String name, - WCSimDetectorConstruction* myDetector, - WCSimWCDAQMessenger* myMessenger, - G4String detectorElement) - : WCSimWCDigitizerBase(name, myDetector, myMessenger, kDigitizerSKI, detectorElement) -{ - DigitizerClassName = "SKI"; - GetVariables(); -} - -WCSimWCDigitizerSKI::~WCSimWCDigitizerSKI(){ -} - -void WCSimWCDigitizerSKI::DigitizeHits(WCSimWCDigitsCollection* WCHCPMT) { - G4cout << "WCSimWCDigitizerSKI::DigitizeHits START WCHCPMT->entries() = " << WCHCPMT->entries() << G4endl; - - double BQcounterInTime = 0; - double BQcounterInTimeAll = 0; - double BQcounterThreshold = 0; - double BQcounterThresholdAll = 0; - double BQcounterTotalCharge = 0; - //loop over entires in WCHCPMT, each entry corresponds to - //the photons on one PMT - for (G4int i = 0 ; i < WCHCPMT->entries() ; i++) - { - - //We must first sort hits by PMT in time. This is very important as the code - //assumes that each hit is in time order from lowest to highest. - (*WCHCPMT)[i]->SortArrayByHitTime(); - int tube = (*WCHCPMT)[i]->GetTubeID(); -#ifdef WCSIMWCDIGITIZER_VERBOSE - if(tube < NPMTS_VERBOSE) { - G4cout << "tube " << tube - << " totalpe = " << (*WCHCPMT)[i]->GetTotalPe() - << " times"; - for(int ip = 0; ip < (*WCHCPMT)[i]->GetTotalPe(); ip++) - G4cout << " " << (*WCHCPMT)[i]->GetTime(ip); - /* - G4cout<<" parents =\t"; - for( G4int ip = 0 ; ip < (*WCHCPMT)[i]->GetTotalPe() ; ip++) - G4cout << " " << (*WCHCPMT)[i]->GetParentID(ip); - */ - G4cout < digi_comp; - - BQcounterTotalCharge+=(*WCHCPMT)[i]->GetTotalPe(); - //loop over the hits on this PMT - for( G4int ip = 0 ; ip < (*WCHCPMT)[i]->GetTotalPe() ; ip++) - { - float time = (*WCHCPMT)[i]->GetTime(ip); - float pe = (*WCHCPMT)[i]->GetPe(ip); - - BQcounterInTimeAll++; - //start the integration time as the time of the first hit - //Hits must be sorted in time - if(ip==0) { - intgr_start=time; - peSmeared = 0; - //Set the limits of the integration window [intgr_start,upperlimit] - upperlimit = intgr_start + DigitizerIntegrationWindow; - } - -#ifdef WCSIMWCDIGITIZER_VERBOSE - if(tube < NPMTS_VERBOSE) - G4cout << "ip " << ip - << " pe " << pe - << " time " << time - << " intgr_start " << intgr_start - << " upperlimit " << upperlimit - << G4endl; -#endif - - bool MakeDigit = false; - if(time >= intgr_start && time <= upperlimit) { - peSmeared += pe; - photon_unique_id = ip; - digi_comp.push_back(photon_unique_id); - -#ifdef WCSIMWCDIGITIZER_VERBOSE - if(tube < NPMTS_VERBOSE) - G4cout<<"INFO: time "<GetTotalPe()){ - MakeDigit = true; - } - BQcounterInTime ++; - } - //if ensures we don't append the same digit multiple times while in the integration window - else if(digi_comp.size()) { - //this hit is outside the integration time window. - //Charge integration is over. The is now a DigitizerDeadTime ns dead - //time period where no hits can be recorded - MakeDigit = true; - } - - //Make digit here - if(MakeDigit) { - int iflag; - WCSimWCDigitizerSKI::Threshold(peSmeared,iflag); - - BQcounterThresholdAll+=peSmeared; - - //Check if previous hit passed the threshold. If so we will digitize the hit - if(iflag == 0) { - //digitize hit - peSmeared *= efficiency; - bool accepted = WCSimWCDigitizerBase::AddNewDigit(tube, digi_unique_id, intgr_start, peSmeared, digi_comp); - if(accepted) { - digi_unique_id++; - BQcounterThreshold+=peSmeared; - } - assert(digi_comp.size()); - digi_comp.clear(); - } - else { - //reject hit -#ifdef WCSIMWCDIGITIZER_VERBOSE - if(tube < NPMTS_VERBOSE) - G4cout << "DIGIT REJECTED with time " << intgr_start << G4endl; -#endif - digi_comp.clear(); - } - } - - //Now try and deal with the next hit - if(time > upperlimit && time <= upperlimit + DigitizerDeadTime) { - //Now we need to reject hits that are after the integration - //period to the end of the veto signal - continue; - } - else if(time > upperlimit + DigitizerDeadTime){ -#ifdef WCSIMWCDIGITIZER_VERBOSE - if(tube < NPMTS_VERBOSE) - G4cout<<"*** PREPARING FOR >1 DIGI ***"<GetTotalPe()) { - int iflag; - WCSimWCDigitizerSKI::Threshold(peSmeared,iflag); - if(iflag == 0) { - //digitize hit - peSmeared *= efficiency; - bool accepted = WCSimWCDigitizerBase::AddNewDigit(tube, digi_unique_id, intgr_start, peSmeared, digi_comp); - if(accepted) { - digi_unique_id++; - } - assert(digi_comp.size()); - digi_comp.clear(); - } - else { - //reject hit -#ifdef WCSIMWCDIGITIZER_VERBOSE - if(tube < NPMTS_VERBOSE) - G4cout << "DIGIT REJECTED with time " << intgr_start << G4endl; -#endif - digi_comp.clear(); - } - } - } - }//ip (totalpe) - }//i (WCHCPMT->entries()) - G4cout << "B.Q Debug, Proportions of pe in time = " << BQcounterInTime/(BQcounterInTimeAll==0?1:BQcounterInTimeAll) << ", and passing threshold = " << BQcounterThreshold/(BQcounterThresholdAll==0?1:BQcounterThresholdAll) << G4endl; - G4cout << "B.Q Debug, Total Charge = " << BQcounterTotalCharge << ", charge passing cuts = " << BQcounterThreshold << G4endl; - G4cout<<"WCSimWCDigitizerSKI::DigitizeHits END DigiStore->entries() " << DigiStore->entries() << "\n"; - -#ifdef WCSIMWCDIGITIZER_VERBOSE - G4cout<<"\n\n\nCHECK DIGI COMP:"<entries() ; idigi++){ - int tubeid = (*DigiStore)[idigi]->GetTubeID(); - if(tubeid < NPMTS_VERBOSE) { - std::map< int, std::vector > comp = (*DigiStore)[idigi]->GetDigiCompositionInfo(); - for(size_t i = 0; i < comp.size(); i++){ - G4cout << "tube " << tubeid - << " gate " << i << " p_id"; - for(size_t iv = 0; iv < comp[i].size(); iv++) { - G4cout << " " << comp[i][iv]; - }//iv - G4cout << G4endl; - }//i - } - }//idigi -#endif -} - diff --git a/libWCSIM/src/GdCaptureGammas.cc b/libWCSIM/src/GdCaptureGammas.cc deleted file mode 100755 index 7277cb1..0000000 --- a/libWCSIM/src/GdCaptureGammas.cc +++ /dev/null @@ -1,263 +0,0 @@ - -/////////////////////////////////////////////////////////////////////////////// -// Spectrum of radiative neutron capture by Gadolinium -// version 1.0.0 -// (Sep.09.2005) - -// Author : karim.zbiri@subatech.in2p3.fr - -//This file contains the gammas spectrum produced in radiative capture of -//neutrons by gadolinium. -//This work is adapted from earlier work in geant3 for chooz 1. - -//First version by Karim Zbiri, April, 2005 -/////////////////////////////////////////////////////////////////////////////// - - -#include "GdCaptureGammas.hh" -#include "Randomize.hh" -#include -#include "G4Gamma.hh" -using namespace std; - -GdCaptureGammas::GdCaptureGammas () {} - -GdCaptureGammas::~GdCaptureGammas () {} - -G4ReactionProductVector * GdCaptureGammas::GetGammas(G4int isotopeA) -{ - G4ReactionProductVector * theGammas = new G4ReactionProductVector; - vector nrj = Initialize(isotopeA); - for(unsigned int i=0; iSetDefinition( G4Gamma::Gamma() ); - - G4double costheta = 2.*G4UniformRand()-1; - G4double theta = acos(costheta); - G4double phi = CLHEP::twopi*G4UniformRand(); - G4double sinth = sin(theta); - - theOne->SetTotalEnergy( nrj[i] ); - G4ThreeVector temp(nrj[i]*sinth*cos(phi), nrj[i]*sinth*sin(phi),nrj[i]*cos(theta) ); - theOne->SetMomentum( temp ) ; - theGammas->push_back(theOne); - } - return theGammas; -} - -vector GdCaptureGammas::Initialize(G4int isotopeA) -{ - vector Eg; - //double DefElem = G4UniformRand()*48.80; - //if(DefElem < 39.83) - if(isotopeA == 157) - Eg = CapGad157(); - else if(isotopeA == 155) - Eg = CapGad155(); - else { - G4cerr << "Unexpected Gd isotope A=" << isotopeA << ". Using 157 instead." << G4endl; - Eg = CapGad157(); - } - - return Eg; -} - -vector GdCaptureGammas::CapGad155() -{ - // gammas from GAD155 - // either 2 gammas - // either a continuum - // total energy = 8.46 MeV - - Elevel = 8.46; - double Nlevel = G4UniformRand(); - vector Egammas; - - if(Nlevel > 0.023) Egammas=continuum(); - else if (Nlevel >= 0.013) Egammas=casca1Gd155(); - else Egammas=casca2Gd155(); - - return Egammas; -} - -vector GdCaptureGammas::CapGad157() -{ - // gammas from GAD155 - // either 2 gammas - // either a continuum - // total energy = 7.87 MeV - - Elevel = 7.87; - double Nlevel = G4UniformRand(); - vector Egammas; - - if(Nlevel > 0.068) Egammas=continuum(); - else if(Nlevel < 0.037) Egammas=casca1Gd157(); - else if(Nlevel < 0.05) Egammas=casca2Gd157(); - else if(Nlevel >= 0.05) Egammas=casca3Gd157(); - - return Egammas; -} - -vector GdCaptureGammas::casca1Gd155() -{ - vector energy; - energy.push_back(7.33); - energy.push_back(1.17); - return energy; -} - -vector GdCaptureGammas::casca2Gd155() -{ - vector energy; - energy.push_back(6.44); - energy.push_back(2.18); - return energy; -} - -vector GdCaptureGammas::casca1Gd157() -{ - vector energy; - energy.push_back(6.74); - energy.push_back(1.11); - return energy; -} - -vector GdCaptureGammas::casca2Gd157() -{ - vector energy; - energy.push_back(5.62); - energy.push_back(2.25); - return energy; -} - -vector GdCaptureGammas::casca3Gd157() -{ - vector energy; - energy.push_back(5.88); - energy.push_back(1.99); - return energy; -} - - -vector GdCaptureGammas::continuum() -{ - // continuum part of gadolinium - //cross sections - vector energy; - const double eps=0.0001,bint=0.01; - const int ngadbin=750; - float sigfond,sigex1,sigftot,sigcont,sigtot; - double dexlevel,auxe,hasard,cint,Eneuve; - int isigma; - do - { - //first excited level and ground level - sigfond=pow(Elevel,3); - sigex1=pow(Elevel-1,3); - sigftot=sigfond+sigex1; - - //continuum - auxe=(Elevel-2.)/bint; - isigma=int(auxe+eps); - sigcont=cumule(isigma); - sigtot=sigfond+sigex1+sigcont; - dexlevel=G4UniformRand()*sigtot; - if(dexlevel < sigfond) - { - energy.push_back(Elevel); Elevel =0.; - return energy;} - if((dexlevel>= sigfond) && (dexlevel <= sigftot)) { - energy.push_back(Elevel-1); - energy.push_back(1); - Elevel=0.; - return energy;} - if(dexlevel > sigftot) - { - hasard=G4UniformRand()*sigcont; - for(int i=0;ihasard) - { - if(Elevel>Eneuve) - { - energy.push_back(Elevel-Eneuve); - Elevel=Eneuve; - } - if(Elevel<2.) exit(0); - if(Elevel <= 2.01) {energy.push_back(Elevel);Elevel =0.; - return energy;} - break; - } - if(i==ngadbin-1) exit(0); - } - } - } while(Elevel > 2.01); - //energy.push_back(Elevel); - return energy; -} - -double GdCaptureGammas::cumule(int ienerg) -{ -// computes the integrated probability from the threshold -// to the energy of indice ienerg - -// input : IENERG indice for the energy with NGADBIN intervalles -// of width BINT - -// output : integrated probabilities cumule - - double norm,A,A2,A4,A6,T1,T2,T3,T4,xi; - // const double eps=0.0001; - const int ngadbin=750; - if(ienerg>ngadbin-1) exit(0); - intc(); - norm=2./(pow(sqrt(37.),8)); - A=sqrt(37.)*sqrt(Elevel-2.); - A2=pow(A,2); - A4=pow(A,4); - A6=pow(A,6); - T1=xint[0][ienerg]*A6; - T2=-3.*xint[1][ienerg]*A4; - T3=3.*xint[2][ienerg]*A2; - T4=-1.0*xint[3][ienerg]; - xi=T1+T2+T3+T4; - return norm*xi; -} - -void GdCaptureGammas::intc() -{ -// computes the table xint(4,750) -// integration -// of the function (X**N) * exp(X) - - -// output : N=1 comes in XINT(1,75) -// 2 2 -// 3 3 -// 4 4 - - double aux[7],ener,xk; - const double ngadbin=750,bint=0.01; - for(int i=0;iGetName()!="Gadolinium") - if((*(G4Element::GetElementTable()))[i]->GetZ() != 64) - { - theCapture[i].Init((*(G4Element::GetElementTable()))[i], dirName); - G4cout<<(*(G4Element::GetElementTable()))[i]->GetName()<GetName()<::iterator - ite = (*theCapture).begin() ; ite != (*theCapture).end() ; ite++ ) - { - delete *ite; - } - (*theCapture).clear(); - } - - #include "G4NeutronHPThermalBoost.hh" - G4HadFinalState * GdNeutronHPCapture::ApplyYourself(const G4HadProjectile& aTrack, G4Nucleus& aNucleus ) - { - G4NeutronHPManager::GetInstance()->OpenReactionWhiteBoard(); - if(getenv("NeutronHPCapture")) G4cout <<" ####### GdNeutronHPCapture called"<GetNumberOfElements(); - G4int index = theMaterial->GetElement(0)->GetIndex(); - if(n!=1) - { - xSec = new G4double[n]; - G4double sum=0; - G4int i; - const G4double * NumAtomsPerVolume = theMaterial->GetVecNbOfAtomsPerVolume(); - G4double rWeight; - G4NeutronHPThermalBoost aThermalE; - for (i=0; iGetElement(i)->GetIndex(); - rWeight = NumAtomsPerVolume[i]; - //G4cout << theMaterial->GetElement(i)->GetName() << G4endl; - xSec[i] = (*(*theCapture)[index]).GetXsec(aThermalE.GetThermalEnergy(aTrack, - theMaterial->GetElement(i), - theMaterial->GetTemperature())); - xSec[i] *= rWeight; - sum+=xSec[i]; - } - G4double random = G4UniformRand(); - G4double running = 0; - for (i=0; iGetElement(i)->GetIndex(); - //if(random<=running/sum) break; - if( sum == 0 || random <= running/sum ) break; - } - if(i==n) i=std::max(0, n-1); - delete [] xSec; - } - //return theCapture[index].ApplyYourself(aTrack); - //G4HadFinalState* result = theCapture[index].ApplyYourself(aTrack); - G4HadFinalState* result = (*(*theCapture)[index]).ApplyYourself(aTrack); - - //Overwrite target parameters - aNucleus.SetParameters(G4NeutronHPManager::GetInstance()->GetReactionWhiteBoard()->GetTargA(),G4NeutronHPManager::GetInstance()->GetReactionWhiteBoard()->GetTargZ()); - const G4Element* target_element = (*G4Element::GetElementTable())[index]; - const G4Isotope* target_isotope=NULL; - G4int iele = target_element->GetNumberOfIsotopes(); - for ( G4int j = 0 ; j != iele ; j++ ) { - target_isotope=target_element->GetIsotope( j ); - if ( target_isotope->GetN() == G4NeutronHPManager::GetInstance()->GetReactionWhiteBoard()->GetTargA() ) break; - } - //G4cout << "Target Material of this reaction is " << theMaterial->GetName() << G4endl; - //G4cout << "Target Element of this reaction is " << target_element->GetName() << G4endl; - //G4cout << "Target Isotope of this reaction is " << target_isotope->GetName() << G4endl; - aNucleus.SetIsotope( target_isotope ); - - G4NeutronHPManager::GetInstance()->CloseReactionWhiteBoard(); - return result; - } - -void GdNeutronHPCapture::BuildPhysicsTable(const G4ParticleDefinition&) -{ - - if ( !G4Threading::IsWorkerThread() ){ - - if ( theCapture == NULL ) theCapture = new std::vector; - - if ( numEle == (G4int)G4Element::GetNumberOfElements() ) return; - - if ( theCapture->size() == G4Element::GetNumberOfElements() ) { - numEle = G4Element::GetNumberOfElements(); - return; - } - - - G4cout << "Capture : start of construction!!!!!!!!" << G4endl; - if ( !getenv("G4NEUTRONHPDATA") ) - throw G4HadronicException(__FILE__, __LINE__, "Please setenv G4NEUTRONHPDATA to point to the neutron cross-section files."); - dirName = getenv("G4NEUTRONHPDATA"); - G4String tString = "/Capture"; - dirName = dirName + tString; - G4cout << "+++++++++++++++++++++++++++++++++++++++++++++++++" << G4endl; - // G4cout <<"Disname="<GetName()!="Gadolinium") - G4cout << (*(G4Element::GetElementTable()))[i]->GetName() << G4endl; - (*theCapture).push_back( new G4NeutronHPChannel ); - (*(*theCapture)[i]).Init((*(G4Element::GetElementTable()))[i], dirName); - if ((*(G4Element::GetElementTable()))[i]->GetZ() != 64) { - (*(*theCapture)[i]).Register(theFS); - // theCapture[i].Register(theGdFS); - } else { - (*(*theCapture)[i]).Register(theGdFS); - } - } - delete theFS; - delete theGdFS; - } - numEle = G4Element::GetNumberOfElements(); - -} \ No newline at end of file diff --git a/libWCSIM/src/GdNeutronHPCaptureFS.cc b/libWCSIM/src/GdNeutronHPCaptureFS.cc deleted file mode 100644 index 3f479d0..0000000 --- a/libWCSIM/src/GdNeutronHPCaptureFS.cc +++ /dev/null @@ -1,213 +0,0 @@ - - -/////////////////////////////////////////////////////////////////////////////// -// Spectrum of radiative neutron capture by Gadolinium -// version 1.0.0 -// (Sep.09.2005) - -// Author : karim.zbiri@subatech.in2p3.fr - -//Modified class from original G4NeutronHPCaptureFS class to deexcite and -//add correctly the secondary to the hadronic final state - -// Karim Zbiri, Aug, 2005 -/////////////////////////////////////////////////////////////////////////////// - - - -#include -#include "GdNeutronHPCaptureFS.hh" -#include "G4Gamma.hh" -#include "G4ReactionProduct.hh" -#include "G4Nucleus.hh" -#include "G4PhotonEvaporation.hh" -#include "G4Fragment.hh" -#include "G4ParticleTable.hh" -#include "G4NeutronHPDataUsed.hh" -#include "G4SystemOfUnits.hh" - - - - G4HadFinalState * GdNeutronHPCaptureFS::ApplyYourself(const G4HadProjectile & theTrack) - { - if ( theResult.Get() == NULL ) theResult.Put( new G4HadFinalState ); - theResult.Get()->Clear(); - - G4int i; - // prepare neutron - G4double eKinetic = theTrack.GetKineticEnergy(); - const G4HadProjectile *incidentParticle = &theTrack; - theNeutron = const_cast(incidentParticle->GetDefinition()) ; - theNeutron.SetMomentum( incidentParticle->Get4Momentum().vect() ); - theNeutron.SetKineticEnergy( eKinetic ); - -// prepare target - G4double eps = 0.0001; - if(targetMass<500*MeV) - targetMass = ( G4NucleiProperties::GetNuclearMass( static_cast(theBaseA+eps) , static_cast(theBaseZ+eps) )) / - G4Neutron::Neutron()->GetPDGMass(); - G4ThreeVector neutronVelocity = 1./G4Neutron::Neutron()->GetPDGMass()*theNeutron.GetMomentum(); - G4double temperature = theTrack.GetMaterial()->GetTemperature(); - theTarget = aNucleus.GetBiasedThermalNucleus(targetMass, neutronVelocity, temperature); - -// go to nucleus rest system - theNeutron.Lorentz(theNeutron, -1*theTarget); - eKinetic = theNeutron.GetKineticEnergy(); - -// dice the photons - - G4ReactionProductVector * thePhotons = NULL; - thePhotons = theFinalgammas.GetGammas(theBaseA); - - // update the nucleus - - G4ThreeVector aCMSMomentum = theNeutron.GetMomentum()+theTarget.GetMomentum(); - G4LorentzVector p4(aCMSMomentum, theTarget.GetTotalEnergy() + theNeutron.GetTotalEnergy()); - nucleus = new G4Fragment(static_cast(theBaseA+1), static_cast(theBaseZ) ,p4); - - // G4Fragment * theOne = NULL; - - G4int nPhotons = 0; - if(thePhotons!=NULL) nPhotons=thePhotons->size(); - - - for(i=0;ioperator[](i)->GetMomentum()); - G4LorentzVector gamma(pGamma,thePhotons->operator[](i)->GetTotalEnergy()); - theOne= new G4Fragment(gamma,G4Gamma::GammaDefinition()); - UpdateNucleus(theOne,thePhotons->operator[](i)->GetTotalEnergy()); - } - - theTwo = new G4DynamicParticle; - theTwo->SetDefinition(G4IonTable::GetIonTable() - ->GetIon(static_cast(theBaseZ), static_cast(theBaseA+1), 0)); - theTwo->SetMomentum(nucleus->GetMomentum()); - - //delete theOne; - -// add them to the final state - - G4int nParticles = nPhotons; - if(1==nPhotons) nParticles = 2; - - // back to lab system - for(i=0; ioperator[](i)->Lorentz(*(thePhotons->operator[](i)), theTarget); - } - - // Recoil, if only one gamma - if (1==nPhotons) - { - G4DynamicParticle * theOne = new G4DynamicParticle; - G4ParticleDefinition * aRecoil = G4IonTable::GetIonTable() - ->GetIon(static_cast(theBaseZ), static_cast(theBaseA+1), 0); - theOne->SetDefinition(aRecoil); - // Now energy; - // Can be done slightly better @ - G4ThreeVector aMomentum = theTrack.Get4Momentum().vect() - +theTarget.GetMomentum() - -thePhotons->operator[](0)->GetMomentum(); - - G4ThreeVector theMomUnit = aMomentum.unit(); - G4double aKinEnergy = theTrack.GetKineticEnergy() - +theTarget.GetKineticEnergy(); // gammas come from Q-value - G4double theResMass = aRecoil->GetPDGMass(); - G4double theResE = aRecoil->GetPDGMass()+aKinEnergy; - G4double theAbsMom = std::sqrt(theResE*theResE - theResMass*theResMass); - G4ThreeVector theMomentum = theAbsMom*theMomUnit; - theOne->SetMomentum(theMomentum); - theResult.Get()->AddSecondary(theOne); - } - - // Now fill in the gammas. - for(i=0; iSetDefinition(thePhotons->operator[](i)->GetDefinition()); - theOne->SetMomentum(thePhotons->operator[](i)->GetMomentum()); - theResult.Get()->AddSecondary(theOne); - delete thePhotons->operator[](i); - } - delete thePhotons; - - //ADD deexcited nucleus - - theResult.Get()->AddSecondary(theTwo); - - // clean up the primary neutron - theResult.Get()->SetStatusChange(stopAndKill); - return theResult.Get(); - } - - - -void GdNeutronHPCaptureFS::UpdateNucleus( const G4Fragment* gamma , G4double eGamma ) -{ - - G4LorentzVector p4Gamma = gamma->GetMomentum(); - G4ThreeVector pGamma(p4Gamma.vect()); - - G4LorentzVector p4Nucleus(nucleus->GetMomentum() ); - - G4double m1 = G4IonTable::GetIonTable()->GetIonMass(static_cast(nucleus->GetZ()), - static_cast(nucleus->GetA())); - G4double m2 = nucleus->GetZ() * G4Proton::Proton()->GetPDGMass() + - (nucleus->GetA()- nucleus->GetZ())*G4Neutron::Neutron()->GetPDGMass(); - - G4double Mass = std::min(m1,m2); - - G4double newExcitation = p4Nucleus.mag() - Mass - eGamma; - - // G4cout<<" Egamma = "<(A), static_cast(Z), M, dirName, tString, dbool); - - G4String filename = aFile.GetName(); - theBaseA = A; - theBaseZ = G4int(Z+.5); - if(!dbool || ( Z<2.5 && ( std::abs(theBaseZ - Z)>0.0001 || std::abs(theBaseA - A)>0.0001))) - { - hasAnyData = false; - hasFSData = false; - hasXsec = false; - return; - } - std::ifstream theData(filename, std::ios::in); - - hasFSData = theFinalStatePhotons.InitMean(theData); - if(hasFSData) - { - targetMass = theFinalStatePhotons.GetTargetMass(); - theFinalStatePhotons.InitAngular(theData); - theFinalStatePhotons.InitEnergies(theData); - } - theData.close(); - } - - diff --git a/libWCSIM/src/TJNuBeamFlux.cc b/libWCSIM/src/TJNuBeamFlux.cc deleted file mode 100644 index 6fb3199..0000000 --- a/libWCSIM/src/TJNuBeamFlux.cc +++ /dev/null @@ -1,111 +0,0 @@ -#include "TJNuBeamFlux.hh" -#include - -//_______________________________________________________________________________________ -RooTrackerVtxBase::RooTrackerVtxBase() : - TObject() -{ -} -//_______________________________________________________________________________________ -JNuBeamFlux::JNuBeamFlux() : - RooTrackerVtxBase() -{ - NuFileName = new TObjString("Not-set"); - this->Reset(); -} -//_______________________________________________________________________________________ -JNuBeamFlux::~JNuBeamFlux() -{ - if(NuFileName != NULL) { delete NuFileName; NuFileName = NULL; } -} -//_______________________________________________________________________________________ -void JNuBeamFlux::Copy(const JNuBeamFlux * flux) -{ - NuFileName->SetString(flux->NuFileName->GetString()); - NuFluxEntry = flux->NuFluxEntry; - NuParentPdg = flux->NuParentPdg; - NuParentDecMode = flux->NuParentDecMode; - NuCospibm = flux->NuCospibm; - NuNorm = flux->NuNorm; - NuCospi0bm = flux->NuCospi0bm; - NuRnu = flux->NuRnu; - NuIdfd = flux->NuIdfd; - NuGipart = flux->NuGipart; - NuGamom0 = flux->NuGamom0; - NuEnusk = flux->NuEnusk; - NuNormsk = flux->NuNormsk; - NuAnorm = flux->NuAnorm; - NuVersion = flux->NuVersion; - NuTuneid = flux->NuTuneid; - NuNtrig = flux->NuNtrig; - NuPint = flux->NuPint; - NuRand = flux->NuRand; - for(int i=0;i<2;i++){ - NuXnu [i] = flux->NuXnu[i]; - NuBpos[i] = flux->NuBpos[i]; - NuBtilt[i] = flux->NuBtilt[i]; - NuBrms[i] = flux->NuBrms[i]; - NuEmit[i] = flux->NuEmit[i]; - NuAlpha[i] = flux->NuAlpha[i]; - } - for(int i=0;i<3;i++){ - NuGpos0[i] = flux->NuGpos0[i]; - NuGvec0[i] = flux->NuGvec0[i]; - NuHcur[i] = flux->NuHcur[i]; - } - for(int i=0;i<4;i++){ - NuParentDecP4[i] = flux->NuParentDecP4[i]; - NuParentDecX4[i] = flux->NuParentDecX4[i]; - NuParentProP4[i] = flux->NuParentProP4[i]; - NuParentProX4[i] = flux->NuParentProX4[i]; - } - NuNg = flux->NuNg; - for(int i=0;iNuGcosbm[i]; - NuGdistc[i] = flux->NuGdistc[i]; - NuGdistal[i] = flux->NuGdistal[i]; - NuGdistti[i] = flux->NuGdistti[i]; - NuGdistfe[i] = flux->NuGdistfe[i]; - NuGpid[i] = flux->NuGpid[i]; - NuGmec[i] = flux->NuGmec[i]; - NuGmat[i] = flux->NuGmat[i]; - for(int j=0;j<3;j++) { - NuGp[i][j] = flux->NuGp[i][j]; - NuGv[i][j] = flux->NuGv[i][j]; - } - } -} -//_______________________________________________________________________________________ -void JNuBeamFlux::Reset() -{ - NuFluxEntry = -1; - NuFileName->SetString("Not-set"); - - NuParentPdg = 0; - NuParentDecMode = -1; - NuCospibm = NuNorm = NuCospi0bm = NuRnu = -999999.; - NuIdfd = NuGipart = -999999; - NuGamom0 = NuEnusk = NuNormsk = NuAnorm = NuVersion = -999999.; - NuTuneid = NuNtrig = NuPint = NuRand = -999999; - for(int i=0;i<2;i++){ - NuXnu [i] = NuBpos[i] = NuBtilt[i] = NuBrms[i] = NuEmit[i] = NuAlpha[i] = -999999.; - } - for(int i=0;i<3;i++) NuGpos0[i] = NuGvec0[i] = NuHcur[i] = -999999.; - for(int i=0;i<4;i++) NuParentDecP4[i] = NuParentDecX4[i] = NuParentProP4[i] = NuParentProX4[i] = -999999.; - NuNg = -1; - for(int i=0;i -#include - -#include "TNRooTrackerVtx.hh" - -using std::endl; -using std::cout; -using std::setw; -using std::setprecision; -using std::setfill; -using std::ios; - -ClassImp(NRooTrackerVtx) - -//_______________________________________________________________________________________ -NRooTrackerVtx::NRooTrackerVtx() : - JNuBeamFlux() -{ - this->Init(); -} -//_______________________________________________________________________________________ -NRooTrackerVtx::~NRooTrackerVtx() -{ - if(GeomPath != NULL) { delete GeomPath; GeomPath = NULL; } - if(GeneratorName != NULL) { delete GeneratorName; GeneratorName = NULL; } - if(EvtCode != NULL) { delete EvtCode; EvtCode = NULL; } - if(OrigFileName != NULL) { delete OrigFileName; OrigFileName = NULL; } - if(OrigTreeName != NULL) { delete OrigTreeName; OrigTreeName = NULL; } -} -//_______________________________________________________________________________________ -void NRooTrackerVtx::Copy(const NRooTrackerVtx * event) -{ - GeomPath->SetString(event->GeomPath->GetString()); - GeneratorName->SetString(event->GeneratorName->GetString()); - EvtCode->SetString(event->EvtCode->GetString()); - OrigFileName->SetString(event->OrigFileName->GetString()); - OrigTreeName->SetString(event->OrigTreeName->GetString()); - - EvtNum = event->EvtNum; - EvtXSec = event->EvtXSec; - EvtDXSec = event->EvtDXSec; - EvtWght = event->EvtWght; - EvtProb = event->EvtProb; - OrigTreePOT = event->OrigTreePOT; - OrigEvtNum = event->OrigEvtNum; - TimeInSpill = event->TimeInSpill; - OrigTreeEntries = event->OrigTreeEntries; - TruthVertexID = event->TruthVertexID; - - for(int i=0; i < 4; i++) { - EvtVtx[i] = event->EvtVtx[i]; - } - - NuNorm = event->NuNorm; - NuEnusk = event->NuEnusk; - NuNormsk = event->NuNormsk; - NuAnorm = event->NuAnorm; - - StdHepN = event->StdHepN; - - //define the dynamic arrays - StdHepPdg = new int[StdHepN]; - StdHepStatus = new int[StdHepN]; - StdHepFd = new int[StdHepN]; - StdHepLd = new int[StdHepN]; - StdHepFm = new int[StdHepN]; - StdHepLm = new int[StdHepN]; - - for(int i=0; i< StdHepN; i++) - { - StdHepPdg [i] = event->StdHepPdgTemp [i]; - StdHepStatus[i] = event->StdHepStatusTemp [i]; - - StdHepFd [i] = event->StdHepFdTemp [i]; - StdHepLd [i] = event->StdHepLdTemp [i]; - StdHepFm [i] = event->StdHepFmTemp [i]; - StdHepLm [i] = event->StdHepLmTemp [i]; - - for(int j=0; j < 4; j++) { StdHepX4 [i][j] = event->StdHepX4 [i][j]; } - for(int j=0; j < 4; j++) { StdHepP4 [i][j] = event->StdHepP4 [i][j]; } - for(int j=0; j < 3; j++) { StdHepPolz [i][j] = event->StdHepPolz [i][j]; } - } - - NEnvc = event->NEnvc; - - //define the dynamic arrays - NEipvc = new int[NEnvc]; - NEiorgvc = new int[NEnvc]; - NEiflgvc = new int[NEnvc]; - NEicrnvc = new int[NEnvc]; - - for(int i=0; i< NEnvc; i++) { - NEipvc[i] = event->NEipvcTemp[i]; - NEiorgvc[i] = event->NEiorgvcTemp[i]; - NEiflgvc[i] = event->NEiflgvcTemp[i]; - NEicrnvc[i] = event->NEicrnvcTemp[i]; - for(int j=0; j < 3; j++) { NEpvc[i][j] = event->NEpvc[i][j]; } - } - - NEcrsx = event->NEcrsx; - NEcrsy = event->NEcrsy; - NEcrsz = event->NEcrsz; - NEcrsphi = event->NEcrsphi; - - NEnvert = event->NEnvert; - //define the dynamic array - NEiflgvert = new int[NEnvert]; - for(int i=0; i< NEnvert; i++) { - NEiflgvert[i] = event->NEiflgvertTemp[i] ; - for(int j=0; j < 3; j++) { NEposvert[i][j] = event->NEposvert[i][j]; } - } - - NEnvcvert = event->NEnvcvert; - - //define the dynamic arrays - NEabspvert = new float[NEnvcvert]; - NEabstpvert = new float[NEnvcvert]; - NEipvert = new int[NEnvcvert]; - NEiverti = new int[NEnvcvert]; - NEivertf = new int[NEnvcvert]; - - for(int i=0; i< NEnvcvert; i++) { - NEabspvert[i] = event->NEabspvertTemp[i]; - NEabstpvert[i] = event->NEabstpvertTemp[i]; - NEipvert[i] = event->NEipvertTemp[i]; - NEiverti[i] = event->NEivertiTemp[i]; - NEivertf[i] = event->NEivertfTemp[i]; - for(int j=0; j < 3; j++) { NEdirvert[i][j] = event->NEdirvert[i][j]; } - } - - // Remember to copy the flux info - JNuBeamFlux * flux = (JNuBeamFlux*) event; - JNuBeamFlux::Copy(flux); - -} -//_______________________________________________________________________________________ -void NRooTrackerVtx::Reset() -{ - - if (GeomPath ) GeomPath -> SetString("Not-set"); - if (GeneratorName) GeneratorName -> SetString("Not-set"); - if (EvtCode ) EvtCode -> SetString("Not-set"); - if (OrigFileName ) OrigFileName -> SetString("Not-set"); - if (OrigTreeName ) OrigTreeName -> SetString("Not-set"); - - EvtNum = -1; - EvtXSec = -1.0; - EvtDXSec = -1.0; - EvtWght = -1.0; - EvtProb = -1.0; - OrigTreePOT = -1.0; - OrigEvtNum = -1; - TimeInSpill = -1.0; - OrigTreeEntries = -1; - TruthVertexID = -1; - - for(int i=0; i < 4; i++){ - EvtVtx[i] = 0.0; - } - - StdHepN = 0; - - for(int i=0; iReset(); -} -//_______________________________________________________________________________________ -void NRooTrackerVtx::Print(const Option_t*) const -{ - - cout << "\nNRooTrackerMCTruth: " << endl; - cout << " --> GeneratorName = " << GeneratorName->GetString() << endl; - cout << " --> EvtCode = " << EvtCode->GetString() << endl; - cout << " --> GeomPath = " << GeomPath->GetString() << endl; - cout << " --> EvtNum = " << EvtNum << endl; - cout << " --> EvtXSec = " << EvtXSec << endl; - cout << " --> EvtDXSec = " << EvtDXSec << endl; - cout << " --> EvtWght = " << EvtWght << endl; - cout << " --> EvtProb = " << EvtProb << endl; - cout << " --> TruthVertexID = " << TruthVertexID << endl; - - - cout << "NRooTrackerStdHep:" << endl; - - for(int i=0; i pdgc: " - << " / ist: " - << setfill(' ') << setw(3) << StdHepStatus[i] - << " / mom: " - << setfill(' ') << setw(3) << StdHepFm[i] - << " / daughters: (" - << setfill(' ') << setw(3) << StdHepFd[i] - << ", " - << setfill(' ') << setw(3) << StdHepLd[i] - << ") / E = " - << setfill(' ') << setw(10) << StdHepP4[i][kNStdHepIdxE] - << ", px = " - << setfill(' ') << setw(10) << StdHepP4[i][kNStdHepIdxPx] - << ", py = " - << setfill(' ') << setw(10) << StdHepP4[i][kNStdHepIdxPy] - << ", pz = " - << setfill(' ') << setw(10) << StdHepP4[i][kNStdHepIdxPz] - << endl; - } - - cout << endl << "Jnu-beam pass through info:" << endl; - cout << " --> NuParentPdg = " << NuParentPdg << endl; - cout << " --> NuParentDecMode = " << NuParentDecMode << endl; - cout << " --> NuParentDecP4[4] = " << NuParentDecP4[0] << ", "<< NuParentDecP4[1] << ", " << NuParentDecP4[2] << ", " << NuParentDecP4[3] << endl; - cout << " --> NuParentDecX4[4] = " << NuParentDecX4[0] << ", "<< NuParentDecX4[1] << ", " << NuParentDecX4[2] << ", " << NuParentDecX4[3] << endl; - cout << " --> NuParentProP4[4] = " << NuParentProP4[0] << ", "<< NuParentProP4[1] << ", " << NuParentProP4[2] << ", " << NuParentProP4[3] << endl; - cout << " --> NuParentProX4[4] = " << NuParentProX4[0] << ", "<< NuParentProX4[1] << ", " << NuParentProX4[2] << ", " << NuParentProX4[3] << endl; - cout << " --> NuGipart = " << NuGipart << endl; - cout << " --> NuGamom0 = " << NuGamom0 << endl; - cout << " --> NuGpos0[3] = " << NuGpos0[0] << ", "<< NuGpos0[1] << ", " << NuGpos0[2] << endl; - cout << " --> NuGvec0[3] = " << NuGvec0[0] << ", "<< NuGvec0[1] << ", " << NuGvec0[2] << endl; - - cout << "NuEnusk = " << NuEnusk << endl; - cout << "NuNorm = " << NuNorm << endl; - cout << "NuNormsk = " << NuNormsk << endl; - cout << "NuAnorm = " << NuAnorm << endl; - - cout << "NuVersion = " << NuVersion << endl; - cout << "NuTuneid = " << NuTuneid << endl;; - cout << "NuPint = " << NuPint << endl; - - cout << "NuBpos[2] = "; - for(int ip = 0; ip < 2; ip++) - cout << NuBpos[ip] << " "; - cout << endl; - - cout << "NuBtilt[2] = "; - for(int ip = 0; ip < 2; ip++) - cout << NuBtilt[ip] << " "; - cout << endl; - - cout << "NuBrms[2] = "; - for(int ip = 0; ip < 2; ip++) - cout << NuBrms[ip] << " "; - cout << endl; - - cout << "NuEmit[2] = "; - for(int ip = 0; ip < 2; ip++) - cout << NuEmit[ip] << " "; - cout << endl; - - cout << "NuAlpha[2] = "; - for(int ip = 0; ip < 2; ip++) - cout << NuAlpha[ip] << " "; - cout << endl; - - cout << "NuHcur[3] = "; - for(int ip = 0; ip < 3; ip++) - cout << NuHcur[ip] << " "; - cout << endl; - - - cout << endl << "NEUT pass through info:" << endl << endl; - - cout << "NEnvc = " << NEnvc << endl; - cout << "ip NEipvc[ip] NEiorgvc[ip] NEiflgvc[ip] NEicrnvc[ip] NEpvc[ip][3]" << endl; - for(int ip = 0; ip < NEnvc; ip++) { - cout << ip << " " << NEipvc[ip] << " " << NEiorgvc[ip] << " "; - cout << NEiflgvc[ip] << " " << NEicrnvc[ip] << " "; - for(int ip2 = 0; ip2 < 3; ip2++){ - cout << NEpvc[ip][ip2] << " "; - } - cout << endl; - } - cout << endl; - - cout << "NEcrsx NEcrsy NEcrsz NEcrsphi" << endl; - cout << NEcrsx << " " << NEcrsy << " " << NEcrsz << " " << NEcrsphi << endl; - - cout << "NEnvert = " << NEnvert << endl; - cout << "iv NEiflgvert[iv] NEposvert[iv][3]" << endl; - - for(int iv = 0; iv < NEnvert; iv++){ - cout << iv << " " << NEiflgvert[iv] << " "; - for(int ip2 = 0; ip2 < 3; ip2++){ - cout << NEposvert[iv][ip2] << " "; - } - cout << endl; - } - cout << endl; - - cout << "NEnvcvert = " << NEnvcvert << endl; - cout << "ip NEabspvert[ip] NEabstpvert[ip] NEipvert[ip] NEiverti[ip] NEivertf[ip] NEdirvert[ip][3]" << endl; - for(int ip = 0; ip < NEnvcvert; ip++){ - cout << ip << " " << NEabspvert[ip] << " " << NEabstpvert[ip] << " " << NEipvert[ip] << " " << NEiverti[ip] << " " << NEivertf[ip] << " "; - for(int ip2 = 0; ip2 < 3; ip2++){ - cout << NEdirvert[ip][ip2] << " "; - } - cout << endl; - } - cout << endl; - - cout << endl << "Pass-through file info:" << endl; - cout << " --> OrigFileName = " << OrigFileName->GetString() << endl; - cout << " --> OrigTreeName = " << OrigTreeName->GetString() << endl; - cout << " --> OrigEvtNum = " << OrigEvtNum << endl; - cout << " --> TimeInSpill = " << TimeInSpill << endl; - cout << " --> OrigTreeEntries = " << OrigTreeEntries << endl; - cout << " --> OrigTreePOT = " << OrigTreePOT << endl; - -} -//_______________________________________________________________________________________ - diff --git a/libWCSIM/src/WCSimConstructCylinder.cc b/libWCSIM/src/WCSimConstructCylinder.cc deleted file mode 100644 index cdcf073..0000000 --- a/libWCSIM/src/WCSimConstructCylinder.cc +++ /dev/null @@ -1,1541 +0,0 @@ -// -*- mode:c++; tab-width:4; -*- -#include "WCSimDetectorConstruction.hh" - -#include "G4Material.hh" -#include "G4Element.hh" -#include "G4Box.hh" -#include "G4Tubs.hh" -#include "G4UnionSolid.hh" -#include "G4Sphere.hh" -#include "G4Trd.hh" -#include "G4IntersectionSolid.hh" -#include "G4Polyhedra.hh" -#include "G4LogicalVolume.hh" -#include "G4ThreeVector.hh" -#include "G4RotationMatrix.hh" -#include "G4PVReplica.hh" -#include "G4PVPlacement.hh" -#include "G4PVParameterised.hh" -#include "G4AssemblyVolume.hh" -#include "G4SubtractionSolid.hh" -#include "globals.hh" -#include "G4VisAttributes.hh" -#include "G4LogicalBorderSurface.hh" -#include "G4LogicalSkinSurface.hh" -#include "G4OpBoundaryProcess.hh" -#include "G4OpticalSurface.hh" -#include "G4UserLimits.hh" -#include "G4ReflectionFactory.hh" -#include "G4GeometryTolerance.hh" -#include "G4GeometryManager.hh" - -#include "WCSimTuningParameters.hh" //jl145 - -#include "G4SystemOfUnits.hh" -#include "G4PhysicalConstants.hh" -//#define DEBUG - -/*********************************************************** - * - * This file contains the functions which construct a - * cylindrical WC detector. It used by both the SK and - * LBNE WC detector modes. It is called in the Construct() - * method in WCSimDetectorConstruction.cc. - * - * Sourcefile for the WCSimDetectorConstruction class - * - ***********************************************************/ - - -G4LogicalVolume* WCSimDetectorConstruction::ConstructCylinder() -{ - G4cout << "**** Building Cylindrical Detector ****" << G4endl; - - //----------------------------------------------------- - // Positions - //----------------------------------------------------- - - debugMode = false; - - WCPosition=0.;//Set the WC tube offset to zero - - WCIDRadius = WCIDDiameter/2.; - // the number of regular cell in phi direction: - WCBarrelRingNPhi = (G4int)(WCBarrelNumPMTHorizontal/WCPMTperCellHorizontal); - // the part of one ring, that is covered by the regular cells: - totalAngle = 2.0*pi*rad*(WCBarrelRingNPhi*WCPMTperCellHorizontal/WCBarrelNumPMTHorizontal) ; - // angle per regular cell: - dPhi = totalAngle/ WCBarrelRingNPhi; - // it's height: - barrelCellHeight = (WCIDHeight-2.*WCBarrelPMTOffset)/WCBarrelNRings; - // the height of all regular cells together: - mainAnnulusHeight = WCIDHeight -2.*WCBarrelPMTOffset -2.*barrelCellHeight; - - //TF: has to change for mPMT vessel: - - G4cout << WCIDRadius << ", " << WCPMTExposeHeight << ", " << mPMT_vessel_cyl_height << ", " << mPMT_vessel_radius << G4endl; -#ifdef DEBUG - G4cout << "HYBRID = " << hybrid << G4endl; -#endif - - if(hybrid){ - if(mPMT_vessel_cyl_height + mPMT_vessel_radius < 1.*mm) - innerAnnulusRadius = WCIDRadius - std::max(WCPMTExposeHeight,WCPMTExposeHeight2)-1.*mm; - else - innerAnnulusRadius = WCIDRadius - std::max(WCPMTExposeHeight,(mPMT_vessel_cyl_height + mPMT_vessel_radius)) -1.*mm; - } - else{ - if(mPMT_vessel_cyl_height + mPMT_vessel_radius < 1.*mm) - innerAnnulusRadius = WCIDRadius - WCPMTExposeHeight -1.*mm; - else - innerAnnulusRadius = WCIDRadius - (mPMT_vessel_cyl_height + mPMT_vessel_radius) -1.*mm; - } - //TF: need to add a Polyhedra on the other side of the outerAnnulusRadius for the OD - outerAnnulusRadius = WCIDRadius + WCBlackSheetThickness + 1.*mm;//+ Stealstructure etc. - - // the radii are measured to the center of the surfaces - // (tangent distance). Thus distances between the corner and the center are bigger. - WCLength = WCIDHeight + 2*2.3*m; //jl145 - reflects top veto blueprint, cf. Farshid Feyzi - WCRadius = (WCIDDiameter/2. + WCBlackSheetThickness + 1.5*m)/cos(dPhi/2.) ; // ToDo: OD - - // now we know the extend of the detector and are able to tune the tolerance - G4GeometryManager::GetInstance()->SetWorldMaximumExtent(WCLength > WCRadius ? WCLength : WCRadius); - G4cout << "Computed tolerance = " - << G4GeometryTolerance::GetInstance()->GetSurfaceTolerance()/mm - << " mm" << G4endl; - - //Decide if adding Gd - water = "Water"; - if (WCAddGd) - { - water = "Doped Water"; - if(!G4Material::GetMaterial("Doped Water", false)) AddDopedWater(); - } - - // Optionally switch on the checkOverlaps. Once the geometry is debugged, switch off for - // faster running. Checking ALL overlaps is crucial. No overlaps are allowed, otherwise - // G4Navigator will be confused and result in wrong photon tracking and resulting yields. - // ToDo: get these options from .mac - checkOverlaps = true; - checkOverlapsPMT = false; - // Optionally place parts of the detector. Very useful for visualization and debugging - // geometry overlaps in detail. - placeBarrelPMTs = true; - placeCapPMTs = true; - placeBorderPMTs = true; - - - //----------------------------------------------------- - // Volumes - //----------------------------------------------------- - // The water barrel is placed in an tubs of air - - G4Tubs* solidWC = new G4Tubs("WC", - 0.0*m, - WCRadius+2.*m, - .5*WCLength+4.2*m, //jl145 - per blueprint - 0.*deg, - 360.*deg); - - G4LogicalVolume* logicWC = - new G4LogicalVolume(solidWC, - G4Material::GetMaterial("Air"), - "WC", - 0,0,0); - - - G4VisAttributes* showColor = new G4VisAttributes(G4Colour(0.0,1.0,0.0)); - logicWC->SetVisAttributes(showColor); - - logicWC->SetVisAttributes(G4VisAttributes::Invisible); //amb79 - - //----------------------------------------------------- - // everything else is contained in this water tubs - //----------------------------------------------------- - G4Tubs* solidWCBarrel = new G4Tubs("WCBarrel", - 0.0*m, - WCRadius+1.*m, // add a bit of extra space - .5*WCLength, //jl145 - per blueprint - 0.*deg, - 360.*deg); - - G4LogicalVolume* logicWCBarrel = - new G4LogicalVolume(solidWCBarrel, - G4Material::GetMaterial(water), - "WCBarrel", - 0,0,0); - - G4VPhysicalVolume* physiWCBarrel = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,0.), - logicWCBarrel, - "WCBarrel", - logicWC, - false, - 0, - checkOverlaps); - -// This volume needs to made invisible to view the blacksheet and PMTs with RayTracer - if (Vis_Choice == "RayTracer") - {logicWCBarrel->SetVisAttributes(G4VisAttributes::Invisible);} - - else - {//{if(!debugMode) - //logicWCBarrel->SetVisAttributes(G4VisAttributes::Invisible);} - } - //----------------------------------------------------- - // Form annular section of barrel to hold PMTs - //---------------------------------------------------- - - - G4double mainAnnulusZ[2] = {-mainAnnulusHeight/2., mainAnnulusHeight/2}; - G4double mainAnnulusRmin[2] = {innerAnnulusRadius, innerAnnulusRadius}; - G4double mainAnnulusRmax[2] = {outerAnnulusRadius, outerAnnulusRadius}; - - G4Polyhedra* solidWCBarrelAnnulus = new G4Polyhedra("WCBarrelAnnulus", - 0.*deg, // phi start - totalAngle, - (G4int)WCBarrelRingNPhi, //NPhi-gon - 2, - mainAnnulusZ, - mainAnnulusRmin, - mainAnnulusRmax); - - G4LogicalVolume* logicWCBarrelAnnulus = - new G4LogicalVolume(solidWCBarrelAnnulus, - G4Material::GetMaterial(water), - "WCBarrelAnnulus", - 0,0,0); - // G4cout << *solidWCBarrelAnnulus << G4endl; - G4VPhysicalVolume* physiWCBarrelAnnulus = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,0.), - logicWCBarrelAnnulus, - "WCBarrelAnnulus", - logicWCBarrel, - false, - 0, - checkOverlaps); -if(!debugMode) - logicWCBarrelAnnulus->SetVisAttributes(G4VisAttributes::Invisible); //amb79 - - //----------------------------------------------------- - // Subdivide the BarrelAnnulus into rings - //----------------------------------------------------- - G4double RingZ[2] = {-barrelCellHeight/2., - barrelCellHeight/2.}; - - G4Polyhedra* solidWCBarrelRing = new G4Polyhedra("WCBarrelRing", - 0.*deg,//+dPhi/2., // phi start - totalAngle, //phi end - (G4int)WCBarrelRingNPhi, //NPhi-gon - 2, - RingZ, - mainAnnulusRmin, - mainAnnulusRmax); - - G4LogicalVolume* logicWCBarrelRing = - new G4LogicalVolume(solidWCBarrelRing, - G4Material::GetMaterial(water), - "WCBarrelRing", - 0,0,0); - - G4VPhysicalVolume* physiWCBarrelRing = - new G4PVReplica("WCBarrelRing", - logicWCBarrelRing, - logicWCBarrelAnnulus, - kZAxis, - (G4int)WCBarrelNRings-2, - barrelCellHeight); - -if(!debugMode) - {G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(0,0.5,1.)); - tmpVisAtt->SetForceWireframe(true);// This line is used to give definition to the rings in OGLSX Visualizer - logicWCBarrelRing->SetVisAttributes(tmpVisAtt); - //If you want the rings on the Annulus to show in OGLSX, then comment out the line below. - logicWCBarrelRing->SetVisAttributes(G4VisAttributes::Invisible); - } -else { - G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(0,0.5,1.)); - tmpVisAtt->SetForceWireframe(true); - logicWCBarrelRing->SetVisAttributes(tmpVisAtt); - } - - //----------------------------------------------------- - // Subdivisions of the BarrelRings are cells - //------------------------------------------------------ - - - G4Polyhedra* solidWCBarrelCell = new G4Polyhedra("WCBarrelCell", - -dPhi/2.+0.*deg, // phi start - dPhi, //total Phi - 1, //NPhi-gon - 2, - RingZ, - mainAnnulusRmin, - mainAnnulusRmax); - //G4cout << *solidWCBarrelCell << G4endl; - G4LogicalVolume* logicWCBarrelCell = - new G4LogicalVolume(solidWCBarrelCell, - G4Material::GetMaterial(water), - "WCBarrelCell", - 0,0,0); - - G4VPhysicalVolume* physiWCBarrelCell = - new G4PVReplica("WCBarrelCell", - logicWCBarrelCell, - logicWCBarrelRing, - kPhi, - (G4int)WCBarrelRingNPhi, - dPhi, - 0.); - - if(!debugMode) - {G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(1.,0.5,0.5)); - tmpVisAtt->SetForceSolid(true);// This line is used to give definition to the cells in OGLSX Visualizer - logicWCBarrelCell->SetVisAttributes(tmpVisAtt); - //If you want the columns on the Annulus to show in OGLSX, then comment out the line below. - logicWCBarrelCell->SetVisAttributes(G4VisAttributes::Invisible); - } - else { - G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(1.,0.5,0.5)); - tmpVisAtt->SetForceWireframe(true); - logicWCBarrelCell->SetVisAttributes(tmpVisAtt); - } - - //----------------------------------------------------------- - // The Blacksheet, a daughter of the cells containing PMTs, - // and also some other volumes to make the edges light tight - //----------------------------------------------------------- - - //------------------------------------------------------------- - // add barrel blacksheet to the normal barrel cells - // ------------------------------------------------------------ - G4double annulusBlackSheetRmax[2] = {(WCIDRadius+WCBlackSheetThickness), - WCIDRadius+WCBlackSheetThickness}; - G4double annulusBlackSheetRmin[2] = {(WCIDRadius), - WCIDRadius}; - G4cout << WCIDRadius << ", " << WCBlackSheetThickness << ", " << barrelCellHeight/2. << G4endl; - G4cout << WCIDHeight << ", " << WCBarrelPMTOffset << ", " << WCBarrelNRings << G4endl; - G4cout << dPhi << ", " << RingZ << G4endl; - G4cout << mainAnnulusRmin[0] << ", " << mainAnnulusRmax[0] << ", " << mainAnnulusZ[0] << G4endl; - G4cout << mainAnnulusRmin[1] << ", " << mainAnnulusRmax[1] << ", " << mainAnnulusZ[1] << G4endl; - - G4Polyhedra* solidWCBarrelCellBlackSheet = new G4Polyhedra("WCBarrelCellBlackSheet", - -dPhi/2., // phi start - dPhi, //total phi - 1, //NPhi-gon - 2, - RingZ, - annulusBlackSheetRmin, - annulusBlackSheetRmax); - - logicWCBarrelCellBlackSheet = - new G4LogicalVolume(solidWCBarrelCellBlackSheet, - G4Material::GetMaterial("Blacksheet"), - "WCBarrelCellBlackSheet", - 0,0,0); - - G4VPhysicalVolume* physiWCBarrelCellBlackSheet = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,0.), - logicWCBarrelCellBlackSheet, - "WCBarrelCellBlackSheet", - logicWCBarrelCell, - false, - 0, - checkOverlaps); - - - G4LogicalBorderSurface * WaterBSBarrelCellSurface - = new G4LogicalBorderSurface("WaterBSBarrelCellSurface", - physiWCBarrelCell, - physiWCBarrelCellBlackSheet, - OpWaterBSSurface); - - new G4LogicalSkinSurface("BSBarrelCellSkinSurface",logicWCBarrelCellBlackSheet, - BSSkinSurface); - - // Change made here to have the if statement contain the !debugmode to be consistent - // This code gives the Blacksheet its color. - -if (Vis_Choice == "RayTracer"){ - - G4VisAttributes* WCBarrelBlackSheetCellVisAtt - = new G4VisAttributes(G4Colour(0.2,0.9,0.2)); // green color - WCBarrelBlackSheetCellVisAtt->SetForceSolid(true); // force the object to be visualized with a surface - WCBarrelBlackSheetCellVisAtt->SetForceAuxEdgeVisible(true); // force auxiliary edges to be shown - if(!debugMode) - logicWCBarrelCellBlackSheet->SetVisAttributes(WCBarrelBlackSheetCellVisAtt); - else - logicWCBarrelCellBlackSheet->SetVisAttributes(G4VisAttributes::Invisible);} - -else { - - G4VisAttributes* WCBarrelBlackSheetCellVisAtt - = new G4VisAttributes(G4Colour(0.2,0.9,0.2)); - if(!debugMode) - logicWCBarrelCellBlackSheet->SetVisAttributes(G4VisAttributes::Invisible); - else - logicWCBarrelCellBlackSheet->SetVisAttributes(WCBarrelBlackSheetCellVisAtt);} - - //----------------------------------------------------------- - // add extra tower if necessary - // --------------------------------------------------------- - - // we have to declare the logical Volumes - // outside of the if block to access it later on - G4LogicalVolume* logicWCExtraTowerCell; - G4LogicalVolume* logicWCExtraBorderCell; - if(!(WCBarrelRingNPhi*WCPMTperCellHorizontal == WCBarrelNumPMTHorizontal)){ - - // as the angles between the corners of the main annulus - // and the corners extra tower are different, we need to adjust the - // tangent distance the surfaces of the extra tower. Otherwise - // the corners of the main annulus and the extra tower would - // not match. - G4double extraTowerRmin[2]; - G4double extraTowerRmax[2]; - for(int i = 0; i < 2 ; i++){ - extraTowerRmin[i] = mainAnnulusRmin[i] != 0 ? mainAnnulusRmin[i]/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.) : 0.; - extraTowerRmax[i] = mainAnnulusRmax[i] != 0 ? mainAnnulusRmax[i]/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.) : 0.; - } - G4Polyhedra* solidWCExtraTower = new G4Polyhedra("WCextraTower", - totalAngle-2.*pi,//+dPhi/2., // phi start - 2.*pi - totalAngle // total angle. - -G4GeometryTolerance::GetInstance()->GetSurfaceTolerance()/(10.*m), - // we need this little Gap between the extra tower and the main annulus - // to avoid a shared surface. Without the gap the photons stuck - // at this place for mare than 25 steps and the howl simulation - // crashes. - 1, //NPhi-gon - 2, - mainAnnulusZ, - extraTowerRmin, - extraTowerRmax); - - G4LogicalVolume* logicWCExtraTower = - new G4LogicalVolume(solidWCExtraTower, - G4Material::GetMaterial(water), - "WCExtraTower", - 0,0,0); - G4VPhysicalVolume* physiWCExtraTower = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,0.), - logicWCExtraTower, - "WCExtraTower", - logicWCBarrel, - false, - 0, - checkOverlaps); - - - - logicWCExtraTower->SetVisAttributes(G4VisAttributes::Invisible); - //------------------------------------------- - // subdivide the extra tower into cells - //------------------------------------------ - - G4Polyhedra* solidWCExtraTowerCell = new G4Polyhedra("WCExtraTowerCell", - totalAngle-2.*pi,//+dPhi/2., // phi start - 2.*pi - totalAngle -G4GeometryTolerance::GetInstance()->GetSurfaceTolerance()/(10.*m), //phi end - 1, //NPhi-gon - 2, - RingZ, - extraTowerRmin, - extraTowerRmax); - //G4cout << * solidWCExtraTowerCell << G4endl; - logicWCExtraTowerCell = - new G4LogicalVolume(solidWCExtraTowerCell, - G4Material::GetMaterial(water), - "WCExtraTowerCell", - 0,0,0); - G4VPhysicalVolume* physiWCTowerCell = - new G4PVReplica("extraTowerCell", - logicWCExtraTowerCell, - logicWCExtraTower, - kZAxis, - (G4int)WCBarrelNRings-2, - barrelCellHeight); - - G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(1.,0.5,0.5)); - tmpVisAtt->SetForceSolid(true);// This line is used to give definition to the cells in OGLSX Visualizer - //logicWCExtraTowerCell->SetVisAttributes(tmpVisAtt); - logicWCExtraTowerCell->SetVisAttributes(G4VisAttributes::Invisible); - //TF vis. - - //--------------------------------------------- - // add blacksheet to this cells - //-------------------------------------------- - - G4double towerBSRmin[2]; - G4double towerBSRmax[2]; - for(int i = 0; i < 2; i++){ - towerBSRmin[i] = annulusBlackSheetRmin[i]/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.); - towerBSRmax[i] = annulusBlackSheetRmax[i]/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.); - } - G4Polyhedra* solidWCTowerBlackSheet = new G4Polyhedra("WCExtraTowerBlackSheet", - totalAngle-2.*pi,//+dPhi/2., // phi start - 2.*pi - totalAngle -G4GeometryTolerance::GetInstance()->GetSurfaceTolerance()/(10.*m), //phi end - 1, //NPhi-gon - 2, - RingZ, - towerBSRmin, - towerBSRmax); - //G4cout << * solidWCTowerBlackSheet << G4endl; - logicWCTowerBlackSheet = - new G4LogicalVolume(solidWCTowerBlackSheet, - G4Material::GetMaterial("Blacksheet"), - "WCExtraTowerBlackSheet", - 0,0,0); - - G4VPhysicalVolume* physiWCTowerBlackSheet = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,0.), - logicWCTowerBlackSheet, - "WCExtraTowerBlackSheet", - logicWCExtraTowerCell, - false, - 0, - checkOverlaps); - - - G4LogicalBorderSurface * WaterBSTowerCellSurface - = new G4LogicalBorderSurface("WaterBSBarrelCellSurface", - physiWCTowerCell, - physiWCTowerBlackSheet, - OpWaterBSSurface); - - new G4LogicalSkinSurface("BSTowerCellSkinSurface",logicWCTowerBlackSheet, - BSSkinSurface); - -// These lines add color to the blacksheet in the extratower. If using RayTracer, comment the first chunk and use the second. The Blacksheet should be green. - - if (Vis_Choice == "OGLSX"){ - - G4VisAttributes* WCBarrelBlackSheetCellVisAtt - = new G4VisAttributes(G4Colour(0.2,0.9,0.2)); // green color - - if(!debugMode) - {logicWCTowerBlackSheet->SetVisAttributes(G4VisAttributes::Invisible);} - else - {logicWCTowerBlackSheet->SetVisAttributes(WCBarrelBlackSheetCellVisAtt);}} - - if (Vis_Choice == "RayTracer"){ - - G4VisAttributes* WCBarrelBlackSheetCellVisAtt - = new G4VisAttributes(G4Colour(0.2,0.9,0.2)); // green color - WCBarrelBlackSheetCellVisAtt->SetForceSolid(true); // force the object to be visualized with a surface - WCBarrelBlackSheetCellVisAtt->SetForceAuxEdgeVisible(true); // force auxiliary edges to be shown - - if(!debugMode) - logicWCTowerBlackSheet->SetVisAttributes(WCBarrelBlackSheetCellVisAtt); - else - logicWCTowerBlackSheet->SetVisAttributes(WCBarrelBlackSheetCellVisAtt);} - - else { - - G4VisAttributes* WCBarrelBlackSheetCellVisAtt - = new G4VisAttributes(G4Colour(0.2,0.9,0.2)); // green color - - if(!debugMode) - {logicWCTowerBlackSheet->SetVisAttributes(G4VisAttributes::Invisible);} - else - {logicWCTowerBlackSheet->SetVisAttributes(WCBarrelBlackSheetCellVisAtt);}} - - -} - - //jl145------------------------------------------------ - // Add top veto volume - //----------------------------------------------------- - - G4bool WCTopVeto = (WCSimTuningParams->GetTopVeto()); - - G4LogicalVolume* logicWCTopVeto; - - if(WCTopVeto){ - - G4double WCTyvekThickness = 1.0*mm; //completely made up - - G4VSolid* solidWCTopVeto; - solidWCTopVeto = - new G4Tubs( "WCTopVeto", - 0.0*m, - WCIDRadius + WCTyvekThickness, - 0.5*m + WCTyvekThickness, - 0.*deg, - 360.*deg); - - logicWCTopVeto = - new G4LogicalVolume(solidWCTopVeto, - G4Material::GetMaterial(water), - "WCTopVeto", - 0,0,0); - - G4VPhysicalVolume* physiWCTopVeto = - new G4PVPlacement( 0, - G4ThreeVector(0.,0.,WCIDHeight/2 - +1.0*m), - logicWCTopVeto, - "WCTopVeto", - logicWCBarrel, - false,0, - checkOverlaps); - - //Add the top veto Tyvek - //----------------------------------------------------- - - G4VSolid* solidWCTVTyvek; - solidWCTVTyvek = - new G4Tubs( "WCTVTyvek", - 0.0*m, - WCIDRadius, - WCTyvekThickness/2, - 0.*deg, - 360.*deg); - - - G4LogicalVolume* logicWCTVTyvek = - new G4LogicalVolume(solidWCTVTyvek, - G4Material::GetMaterial("Tyvek"), - "WCTVTyvek", - 0,0,0); - - //Bottom - G4VPhysicalVolume* physiWCTVTyvekBot = - new G4PVPlacement( 0, - G4ThreeVector(0.,0.,-0.5*m - -WCTyvekThickness/2), - logicWCTVTyvek, - "WCTVTyvekBot", - logicWCTopVeto, - false,0, - checkOverlaps); - - G4LogicalBorderSurface * WaterTyTVSurfaceBot = - new G4LogicalBorderSurface( "WaterTyTVSurfaceBot", - physiWCTopVeto, - physiWCTVTyvekBot, - OpWaterTySurface); - - //Top - G4VPhysicalVolume* physiWCTVTyvekTop = - new G4PVPlacement( 0, - G4ThreeVector(0.,0.,0.5*m - +WCTyvekThickness/2), - logicWCTVTyvek, - "WCTVTyvekTop", - logicWCTopVeto, - false,0, - checkOverlaps); - - G4LogicalBorderSurface * WaterTyTVSurfaceTop = - new G4LogicalBorderSurface( "WaterTyTVSurfaceTop", - physiWCTopVeto, - physiWCTVTyvekTop, - OpWaterTySurface); - - //Side - G4VSolid* solidWCTVTyvekSide; - solidWCTVTyvekSide = - new G4Tubs( "WCTVTyvekSide", - WCIDRadius, - WCIDRadius + WCTyvekThickness, - 0.5*m + WCTyvekThickness, - 0.*deg, - 360.*deg); - - - G4LogicalVolume* logicWCTVTyvekSide = - new G4LogicalVolume(solidWCTVTyvekSide, - G4Material::GetMaterial("Tyvek"), - "WCTVTyvekSide", - 0,0,0); - - G4VPhysicalVolume* physiWCTVTyvekSide = - new G4PVPlacement( 0, - G4ThreeVector(0.,0.,0.), - logicWCTVTyvekSide, - "WCTVTyvekSide", - logicWCTopVeto, - false,0, - checkOverlaps); - - G4LogicalBorderSurface * WaterTyTVSurfaceSide = - new G4LogicalBorderSurface( "WaterTyTVSurfaceSide", - physiWCTopVeto, - physiWCTVTyvekSide, - OpWaterTySurface); - - } - - // - // - //jl145------------------------------------------------ - - - //////////// M Fechner : I need to declare the PMT here in order to - // place the PMTs in the truncated cells - //----------------------------------------------------- - // The PMT - //----------------------------------------------------- - - ////////////J Felde: The PMT logical volume is now constructed separately - // from any specific detector geometry so that any geometry can use the same definition. - // K.Zbiri: The PMT volume and the PMT glass are now put in parallel. - // The PMT glass is the sensitive volume in this new configuration. - - // TF: Args are set to properties of the class which is somehow global (see the ConstructDetector.hh) - // They are set in the WCSimDetectorConfigs and are property of the PMT. - G4LogicalVolume* logicWCPMT; - if(nID_PMTs<=1) logicWCPMT = ConstructPMT(WCPMTName, WCIDCollectionName,"tank",nID_PMTs); - else logicWCPMT = ConstructMultiPMT(WCPMTName, WCIDCollectionName,"tank",nID_PMTs); - if(!logicWCPMT){ - G4cerr << "Overlapping PMTs in multiPMT" << G4endl; - return NULL; - } - G4LogicalVolume* logicWCPMT2; -#ifdef DEBUG - G4cout << "HYBRID2 = " << hybrid << G4endl; -#endif - if(hybrid){ - G4cout<<"First type of PMT LV is created. Now creating the LV for the second type of PMT"<SetForceSolid(true); - WClogic->SetForceAuxEdgeVisible(true); - - //logicWCPMT->SetVisAttributes(WClogic); - //logicWCPMT->SetVisAttributes(G4VisAttributes::Invisible); - - //jl145------------------------------------------------ - // Add top veto PMTs - //----------------------------------------------------- - - if(WCTopVeto){ - - G4double WCTVPMTSpacing = (WCSimTuningParams->GetTVSpacing())*cm; - G4double WCTVEdgeLimit = WCCapEdgeLimit; - G4int TVNCell = WCTVEdgeLimit/WCTVPMTSpacing + 2; - - int icopy = 0; - -#ifdef DEBUG - G4cout<<"Debug B.Q: Constructing the cells for veto"<rotateY(90.*deg); //if mPMT: perp to wall - else if(orientation == VERTICAL) - WCPMTRotation->rotateY(0.*deg); //if mPMT: vertical/aligned to wall - else if(orientation == HORIZONTAL) - WCPMTRotation->rotateX(90.*deg); //if mPMT: horizontal to wall - - - - - - - - - - G4double barrelCellWidth = 2.*WCIDRadius*tan(dPhi/2.); - G4double horizontalSpacing = barrelCellWidth/WCPMTperCellHorizontal; - G4double verticalSpacing = barrelCellHeight/WCPMTperCellVertical; - - if(placeBarrelPMTs){ - -#ifdef DEBUG - G4cout << "Debug B.Q : Number of PMT horizontal = " << WCPMTperCellHorizontal << ", number of PMT vertical = " << WCPMTperCellVertical << G4endl; - G4cout << "Debug B.Q : Horizontal spacing = "<< horizontalSpacing << " vertical spacing = " << verticalSpacing << G4endl; -#endif - - for(G4double i = 0; i < WCPMTperCellHorizontal; i++){ - for(G4double j = 0; j < WCPMTperCellVertical; j++){ - G4ThreeVector PMTPosition = G4ThreeVector(WCIDRadius, - -barrelCellWidth/2.+(i+0.5)*horizontalSpacing, - -barrelCellHeight/2.+(j+0.5)*verticalSpacing); - - G4VPhysicalVolume* physiWCBarrelPMT = - new G4PVPlacement(WCPMTRotation, // its rotation - PMTPosition, - (i==j && hybrid && WCPMTPercentCoverage2!=0)?logicWCPMT2:logicWCPMT, // its logical volume - pmtname,//"WCPMT", // its name - logicWCBarrelCell, // its mother volume - false, // no boolean operations - (int)(i*WCPMTperCellVertical+j), - checkOverlapsPMT); - - // logicWCPMT->GetDaughter(0),physiCapPMT is the glass face. If you add more - // daugter volumes to the PMTs (e.g. a acryl cover) you have to check, if - // this is still the case. - } - } - - //------------------------------------------------------------- - // Add PMTs in extra Tower if necessary - //------------------------------------------------------------ - - - if(!(WCBarrelRingNPhi*WCPMTperCellHorizontal == WCBarrelNumPMTHorizontal)){ - -#ifdef DEBUG - G4cout << "Debug B.Q: Wish to place extra PMTs" << G4endl; - G4cout << "Debug B.Q : WCBarrelRingNPhi = " << WCBarrelRingNPhi << ", WCBarrelNumPMTHorizontal = " << WCBarrelNumPMTHorizontal << G4endl; -#endif - - G4RotationMatrix* WCPMTRotation = new G4RotationMatrix; - WCPMTRotation->rotateY(90.*deg); - WCPMTRotation->rotateX((2*pi-totalAngle)/2.);//align the PMT with the Cell - //Stopped here: B.Q - G4double towerWidth = WCIDRadius*tan(2*pi-totalAngle); - - G4double horizontalSpacing = towerWidth/(WCBarrelNumPMTHorizontal-WCBarrelRingNPhi*WCPMTperCellHorizontal); - G4double verticalSpacing = barrelCellHeight/WCPMTperCellVertical; - - for(G4double i = 0; i < (WCBarrelNumPMTHorizontal-WCBarrelRingNPhi*WCPMTperCellHorizontal); i++){ - for(G4double j = 0; j < WCPMTperCellVertical; j++){ - G4ThreeVector PMTPosition = G4ThreeVector(WCIDRadius/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.), - towerWidth/2.-(i+0.5)*horizontalSpacing, - -barrelCellHeight/2.+(j+0.5)*verticalSpacing); - PMTPosition.rotateZ(-(2*pi-totalAngle)/2.); // align with the symmetry - //axes of the cell - - G4VPhysicalVolume* physiWCBarrelPMT = - new G4PVPlacement(WCPMTRotation, // its rotation - PMTPosition, - (i==j && hybrid && WCPMTPercentCoverage2!=0)?logicWCPMT2:logicWCPMT, // its logical volume - pmtname, // its name - logicWCExtraTowerCell, // its mother volume - false, // no boolean operations - (int)(i*WCPMTperCellVertical+j), - checkOverlapsPMT); - - // logicWCPMT->GetDaughter(0),physiCapPMT is the glass face. If you add more - // daughter volumes to the PMTs (e.g. a acryl cover) you have to check, if - // this is still the case. - } - } - - } - - }//end if placeBarrelPMTs - - G4LogicalVolume* logicTopCapAssembly = ConstructCaps(-1); - G4LogicalVolume* logicBottomCapAssembly = ConstructCaps(1); - - // These lines make the large cap volume invisible to view the caps blacksheets. Need to make invisible for - // RayTracer - if (Vis_Choice == "RayTracer"){ - logicBottomCapAssembly->SetVisAttributes(G4VisAttributes::Invisible); - logicTopCapAssembly->SetVisAttributes(G4VisAttributes::Invisible); - } else { - logicBottomCapAssembly->SetVisAttributes(G4VisAttributes::Invisible); - logicTopCapAssembly->SetVisAttributes(G4VisAttributes::Invisible); - } - - G4VPhysicalVolume* physiTopCapAssembly = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,(mainAnnulusHeight/2.+ capAssemblyHeight/2.)), - logicTopCapAssembly, - "TopCapAssembly", - logicWCBarrel, - false, 0, - checkOverlaps); - - G4VPhysicalVolume* physiBottomCapAssembly = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,(-mainAnnulusHeight/2.- capAssemblyHeight/2.)), - logicBottomCapAssembly, - "BottomCapAssembly", - logicWCBarrel, - false, 0, - checkOverlaps); - - return logicWC; -} - - -G4LogicalVolume* WCSimDetectorConstruction::ConstructCaps(G4int zflip) -{ - capAssemblyHeight = (WCIDHeight-mainAnnulusHeight)/2+1*mm+WCBlackSheetThickness; - - G4Tubs* solidCapAssembly = new G4Tubs("CapAssembly", - 0.0*m, - outerAnnulusRadius/cos(dPhi/2.), - capAssemblyHeight/2, - 0.*deg, - 360.*deg); - - G4LogicalVolume* logicCapAssembly = - new G4LogicalVolume(solidCapAssembly, - G4Material::GetMaterial(water), - "CapAssembly", - 0,0,0); - - - //---------------------------------------------------- - // extra rings for the top and bottom of the annulus - //--------------------------------------------------- - G4double borderAnnulusZ[3] = {(-barrelCellHeight/2.-(WCIDRadius-innerAnnulusRadius))*zflip, - -barrelCellHeight/2.*zflip, - barrelCellHeight/2.*zflip}; - G4double borderAnnulusRmin[3] = { WCIDRadius, innerAnnulusRadius, innerAnnulusRadius}; - G4double borderAnnulusRmax[3] = {outerAnnulusRadius, outerAnnulusRadius,outerAnnulusRadius}; - G4Polyhedra* solidWCBarrelBorderRing = new G4Polyhedra("WCBarrelBorderRing", - 0.*deg, // phi start - totalAngle, - (G4int)WCBarrelRingNPhi, //NPhi-gon - 3, - borderAnnulusZ, - borderAnnulusRmin, - borderAnnulusRmax); - G4LogicalVolume* logicWCBarrelBorderRing = - new G4LogicalVolume(solidWCBarrelBorderRing, - G4Material::GetMaterial(water), - "WCBarrelRing", - 0,0,0); - //G4cout << *solidWCBarrelBorderRing << G4endl; - - G4VPhysicalVolume* physiWCBarrelBorderRing = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,(capAssemblyHeight/2.- barrelCellHeight/2.)*zflip), - logicWCBarrelBorderRing, - "WCBarrelBorderRing", - logicCapAssembly, - false, 0, - checkOverlaps); - - - - if(!debugMode){ - - G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(1.,0.5,0.5)); - tmpVisAtt->SetForceSolid(true);// This line is used to give definition to the cells in OGLSX Visualizer - //logicWCBarrelBorderRing->SetVisAttributes(tmpVisAtt); - logicWCBarrelBorderRing->SetVisAttributes(G4VisAttributes::Invisible); - //TF vis. - } - - //---------------------------------------------------- - // Subdivide border rings into cells - // -------------------------------------------------- - G4Polyhedra* solidWCBarrelBorderCell = new G4Polyhedra("WCBarrelBorderCell", - -dPhi/2., // phi start - dPhi, //total angle - 1, //NPhi-gon - 3, - borderAnnulusZ, - borderAnnulusRmin, - borderAnnulusRmax); - - G4LogicalVolume* logicWCBarrelBorderCell = - new G4LogicalVolume(solidWCBarrelBorderCell, - G4Material::GetMaterial(water), - "WCBarrelBorderCell", - 0,0,0); - //G4cout << *solidWCBarrelBorderCell << G4endl; - G4VPhysicalVolume* physiWCBarrelBorderCell = - new G4PVReplica("WCBarrelBorderCell", - logicWCBarrelBorderCell, - logicWCBarrelBorderRing, - kPhi, - (G4int)WCBarrelRingNPhi, - dPhi, - 0.); - -// These lines of code below will turn the border rings invisible. - -// used for RayTracer - if (Vis_Choice == "RayTracer"){ - - if(!debugMode){ - G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(1.,0.5,0.5)); - tmpVisAtt->SetForceSolid(true); - logicWCBarrelBorderCell->SetVisAttributes(tmpVisAtt); - logicWCBarrelBorderCell->SetVisAttributes(G4VisAttributes::Invisible);} - else { - G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(1.,0.5,0.5)); - tmpVisAtt->SetForceWireframe(true); - logicWCBarrelBorderCell->SetVisAttributes(tmpVisAtt); - logicWCBarrelBorderCell->SetVisAttributes(G4VisAttributes::Invisible);}} - -// used for OGLSX - else { - - if(!debugMode) - { - G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(1.,0.5,0.5)); - tmpVisAtt->SetForceSolid(true); - logicWCBarrelBorderCell->SetVisAttributes(tmpVisAtt); - logicWCBarrelBorderCell->SetVisAttributes(G4VisAttributes::Invisible); - } - else { - G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(1.,0.5,0.5)); - //tmpVisAtt->SetForceWireframe(true); - tmpVisAtt->SetForceSolid(true); - logicWCBarrelBorderCell->SetVisAttributes(tmpVisAtt);}} - - - //------------------------------------------------------------ - // add blacksheet to the border cells. - // We can use the same logical volume as for the normal - // barrel cells. - // --------------------------------------------------------- - - - G4VPhysicalVolume* physiWCBarrelBorderCellBlackSheet = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,0.), - logicWCBarrelCellBlackSheet, - "WCBarrelCellBlackSheet", - logicWCBarrelBorderCell, - false, - 0, - checkOverlaps); - - G4LogicalBorderSurface * WaterBSBarrelBorderCellSurface - = new G4LogicalBorderSurface("WaterBSBarrelCellSurface", - physiWCBarrelBorderCell, - physiWCBarrelBorderCellBlackSheet, - OpWaterBSSurface); - - // we have to declare the logical Volumes - // outside of the if block to access it later on - G4LogicalVolume* logicWCExtraTowerCell; - G4LogicalVolume* logicWCExtraBorderCell; - if(!(WCBarrelRingNPhi*WCPMTperCellHorizontal == WCBarrelNumPMTHorizontal)){ - //---------------------------------------------- - // also the extra tower need special cells at the - // top and th bottom. - // (the top cell is created later on by reflecting the - // bottom cell) - //--------------------------------------------- - G4double extraBorderRmin[3]; - G4double extraBorderRmax[3]; - for(int i = 0; i < 3; i++){ - extraBorderRmin[i] = borderAnnulusRmin[i]/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.); - extraBorderRmax[i] = borderAnnulusRmax[i]/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.); - } - G4Polyhedra* solidWCExtraBorderCell = new G4Polyhedra("WCspecialBarrelBorderCell", - totalAngle-2.*pi,//+dPhi/2., // phi start - 2.*pi - totalAngle -G4GeometryTolerance::GetInstance()->GetSurfaceTolerance()/(10.*m), //total phi - 1, //NPhi-gon - 3, - borderAnnulusZ, - extraBorderRmin, - extraBorderRmax); - - logicWCExtraBorderCell = - new G4LogicalVolume(solidWCExtraBorderCell, - G4Material::GetMaterial(water), - "WCspecialBarrelBorderCell", - 0,0,0); - //G4cout << *solidWCExtraBorderCell << G4endl; - - G4VPhysicalVolume* physiWCExtraBorderCell = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,(capAssemblyHeight/2.- barrelCellHeight/2.)*zflip), - logicWCExtraBorderCell, - "WCExtraTowerBorderCell", - logicCapAssembly, - false, 0, - checkOverlaps); - - G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(1.,0.5,0.5)); - tmpVisAtt->SetForceSolid(true);// This line is used to give definition to the cells in OGLSX Visualizer - logicWCExtraBorderCell->SetVisAttributes(tmpVisAtt); - logicWCExtraBorderCell->SetVisAttributes(G4VisAttributes::Invisible); - //TF vis. - - - G4VPhysicalVolume* physiWCExtraBorderBlackSheet = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,0.), - logicWCTowerBlackSheet, - "WCExtraTowerBlackSheet", - logicWCExtraBorderCell, - false, - 0, - checkOverlaps); - - G4LogicalBorderSurface * WaterBSExtraBorderCellSurface - = new G4LogicalBorderSurface("WaterBSBarrelCellSurface", - physiWCExtraBorderCell, - physiWCExtraBorderBlackSheet, - OpWaterBSSurface); - - } - //------------------------------------------------------------ - // add caps - // ----------------------------------------------------------- - //crucial to match with borderAnnulusZ - G4double capZ[4] = { (-WCBlackSheetThickness-1.*mm)*zflip, - (WCBarrelPMTOffset - (WCIDRadius-innerAnnulusRadius))*zflip, - (WCBarrelPMTOffset - (WCIDRadius-innerAnnulusRadius))*zflip, - WCBarrelPMTOffset*zflip} ; - - G4double capRmin[4] = { 0. , 0., 0., 0.} ; - G4double capRmax[4] = {outerAnnulusRadius, outerAnnulusRadius, WCIDRadius, innerAnnulusRadius}; - G4VSolid* solidWCCap; - -#ifdef DEBUG - G4cout << "B.Q Cap: " << totalAngle << ", " << WCBarrelRingNPhi << ", " << outerAnnulusRadius << ", " << innerAnnulusRadius << ", " << WCBlackSheetThickness << ", " << zflip << ", " << WCBarrelPMTOffset << G4endl; -#endif - - if(WCBarrelRingNPhi*WCPMTperCellHorizontal == WCBarrelNumPMTHorizontal){ - solidWCCap - = new G4Polyhedra("WCCap", - 0.*deg, // phi start - totalAngle, //phi end - (int)WCBarrelRingNPhi, //NPhi-gon - 4, // 2 z-planes - capZ, //position of the Z planes - capRmin, // min radius at the z planes - capRmax// max radius at the Z planes - ); - } else { - // if there is an extra tower, the cap volume is a union of - // to polyhedra. We have to unite both parts, because there are - // PMTs that are on the border between both parts. - G4Polyhedra* mainPart - = new G4Polyhedra("WCCapMainPart", - 0.*deg, // phi start - totalAngle, //phi end - (int)WCBarrelRingNPhi, //NPhi-gon - 4, // 2 z-planes - capZ, //position of the Z planes - capRmin, // min radius at the z planes - capRmax// max radius at the Z planes - ); - G4double extraCapRmin[4]; - G4double extraCapRmax[4]; - for(int i = 0; i < 4 ; i++){ - extraCapRmin[i] = capRmin[i] != 0. ? capRmin[i]/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.) : 0.; - extraCapRmax[i] = capRmax[i] != 0. ? capRmax[i]/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.) : 0.; - } - G4Polyhedra* extraSlice - = new G4Polyhedra("WCCapExtraSlice", - totalAngle-2.*pi, // phi start - 2.*pi - totalAngle -G4GeometryTolerance::GetInstance()->GetSurfaceTolerance()/(10.*m), //total phi - // fortunately there are no PMTs an the gap! - 1, //NPhi-gon - 4, // z-planes - capZ, //position of the Z planes - extraCapRmin, // min radius at the z planes - extraCapRmax// max radius at the Z planes - ); - solidWCCap = - new G4UnionSolid("WCCap", mainPart, extraSlice); - - //G4cout << *solidWCCap << G4endl; - - } - // G4cout << *solidWCCap << G4endl; - G4LogicalVolume* logicWCCap = - new G4LogicalVolume(solidWCCap, - G4Material::GetMaterial(water), - "WCCapPolygon", - 0,0,0); - - G4VPhysicalVolume* physiWCCap = - new G4PVPlacement(0, // no rotation - G4ThreeVector(0.,0.,(-capAssemblyHeight/2.+1*mm+WCBlackSheetThickness)*zflip), // its position - logicWCCap, // its logical volume - "WCCap", // its name - logicCapAssembly, // its mother volume - false, // no boolean operations - 0, // Copy # - checkOverlaps); - - -// used for RayTracer - if (Vis_Choice == "RayTracer"){ - if(!debugMode){ - G4VisAttributes* tmpVisAtt2 = new G4VisAttributes(G4Colour(1,0.5,0.5)); - tmpVisAtt2->SetForceSolid(true); - logicWCCap->SetVisAttributes(tmpVisAtt2); - logicWCCap->SetVisAttributes(G4VisAttributes::Invisible); - - } else{ - - G4VisAttributes* tmpVisAtt2 = new G4VisAttributes(G4Colour(0.6,0.5,0.5)); - tmpVisAtt2->SetForceSolid(true); - logicWCCap->SetVisAttributes(tmpVisAtt2);}} - -// used for OGLSX - else{ - if(!debugMode){ - - G4VisAttributes* tmpVisAtt = new G4VisAttributes(G4Colour(1.,0.5,0.5)); - tmpVisAtt->SetForceSolid(true);// This line is used to give definition to the cells in OGLSX Visualizer - //logicWCCap->SetVisAttributes(tmpVisAtt); - logicWCCap->SetVisAttributes(G4VisAttributes::Invisible); - //TF vis. - - } else - {G4VisAttributes* tmpVisAtt2 = new G4VisAttributes(G4Colour(.6,0.5,0.5)); - tmpVisAtt2->SetForceWireframe(true); - logicWCCap->SetVisAttributes(tmpVisAtt2);}} - - //--------------------------------------------------------------------- - // add cap blacksheet - // ------------------------------------------------------------------- - - G4double capBlackSheetZ[4] = {-WCBlackSheetThickness*zflip, 0., 0., (WCBarrelPMTOffset - (WCIDRadius-innerAnnulusRadius)) *zflip}; - G4double capBlackSheetRmin[4] = {0., 0., WCIDRadius, WCIDRadius}; - G4double capBlackSheetRmax[4] = {WCIDRadius+WCBlackSheetThickness, - WCIDRadius+WCBlackSheetThickness, - WCIDRadius+WCBlackSheetThickness, - WCIDRadius+WCBlackSheetThickness}; - G4VSolid* solidWCCapBlackSheet; - if(WCBarrelRingNPhi*WCPMTperCellHorizontal == WCBarrelNumPMTHorizontal){ - solidWCCapBlackSheet - = new G4Polyhedra("WCCapBlackSheet", - 0.*deg, // phi start - totalAngle, //total phi - WCBarrelRingNPhi, //NPhi-gon - 4, // z-planes - capBlackSheetZ, //position of the Z planes - capBlackSheetRmin, // min radius at the z planes - capBlackSheetRmax// max radius at the Z planes - ); - // G4cout << *solidWCCapBlackSheet << G4endl; - } else { - // same as for the cap volume - G4Polyhedra* mainPart - = new G4Polyhedra("WCCapBlackSheetMainPart", - 0.*deg, // phi start - totalAngle, //phi end - WCBarrelRingNPhi, //NPhi-gon - 4, // z-planes - capBlackSheetZ, //position of the Z planes - capBlackSheetRmin, // min radius at the z planes - capBlackSheetRmax// max radius at the Z planes - ); - G4double extraBSRmin[4]; - G4double extraBSRmax[4]; - for(int i = 0; i < 4 ; i++){ - extraBSRmin[i] = capBlackSheetRmin[i] != 0. ? capBlackSheetRmin[i]/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.) : 0.; - extraBSRmax[i] = capBlackSheetRmax[i] != 0. ? capBlackSheetRmax[i]/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.) : 0.; - } - G4Polyhedra* extraSlice - = new G4Polyhedra("WCCapBlackSheetextraSlice", - totalAngle-2.*pi, // phi start - 2.*pi - totalAngle -G4GeometryTolerance::GetInstance()->GetSurfaceTolerance()/(10.*m), // - WCBarrelRingNPhi, //NPhi-gon - 4, // z-planes - capBlackSheetZ, //position of the Z planes - extraBSRmin, // min radius at the z planes - extraBSRmax// max radius at the Z planes - ); - - solidWCCapBlackSheet = - new G4UnionSolid("WCCapBlackSheet", mainPart, extraSlice); - } - G4LogicalVolume* logicWCCapBlackSheet = - new G4LogicalVolume(solidWCCapBlackSheet, - G4Material::GetMaterial("Blacksheet"), - "WCCapBlackSheet", - 0,0,0); - G4VPhysicalVolume* physiWCCapBlackSheet = - new G4PVPlacement(0, - G4ThreeVector(0.,0.,0.), - logicWCCapBlackSheet, - "WCCapBlackSheet", - logicWCCap, - false, - 0, - checkOverlaps); - - G4LogicalBorderSurface * WaterBSBottomCapSurface - = new G4LogicalBorderSurface("WaterBSCapPolySurface", - physiWCCap,physiWCCapBlackSheet, - OpWaterBSSurface); - - new G4LogicalSkinSurface("BSBottomCapSkinSurface",logicWCCapBlackSheet, - BSSkinSurface); - - G4VisAttributes* WCCapBlackSheetVisAtt - = new G4VisAttributes(G4Colour(0.9,0.2,0.2)); - -// used for OGLSX - if (Vis_Choice == "OGLSX"){ - - G4VisAttributes* WCCapBlackSheetVisAtt - = new G4VisAttributes(G4Colour(0.9,0.2,0.2)); - - if(!debugMode) - logicWCCapBlackSheet->SetVisAttributes(G4VisAttributes::Invisible); - else - logicWCCapBlackSheet->SetVisAttributes(WCCapBlackSheetVisAtt);} - -// used for RayTracer (makes the caps blacksheet yellow) - if (Vis_Choice == "RayTracer"){ - - G4VisAttributes* WCCapBlackSheetVisAtt - = new G4VisAttributes(G4Colour(1.0,1.0,0.0)); - - if(!debugMode) - //logicWCCapBlackSheet->SetVisAttributes(G4VisAttributes::Invisible); //Use this line if you want to make the blacksheet on the caps invisible to view through - logicWCCapBlackSheet->SetVisAttributes(WCCapBlackSheetVisAtt); - else - logicWCCapBlackSheet->SetVisAttributes(WCCapBlackSheetVisAtt);} - - //--------------------------------------------------------- - // Add top and bottom PMTs - // ----------------------------------------------------- - - G4LogicalVolume* logicWCPMT; - if(nID_PMTs<=1) logicWCPMT = ConstructPMT(WCPMTName, WCIDCollectionName,"tank",nID_PMTs); - else logicWCPMT = ConstructMultiPMT(WCPMTName, WCIDCollectionName,"tank",nID_PMTs); - G4LogicalVolume* logicWCPMT2; - if(hybrid){ - if(nID_PMTs2<=1) logicWCPMT2 = ConstructPMT(WCPMTName2, WCIDCollectionName2,"tankPMT2",nID_PMTs2); - else logicWCPMT2 = ConstructMultiPMT(WCPMTName2, WCIDCollectionName2,"tankPMT2",nID_PMTs2); - } - //G4LogicalVolume* logicWCPMT = ConstructPMT(WCPMTName, WCIDCollectionName); - G4String pmtname = "WCMultiPMT"; - - // If using RayTracer and want to view the detector without caps, comment out the top and bottom PMT's - G4double xoffset; - G4double yoffset; - G4int icopy = 0; - - G4RotationMatrix* WCCapPMTRotation = new G4RotationMatrix; - //if mPMT: perp to wall - if(orientation == PERPENDICULAR){ - if(zflip==-1){ - WCCapPMTRotation->rotateY(180.*deg); - } - } - else if (orientation == VERTICAL) - WCCapPMTRotation->rotateY(90.*deg); - else if (orientation == HORIZONTAL) - WCCapPMTRotation->rotateX(90.*deg); - - // loop over the cap - if(placeCapPMTs){ - G4int CapNCell = WCCapEdgeLimit/WCCapPMTSpacing + 2; -#ifdef DEBUG - G4cout << "Debug B.Q, wccap edge = " << WCCapEdgeLimit << ", spacing = " << WCCapPMTSpacing << ", CapNCell = " << CapNCell << ", PMT radius = " << WCPMTRadius << G4endl; - // G4cout << "Debug B.Q, test = " << ((horizontalModulo == verticalModulo) && hybrid && WCPMTPercentCoverage2!=0) << G4endl; -#endif - for ( int i = -CapNCell ; i < CapNCell; i++) { - for (int j = -CapNCell ; j < CapNCell; j++) { - - - xoffset = i*WCCapPMTSpacing + WCCapPMTSpacing*0.5; - yoffset = j*WCCapPMTSpacing + WCCapPMTSpacing*0.5; - - - G4ThreeVector cellpos = G4ThreeVector(xoffset, yoffset, 0); - // G4double WCBarrelEffRadius = WCIDDiameter/2. - WCCapPMTSpacing; - // double comp = xoffset*xoffset + yoffset*yoffset - // - 2.0 * WCBarrelEffRadius * sqrt(xoffset*xoffset+yoffset*yoffset) - // + WCBarrelEffRadius*WCBarrelEffRadius; - // if ( (comp > WCPMTRadius*WCPMTRadius) && ((sqrt(xoffset*xoffset + yoffset*yoffset) + WCPMTRadius) < WCCapEdgeLimit) ) { - double dcenter = hybrid?std::max(WCPMTRadius,WCPMTRadius2):WCPMTRadius; - dcenter+=sqrt(xoffset*xoffset + yoffset*yoffset); - if (dcenter < WCCapEdgeLimit) - - // for debugging boundary cases: - // && ((sqrt(xoffset*xoffset + yoffset*yoffset) + WCPMTRadius) > (WCCapEdgeLimit-100)) ) - { - - //B.Q for Hybrid - G4int horizontalModulo = (i+CapNCell) - WCPMTperCellHorizontal*std::floor((i+CapNCell)/WCPMTperCellHorizontal); - G4int verticalModulo = (j+CapNCell) - WCPMTperCellVertical*std::floor((j+CapNCell)/WCPMTperCellVertical); - -#ifdef DEBUG - //G4cout << "Debug B.Q, xoffset = " << xoffset << ", yoffset = " << yoffset << ", distance = " << dcenter << G4endl; - //G4cout << "Debug B.Q, xoffset = " << xoffset << ", yoffset = " << yoffset << ", CapNCell = " << CapNCell << ", spacing = " << WCCapPMTSpacing << G4endl; - // G4cout << "Debug B.Q, test = " << ((horizontalModulo == verticalModulo) && hybrid && WCPMTPercentCoverage2!=0) << G4endl; -#endif - G4VPhysicalVolume* physiCapPMT = - new G4PVPlacement(WCCapPMTRotation, - cellpos,// its position - ((horizontalModulo == verticalModulo) && hybrid && WCPMTPercentCoverage2!=0)?logicWCPMT2:logicWCPMT, // its logical volume - pmtname, // its name - logicWCCap, // its mother volume - false, // no boolean os - icopy, // every PMT need a unique id. - checkOverlapsPMT); - /* - G4VPhysicalVolume* physiCapPMT = - new G4PVPlacement(WCCapPMTRotation, - cellpos,// its position - logicWCPMT, // its logical volume - pmtname, // its name - logicWCCap, // its mother volume - false, // no boolean os - icopy, // every PMT need a unique id. - checkOverlapsPMT); - */ - // logicWCPMT->GetDaughter(0),physiCapPMT is the glass face. If you add more - // daugter volumes to the PMTs (e.g. a acryl cover) you have to check, if - // this is still the case. - - // ((i%WCPMTperCellHorizontal == j%WCPMTperCellVertical) && hybrid)?logicWCPMT2:logicWCPMT, // its logical volume - - icopy++; - } - } - } - - - G4cout << "total on cap: " << icopy << "\n"; - G4cout << "Coverage was calculated to be: " << (icopy*WCPMTRadius*WCPMTRadius/(WCIDRadius*WCIDRadius)) << "\n"; - }//end if placeCapPMTs - - /////////////// Barrel PMT placement - G4RotationMatrix* WCPMTRotation = new G4RotationMatrix; - if(orientation == PERPENDICULAR) - WCPMTRotation->rotateY(90.*deg); //if mPMT: perp to wall - else if(orientation == VERTICAL) - WCPMTRotation->rotateY(0.*deg); //if mPMT: vertical/aligned to wall - else if(orientation == HORIZONTAL) - WCPMTRotation->rotateX(90.*deg); //if mPMT: horizontal to wall - - - G4double barrelCellWidth = 2.*WCIDRadius*tan(dPhi/2.); - G4double horizontalSpacing = barrelCellWidth/WCPMTperCellHorizontal; - G4double verticalSpacing = barrelCellHeight/WCPMTperCellVertical; - - if(placeBorderPMTs){ -#ifdef DEBUG - G4cout << "Debug B.Q : place border pmt, spacing horiz = " << horizontalSpacing << ", vertical = " << verticalSpacing*zflip << G4endl; -#endif - - for(G4double i = 0; i < WCPMTperCellHorizontal; i++){ - for(G4double j = 0; j < WCPMTperCellVertical; j++){ - G4ThreeVector PMTPosition = G4ThreeVector(WCIDRadius, - -barrelCellWidth/2.+(i+0.5)*horizontalSpacing, - (-barrelCellHeight/2.+(j+0.5)*verticalSpacing)*zflip); - - G4VPhysicalVolume* physiWCBarrelBorderPMT = - new G4PVPlacement(WCPMTRotation, // its rotation - PMTPosition, - (i==j && hybrid && WCPMTPercentCoverage2!=0)?logicWCPMT2:logicWCPMT, // its logical volume - pmtname, // its name - logicWCBarrelBorderCell, // its mother volume - false, // no boolean operations - (int)(i*WCPMTperCellVertical+j), - checkOverlapsPMT); - - // logicWCPMT->GetDaughter(0),physiCapPMT is the glass face. If you add more - // daugter volumes to the PMTs (e.g. a acryl cover) you have to check, if - // this is still the case. - } - } - //------------------------------------------------------------- - // Add PMTs in extra Tower if necessary - //------------------------------------------------------------ - - - if(!(WCBarrelRingNPhi*WCPMTperCellHorizontal == WCBarrelNumPMTHorizontal)){ - -#ifdef DEBUG - G4cout << "Debug B.Q : Add extra tower cap " << G4endl; -#endif - - G4RotationMatrix* WCPMTRotation = new G4RotationMatrix; - WCPMTRotation->rotateY(90.*deg); - WCPMTRotation->rotateX((2*pi-totalAngle)/2.);//align the PMT with the Cell - - G4double towerWidth = WCIDRadius*tan(2*pi-totalAngle); - - G4double horizontalSpacing = towerWidth/(WCBarrelNumPMTHorizontal-WCBarrelRingNPhi*WCPMTperCellHorizontal); - G4double verticalSpacing = barrelCellHeight/WCPMTperCellVertical; - - for(G4double i = 0; i < (WCBarrelNumPMTHorizontal-WCBarrelRingNPhi*WCPMTperCellHorizontal); i++){ - for(G4double j = 0; j < WCPMTperCellVertical; j++){ - G4ThreeVector PMTPosition = G4ThreeVector(WCIDRadius/cos(dPhi/2.)*cos((2.*pi-totalAngle)/2.), - towerWidth/2.-(i+0.5)*horizontalSpacing, - (-barrelCellHeight/2.+(j+0.5)*verticalSpacing)*zflip); - PMTPosition.rotateZ(-(2*pi-totalAngle)/2.); // align with the symmetry - //axes of the cell - - G4VPhysicalVolume* physiWCBarrelBorderPMT = - new G4PVPlacement(WCPMTRotation, // its rotation - PMTPosition, - (i==j && hybrid && WCPMTPercentCoverage2!=0)?logicWCPMT2:logicWCPMT, // its logical volume - "WCPMT", // its name - logicWCExtraBorderCell, // its mother volume - false, // no boolean operations - (int)(i*WCPMTperCellVertical+j), - checkOverlapsPMT); - - // logicWCPMT->GetDaughter(0),physiCapPMT is the glass face. If you add more - // daugter volumes to the PMTs (e.g. a acryl cover) you have to check, if - // this is still the case. - } - } - - } - }//end if placeBorderPMTs -return logicCapAssembly; - -} - - diff --git a/libWCSIM/src/WCSimConstructEggShapedHyperK.cc b/libWCSIM/src/WCSimConstructEggShapedHyperK.cc deleted file mode 100644 index aae8614..0000000 --- a/libWCSIM/src/WCSimConstructEggShapedHyperK.cc +++ /dev/null @@ -1,723 +0,0 @@ -// -*- mode:c++; tab-width:4; -*- -#include "WCSimDetectorConstruction.hh" - -#include "G4SystemOfUnits.hh" -#include "G4ThreeVector.hh" -#include "G4RotationMatrix.hh" -#include "G4Transform3D.hh" - -#include "G4Material.hh" -#include "G4NistManager.hh" - -#include "G4Box.hh" -#include "G4Trap.hh" -#include "G4Tubs.hh" -#include "G4Sphere.hh" - -#include "G4UnionSolid.hh" -#include "G4SubtractionSolid.hh" - -#include "G4LogicalVolume.hh" -#include "G4PVPlacement.hh" -#include "G4PVReplica.hh" - -#include "G4LogicalSkinSurface.hh" - -#include "G4VisAttributes.hh" - -#include "G4SDManager.hh" -#include "WCSimWCSD.hh" - - -/*********************************************************** - * - * This file contains the functions which construct an - * egg-shaped HyperK detector. It used by the HK detector - * configuration modes. It is called in the Construct() - * method in WCSimDetectorConstruction.cc. - * - * Sourcefile for the WCSimDetectorConstruction class - * - ***********************************************************/ - - -G4LogicalVolume* WCSimDetectorConstruction::ConstructEggShapedHyperK() -{ - G4cout << "**** Construct egg-shaped HyperK Detector ****" << G4endl; - - PMTCopyNo = 0; - wallSlabCopyNo = 0; - - // ------------- Volumes -------------- - - checkOverlaps = true; - - // HyperK Volume - - G4LogicalVolume* eggShapedHyperKLV - = new G4LogicalVolume(new G4Box("EggShapedHyperKBox", - waterTank_Height/2., - waterTank_Height/2., - waterTank_Length/2.+blackSheetThickness), - FindMaterial("G4_AIR"), - "EggShapedHyperK"); - eggShapedHyperKLV->SetVisAttributes(G4VisAttributes::Invisible); - - new G4LogicalSkinSurface("WaterBSSurface",eggShapedHyperKLV,OpWaterBSSurface); - -// G4VPhysicalVolume* eggShapedHyperKPV -// = new G4PVPlacement(0,G4ThreeVector(),eggShapedHyperKLV, -// "EggShapedHyperK",0,false,0,checkOverlaps); - - // +/- Z Black Sheets - - G4LogicalVolume* blackSheetZLV - = new G4LogicalVolume(new G4Box("blackSheetZ", - waterTank_Height/2., - waterTank_Height/2., - blackSheetThickness/2.), - FindMaterial("G4_POLYETHYLENE"), - "blackSheetZ"); -// blackSheetZLV->SetVisAttributes(new G4VisAttributes(G4Colour(0,0.5,1.))); - - new G4LogicalSkinSurface("WaterBSSurface",blackSheetZLV,OpWaterBSSurface); - - G4double z = waterTank_Length/2.+blackSheetThickness/2.; - - new G4PVPlacement(0,G4ThreeVector(0,0, z),blackSheetZLV, - "blackSheetZ",eggShapedHyperKLV,false,0,checkOverlaps); - new G4PVPlacement(0,G4ThreeVector(0,0,-z),blackSheetZLV, - "blackSheetZ",eggShapedHyperKLV,false,1,checkOverlaps); - -// ----------------------------------------------------------------- -// G4LogicalVolume* pmtLV = ConstructPMT(innerPMT_Radius,innerPMT_Expose); -// new G4PVPlacement(0,G4ThreeVector(0.,0.,innerPMT_Expose/2.), -// pmtLV,"PMT",eggShapedHyperKLV,false,0,checkOverlaps); -// return eggShapedHyperKLV; -// ----------------------------------------------------------------- - - // The egg-shaped Water Tank - - G4VSolid* waterTank_top = ConstructHalf(waterTank_TopR,waterTank_UpperA); - G4VSolid* waterTank_bot = ConstructHalf(waterTank_BotR,waterTank_LowerB); - - G4RotationMatrix rotm = G4RotationMatrix(); - rotm.rotateZ(180.*degree); - - G4VSolid* waterTank_union - = new G4UnionSolid("Union",waterTank_top,waterTank_bot, - G4Transform3D(rotm, - G4ThreeVector(0,-waterTank_Height/2.,0))); - - waterTankLV - = new G4LogicalVolume(waterTank_union,FindMaterial("G4_WATER"),"Tank"); - waterTankLV->SetVisAttributes(G4VisAttributes::Invisible); - - new G4PVPlacement(0,G4ThreeVector(0,waterTank_Height/4.,0), - waterTankLV,"Tank",eggShapedHyperKLV,false,0,checkOverlaps); - - G4RotationMatrix* g4rot; - G4LogicalVolume* pmtCellLV; - - //Construct the PMTs - - G4LogicalVolume* logicWCPMT = ConstructPMT(WCPMTName, WCIDCollectionName); - G4LogicalVolume* logicWCPMT_OD = ConstructPMT(outerPMT_Name, WCODCollectionName); - - // Radial PMTs - - pmtCellLV = ConstructRadialPMT(true, innerPMT_TopR, innerPMT_Height, - waterTank_UpperA, innerPMT_Expose, - innerPMT_Rpitch, innerPMT_Apitch); - - g4rot = new G4RotationMatrix(); - g4rot->rotateY(90.*deg); - - G4double ratio = innerPMT_Radius/innerPMT_TopR; - G4double r = innerPMT_TopR*std::sqrt(1.-ratio*ratio); - r -= innerPMT_Expose/2.; - - new G4PVPlacement(g4rot, - G4ThreeVector(r,0.,0.), - logicWCPMT, - "PMT", - pmtCellLV, - false,PMTCopyNo++,checkOverlaps); - - pmtCellLV = ConstructRadialPMT(false, innerPMT_BotR, innerPMT_Height, - waterTank_LowerB,innerPMT_Expose, - innerPMT_Rpitch, innerPMT_Apitch); - - ratio = innerPMT_Radius/innerPMT_BotR; - r = innerPMT_BotR*std::sqrt(1.-ratio*ratio); - r -= innerPMT_Expose/2.; - - new G4PVPlacement(g4rot, - G4ThreeVector(r,0.,0.), - logicWCPMT, - "PMT", - pmtCellLV, - false,PMTCopyNo++,checkOverlaps); - - pmtCellLV = ConstructRadialPMT(true, outerPMT_TopR, outerPMT_Height, - waterTank_UpperA, outerPMT_Expose, - outerPMT_TopRpitch, outerPMT_Apitch); - - g4rot = new G4RotationMatrix(); - g4rot->rotateY(-90.*deg); - - G4double phi = outerPMT_TopRpitch/outerPMT_TopR/3; - G4double dphi = innerPMT_Radius/innerPMT_TopR; - - G4double angle = -phi + dphi; - - ratio = outerPMT_Radius/outerPMT_TopR; - r = outerPMT_TopR*std::sqrt(1.-ratio*ratio); - // don't understand this reall.... - r -= outerPMT_Expose; - - G4double x = r * std::cos(angle); - G4double y = r * std::sin(angle); -// G4double x = (outerPMT_TopR+outerPMT_Expose/2.) * std::cos(angle); -// G4double y = (outerPMT_TopR+outerPMT_Expose/2.) * std::sin(angle); - z = -outerPMT_Apitch/4.; - - new G4PVPlacement(g4rot, G4ThreeVector(x,y,z), - logicWCPMT_OD, - "PMT", - pmtCellLV, - false,PMTCopyNo++,checkOverlaps); - - angle = phi - dphi; - - x = r * std::cos(angle); - y = r * std::sin(angle); -// x = (outerPMT_TopR+outerPMT_Expose/2.) * std::cos(angle); -// y = (outerPMT_TopR+outerPMT_Expose/2.) * std::sin(angle); - z = outerPMT_Apitch/4.; - - new G4PVPlacement(g4rot, G4ThreeVector(x,y,z), - logicWCPMT_OD, - "PMT", - pmtCellLV, - false,PMTCopyNo++,checkOverlaps); - - pmtCellLV = ConstructRadialPMT(false, outerPMT_BotR, outerPMT_Height, - waterTank_LowerB, outerPMT_Expose, - outerPMT_BotRpitch, outerPMT_Apitch); - - phi = outerPMT_BotRpitch/outerPMT_BotR/3; - dphi = innerPMT_Radius/innerPMT_BotR; - - angle = -phi + dphi; - - ratio = outerPMT_Radius/outerPMT_BotR; - r = outerPMT_BotR*std::sqrt(1.-ratio*ratio); - // don't understand this reall.... - r -= outerPMT_Expose; - - x = r * std::cos(angle); - y = r * std::sin(angle); -// x = (outerPMT_BotR+outerPMT_Expose/2.) * std::cos(angle); -// y = (outerPMT_BotR+outerPMT_Expose/2.) * std::sin(angle); - z = -outerPMT_Apitch/4.; - - new G4PVPlacement(g4rot, G4ThreeVector(x,y,z), - logicWCPMT_OD, - "PMT", - pmtCellLV, - false,PMTCopyNo++,checkOverlaps); - - angle = phi - dphi; - - x = r * std::cos(angle); - y = r * std::sin(angle); -// x = (outerPMT_BotR+outerPMT_Expose/2.) * std::cos(angle); -// y = (outerPMT_BotR+outerPMT_Expose/2.) * std::sin(angle); - z = outerPMT_Apitch/4.; - - new G4PVPlacement(g4rot, G4ThreeVector(x,y,z), - logicWCPMT_OD, - "PMT", - pmtCellLV, - false,PMTCopyNo++,checkOverlaps); - - // End Wall PMTs - - pmtCellLV = ConstructEndWallPMT(); - - g4rot = new G4RotationMatrix(); - g4rot->rotateY(180.*deg); - - new G4PVPlacement(g4rot, G4ThreeVector(0.,0.,-innerPMT_Expose/2.), - logicWCPMT, - "PMT", - pmtCellLV, - false,PMTCopyNo++,checkOverlaps); - - // Top and Bottom PMTs - - pmtCellLV = ConstructCeilingPMT(true, innerPMT_TopW, innerPMT_Height, - innerPMT_Apitch, innerPMT_Expose); - - g4rot = new G4RotationMatrix(); - g4rot->rotateX(-90.*deg); - - new G4PVPlacement(g4rot, G4ThreeVector(0.,0.,0.), - logicWCPMT, - "PMT", - pmtCellLV, - false,PMTCopyNo++,checkOverlaps); - - pmtCellLV = ConstructCeilingPMT(false, innerPMT_BotW, innerPMT_Height, - innerPMT_Apitch, innerPMT_Expose); - - new G4PVPlacement(g4rot, G4ThreeVector(0.,0.,0.), - logicWCPMT, - "PMT", - pmtCellLV, - false,PMTCopyNo++,checkOverlaps); - - pmtCellLV = ConstructCeilingPMT(true, outerPMT_TopW, outerPMT_Height, - outerPMT_Apitch, outerPMT_Expose); - - g4rot = new G4RotationMatrix(); - g4rot->rotateX(90.*deg); - - new G4PVPlacement(g4rot, G4ThreeVector(0.,0.,0.), - logicWCPMT_OD, - "PMT", - pmtCellLV, - false,PMTCopyNo++,checkOverlaps); - - pmtCellLV = ConstructCeilingPMT(false, outerPMT_BotW, outerPMT_Height, - outerPMT_Apitch, outerPMT_Expose); - - new G4PVPlacement(g4rot, G4ThreeVector(0.,0.,0.), - logicWCPMT_OD, - "PMT", - pmtCellLV, - false,PMTCopyNo++,checkOverlaps); - - // - // always return the physical HyperK volume - // - return eggShapedHyperKLV; -} - -G4VSolid* WCSimDetectorConstruction::ConstructHalf(G4double waterTank_Radius, - G4double waterTank_OffSet) -{ - G4double sinPhi = (waterTank_Height/2.)/waterTank_Radius; - G4double cosPhi = std::sqrt(1.-sinPhi*sinPhi); - - G4double sinPhiHalf = std::sqrt((1.-cosPhi)/2.); - G4double cosPhiHalf = std::sqrt((1.+cosPhi)/2.); - - G4double phi = std::asin(sinPhi); - - G4Trap* waterTank_trap - = new G4Trap("TankTrap", - waterTank_Length/2.,0.,0., - waterTank_Height/4., - waterTank_Radius-waterTank_OffSet, - waterTank_Radius*cosPhi-waterTank_OffSet, - 0, - waterTank_Height/4., - waterTank_Radius-waterTank_OffSet, - waterTank_Radius*cosPhi-waterTank_OffSet, - 0); - - const G4double epsilon = 1*mm; - G4VSolid* waterTank_slice - = new G4SubtractionSolid("TankSlice", - new G4Tubs("WaterTubs", - 0.,waterTank_Radius, - waterTank_Length/2., - -phi/2.,phi), - new G4Box("WaterBox", - waterTank_Radius*cosPhiHalf, - waterTank_Radius*sinPhiHalf, - waterTank_Length/2.+epsilon)); - - G4RotationMatrix rotm = G4RotationMatrix(); - rotm.rotateZ(phi/2.); - - G4VSolid* waterTank_union1 - = new G4UnionSolid("Union1",waterTank_trap,waterTank_slice, - G4Transform3D(rotm,G4ThreeVector(-waterTank_OffSet, - -waterTank_Height/4., - 0))); - - rotm = G4RotationMatrix(); - rotm.rotateZ(phi/2.); - rotm.rotateY(180.*degree); - - G4VSolid* waterTank_union2 - = new G4UnionSolid("Union2",waterTank_union1,waterTank_slice, - G4Transform3D(rotm,G4ThreeVector(waterTank_OffSet, - -waterTank_Height/4., - 0))); - return waterTank_union2; -} - - -G4LogicalVolume* WCSimDetectorConstruction::ConstructRadialPMT(G4bool top, - G4double radius, - G4double height, - G4double offset, - G4double expose, - G4double pitchR, - G4double pitchZ) -{ - G4bool inner = false; - if (height == innerPMT_Height) inner = true; - - G4double sinPhi = height/radius; - - G4double phi = std::asin(sinPhi); - G4int nphi = (G4int)(radius*phi/pitchR); - - if(radius == innerPMT_TopR){ - nphi = nphi + 1; - innerPMT_TopN = nphi; - } - if(radius == innerPMT_BotR){ - innerPMT_BotN = nphi; - } - if(radius == outerPMT_TopR){ - nphi = innerPMT_TopN/3; - } - if(radius == outerPMT_BotR){ - nphi = innerPMT_BotN/3; - } - - phi = nphi * pitchR/radius; - - G4double dphi = pitchR/radius; - - G4double length = waterTank_Length; - G4int nz = (G4int)(length/pitchZ); - length = nz * pitchZ; - - G4LogicalVolume* pmtAnnulusLV - = new G4LogicalVolume(new G4Tubs("PMTAnnulus", - radius-2.*expose,radius, - length/2., - -phi/2.,phi), - FindMaterial("G4_WATER"), - "PMTAnnulus"); - pmtAnnulusLV->SetVisAttributes(G4VisAttributes::Invisible); - - G4LogicalVolume* blackSheetALV = NULL; - if (inner) { - blackSheetALV = - new G4LogicalVolume(new G4Tubs("blackSheetA", - radius,radius+blackSheetThickness, - length/2., - -phi/2.,phi), - FindMaterial("G4_POLYETHYLENE"), - "blackSheetA"); -// blackSheetALV->SetVisAttributes(new G4VisAttributes(G4Colour(0,0.5,1.))); - new G4LogicalSkinSurface("WaterBSSurface",blackSheetALV,OpWaterBSSurface); - } - - G4LogicalVolume* pmtRingLV - = new G4LogicalVolume(new G4Tubs("PMTRing", - radius-2.*expose,radius, - pitchZ/2., - -phi/2.,phi), - FindMaterial("G4_WATER"), - "PMTRings"); - pmtRingLV->SetVisAttributes(G4VisAttributes::Invisible); - - G4LogicalVolume* pmtCellLV - = new G4LogicalVolume(new G4Tubs("PMTCell", - radius-2.*expose,radius, - pitchZ/2., - -dphi/2.,dphi), - FindMaterial("G4_WATER"), - "PMTCell"); - pmtCellLV->SetVisAttributes(G4VisAttributes::Invisible); - -//----------------------------------------------------------------- -// -// G4Tubs* cell_solid = new G4Tubs("Cell", -// radius-2.*expose,radius, -// pitchZ/2., -// -dphi/2.,dphi); -// G4LogicalVolume* cellLV = new G4LogicalVolume(cell_solid, -// FindMaterial("G4_WATER"), -// "Cell"); -// G4VisAttributes* red = new G4VisAttributes(G4Colour(255/255.,0/255.,0/255.)); -// cellLV->SetVisAttributes(red); -// G4RotationMatrix* g4rot = new G4RotationMatrix(); -// G4double angle = -phi/2.+dphi/2.; -// g4rot->rotateZ(angle); -// *g4rot = g4rot->inverse(); -// for (G4int i=0; irotateZ(-phi/2.); - if(!top)rotm->rotateZ(180.*degree); - - G4int t = 1; - if(!top)t = -1; - - new G4PVPlacement(rotm,G4ThreeVector(-t*offset,-waterTank_Height/4.,0), - pmtAnnulusLV,"PMTAnnulus",waterTankLV,false,0,checkOverlaps); - if (inner) - new G4PVPlacement(rotm,G4ThreeVector(-t*offset,-waterTank_Height/4.,0), - blackSheetALV,"blackSheetA",waterTankLV,false,0,checkOverlaps); - - rotm = new G4RotationMatrix(); - rotm->rotateZ(phi/2.); - if(!top)rotm->rotateZ(180.*degree); - rotm->rotateY(180.*degree); - - new G4PVPlacement(rotm,G4ThreeVector( t* offset,-waterTank_Height/4.,0), - pmtAnnulusLV,"PMTAnnulus",waterTankLV,false,1,checkOverlaps); - if (inner) - new G4PVPlacement(rotm,G4ThreeVector( t* offset,-waterTank_Height/4.,0), - blackSheetALV,"blackSheetA",waterTankLV,false,1,checkOverlaps); - -// return cellLV; - return pmtCellLV; -} - -G4LogicalVolume* WCSimDetectorConstruction::ConstructEndWallPMT() -{ - G4LogicalVolume* pmtCellLV - = new G4LogicalVolume(new G4Box("PMTCell", - innerPMT_Apitch/2., - innerPMT_Apitch/2., - innerPMT_Apitch/2.), - FindMaterial("G4_WATER"), - "PMTCell"); - pmtCellLV->SetVisAttributes(G4VisAttributes::Invisible); - - G4RotationMatrix* rotm = new G4RotationMatrix(); - rotm->rotateY(180.*degree); - - G4int i = 0; - G4double x = innerPMT_Apitch/2.; - G4ThreeVector xv = G4ThreeVector(x,0.,0.); - - while (x < innerPMT_TopR - waterTank_UpperA) - - { - G4int j = 0; - G4double y = innerPMT_Apitch/2.; - G4ThreeVector ov = G4ThreeVector(waterTank_UpperA,0.,0.); - G4ThreeVector yv = G4ThreeVector(0.,y,0.); - -// while (y < innerPMT_Height - innerPMT_Radius) - while (y < innerPMT_Height - 2*innerPMT_Expose) - { - G4ThreeVector dv = ov + xv + yv; -// if (dv.mag() > innerPMT_TopR - innerPMT_Radius) break; - if (dv.mag() > innerPMT_TopR - 2*innerPMT_Expose) break; - j++; - y+=innerPMT_Apitch; - yv = G4ThreeVector(0.,y,0.); - } - G4int ny = j; - j--; - y-=innerPMT_Apitch; - - G4double yt = y; - - j = 0; - y = innerPMT_Apitch/2.; - ov = G4ThreeVector(waterTank_LowerB,0.,0.); - yv = G4ThreeVector(0.,y,0.); - -// while (y < innerPMT_Height - innerPMT_Radius) - while (y < innerPMT_Height - 2*innerPMT_Expose) - { - G4ThreeVector dv = ov + xv + yv; -// if (dv.mag() > innerPMT_BotR - innerPMT_Radius) break; - if (dv.mag() > innerPMT_BotR - 2*innerPMT_Expose) break; - j++; - y+=innerPMT_Apitch; - yv = G4ThreeVector(0.,y,0.); - } - j--; - ny += j; - y-=innerPMT_Apitch; - - G4double yb = y; - - G4LogicalVolume* pmtSlabLV - = new G4LogicalVolume(new G4Box("PMTSlab", - innerPMT_Apitch/2., - (yt+yb)/2., - innerPMT_Apitch/2.), - FindMaterial("G4_WATER"), - "PMTSlab"); - pmtSlabLV->SetVisAttributes(G4VisAttributes::Invisible); - - new G4PVReplica("PMTCell", - pmtCellLV, - pmtSlabLV, - kYAxis, - ny, - innerPMT_Apitch); - - G4double xpos = x-innerPMT_Apitch/2.; - G4double ypos = (yt-yb)/2. - waterTank_Height/4. ; - G4double zpos = waterTank_Length/2.-innerPMT_Apitch/2.; - - new G4PVPlacement(0, G4ThreeVector(xpos,ypos, zpos), - pmtSlabLV,"PMTSlab",waterTankLV,false,wallSlabCopyNo++,checkOverlaps); - new G4PVPlacement(rotm,G4ThreeVector(xpos,ypos,-zpos), - pmtSlabLV,"PMTSlab",waterTankLV,false,wallSlabCopyNo++,checkOverlaps); - if ( i > 0 ) { - xpos = -xpos; - new G4PVPlacement(0, G4ThreeVector(xpos,ypos, zpos), - pmtSlabLV,"PMTSlab",waterTankLV,false,wallSlabCopyNo++,checkOverlaps); - new G4PVPlacement(rotm,G4ThreeVector(xpos,ypos,-zpos), - pmtSlabLV,"PMTSlab",waterTankLV,false,wallSlabCopyNo++,checkOverlaps); - } - - i++; - x+=innerPMT_Apitch; - xv = G4ThreeVector(x,0.,0.); - - } - - return pmtCellLV; -} - -G4LogicalVolume* WCSimDetectorConstruction::ConstructCeilingPMT(G4bool top, - G4double width, - G4double height, - G4double pitch, - G4double size) -{ - G4bool inner = false; - if (height == innerPMT_Height) inner = true; - - G4int nx = (G4int)(2.*width/pitch); - - // ***** to fix the overlap with annulus ***** - nx = nx - 1; - - G4double xlength = nx * pitch; - G4int nz = (G4int)(waterTank_Length/pitch); - G4double zlength = nz * pitch; - - G4LogicalVolume* pmtSlabLV - = new G4LogicalVolume(new G4Box("PMTSlab", - xlength/2., - size/2., - zlength/2.), - FindMaterial("G4_WATER"), - "PMTSlab"); - pmtSlabLV->SetVisAttributes(G4VisAttributes::Invisible); - - G4LogicalVolume* blackSheetYLV = NULL; - if (inner) { - blackSheetYLV = - new G4LogicalVolume(new G4Box("blackSheetY", - xlength/2., - blackSheetThickness/2., - zlength/2.), - FindMaterial("G4_POLYETHYLENE"), - "blackSheetY"); -// blackSheetYLV->SetVisAttributes(new G4VisAttributes(G4Colour(0,0.5,1.))); - new G4LogicalSkinSurface("WaterBSSurface",blackSheetYLV,OpWaterBSSurface); - } - - G4LogicalVolume* pmtSliceLV - = new G4LogicalVolume(new G4Box("PMTSlice", - xlength/2., - size/2., - pitch/2.), - FindMaterial("G4_WATER"), - "PMTSlab"); - pmtSliceLV->SetVisAttributes(G4VisAttributes::Invisible); - - G4LogicalVolume* pmtCellLV - = new G4LogicalVolume(new G4Box("PMTCell", - pitch/2., - size/2., - pitch/2.), - FindMaterial("G4_WATER"), - "PMTCell"); - pmtCellLV->SetVisAttributes(G4VisAttributes::Invisible); - - G4VPhysicalVolume* pmtSlicePV - = new G4PVReplica("PMTSlice", - pmtSliceLV, - pmtSlabLV, - kZAxis, - nz, - pitch); - - new G4PVReplica("PMTCell", - pmtCellLV, - pmtSlicePV, - kXAxis, - nx, - pitch); - - G4double xpos = 0.0*mm; - G4double ypos = (height-size/2.) - waterTank_Height/4.; - G4double zpos = 0.0*mm; - - if (top) { - new G4PVPlacement(0,G4ThreeVector(xpos,ypos,zpos), - pmtSlabLV,"PMTSlab",waterTankLV,false,wallSlabCopyNo++,checkOverlaps); - ypos += size/2. + blackSheetThickness/2.; - if (inner) new G4PVPlacement(0,G4ThreeVector(xpos,ypos,zpos), - blackSheetYLV,"blackSheet",waterTankLV,false,wallSlabCopyNo++,checkOverlaps); - } else { - ypos = (-height+size/2.) - waterTank_Height/4.; - G4RotationMatrix* rotm = new G4RotationMatrix(); - rotm->rotateZ(180.*degree); - new G4PVPlacement(rotm,G4ThreeVector(xpos,ypos,zpos), - pmtSlabLV,"PMTSlab",waterTankLV,false,wallSlabCopyNo++,checkOverlaps); - ypos -= size/2. + blackSheetThickness/2.; - if (inner) new G4PVPlacement(0,G4ThreeVector(xpos,ypos,zpos), - blackSheetYLV,"blackSheet",waterTankLV,false,wallSlabCopyNo++,checkOverlaps); - - } - - return pmtCellLV; -} - -/** - * Mapping allows HK and SK to eventually have different materials - */ -G4Material* WCSimDetectorConstruction::FindMaterial(G4String name) -{ - if (name == "G4_AIR") return FindMaterial("Air"); - if (name == "G4_WATER") return FindMaterial("Water"); - if (name == "G4_POLYETHYLENE") return FindMaterial("Blacksheet"); - G4Material* material = G4Material::GetMaterial(name,true); - return material; -} diff --git a/libWCSIM/src/WCSimConstructGeometryTables.cc b/libWCSIM/src/WCSimConstructGeometryTables.cc deleted file mode 100644 index 7a9b0ac..0000000 --- a/libWCSIM/src/WCSimConstructGeometryTables.cc +++ /dev/null @@ -1,491 +0,0 @@ -#include "WCSimDetectorConstruction.hh" -#include "WCSimPmtInfo.hh" - -#include "G4Material.hh" -#include "G4Element.hh" -#include "G4Box.hh" -#include "G4LogicalVolume.hh" -#include "G4VPhysicalVolume.hh" -#include "G4PVPlacement.hh" -#include "G4ThreeVector.hh" -#include "G4Vector3D.hh" -#include "globals.hh" -#include "G4VisAttributes.hh" -#include "G4Tubs.hh" -#include "G4Sphere.hh" -#include "G4VPVParameterisation.hh" - -#include "G4PhysicalConstants.hh" -#include "G4SystemOfUnits.hh" - -#include -#include - - -using std::setw; -// These routines are object registration routines that you can pass -// to the traversal code. - -void WCSimDetectorConstruction::PrintGeometryTree -(G4VPhysicalVolume* aPV ,int aDepth, int /*replicaNo*/, - const G4Transform3D& aTransform) -{ - for (int levels = 0; levels < aDepth; levels++) G4cout << " "; - G4cout << aPV->GetName() << " Level:" << aDepth - << " Pos:" << aTransform.getTranslation() - << " Rot:" << aTransform.getRotation().getTheta()/deg - << "," << aTransform.getRotation().getPhi()/deg - << "," << aTransform.getRotation().getPsi()/deg - << G4endl; -} - -void WCSimDetectorConstruction::GetWCGeom -(G4VPhysicalVolume* aPV ,int aDepth, int /*replicaNo*/, - const G4Transform3D& aTransform) -{ - - // Grab mishmash of useful information from the tree while traversing it - // This information will later be written to the geometry file - // (Alternatively one might define accessible constants) - - if ((aPV->GetName() == "WCBox")) { // last condition is the HyperK Envelope name. - // Stash info in data member - WCOffset = G4ThreeVector(aTransform.getTranslation().getX()/cm, - aTransform.getTranslation().getY()/cm, - aTransform.getTranslation().getZ()/cm); - WCXRotation = G4ThreeVector(aTransform.getRotation().xx(), aTransform.getRotation().xy(), aTransform.getRotation().xz()); - WCYRotation = G4ThreeVector(aTransform.getRotation().yx(), aTransform.getRotation().yy(), aTransform.getRotation().yz()); - WCZRotation = G4ThreeVector(aTransform.getRotation().zx(), aTransform.getRotation().zy(), aTransform.getRotation().zz()); - } - - // Stash info in data member - // AH Need to store this in CM for it to be understood by SK code - WCPMTSize = WCPMTRadius/cm;// I think this is just a variable no if needed - WCPMTSize2 = WCPMTRadius2/cm;// I think this is just a variable no if needed - - // Note WC can be off-center... get both extremities - static G4float zmin=100000,zmax=-100000.; - static G4float xmin=100000,xmax=-100000.; - static G4float ymin=100000,ymax=-100000.; - if (aDepth == 0) { // Reset for this traversal - xmin=100000,xmax=-100000.; - ymin=100000,ymax=-100000.; - zmin=100000,zmax=-100000.; - } - - if ((aPV->GetName() == "WCCapBlackSheet") || (aPV->GetName().find("glassFaceWCPMT") != std::string::npos)){ - G4float x = aTransform.getTranslation().getX()/cm; - G4float y = aTransform.getTranslation().getY()/cm; - G4float z = aTransform.getTranslation().getZ()/cm; - - if (xxmax){xmax=x;} - - if (yymax){ymax=y;} - - if (zzmax){zmax=z;} - - WCCylInfo[0] = xmax-xmin; - WCCylInfo[1] = ymax-ymin; - WCCylInfo[2] = zmax-zmin; - // G4cout << "determine height: " << zmin << " " << zmax << " " << aPV->GetName()<<" " << z << G4endl; - } -} - -void WCSimDetectorConstruction::DescribeAndRegisterPMT(G4VPhysicalVolume* aPV ,int aDepth, int replicaNo, - const G4Transform3D& aTransform) -{ - static std::string replicaNoString[20]; - - std::stringstream depth; - std::stringstream pvname; - - depth << replicaNo; - pvname << aPV->GetName(); - - replicaNoString[aDepth] = pvname.str() + "-" + depth.str(); - - - //TF: To Consider: add a separate table for mPMT positions? Need to use its orientation anyway - // Could be useful for the near future. Need to add an == WCMultiPMT here then. - if (aPV->GetName()== WCIDCollectionName || aPV->GetName()== WCIDCollectionName2 ||aPV->GetName()== WCODCollectionName ) - { - - // First increment the number of PMTs in the tank. - //totalNumPMTs++; - if(aPV->GetName()== WCIDCollectionName) totalNumPMTs++; - else if(aPV->GetName()== WCIDCollectionName2) totalNumPMTs2++; - - // Put the location of this tube into the location map so we can find - // its ID later. It is coded by its tubeTag string. - // This scheme must match that used in WCSimWCSD::ProcessHits() - - std::string tubeTag; - int mPMT_pmtno = -1; - bool foundString = false; - // - - for (int i=0; i <= aDepth; i++){ - tubeTag += ":" + replicaNoString[i]; - std::string::size_type position = replicaNoString[i].find("pmt-"); - if( position == 0) { - foundString = true; - mPMT_pmtno = atoi(replicaNoString[i].substr(position+4).c_str())+1; - if(mPMT_pmtno == 1) - if(aPV->GetName()== WCIDCollectionName) totalNum_mPMTs++; - else if(aPV->GetName()== WCIDCollectionName2) totalNum_mPMTs2++; - } - } - if(!foundString){ - // to distinguish mPMT PMTs from single PMTs: - mPMT_pmtno = 0; - if(aPV->GetName()== WCIDCollectionName) totalNum_mPMTs++; - else if(aPV->GetName()== WCIDCollectionName2) totalNum_mPMTs2++; - } - - // G4cout << tubeTag << G4endl; - if(aPV->GetName()== WCIDCollectionName){ - if ( tubeLocationMap.find(tubeTag) != tubeLocationMap.end() ) { - G4cerr << "Repeated tube tag: " << tubeTag << G4endl; - G4cerr << "Assigned to both tube #" << tubeLocationMap[tubeTag] << " and #" << totalNumPMTs << G4endl; - G4cerr << "Cannot continue -- hits will not be recorded correctly." << G4endl; - G4cerr << "Please make sure that logical volumes with multiple placements are each given a unique copy number" << G4endl; - assert(false); - } - tubeLocationMap[tubeTag] = totalNumPMTs; - // Put the transform for this tube into the map keyed by its ID - tubeIDMap[totalNumPMTs] = aTransform; - - - mPMTIDMap[totalNumPMTs] = std::make_pair(totalNum_mPMTs,mPMT_pmtno); - //G4cout << "depth " << depth.str() << G4endl; - //G4cout << "tubeLocationmap[" << tubeTag << "]= " << tubeLocationMap[tubeTag] << "\n"; - - // Print - // G4cout << "Tube: "<GetName()== WCIDCollectionName2){ - if ( tubeLocationMap2.find(tubeTag) != tubeLocationMap2.end() ) { - G4cerr << "Repeated tube tag: " << tubeTag << G4endl; - G4cerr << "Assigned to both tube #" << tubeLocationMap2[tubeTag] << " and #" << totalNumPMTs2 << G4endl; - G4cerr << "Cannot continue -- hits will not be recorded correctly." << G4endl; - G4cerr << "Please make sure that logical volumes with multiple placements are each given a unique copy number" << G4endl; - assert(false); - } - tubeLocationMap2[tubeTag] = totalNumPMTs2; - // Put the transform for this tube into the map keyed by its ID - tubeIDMap2[totalNumPMTs2] = aTransform; - - - mPMTIDMap2[totalNumPMTs2] = std::make_pair(totalNum_mPMTs2,mPMT_pmtno); - //G4cout << "depth " << depth.str() << G4endl; - //G4cout << "tubeLocationmap[" << tubeTag << "]= " << tubeLocationMap[tubeTag] << "\n"; - - // Print - // G4cout << "Tube: "< 0)//veto pmt - {cylLocation=3;} - else if (pmtOrientation.z()==1.0)//bottom - {cylLocation=2;} - else if (pmtOrientation.z()==-1.0)//top - {cylLocation=0;} - else // barrel - {cylLocation=1;} - - geoFile.precision(9); - geoFile << setw(4) << tubeID - << " " << setw(4) << mPMTIDMap[tubeID].first - << " " << setw(4) << mPMTIDMap[tubeID].second - << " " << setw(8) << newTransform.getTranslation().getX()/cm - << " " << setw(8) << newTransform.getTranslation().getY()/cm - << " " << setw(8) << newTransform.getTranslation().getZ()/cm - << " " << setw(7) << pmtOrientation.x() - << " " << setw(7) << pmtOrientation.y() - << " " << setw(7) << pmtOrientation.z() - << " " << setw(3) << cylLocation - << G4endl; - - WCSimPmtInfo *new_pmt = new WCSimPmtInfo(cylLocation, - newTransform.getTranslation().getX()/cm, - newTransform.getTranslation().getY()/cm, - newTransform.getTranslation().getZ()/cm, - pmtOrientation.x(), - pmtOrientation.y(), - pmtOrientation.z(), - tubeID, - mPMTIDMap[tubeID].first, - mPMTIDMap[tubeID].second); - - fpmts.push_back(new_pmt); - - } - - //Record location of the second PMT type for the hybrid configuration - for (unsigned int i=0;i 0)//veto pmt - {cylLocation=3;} - else if (pmtOrientation.z()==1.0)//bottom - {cylLocation=2;} - else if (pmtOrientation.z()==-1.0)//top - {cylLocation=0;} - else // barrel - {cylLocation=1;} - - - geoFile.precision(9); - geoFile << setw(4) << tubeID - << " " << setw(4) << mPMTIDMap2[tubeID].first - << " " << setw(4) << mPMTIDMap2[tubeID].second - << " " << setw(8) << newTransform.getTranslation().getX()/cm - << " " << setw(8) << newTransform.getTranslation().getY()/cm - << " " << setw(8) << newTransform.getTranslation().getZ()/cm - << " " << setw(7) << pmtOrientation.x() - << " " << setw(7) << pmtOrientation.y() - << " " << setw(7) << pmtOrientation.z() - << " " << setw(3) << cylLocation - << G4endl; - - WCSimPmtInfo *new_pmt = new WCSimPmtInfo(cylLocation, - newTransform.getTranslation().getX()/cm, - newTransform.getTranslation().getY()/cm, - newTransform.getTranslation().getZ()/cm, - pmtOrientation.x(), - pmtOrientation.y(), - pmtOrientation.z(), - tubeID, - mPMTIDMap2[tubeID].first, - mPMTIDMap2[tubeID].second); - - fpmts2.push_back(new_pmt); - - } - - geoFile.close(); - - std::cout << "Geofile written" << std::endl; -} - - -// Code for traversing the geometry tree. This code is very general you pass -// it a function and it will call the function with the information on each -// object it finds. -// -// The traversal code comes from a combination of me/G4Lab project & -// from source/visualization/modeling/src/G4PhysicalVolumeModel.cc -// -// If you are trying to understand how passing the function works you need -// to understand pointers to member functions... -// -// Also notice that DescriptionFcnPtr is a (complicated) typedef. -// - -void WCSimDetectorConstruction::TraverseReplicas -(G4VPhysicalVolume* aPV, int aDepth, const G4Transform3D& aTransform, - DescriptionFcnPtr registrationRoutine) -{ - // Recursively visit all of the geometry below the physical volume - // pointed to by aPV including replicas. - - G4ThreeVector originalTranslation = aPV->GetTranslation(); - G4RotationMatrix* pOriginalRotation = aPV->GetRotation(); - - if (aPV->IsReplicated() ) - { - EAxis axis; - G4int nReplicas; - G4double width, offset; - G4bool consuming; - - aPV->GetReplicationData(axis,nReplicas,width,offset,consuming); - - for (int n = 0; n < nReplicas; n++) - { - //TF: for mPMT parameterization - // because kZAxis is a dummy value altered by Parameterised volume in MultiPMTParam - if(aPV->IsParameterised() ){ - //G4cout << "Replica No. " << n << G4endl; - //use the actual translation and rotation of the replica. - (aPV->GetParameterisation())->ComputeTransformation(n,aPV); - } - - else{ - - switch(axis) { - default: - case kXAxis: - aPV->SetTranslation(G4ThreeVector - (-width*(nReplicas-1)*0.5+n*width,0,0)); - aPV->SetRotation(0); - break; - case kYAxis: - aPV->SetTranslation(G4ThreeVector - (0,-width*(nReplicas-1)*0.5+n*width,0)); - aPV->SetRotation(0); - break; - case kZAxis: - aPV->SetTranslation(G4ThreeVector - (0,0,-width*(nReplicas-1)*0.5+n*width)); - aPV->SetRotation(0); - break; - case kRho: - //Lib::Out::putL("GeometryVisitor::visit: WARNING:"); - //Lib::Out::putL(" built-in replicated volumes replicated"); - //Lib::Out::putL(" in radius are not yet properly visualizable."); - aPV->SetTranslation(G4ThreeVector(0,0,0)); - aPV->SetRotation(0); - break; - case kPhi: - { - G4RotationMatrix rotation; - rotation.rotateZ(-(offset+(n+0.5)*width)); - // Minus Sign because for the physical volume we need the - // coordinate system rotation. - aPV->SetTranslation(G4ThreeVector(0,0,0)); - aPV->SetRotation(&rotation); - } - break; - - } // axis switch - } - DescribeAndDescendGeometry(aPV, aDepth, n, aTransform, - registrationRoutine); - - } // num replicas for loop - } // if replicated - else - DescribeAndDescendGeometry(aPV, aDepth, aPV->GetCopyNo(), aTransform, - registrationRoutine); - - // Restore original transformation... - aPV->SetTranslation(originalTranslation); - aPV->SetRotation(pOriginalRotation); -} - -void WCSimDetectorConstruction::DescribeAndDescendGeometry -(G4VPhysicalVolume* aPV ,int aDepth, int replicaNo, - const G4Transform3D& aTransform, DescriptionFcnPtr registrationRoutine) -{ - - // Calculate the new transform relative to the old transform - G4Transform3D* transform = - new G4Transform3D(*(aPV->GetObjectRotation()), aPV->GetTranslation()); - - G4Transform3D newTransform = aTransform * (*transform); - delete transform; - - /* - G4cout << aPV->GetObjectRotation()->getPhi() << " " << aPV->GetObjectRotation()->getTheta() << " " - << aPV->GetObjectRotation()->getPsi() << G4endl; - G4cout << aPV->GetTranslation().x() << " " << aPV->GetTranslation().y() << " " - << aPV->GetTranslation().z() << G4endl; - */ - - // Call the routine we use to print out geometry descriptions, make - // tables, etc. The routine was passed here as a parameter. It needs to - // be a member function of the class - - (this->*registrationRoutine)(aPV, aDepth, replicaNo, newTransform); - - int nDaughters = aPV->GetLogicalVolume()->GetNoDaughters(); - - for (int iDaughter = 0; iDaughter < nDaughters; iDaughter++) - TraverseReplicas(aPV->GetLogicalVolume()->GetDaughter(iDaughter), - aDepth+1, newTransform, registrationRoutine); -} - - - -G4double WCSimDetectorConstruction::GetGeo_Dm(G4int i){ - if (i>=0&&i<=2){ - return WCCylInfo[i]; - }else if(i==3){ - return innerradius; - }else{ - return 0; - } -} diff --git a/libWCSIM/src/WCSimConstructMaterials.cc b/libWCSIM/src/WCSimConstructMaterials.cc deleted file mode 100644 index 4a5ef28..0000000 --- a/libWCSIM/src/WCSimConstructMaterials.cc +++ /dev/null @@ -1,1264 +0,0 @@ -#include "WCSimDetectorConstruction.hh" -#include "WCSimTuningParameters.hh" -#include "WCSimPMTObject.hh" - -#include "G4Material.hh" -#include "G4Element.hh" -#include "globals.hh" -#include "G4UnitsTable.hh" -#include "G4NistManager.hh" - -#include "G4LogicalBorderSurface.hh" -#include "G4LogicalSkinSurface.hh" -#include "G4OpBoundaryProcess.hh" - -#include "G4PhysicalConstants.hh" -#include "G4SystemOfUnits.hh" - -void WCSimDetectorConstruction::ConstructMaterials() -{ - //****Materials Definitions**** - - G4double density; - G4double a; - - - G4NistManager *nist_man = G4NistManager::Instance(); - - - //---Vacuum - - density = universe_mean_density; //from PhysicalConstants.h - G4double pressure = 1.e-19*pascal; - G4double temperature = 0.1*kelvin; - a = 1.01*g/mole; - G4Material* Vacuum = - new G4Material("Vacuum", 1., a, density, - kStateGas,temperature,pressure); - - //---Water - - a = 1.01*g/mole; - G4Element* elH - = new G4Element("Hydrogen","H", 1,a); - - a = 16.00*g/mole; - G4Element* elO - = new G4Element("Oxygen","O", 8,a); - - density = 1.00*g/cm3; - G4Material* Water - = new G4Material("Water",density,2); - Water->AddElement(elH, 2); - Water->AddElement(elO, 1); - - //---Ice - - density = 0.92*g/cm3;//Ice - G4Material* Ice = new G4Material("Ice",density,2); - Ice->AddElement(elH, 2); - Ice->AddElement(elO, 1); - - //---Steel - - a= 12.01*g/mole; - G4Element* elC - = new G4Element("Carbon","C", 6,a); - - a = 55.85*g/mole; - G4Element* elFe - = new G4Element("Iron","Fe", 26,a); - - density = 7.8*g/cm3; - G4Material* Steel - = new G4Material("Steel",density,2); - Steel->AddElement(elC, 1.*perCent); - Steel->AddElement(elFe, 99.*perCent); - - //---Stainless steel 304L (simple example, particular alloy can be different) - - a = 51.9961*g/mole; - G4Element* elCr = new G4Element("Chromium", "Cr", 24., a); - - a = 58.6934*g/mole; - G4Element* elNi = new G4Element("Nickel", "Ni", 28., a); - - a = 54.938*g/mole; - G4Element* elMn = new G4Element("Manganese", "Mn", 25., a); - - a = 30.974*g/mole; - G4Element* elP = new G4Element("Phosphore", "P", 15., a); - - a = 28.09*g/mole; - G4Element* elSi = new G4Element("Silicon", "Si", 14., a); - - a = 32.065*g/mole; - G4Element* elS = new G4Element("Sulphur", "S", 16., a); - - density = 7.81*g/cm3; - G4Material* StainlessSteel = new G4Material("StainlessSteel", density, 8); - - StainlessSteel->AddElement(elFe, 70.44*perCent); - StainlessSteel->AddElement(elCr, 18*perCent); - StainlessSteel->AddElement(elC, 0.08*perCent); - StainlessSteel->AddElement(elNi, 8*perCent); - StainlessSteel->AddElement(elP, 0.45*perCent); - StainlessSteel->AddElement(elSi, 1*perCent); - StainlessSteel->AddElement(elMn, 2*perCent); - StainlessSteel->AddElement(elS, 0.03*perCent); - - G4MaterialPropertiesTable *mpt = new G4MaterialPropertiesTable(); - - const G4int nEntries = 2; - G4double photonEnergy[nEntries] = {1.*eV , 7.*eV}; - - //G4double rindex_Steel[nEntries] = {1.462 , 1.462}; // No I haven't gone mad - G4double abslength_Steel[nEntries] = {.001*mm , .001*mm}; - //mpt->AddProperty("RINDEX", photonEnergy, rindex_Steel, nEntries); - mpt->AddProperty("ABSLENGTH", photonEnergy, abslength_Steel, nEntries); - - StainlessSteel->SetMaterialPropertiesTable(mpt); - - //---Solid Dry Ice - - density = 1.563*g/cm3; - G4Material* DryIce = new G4Material("SolidDryIce", density, 2); - DryIce->AddElement(elC, 1); - DryIce->AddElement(elO, 2); - - - - //---Air - - a = 14.01*g/mole; - G4Element* elN - = new G4Element("Nitrogen","N", 7,a); - - density = 1.290*mg/cm3; - G4Material* Air - = new G4Material("Air",density,2); - Air->AddElement(elN, 70.*perCent); - Air->AddElement(elO, 30.*perCent); - - //---Plastic - - density = 0.95*g/cm3; - G4Material* Plastic - = new G4Material("Plastic",density,2); - Plastic->AddElement(elC, 1); - Plastic->AddElement(elH, 2); - - //---Aluminum (element and material) - - a = 26.98*g/mole; - G4Element* elAl = new G4Element("Aluminum", "Al", 13, a); - - density = 2.7*g/cm3; - G4Material* Aluminum - = new G4Material("Aluminum",density,1); - Aluminum->AddElement(elAl, 1); - - - // TF: --Silver (coating for reflector): higher reflectivity, but cutoff at 400nm + bad for water - // Also exposure to air not good for Silver - // cfr. Wikipedia "Optical Coating" - - a = 107.8682*g/mole; - G4Element* elAg = new G4Element("Silver","Ag", 47, a); - - density = 10.5*g/cm3; - G4Material *Silver - = new G4Material("Silver",density,1); - Silver->AddElement(elAg, 1); - - // ToDo: Add combinations of Al with silver coatings - // ... - - - // TF: Add Wacker SilGel for optical coupling (from wacker.com) - // ToDo: update once final type of gel is decided and/or play with the gel in MC!!! - density = 0.98*g/cm3; - G4Material *SilGel - = new G4Material("SilGel",density,1); - SilGel->AddElement(elSi, 1); - - // TF: Add Acrylic from G4 database: - G4Material* Acrylic - = nist_man->FindOrBuildMaterial("G4_PLEXIGLASS"); - - - - - //---Black sheet - - density = 0.95*g/cm3; - G4Material* Blacksheet - = new G4Material("Blacksheet",density,2); - Blacksheet->AddElement(elC, 1); - Blacksheet->AddElement(elH, 2); - - //---Tyvek - jl145 - - density = 0.38*g/cm3; //cf. DuPont product handbook - G4Material* Tyvek - = new G4Material("Tyvek",density,2); - Tyvek->AddElement(elC, 1); //polyethylene - Tyvek->AddElement(elH, 2); - - //---Glass - - density = 2.20*g/cm3; - G4Material* SiO2 = new G4Material("SiO2",density,2); - SiO2->AddElement(elSi, 1); - SiO2->AddElement(elO , 2); - - a = 10.81*g/mole; - G4Element* elB = new G4Element("Boron", "B", 5, a); - - density = 2.46*g/cm3; - G4Material* B2O3 = new G4Material("B2O3",density,2); - B2O3->AddElement(elB, 2); - B2O3->AddElement(elO, 3); - - a = 22.99*g/mole; - G4Element* elNa = new G4Element("Sodium", "Na", 11, a); - - density = 2.27*g/cm3; - G4Material* Na2O = new G4Material("Na2O",density,2); - Na2O->AddElement(elNa, 2); - Na2O->AddElement(elO, 1); - - density = 4.00*g/cm3; - G4Material* Al2O3 = new G4Material("Al2O3",density,2); - Al2O3->AddElement(elAl, 2); - Al2O3->AddElement(elO, 3); - -// G4Material* blackAcryl -// = new G4Material("blackAcryl", density, 3); -// blackAcryl -> AddElement(elH, 6); -// blackAcryl -> AddElement(elC, 4); -// blackAcryl -> AddElement(elO, 2); - - density = 2.23*g/cm3; - G4Material* Glass - = new G4Material("Glass",density,4); - //G4Material* Glass - //= new G4Material("Glass",density,8); //Put in 8 materials later - - Glass->AddMaterial(SiO2, 80.6*perCent); - Glass->AddMaterial(B2O3, 13.0*perCent); - Glass->AddMaterial(Na2O, 4.0*perCent); - Glass->AddMaterial(Al2O3, 2.4*perCent); - //Glass->AddMaterial(Al2O3, 2.3*perCent); - //Put in 2.3 percent if the other 4 materials = 0.1 percent - - //---Rock - - // a = 16.00*g/mole; G4Element* elO = new G4Element("Oxygen","O", 8,a); - // a = 28.09*g/mole; G4Element* elSi = new G4Element("Silicon", "Si", 14., a); - // a = 26.98*g/mole; G4Element* elAl = new G4Element("Aluminum", "Al", 13, a); - // a = 55.85*g/mole; G4Element* elFe = new G4Element("Iron","Fe", 26,a); - a = 40.08*g/mole; G4Element* elCa = new G4Element("Calcium","Ca", 20,a); - // a = 22.99*g/mole; G4Element* elNa = new G4Element("Sodium", "Na", 11, a); - a = 39.10*g/mole; G4Element* elK = new G4Element("Potassium","K", 19,a); - a = 24.30*g/mole; G4Element* elMg = new G4Element("Magnesium","Mg",12,a); - - density = 2.7*g/cm3; - G4Material* Rock = new G4Material("Rock", density, 8); - - //From Daya-Bay - Rock->AddElement(elO, 48.50*perCent); - Rock->AddElement(elSi, 34.30*perCent); - Rock->AddElement(elAl, 8.00*perCent); - Rock->AddElement(elFe, 2.00*perCent); - Rock->AddElement(elCa, 0.20*perCent); - Rock->AddElement(elNa, 2.40*perCent); - Rock->AddElement(elK, 4.50*perCent); - Rock->AddElement(elMg, 0.10*perCent); - - - - // Potential alternative material definitions for HK - // Get nist material manager - //G4NistManager* nist = G4NistManager::Instance(); - //nist->FindOrBuildMaterial("G4_AIR"); - //nist->FindOrBuildMaterial("G4_WATER"); - //nist->FindOrBuildMaterial("G4_Galactic"); - //nist->FindOrBuildMaterial("G4_Pyrex_Glass"); - //nist->FindOrBuildMaterial("G4_POLYETHYLENE"); - - - - - - - -// ------------------------------------------------------------- -// Generate & Add Material Properties Table -// ------------------------------------------------------------- - - - const G4int NUMENTRIES = 32; - - G4double PPCKOV[NUMENTRIES] = - { 2.034E-9*GeV, 2.068E-9*GeV, 2.103E-9*GeV, 2.139E-9*GeV, - 2.177E-9*GeV, 2.216E-9*GeV, 2.256E-9*GeV, 2.298E-9*GeV, - 2.341E-9*GeV, 2.386E-9*GeV, 2.433E-9*GeV, 2.481E-9*GeV, - 2.532E-9*GeV, 2.585E-9*GeV, 2.640E-9*GeV, 2.697E-9*GeV, - 2.757E-9*GeV, 2.820E-9*GeV, 2.885E-9*GeV, 2.954E-9*GeV, - 3.026E-9*GeV, 3.102E-9*GeV, 3.181E-9*GeV, 3.265E-9*GeV, - 3.353E-9*GeV, 3.446E-9*GeV, 3.545E-9*GeV, 3.649E-9*GeV, - 3.760E-9*GeV, 3.877E-9*GeV, 4.002E-9*GeV, 4.136E-9*GeV }; - - - // default values - /* - G4double RINDEX1[NUMENTRIES] = - { 1.3435, 1.344, 1.3445, 1.345, 1.3455, - 1.346, 1.3465, 1.347, 1.3475, 1.348, - 1.3485, 1.3492, 1.35, 1.3505, 1.351, - 1.3518, 1.3522, 1.3530, 1.3535, 1.354, - 1.3545, 1.355, 1.3555, 1.356, 1.3568, - 1.3572, 1.358, 1.3585, 1.359, 1.3595, - 1.36, 1.3608}; - */ - - // M Fechner, values from refsg.F in apdetsim - /* G4double RINDEX1[NUMENTRIES] = - { 1.33332, 1.333364, 1.333396, 1.3343, 1.33465, - 1.33502, 1.3354, 1.33579, 1.3362, 1.33663, 1.33709, - 1.33756, 1.33806, 1.3386, 1.33915, 1.33974, - 1.34037, 1.34105, 1.34176, 1.34253, 1.34336, - 1.34425, 1.34521, 1.34626, 1.3474, 1.34864, - 1.35002, 1.35153, 1.35321, 1.35507, 1.35717, 1.35955 }; - */ - - //From SKDETSIM water absorption - const G4int NUMENTRIES_water=60; - - G4double ENERGY_water[NUMENTRIES_water] = - { 1.56962e-09*GeV, 1.58974e-09*GeV, 1.61039e-09*GeV, 1.63157e-09*GeV, - 1.65333e-09*GeV, 1.67567e-09*GeV, 1.69863e-09*GeV, 1.72222e-09*GeV, - 1.74647e-09*GeV, 1.77142e-09*GeV, 1.7971e-09*GeV, 1.82352e-09*GeV, - 1.85074e-09*GeV, 1.87878e-09*GeV, 1.90769e-09*GeV, 1.93749e-09*GeV, - 1.96825e-09*GeV, 1.99999e-09*GeV, 2.03278e-09*GeV, 2.06666e-09*GeV, - 2.10169e-09*GeV, 2.13793e-09*GeV, 2.17543e-09*GeV, 2.21428e-09*GeV, - 2.25454e-09*GeV, 2.29629e-09*GeV, 2.33962e-09*GeV, 2.38461e-09*GeV, - 2.43137e-09*GeV, 2.47999e-09*GeV, 2.53061e-09*GeV, 2.58333e-09*GeV, - 2.63829e-09*GeV, 2.69565e-09*GeV, 2.75555e-09*GeV, 2.81817e-09*GeV, - 2.88371e-09*GeV, 2.95237e-09*GeV, 3.02438e-09*GeV, 3.09999e-09*GeV, - 3.17948e-09*GeV, 3.26315e-09*GeV, 3.35134e-09*GeV, 3.44444e-09*GeV, - 3.54285e-09*GeV, 3.64705e-09*GeV, 3.75757e-09*GeV, 3.87499e-09*GeV, - 3.99999e-09*GeV, 4.13332e-09*GeV, 4.27585e-09*GeV, 4.42856e-09*GeV, - 4.59258e-09*GeV, 4.76922e-09*GeV, 4.95999e-09*GeV, 5.16665e-09*GeV, - 5.39129e-09*GeV, 5.63635e-09*GeV, 5.90475e-09*GeV, 6.19998e-09*GeV }; - - - - // Air - G4double RINDEX_air[NUMENTRIES_water] = - { 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, - 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, - 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, - 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, - 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, - 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, - 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, - 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, - 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, - 1.0, 1.0, 1.0, 1.0, 1.0, 1.0}; - - - // M Fechner : new ; define the water refraction index using refsg.F - //from skdetsim using the whole range. - G4double RINDEX1[NUMENTRIES_water] = - {1.32885, 1.32906, 1.32927, 1.32948, 1.3297, 1.32992, 1.33014, - 1.33037, 1.3306, 1.33084, 1.33109, 1.33134, 1.3316, 1.33186, 1.33213, - 1.33241, 1.3327, 1.33299, 1.33329, 1.33361, 1.33393, 1.33427, 1.33462, - 1.33498, 1.33536, 1.33576, 1.33617, 1.3366, 1.33705, 1.33753, 1.33803, - 1.33855, 1.33911, 1.3397, 1.34033, 1.341, 1.34172, 1.34248, 1.34331, - 1.34419, 1.34515, 1.3462, 1.34733, 1.34858, 1.34994, 1.35145, 1.35312, - 1.35498, 1.35707, 1.35943, 1.36211, 1.36518, 1.36872, 1.37287, 1.37776, - 1.38362, 1.39074, 1.39956, 1.41075, 1.42535}; - - - G4double ABWFF = 1.0; - - // Get from the tuning parameters - ABWFF = WCSimTuningParams->GetAbwff(); - - //T. Akiri: Values from Skdetsim - G4double ABSORPTION_water[NUMENTRIES_water] = - { - 16.1419*cm*ABWFF, 18.278*cm*ABWFF, 21.0657*cm*ABWFF, 24.8568*cm*ABWFF, 30.3117*cm*ABWFF, - 38.8341*cm*ABWFF, 54.0231*cm*ABWFF, 81.2306*cm*ABWFF, 120.909*cm*ABWFF, 160.238*cm*ABWFF, - 193.771*cm*ABWFF, 215.017*cm*ABWFF, 227.747*cm*ABWFF, 243.85*cm*ABWFF, 294.036*cm*ABWFF, - 321.647*cm*ABWFF, 342.81*cm*ABWFF, 362.827*cm*ABWFF, 378.041*cm*ABWFF, 449.378*cm*ABWFF, - 739.434*cm*ABWFF, 1114.23*cm*ABWFF, 1435.56*cm*ABWFF, 1611.06*cm*ABWFF, 1764.18*cm*ABWFF, - 2100.95*cm*ABWFF, 2292.9*cm*ABWFF, 2431.33*cm*ABWFF, 3053.6*cm*ABWFF, 4838.23*cm*ABWFF, - 6539.65*cm*ABWFF, 7682.63*cm*ABWFF, 9137.28*cm*ABWFF, 12220.9*cm*ABWFF, 15270.7*cm*ABWFF, - 19051.5*cm*ABWFF, 23671.3*cm*ABWFF, 29191.1*cm*ABWFF, 35567.9*cm*ABWFF, 42583*cm*ABWFF, - 49779.6*cm*ABWFF, 56465.3*cm*ABWFF, 61830*cm*ABWFF, 65174.6*cm*ABWFF, 66143.7*cm*ABWFF, - 64820*cm*ABWFF, 61635*cm*ABWFF, 57176.2*cm*ABWFF, 52012.1*cm*ABWFF, 46595.7*cm*ABWFF, - 41242.1*cm*ABWFF, 36146.3*cm*ABWFF, 31415.4*cm*ABWFF, 27097.8*cm*ABWFF, 23205.7*cm*ABWFF, - 19730.3*cm*ABWFF, 16651.6*cm*ABWFF, 13943.6*cm*ABWFF, 11578.1*cm*ABWFF, 9526.13*cm*ABWFF - }; - - //Xin Qian: proposed value for absorption length - // G4double ABSORPTION_water[NUMENTRIES_water] = -// {22.8154*cm*ABWFF, 28.6144*cm*ABWFF, 35.9923*cm*ABWFF, 45.4086*cm*ABWFF, 57.4650*cm*ABWFF, -// 72.9526*cm*ABWFF, 75*cm*ABWFF, 81.2317*cm*ABWFF, 120.901*cm*ABWFF, 160.243*cm*ABWFF, -// 193.797*cm*ABWFF, 215.045*cm*ABWFF, 227.786*cm*ABWFF, 243.893*cm*ABWFF, 294.113*cm*ABWFF, -// 321.735*cm*ABWFF, 342.931*cm*ABWFF, 362.967*cm*ABWFF, 378.212*cm*ABWFF, 449.602*cm*ABWFF, -// 740.143*cm*ABWFF, 1116.06*cm*ABWFF, 1438.78*cm*ABWFF, 1615.48*cm*ABWFF, 1769.86*cm*ABWFF, -// 2109.67*cm*ABWFF, 2304.13*cm*ABWFF, 2444.97*cm*ABWFF, 3076.83*cm*ABWFF, 4901.5*cm*ABWFF, -// 6666.57*cm*ABWFF, 7873.95*cm*ABWFF, 9433.81*cm*ABWFF, 10214.5*cm*ABWFF, 10845.8*cm*ABWFF, -// 15746.9*cm*ABWFF, 20201.8*cm*ABWFF, 22025.8*cm*ABWFF, 21142.2*cm*ABWFF, 15083.9*cm*ABWFF, -// 11751*cm*ABWFF, 8795.34*cm*ABWFF, 8741.23*cm*ABWFF, 7102.37*cm*ABWFF, 6060.68*cm*ABWFF, -// 4498.56*cm*ABWFF, 3039.56*cm*ABWFF, 2232.2*cm*ABWFF, 1938*cm*ABWFF, 1811.58*cm*ABWFF, -// 1610.32*cm*ABWFF, 1338.7*cm*ABWFF, 1095.3*cm*ABWFF, 977.525*cm*ABWFF, 965.258*cm*ABWFF, -// 1082.86*cm*ABWFF, 876.434*cm*ABWFF, 633.723*cm*ABWFF, 389.87*cm*ABWFF, 142.011*cm*ABWFF -// }; - - /*G4double ABSORPTION_water[NUMENTRIES_water] = //old - { 22.8154*cm*ABWFF, 28.6144*cm*ABWFF, 35.9923*cm*ABWFF, 45.4086*cm*ABWFF, 57.4650*cm*ABWFF, - 72.9526*cm*ABWFF, 92.9156*cm*ABWFF, 118.737*cm*ABWFF, 152.255*cm*ABWFF, 195.925*cm*ABWFF, - 202.429*cm*ABWFF, 224.719*cm*ABWFF, 236.407*cm*ABWFF, 245.700*cm*ABWFF, 289.017*cm*ABWFF, - 305.810*cm*ABWFF, 316.456*cm*ABWFF, 326.797*cm*ABWFF, 347.222*cm*ABWFF, 414.938*cm*ABWFF, - 636.943*cm*ABWFF, 934.579*cm*ABWFF, 1245.33*cm*ABWFF, 1402.52*cm*ABWFF, 1550.39*cm*ABWFF, - 1745.20*cm*ABWFF, 1883.24*cm*ABWFF, 2016.13*cm*ABWFF, 2442.18*cm*ABWFF, 3831.28*cm*ABWFF, - 4652.89*cm*ABWFF, 5577.04*cm*ABWFF, 6567.08*cm*ABWFF, 7559.88*cm*ABWFF, 8470.06*cm*ABWFF, - 9205.54*cm*ABWFF, 9690.95*cm*ABWFF, 9888.36*cm*ABWFF, 9804.50*cm*ABWFF, 9482.17*cm*ABWFF, - 8982.77*cm*ABWFF, 8369.39*cm*ABWFF, 7680.31*cm*ABWFF, 6902.11*cm*ABWFF, 6183.84*cm*ABWFF, - 5522.27*cm*ABWFF, 4914.33*cm*ABWFF, 4357.09*cm*ABWFF, 3847.72*cm*ABWFF, 3383.51*cm*ABWFF, - 2961.81*cm*ABWFF, 2580.08*cm*ABWFF, 2235.83*cm*ABWFF, 1926.66*cm*ABWFF, 1650.21*cm*ABWFF, - 1404.21*cm*ABWFF, 1186.44*cm*ABWFF, 994.742*cm*ABWFF, 827.027*cm*ABWFF, 681.278*cm*ABWFF}; - */ - - /* - G4double ABSORPTION_water[NUMENTRIES_water] = //new - {25.3504*cm, 31.7938*cm, 39.9915*cm, 50.454*cm, 63.85*cm, - 81.0584*cm, 103.24*cm, 131.93*cm, 169.172*cm, 217.694*cm, - 224.921*cm, 249.688*cm, 262.674*cm, 273*cm, 321.13*cm, 339.789*cm, - 351.617*cm, 363.108*cm, 385.802*cm, 461.042*cm, 707.714*cm, - 1038.42*cm, 1383.7*cm, 1558.36*cm, 1722.65*cm, 1939.11*cm, - 2092.49*cm, 2240.14*cm, 2962.96*cm, 4967.03*cm, 6368.58*cm, - 8207.56*cm, 10634.2*cm, 13855.3*cm, 18157.3*cm, 23940.2*cm, - 31766*cm, 42431.6*cm, 57074.9*cm, 77335.9*cm, 105598*cm, - 145361*cm, 192434*cm, 183898*cm, 176087*cm, 168913*cm, 162301*cm, - 156187*cm, 150516*cm, 145243*cm, 140327*cm, 135733*cm, 131430*cm, - 127392*cm, 123594*cm, 120016*cm, 116640*cm, 113448*cm, 110426*cm, - 107562*cm}; - */ - // M Fechner: Rayleigh scattering -- as of version 4.6.2 of GEANT, - // one may use one's own Rayleigh scattering lengths (the buffer is no - // longer overwritten for "water", see 4.6.2 release notes) - - // RAYFF = 1/ARAS, for those of you who know SKdetsim... - // actually that's not quite right because the scattering models - // are different; in G4 there is no scattering depolarization - // std value at SK = 0.6. But Mie scattering is implemented - // in SKdetsim and not in G4 - - - // april 2005 : reduced reflections, let's increase scattering... - // sep 09: for the large detector like superK the old values are muc better - //G4double RAYFF = 1.0/1.65; //old - // G4double RAYFF = 1.0/1.5; - - G4double RAYFF = 0.625; - - // Get from the tuning parameters - RAYFF = WCSimTuningParams->GetRayff(); - // printf("RAYFF: %f\n",RAYFF); - - //T. Akiri: Values from Skdetsim - G4double RAYLEIGH_water[NUMENTRIES_water] = { - 386929*cm*RAYFF, 366249*cm*RAYFF, 346398*cm*RAYFF, 327355*cm*RAYFF, 309097*cm*RAYFF, - 291603*cm*RAYFF, 274853*cm*RAYFF, 258825*cm*RAYFF, 243500*cm*RAYFF, 228856*cm*RAYFF, - 214873*cm*RAYFF, 201533*cm*RAYFF, 188816*cm*RAYFF, 176702*cm*RAYFF, 165173*cm*RAYFF, - 154210*cm*RAYFF, 143795*cm*RAYFF, 133910*cm*RAYFF, 124537*cm*RAYFF, 115659*cm*RAYFF, - 107258*cm*RAYFF, 99318.2*cm*RAYFF, 91822.2*cm*RAYFF, 84754*cm*RAYFF, 78097.3*cm*RAYFF, - 71836.5*cm*RAYFF, 65956*cm*RAYFF, 60440.6*cm*RAYFF, 55275.4*cm*RAYFF, 50445.6*cm*RAYFF, - 45937*cm*RAYFF, 41735.2*cm*RAYFF, 37826.6*cm*RAYFF, 34197.6*cm*RAYFF, 30834.9*cm*RAYFF, - 27725.4*cm*RAYFF, 24856.6*cm*RAYFF, 22215.9*cm*RAYFF, 19791.3*cm*RAYFF, 17570.9*cm*RAYFF, - 15543*cm*RAYFF, 13696.6*cm*RAYFF, 12020.5*cm*RAYFF, 10504.1*cm*RAYFF, 9137.15*cm*RAYFF, - 7909.45*cm*RAYFF, 6811.3*cm*RAYFF, 5833.25*cm*RAYFF, 4966.2*cm*RAYFF, 4201.36*cm*RAYFF, - 3530.28*cm*RAYFF, 2944.84*cm*RAYFF, 2437.28*cm*RAYFF, 2000.18*cm*RAYFF, 1626.5*cm*RAYFF, - 1309.55*cm*RAYFF, 1043.03*cm*RAYFF, 821.016*cm*RAYFF, 637.97*cm*RAYFF, 488.754*cm*RAYFF - }; - - /*G4double RAYLEIGH_water[NUMENTRIES_water] = { - 167024.4*cm*RAYFF, 158726.7*cm*RAYFF, 150742*cm*RAYFF, - 143062.5*cm*RAYFF, 135680.2*cm*RAYFF, 128587.4*cm*RAYFF, - 121776.3*cm*RAYFF, 115239.5*cm*RAYFF, 108969.5*cm*RAYFF, - 102958.8*cm*RAYFF, 97200.35*cm*RAYFF, 91686.86*cm*RAYFF, - 86411.33*cm*RAYFF, 81366.79*cm*RAYFF, 76546.42*cm*RAYFF, - 71943.46*cm*RAYFF, 67551.29*cm*RAYFF, 63363.36*cm*RAYFF, - 59373.25*cm*RAYFF, 55574.61*cm*RAYFF, 51961.24*cm*RAYFF, - 48527.00*cm*RAYFF, 45265.87*cm*RAYFF, 42171.94*cm*RAYFF, - 39239.39*cm*RAYFF, 36462.50*cm*RAYFF, 33835.68*cm*RAYFF, - 31353.41*cm*RAYFF, 29010.30*cm*RAYFF, 26801.03*cm*RAYFF, - 24720.42*cm*RAYFF, 22763.36*cm*RAYFF, 20924.88*cm*RAYFF, - 19200.07*cm*RAYFF, 17584.16*cm*RAYFF, 16072.45*cm*RAYFF, - 14660.38*cm*RAYFF, 13343.46*cm*RAYFF, 12117.33*cm*RAYFF, - 10977.70*cm*RAYFF, 9920.416*cm*RAYFF, 8941.407*cm*RAYFF, - 8036.711*cm*RAYFF, 7202.470*cm*RAYFF, 6434.927*cm*RAYFF, - 5730.429*cm*RAYFF, 5085.425*cm*RAYFF, 4496.467*cm*RAYFF, - 3960.210*cm*RAYFF, 3473.413*cm*RAYFF, 3032.937*cm*RAYFF, - 2635.746*cm*RAYFF, 2278.907*cm*RAYFF, 1959.588*cm*RAYFF, - 1675.064*cm*RAYFF, 1422.710*cm*RAYFF, 1200.004*cm*RAYFF, - 1004.528*cm*RAYFF, 833.9666*cm*RAYFF, 686.1063*cm*RAYFF - };*/ - - - // Get from the tuning parameters - G4double MIEFF = WCSimTuningParams->GetMieff(); - //G4double MIEFF = 0.0; - // printf("MIEFF: %f\n",MIEFF); - - //Values extracted from Skdetsim - G4double MIE_water[NUMENTRIES_water] = { - 7790020*cm*MIEFF, 7403010*cm*MIEFF, 7030610*cm*MIEFF, 6672440*cm*MIEFF, 6328120*cm*MIEFF, - 5997320*cm*MIEFF, 5679650*cm*MIEFF, 5374770*cm*MIEFF, 5082340*cm*MIEFF, 4802000*cm*MIEFF, - 4533420*cm*MIEFF, 4276280*cm*MIEFF, 4030220*cm*MIEFF, 3794950*cm*MIEFF, 3570120*cm*MIEFF, - 3355440*cm*MIEFF, 3150590*cm*MIEFF, 2955270*cm*MIEFF, 2769170*cm*MIEFF, 2592000*cm*MIEFF, - 2423470*cm*MIEFF, 2263300*cm*MIEFF, 2111200*cm*MIEFF, 1966900*cm*MIEFF, 1830120*cm*MIEFF, - 1700610*cm*MIEFF, 1578100*cm*MIEFF, 1462320*cm*MIEFF, 1353040*cm*MIEFF, 1250000*cm*MIEFF, - 1152960*cm*MIEFF, 1061680*cm*MIEFF, 975936*cm*MIEFF, 895491*cm*MIEFF, 820125*cm*MIEFF, - 749619*cm*MIEFF, 683760*cm*MIEFF, 622339*cm*MIEFF, 565152*cm*MIEFF, 512000*cm*MIEFF, - 462688*cm*MIEFF, 417027*cm*MIEFF, 374832*cm*MIEFF, 335923*cm*MIEFF, 300125*cm*MIEFF, - 267267*cm*MIEFF, 237184*cm*MIEFF, 209715*cm*MIEFF, 184704*cm*MIEFF, 162000*cm*MIEFF, - 141456*cm*MIEFF, 122931*cm*MIEFF, 106288*cm*MIEFF, 91395.2*cm*MIEFF, 78125*cm*MIEFF, - 66355.2*cm*MIEFF, 55968.2*cm*MIEFF, 46851.2*cm*MIEFF, 38896.2*cm*MIEFF, 32000*cm*MIEFF - }; - - //Mie scattering length values when assuming 10 times larger than Rayleigh scattering. - /*G4double MIE_water[NUMENTRIES_water] = { - 3869290*cm*MIEFF, 3662490*cm*MIEFF, 3463980*cm*MIEFF, 3273550*cm*MIEFF, 3090970*cm*MIEFF, - 2916030*cm*MIEFF, 2748530*cm*MIEFF, 2588250*cm*MIEFF, 2435000*cm*MIEFF, 2288560*cm*MIEFF, - 2148730*cm*MIEFF, 2015330*cm*MIEFF, 1888160*cm*MIEFF, 1767020*cm*MIEFF, 1651730*cm*MIEFF, - 1542100*cm*MIEFF, 1437950*cm*MIEFF, 1339100*cm*MIEFF, 1245370*cm*MIEFF, 1156590*cm*MIEFF, - 1072580*cm*MIEFF, 993182*cm*MIEFF, 918222*cm*MIEFF, 847540*cm*MIEFF, 780973*cm*MIEFF, - 718365*cm*MIEFF, 65956*cm*MIEFF, 604406*cm*MIEFF, 552754*cm*MIEFF, 504456*cm*MIEFF, - 459370*cm*MIEFF, 417352*cm*MIEFF, 378266*cm*MIEFF, 341976*cm*MIEFF, 308349*cm*MIEFF, - 277254*cm*MIEFF, 248566*cm*MIEFF, 222159*cm*MIEFF, 197913*cm*MIEFF, 175709*cm*MIEFF, - 155430*cm*MIEFF, 136966*cm*MIEFF, 120205*cm*MIEFF, 105041*cm*MIEFF, 91371.5*cm*MIEFF, - 79094.5*cm*MIEFF, 68113*cm*MIEFF, 58332.5*cm*MIEFF, 49662*cm*MIEFF, 42013.6*cm*MIEFF, - 35302.8*cm*MIEFF, 29448.4*cm*MIEFF, 24372.8*cm*MIEFF, 20001.8*cm*MIEFF, 16265*cm*MIEFF, - 13095.5*cm*MIEFF, 10430.3*cm*MIEFF, 8210.16*cm*MIEFF, 6379.7*cm*MIEFF, 4887.54*cm*MIEFF - }; - */ - - G4double MIE_water_const[3]={0.4,0.,1};// gforward, gbackward, forward backward ratio - - - //From SKDETSIM - /* - G4double RINDEX_glass[NUMENTRIES] = - { 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600, 1.600, 1.600 }; - */ - // M Fechner : unphysical, I want to reduce reflections - // back to the old value 1.55 - - - /* TF: UNPHYSICAL - G4double RINDEX_glass[NUMENTRIES_water] = - { 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, 1.600, - 1.600, 1.600 }; - */ - - // Based on analytical formula from http://refractiveindex.info/?shelf=glass&book=HIKARI-BK&page=J-BK7A for Borosilicate glass - G4double RINDEX_glass[NUMENTRIES_water] = - { 1.5110, 1.5112, 1.5114, 1.5116, 1.5118, - 1.5121, 1.5123, 1.5126, 1.5128, 1.5131, - 1.5133, 1.5136, 1.5139, 1.5142, 1.5145, - 1.5148, 1.5152, 1.5155, 1.5159, 1.5163, - 1.5167, 1.5171, 1.5176, 1.5180, 1.5185, - 1.5190, 1.5196, 1.5202, 1.5208, 1.5214, - 1.5221, 1.5228, 1.5236, 1.5244, 1.5253, - 1.5263, 1.5273, 1.5284, 1.5296, 1.5309, - 1.5323, 1.5338, 1.5354, 1.5372, 1.5392, - 1.5414, 1.5438, 1.5465, 1.5495, 1.5528, - 1.5566, 1.5608, 1.5657, 1.5713, 1.5777, - 1.5853, 1.5941, 1.6047, 1.6173, 1.6328 }; - - - //G4double RINDEX_blacksheet[NUMENTRIES] = - //{ 2.500, 2.500, 2.500, 2.500, 2.500, 2.500, 2.500, - // 2.500, 2.500, 2.500, 2.500, 2.500, 2.500, 2.500, - // 2.500, 2.500, 2.500, 2.500, 2.500, 2.500, 2.500, - // 2.500, 2.500, 2.500, 2.500, 2.500, 2.500, 2.500, - // 2.500, 2.500, 2.500, 2.500 }; - - - //G4double ABSORPTION1[NUMENTRIES] = - //{344.8*cm, 408.2*cm, 632.9*cm, 917.4*cm, 1234.6*cm, 1388.9*cm, - // 1515.2*cm, 1724.1*cm, 1886.8*cm, 2000.0*cm, 2631.6*cm, 3571.4*cm, - // 4545.5*cm, 4761.9*cm, 5263.2*cm, 5263.2*cm, 5555.6*cm, 5263.2*cm, - // 5263.2*cm, 4761.9*cm, 4545.5*cm, 4166.7*cm, 3703.7*cm, 3333.3*cm, - // 3000.0*cm, 2850.0*cm, 2700.0*cm, 2450.0*cm, 2200.0*cm, 1950.0*cm, - // 1750.0*cm, 1450.0*cm }; - - /* - G4double ABSORPTION_glass[NUMENTRIES] = - { 100.0*cm, 110.0*cm, 120.0*cm, 130.0*cm, 140.0*cm, 150.0*cm, 160.0*cm, - 165.0*cm, 170.0*cm, 175.0*cm, 180.0*cm, 185.0*cm, 190.0*cm, 195.0*cm, - 200.0*cm, 200.0*cm, 200.0*cm, 200.0*cm, 200.0*cm, 195.0*cm, 190.0*cm, - 185.0*cm, 180.0*cm, 175.0*cm, 170.0*cm, 160.0*cm, 150.0*cm, 140.0*cm, - 130.0*cm, 120.0*cm, 110.0*cm, 100.0*cm }; - */ - // M Fechner : the quantum efficiency already takes glass abs into account - - G4double ABSORPTION_glass[NUMENTRIES_water]= - { 1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm, - 1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm, - 1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm, - 1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm, - 1.0e9*cm, 1.0e9*cm, - 1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm, - 1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm, - 1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm, - 1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm,1.0e9*cm, - 1.0e9*cm, 1.0e9*cm }; - - G4double BLACKABS_blacksheet[NUMENTRIES_water] = - { 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, - 1.0e-9*cm, - 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, - 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, - 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, - 1.0e-9*cm, 1.0e-9*cm, - 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, - 1.0e-9*cm, - 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, - 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, - 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, 1.0e-9*cm, - 1.0e-9*cm, 1.0e-9*cm}; - - - //The following reflectivity for blacksheet is obtained from skdetsim - //There is a fudge factor of 2.7 for the reflectivity of blacksheet - //depending on whether SK1 or SK2 simulation is used. - //The BlackSheetFudgeFactor is set to true if you want to use the - //SK2 values, false if not. - //G4double SK1SK2FF = 1.0; - //G4bool BlackSheetFudgeFactor=false; - //G4bool BlackSheetFudgeFactor=true; - // if (BlackSheetFudgeFactor) SK1SK2FF=SK1SK2FF*2.7; - //if (BlackSheetFudgeFactor) SK1SK2FF=SK1SK2FF*1.55; - - //July 1, 2010, F. Beroz: changed SK1SK2FF to BSRFF to avoid confusion, - // since this parameter is not from SK. - - G4double BSRFF = 1.0; - - // Get from the tuning parameters - BSRFF = WCSimTuningParams->GetBsrff(); - - G4double REFLECTIVITY_blacksheet[NUMENTRIES_water] = - { 0.055*BSRFF, 0.055*BSRFF, - 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, - 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, - 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, - 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, - 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, - 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, - 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, 0.055*BSRFF, - 0.055*BSRFF, 0.057*BSRFF, 0.059*BSRFF, 0.060*BSRFF, - 0.059*BSRFF, 0.058*BSRFF, 0.057*BSRFF, 0.055*BSRFF, - 0.050*BSRFF, 0.045*BSRFF, 0.045*BSRFF, 0.045*BSRFF, - 0.045*BSRFF, 0.045*BSRFF, 0.045*BSRFF, 0.045*BSRFF, - 0.045*BSRFF, 0.045*BSRFF, 0.045*BSRFF, 0.045*BSRFF, - 0.045*BSRFF, 0.045*BSRFF, 0.045*BSRFF, 0.045*BSRFF, - 0.045*BSRFF, 0.045*BSRFF, 0.045*BSRFF, 0.045*BSRFF, - 0.045*BSRFF, 0.045*BSRFF }; - - /* - G4double REFLECTIVITY_blacksheet[NUMENTRIES] = - { 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, - 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, - 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, - 0.055*SK1SK2FF, 0.057*SK1SK2FF, 0.059*SK1SK2FF, 0.060*SK1SK2FF, - 0.059*SK1SK2FF, 0.058*SK1SK2FF, 0.057*SK1SK2FF, 0.055*SK1SK2FF, - 0.050*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, - 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, - 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF }; - */ - - /* - G4double REFLECTIVITY_blacksheet[NUMENTRIES_water] = - { 0.055*SK1SK2FF, 0.055*SK1SK2FF, - 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, - 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, - 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, - 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, - 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, - 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, - 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, 0.055*SK1SK2FF, - 0.055*SK1SK2FF, 0.057*SK1SK2FF, 0.059*SK1SK2FF, 0.060*SK1SK2FF, - 0.059*SK1SK2FF, 0.058*SK1SK2FF, 0.057*SK1SK2FF, 0.055*SK1SK2FF, - 0.050*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, - 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, - 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, - 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, - 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF, 0.045*SK1SK2FF , - 0.045*SK1SK2FF, 0.045*SK1SK2FF }; - */ - - //utter fiction at this stage - G4double EFFICIENCY[NUMENTRIES_water] = - { 0.001*m }; - - //utter fiction at this stage, does not matter - G4double RAYLEIGH_air[NUMENTRIES_water] = - { 0.001*m,0.001*m,0.001*m,0.001*m,0.001*m,0.001*m, - 0.001*m,0.001*m,0.001*m,0.001*m,0.001*m,0.001*m, - 0.001*m,0.001*m,0.001*m,0.001*m,0.001*m,0.001*m, - 0.001*m,0.001*m,0.001*m,0.001*m,0.001*m,0.001*m, - 0.001*m,0.001*m,0.001*m,0.001*m,0.001*m,0.001*m, - 0.001*m,0.001*m,0.001*m,0.001*m,0.001*m,0.001*m, - 0.001*m,0.001*m,0.001*m,0.001*m,0.001*m,0.001*m, - 0.001*m,0.001*m,0.001*m,0.001*m,0.001*m,0.001*m, - 0.001*m,0.001*m,0.001*m,0.001*m,0.001*m,0.001*m, - 0.001*m,0.001*m,0.001*m,0.001*m,0.001*m,0.001*m}; - - //utter fiction at this stage, does not matter - G4double MIE_air[NUMENTRIES_water] = - { 0.1*m,0.1*m,0.1*m,0.1*m,0.1*m,0.1*m, - 0.1*m,0.1*m,0.1*m,0.1*m,0.1*m,0.1*m, - 0.1*m,0.1*m,0.1*m,0.1*m,0.1*m,0.1*m, - 0.1*m,0.1*m,0.1*m,0.1*m,0.1*m,0.1*m, - 0.1*m,0.1*m,0.1*m,0.1*m,0.1*m,0.1*m, - 0.1*m,0.1*m,0.1*m,0.1*m,0.1*m,0.1*m, - 0.1*m,0.1*m,0.1*m,0.1*m,0.1*m,0.1*m, - 0.1*m,0.1*m,0.1*m,0.1*m,0.1*m,0.1*m, - 0.1*m,0.1*m,0.1*m,0.1*m,0.1*m,0.1*m, - 0.1*m,0.1*m,0.1*m,0.1*m,0.1*m,0.1*m}; - - G4double MIE_air_const[3]={0.99,0.99,0.8};// gforward, gbackward, forward backward ratio - - - //Not used yet, fictional values - //G4double SPECULARLOBECONSTANT1[NUMENTRIES] = - //{ 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001 }; - - //Not used yet, fictional values - //G4double SPECULARSPIKECONSTANT1[NUMENTRIES] = - //{ 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001 }; - - //Not used yet, fictional values - //G4double BACKSCATTERCONSTANT1[NUMENTRIES] = - //{ 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, 0.001, - // 0.001, 0.001, 0.001, 0.001 }; - - G4double EFFICIENCY_blacksheet[NUMENTRIES_water] = - { 0.0 }; - - //TF: Al, Ag and coatings - G4double REFLECTIVITY_aluminium[NUMENTRIES_water] = //start simple with flat 90% - { 0.9,0.9,0.9,0.9,0.9,0.9, - 0.9,0.9,0.9,0.9,0.9,0.9, - 0.9,0.9,0.9,0.9,0.9,0.9, - 0.9,0.9,0.9,0.9,0.9,0.9, - 0.9,0.9,0.9,0.9,0.9,0.9, - 0.9,0.9,0.9,0.9,0.9,0.9, - 0.9,0.9,0.9,0.9,0.9,0.9, - 0.9,0.9,0.9,0.9,0.9,0.9, - 0.9,0.9,0.9,0.9,0.9,0.9, - 0.9,0.9,0.9,0.9,0.9,0.9}; - - // TF: Properties acrylic from skdetsim: refractive index and absorption is enough - - // photon energies between 295nm and 600nm, using hc/(e*lambda) up to 3 digits - // ToDo: remove comment: python magic: L = ['%.3f*eV' %(h*c/(i*1e-9*eC)) for i in range(295,601)] - // L[::-1] to revert list - G4double ENERGY_skAcrylic[306] = - { 2.066*eV, 2.070*eV, 2.073*eV, 2.077*eV, 2.080*eV, 2.084*eV, 2.087*eV, - 2.091*eV, 2.094*eV, 2.098*eV, 2.101*eV, 2.105*eV, 2.109*eV, 2.112*eV, - 2.116*eV, 2.119*eV, 2.123*eV, 2.127*eV, 2.130*eV, 2.134*eV, 2.138*eV, - 2.141*eV, 2.145*eV, 2.149*eV, 2.153*eV, 2.156*eV, 2.160*eV, 2.164*eV, - 2.168*eV, 2.171*eV, 2.175*eV, 2.179*eV, 2.183*eV, 2.187*eV, 2.191*eV, - 2.194*eV, 2.198*eV, 2.202*eV, 2.206*eV, 2.210*eV, 2.214*eV, 2.218*eV, - 2.222*eV, 2.226*eV, 2.230*eV, 2.234*eV, 2.238*eV, 2.242*eV, 2.246*eV, - 2.250*eV, 2.254*eV, 2.258*eV, 2.262*eV, 2.267*eV, 2.271*eV, 2.275*eV, - 2.279*eV, 2.283*eV, 2.288*eV, 2.292*eV, 2.296*eV, 2.300*eV, 2.305*eV, - 2.309*eV, 2.313*eV, 2.317*eV, 2.322*eV, 2.326*eV, 2.331*eV, 2.335*eV, - 2.339*eV, 2.344*eV, 2.348*eV, 2.353*eV, 2.357*eV, 2.362*eV, 2.366*eV, - 2.371*eV, 2.375*eV, 2.380*eV, 2.384*eV, 2.389*eV, 2.394*eV, 2.398*eV, - 2.403*eV, 2.407*eV, 2.412*eV, 2.417*eV, 2.422*eV, 2.426*eV, 2.431*eV, - 2.436*eV, 2.441*eV, 2.445*eV, 2.450*eV, 2.455*eV, 2.460*eV, 2.465*eV, - 2.470*eV, 2.475*eV, 2.480*eV, 2.485*eV, 2.490*eV, 2.495*eV, 2.500*eV, - 2.505*eV, 2.510*eV, 2.515*eV, 2.520*eV, 2.525*eV, 2.530*eV, 2.535*eV, - 2.541*eV, 2.546*eV, 2.551*eV, 2.556*eV, 2.562*eV, 2.567*eV, 2.572*eV, - 2.578*eV, 2.583*eV, 2.588*eV, 2.594*eV, 2.599*eV, 2.605*eV, 2.610*eV, - 2.616*eV, 2.621*eV, 2.627*eV, 2.632*eV, 2.638*eV, 2.644*eV, 2.649*eV, - 2.655*eV, 2.661*eV, 2.666*eV, 2.672*eV, 2.678*eV, 2.684*eV, 2.689*eV, - 2.695*eV, 2.701*eV, 2.707*eV, 2.713*eV, 2.719*eV, 2.725*eV, 2.731*eV, - 2.737*eV, 2.743*eV, 2.749*eV, 2.755*eV, 2.761*eV, 2.768*eV, 2.774*eV, - 2.780*eV, 2.786*eV, 2.792*eV, 2.799*eV, 2.805*eV, 2.811*eV, 2.818*eV, - 2.824*eV, 2.831*eV, 2.837*eV, 2.844*eV, 2.850*eV, 2.857*eV, 2.863*eV, - 2.870*eV, 2.877*eV, 2.883*eV, 2.890*eV, 2.897*eV, 2.904*eV, 2.910*eV, - 2.917*eV, 2.924*eV, 2.931*eV, 2.938*eV, 2.945*eV, 2.952*eV, 2.959*eV, - 2.966*eV, 2.973*eV, 2.980*eV, 2.988*eV, 2.995*eV, 3.002*eV, 3.009*eV, - 3.017*eV, 3.024*eV, 3.031*eV, 3.039*eV, 3.046*eV, 3.054*eV, 3.061*eV, - 3.069*eV, 3.077*eV, 3.084*eV, 3.092*eV, 3.100*eV, 3.107*eV, 3.115*eV, - 3.123*eV, 3.131*eV, 3.139*eV, 3.147*eV, 3.155*eV, 3.163*eV, 3.171*eV, - 3.179*eV, 3.187*eV, 3.195*eV, 3.204*eV, 3.212*eV, 3.220*eV, 3.229*eV, - 3.237*eV, 3.246*eV, 3.254*eV, 3.263*eV, 3.271*eV, 3.280*eV, 3.289*eV, - 3.297*eV, 3.306*eV, 3.315*eV, 3.324*eV, 3.333*eV, 3.342*eV, 3.351*eV, - 3.360*eV, 3.369*eV, 3.378*eV, 3.388*eV, 3.397*eV, 3.406*eV, 3.416*eV, - 3.425*eV, 3.434*eV, 3.444*eV, 3.454*eV, 3.463*eV, 3.473*eV, 3.483*eV, - 3.493*eV, 3.502*eV, 3.512*eV, 3.522*eV, 3.532*eV, 3.542*eV, 3.553*eV, - 3.563*eV, 3.573*eV, 3.583*eV, 3.594*eV, 3.604*eV, 3.615*eV, 3.625*eV, - 3.636*eV, 3.647*eV, 3.657*eV, 3.668*eV, 3.679*eV, 3.690*eV, 3.701*eV, - 3.712*eV, 3.723*eV, 3.734*eV, 3.746*eV, 3.757*eV, 3.769*eV, 3.780*eV, - 3.792*eV, 3.803*eV, 3.815*eV, 3.827*eV, 3.839*eV, 3.850*eV, 3.862*eV, - 3.875*eV, 3.887*eV, 3.899*eV, 3.911*eV, 3.924*eV, 3.936*eV, 3.949*eV, - 3.961*eV, 3.974*eV, 3.987*eV, 3.999*eV, 4.012*eV, 4.025*eV, 4.039*eV, - 4.052*eV, 4.065*eV, 4.078*eV, 4.092*eV, 4.105*eV, 4.119*eV, 4.133*eV, - 4.147*eV, 4.161*eV, 4.175*eV, 4.189*eV, 4.203*eV }; - - // reverted array from skdetsim acrnsg.F because that one is function of wavelength - // instead of photon energy here. - // Real part of refractive index (2002: R. Nambu) - /* G4double RINDEX_skAcrylic[306] = - { 1.50443324, 1.504443389, 1.504455332, 1.504469061, 1.50448457, 1.504501851, - 1.504520896, 1.5045417, 1.504564254, 1.504588551, 1.504614584, 1.504642346, - 1.50467183, 1.504703029, 1.504735934, 1.50477054, 1.504806839, 1.504844823, - 1.504884485, 1.504925819, 1.504968817, 1.505013472, 1.505059776, 1.505107723, - 1.505157305, 1.505208515, 1.505261346, 1.505315791, 1.505371841, 1.505429491, - 1.505488733, 1.50554956, 1.505611965, 1.505675939, 1.505741477, 1.505808571, - 1.505877213, 1.505947398, 1.506019116, 1.506092362, 1.506167127, 1.506243405, - 1.506321189, 1.506400471, 1.506481244, 1.506563501, 1.506647235, 1.506732438, - 1.506819104, 1.506907224, 1.506996793, 1.507087802, 1.507180244, 1.507274113, - 1.507369401, 1.5074661, 1.507564205, 1.507663706, 1.507764598, 1.507866873, - 1.507970524, 1.508075544, 1.508181924, 1.508289659, 1.508398741, 1.508509163, - 1.508620917, 1.508733997, 1.508848395, 1.508964104, 1.509081116, 1.509199426, - 1.509319024, 1.509439905, 1.50956206, 1.509685483, 1.509810167, 1.509936104, - 1.510063288, 1.51019171, 1.510321364, 1.510452242, 1.510584337, 1.510717643, - 1.510852152, 1.510987856, 1.511124749, 1.511262823, 1.511402071, 1.511542485, - 1.51168406, 1.511826787, 1.511970659, 1.512115669, 1.51226181, 1.512409075, - 1.512557456, 1.512706946, 1.512857538, 1.513009225, 1.513162, 1.513315855, - 1.513470783, 1.513626777, 1.51378383, 1.513941935, 1.514101083, 1.514261269, - 1.514422485, 1.514584724, 1.514747978, 1.514912241, 1.515077504, 1.515243762, - 1.515411006, 1.51557923, 1.515748426, 1.515918587, 1.516089706, 1.516261775, - 1.516434788, 1.516608738, 1.516783616, 1.516959417, 1.517136132, 1.517313754, - 1.517492277, 1.517671692, 1.517851994, 1.518033174, 1.518215226, 1.518398141, - 1.518581914, 1.518766537, 1.518952002, 1.519138302, 1.519325431, 1.519513381, - 1.519702144, 1.519891714, 1.520082084, 1.520273245, 1.520465192, 1.520657916, - 1.520851411, 1.521045669, 1.521240683, 1.521436447, 1.521632951, 1.521830191, - 1.522028158, 1.522226844, 1.522426244, 1.522626349, 1.522827153, 1.523028648, - 1.523230828, 1.523433684, 1.523637209, 1.523841398, 1.524046241, 1.524251732, - 1.524457865, 1.524664631, 1.524872023, 1.525080035, 1.525288658, 1.525497887, - 1.525707713, 1.525918129, 1.526129129, 1.526340704, 1.526552849, 1.526765554, - 1.526978815, 1.527192622, 1.52740697, 1.52762185, 1.527837255, 1.52805318, - 1.528269615, 1.528486554, 1.52870399, 1.528921915, 1.529140323, 1.529359205, - 1.529578556, 1.529798368, 1.530018632, 1.530239344, 1.530460494, 1.530682076, - 1.530904083, 1.531126507, 1.531349341, 1.531572579, 1.531796212, 1.532020234, - 1.532244638, 1.532469415, 1.53269456, 1.532920065, 1.533145922, 1.533372124, - 1.533598665, 1.533825537, 1.534052732, 1.534280244, 1.534508066, 1.534736189, - 1.534964608, 1.535193314, 1.535422301, 1.535651561, 1.535881088, 1.536110873, - 1.53634091, 1.536571192, 1.536801711, 1.53703246, 1.537263432, 1.537494619, - 1.537726015, 1.537957613, 1.538189404, 1.538421383, 1.538653541, 1.538885871, - 1.539118367, 1.539351021, 1.539583826, 1.539816774, 1.540049859, 1.540283073, - 1.540516409, 1.54074986, 1.540983419, 1.541217077, 1.54145083, 1.541684668, - 1.541918585, 1.542152573, 1.542386626, 1.542620736, 1.542854896, 1.543089099, - 1.543323337, 1.543557604, 1.543791892, 1.544026194, 1.544260503, 1.544494811, - 1.544729111, 1.544963397, 1.54519766, 1.545431894, 1.545666092, 1.545900246, - 1.546134349, 1.546368394, 1.546602374, 1.546836281, 1.547070108, 1.547303849, - 1.547537495, 1.54777104, 1.548004477, 1.548237797, 1.548470995, 1.548704063, - 1.548936993, 1.549169778, 1.549402412, 1.549634887, 1.549867196, 1.550099331, - 1.550331286, 1.550563052, 1.550794624, 1.551025994, 1.551257154, 1.551488097, - 1.551718817, 1.551949306, 1.552179556, 1.552409561, 1.552639313, 1.552868806, - 1.553098031, 1.553326982, 1.553555652, 1.553784032, 1.554012117, 1.554239899, - 1.554467371, 1.554694525, 1.554921354, 1.555147851, 1.55537401, 1.555599822, - 1.55582528, 1.556050378, 1.556275107, 1.556499462, 1.556723434, 1.556947017 }; - */ - - // TF: from http://refractiveindex.info/?shelf=organic&book=poly%28methyl_methacrylate%29&page=Szczurowski - G4double RINDEX_skAcrylic[306] = - {1.4901, 1.4901, 1.4902, 1.4902, 1.4902, 1.4903, 1.4903, 1.4904, 1.4904, 1.4904, - 1.4905, 1.4905, 1.4906, 1.4906, 1.4907, 1.4907, 1.4908, 1.4908, 1.4908, 1.4909, - 1.4909, 1.4910, 1.4910, 1.4911, 1.4911, 1.4912, 1.4912, 1.4913, 1.4913, 1.4913, - 1.4914, 1.4914, 1.4915, 1.4915, 1.4916, 1.4916, 1.4917, 1.4917, 1.4918, 1.4918, - 1.4919, 1.4919, 1.4920, 1.4920, 1.4921, 1.4921, 1.4922, 1.4922, 1.4923, 1.4923, - 1.4924, 1.4924, 1.4925, 1.4926, 1.4926, 1.4927, 1.4927, 1.4928, 1.4928, 1.4929, - 1.4929, 1.4930, 1.4931, 1.4931, 1.4932, 1.4932, 1.4933, 1.4933, 1.4934, 1.4935, - 1.4935, 1.4936, 1.4936, 1.4937, 1.4938, 1.4938, 1.4939, 1.4940, 1.4940, 1.4941, - 1.4941, 1.4942, 1.4943, 1.4943, 1.4944, 1.4945, 1.4945, 1.4946, 1.4947, 1.4947, - 1.4948, 1.4949, 1.4949, 1.4950, 1.4951, 1.4951, 1.4952, 1.4953, 1.4954, 1.4954, - 1.4955, 1.4956, 1.4956, 1.4957, 1.4958, 1.4959, 1.4959, 1.4960, 1.4961, 1.4962, - 1.4962, 1.4963, 1.4964, 1.4965, 1.4965, 1.4966, 1.4967, 1.4968, 1.4969, 1.4970, - 1.4970, 1.4971, 1.4972, 1.4973, 1.4974, 1.4974, 1.4975, 1.4976, 1.4977, 1.4978, - 1.4979, 1.4980, 1.4980, 1.4981, 1.4982, 1.4983, 1.4984, 1.4985, 1.4986, 1.4987, - 1.4988, 1.4989, 1.4990, 1.4991, 1.4992, 1.4993, 1.4994, 1.4995, 1.4996, 1.4997, - 1.4997, 1.4998, 1.5000, 1.5001, 1.5002, 1.5003, 1.5004, 1.5005, 1.5006, 1.5007, - 1.5008, 1.5009, 1.5010, 1.5011, 1.5012, 1.5013, 1.5015, 1.5016, 1.5017, 1.5018, - 1.5019, 1.5020, 1.5022, 1.5023, 1.5024, 1.5025, 1.5026, 1.5028, 1.5029, 1.5030, - 1.5031, 1.5033, 1.5034, 1.5035, 1.5036, 1.5038, 1.5039, 1.5040, 1.5042, 1.5043, - 1.5044, 1.5046, 1.5047, 1.5049, 1.5050, 1.5051, 1.5053, 1.5054, 1.5056, 1.5057, - 1.5059, 1.5060, 1.5062, 1.5063, 1.5065, 1.5066, 1.5068, 1.5069, 1.5071, 1.5073, - 1.5074, 1.5076, 1.5077, 1.5079, 1.5081, 1.5082, 1.5084, 1.5086, 1.5088, 1.5089, - 1.5091, 1.5093, 1.5095, 1.5096, 1.5098, 1.5100, 1.5102, 1.5104, 1.5106, 1.5108, - 1.5109, 1.5111, 1.5113, 1.5115, 1.5117, 1.5119, 1.5121, 1.5123, 1.5125, 1.5127, - 1.5130, 1.5132, 1.5134, 1.5136, 1.5138, 1.5141, 1.5143, 1.5145, 1.5147, 1.5149, - 1.5152, 1.5154, 1.5157, 1.5159, 1.5161, 1.5164, 1.5166, 1.5169, 1.5171, 1.5174, - 1.5176, 1.5179, 1.5181, 1.5184, 1.5187, 1.5190, 1.5192, 1.5195, 1.5198, 1.5201, - 1.5203, 1.5206, 1.5209, 1.5212, 1.5215, 1.5218, 1.5221, 1.5224, 1.5227, 1.5230, - 1.5234, 1.5237, 1.5240, 1.5243, 1.5247, 1.5250, 1.5254, 1.5257, 1.5260, 1.5264, - 1.5267, 1.5271, 1.5275, 1.5279, 1.5282, 1.5286, 1.5290, 1.5294, 1.5297, 1.5302, - 1.5306, 1.5310, 1.5314, 1.5318, 1.5322, 1.5326}; - - - // Table from acrsg.F is abs coeff, so converted to abs length in m and inverted as - // it was original a function of wavelength and now photon energy in increased energies. - G4double ABSORPTION_skAcrylic[306] = - {3472.22*m, 3355.70*m, 3300.33*m, 3412.97*m, 3496.50*m, 3649.64*m, 3861.00*m, 3921.57*m, 3846.15*m, 3703.70*m, - 3448.28*m, 3048.78*m, 2932.55*m, 2770.08*m, 2808.99*m, 2906.98*m, 2898.55*m, 2906.98*m, 3003.00*m, 3164.56*m, - 3184.71*m, 3246.75*m, 3086.42*m, 2994.01*m, 2994.01*m, 2967.36*m, 3030.30*m, 3039.51*m, 3105.59*m, 2958.58*m, - 3012.05*m, 3039.51*m, 2967.36*m, 2932.55*m, 2923.98*m, 2890.17*m, 2857.14*m, 2865.33*m, 2915.45*m, 2898.55*m, - 2881.84*m, 2754.82*m, 2702.70*m, 2754.82*m, 2890.17*m, 2840.91*m, 2659.57*m, 2604.17*m, 2624.67*m, 2857.14*m, - 2849.00*m, 2849.00*m, 2610.97*m, 2597.40*m, 2688.17*m, 2590.67*m, 2439.02*m, 2380.95*m, 2398.08*m, 2320.19*m, - 2369.67*m, 2293.58*m, 2267.57*m, 2262.44*m, 2320.19*m, 2309.47*m, 2237.14*m, 2247.19*m, 2257.34*m, 2336.45*m, - 2347.42*m, 2421.31*m, 2439.02*m, 2392.34*m, 2392.34*m, 2500.00*m, 2500.00*m, 2512.56*m, 2481.39*m, 2444.99*m, - 2155.17*m, 2053.39*m, 2008.03*m, 2040.82*m, 1988.07*m, 2004.01*m, 1976.28*m, 1834.86*m, 1798.56*m, 1869.16*m, - 1808.32*m, 1754.39*m, 1805.05*m, 1841.62*m, 1872.66*m, 1964.64*m, 1976.28*m, 1893.94*m, 1890.36*m, 1908.40*m, - 1897.53*m, 1855.29*m, 1886.79*m, 1904.76*m, 1841.62*m, 1855.29*m, 1834.86*m, 1805.05*m, 1733.10*m, 1779.36*m, - 1754.39*m, 1788.91*m, 1831.50*m, 1834.86*m, 1869.16*m, 1912.05*m, 2004.01*m, 1968.50*m, 1915.71*m, 1818.18*m, - 1795.33*m, 1766.78*m, 1808.32*m, 1766.78*m, 1754.39*m, 1751.31*m, 1776.20*m, 1838.24*m, 1872.66*m, 1919.39*m, - 1901.14*m, 1988.07*m, 1926.78*m, 1893.94*m, 1901.14*m, 1851.85*m, 1818.18*m, 1769.91*m, 1727.12*m, 1724.14*m, - 1709.40*m, 1689.19*m, 1692.05*m, 1680.67*m, 1697.79*m, 1727.12*m, 1763.67*m, 1697.79*m, 1692.05*m, 1666.67*m, - 1600.00*m, 1636.66*m, 1615.51*m, 1610.31*m, 1538.46*m, 1501.50*m, 1519.76*m, 1492.54*m, 1515.15*m, 1457.73*m, - 1479.29*m, 1461.99*m, 1470.59*m, 1470.59*m, 1453.49*m, 1512.86*m, 1522.07*m, 1589.83*m, 1577.29*m, 1597.44*m, - 1600.00*m, 1672.24*m, 1663.89*m, 1647.45*m, 1661.13*m, 1607.72*m, 1543.21*m, 1488.10*m, 1459.85*m, 1373.63*m, - 1326.26*m, 1373.63*m, 1410.44*m, 1406.47*m, 1396.65*m, 1308.90*m, 1328.02*m, 1386.96*m, 1347.71*m, 1297.02*m, - 1262.63*m, 1234.57*m, 1283.70*m, 1319.26*m, 1310.62*m, 1319.26*m, 1402.52*m, 1459.85*m, 1464.13*m, 1472.75*m, - 1420.45*m, 1432.66*m, 1400.56*m, 1362.40*m, 1358.70*m, 1358.70*m, 1379.31*m, 1360.54*m, 1362.40*m, 1355.01*m, - 1342.28*m, 1367.99*m, 1293.66*m, 1221.00*m, 1197.60*m, 1165.50*m, 1119.82*m, 1107.42*m, 1108.65*m, 1040.58*m, - 1033.06*m, 1000.00*m, 952.38*m, 925.93*m, 909.09*m, 869.57*m, 800.00*m, 781.25*m, 769.23*m, 729.93*m, 735.29*m, - 729.93*m, 709.22*m, 684.93*m, 657.89*m, 645.16*m, 625.00*m, 581.40*m, 568.18*m, 558.66*m, 543.48*m, 540.54*m, 523.56*m, - 505.05*m, 490.20*m, 485.44*m, 462.96*m, 450.45*m, 442.48*m, 429.18*m, 420.17*m, 411.52*m, 411.52*m, 393.70*m, 393.70*m, - 386.10*m, 381.68*m, 355.87*m, 343.64*m, 331.13*m, 343.64*m, 338.98*m, 335.57*m, 326.80*m, 313.48*m, 303.95*m, 294.12*m, - 292.40*m, 280.90*m, 273.97*m, 270.27*m, 265.25*m, 261.78*m, 256.41*m, 250.00*m, 242.13*m, 237.53*m, 233.10*m, 228.83*m, - 226.76*m, 223.21*m, 219.30*m, 215.05*m, 211.86*m, 208.77*m, 206.61*m, 201.21*m, 196.85*m, 193.42*m, 189.75*m, 186.22*m, - 182.15*m, 175.75*m, 168.92*m, 162.60*m, 154.80*m, 144.51*m, 133.51*m, 120.19*m, 105.93*m, 90.91*m, 75.76*m, 61.35*m, - 48.54*m, 37.88*m, 29.41*m}; - - - // Can not use AddConstProperty for RINDEX (see http://hypernews.slac.stanford.edu/HyperNews/geant4/get/opticalphotons/379/1.html) - // ToDo: update with actual wavelength dependence, once known. - G4double RINDEX_SilGel[NUMENTRIES_water] = - {1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, - 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, - 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, - 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, - 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, - 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404, 1.404}; - - // From Lew Classen (Erlangen) with PHOTON_NRG_MIN = .1*eV, and MAX = 7*eV - G4double ENERGY_SilGel[18] = { .1*eV, - 2.034*eV,2.13*eV,2.18*eV,2.22*eV,2.25*eV,2.3*eV, - 2.34*eV, 2.43*eV, 2.53*eV, 2.67*eV, 2.79*eV, - 3.1*eV, 3.28*eV, 3.56*eV, 3.77*eV,4.136*eV, 7.*eV}; - G4double ABSORPTION_SilGel[18] = { 0*m, - .975*m, .975*m, .966*m, .935*m, .890*m, .898*m, - .876*m, .844*m, .786*m, .727*m, .674*m, - .566*m, .485*m, .360*m, .220*m, .220*m, 0*m }; - - // If necessary: - //Reference: A. D. Rakić, A. B. Djurišic, J. M. Elazar, and M. L. Majewski. Optical properties of metallic films for vertical-cavity optoelectronic devices, Appl. Opt. 37, 5271-5283 (1998) - // from http://refractiveindex.info/?shelf=3d&book=metals&page=aluminium - /* - G4double ENERGY_Al[64] = - {1.644*eV, 1.678*eV, 1.713*eV, 1.748*eV, 1.785*eV, 1.822*eV, - 1.860*eV, 1.898*eV, 1.938*eV, 1.978*eV, 2.019*eV, 2.061*eV, - 2.104*eV, 2.148*eV, 2.193*eV, 2.238*eV, 2.285*eV, 2.332*eV, - 2.381*eV, 2.430*eV, 2.481*eV, 2.532*eV, 2.585*eV, 2.639*eV, - 2.693*eV, 2.749*eV, 2.807*eV, 2.865*eV, 2.925*eV, 2.985*eV, - 3.047*eV, 3.111*eV, 3.175*eV, 3.241*eV, 3.309*eV, 3.378*eV, - 3.448*eV, 3.519*eV, 3.593*eV, 3.667*eV, 3.744*eV, 3.821*eV, - 3.901*eV, 3.982*eV, 4.065*eV, 4.149*eV, 4.235*eV, 4.324*eV, - 4.413*eV, 4.505*eV, 4.599*eV, 4.694*eV, 4.792*eV, 4.892*eV, - 4.993*eV, 5.097*eV, 5.203*eV, 5.311*eV, 5.422*eV, 5.534*eV, - 5.649*eV, 5.767*eV, 5.887*eV, 6.009*eV}; - - G4double RINDEX_Al[64] = - {2.299, 2.096, 1.911, 1.752, 1.623, 1.518, 1.432, - 1.361, 1.299, 1.245, 1.196, 1.150, 1.106, 1.063, - 1.022, 0.982, 0.942, 0.903, 0.865, 0.827, 0.790, - 0.754, 0.719, 0.686, 0.653, 0.622, 0.592, 0.564, - 0.536, 0.511, 0.486, 0.463, 0.441, 0.420, 0.401, - 0.382, 0.365, 0.349, 0.333, 0.318, 0.305, 0.292, - 0.279, 0.268, 0.257, 0.246, 0.236, 0.227, 0.218, - 0.210, 0.202, 0.194, 0.187, 0.180, 0.173, 0.166, - 0.160, 0.154, 0.149, 0.143, 0.138, 0.133, 0.128, 0.123}; - */ - // ------------- Surfaces -------------- - - OpWaterBSSurface = - new G4OpticalSurface("WaterBSCellSurface"); - - OpWaterBSSurface->SetType(dielectric_dielectric); - OpWaterBSSurface->SetModel(unified); - OpWaterBSSurface->SetFinish(groundfrontpainted); - OpWaterBSSurface->SetSigmaAlpha(0.1); - - BSSkinSurface = - new G4OpticalSurface("BSSkinSurface"); - - BSSkinSurface->SetType(dielectric_dielectric); - BSSkinSurface->SetModel(unified); - BSSkinSurface->SetFinish(groundfrontpainted); - BSSkinSurface->SetSigmaAlpha(0.1); - - const G4int NUM = 2; - // G4double PP[NUM] = - //{ 2.038E-9*GeV, 4.144E-9*GeV }; - - G4double PP[NUM] = { 1.4E-9*GeV,6.2E-9*GeV}; - G4double RINDEX_blacksheet[NUM] = - { 1.76, 1.76 }; - // { 1.6, 1.6 };//B.Q - - G4double SPECULARLOBECONSTANT[NUM] = - { 0.3, 0.3 }; - G4double SPECULARSPIKECONSTANT[NUM] = - { 0.2, 0.2 }; - G4double BACKSCATTERCONSTANT[NUM] = - { 0.2, 0.2 }; - - OpGlassCathodeSurface = - new G4OpticalSurface("GlassCathodeSurface"); - OpGlassCathodeSurface->SetType(dielectric_dielectric); - OpGlassCathodeSurface->SetModel(unified); - // OpGlassCathodeSurface->SetFinish(groundbackpainted); - OpGlassCathodeSurface->SetFinish(polished); - //OpGlassCathodeSurface->SetSigmaAlpha(0.002); - // was 1.0 - // totally unphysical anyway - G4double RINDEX_cathode[NUM] = - { 1.0, 1.0 }; - /* - G4double SPECULARLOBECONSTANT_glasscath[NUM] = - { 1.0, 1.0 }; - // { 0.3, 0.3 }; - G4double SPECULARSPIKECONSTANT_glasscath[NUM] = - { 0.0, 0.0 }; - // { 0.2, 0.2 }; - G4double BACKSCATTERCONSTANT_glasscath[NUM] = - {0.0, 0.0}; - // { 0.2, 0.2 }; - */ - - - G4double RGCFF = 0.0; - RGCFF = WCSimTuningParams->GetRgcff(); //defaults in mac: 0.32 and flat - - - G4double REFLECTIVITY_glasscath[NUM] = - //{ 0.0+RGCFF, 0.0+RGCFF }; - //{ RGCFF, RGCFF }; - //{ 0.0, 0.0 }; - { 1.0*RGCFF, 1.0*RGCFF }; - - - G4double EFFICIENCY_glasscath[NUM] = - { 0.0, 0.0 }; - - - // jl145 ---- - // - OpWaterTySurface = - new G4OpticalSurface("WaterTyCellSurface"); - - OpWaterTySurface->SetType(dielectric_dielectric); - OpWaterTySurface->SetModel(unified); - OpWaterTySurface->SetFinish(groundbackpainted); //a guess, but seems to work - OpWaterTySurface->SetSigmaAlpha(0.5); //cf. A. Chavarria's ~30deg - - G4double RINDEX_tyvek[NUM] = - { 1.5, 1.5 }; // polyethylene permittivity is ~2.25 - G4double TySPECULARLOBECONSTANT[NUM] = - { 0.75, 0.75 }; // crudely estimated from A. Chavarria's thesis - G4double TySPECULARSPIKECONSTANT[NUM] = - { 0.0, 0.0 }; - G4double TyBACKSCATTERCONSTANT[NUM] = - { 0.0, 0.0 }; - // Lambertian prob is therefore 0.25 - - G4double TyREFLECTIVITY[NUM] = - { 0.94, 0.94 }; //cf. DuPont - // - // ---- - - - G4MaterialPropertiesTable *myMPT1 = new G4MaterialPropertiesTable(); - // M Fechner : new ; wider range for lambda - myMPT1->AddProperty("RINDEX", ENERGY_water, RINDEX1, NUMENTRIES_water); - myMPT1->AddProperty("ABSLENGTH",ENERGY_water, ABSORPTION_water, NUMENTRIES_water); - // M Fechner: new, don't let G4 compute it. - myMPT1->AddProperty("RAYLEIGH",ENERGY_water,RAYLEIGH_water,NUMENTRIES_water); - - // myMPT1->AddProperty("MIEHG",ENERGY_water,MIE_water,NUMENTRIES_water); -// myMPT1->AddConstProperty("MIEHG_FORWARD",MIE_water_const[0]); -// myMPT1->AddConstProperty("MIEHG_BACKWARD",MIE_water_const[1]); -// myMPT1->AddConstProperty("MIEHG_FORWARD_RATIO",MIE_water_const[2]); - - - Water->SetMaterialPropertiesTable(myMPT1); - // myMPT1->DumpTable(); - - G4MaterialPropertiesTable *myMPT2 = new G4MaterialPropertiesTable(); - myMPT2->AddProperty("RINDEX", ENERGY_water, RINDEX_air, NUMENTRIES_water); - // M Fechner : what is that ????? - myMPT2->AddProperty("ABSLENGTH", ENERGY_water, BLACKABS_blacksheet, NUMENTRIES_water); - myMPT2->AddProperty("RAYLEIGH",ENERGY_water, RAYLEIGH_air, NUMENTRIES_water); - - // myMPT2->AddProperty("MIEHG",ENERGY_water,MIE_air,NUMENTRIES_water); -// myMPT2->AddConstProperty("MIEHG_FORWARD",MIE_air_const[0]); -// myMPT2->AddConstProperty("MIEHG_BACKWARD",MIE_air_const[1]); -// myMPT2->AddConstProperty("MIEHG_FORWARD_RATIO",MIE_air_const[2]); - - Air->SetMaterialPropertiesTable(myMPT2); - - G4MaterialPropertiesTable *myMPT3 = new G4MaterialPropertiesTable(); - myMPT3->AddProperty("ABSLENGTH", ENERGY_water, BLACKABS_blacksheet, NUMENTRIES_water); - myMPT3->AddProperty("REFLECTIVITY", ENERGY_water, REFLECTIVITY_blacksheet, NUMENTRIES_water); - myMPT3->AddProperty("EFFICIENCY", ENERGY_water, EFFICIENCY, NUMENTRIES_water); - Plastic->SetMaterialPropertiesTable(myMPT3); - - G4MaterialPropertiesTable *myMPT4 = new G4MaterialPropertiesTable(); - myMPT4->AddProperty("ABSLENGTH", ENERGY_water, BLACKABS_blacksheet, NUMENTRIES_water); - //myMPT4->AddProperty("RINDEX", ENERGY_water, RINDEX_blacksheet, NUMENTRIES_water); //TF - never required if BS is properly defined as surface. - Blacksheet->SetMaterialPropertiesTable(myMPT4); - - G4MaterialPropertiesTable *myMPT5 = new G4MaterialPropertiesTable(); - myMPT5->AddProperty("RINDEX", ENERGY_water, RINDEX_glass, NUMENTRIES_water); - myMPT5->AddProperty("ABSLENGTH",ENERGY_water, ABSORPTION_glass, NUMENTRIES_water); - Glass->SetMaterialPropertiesTable(myMPT5); - - // jl145 ---- - // Abs legnth is same as blacksheet, very small. - G4MaterialPropertiesTable *myMPT6 = new G4MaterialPropertiesTable(); - myMPT6->AddProperty("ABSLENGTH", ENERGY_water, BLACKABS_blacksheet, NUMENTRIES_water); - Tyvek->SetMaterialPropertiesTable(myMPT6); - - /// SilGel : Currently based on WackerSilGel 612, BUT should be adjusted to best one (R&D) - G4MaterialPropertiesTable *SilGelPropTable = new G4MaterialPropertiesTable(); - SilGelPropTable->AddProperty("RINDEX", ENERGY_water, RINDEX_SilGel, NUMENTRIES_water); - SilGelPropTable->AddProperty("ABSLENGTH",ENERGY_SilGel, ABSORPTION_SilGel, 18); //ToDo: get measurement of optical properties of the optical gel. From slides: better than 40cm above 350nm. - SilGelPropTable->AddProperty("RAYLEIGH",ENERGY_water,RAYLEIGH_water,NUMENTRIES_water); //ToDo: get actual Rayleigh scattering in gel - SilGel->SetMaterialPropertiesTable(SilGelPropTable); - - //Acrylic - G4MaterialPropertiesTable *AcrPropTable = new G4MaterialPropertiesTable(); - AcrPropTable->AddProperty("RINDEX", ENERGY_skAcrylic, RINDEX_skAcrylic, 306); - AcrPropTable->AddProperty("ABSLENGTH", ENERGY_skAcrylic, ABSORPTION_skAcrylic, 306); - Acrylic->SetMaterialPropertiesTable(AcrPropTable); - - - - // ------------- Surfaces -------------- - - // Blacksheet - G4MaterialPropertiesTable *myST1 = new G4MaterialPropertiesTable(); - myST1->AddProperty("RINDEX", ENERGY_water, RINDEX_blacksheet, NUMENTRIES_water); - myST1->AddProperty("SPECULARLOBECONSTANT", PP, SPECULARLOBECONSTANT, NUM); - myST1->AddProperty("SPECULARSPIKECONSTANT", PP, SPECULARSPIKECONSTANT, NUM); - myST1->AddProperty("BACKSCATTERCONSTANT", PP, BACKSCATTERCONSTANT, NUM); - myST1->AddProperty("REFLECTIVITY", ENERGY_water, REFLECTIVITY_blacksheet, NUMENTRIES_water); - myST1->AddProperty("EFFICIENCY", ENERGY_water, EFFICIENCY_blacksheet, NUMENTRIES_water); - OpWaterBSSurface->SetMaterialPropertiesTable(myST1); - - BSSkinSurface->SetMaterialPropertiesTable(myST1); - - //Glass to Cathode surface inside PMTs - G4MaterialPropertiesTable *myST2 = new G4MaterialPropertiesTable(); - myST2->AddProperty("RINDEX", PP, RINDEX_cathode, NUM); - // myST2->AddProperty("SPECULARLOBECONSTANT", PP, SPECULARLOBECONSTANT_glasscath, NUM); - // myST2->AddProperty("SPECULARSPIKECONSTANT", PP, SPECULARSPIKECONSTANT_glasscath, NUM); - // myST2->AddProperty("BACKSCATTERCONSTANT", PP, BACKSCATTERCONSTANT_glasscath, NUM); - myST2->AddProperty("REFLECTIVITY", PP, REFLECTIVITY_glasscath, NUM); - myST2->AddProperty("EFFICIENCY", PP, EFFICIENCY_glasscath, NUM); - //myST2->AddProperty("ABSLENGTH", PP, abslength_paint , NUM); - OpGlassCathodeSurface->SetMaterialPropertiesTable(myST2); - - //Tyvek - jl145 - G4MaterialPropertiesTable *myST3 = new G4MaterialPropertiesTable(); - myST3->AddProperty("RINDEX", PP, RINDEX_tyvek, NUM); - myST3->AddProperty("SPECULARLOBECONSTANT", PP, TySPECULARLOBECONSTANT, NUM); - myST3->AddProperty("SPECULARSPIKECONSTANT", PP, TySPECULARSPIKECONSTANT, NUM); - myST3->AddProperty("BACKSCATTERCONSTANT", PP, TyBACKSCATTERCONSTANT, NUM); - myST3->AddProperty("REFLECTIVITY", PP, TyREFLECTIVITY, NUM); - myST3->AddProperty("EFFICIENCY", PP, EFFICIENCY_blacksheet, NUM); - //use same efficiency as blacksheet, which is 0 - OpWaterTySurface->SetMaterialPropertiesTable(myST3); - - - // Surfaces for Al, Ag and future combinations: - ReflectorSkinSurface = - new G4OpticalSurface("ReflectorSurface"); - ReflectorSkinSurface->SetType(dielectric_metal); - ReflectorSkinSurface->SetModel(unified); - ReflectorSkinSurface->SetFinish(polished); - - G4MaterialPropertiesTable *AlPropTable = new G4MaterialPropertiesTable(); - //AlPropTable->AddProperty("RINDEX", ENERGY_Al, RINDEX_Al, 64); //not necessary for boundaries if kept within their MotherVolumes. - AlPropTable->AddProperty("REFLECTIVITY", ENERGY_water, REFLECTIVITY_aluminium, NUMENTRIES_water); - ReflectorSkinSurface->SetMaterialPropertiesTable(AlPropTable); - - OpGelFoamSurface = - new G4OpticalSurface("GelFoamSurface"); - - OpGelFoamSurface->SetType(dielectric_dielectric); - OpGelFoamSurface->SetModel(unified); - OpGelFoamSurface->SetFinish(groundfrontpainted); - OpGelFoamSurface->SetSigmaAlpha(0.1); - - OpGelFoamSurface->SetMaterialPropertiesTable(myST1); //TF: same as water-blacksheet for now - - - - //ToDo: - G4MaterialPropertiesTable *AgPropTable = new G4MaterialPropertiesTable(); - G4MaterialPropertiesTable *AlAg1PropTable = new G4MaterialPropertiesTable(); -} - -void WCSimDetectorConstruction::AddDopedWater(G4double percentGd){ - G4Material * Water = G4Material::GetMaterial("Water"); - G4Material * DopedWater = G4Material::GetMaterial("Doped Water", false); - //Delete old doped water if it exists - if(DopedWater) delete DopedWater; - //Create new doped water material - DopedWater = new G4Material("Doped Water",1.00*g/cm3,2); - //Gd doped water has the same optical properties as pure water - DopedWater->SetMaterialPropertiesTable(Water->GetMaterialPropertiesTable()); - //Set concentration - G4double a = 157.25*g/mole; - G4Element* Gd = new G4Element("Gadolinium","Gd", 64,a); - DopedWater->AddMaterial(Water,(100.-percentGd)*perCent); - DopedWater->AddElement(Gd,percentGd*perCent); -} diff --git a/libWCSIM/src/WCSimConstructMultiPMT.cc b/libWCSIM/src/WCSimConstructMultiPMT.cc deleted file mode 100644 index bdcd397..0000000 --- a/libWCSIM/src/WCSimConstructMultiPMT.cc +++ /dev/null @@ -1,885 +0,0 @@ -#include "WCSimDetectorConstruction.hh" - -#include "G4Box.hh" -#include "G4Sphere.hh" -#include "G4SubtractionSolid.hh" -#include "G4UnionSolid.hh" -#include "G4LogicalVolume.hh" -#include "G4VisAttributes.hh" -#include "G4Material.hh" -#include "G4Polycone.hh" -#include "G4Tubs.hh" -#include "G4PVPlacement.hh" -#include "G4LogicalSkinSurface.hh" - -#include "G4SDManager.hh" -#include "WCSimWCSD.hh" - -#include "G4SystemOfUnits.hh" - -#include "WCSimMultiPMTParameterisation.hh" -#include "G4PVParameterised.hh" - - -//mPMT logical volume construction. -//A function of the WCSimDetectorConstruction class - - -// Keep track of created logical volumes (PMTs and mPMTs). -// If already created, call from map! -WCSimDetectorConstruction::PMTMap_t WCSimDetectorConstruction::PMTLogicalVolumes; - -// Options: - PMT type (let's do ONE type per multiPMT) -// - expose: how much the individual PMTs stick out of the blacksheet. -// Let's keep this approach for now and make the surface of our multiPMT a black sheet. - -// For MultiPMT: expose have different meaning, namely where to locate the sphere on the blacksheet ID/OD separator -G4LogicalVolume* WCSimDetectorConstruction::ConstructMultiPMT(G4String PMTName, G4String CollectionName, G4String detectorElement, G4int nIDPMTs) -{ - - - G4bool showme = true; - - std::cout<<"TEMP: Collection Name = "<second; - } - - G4cout << "Create mPMT" << G4endl; - G4cout << "Vessel height: " << mPMT_vessel_cyl_height << G4endl; - G4cout << "Vessel radius: " << mPMT_vessel_radius << G4endl; - G4cout << "Vessel radius curv: " << mPMT_vessel_radius_curv << G4endl; - G4cout << "Number of ID PMTs: " << nIDPMTs << G4endl; - - //All components of the PMT are now contained in a single logical volume logicWCPMT. - //Origin is on the blacksheet, faces positive z-direction. - - WCSimPMTObject *PMT = GetPMTPointer(CollectionName); - G4double expose = PMT->GetExposeHeight(); // THIS is the PMT expose, which I'm currently using for pressure mPMT_vessel construction - - // expose = 0.0153*m; //TF: to compare same size mPMT_vessels for different size PMTs - // this is NOT the "expose" of the mPMT - - //////////////////////////////////////////////////// - /// 1) Outer logical volume: fill with water /// - /// (cfr. same starting point as ConstructPMT) /// - //////////////////////////////////////////////////// - - // origin on blacksheet wall is origin of spherical top for ID, - // change if we want to use the cylindrical part in ID. - - // NEW: possibility of spherical Caps == sphere minus cylinder - G4double mPMT_vessel_cap_height = mPMT_vessel_radius_curv - sqrt(mPMT_vessel_radius_curv * mPMT_vessel_radius_curv - mPMT_vessel_radius*mPMT_vessel_radius); - - - G4double mPMT_zRange_outer[2] = {0, // must be zero or it will collide with the blacksheet and mess up the G4Navigator - mPMT_vessel_cyl_height + mPMT_vessel_cap_height}; - G4double mPMT_RRange_outer[2] = {mPMT_vessel_radius, - mPMT_vessel_radius}; - G4double mPMT_rRange_outer[2] = {0., 0.}; - - // Although G4Tubs is more natural, Polycone is used to be in control of z position - // and because z = 0 is position of bottom of cylinder, vs center in G4Tubs - G4Polycone* solidMultiPMT = - new G4Polycone("WCmPMT", - 0.0*deg, - 360.0*deg, - 2, - mPMT_zRange_outer, - mPMT_rRange_outer, // R Inner - mPMT_RRange_outer);// R Outer - - G4LogicalVolume *logicWCMultiPMT = - new G4LogicalVolume( solidMultiPMT, - G4Material::GetMaterial("Water"), - "WCMultiPMT", - 0,0,0); - - - - //////////////////////////////////////////////// - /// 2)Acrylic/Glass shell of the mPMT vessel /// - //////////////////////////////////////////////// - G4Sphere* mPMT_top_sphere_vessel = - new G4Sphere( "WCmPMT_tsphere_vessel", - 0, - mPMT_vessel_radius_curv, - 0.0*deg,360.0*deg, - 0.0*deg,90.0*deg); - - - G4SubtractionSolid * mPMT_top_cap_vessel; - //G4LogicalVolume * logic_mPMT_top_sphere_vessel; - G4double cap_position_offset = 0.*cm; - cap_position_offset = -1.0* (mPMT_vessel_radius_curv - mPMT_vessel_cap_height); //lower position of cap by the piece that we cut off. - G4LogicalVolume *logic_mPMT_vessel; - - // CAP - if(mPMT_vessel_radius_curv - mPMT_vessel_radius > .1*mm){ - G4Box* solidCutOffTubs = - new G4Box( "cutOffTubs", - mPMT_vessel_radius_curv + 1.*cm, // x/2 - mPMT_vessel_radius_curv + 1.*cm, // y/2 - mPMT_vessel_radius_curv - mPMT_vessel_cap_height); // z/2 - - mPMT_top_cap_vessel = - new G4SubtractionSolid( "InteriorWCPMT", - mPMT_top_sphere_vessel, - solidCutOffTubs); - - - //std::cout << "DEBUG MODE " << vessel_radius_curv - vessel_cap_height << " " < .1*mm){ - G4Box* solidCutOffContainer = - new G4Box( "cutOffContainer", - outerR_curv_container + 1.*cm, - outerR_curv_container + 1.*cm, - mPMT_vessel_radius_curv - mPMT_vessel_cap_height); - - mPMT_top_cap_container = - new G4SubtractionSolid( "InteriorWCPMT", - mPMT_top_sphere_container, - solidCutOffContainer); - - - // CAP wo CYLINDER inner - if( mPMT_vessel_cyl_height < 0.01*mm){ - - logic_mPMT_container = - new G4LogicalVolume( mPMT_top_cap_container, - G4Material::GetMaterial(mPMT_inner_material),//"SilGel"), //whole area between pressure vessel and support structure is optical gel! - "WCMultiPMT_container", - 0,0,0); - - } else{ - // CAP + CYLINDER inner - G4Tubs* mPMT_cylinder_container = - new G4Tubs("WCmPMT_cyl_container", - innerR_container, - outerR_container, - mPMT_vessel_cyl_height/2., - 0.0*deg, - 360.0*deg); - - - G4UnionSolid *union_container = - new G4UnionSolid("ContainerUnion",mPMT_cylinder_container,mPMT_top_cap_container,0,G4ThreeVector(0,0,cap_position_offset+mPMT_vessel_cyl_height/2)); - - G4cout << mPMT_cylinder_container << ", " << mPMT_top_cap_container << ", " << cap_position_offset+mPMT_vessel_cyl_height/2 << G4endl; - G4cout << innerR_curv_container << ", " << outerR_curv_container << ", " << innerR_container << ", " << outerR_container << G4endl; - logic_mPMT_container = - new G4LogicalVolume( union_container, - G4Material::GetMaterial(mPMT_inner_material),//"SilGel"), //whole area between pressure vessel and support structure is optical gel! - "WCMultiPMT_container", - 0,0,0); - - } - } else{ - //SPHERE wo CYLINDER inner - if( mPMT_vessel_cyl_height < 0.01*mm){ - logic_mPMT_container = - new G4LogicalVolume( mPMT_top_sphere_container, - G4Material::GetMaterial(mPMT_inner_material),//"SilGel"), //whole area between pressure vessel and support structure is optical gel! - "WCMultiPMT_container", - 0,0,0); - - - }else{ - // SPHERE w CYLINDER inner - G4Tubs* mPMT_cylinder_container = - new G4Tubs("WCmPMT_cyl_container", - innerR_container, - outerR_container, - mPMT_vessel_cyl_height/2., - 0.0*deg, - 360.0*deg); - - G4UnionSolid *union_container = - new G4UnionSolid("ContainerUnion",mPMT_cylinder_container,mPMT_top_sphere_container,0,G4ThreeVector(0,0,mPMT_vessel_cyl_height/2)); - - logic_mPMT_container = - new G4LogicalVolume( union_container, - G4Material::GetMaterial(mPMT_inner_material),//"SilGel"), //whole area between pressure vessel and support structure is optical gel! - "WCMultiPMT_container", - 0,0,0); - - } - } - - - ///////////////////////////////////////////////////// - /// 4) Fill the mPMT mpmt_vessel with single (ID) PMTs /// - ///////////////////////////////////////////////////// - - /* - * ToDo : Also have an alternate filling option according to some array. BUT do that - * in the parametrization somehow (eg. looping over an array or so) - */ - - // individual PMTs z=0 is offset by position_z_offset (see ConstructPMT) for n>1 - G4double pmtDistance = innerR_curv_container; // Inner radius od the DOM - pmtDistance -= (mPMT_vessel_radius_curv - mPMT_outer_material_d - 54.*mm); - G4cout << "Distance from the Z axis and corrected for z0 of individual PMTs = " << pmtDistance << " mm" << G4endl; - - G4LogicalVolume* logicWCPMT = ConstructPMT(PMTName, CollectionName,detectorElement,nIDPMTs); - - G4int NbOfTotPmt = 0; - if(nIDPMTs == 1){ - - ///////////////////////////////////////// - /// For single PMT WITH acrylic cover /// - ///////////////////////////////////////// - - double single_pmt_z_offset = 0.; - // raise/lower PMT to be inside the container - if(mPMT_vessel_cyl_height < 0.01*mm) - single_pmt_z_offset = -cap_position_offset; - else - single_pmt_z_offset = -mPMT_vessel_cyl_height/2; - // singlePMT: OK now! - new G4PVPlacement(0, - G4ThreeVector(0, 0, single_pmt_z_offset), - logicWCPMT, // its logical volume - "pmt", - logic_mPMT_container, // Mother logical volume, - false, - 0, - checkOverlaps); - - - //Place Container - new G4PVPlacement(0, // its rotation - G4ThreeVector(0,0,0), // its position - logic_mPMT_container, // its logical volume - "WCPMT_container", // its name - logic_mPMT_vessel, // its mother volume - false, // no boolean os - 0, - checkOverlaps); - - - /* Set all visualization here for better overview. */ - // Gray wireframe visual style - G4VisAttributes* VisAttGrey = new G4VisAttributes(G4Colour(0.2,0.2,0.2)); - //VisAttGrey->SetForceWireframe(true); - VisAttGrey->SetForceSolid(true); - G4VisAttributes* WCPMTVisAtt5 = new G4VisAttributes(G4Colour(.0,1.,1.)); - WCPMTVisAtt5->SetForceSolid(true); - G4VisAttributes* VisAttYellow = new G4VisAttributes(G4Colour(1.0,1.,0.)); - VisAttYellow->SetForceSolid(true); - - if(showme){ - logicWCMultiPMT->SetVisAttributes(VisAttGrey); - - logic_mPMT_vessel->SetVisAttributes(WCPMTVisAtt5); - logic_mPMT_container->SetVisAttributes(VisAttYellow); - } - - // Keep track of already created mPMT logical volumes in same map as for PMTs - PMTLogicalVolumes[key] = logicWCMultiPMT; - - - return logicWCMultiPMT; - } - else{ - NbOfTotPmt = FillCircles(); - if(NbOfTotPmt != nIDPMTs){ - G4cerr << "ERROR: Requested " << nIDPMTs << " PMTs but only acquired " << NbOfTotPmt << - " PMT from config file " << G4endl; - return NULL; - } - // OLDER version: - // CountPMT() needs number of PMTs for circle 1 - // This is the only free param of the OLDER fill-PMT alg - //NbOfTotPmt = CountPMT(nID_PMTs); // Total number of PMTs in the circles, defines vNiC and vAlpha - - } - - //Fill top sphere (or let the parametrisation figure out the whole filling? Yes!) - // raise/lower PMT to be inside the container - double pmt_z_offset = 0.; - if(mPMT_vessel_cyl_height >= 0.01*mm){ - pmt_z_offset = cap_position_offset+mPMT_vessel_cyl_height/2; // and same for inner structure - } - G4VPVParameterisation* pmtParam_id = - new WCSimMultiPMTParameterisation( - nIDPMTs, // NoPMTs - pmtDistance, // Radius of PMT translated positions on sphere - vNiC, // # PMTs in each circle - vAlpha, // Their tilt angle - vCircle, // Circle number - vAzimOffset, // The offset in azimuth angle of first PMT in that circle - pmt_z_offset); // Position offsets of circles, should become array! - - - // dummy value : kZAxis -- modified by parameterised volume - - // Mother Volume can only have ONE parametrization, - // hence need to cut in pieces OR parametrize all at once. - // TF: cut in pieces is better for different OD/ID types. But different parametrization objects needed? - - - G4VPhysicalVolume* hemisphere = - new G4PVParameterised("pmt", // their name - logicWCPMT, // their logical volume - logic_mPMT_container, // Mother logical volume - kZAxis, // Are placed along this axis - NbOfTotPmt, // Number of PMTs (replica's) - pmtParam_id, // The parametrisation - true); // checking overlaps - - - // Need a 4mm tolerance : still not perfect though because overlaps with mother volume happen still.. - if(hemisphere->CheckOverlaps(1000,4,1)){ - G4cerr << "Hemisphere has overlapping PMTs: Retry" << G4endl; - //return NULL; - } - - /* TESTING single case - G4RotationMatrix* rotm = new G4RotationMatrix(); - rotm->rotateY(-36.*CLHEP::degree); - new G4PVPlacement( rotm, // its rotation - G4ThreeVector(0,0,pmt_z_offset), // its position - logicWCPMT, // its logical volume - "test1", // its name - logic_mPMT_container, // its mother volume - false, // no boolean os - 0, - checkOverlaps); - */ - - - - - // Place Container: - /* - new G4PVPlacement( 0, // its rotation - G4ThreeVector(0,0,0), // its position - logic_mPMT_top_sphere_container, // its logical volume - "WCPMT_top_container", // its name - logic_mPMT_top_sphere_vessel, // its mother volume - false, // no boolean os - 0, - checkOverlaps); - */ - new G4PVPlacement( 0, // its rotation - G4ThreeVector(0,0,0), // its position - logic_mPMT_container, // its logical volume - "WCPMT_container", // its name - logic_mPMT_vessel, // its mother volume - false, // no boolean os - 0, - checkOverlaps); - - - - - - /////////////////////////////////////////////////////////////// - /// 5) Cover the surface with black material/paint/...sheet /// - /////////////////////////////////////////////////////////////// - - // Note: no union solid needed here as no object lives in this volume - - if(!addPMTBase){ - ///// - //Defines a cylinder - // origin is in the center of upward cylinder - if(mPMT_vessel_cyl_height > 0){ - G4Tubs* inner_cylinder = - new G4Tubs("inn_cyl", - 0., - innerR_container, - mPMT_vessel_cyl_height/2, - 0.0*deg, - 360.0*deg); - - G4LogicalVolume *logic_mPMT_cylinder_inner = - new G4LogicalVolume( inner_cylinder, - G4Material::GetMaterial("Blacksheet"), - "WCMultiPMT_inner_cyl", - 0,0,0); - - - //Place Inner Cylinder: - new G4PVPlacement( 0, // its rotation - G4ThreeVector(0,0,0), // its position - logic_mPMT_cylinder_inner, // its logical volume - "WCPMT_inner_cyl", // its name - logic_mPMT_vessel, // its mother volume - false, // no boolean os - 0, - checkOverlaps); - - new G4LogicalSkinSurface("FoamSurfaceCyl",logic_mPMT_cylinder_inner, OpGelFoamSurface); - G4VisAttributes* VisAttRed = new G4VisAttributes(G4Colour(1.0,0.,0.)); - VisAttRed->SetForceSolid(true); - logic_mPMT_cylinder_inner->SetVisAttributes(VisAttRed); - - } - G4Sphere* inner_top_sphere = - new G4Sphere( "WCmPMT_tsphere", - 0,innerR_curv_container, - 0.0*deg,360.0*deg, - 0.0*deg,90.0*deg); - - - - G4SubtractionSolid * inner_top_cap; - G4LogicalVolume * logic_mPMT_top_sphere_inner; - if(mPMT_vessel_radius_curv - mPMT_vessel_radius > .1*mm){ - - G4Box* solidCutOffInner = - new G4Box( "cutOffInner", - innerR_curv_container + 1.*cm, // x/2 - innerR_curv_container + 1.*cm, // y/2 - mPMT_vessel_radius_curv - mPMT_vessel_cap_height); // z/2 - - inner_top_cap = - new G4SubtractionSolid( "InteriorWCPMT", - inner_top_sphere, - solidCutOffInner); - - - logic_mPMT_top_sphere_inner = - new G4LogicalVolume( inner_top_cap, - G4Material::GetMaterial("Blacksheet"), - "WCMultiPMT_inner_top", - 0,0,0); - - - } else{ - logic_mPMT_top_sphere_inner = - new G4LogicalVolume( inner_top_sphere, - G4Material::GetMaterial("Blacksheet"), - "WCMultiPMT_inner_top", - 0,0,0); - } - - - //Place Inner Structure - new G4PVPlacement( 0, // its rotation - G4ThreeVector(0,0,pmt_z_offset), // its position - logic_mPMT_top_sphere_inner, // its logical volume - "WCPMT_inner_top", // its name - logic_mPMT_vessel, // its mother volume - false, // no boolean os - 0, - checkOverlaps); - - new G4LogicalSkinSurface("FoamSurfaceTop",logic_mPMT_top_sphere_inner, OpGelFoamSurface); - - G4VisAttributes* VisAttRed = new G4VisAttributes(G4Colour(1.0,0.,0.)); - VisAttRed->SetForceSolid(true); - logic_mPMT_top_sphere_inner->SetVisAttributes(VisAttRed); - } - - /* Set all visualization here for better overview. */ - // Gray wireframe visual style - G4VisAttributes* VisAttGrey = new G4VisAttributes(G4Colour(0.2,0.2,0.2)); - //VisAttGrey->SetForceWireframe(true); - VisAttGrey->SetForceSolid(true); - G4VisAttributes* WCPMTVisAtt5 = new G4VisAttributes(G4Colour(.0,1.,1.)); - WCPMTVisAtt5->SetForceSolid(true); - G4VisAttributes* VisAttYellow = new G4VisAttributes(G4Colour(1.0,1.,0.)); - VisAttYellow->SetForceSolid(true); - - - if(showme){ - logicWCMultiPMT->SetVisAttributes(VisAttGrey); - logic_mPMT_vessel->SetVisAttributes(WCPMTVisAtt5); - logic_mPMT_container->SetVisAttributes(VisAttYellow); - - } - - // Keep track of already created mPMT logical volumes in same map as for PMTs - PMTLogicalVolumes[key] = logicWCMultiPMT; - - return logicWCMultiPMT; -} - -/* - * Different approach (more practical) than the one below using CountPMT() - * We know total number of PMTs and have a minimum alpha. - * Using findClosestPacking.py to find combination with maximum viewing angle. - * This results in the number of circles and the number of PMTs per circle. - * Need to fill the vectors - */ -G4int WCSimDetectorConstruction::FillCircles(void){ - - //Read in text file (instead of setting 8 numbers) with configuration - // and pre-calculated values - //Fill vAlpha, vNiC, vCircle, vAzimOffset - std::vector vEta; - vEta.clear(); - - std::ifstream config(config_file); - if(!config){ - G4cerr << "Can not find mPMT configfile: " << config_file << G4endl; - return 0; - } - std::string line, item; - getline(config,line); - std::istringstream value(line); - while(getline(value, item, ' ')){ - if(atoi(item.c_str()) > 0) - vNiC.push_back(atoi(item.c_str())); - } - getline(config,line); - std::istringstream value2(line); - while(getline(value2, item, ' ')){ - vEta.push_back(atof(item.c_str())*CLHEP::deg); - } - G4double min_eta = *std::min_element(vEta.begin(),vEta.end()); - G4cout << "Min viewing angle eta from input file for mPMT filling is : " << min_eta << G4endl; - - getline(config,line); - std::istringstream value3(line); - while(getline(value3, item, ' ')){ - vAlpha.push_back(atof(item.c_str())*CLHEP::deg); - } - - //for(int i = 0; i < vNiC.size(); i++) { - // std::cout << "test " << vNiC[i] << " " << vAlpha[i]/CLHEP::deg << " " << vEta[i]/CLHEP::deg << std::endl; - //} - - /// NEW: optionally add an azimuth offset line for the first PMT of each circle (by default 0) - // necessary for hexagonal orientation - getline(config,line); - if(!line.empty()){ - std::istringstream value4(line); - while(getline(value4, item, ' ')){ - vAzimOffset.push_back(atof(item.c_str())*CLHEP::deg); - } - } - else { - for(size_t i = 0; i < vNiC.size();i++){ - vAzimOffset.push_back(0.*CLHEP::deg); - } - assert(vAzimOffset.size() == vAlpha.size()); - } - - // Total number of pmt in the circles - G4int TotPmt = 0; - for(size_t i = 0; i < vNiC.size();i++){ - TotPmt+=vNiC[i]; - for(int j = 0; j < vNiC[i]; j++){ - vCircle.push_back(i); //number circles internally between 0 and N-1 - } - } - - //std::cout << "Test: vNic: " << vNiC.size() << " vAlpha: " << vAlpha.size() << " vCircle: " << vCircle.size() << std::endl; - - return TotPmt; -} - - - - - -/* - * Method based on the number of PMTs in the first circle - * Then the rest will be filled according to closest packing - * Result are filled vNiC, vAlpha and vCircle vectors. - * Return total number of PMTs - * - * Downside : - uses same fEta (from first circle) for all. - * - assumes the viewing angle eta from the first circle - * is the minimum one, which is not always true. - * - needs an empirical number of PMTs in circle 1 to start with - * and cannot properly deal with a minimum opening angle. - * - calculate configuration for each WCSim MC (slow). - * => DEPRECATE at some point. - */ -G4int WCSimDetectorConstruction::CountPMT(G4int NoPmt) -{ - - G4double fEta = ComputeEta(NoPmt); // PMT viewing angle, viewing_angle (CONVERT TO DEG) - - G4int NoCircle = 1; - G4double alphaNext = fEta; - G4double alphaPrev = alphaNext; - G4double NiCNext = NoPmt; - - // Only fill the ones where alpha is above the minimum (to look over neighbouring mPMTs) - // Theta_min = eta + atan(R/r) with R radius of sphere, and r the distance between mPMTs - G4double theta_min = 13.*CLHEP::pi/180; //atan((vessel_radius_curv)/id_spacing) - if(alphaNext > theta_min + fEta){ - vAlpha.push_back(alphaNext); - vNiC.push_back(NoPmt); - G4cout << "Circle n. " << NoCircle << " - Number of PMTs: " << vNiC[NoCircle-1] << " - alpha: " << vAlpha[NoCircle-1]*180/3.141592 << " - eta " << fEta*180/3.141592 << G4endl; - NoCircle++; - } - - // Recursive calculation of vector contents - do - { - alphaNext = ComputeAlpha(alphaPrev, fEta); - NiCNext = ComputeNiC(alphaNext, fEta); - if(NiCNext<2){break;} // Prevents negative values of NiC - if(alphaNext > theta_min +fEta){ - vAlpha.push_back(alphaNext); - vNiC.push_back(NiCNext); - G4cout << "Circle n. " << NoCircle << " - Number of PMTs: " << vNiC[NoCircle-1] << " - alpha: " << vAlpha[NoCircle-1]*180/3.141592 << " - eta " << fEta*180/3.141592 << G4endl; - std::cout << "Circle n. " << NoCircle << " - Number of PMTs: " << vNiC[NoCircle-1] << " - alpha: " << vAlpha[NoCircle-1]*180/3.141592 << " - eta " << fEta*180/3.141592 << std::endl; - NoCircle++; - } - alphaPrev = alphaNext; - } while(NiCNext > 2); // No less than 3 detectors in each circle - - // TF: If possible, add top: - if(vAlpha[NoCircle-1]+2*fEta < 90.){ - alphaNext = CLHEP::pi/2; - NiCNext = 1; - vAlpha.push_back(alphaNext); - vNiC.push_back(NiCNext); - G4cout << "Circle n. " << NoCircle << " - Number of PMTs: " << vNiC[NoCircle-1] << " - alpha: " << vAlpha[NoCircle-1]*180/3.141592 << " - eta " << fEta*180/3.141592 << G4endl; - std::cout << "Circle n. " << NoCircle << " - Number of PMTs: " << vNiC[NoCircle-1] << " - alpha: " << vAlpha[NoCircle-1]*180/3.141592 << " - eta " << fEta*180/3.141592 << std::endl; - NoCircle++; - } - - - - // Total number of pmt in the circles - G4int TotPmt = 0; - for(size_t i = 0; i < vNiC.size();i++){ - TotPmt+=vNiC[i]; - for(int j = 0; j < vNiC[i]; j++){ - vCircle.push_back(i); //number circles internally between 0 and N-1 - } - } - - std::cout << "Total number of pmt: " << TotPmt << std::endl; - std::cout << "Percentage of covered hemispherical surface = " << TotPmt*(1-std::cos(fEta))*100 << "%" << std::endl; - std::cout << "Percentage of covered hemispherical surface above theta_min = " << TotPmt*(1-std::cos(fEta))*100/(1-std::cos(CLHEP::pi/2-theta_min)) << "%" << std::endl; - std::cout << "Test: vNic: " << vNiC.size() << " vAlpha: " << vAlpha.size() << " vCircle: " << vCircle.size() << std::endl; - - return TotPmt; -} - - - - -//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... - -//NoPmt is number of PMT is first circle -G4double WCSimDetectorConstruction::ComputeEta(G4int NoPmt) -{ - // From eq. 11 in Meas. Sci. Technol. 10 (1999) 1015–1019, where a_{c-1} = n_{c-1} = 0. - G4double cospi = std::cos(CLHEP::pi/NoPmt); - G4double etaLocal = std::atan(std::sqrt(1 - cospi*cospi)); - return etaLocal; -} - -//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... - -G4double WCSimDetectorConstruction::ComputeAlpha (G4double alphaOfPrevC, G4double Eta) -{ - G4double alphaLocal = alphaOfPrevC + 2*Eta; - return alphaLocal; -} - -//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo...... - -G4int WCSimDetectorConstruction::ComputeNiC (G4double alphaOfCircle, G4double Eta) -{ - G4double cos2eta = std::cos(Eta) * std::cos(Eta); - G4double sin2alpha = std::sin(alphaOfCircle) * std::sin(alphaOfCircle); - G4double cosalpha = std::cos(alphaOfCircle); - G4double arccos = std::acos(std::sqrt(cos2eta-sin2alpha)/cosalpha); //eq. 10 - G4int NiCLocal = std::floor(CLHEP::pi/arccos); //eq. 7 - return NiCLocal; -} diff --git a/libWCSIM/src/WCSimConstructPMT.cc b/libWCSIM/src/WCSimConstructPMT.cc deleted file mode 100755 index 07b2fb6..0000000 --- a/libWCSIM/src/WCSimConstructPMT.cc +++ /dev/null @@ -1,493 +0,0 @@ -#include "WCSimDetectorConstruction.hh" - -#include "G4Box.hh" -#include "G4Sphere.hh" -#include "G4SubtractionSolid.hh" -#include "G4LogicalVolume.hh" -#include "G4VisAttributes.hh" -#include "G4Material.hh" -#include "G4Polycone.hh" -#include "G4Cons.hh" -#include "G4PVPlacement.hh" -#include "G4LogicalBorderSurface.hh" - -#include "G4SDManager.hh" -#include "WCSimWCSD.hh" -#include "WCSimPMTObject.hh" - -#include "G4SystemOfUnits.hh" - -#include "G4LogicalSkinSurface.hh" - -//PMT logical volume construction. -//WCSimDetectorConstruction::PMTMap_t WCSimDetectorConstruction::PMTLogicalVolumes; - -//ToDo: use enum instead of string for PMTtype, and PMTs should be objects of one simple class. -G4LogicalVolume* WCSimDetectorConstruction::ConstructPMT(G4String PMTName, G4String CollectionName, G4String detectorElement, G4int nIDPMTs)//Modified by B.Q, 2018/12 to incorporate number of ID PMTs in the function input -{ - PMTKey_t key(PMTName,CollectionName); - - // Return pre-created PMT Logical Volume if it already exists. - PMTMap_t::iterator it = PMTLogicalVolumes.find(key); - if (it != PMTLogicalVolumes.end()) { - //G4cout << "Restore PMT" << G4endl; - return it->second; - } - - //G4cout << "Create PMT" << G4endl; - - - if (Vis_Choice == "RayTracer"){ - // Blue wireframe visual style - // Used in the RayTracer visualizer - G4VisAttributes* WCPMTVisAtt = new G4VisAttributes(G4Colour(0.0,0.0,1.0)); - WCPMTVisAtt->SetForceSolid(true); // force the object to be visualized with a surface - WCPMTVisAtt->SetForceAuxEdgeVisible(true); // force auxiliary edges to be shown - } - - - G4double expose = 0.; - G4double radius = 0.; - G4double glassThickness = 0.; - - WCSimPMTObject *PMT = GetPMTPointer(CollectionName); - expose = PMT->GetExposeHeight(); - radius = PMT->GetRadius(); //r at height = expose - glassThickness = PMT->GetPMTGlassThickness(); - - //sphereRadius R: radius of curvature, based on spherical approx near exposeHeight - //if radius of spherical cap at exposeHeight = PMTradius r (eg. 20"/2) - //then from r = sqrt(R*R - (R-expose)*(R-expose)), we get: - G4double sphereRadius = (expose*expose + radius*radius)/(2*expose); - G4double PMTOffset = sphereRadius - expose; - - //Optional reflectorCone: - G4double reflectorRadius = radius + id_reflector_height * tan(id_reflector_angle); - G4double reflectorThickness = 0.5*CLHEP::mm; - if((reflectorRadius - radius) < 1.*CLHEP::mm) - reflectorThickness = 0.*CLHEP::mm; - - - G4cout << "========================================================" << G4endl; - G4cout << "Expose height for PMT " << CollectionName << " : " << expose << " mm" << G4endl; - G4cout << "Radius at expose height : " << radius << " mm" << G4endl; - G4cout << "Radius of curvature : " << sphereRadius << " mm" << G4endl; - G4cout << "Reflector radius : " << reflectorRadius << " mm" << G4endl; - G4cout << "Reflector thickness : " << reflectorThickness << " mm" << G4endl; - G4cout << "========================================================" << G4endl; - - - G4VSolid *solidWCPMT; - G4double position_z_offset = 0.; // for positioning PMTs - G4double wcpmt_z_offset = 0.; // for positioning single PMT support (mPMT) - G4bool addPMTBase = false; - G4double pmtModuleHeight = 54.*CLHEP::mm; //includes puck and single PMT support, not PMT base. - - G4cout << "Number of PMTs per Vessel, in ConstructPMT = " << nIDPMTs << G4endl; - - if(nIDPMTs == 1){ - - //All components of the PMT are now contained in a single logical volume logicWCPMT. - //Origin is on the blacksheet, faces positive z-direction. - - G4double PMTHolderZ[2] = {0, std::max(expose,id_reflector_height+id_reflector_z_offset)}; - G4double PMTHolderR[2] = {std::max(radius,reflectorRadius) + reflectorThickness, - std::max(radius,reflectorRadius) + reflectorThickness}; - G4double PMTHolderr[2] = {0,0}; - - - solidWCPMT = - new G4Polycone("WCPMT", - 0.0*deg, - 360.0*deg, - 2, - PMTHolderZ, - PMTHolderr, // R Inner - PMTHolderR);// R Outer - - } else{ - - // NEW: reduce overlaps by using curved surfaces, but only for mPMTs because - // the single PMTs will rest on a curved PMT support (in WCSim as the polygon is - // quite non trivial to model and not so important to get right initially, later - // we'll directly use the SolidWorks exported STL to CADMesh as Tesselated surface - // but that's not urgent now) - - ////////////////////////////// - /// Optional: Add PMT base /// - ////////////////////////////// - - // ToDo: Base is PMT property! Should not hard coded here. - G4double baseHeight = 0.; - G4double baseRadius = 0.; - if(addPMTBase){ - G4double basePinLength = 28.*CLHEP::mm; //TF measurement: pins + base + transformer. - baseHeight = 97.*CLHEP::mm - expose + basePinLength; //97mm includes the PMT top as well. - baseRadius = 26.*CLHEP::mm; //for R121990-02 - } else { - // version without a base but with optional reflectorCone - baseHeight = expose; - baseRadius = std::max(radius,reflectorRadius) + reflectorThickness; //radius + reflectorThickness; too tight - } - - // ToDo: extend the shell - ///////////////////////////// - - G4cout << mPMT_vessel_radius_curv << ", " << mPMT_outer_material_d << ", " << expose << ", " << dist_pmt_vessel << G4endl; - - // for single PMTs no overlap issues and will rest on flat black sheets - // Need spherical shell for curved top to better match (and reduce overlaps) with - // gel layer ("container") and vessel. - solidWCPMT = - new G4Sphere("WCPMT", - mPMT_vessel_radius_curv - mPMT_outer_material_d - pmtModuleHeight, //rMin, 54mm is position of support structure - mPMT_vessel_radius_curv - mPMT_outer_material_d, //rMax - 0.0*deg, //phiStart - 360.0*deg, //Deltaphi - 0.0*deg, //thetaStart - mPMT_pmt_openingAngle); //Deltatheta - // Once we have a full PMT support in (not urgent), - // replacing the current "inner" black sphere support by using - // G4Tesselated from CADMesh'ing an STL, - // Cutoff the bottom to make it flat. - - - - //z=0 for spherical shell needs to be shifted - wcpmt_z_offset = mPMT_vessel_radius_curv - mPMT_outer_material_d - pmtModuleHeight; - position_z_offset = mPMT_vessel_radius_curv - mPMT_outer_material_d - expose - dist_pmt_vessel; - G4cout << mPMT_vessel_radius_curv << ", " << mPMT_outer_material_d << ", " << expose << ", " << dist_pmt_vessel << G4endl; - } - - /* - G4Material *material_around_pmt = G4Material::GetMaterial("Water"); - if(id_reflector_height > 0.1*CLHEP::mm && - (vessel_radius > 1.*CLHEP::mm || vessel_cyl_height > 1.*CLHEP::mm)) //or make this a user option? - material_around_pmt = ; //"SilGel" - */ - - //ToDo: define defaults for single PMTs and rename variable (optional gel for single 20" PMT) - G4LogicalVolume* logicWCPMT = - new G4LogicalVolume( solidWCPMT, - G4Material::GetMaterial(mPMT_material_pmtAssembly), // Default: SilGel for n > 1, Water for n = 1 - "WCPMT", - 0,0,0); - - - - ////////////////////////// - // Visualize logicWCPMT // - ////////////////////////// - - if (Vis_Choice == "RayTracer"){ - // Makes the volume containing the PMT visible, solid, and forces the auxiliary edges to be viewed. - G4VisAttributes* WCPMTVisAtt = new G4VisAttributes(G4Colour(0.0,0.0,1.0)); - WCPMTVisAtt->SetForceSolid(true); // force the object to be visualized with a surface - WCPMTVisAtt->SetForceAuxEdgeVisible(true); // force auxiliary edges to be shown - - logicWCPMT->SetVisAttributes(WCPMTVisAtt); - } else{ - G4VisAttributes* WCPMTVisAtt2 = new G4VisAttributes(G4Colour(1.0,0.0,0.0)); - WCPMTVisAtt2->SetForceSolid(true); - if(addPMTBase) - logicWCPMT->SetVisAttributes(WCPMTVisAtt2); - else{ - // Makes the volume containg the PMT invisible for normal visualization - G4VisAttributes* WCPMTVisAttGrey = new G4VisAttributes(G4Colour(0.2,0.2,0.2)); - WCPMTVisAttGrey->SetForceSolid(true); - logicWCPMT->SetVisAttributes(G4VisAttributes::Invisible); - logicWCPMT->SetVisAttributes(WCPMTVisAttGrey); //TF debug overlaps with this volume - } - } - - //Need a volume to cut away excess behind blacksheet - G4Box* solidCutOffTubs = - new G4Box( "cutOffTubs", - sphereRadius+1.*cm, - sphereRadius+1.*cm, - PMTOffset); - - //////////////////////////// - /// Create PMT Interior /// - //////////////////////////// - - G4Sphere* tmpSolidInteriorWCPMT = - new G4Sphere( "tmpInteriorWCPMT", - 0.0*m,(sphereRadius-glassThickness), - 0.0*deg,360.0*deg, - 0.0*deg,90.0*deg); - - G4SubtractionSolid* solidInteriorWCPMT = - new G4SubtractionSolid( "InteriorWCPMT", - tmpSolidInteriorWCPMT, - solidCutOffTubs); - - // "Air" here is not true air, but a modified material - // with n = 1 and a very short absorption length - G4LogicalVolume* logicInteriorWCPMT = - new G4LogicalVolume( solidInteriorWCPMT, - G4Material::GetMaterial("Air"), - "InteriorWCPMT", - 0,0,0); - - G4VPhysicalVolume* physiInteriorWCPMT = - new G4PVPlacement(0, - G4ThreeVector(0, 0, -1.0*PMTOffset + position_z_offset), - logicInteriorWCPMT, - "InteriorWCPMT", - logicWCPMT, - false, - 0, - checkOverlaps); - - - ////////////////////////////////// - // Visualize logicInteriorWCPMT // - ///////////////////////////////// - - if (Vis_Choice == "RayTracer"){ - // Adding color and forcing the inner portion of the PMT's to be viewed - G4VisAttributes* WCPMTVisAtt = new G4VisAttributes(G4Colour(0.0,0.0,1.0)); - WCPMTVisAtt->SetForceSolid(true); // force the object to be visualized with a surface - WCPMTVisAtt->SetForceAuxEdgeVisible(true); // force auxiliary edges to be shown - - logicInteriorWCPMT->SetVisAttributes(WCPMTVisAtt); - } else { - // Making the inner portion of the detector invisible for OGLSX visualization - logicInteriorWCPMT->SetVisAttributes(G4VisAttributes::Invisible); - } - - ///////////////////////////// - /// Create PMT Glass Face /// - ///////////////////////////// - - G4Sphere* tmpGlassFaceWCPMT = - new G4Sphere( "tmpGlassFaceWCPMT", - (sphereRadius-glassThickness), - sphereRadius, - 0.0*deg,360.0*deg, - 0.0*deg,90.0*deg); - - G4SubtractionSolid* solidGlassFaceWCPMT = - new G4SubtractionSolid( CollectionName, - tmpGlassFaceWCPMT, - solidCutOffTubs); - - G4LogicalVolume *logicGlassFaceWCPMT = - new G4LogicalVolume( solidGlassFaceWCPMT, - G4Material::GetMaterial("Glass"), - CollectionName, - 0,0,0); - - G4VPhysicalVolume* physiGlassFaceWCPMT = - new G4PVPlacement(0, - G4ThreeVector(0, 0, -1.0*PMTOffset+position_z_offset), - logicGlassFaceWCPMT, - CollectionName, - logicWCPMT, - false, - 0, - checkOverlaps); - - // For either visualization type, logicGlassFaceWCPMT will either be visible - // or invisible depending on which line is commented at the end of the respective if statements. - - if (Vis_Choice == "OGLSX") - { // Gray wireframe visual style - // used in OGLSX visualizer - G4VisAttributes* WCPMTVisAtt = new G4VisAttributes(G4Colour(0.2,0.2,0.2)); - WCPMTVisAtt->SetForceWireframe(true); - //logicGlassFaceWCPMT->SetVisAttributes(G4VisAttributes::Invisible); - logicGlassFaceWCPMT->SetVisAttributes(WCPMTVisAtt);} - - if (Vis_Choice == "RayTracer"){ - // Blue wireframe visual style - // Used in the RayTracer visualizer - G4VisAttributes* WCPMTVisAtt = new G4VisAttributes(G4Colour(0.0,0.0,1.0)); - WCPMTVisAtt->SetForceSolid(true); // force the object to be visualized with a surface - WCPMTVisAtt->SetForceAuxEdgeVisible(true); // force auxiliary edges to be shown - //logicGlassFaceWCPMT->SetVisAttributes(G4VisAttributes::Invisible); - - logicGlassFaceWCPMT->SetVisAttributes(WCPMTVisAtt); - } else { - - // Gray wireframe visual style - // used in OGLSX visualizer - //G4VisAttributes* WCPMTVisAtt = new G4VisAttributes(G4Colour(0.2,0.2,0.2)); //better for seeing hits - //WCPMTVisAtt->SetForceWireframe(true); - G4VisAttributes* WCPMTVisAtt = new G4VisAttributes(G4Colour(0.0,1.0,0.0)); //better for seeing geometry - WCPMTVisAtt->SetForceSolid(true); - //logicGlassFaceWCPMT->SetVisAttributes(G4VisAttributes::Invisible); - logicGlassFaceWCPMT->SetVisAttributes(WCPMTVisAtt); - - } - - //Add Logical Border Surface - new G4LogicalBorderSurface("GlassCathodeSurface", - physiGlassFaceWCPMT, - physiInteriorWCPMT, - OpGlassCathodeSurface); - - - - ////////////////////////// - /// Optional Reflector /// - ////////////////////////// - - if(id_reflector_height > 0.1*CLHEP::mm - && (reflectorRadius-radius) > -5*CLHEP::mm){ - - /* Some details: - * 1.1mm is the gap between reflector and PMT for mechanical construction - * (KM3NeT support matrix value) - * three degrees of freedom: height, z position and opening angle - */ - G4Cons* reflectorCone = - new G4Cons("WCPMT_reflect", - radius + 1.1*CLHEP::mm, //rmin - radius + 1.1*CLHEP::mm + reflectorThickness, //rmax - reflectorRadius + 1.1*CLHEP::mm, //Rmin - reflectorRadius + 1.1*CLHEP::mm + reflectorThickness, //Rmax - id_reflector_height/2, //z/2 - 0, 2*CLHEP::pi); - - - G4LogicalVolume* logicReflector = - new G4LogicalVolume( reflectorCone, - G4Material::GetMaterial("Aluminum"), //It actually is Al+ Ag evaporation - "reflectorCone", - 0,0,0); - - // Visualize - G4VisAttributes* WCPMTVisAtt3 = new G4VisAttributes(G4Colour(0.0,0.0,1.0)); - WCPMTVisAtt3->SetForceSolid(true); - logicReflector->SetVisAttributes(WCPMTVisAtt3); - - new G4LogicalSkinSurface("ReflectorLogSkinSurface",logicReflector,ReflectorSkinSurface); - - // reflector PMT Placement: - new G4PVPlacement(0, - G4ThreeVector(0, 0, id_reflector_z_offset+id_reflector_height/2+position_z_offset), - logicReflector, - "reflectorWCPMT", - logicWCPMT, - false, - 0, - checkOverlaps); - } - - - G4cout << id_reflector_height << ", " << id_reflector_height - radius << ", " << radius << ", " << reflectorThickness << ", " << reflectorRadius << ", " << id_reflector_z_offset << ", " << position_z_offset << G4endl; - - if(nIDPMTs > 1){ - //////////////////// - /// 1-PMT support // - //////////////////// - //54mm is position of full PMT support wrt inside of pressure vessel. - G4Cons * solidWCPMTsupport = - new G4Cons("WCPMTsupport", - 0., //rmin1 - tan(mPMT_pmt_openingAngle)*(mPMT_vessel_radius_curv - mPMT_outer_material_d - pmtModuleHeight), //rmax1 - 0., //rmin2 - tan(mPMT_pmt_openingAngle)*(mPMT_vessel_radius_curv - mPMT_outer_material_d - -expose - dist_pmt_vessel), //rmax2 - (pmtModuleHeight - expose - dist_pmt_vessel)/2, //h/2 - 0.0*deg, //phiStart - 360.0*deg); //Deltaphi - - G4LogicalVolume* logicWCPMTsupport = - new G4LogicalVolume(solidWCPMTsupport, - G4Material::GetMaterial("Blacksheet"), - "WCPMTsupport", - 0,0,0); - - new G4LogicalSkinSurface("FoamLogSkinSurface",logicWCPMTsupport,OpGelFoamSurface); - - new G4PVPlacement(0, - G4ThreeVector(0, 0, wcpmt_z_offset+(pmtModuleHeight - expose - dist_pmt_vessel)/2), - logicWCPMTsupport, - "WCPMTsupport", - logicWCPMT, - false, - 0, - checkOverlaps); - - // Visualize - G4VisAttributes* WCPMTVisAtt_sup = new G4VisAttributes(G4Colour(0.3,0.3,0.3)); - WCPMTVisAtt_sup->SetForceSolid(true); - logicWCPMTsupport->SetVisAttributes(WCPMTVisAtt_sup); - - - //Reflector support - if(id_reflector_height > 0.1*CLHEP::mm - && (reflectorRadius-radius) > -5*CLHEP::mm){ - - G4double ReflectorHolderZ[3] = {0, id_reflector_z_offset, id_reflector_z_offset+id_reflector_height}; - G4double ReflectorHolderR[3] = {tan(mPMT_pmt_openingAngle)*(mPMT_vessel_radius_curv - mPMT_outer_material_d - -expose - dist_pmt_vessel), - tan(mPMT_pmt_openingAngle)*(mPMT_vessel_radius_curv - mPMT_outer_material_d - -expose - dist_pmt_vessel+id_reflector_z_offset), - tan(mPMT_pmt_openingAngle)*(mPMT_vessel_radius_curv - mPMT_outer_material_d - -expose - dist_pmt_vessel+id_reflector_z_offset - +id_reflector_height)}; - - G4double ReflectorHolderr[3] = {radius + 1.1*CLHEP::mm + reflectorThickness, - radius + 1.1*CLHEP::mm + reflectorThickness, - reflectorRadius + 1.1*CLHEP::mm + reflectorThickness}; - - G4Polycone * solidWCPMTsupport2 = - new G4Polycone("WCPMTsupport2", - 0.0*deg, - 360.0*deg, - 3, - ReflectorHolderZ, - ReflectorHolderr, // R Inner - ReflectorHolderR);// R Outer - - - G4LogicalVolume* logicWCPMTsupport2 = - new G4LogicalVolume(solidWCPMTsupport2, - G4Material::GetMaterial("Blacksheet"), - "WCPMTsupport", - 0,0,0); - - new G4LogicalSkinSurface("FoamLogSkinSurface2",logicWCPMTsupport2,OpGelFoamSurface); - logicWCPMTsupport2->SetVisAttributes(WCPMTVisAtt_sup); - - new G4PVPlacement(0, - G4ThreeVector(0, 0, position_z_offset), - logicWCPMTsupport2, - "WCPMTsupport2", - logicWCPMT, - false, - 0, - checkOverlaps); - } - - } - - ///////////////////////// - // Sensitive detector /// - ///////////////////////// - - // Instantiate a new sensitive detector - // and register this sensitive detector volume with the SD Manager. - G4SDManager* SDman = G4SDManager::GetSDMpointer(); - G4String SDName = "/WCSim/"; - SDName += CollectionName; - - // If there is no such sensitive detector with that SDName yet, - // make a new one - if( ! SDman->FindSensitiveDetector(SDName, false) ) { - - aWCPMT = new WCSimWCSD(CollectionName,SDName,this,detectorElement); - SDman->AddNewDetector( aWCPMT ); - } - - logicGlassFaceWCPMT->SetSensitiveDetector( aWCPMT ); - - - PMTLogicalVolumes[key] = logicWCPMT; - - return logicWCPMT; -} diff --git a/libWCSIM/src/WCSimDarkRateMessenger.cc b/libWCSIM/src/WCSimDarkRateMessenger.cc deleted file mode 100644 index e35d62d..0000000 --- a/libWCSIM/src/WCSimDarkRateMessenger.cc +++ /dev/null @@ -1,193 +0,0 @@ -#include "WCSimDarkRateMessenger.hh" -#include "WCSimWCAddDarkNoise.hh" - -#include "G4UIdirectory.hh" -#include "G4UIcommand.hh" -#include "G4UIcmdWithADoubleAndUnit.hh" -#include "G4UIcmdWithADouble.hh" -#include "G4UIcmdWithAnInteger.hh" -#include "G4UIcmdWithAString.hh" - -WCSimDarkRateMessenger::WCSimDarkRateMessenger(WCSimWCAddDarkNoise* darkratepoint) - : WCSimAddDarkNoise(darkratepoint), WCSimDir(0), SetFrequency(0), SetConversionRate(0), SetDarkMode(0), SetDarkLow(0), SetDarkHigh(0), SetDarkWindow(0), SetDetectorElement(0) -{ - //Initialize(); -} - -WCSimDarkRateMessenger* WCSimDarkRateMessenger::iInstance = NULL; - -WCSimDarkRateMessenger* WCSimDarkRateMessenger::GetInstance(){ - //G4cout<<"making the static instance"<SetGuidance("Commands to change the dark noise frequency of the simulation"); - - double const conversion_to_kHz = 1000000; - double defaultFrequency = 0; - if(SetFrequency==0){SetFrequency = new G4UIcmdWithADoubleAndUnit("/DarkRate/SetDarkRate",this);} - SetFrequency->SetGuidance("Commands to change the dark noise frequency of the simulation"); - SetFrequency->SetParameterName("DarkRate",false); - SetFrequency->SetDefaultValue(defaultFrequency * conversion_to_kHz); - //kilohertz is 10e-6 (CLHEP units: 1kHz = 1000 / (1e9 ns) = 1e-6 as ns := 1.) - SetFrequency->SetUnitCategory("Frequency"); - SetFrequency->SetDefaultUnit("kHz"); - SetFrequency->SetUnitCandidates("Hz kHz MHz GHz"); - - - double defaultConvRate = 1; - if(SetConversionRate==0){ SetConversionRate = new G4UIcmdWithADouble("/DarkRate/SetConvert",this);} - SetConversionRate->SetGuidance("Calibrate the frequency of dark noise before applying the threshold (typically 0.25pe)"); - SetConversionRate->SetParameterName("DigiCorr",false); - SetConversionRate->SetDefaultValue(defaultConvRate); - - //Mode 0 - Add dark rate in window defined by /DarkRate/SetDarkLow and /DarkRate/SetDarkHigh - //If not set default is 0 and 100000ns - //Mode 1 - Add dark rate to a window of size /DarkRate/SetDarkWindow around each hit - int defaultDarkMode = 0; - if(SetDarkMode==0){SetDarkMode = new G4UIcmdWithAnInteger("/DarkRate/SetDarkMode",this);} - SetDarkMode->SetGuidance("Set the mode of adding dark noise to the event"); - SetDarkMode->SetParameterName("DarkMode",false); - SetDarkMode->SetDefaultValue(defaultDarkMode); - SetNewValue(SetDarkMode, G4UIcommand::ConvertToString(defaultDarkMode)); - - double defaultDarkLow = 0; - if(SetDarkLow==0){SetDarkLow = new G4UIcmdWithADoubleAndUnit("/DarkRate/SetDarkLow",this);} - SetDarkLow->SetGuidance("Set the lower GEANT time limit to add dark noise"); - SetDarkLow->SetParameterName("DarkLow",false); - SetDarkLow->SetDefaultValue(defaultDarkLow); - SetDarkLow->SetUnitCategory("Time"); - SetDarkLow->SetDefaultUnit("ns"); - SetNewValue(SetDarkLow, G4UIcommand::ConvertToString(defaultDarkLow, "ns")); - - double defaultDarkHigh = 100000; - if(SetDarkHigh==0){SetDarkHigh = new G4UIcmdWithADoubleAndUnit("/DarkRate/SetDarkHigh",this);} - SetDarkHigh->SetGuidance("Set the upper GEANT time limit to add dark noise"); - SetDarkHigh->SetParameterName("DarkHigh",false); - SetDarkHigh->SetDefaultValue(defaultDarkHigh); - SetDarkHigh->SetUnitCategory("Time"); - SetDarkHigh->SetDefaultUnit("ns"); - SetNewValue(SetDarkHigh, G4UIcommand::ConvertToString(defaultDarkHigh, "ns")); - - double defaultDarkWindow = 5000; - if(SetDarkWindow==0){SetDarkWindow = new G4UIcmdWithADoubleAndUnit("/DarkRate/SetDarkWindow",this);} - SetDarkWindow->SetGuidance("Set the window width to add dark noise"); - SetDarkWindow->SetParameterName("DarkWindow",false); - SetDarkWindow->SetDefaultValue(defaultDarkWindow); - SetDarkWindow->SetUnitCategory("Time"); - SetDarkWindow->SetDefaultUnit("ns"); - SetNewValue(SetDarkWindow, G4UIcommand::ConvertToString(defaultDarkWindow, "ns")); - - G4String defaultDetectorElement = "tank"; - if(SetDetectorElement==0){ SetDetectorElement = new G4UIcmdWithAString("/DarkRate/SetDetectorElement", this); } - SetDetectorElement->SetGuidance("Set the detector element for which trigger settings will be applied"); - G4String theguidance = "Available choices are:\n"; - G4String thecandidates=""; - for(std::map::iterator it=darknoiseptrs.begin(); it!=darknoiseptrs.end(); ++it){ - theguidance+=it->first; theguidance+="\n"; - thecandidates+=it->first; thecandidates+=" "; - } - SetDetectorElement->SetGuidance(theguidance); - SetDetectorElement->SetCandidates(thecandidates); - SetDetectorElement->SetParameterName("detectorElement", false); - SetDetectorElement->AvailableForStates(G4State_PreInit, G4State_Idle); - SetDetectorElement->SetDefaultValue(defaultDetectorElement); - - initaliseString = ""; -} - -WCSimDarkRateMessenger::~WCSimDarkRateMessenger() -{ - - delete SetFrequency; - delete SetConversionRate; - delete SetDarkMode; - delete SetDarkLow; - delete SetDarkHigh; - delete SetDarkWindow; - delete WCSimDir; - delete SetDetectorElement; -} - -void WCSimDarkRateMessenger::SetNewValue(G4UIcommand* command,G4String newValue) -{ - - for(std::map::iterator it=darknoiseptrs.begin(); it!=darknoiseptrs.end(); ++it){ - WCSimAddDarkNoise = (WCSimWCAddDarkNoise*) it->second; - - if(command == SetFrequency){ - // Since kHz is 10e-3 for this class we must multiply by a 10e6 factor - // to make default units in kHz - // ToDo: make this consistent with CLHEP unit system. Makes it easier to just using CLHEP! - double const conversion_to_kHz = 1000000; - WCSimAddDarkNoise->SetDarkRate(conversion_to_kHz * SetFrequency->GetNewDoubleValue(newValue)); - G4cout << "Setting Dark Rate " << conversion_to_kHz * SetFrequency->GetNewDoubleValue(newValue) - << initaliseString.c_str() << G4endl; - } - else if(command == SetConversionRate){ - WCSimAddDarkNoise->SetConversion(SetConversionRate->GetNewDoubleValue(newValue)); - G4cout << "Setting Dark Rate Conversion value " << SetConversionRate->GetNewDoubleValue(newValue) - << initaliseString.c_str() << G4endl; - } - else if(command == SetDarkMode){ - WCSimAddDarkNoise->SetDarkMode(SetDarkMode->GetNewIntValue(newValue)); - G4cout << "Setting DarkMode value " << SetDarkMode->GetNewIntValue(newValue) - << initaliseString.c_str() << G4endl; - } - else if(command == SetDarkLow){ - WCSimAddDarkNoise->SetDarkLow(SetDarkLow->GetNewDoubleValue(newValue)); - G4cout << "Setting DarkLow value " << SetDarkLow->GetNewDoubleValue(newValue) - << initaliseString.c_str() << G4endl; - } - else if(command == SetDarkHigh){ - WCSimAddDarkNoise->SetDarkHigh(SetDarkHigh->GetNewDoubleValue(newValue)); - G4cout << "Setting DarkHigh value " << SetDarkHigh->GetNewDoubleValue(newValue) - << initaliseString.c_str() << G4endl; - } - else if(command == SetDarkWindow){ - WCSimAddDarkNoise->SetDarkWindow(SetDarkWindow->GetNewDoubleValue(newValue)); - G4cout << "Setting DarkWindow value " << SetDarkWindow->GetNewDoubleValue(newValue) - << initaliseString.c_str() << G4endl; - } - else if(command == SetDetectorElement){ - if(darknoiseptrs.count(newValue)>0){ - WCSimAddDarkNoise=darknoiseptrs[newValue]; - G4cout << "Setting detectorElement value " << newValue - << initaliseString.c_str() << G4endl; - } - } - } -} - -void WCSimDarkRateMessenger::AddDarkRateInstance(WCSimWCAddDarkNoise* darkratepoint, G4String detectorElementin){ - if(darknoiseptrs.count(detectorElementin)>0){ - G4cout<<"new dark rate messenger for existing detector Element "<(detectorElementin,darkratepoint)); - WCSimAddDarkNoise=darkratepoint; - detectorElement=detectorElementin; - } -} - -void WCSimDarkRateMessenger::RemoveDarkRateInstance(G4String detectorElementin){ - if(darknoiseptrs.count(detectorElementin)>0){ - G4cout<<"deleting dark rate messenger for detector Element "<::iterator thepointer = darknoiseptrs.find(detectorElementin); - darknoiseptrs.erase(thepointer); - if(darknoiseptrs.size()==0){ delete this; } - } else { - G4cout<<"Attempt to remove nonexistant element "<GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 33.6815*m; //16.900*2*cos(2*pi*rad/75)*m; //inner detector diameter - WCIDHeight = 36.200*m; //"" "" height - WCBarrelPMTOffset = WCPMTRadius; //offset from vertical - WCBarrelNumPMTHorizontal = 150; - WCBarrelNRings = 17.; - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCCapPMTSpacing = 0.707*m; // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - InitSinglePMT(); -} - - -// Note: the actual coverage is 20.27% -void WCSimDetectorConstruction::SuperK_20inchPMT_20perCent() -{ - WCDetectorName = "SuperK_20inchPMT_20perCent"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("PMT20inch", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 33.6815*m; //16.900*2*cos(2*pi*rad/75)*m; //inner detector diameter - WCIDHeight = 36.200*m; //"" "" height - WCBarrelPMTOffset = 0.0715*m; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 20.27; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - InitSinglePMT(); -} - - -// Note: the actual coverage is 20.27% -void WCSimDetectorConstruction::SuperK_20inchBandL_20perCent() -{ - WCDetectorName = "SuperK_20inchBandL_20perCent"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("BoxandLine20inchHQE", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 33.6815*m; //16.900*2*cos(2*pi*rad/75)*m; //inner detector diameter - WCIDHeight = 36.200*m; //"" "" height - WCBarrelPMTOffset = 0.0715*m; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 20.27; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - InitSinglePMT(); -} - - -// Note: the actual coverage is 14.59% -void WCSimDetectorConstruction::SuperK_12inchBandL_15perCent() -{ - WCDetectorName = "SuperK_12inchBandL_15perCent"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("BoxandLine12inchHQE", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 33.6815*m; //16.900*2*cos(2*pi*rad/75)*m; //inner detector diameter - WCIDHeight = 36.200*m; //"" "" height - WCBarrelPMTOffset = 0.0715*m; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 14.59; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - InitSinglePMT(); -} - - -// Note: the actual coverage is 13.51% -void WCSimDetectorConstruction::SuperK_20inchBandL_14perCent() -{ - WCDetectorName = "SuperK_20inchBandL_14perCent"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("BoxandLine20inchHQE", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 33.6815*m; //16.900*2*cos(2*pi*rad/75)*m; //inner detector diameter - WCIDHeight = 36.200*m; //"" "" height - WCBarrelPMTOffset = 0.0715*m; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 13.51; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - InitSinglePMT(); -} - -void WCSimDetectorConstruction::Cylinder_60x74_20inchBandL_14perCent() -{ - WCDetectorName = "Cylinder_60x74_20inchBandL_14perCent()"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("BoxandLine20inchHQE", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 74.0*m; - WCIDHeight = 60.0*m; - WCBarrelPMTOffset = WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 13.51; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - InitSinglePMT(); -} - -void WCSimDetectorConstruction::Cylinder_60x74_20inchBandL_40perCent() -{ - WCDetectorName = "Cylinder_60x74_20inchBandL_40perCent"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("BoxandLine20inchHQE", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 74.0*m; - WCIDHeight = 60.0*m; - WCBarrelPMTOffset = WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 40.0; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - InitSinglePMT(); -} - -void WCSimDetectorConstruction::Cylinder_12inchHPD_15perCent() -{ - WCDetectorName = "Cylinder_12inchHPD_15perCent"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - // cylindrical detector with a height of 100m and a diameter of 69m - // with 12" HPD and 14.59% photocoverage - WCSimPMTObject * PMT = CreatePMTObject("HPD12inchHQE", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 69.0*m; - WCIDHeight = 100.0*m; - WCBarrelPMTOffset = WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 14.59; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - InitSinglePMT(); -} - -void WCSimDetectorConstruction::SetHyperKGeometry() -{ - WCDetectorName = "HyperK"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("BoxandLine20inchHQE", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 70.8*m; // = 74m - 2*(60cm ID wall + 1m OD) - WCIDHeight = 54.8*m; // = 60m - 2*(60cm ID wall + 2m OD) - WCBarrelPMTOffset = WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 40.0; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; -} -//B. Quilain, HK with 20'' PMTs -void WCSimDetectorConstruction::SetHyperK_SKPMTGeometry() -{ - WCDetectorName = "HyperK_SKPMT"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("PMT20inch", WCIDCollectionName); - - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 70.8*m; // = 74m - 2*(60cm ID wall + 1m OD) - WCIDHeight = 54.8*m; // = 60m - 2*(60cm ID wall + 2m OD) - WCBarrelPMTOffset = WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 40.0; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; -} - -//B. Quilain, HK with 10'' PMTs -void WCSimDetectorConstruction::SetHyperK_10inchGeometry() -{ - WCDetectorName = "HyperK_10inch"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("PMT10inch", WCIDCollectionName); - - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 70.8*m; // = 74m - 2*(60cm ID wall + 1m OD) - WCIDHeight = 54.8*m; // = 60m - 2*(60cm ID wall + 2m OD) - WCBarrelPMTOffset = WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 40.0;//10.0; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; -} - -//B. Quilain, HK with 8'' PMTs -void WCSimDetectorConstruction::SetHyperK_8inchGeometry() -{ - WCDetectorName = "HyperK_8inch"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("PMT8inch", WCIDCollectionName); - - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 70.8*m; // = 74m - 2*(60cm ID wall + 1m OD) - WCIDHeight = 54.8*m; // = 60m - 2*(60cm ID wall + 2m OD) - WCBarrelPMTOffset = WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 6.4;//40.0; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; -} - -//B. Quilain, HK with 3'' PMTs -void WCSimDetectorConstruction::SetHyperK_3inchGeometry() -{ - WCDetectorName = "HyperK_3inch"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("PMT3inchR12199_02", WCIDCollectionName); - - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - WCIDDiameter = 70.8*m; // = 74m - 2*(60cm ID wall + 1m OD) - WCIDHeight = 54.8*m; // = 60m - 2*(60cm ID wall + 2m OD) - WCBarrelPMTOffset = WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 1;//40.0; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; -} -// B.Quilain, HK with mPMTs -// These are defaults that can be altered through the macros -void WCSimDetectorConstruction::SetHyperK_mPMTGeometry() -{ - WCDetectorName = "HyperK_mPMT"; - hybrid=false; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - mPMT_ID_PMT = "PMT3inchR14374"; //can be changed in macro through mPMT settings. - mPMT_OD_PMT = "PMT3inchR14374"; - //mPMT_ID_PMT = "PMT3inchR12199_02"; //can be changed in macro through mPMT settings. - //mPMT_OD_PMT = "PMT3inchR12199_02"; - WCSimPMTObject * PMT = CreatePMTObject(mPMT_ID_PMT, WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius();//Radius of a small PMT, not of a module - - //mPMT params go first because detector depends on it: - mPMT_vessel_cyl_height = 38.*CLHEP::mm; //option A, option B would be 277 mm - mPMT_vessel_radius_curv = 342.*CLHEP::mm; //needs to include the vessel thickness, as we construct from outside inwards. - mPMT_vessel_radius = 254.*CLHEP::mm; - dist_pmt_vessel = 2*CLHEP::mm; - orientation = PERPENDICULAR; - mPMT_outer_material = "G4_PLEXIGLASS"; - mPMT_inner_material = "Air"; // TODO: real air, hence update abs_length - mPMT_material_pmtAssembly = "SilGel"; - mPMT_outer_material_d = 10*CLHEP::mm; - - // Radius of cone at z=reflectorHeight - id_reflector_height = 0;//6.53*CLHEP::mm; // for a radius of 7.25mm, for hex: 5.4mm (radius of 6mm) - id_reflector_z_offset = 0;//4.8*CLHEP::mm; //from KM3Net CAD drawings - id_reflector_angle = 0;//48.*CLHEP::deg; // Need to be remeasured for different PMT curvature - mPMT_pmt_openingAngle = 8.7*CLHEP::deg; // for hex: 8.5deg - G4double mPMT_vessel_tot_height = mPMT_vessel_radius + mPMT_vessel_cyl_height; - - // parameters related to filling the ID mPMT - nID_PMTs = 19; - config_file = wcsimdir_path+"/mPMT-configfiles/mPMTconfig_19_nuPrism_3ring.txt"; // for smaller reflector, use: mPMTconfig_19_nuPrism.txt (hex) - - WCBarrelPMTOffset = mPMT_vessel_tot_height; - WCIDVerticalPosition = 0.; - - WCIDDiameter = 70.8*m; // = 74m - 2*(60cm ID wall + 1m OD) - WCIDHeight = 54.8*m; // = 60m - 2*(60cm ID wall + 2m OD) - WCPMTperCellHorizontal= 1;//4; - WCPMTperCellVertical = 1;//3; - WCPMTPercentCoverage = 5.0;//40.0; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*mPMT_vessel_radius)); - G4cout << "## Number of Horizontal PMT along diameter = " << WCBarrelNumPMTHorizontal << ", diameter = "<< WCIDDiameter << ", coverage = "<< WCPMTPercentCoverage << ", PMT radius = " << WCPMTRadius << ", test = " << WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius) << G4endl; - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; -} - -// B.Quilain, HK with hybrid configuration: mPMTs + B&L -// These are defaults that can be altered through the macros -void WCSimDetectorConstruction::SetHyperK_HybridmPMTGeometry() -{ - //Box&Line - WCDetectorName = "HyperK_HybridmPMT"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("BoxandLine20inchHQE", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - nID_PMTs = 1; - //WCPMTVesselCycHeight = 0; - //WCPMTVesselRadiusCurv = 0; - //WCPMTVesselRadius = 0; - //WCPMTDistPMTVessel = 0; - - //WCDetectorName2 = "HyperK_HybridmPMT"; - //WCIDCollectionName2 = WCDetectorName +"-glassFaceWCPMT"; - //WCSimPMTObject * PMT2 = CreatePMTObject("BoxandLine20inchHQE", WCIDCollectionName2); - //WCPMTName2 = PMT2->GetPMTName(); - //WCPMTExposeHeight2 = PMT2->GetExposeHeight(); - //WCPMTRadius2 = PMT2->GetRadius(); - //nID_PMTs = 1; - - //mPMT - hybrid = true; - WCIDCollectionName2 = WCDetectorName +"-glassFaceWCPMT2"; - mPMT_ID_PMT = "PMT3inchR14374"; //can be changed in macro through mPMT settings. - mPMT_OD_PMT = "PMT3inchR14374"; - //mPMT_ID_PMT = "PMT3inchR12199_02"; //can be changed in macro through mPMT settings. - //mPMT_OD_PMT = "PMT3inchR12199_02"; - WCSimPMTObject * PMT2 = CreatePMTObject(mPMT_ID_PMT, WCIDCollectionName2); - WCPMTName2 = PMT2->GetPMTName(); - WCPMTExposeHeight2 = PMT2->GetExposeHeight(); - WCPMTRadius2 = PMT2->GetRadius();//Radius of a small PMT, not of a module - nID_PMTs2 = 19; - //WCPMTVesselCycHeight = 38.*CLHEP::mm; //option A, option B would be 277 mm - //WCPMTVesselRadiusCurv = 342.*CLHEP::mm; //needs to include the vessel thickness, as we construct from outside inwards. - //WCPMTVesselRadius = 254.*CLHEP::mm; - //WCPMTDistPMTVessel = 2*CLHEP::mm; - - //mPMT params go first because detector depends on it: - mPMT_vessel_cyl_height = 38.*CLHEP::mm; //option A, option B would be 277 mm - mPMT_vessel_radius_curv = 342.*CLHEP::mm; //needs to include the vessel thickness, as we construct from outside inwards. - mPMT_vessel_radius = 254.*CLHEP::mm; - dist_pmt_vessel = 2*CLHEP::mm; - - orientation = PERPENDICULAR; - mPMT_outer_material = "G4_PLEXIGLASS"; - mPMT_inner_material = "Air"; // TODO: real air, hence update abs_length - mPMT_material_pmtAssembly = "SilGel"; - mPMT_outer_material_d = 10*CLHEP::mm; - // Radius of cone at z=reflectorHeight - id_reflector_height = 0;//6.53*CLHEP::mm; // for a radius of 7.25mm, for hex: 5.4mm (radius of 6mm) - id_reflector_z_offset = 0;//4.8*CLHEP::mm; //from KM3Net CAD drawings - id_reflector_angle = 0;//48.*CLHEP::deg; // Need to be remeasured for different PMT curvature - mPMT_pmt_openingAngle = 8.7*CLHEP::deg; // for hex: 8.5deg - // parameters related to filling the ID mPMT - config_file = wcsimdir_path+"/mPMT-configfiles/mPMTconfig_19_nuPrism_3ring.txt"; // for smaller reflector, use: mPMTconfig_19_nuPrism.txt (hex) - - G4double mPMT_vessel_tot_height = mPMT_vessel_radius + mPMT_vessel_cyl_height; - - - WCIDDiameter = 70.8*m; // = 74m - 2*(60cm ID wall + 1m OD) - WCIDHeight = 54.8*m; // = 60m - 2*(60cm ID wall + 2m OD) - WCBarrelPMTOffset = std::max(WCPMTRadius,mPMT_vessel_tot_height); //offset from vertical - WCIDVerticalPosition = 0.; - - WCPMTPercentCoverage = 20.0;//20.0;//40.0;//40.0; - WCPMTPercentCoverage2 = 5.;//5.0;//20.0; - WCPMTperCellHorizontal = round((WCPMTPercentCoverage+WCPMTPercentCoverage2) / (WCPMTPercentCoverage2==0?(WCPMTPercentCoverage+WCPMTPercentCoverage2):WCPMTPercentCoverage2)); - WCPMTperCellVertical = WCPMTperCellHorizontal; - //WCPMTperCellHorizontal= ((int) WCPMTPercentCoverage2/(WCPMTPercentCoverage+WCPMTPercentCoverage2));//Should be in agreement with coverage of PMT 1 & 2: here, 4/5 PMTs are B&L + 1/5 mPMT - //WCPMTperCellVertical = ((int) WCPMTPercentCoverage2/(WCPMTPercentCoverage+WCPMTPercentCoverage2));//Should be in agreement with coverage of PMT 1 & 2: here, 4/5 PMTs are B&L + 1/5 mPMT - - //WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*(WCPMTPercentCoverage+WCPMTPercentCoverage2))/(10.0*std::max(WCPMTRadius,mPMT_vessel_tot_height))); - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*(WCPMTPercentCoverage+WCPMTPercentCoverage2))/(10.0*std::max(WCPMTRadius,mPMT_vessel_radius))); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - //G4cout << "## Number of Horizontal PMT along diameter = " << WCBarrelNumPMTHorizontal << ", diameter = "<< WCIDDiameter << ", coverage = "<< WCPMTPercentCoverage << ", PMT radius = " << WCPMTRadius << ", test = " << WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius) << G4endl; - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - 5*std::max(WCPMTExposeHeight,mPMT_vessel_tot_height);//3.0*cm;//WCIDDiameter/2.0 - std::max(WCPMTRadius,mPMT_vessel_tot_height);//WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; -} -// B.Quilain, HK with hybrid configuration: mPMTs + B&L -// These are defaults that can be altered through the macros -void WCSimDetectorConstruction::SetHyperK_HybridmPMT10PCGeometry() -{ - //Box&Line - WCDetectorName = "HyperK_HybridmPMT"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("BoxandLine20inchHQE", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - nID_PMTs = 1; - //WCPMTVesselCycHeight = 0; - //WCPMTVesselRadiusCurv = 0; - //WCPMTVesselRadius = 0; - //WCPMTDistPMTVessel = 0; - - //WCDetectorName2 = "HyperK_HybridmPMT"; - //WCIDCollectionName2 = WCDetectorName +"-glassFaceWCPMT"; - //WCSimPMTObject * PMT2 = CreatePMTObject("BoxandLine20inchHQE", WCIDCollectionName2); - //WCPMTName2 = PMT2->GetPMTName(); - //WCPMTExposeHeight2 = PMT2->GetExposeHeight(); - //WCPMTRadius2 = PMT2->GetRadius(); - //nID_PMTs = 1; - - //mPMT - hybrid = true; - WCIDCollectionName2 = WCDetectorName +"-glassFaceWCPMT2"; - mPMT_ID_PMT = "PMT3inchR14374"; //can be changed in macro through mPMT settings. - mPMT_OD_PMT = "PMT3inchR14374"; - //mPMT_ID_PMT = "PMT3inchR12199_02"; //can be changed in macro through mPMT settings. - //mPMT_OD_PMT = "PMT3inchR12199_02"; - WCSimPMTObject * PMT2 = CreatePMTObject(mPMT_ID_PMT, WCIDCollectionName2); - WCPMTName2 = PMT2->GetPMTName(); - WCPMTExposeHeight2 = PMT2->GetExposeHeight(); - WCPMTRadius2 = PMT2->GetRadius();//Radius of a small PMT, not of a module - nID_PMTs2 = 19; - //WCPMTVesselCycHeight = 38.*CLHEP::mm; //option A, option B would be 277 mm - //WCPMTVesselRadiusCurv = 342.*CLHEP::mm; //needs to include the vessel thickness, as we construct from outside inwards. - //WCPMTVesselRadius = 254.*CLHEP::mm; - //WCPMTDistPMTVessel = 2*CLHEP::mm; - - //mPMT params go first because detector depends on it: - mPMT_vessel_cyl_height = 38.*CLHEP::mm; //option A, option B would be 277 mm - mPMT_vessel_radius_curv = 342.*CLHEP::mm; //needs to include the vessel thickness, as we construct from outside inwards. - mPMT_vessel_radius = 254.*CLHEP::mm; - dist_pmt_vessel = 2*CLHEP::mm; - - orientation = PERPENDICULAR; - mPMT_outer_material = "G4_PLEXIGLASS"; - mPMT_inner_material = "Air"; // TODO: real air, hence update abs_length - mPMT_material_pmtAssembly = "SilGel"; - mPMT_outer_material_d = 10*CLHEP::mm; - // Radius of cone at z=reflectorHeight - id_reflector_height = 0;//6.53*CLHEP::mm; // for a radius of 7.25mm, for hex: 5.4mm (radius of 6mm) - id_reflector_z_offset = 0;//4.8*CLHEP::mm; //from KM3Net CAD drawings - id_reflector_angle = 0;//48.*CLHEP::deg; // Need to be remeasured for different PMT curvature - mPMT_pmt_openingAngle = 8.7*CLHEP::deg; // for hex: 8.5deg - // parameters related to filling the ID mPMT - config_file = wcsimdir_path+"/mPMT-configfiles/mPMTconfig_19_nuPrism_3ring.txt"; // for smaller reflector, use: mPMTconfig_19_nuPrism.txt (hex) - - G4double mPMT_vessel_tot_height = mPMT_vessel_radius + mPMT_vessel_cyl_height; - - - WCIDDiameter = 70.8*m; // = 74m - 2*(60cm ID wall + 1m OD) - WCIDHeight = 54.8*m; // = 60m - 2*(60cm ID wall + 2m OD) - WCBarrelPMTOffset = std::max(WCPMTRadius,mPMT_vessel_tot_height); //offset from vertical - WCIDVerticalPosition = 0.; - - WCPMTPercentCoverage = 20.0;//40.0;//40.0; - WCPMTPercentCoverage2 = 10.0;//20.0; - WCPMTperCellHorizontal = round((WCPMTPercentCoverage+WCPMTPercentCoverage2) / (WCPMTPercentCoverage2==0?(WCPMTPercentCoverage+WCPMTPercentCoverage2):WCPMTPercentCoverage2)); - WCPMTperCellVertical = WCPMTperCellHorizontal; - //WCPMTperCellHorizontal= ((int) WCPMTPercentCoverage2/(WCPMTPercentCoverage+WCPMTPercentCoverage2));//Should be in agreement with coverage of PMT 1 & 2: here, 4/5 PMTs are B&L + 1/5 mPMT - //WCPMTperCellVertical = ((int) WCPMTPercentCoverage2/(WCPMTPercentCoverage+WCPMTPercentCoverage2));//Should be in agreement with coverage of PMT 1 & 2: here, 4/5 PMTs are B&L + 1/5 mPMT - - //WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*(WCPMTPercentCoverage+WCPMTPercentCoverage2))/(10.0*std::max(WCPMTRadius,mPMT_vessel_tot_height))); - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*(WCPMTPercentCoverage+WCPMTPercentCoverage2))/(10.0*std::max(WCPMTRadius,mPMT_vessel_radius))); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - //G4cout << "## Number of Horizontal PMT along diameter = " << WCBarrelNumPMTHorizontal << ", diameter = "<< WCIDDiameter << ", coverage = "<< WCPMTPercentCoverage << ", PMT radius = " << WCPMTRadius << ", test = " << WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius) << G4endl; - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - 5*std::max(WCPMTExposeHeight,mPMT_vessel_tot_height);//3.0*cm;//WCIDDiameter/2.0 - std::max(WCPMTRadius,mPMT_vessel_tot_height);//WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; -} -// B.Quilain, HK with hybrid configuration: mPMTs + B&L -// These are defaults that can be altered through the macros -void WCSimDetectorConstruction::SetHyperK_HybridFakeGeometry() -{ - WCDetectorName = "HyperK_HybridFake"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - mPMT_ID_PMT = "PMT3inchR12199_02"; //can be changed in macro through mPMT settings. - mPMT_OD_PMT = "PMT3inchR12199_02"; - WCSimPMTObject * PMT = CreatePMTObject(mPMT_ID_PMT, WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius();//Radius of a small PMT, not of a module - nID_PMTs = 1; - - //mPMT params go first because detector depends on it: - mPMT_vessel_cyl_height = 38.*CLHEP::mm; //option A, option B would be 277 mm - mPMT_vessel_radius_curv = 342.*CLHEP::mm; //needs to include the vessel thickness, as we construct from outside inwards. - mPMT_vessel_radius = 254.*CLHEP::mm; - dist_pmt_vessel = 2*CLHEP::mm; - orientation = PERPENDICULAR; - mPMT_outer_material = "G4_PLEXIGLASS"; - mPMT_inner_material = "Air"; // TODO: real air, hence update abs_length - mPMT_material_pmtAssembly = "SilGel"; - mPMT_outer_material_d = 10*CLHEP::mm; - - // Radius of cone at z=reflectorHeight - id_reflector_height = 6.53*CLHEP::mm; // for a radius of 7.25mm, for hex: 5.4mm (radius of 6mm) - id_reflector_z_offset = 4.8*CLHEP::mm; //from KM3Net CAD drawings - id_reflector_angle = 48.*CLHEP::deg; // Need to be remeasured for different PMT curvature - mPMT_pmt_openingAngle = 8.7*CLHEP::deg; // for hex: 8.5deg - G4double mPMT_vessel_tot_height = mPMT_vessel_radius + mPMT_vessel_cyl_height; - - //Box&Line - //WCDetectorName2 = "HyperK_HybridmPMT"; - WCIDCollectionName2 = WCDetectorName +"-glassFaceWCPMT2"; - WCSimPMTObject * PMT2 = CreatePMTObject("BoxandLine20inchHQE", WCIDCollectionName2); - WCPMTName2 = PMT2->GetPMTName(); - WCPMTExposeHeight2 = PMT2->GetExposeHeight(); - WCPMTRadius2 = PMT2->GetRadius(); - nID_PMTs2 = 1; - //WCPMTVesselCycHeight = 0; - //WCPMTVesselRadiusCurv = 0; - //WCPMTVesselRadius = 0; - //WCPMTDistPMTVessel = 0; - - //WCDetectorName2 = "HyperK_HybridmPMT"; - //WCIDCollectionName2 = WCDetectorName +"-glassFaceWCPMT"; - //WCSimPMTObject * PMT2 = CreatePMTObject("BoxandLine20inchHQE", WCIDCollectionName2); - //WCPMTName2 = PMT2->GetPMTName(); - //WCPMTExposeHeight2 = PMT2->GetExposeHeight(); - //WCPMTRadius2 = PMT2->GetRadius(); - //nID_PMTs = 1; - - - WCIDDiameter = 70.8*m; // = 74m - 2*(60cm ID wall + 1m OD) - WCIDHeight = 54.8*m; // = 60m - 2*(60cm ID wall + 2m OD) - WCBarrelPMTOffset = std::max(WCPMTRadius,mPMT_vessel_tot_height); //offset from vertical - WCIDVerticalPosition = 0.; - - WCPMTPercentCoverage = 20.0;//20.0;//40.0;//40.0; - WCPMTPercentCoverage2 = 0;//5.0;//20.0; - WCPMTperCellHorizontal = round((WCPMTPercentCoverage+WCPMTPercentCoverage2) / (WCPMTPercentCoverage2==0?(WCPMTPercentCoverage+WCPMTPercentCoverage2):WCPMTPercentCoverage2)); - WCPMTperCellVertical = WCPMTperCellHorizontal; - //WCPMTperCellHorizontal= ((int) WCPMTPercentCoverage2/(WCPMTPercentCoverage+WCPMTPercentCoverage2));//Should be in agreement with coverage of PMT 1 & 2: here, 4/5 PMTs are B&L + 1/5 mPMT - //WCPMTperCellVertical = ((int) WCPMTPercentCoverage2/(WCPMTPercentCoverage+WCPMTPercentCoverage2));//Should be in agreement with coverage of PMT 1 & 2: here, 4/5 PMTs are B&L + 1/5 mPMT - - //WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*(WCPMTPercentCoverage+WCPMTPercentCoverage2))/(10.0*std::max(WCPMTRadius,mPMT_vessel_tot_height))); - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*(WCPMTPercentCoverage+WCPMTPercentCoverage2))/(10.0*std::max(WCPMTRadius,mPMT_vessel_radius))); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - //G4cout << "## Number of Horizontal PMT along diameter = " << WCBarrelNumPMTHorizontal << ", diameter = "<< WCIDDiameter << ", coverage = "<< WCPMTPercentCoverage << ", PMT radius = " << WCPMTRadius << ", test = " << WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius) << G4endl; - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - 5*std::max(WCPMTExposeHeight,mPMT_vessel_tot_height);//3.0*cm;//WCIDDiameter/2.0 - std::max(WCPMTRadius,mPMT_vessel_tot_height);//WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; -} -void WCSimDetectorConstruction::SetEggShapedHyperKGeometry() -{ - WCDetectorName = "EggShapedHyperK"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCODCollectionName = WCDetectorName + "-glassFaceWCPMT_OD"; - WCSimPMTObject * PMT = CreatePMTObject("PMT20inch", WCIDCollectionName); - WCSimPMTObject * outerPMT = CreatePMTObject("PMT8inch", WCODCollectionName); - WCPMTName = PMT->GetPMTName(); - innerPMT_Expose = PMT->GetExposeHeight(); - innerPMT_Radius = PMT->GetRadius(); - waterTank_TopR = 32000.*mm; - waterTank_BotR = 30000.*mm; - waterTank_Height = 48000.*mm; - waterTank_UpperA = 8000.*mm; - waterTank_LowerB = 6000.*mm; - waterTank_Length = 49500.*mm; - - innerPMT_TopR = 29095.*mm; - innerPMT_BotR = 27095.*mm; - innerPMT_TopW = 12038.*mm; - innerPMT_BotW = 11004.*mm; - innerPMT_Height = 21095.*mm; - innerPMT_Rpitch = 990.*mm; - innerPMT_Apitch = 990.*mm; - - outerPMT_Name = outerPMT->GetPMTName(); - outerPMT_Expose = outerPMT->GetExposeHeight(); - outerPMT_Radius = outerPMT->GetRadius(); - outerPMT_TopR = innerPMT_TopR + 900.*mm; - outerPMT_BotR = innerPMT_BotR + 900.*mm; - outerPMT_TopW = 12394.*mm; - outerPMT_BotW = 11319.*mm; - outerPMT_Height = innerPMT_Height + 900.*mm; - outerPMT_TopRpitch = 3. * innerPMT_Rpitch * (outerPMT_TopR/innerPMT_TopR); - outerPMT_BotRpitch = 3. * innerPMT_Rpitch * (outerPMT_BotR/innerPMT_BotR); - outerPMT_Apitch = 2. * innerPMT_Apitch; - - blackSheetThickness = 20.*mm; - - innerPMT_TopN = 0; - innerPMT_BotN = 0; - - isEggShapedHyperK = true; // Tell DetectorConstruction to build egg-shaped HK geometry - - MatchWCSimAndEggShapedHyperK(); - InitSinglePMT(); -} - -void WCSimDetectorConstruction::SetEggShapedHyperKGeometry_withHPD() -{ - WCDetectorName = "EggShapedHyperK_withHPD"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCODCollectionName = WCDetectorName + "-glassFaceWCPMT_OD"; - WCSimPMTObject * PMT = CreatePMTObject("HPD20inchHQE", WCIDCollectionName); - WCSimPMTObject * outerPMT = CreatePMTObject("PMT8inch", WCODCollectionName); - WCPMTName = PMT->GetPMTName(); - innerPMT_Expose = PMT->GetExposeHeight(); - innerPMT_Radius = PMT->GetRadius(); - waterTank_TopR = 32000.*mm; - waterTank_BotR = 30000.*mm; - waterTank_Height = 48000.*mm; - waterTank_UpperA = 8000.*mm; - waterTank_LowerB = 6000.*mm; - waterTank_Length = 49500.*mm; - - - innerPMT_TopR = 29095.*mm; - innerPMT_BotR = 27095.*mm; - innerPMT_TopW = 12038.*mm; - innerPMT_BotW = 11004.*mm; - innerPMT_Height = 21095.*mm; - innerPMT_Rpitch = 990.*mm; - innerPMT_Apitch = 990.*mm; - - - outerPMT_Expose = outerPMT->GetExposeHeight(); - outerPMT_Radius = outerPMT->GetRadius(); - outerPMT_TopR = innerPMT_TopR + 900.*mm; - outerPMT_BotR = innerPMT_BotR + 900.*mm; - outerPMT_TopW = 12394.*mm; - outerPMT_BotW = 11319.*mm; - outerPMT_Height = innerPMT_Height + 900.*mm; - outerPMT_TopRpitch = 3. * innerPMT_Rpitch * (outerPMT_TopR/innerPMT_TopR); - outerPMT_BotRpitch = 3. * innerPMT_Rpitch * (outerPMT_BotR/innerPMT_BotR); - outerPMT_Apitch = 2. * innerPMT_Apitch; - - blackSheetThickness = 20.*mm; - - innerPMT_TopN = 0; - innerPMT_BotN = 0; - - isEggShapedHyperK = true; // Tell DetectorConstruction to build egg-shaped HK geometry - - MatchWCSimAndEggShapedHyperK(); - InitSinglePMT(); -} - - -/** - * Transfer egg-shaped HK variables needed elsewhere - * to their generic WC equivalents. - * This should be included in all egg-shaped HK configurations. - */ -void WCSimDetectorConstruction::MatchWCSimAndEggShapedHyperK() -{ - WCLength = waterTank_Length; - WCPosition = 0.; - WCPMTRadius = innerPMT_Radius; -} - - -void WCSimDetectorConstruction::SetNuPrismGeometry(G4String PMTType, G4double PMTCoverage, G4double detectorHeight, G4double detectorDiameter, G4double verticalPosition) -{ - WCDetectorName = "NuPRISM"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject(PMTType, WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - - WCIDHeight = detectorHeight; - WCIDDiameter = detectorDiameter; - WCIDVerticalPosition = verticalPosition; - - WCBarrelPMTOffset = WCPMTRadius; - WCPMTperCellHorizontal = 1.0; - WCPMTperCellVertical = 1.0; - WCPMTPercentCoverage = PMTCoverage; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage/100.0)/WCPMTRadius); - - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter)))/WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - InitSinglePMT(); -} - -void WCSimDetectorConstruction::SetDefaultNuPrismGeometry() -{ - SetNuPrismGeometry("PMT8inch", 40.0, 10*m, 8*m, 0*m); -} - -// NuPRISM with mPMTs -// These are defaults that can be altered through the macros -void WCSimDetectorConstruction::SetNuPrism_mPMTGeometry() -{ - WCDetectorName = "NuPRISM_mPMT"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - mPMT_ID_PMT = "PMT3inchR12199_02"; //can be changed in macro through mPMT settings. - mPMT_OD_PMT = "PMT3inchR12199_02"; - WCSimPMTObject * PMT = CreatePMTObject(mPMT_ID_PMT, WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - - //mPMT params go first because detector depends on it: - mPMT_vessel_cyl_height = 38.*CLHEP::mm; //option A, option B would be 277 mm - mPMT_vessel_radius_curv = 342.*CLHEP::mm; //needs to include the vessel thickness, as we construct from outside inwards. - mPMT_vessel_radius = 254.*CLHEP::mm; - dist_pmt_vessel = 2*CLHEP::mm; - orientation = PERPENDICULAR; - mPMT_outer_material = "G4_PLEXIGLASS"; - mPMT_inner_material = "Air"; // TODO: real air, hence update abs_length - mPMT_material_pmtAssembly = "SilGel"; - mPMT_outer_material_d = 10*CLHEP::mm; - - // Radius of cone at z=reflectorHeight - //id_reflector_height = 0.0*CLHEP::mm; // for a radius of 7.25mm, for hex: 5.4mm (radius of 6mm) - //B.Q - //id_reflector_height = mPMT_ID_reflector_height; - //id_reflector_angle = mPMT_ID_reflector_angle; - //id_reflector_height = 8.*CLHEP::mm; // for a radius of 7.25mm, for hex: 5.4mm (radius of 6mm) - id_reflector_height = 6.53*CLHEP::mm; // for a radius of 7.25mm, for hex: 5.4mm (radius of 6mm) - id_reflector_z_offset = 4.8*CLHEP::mm; //from KM3Net CAD drawings - id_reflector_angle = 48.*CLHEP::deg; // Need to be remeasured for different PMT curvature - //id_reflector_height = 3.*CLHEP::mm; // for a radius of 7.25mm, for hex: 5.4mm (radius of 6mm) - //id_reflector_angle = 0.*CLHEP::deg; // Need to be remeasured for different PMT curvature - //End B.Q - - mPMT_pmt_openingAngle = 8.7*CLHEP::deg; // for hex: 8.5deg - G4double mPMT_vessel_tot_height = mPMT_vessel_radius + mPMT_vessel_cyl_height; - - // parameters related to filling the ID mPMT - nID_PMTs = 19; - config_file = wcsimdir_path+"/mPMT-configfiles/mPMTconfig_19_nuPrism_3ring.txt"; // for smaller reflector, use: mPMTconfig_19_nuPrism.txt (hex) - - WCIDHeight = 10.42*CLHEP::m; - WCIDDiameter = 7.42*CLHEP::m; - WCIDVerticalPosition = 0.; - - WCBarrelPMTOffset = mPMT_vessel_tot_height; - WCPMTperCellHorizontal = 2.0; // 2 per phi - WCPMTperCellVertical = 1.0; - - // Numbers below are based on R.Henderson's 832 module tank design - WCBarrelNumPMTHorizontal = 40; - WCBarrelNRings = 16; - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); - // This is tuned to give a 6-5-5-4-3-1 config in one quarter of the cap - // And results in 832 mPMT modules as currently in Robert's Design - WCCapEdgeLimit = 3.3*m; - WCBlackSheetThickness = 2.0*cm; // deprecate soon. - WCAddGd = false; -} - -// Short version of NuPRISM with mPMTs: 6 m tall ID -// These are defaults that can be altered through the macros -void WCSimDetectorConstruction::SetNuPrismShort_mPMTGeometry() -{ - WCDetectorName = "NuPRISMShort_mPMT"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - mPMT_ID_PMT = "PMT3inchR12199_02"; //can be changed in macro through mPMT settings. - mPMT_OD_PMT = "PMT3inchR12199_02"; - WCSimPMTObject * PMT = CreatePMTObject(mPMT_ID_PMT, WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - - //mPMT params go first because detector depends on it: - vessel_cyl_height = 38.*CLHEP::mm; //option A, option B would be 277 mm - vessel_radius_curv = 342.*CLHEP::mm; //needs to include the vessel thickness, as we construct from outside inwards. - vessel_radius = 254.*CLHEP::mm; - dist_pmt_vessel = 2*CLHEP::mm; - orientation = PERPENDICULAR; - mPMT_outer_material = "G4_PLEXIGLASS"; - mPMT_inner_material = "Air"; // TODO: real air, hence update abs_length - mPMT_material_pmtAssembly = "SilGel"; - mPMT_outer_material_d = 10*CLHEP::mm; - - // Radius of cone at z=reflectorHeight - id_reflector_height = 6.53*CLHEP::mm; // for a radius of 7.25mm, for hex: 5.4mm (radius of 6mm) - id_reflector_z_offset = 4.8*CLHEP::mm; //from KM3Net CAD drawings - id_reflector_angle = 48.*CLHEP::deg; // Need to be remeasured for different PMT curvature - mPMT_pmt_openingAngle = 8.7*CLHEP::deg; // for hex: 8.5deg - G4double vessel_tot_height = vessel_radius + vessel_cyl_height; - - // parameters related to filling the ID mPMT - nID_PMTs = 19; - config_file = wcsimdir_path+"/mPMT-configfiles/mPMTconfig_19_nuPrism_3ring.txt"; // for smaller reflector, use: mPMTconfig_19_nuPrism.txt (hex) - - WCIDHeight = 6.0*CLHEP::m; - WCIDDiameter = 8.0*CLHEP::m; - WCIDVerticalPosition = 0.; - - WCBarrelPMTOffset = vessel_tot_height; - WCPMTperCellHorizontal = 2.0; // 2 per phi - WCPMTperCellVertical = 1.0; - - // Numbers below are based on R.Henderson's 832 module tank design - WCBarrelNumPMTHorizontal = 40; - WCBarrelNRings = 9; // Scaled from 10.42 m tall detector. - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); - // This is tuned to give a 6-5-5-4-3-1 config in one quarter of the cap - // And results in 832 mPMT modules as currently in Robert's Design - WCCapEdgeLimit = 3.3*m; - WCBlackSheetThickness = 2.0*cm; // deprecate soon. - WCAddGd = false; -} - -void WCSimDetectorConstruction::SetTestSinglemPMTGeometry() -{ - WCDetectorName = "TestSinglemPMT"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - - mPMT_ID_PMT = "PMT3inchR12199_02"; //"BoxandLine20inchHQE";// (combine with nPMT=1); - mPMT_OD_PMT = "PMT8inch"; // Only used for the unique string name of mPMT for now! - - WCSimPMTObject * PMT = CreatePMTObject(mPMT_ID_PMT, WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = PMT->GetRadius(); - - //mPMT params - vessel_cyl_height = 38.*CLHEP::mm; - vessel_radius_curv = 300.*CLHEP::mm; - vessel_radius = 254.*CLHEP::mm; - dist_pmt_vessel = 2*CLHEP::mm; - orientation = PERPENDICULAR; - mPMT_outer_material = "G4_PLEXIGLASS"; - mPMT_inner_material = "SilGel"; - mPMT_material_pmtAssembly = "SilGel"; - mPMT_outer_material_d = 10.*CLHEP::mm; - // Radius of cone at z=reflectorHeight - id_reflector_height = 10*CLHEP::mm; //10. > previous 7mm (deprecated number from JINST) - id_reflector_z_offset = 4.8*CLHEP::mm; //from KM3Net CAD drawings - id_reflector_angle = 48*CLHEP::pi/180.*CLHEP::rad; // Based on KM3Net reflector specs - mPMT_pmt_openingAngle = 8.7*CLHEP::deg; - - // BarrelPMTOffset/WCCapEdgeLimit needs PMT/mPMT height - G4double vessel_tot_height = vessel_radius + vessel_cyl_height; - // parameters related to filling the ID mPMT - nID_PMTs = 33; - config_file = wcsimdir_path+"/mPMT-configfiles/mPMTconfig_33_13_1.txt"; - - WCIDDiameter = 10.0*m; - WCIDHeight = 30.0*m; - WCBarrelPMTOffset = vessel_tot_height; - WCBarrelNumPMTHorizontal = 3; - WCBarrelNRings = 3; - WCPMTperCellHorizontal = 1; - WCPMTperCellVertical = 1; - WCPMTPercentCoverage = 10.; - WCCapEdgeLimit = WCIDDiameter/2.0 - vessel_tot_height; - WCCapPMTSpacing = WCIDDiameter*2.0; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; -} - -// Note: using vessel_radius (mPMT radius) instead of PMT radius for detector construction. Important! -void WCSimDetectorConstruction::Cylinder_60x74_3inchmPMT_14perCent() -{ - WCDetectorName = "Cylinder_60x74_3inchmPMT_14perCent"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - mPMT_ID_PMT = "PMT3inchR12199_02"; - mPMT_OD_PMT = "PMT8inch"; - WCSimPMTObject * PMT = CreatePMTObject(mPMT_ID_PMT, WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = 0.254*m;//outer_module_radius;//PMT-T>GetRadius(); //Actually, this name is very misleading, because it should be moduleRadius (eg. PMT + acrylic), this is used in two places in WCConstructCylinder - - - - //mPMT params go first because detector depends on it: - vessel_cyl_height = 0.*CLHEP::mm; // the 50mm should exist only for OD and extends behind the iD wall - vessel_radius_curv = 270.*CLHEP::mm; - vessel_radius = 270.*CLHEP::mm; - dist_pmt_vessel = 5*CLHEP::mm; - orientation = PERPENDICULAR; - mPMT_outer_material = "G4_PLEXIGLASS"; - mPMT_inner_material = "SilGel"; - mPMT_outer_material_d = 0.03*2* vessel_radius *CLHEP::mm; //using NEMO article safety margin of 0.03 for t/D - // Radius of cone at z=reflectorHeight - id_reflector_height = 7.5*CLHEP::mm; //7.5mm from KM3Net JINST - id_reflector_z_offset = 1.1*CLHEP::mm; //from KM3Net CAD drawings - id_reflector_angle = CLHEP::pi/4*CLHEP::rad; // Based on KM3Net JINST: 45 deg wrt normal, so 7.5mm xtra - G4double vessel_tot_height = vessel_radius + vessel_cyl_height; - - // parameters related to filling the ID mPMT - nID_PMTs = 33; - config_file = wcsimdir_path+"/mPMT-configfiles/mPMTconfig_33_13_1.txt"; - - // Note: in principal WCPMTRadius and WCCapEdgeLimit could also use outer_module_radius - // but then total n mPMT modules change slightly. This configuration wants to have NO overlaps - // and use the same number of mPMT modules as 20in PMTs, so use same settings. - // Most important one is WCBarrelNumPMTHorizontal - // WCBarrelPMTOffset does not change total No. of modules, so use actual outer_module_radius. - WCIDDiameter = 74.0*m; - WCIDHeight = 60.0*m; - WCBarrelPMTOffset = vessel_tot_height; //mPMT cylinder Radius //WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 13.51; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*0.254*m)); //Used same radius as 20in to have same granularity in mPMTs (28% difference otherwise - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - 0.254*m;//outer_module_radius;//WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - -} - -// Note: using vessel_radius (mPMT radius) instead of PMT radius for detector construction. Important! -void WCSimDetectorConstruction::Cylinder_60x74_3inchmPMT_40perCent() -{ - WCDetectorName = "Cylinder_60x74_3inchmPMT_40perCent"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - mPMT_ID_PMT = "PMT3inchR12199_02"; - mPMT_OD_PMT = "PMT8inch"; - WCSimPMTObject * PMT = CreatePMTObject(mPMT_ID_PMT, WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = 0.254*m; - - //mPMT params: - vessel_cyl_height = 0.*CLHEP::mm; // the 50mm should exist only for OD and extends behind the iD wall - vessel_radius_curv = 280.*CLHEP::mm; - vessel_radius = 280.*CLHEP::mm; - dist_pmt_vessel = 5*CLHEP::mm; - orientation = PERPENDICULAR; - mPMT_outer_material = "G4_PLEXIGLASS"; - mPMT_inner_material = "SilGel"; - mPMT_outer_material_d = 0.03*2*vessel_radius*CLHEP::mm; //using NEMO article safety margin of 0.03 for t/D - // Radius of cone at z=reflectorHeight - id_reflector_height = 7.5*CLHEP::mm; //7.5mm from KM3Net JINST - id_reflector_z_offset = 1.1*CLHEP::mm; //from KM3Net CAD drawings - id_reflector_angle = CLHEP::pi/4*CLHEP::rad; // Based on KM3Net JINST: 45 deg wrt normal, so 7.5mm xtra - G4double vessel_tot_height = vessel_radius + vessel_cyl_height; - - // parameters related to filling the ID mPMT - nID_PMTs = 34; - config_file = wcsimdir_path+"/mPMT-configfiles/mPMTconfig_34_22.5_2.txt"; - - - WCIDDiameter = 74.0*m; - WCIDHeight = 60.0*m; - WCBarrelPMTOffset = vessel_tot_height; //mPMT cylinder Radius //WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 40.0; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*0.254*m));//WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - 0.254*m;//WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - -} - -// Uniform distribution of 3" PMTs -void WCSimDetectorConstruction::Cylinder_60x74_3inch_14perCent() -{ - WCDetectorName = "Cylinder_60x74_3inch_14perCent"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("PMT3inchR12199_02", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = .04*m; //PMT->GetRadius(); - WCIDDiameter = 74.0*m; - WCIDHeight = 60.0*m; - WCBarrelPMTOffset = WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 13.51; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - InitSinglePMT(); -} - -void WCSimDetectorConstruction::Cylinder_60x74_3inch_40perCent() -{ - WCDetectorName = "Cylinder_60x74_3inch_40perCent"; - WCIDCollectionName = WCDetectorName +"-glassFaceWCPMT"; - WCSimPMTObject * PMT = CreatePMTObject("PMT3inchR12199_02", WCIDCollectionName); - WCPMTName = PMT->GetPMTName(); - WCPMTExposeHeight = PMT->GetExposeHeight(); - WCPMTRadius = .04*m; //PMT->GetRadius(); - WCIDDiameter = 74.0*m; - WCIDHeight = 60.0*m; - WCBarrelPMTOffset = WCPMTRadius; //offset from vertical - WCPMTperCellHorizontal= 4; - WCPMTperCellVertical = 3; - WCPMTPercentCoverage = 40.; - WCBarrelNumPMTHorizontal = round(WCIDDiameter*sqrt(pi*WCPMTPercentCoverage)/(10.0*WCPMTRadius)); - WCBarrelNRings = round(((WCBarrelNumPMTHorizontal*((WCIDHeight-2*WCBarrelPMTOffset)/(pi*WCIDDiameter))) - /WCPMTperCellVertical)); - WCCapPMTSpacing = (pi*WCIDDiameter/WCBarrelNumPMTHorizontal); // distance between centers of top and bottom pmts - WCCapEdgeLimit = WCIDDiameter/2.0 - WCPMTRadius; - WCBlackSheetThickness = 2.0*cm; - WCAddGd = false; - - InitSinglePMT(); -} - - - -void WCSimDetectorConstruction::InitSinglePMT(){ - - vessel_cyl_height = 0.*mm; - vessel_radius_curv = 0.1*mm; - vessel_radius = 0.1*mm; - dist_pmt_vessel = 0.*mm; - orientation = PERPENDICULAR; - mPMT_ID_PMT = ""; - mPMT_OD_PMT = ""; - mPMT_outer_material = ""; - mPMT_inner_material = ""; - mPMT_outer_material_d = 0.*CLHEP::mm; - id_reflector_height = 0.*CLHEP::mm; - id_reflector_z_offset = 0.*CLHEP::mm; - id_reflector_angle = 0.*CLHEP::rad; - nID_PMTs = 1; - config_file = ""; - -} diff --git a/libWCSIM/src/WCSimDetectorConstruction.cc b/libWCSIM/src/WCSimDetectorConstruction.cc deleted file mode 100644 index 55702dd..0000000 --- a/libWCSIM/src/WCSimDetectorConstruction.cc +++ /dev/null @@ -1,435 +0,0 @@ -#include "WCSimDetectorConstruction.hh" -#include "WCSimDetectorMessenger.hh" -#include "WCSimTuningParameters.hh" - -#include "G4Element.hh" -#include "G4Box.hh" -#include "G4LogicalVolume.hh" -#include "G4VPhysicalVolume.hh" -#include "G4PVPlacement.hh" -#include "G4ThreeVector.hh" -#include "globals.hh" -#include "G4VisAttributes.hh" - -#include "G4RunManager.hh" -#include "G4PhysicalVolumeStore.hh" -#include "G4LogicalVolumeStore.hh" -#include "G4SolidStore.hh" - -#include "G4PhysicalConstants.hh" -#include "G4SystemOfUnits.hh" - -std::map WCSimDetectorConstruction::tubeIDMap; -std::map > WCSimDetectorConstruction::mPMTIDMap; -std::map WCSimDetectorConstruction::tubeIDMap2; -std::map > WCSimDetectorConstruction::mPMTIDMap2; -//std::map WCSimDetectorConstruction::tubeCylLocation; -//hash_map > -//WCSimDetectorConstruction::tubeLocationMap_old; //deprecated !! - -// std::hash is default hash function actually (http://en.cppreference.com/w/cpp/utility/hash) -// with operator() already properly defined. -std::unordered_map > -WCSimDetectorConstruction::tubeLocationMap; -std::unordered_map > -WCSimDetectorConstruction::tubeLocationMap2; - -WCSimDetectorConstruction::WCSimDetectorConstruction(G4int DetConfig,WCSimTuningParameters* WCSimTuningPars):WCSimTuningParams(WCSimTuningPars) -{ - - // Decide if (only for the case of !1kT detector) should be upright or horizontal - isUpright = false; - isEggShapedHyperK = false; - isNuPrism = false; - - debugMode = false; - - myConfiguration = DetConfig; - - //----------------------------------------------------- - // Create Materials - //----------------------------------------------------- - - ConstructMaterials(); - - //----------------------------------------------------- - // Initialize things related to the tubeID - //----------------------------------------------------- - - WCSimDetectorConstruction::tubeIDMap.clear(); - WCSimDetectorConstruction::mPMTIDMap.clear(); - WCSimDetectorConstruction::tubeLocationMap.clear(); - - WCSimDetectorConstruction::tubeIDMap2.clear(); - WCSimDetectorConstruction::mPMTIDMap2.clear(); - WCSimDetectorConstruction::tubeLocationMap2.clear(); - - WCSimDetectorConstruction::PMTLogicalVolumes.clear(); - - totalNumPMTs = 0; - totalNumPMTs2 = 0; - WCPMTExposeHeight= 0.; - WCPMTExposeHeight2= 0.; - - //--------------------------------------------------- - // Need to define defaults for all mPMT parameters - // defaults are chosen that they are valid for a SK detector - //-------------------------------------------------- - vessel_cyl_height = 0.1*mm; // LATER: when tested, default WITH PMT cover! - vessel_radius_curv = 0.1*mm; - vessel_radius = 0.1*mm; - - mPMT_vessel_cyl_height = 0.1*mm; // LATER: when tested, default WITH PMT cover! - mPMT_vessel_radius_curv = 0.1*mm; - mPMT_vessel_radius = 0.1*mm; - dist_pmt_vessel = 0.*mm; - orientation = PERPENDICULAR; - mPMT_ID_PMT = "PMT3inchR12199_02"; - mPMT_OD_PMT = ""; - mPMT_outer_material = "Water"; - mPMT_inner_material = ""; - mPMT_outer_material_d = 0.*CLHEP::mm; - // Radius of cone at z=reflectorHeight - id_reflector_height = 0.*CLHEP::mm; - id_reflector_z_offset = 0.*CLHEP::mm; - id_reflector_angle = 0.*CLHEP::deg; - // parameters related to filling the ID mPMT - nID_PMTs = 1; //per mPMT - nID_PMTs2 = 1; //per mPMT - config_file = ""; - fix_nModules = false; - mPMT_material_pmtAssembly = "Water"; - mPMT_pmt_openingAngle = 0.*CLHEP::deg; - - - - //SetTestSinglemPMTGeometry(); - SetSuperKGeometry(); - //SetHyperKGeometry(); - - //----------------------------------------------------- - // Set whether or not Pi0-specific info is saved - //----------------------------------------------------- - - SavePi0Info(false); - - //----------------------------------------------------- - // Set whether or not neutron capture info is saved - //----------------------------------------------------- - - SaveCaptureInfo(false); - - //----------------------------------------------------- - // Set the default method for implementing the PMT QE - //----------------------------------------------------- - SetPMT_QE_Method(1); - - //default is to use collection efficiency - SetPMT_Coll_Eff(1); - // set default visualizer to OGLSX - SetVis_Choice("OGLSX"); - - - //----------------------------------------------------- - // Make the detector messenger to allow changing geometry - //----------------------------------------------------- - - messenger = new WCSimDetectorMessenger(this); - - - // Get WCSIMDIR - wcsimdir_path = std::getenv("WCSIMDIR"); -} - -#include "G4GeometryManager.hh" -#include "G4LogicalBorderSurface.hh" -#include "G4LogicalSkinSurface.hh" - -void WCSimDetectorConstruction::UpdateGeometry() -{ - - - G4bool geomChanged = true; - G4RunManager::GetRunManager()->DefineWorldVolume(Construct(), geomChanged); - // ToDo: need some error catching here for NULL Construct() cases - -} - - - -WCSimDetectorConstruction::~WCSimDetectorConstruction(){ - for (unsigned int i=0;iGet("Settings"); -// if (!fSettingsTree){ -// G4ThreeVector trans; -// return trans; -// } -// -// float NuBeamAng; -// float NuIdfdPos[3]; -// for(int i = 0; i < 3; ++i) NuIdfdPos[i] = 0; -// fSettingsTree->SetBranchAddress("NuIdfdPos",NuIdfdPos); -// fSettingsTree->SetBranchAddress("NuBeamAng",&NuBeamAng); -// -// fSettingsTree->GetEntry(0); -// -// G4ThreeVector trans(0, (NuIdfdPos[1] + TMath::Tan(NuBeamAng)*600.0), NuIdfdPos[2] + 600.0); -// -// inFile->Close(); -// fInputSettingsFilename = ""; -// fSettingsTree = NULL; -// -// return trans; -// -//} - -G4VPhysicalVolume* WCSimDetectorConstruction::Construct() -{ - G4GeometryManager::GetInstance()->OpenGeometry(); - - G4PhysicalVolumeStore::GetInstance()->Clean(); - G4LogicalVolumeStore::GetInstance()->Clean(); - G4SolidStore::GetInstance()->Clean(); - G4LogicalBorderSurface::CleanSurfaceTable(); - G4LogicalSkinSurface::CleanSurfaceTable(); - WCSimDetectorConstruction::PMTLogicalVolumes.clear(); - //TF: for new mPMT (or make this into a function?) - vNiC.clear(); - vAlpha.clear(); - vCircle.clear(); - vAzimOffset.clear(); - - totalNumPMTs = 0; - totalNumPMTs2 = 0; - - //----------------------------------------------------- - // Create Logical Volumes - //----------------------------------------------------- - - // First create the logical volumes of the sub detectors. After they are - // created their size will be used to make the world volume. - // Note the order is important because they rearrange themselves depending - // on their size and detector ordering. - - G4LogicalVolume* logicWCBox; - - // Select between HyperK and cylinder - if (isEggShapedHyperK) logicWCBox = ConstructEggShapedHyperK(); - else logicWCBox = ConstructCylinder(); - - if(!logicWCBox){ - G4cerr << "Something went wrong in ConstructCylinder" << G4endl; - return NULL; - } - G4cout << " WCLength = " << WCLength/m << " m"<< G4endl; - - //------------------------------- - - // Now make the detector Hall. The lengths of the subdectors - // were set above. - - G4ThreeVector position(0,0,0);// = GetTranslationFromSettings(); - - if(isNuPrism) position.setY(WCIDVerticalPosition); - - std::cout << "position Y = " << position.y() << std::endl; - - - G4double expHallLength = 3.*WCLength; //jl145 - extra space to simulate cosmic muons more easily - - G4cout << " expHallLength = " << expHallLength / m << G4endl; - G4double expHallHalfLength = 0.5*expHallLength; - - G4Box* solidExpHall = new G4Box("expHall", - expHallHalfLength + fabs(position.x()), - expHallHalfLength + fabs(position.y()), - expHallHalfLength + fabs(position.z())); - - G4LogicalVolume* logicExpHall = - new G4LogicalVolume(solidExpHall, - G4Material::GetMaterial("Vacuum"), - "expHall", - 0,0,0); - - // Now set the visualization attributes of the logical volumes. - - // logicWCBox->SetVisAttributes(G4VisAttributes::Invisible); - logicExpHall->SetVisAttributes(G4VisAttributes::Invisible); - - //----------------------------------------------------- - // Create and place the physical Volumes - //----------------------------------------------------- - - // Experimental Hall - G4VPhysicalVolume* physiExpHall = - new G4PVPlacement(0,G4ThreeVector(), - logicExpHall, - "expHall", - 0,false,0,true); - - // Water Cherenkov Detector (WC) mother volume - // WC Box, nice to turn on for x and y views to provide a frame: - - G4RotationMatrix* rotationMatrix = new G4RotationMatrix; - G4ThreeVector genPosition = G4ThreeVector(0., 0., WCPosition); - - if(isNuPrism){ - rotationMatrix->rotateX(90.*deg); - rotationMatrix->rotateY(0.*deg); - rotationMatrix->rotateZ(0.*deg); - genPosition.setX(position.x()); - genPosition.setY(position.y()); - genPosition.setZ(position.z()); - } - - G4Transform3D transform(*rotationMatrix, genPosition); - - G4VPhysicalVolume* physiWCBox = - new G4PVPlacement(transform, - logicWCBox, - "WCBox", - logicExpHall, - false, - 0); - - // Reset the tubeID and tubeLocation maps before refilling them - tubeIDMap.clear(); - mPMTIDMap.clear(); - tubeLocationMap.clear(); - - tubeIDMap2.clear(); - mPMTIDMap2.clear(); - tubeLocationMap2.clear(); - - - // Traverse and print the geometry Tree - - // TraverseReplicas(physiWCBox, 0, G4Transform3D(), - // &WCSimDetectorConstruction::PrintGeometryTree) ; - - TraverseReplicas(physiWCBox, 0, G4Transform3D(), - &WCSimDetectorConstruction::DescribeAndRegisterPMT) ; - - - TraverseReplicas(physiWCBox, 0, G4Transform3D(), - &WCSimDetectorConstruction::GetWCGeom) ; - - DumpGeometryTableToFile(); - - - - - // Return the pointer to the physical experimental hall - return physiExpHall; -} - -WCSimPMTObject *WCSimDetectorConstruction::CreatePMTObject(G4String PMTType, G4String CollectionName) -{ - if (PMTType == "PMT20inch"){ - WCSimPMTObject* PMT = new PMT20inch; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "PMT8inch"){ - WCSimPMTObject* PMT = new PMT8inch; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "PMT5inch"){ - WCSimPMTObject* PMT = new PMT5inch; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "PMT3inch"){ - WCSimPMTObject* PMT = new PMT3inch; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "PMT3inchGT"){ - WCSimPMTObject* PMT = new PMT3inchGT; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "PMT3inchR12199_02"){ - WCSimPMTObject* PMT = new PMT3inchR12199_02; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "PMT3inchR14374"){ - WCSimPMTObject* PMT = new PMT3inchR14374; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "PMT10inch"){ - WCSimPMTObject* PMT = new PMT10inch; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "PMT10inchHQE"){ - WCSimPMTObject* PMT = new PMT10inchHQE; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "PMT12inchHQE"){ - WCSimPMTObject* PMT = new PMT12inchHQE; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "HPD20inchHQE"){ - WCSimPMTObject* PMT = new HPD20inchHQE; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "HPD12inchHQE"){ - WCSimPMTObject* PMT = new HPD12inchHQE; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "BoxandLine20inchHQE"){ - WCSimPMTObject* PMT = new BoxandLine20inchHQE; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "BoxandLine12inchHQE"){ - WCSimPMTObject* PMT = new BoxandLine12inchHQE; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "PMT4inchR12199_02"){ - WCSimPMTObject* PMT = new PMT4inchR12199_02; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - else if (PMTType == "PMT5inchR12199_02"){ - WCSimPMTObject* PMT = new PMT5inchR12199_02; - WCSimDetectorConstruction::SetPMTPointer(PMT, CollectionName); - return PMT; - } - - else { G4cout << PMTType << " is not a recognized PMT Type. Exiting WCSim." << G4endl; exit(1);} -} - -void WCSimDetectorConstruction::SaveOptionsToOutput(WCSimRootOptions * wcopt) -{ - wcopt->SetDetectorName(WCDetectorName); - wcopt->SetSavePi0(pi0Info_isSaved); - wcopt->SetPMTQEMethod(PMT_QE_Method); - wcopt->SetPMTCollEff(PMT_Coll_Eff); -} diff --git a/libWCSIM/src/WCSimDetectorMessenger.cc b/libWCSIM/src/WCSimDetectorMessenger.cc deleted file mode 100644 index c4fbbb3..0000000 --- a/libWCSIM/src/WCSimDetectorMessenger.cc +++ /dev/null @@ -1,701 +0,0 @@ -#include "WCSimDetectorMessenger.hh" - -#include "WCSimDetectorConstruction.hh" -#include "G4UIdirectory.hh" -#include "G4UIcommand.hh" -#include "G4UIparameter.hh" - -WCSimDetectorMessenger::WCSimDetectorMessenger(WCSimDetectorConstruction* WCSimDet) -:WCSimDetector(WCSimDet) -{ - WCSimDir = new G4UIdirectory("/WCSim/"); - WCSimDir->SetGuidance("Commands to change the geometry of the simulation"); - - - PMTConfig = new G4UIcmdWithAString("/WCSim/WCgeom",this); - PMTConfig->SetGuidance("Set the geometry configuration for the WC."); - PMTConfig->SetGuidance("Available options are:\n" - "SuperK\n" - "SuperK_20inchPMT_20perCent\n" - "SuperK_20inchBandL_20perCent\n" - "SuperK_12inchBandL_15perCent\n" - "SuperK_20inchBandL_14perCent\n" - "Cylinder_60x74_20inchBandL_14perCent\n" - "Cylinder_60x74_20inchBandL_40perCent\n" - "Cylinder_12inchHPD_15perCent\n" - "HyperK\n" - "HyperK_3inch\n" - "HyperK_8inch\n" - "HyperK_10inch\n" - "HyperK_SKPMT\n" - "HyperK_mPMT\n" - "HyperK_HybridmPMT\n" - "HyperK_HybridmPMT10PC\n" - "HyperK_HybridFake\n" - "EggShapedHyperK\n" - "EggShapedHyperK_withHPD\n" - "nuPRISM\n" - "nuPRISM_mPMT\n" - "nuPRISMShort_mPMT\n" - "Cylinder_60x74_3inchmPMT_14perCent\n" - "Cylinder_60x74_3inchmPMT_40perCent\n" - "Cylinder_60x74_3inch_14perCent\n" - "Cylinder_60x74_3inch_40perCent\n" - "TestSinglemPMT\n" - ); - PMTConfig->SetParameterName("PMTConfig", false); - PMTConfig->SetCandidates("SuperK " - "SuperK_20inchPMT_20perCent " - "SuperK_20inchBandL_20perCent " - "SuperK_12inchBandL_15perCent " - "SuperK_20inchBandL_14perCent " - "Cylinder_60x74_20inchBandL_14perCent\n" - "Cylinder_60x74_20inchBandL_40perCent\n" - "Cylinder_12inchHPD_15perCent " - "HyperK " - "HyperK_3inch " - "HyperK_8inch " - "HyperK_10inch " - "HyperK_SKPMT " - "HyperK_mPMT " - "HyperK_HybridmPMT " - "HyperK_HybridmPMT10PC " - "HyperK_HybridFake " - "EggShapedHyperK " - "EggShapedHyperK_withHPD " - "nuPRISM " - "nuPRISM_mPMT " - "nuPRISMShort_mPMT " - "Cylinder_60x74_3inchmPMT_14perCent " - "Cylinder_60x74_3inchmPMT_40perCent " - "Cylinder_60x74_3inch_14perCent " - "Cylinder_60x74_3inch_40perCent " - "TestSinglemPMT\n" - ); - PMTConfig->AvailableForStates(G4State_PreInit, G4State_Idle); - - WCVisChoice = new G4UIcmdWithAString("/WCSim/WCVisChoice",this); - WCVisChoice->SetGuidance("Set the visualization style for the WC."); - WCVisChoice->SetGuidance("Available options are:\n" - "OGLSX\n" - "RayTracer\n" - ); - WCVisChoice->SetParameterName("WCVisChoice", false); - WCVisChoice->SetCandidates("OGLSX " - "RayTracer " - ); - WCVisChoice->AvailableForStates(G4State_PreInit, G4State_Idle); - - DopedWater = new G4UIcmdWithABool("/WCSim/DopedWater", this); - DopedWater->SetGuidance("Set whether water is doped with Gadolinium"); - DopedWater->SetParameterName("DopedWater", false); - DopedWater->AvailableForStates(G4State_PreInit, G4State_Idle); - - DopingConcentration = new G4UIcmdWithADouble("/WCSim/DopingConcentration", this); - DopingConcentration->SetGuidance("Set percentage concentration Gadolinium doping"); - DopingConcentration->SetParameterName("DopingConcentration", false); - DopingConcentration->AvailableForStates(G4State_PreInit, G4State_Idle); - - PMTSize = new G4UIcmdWithAString("/WCSim/WCPMTsize",this); - PMTSize->SetGuidance("Set alternate PMT size for the WC (Must be entered after geometry details are set)."); - PMTSize->SetGuidance("Available options 20inch 10inch"); - PMTSize->SetParameterName("PMTSize", false); - PMTSize->SetCandidates("20inch 10inch"); - PMTSize->AvailableForStates(G4State_PreInit, G4State_Idle); - - PMTSize2 = new G4UIcmdWithAString("/WCSim/WCPMTsize2",this);//B.Q, for the second PMT type - PMTSize2->SetGuidance("Set alternate PMT size for the WC (Must be entered after geometry details are set)."); - PMTSize2->SetGuidance("Available options 20inch 10inch"); - PMTSize2->SetParameterName("PMTSize2", false); - PMTSize2->SetCandidates("20inch 10inch"); - PMTSize2->AvailableForStates(G4State_PreInit, G4State_Idle); - - SavePi0 = new G4UIcmdWithAString("/WCSim/SavePi0", this); - SavePi0->SetGuidance("true or false"); - SavePi0->SetParameterName("SavePi0",false); - SavePi0->SetCandidates("true false"); - SavePi0->AvailableForStates(G4State_PreInit, G4State_Idle); - - SaveCapture = new G4UIcmdWithAString("/WCSim/SaveCapture", this); - SaveCapture->SetGuidance("true or false"); - SaveCapture->SetParameterName("SaveCapture",false); - SaveCapture->SetCandidates("true false"); - SaveCapture->AvailableForStates(G4State_PreInit, G4State_Idle); - - - PMTQEMethod = new G4UIcmdWithAString("/WCSim/PMTQEMethod", this); - PMTQEMethod->SetGuidance("Set the PMT configuration."); - PMTQEMethod->SetGuidance("Available options are:\n" - "Stacking_Only\n" - "Stacking_And_SensitiveDetector\n" - "SensitiveDetector_Only\n" - "DoNotApplyQE"); - - PMTQEMethod->SetParameterName("PMTQEMethod", false); - PMTQEMethod->SetCandidates("Stacking_Only " - "Stacking_And_SensitiveDetector " - "SensitiveDetector_Only " - "DoNotApplyQE "); - PMTQEMethod->AvailableForStates(G4State_PreInit, G4State_Idle); - - PMTCollEff = new G4UIcmdWithAString("/WCSim/PMTCollEff", this); - PMTCollEff->SetGuidance("Set the PMT configuration."); - PMTCollEff->SetGuidance("Available options are:\n" - "on\n" - "off\n"); - PMTCollEff->SetParameterName("PMTCollEff", false); - PMTCollEff->SetCandidates("on " - "off "); - PMTCollEff->AvailableForStates(G4State_PreInit, G4State_Idle); - - waterTank_Length = new G4UIcmdWithADoubleAndUnit("/WCSim/EggShapedHyperK/waterTank_Length", this); - waterTank_Length->SetGuidance("Set the Length of the egg-shaped HyperK detector (unit: mm cm m)."); - waterTank_Length->SetParameterName("waterTank_length", true); - waterTank_Length->SetDefaultValue(49500.); - waterTank_Length->SetUnitCategory("Length"); - waterTank_Length->SetDefaultUnit("mm"); - waterTank_Length->SetUnitCandidates("mm cm m"); - - WCConstruct = new G4UIcmdWithoutParameter("/WCSim/Construct", this); - WCConstruct->SetGuidance("Update detector construction with new settings."); - - // Params: - Cylinder height and radius of mPMT: DONE - // - Type of ID PMT (should be enum) and Type of OD PMT: Need to reorganize the PMTpart. - // - Params related to filling PMT: txt filename - // - Orientation of mPMT wrt wall : horizontal/vertical/perp - // - Reflector OFF/Winston/other - // - material of mPMT wall: water/glass/acrylic/ - // - acrylic cover ON/OFF - - - mPMTDir = new G4UIdirectory("/WCSim/mPMT/"); - mPMTDir->SetGuidance("Commands to change the properties of the multiPMT."); - - - mPMT_CylHeight = new G4UIcmdWithADoubleAndUnit("/WCSim/mPMT/VesselCylHeight",this); - mPMT_CylHeight->SetGuidance("Set height of cylinder of pressure vessel of multiPMT object."); - mPMT_CylHeight->SetParameterName("CylHeight", true); - mPMT_CylHeight->SetDefaultValue(50.); //mDOM (PINGU), 0 for KM3Net - mPMT_CylHeight->SetUnitCategory("Length"); - mPMT_CylHeight->SetDefaultUnit("mm"); - mPMT_CylHeight->SetUnitCandidates("mm cm m"); - - mPMT_CylRadius = new G4UIcmdWithADoubleAndUnit("/WCSim/mPMT/VesselRadius",this); - mPMT_CylRadius->SetGuidance("Set vessel radius of multiPMT object."); - mPMT_CylRadius->SetParameterName("CylRadius", true); - mPMT_CylRadius->SetDefaultValue(325.); - mPMT_CylRadius->SetUnitCategory("Length"); - mPMT_CylRadius->SetDefaultUnit("mm"); - mPMT_CylRadius->SetUnitCandidates("mm cm m"); - - mPMT_CylRadiusCurv = new G4UIcmdWithADoubleAndUnit("/WCSim/mPMT/VesselRadiusCurv",this); - mPMT_CylRadiusCurv->SetGuidance("Set vessel radius of curvature of multiPMT object."); - mPMT_CylRadiusCurv->SetParameterName("CylRadiusCurv", true); - mPMT_CylRadiusCurv->SetDefaultValue(325.); - mPMT_CylRadiusCurv->SetUnitCategory("Length"); - mPMT_CylRadiusCurv->SetDefaultUnit("mm"); - mPMT_CylRadiusCurv->SetUnitCandidates("mm cm m"); - - mPMT_orientation = new G4UIcmdWithAString("/WCSim/mPMT/orientation",this); - mPMT_orientation->SetGuidance("Set orientation of multiPMT cylinder wrt wall."); - mPMT_orientation->SetParameterName("orientation", true); - mPMT_orientation->SetCandidates("Horizontal " - "Vertical " - "Perpendicular " - ); - - // untested/unfinished - // This makes it possible to overwrite the DetectorConfigs - - mPMT_PMTtype_inner = new G4UIcmdWithAString("/WCSim/mPMT/PMTtype_inner",this); - mPMT_PMTtype_inner->SetGuidance("Set type of PMT for ID."); - mPMT_PMTtype_inner->SetParameterName("PMTtype_inner", true); - mPMT_PMTtype_inner->SetCandidates("PMT20inch " - "PMT8inch " - "PMT10inch " - "PMT10inchHQE " - "PMT12inchHQE " - "HPD20inchHQE " - "HPD12inchHQE " - "BoxandLine20inchHQE " - "BoxandLine12inchHQE " - "PMT3inchR12199_02 " - "PMT3inchR14374 " - "PMT4inchR12199_02 " - "PMT5inchR12199_02 " - ); - - mPMT_PMTtype_outer = new G4UIcmdWithAString("/WCSim/mPMT/PMTtype_outer",this); - mPMT_PMTtype_outer->SetGuidance("Set type of PMT for OD."); - mPMT_PMTtype_outer->SetParameterName("PMTtype_outer", true); - mPMT_PMTtype_outer->SetCandidates("PMT20inch " - "PMT8inch " - "PMT10inch " - "PMT10inchHQE " - "PMT12inchHQE " - "HPD20inchHQE " - "HPD12inchHQE " - "BoxandLine20inchHQE " - "BoxandLine12inchHQE " - "PMT3inchR12199_02 " - "PMT3inchR14374 " - "PMT4inchR12199_02 " - "PMT5inchR12199_02 " - ); - - - mPMT_material_outer = new G4UIcmdWithAString("/WCSim/mPMT/material_outer",this); - mPMT_material_outer->SetGuidance("Set material of outer capsule of multiPMT."); - mPMT_material_outer->SetParameterName("material_outer", true); - mPMT_material_outer->SetCandidates("Water " - "Glass " - "Acrylic "); - - mPMT_material_inner = new G4UIcmdWithAString("/WCSim/mPMT/material_inner",this); - mPMT_material_inner->SetGuidance("Set material of region in between PMTs."); - mPMT_material_inner->SetParameterName("material_inner", true); - mPMT_material_inner->SetCandidates("Water " - "SilGel " - "Air " ); - - // Thickness of outer and inner material - mPMT_material_outer_thickness = new G4UIcmdWithADoubleAndUnit("/WCSim/mPMT/material_outer_thickness",this); - mPMT_material_outer_thickness->SetGuidance("Set thickness of outer (pressure) vessel of multiPMT."); - mPMT_material_outer_thickness->SetParameterName("material_outer_thickness", true); - mPMT_material_outer_thickness->SetDefaultValue(20.); - mPMT_material_outer_thickness->SetUnitCategory("Length"); - mPMT_material_outer_thickness->SetDefaultUnit("mm"); - mPMT_material_outer_thickness->SetUnitCandidates("mm cm m"); - - - mPMT_ID_reflector_height = new G4UIcmdWithADoubleAndUnit("/WCSim/mPMT/reflectorHeightID",this); - mPMT_ID_reflector_height->SetGuidance("Set height of reflector cone for each ID PMT."); - mPMT_ID_reflector_height->SetParameterName("reflectorHeightID", true); - mPMT_ID_reflector_height->SetDefaultValue(7.5); - mPMT_ID_reflector_height->SetUnitCategory("Length"); - mPMT_ID_reflector_height->SetDefaultUnit("mm"); - mPMT_ID_reflector_height->SetUnitCandidates("mm cm m"); - - mPMT_ID_reflector_z_offset = new G4UIcmdWithADoubleAndUnit("/WCSim/mPMT/reflectorZoffsetID",this); - mPMT_ID_reflector_z_offset->SetGuidance("Set z position offset of reflector cone for each ID PMT."); - mPMT_ID_reflector_z_offset->SetParameterName("reflectorZoffsetID", true); - mPMT_ID_reflector_z_offset->SetDefaultValue(12); - mPMT_ID_reflector_z_offset->SetUnitCategory("Length"); - mPMT_ID_reflector_z_offset->SetDefaultUnit("mm"); - mPMT_ID_reflector_z_offset->SetUnitCandidates("mm cm m"); - - - mPMT_ID_reflector_angle = new G4UIcmdWithADoubleAndUnit("/WCSim/mPMT/reflectorAngleID",this); - mPMT_ID_reflector_angle->SetGuidance("Set angle of reflector cone wrt PMT axis for each ID PMT."); - mPMT_ID_reflector_angle->SetParameterName("reflectorAngleID", true); - mPMT_ID_reflector_angle->SetDefaultValue(45.); - mPMT_ID_reflector_angle->SetUnitCategory("Angle"); - mPMT_ID_reflector_angle->SetDefaultUnit("deg"); - mPMT_ID_reflector_angle->SetUnitCandidates("deg rad"); - - - // Options related to mPMT filling: - // Total number of ID PMTs: used to cross check with provided configfile - mPMT_nID_PMTs = new G4UIcmdWithAnInteger("/WCSim/mPMT/nID_PMTs",this); - mPMT_nID_PMTs->SetGuidance("Set number of ID mPMT PMTs."); - mPMT_nID_PMTs->SetParameterName("nID_PMTs", true); - mPMT_nID_PMTs->SetDefaultValue(20.); - - mPMT_config = new G4UIcmdWithAString("/WCSim/mPMT/configfile",this); - mPMT_config->SetGuidance("Set filename for config file with viewing angles and tilt angles."); - mPMT_config->SetParameterName("mPMTconfig", true); - - // NEW! - mPMT_numModulesFixed = new G4UIcmdWithABool("/WCSim/mPMT/keep_num_mPMTs_fixed",this); - mPMT_numModulesFixed->SetGuidance("Keep the number of mPMT modules in the detector fixed even though mPMT dimensions change."); - mPMT_numModulesFixed->SetParameterName("numModulesFixed",true); - mPMT_numModulesFixed->SetDefaultValue(false); - - mPMT_pmtOpeningAngle = new G4UIcmdWithADoubleAndUnit("/WCSim/mPMT/pmt_opening_angle",this); - mPMT_pmtOpeningAngle->SetGuidance("Set size of PMT sub-assembly (opening angle). Must be able to contain reflector if wanted."); - mPMT_pmtOpeningAngle->SetParameterName("pmtOpeningAngle", true); - mPMT_pmtOpeningAngle->SetDefaultValue(8.7); - mPMT_pmtOpeningAngle->SetUnitCategory("Angle"); - mPMT_pmtOpeningAngle->SetDefaultUnit("deg"); - mPMT_pmtOpeningAngle->SetUnitCandidates("deg rad"); - - mPMT_material_pmtAssembly = new G4UIcmdWithAString("/WCSim/mPMT/material_pmtAssembly",this); - mPMT_material_pmtAssembly->SetGuidance("Set material of region around the PMT itself."); - mPMT_material_pmtAssembly->SetParameterName("material_pmtAssembly", true); - mPMT_material_pmtAssembly->SetCandidates("Water " - "SilGel " - "Air " ); - - // B.Q: Add the hybrid configuration - SetHybridPMT = new G4UIcmdWithABool("/WCSim/mPMT/SetHybridPMT", this); - SetHybridPMT->SetGuidance("Set the possibility of having hybrid PMT types"); - SetHybridPMT->SetGuidance("Should be 1 or 0"); - SetHybridPMT->SetParameterName("HybridPMT", false); - //SetHybridPMT->SetCandidates(); - SetHybridPMT->SetDefaultValue(false); - - // First, the PMT type - SetPMTType = new G4UIcmdWithAString("/WCSim/nuPRISM/SetPMTType", this); - SetPMTType->SetGuidance("Set the type of PMT to be used for nuPRISM"); - SetPMTType->SetGuidance("Available options are:\n" - "PMT3inch\n" - "PMT3inchGT\n" - "PMT3inchR12199_02\n" - "PMT3inchR14374\n" - "PMT5inch\n" - "PMT8inch\n" - "PMT10inchHQE\n" - "PMT10inch\n" - "PMT12inchHQE\n" - "HPD20inchHQE\n" - "PMT20inch\n"); - SetPMTType->SetParameterName("PMTType", false); - SetPMTType->SetCandidates("PMT3inch PMT3inchGT PMT3inchR12199_02 PMT3inchR14374 PMT5inch PMT8inch PMT10inchHQE PMT10inch PMT12inchHQE HPD20inchHQE PMT20inch"); - SetPMTType->SetDefaultValue("PMT10inch"); - - // Next, the PMT coverage - SetPMTCoverage = new G4UIcmdWithAString("/WCSim/nuPRISM/SetPMTPercentCoverage", this); - SetPMTCoverage->SetGuidance("Set the PMT percentage coverage to be used for nuPRISM"); - SetPMTCoverage->SetGuidance("Any number is allowed (such as 40)"); - SetPMTCoverage->SetParameterName("PMTCoverage", false); - SetPMTCoverage->SetDefaultValue("40"); - - // Set the vertical position of the nuPRISM-lite detector - SetDetectorVerticalPosition = new G4UIcmdWithADoubleAndUnit("/WCSim/nuPRISM/SetDetectorVerticalPosition", this); - SetDetectorVerticalPosition->SetGuidance("Set the vertical position of the nuPRISM inner detector (unit mm cm m)."); - SetDetectorVerticalPosition->SetGuidance("The default will be 0m, so particle guns are easy to create."); - SetDetectorVerticalPosition->SetParameterName("DetectorVerticalPosition", false); - SetDetectorVerticalPosition->SetDefaultValue(0.0); - SetDetectorVerticalPosition->SetUnitCategory("Length"); - SetDetectorVerticalPosition->SetDefaultUnit("m"); - - // Set the height of the nuPRISM-lite detector - SetDetectorHeight = new G4UIcmdWithADoubleAndUnit("/WCSim/nuPRISM/SetDetectorHeight", this); - SetDetectorHeight->SetGuidance("Set the height of the nuPRISM inner detector (unit mm cm m)."); - SetDetectorHeight->SetGuidance("The default will be 52.4m, the full height of nuPRISM."); - SetDetectorHeight->SetParameterName("DetectorHeight", false); - SetDetectorHeight->SetDefaultValue(52.4); - SetDetectorHeight->SetUnitCategory("Length"); - SetDetectorHeight->SetDefaultUnit("m"); - - // Set the diameter of the nuPRISM-lite detector - SetDetectorDiameter = new G4UIcmdWithADoubleAndUnit("/WCSim/nuPRISM/SetDetectorDiameter", this); - SetDetectorDiameter->SetGuidance("Set the diameter of the nuPRISM inner detector (unit mm cm m)."); - SetDetectorDiameter->SetGuidance("The default will be 6m, the nominal diameter of nuPRISM."); - SetDetectorDiameter->SetParameterName("DetectorDiameter", false); - SetDetectorDiameter->SetDefaultValue(6.); - SetDetectorDiameter->SetUnitCategory("Length"); - SetDetectorDiameter->SetDefaultUnit("m"); -} - -WCSimDetectorMessenger::~WCSimDetectorMessenger() -{ - delete PMTConfig; - delete SavePi0; - delete SaveCapture; - delete PMTQEMethod; - delete PMTCollEff; - delete waterTank_Length; - delete WCVisChoice; - - delete SetDetectorDiameter; - delete SetDetectorHeight; - delete SetDetectorVerticalPosition; - delete SetPMTCoverage; - delete SetPMTType; - delete SetHybridPMT; - - delete tubeCmd; - delete distortionCmd; - - delete mPMT_CylHeight; - delete mPMT_CylRadius; - delete WCSimDir; - delete mPMTDir; -} - -void WCSimDetectorMessenger::SetNewValue(G4UIcommand* command,G4String newValue) -{ - if( command == PMTConfig ) { - WCSimDetector->SetIsUpright(false); - WCSimDetector->SetIsEggShapedHyperK(false); - if ( newValue == "SuperK") { - WCSimDetector->SetSuperKGeometry(); - } else if ( newValue == "SuperK_20inchPMT_20perCent" ){ - WCSimDetector->SuperK_20inchPMT_20perCent(); - } else if ( newValue == "SuperK_20inchBandL_20perCent" ){ - WCSimDetector->SuperK_20inchBandL_20perCent(); - } else if ( newValue == "SuperK_12inchBandL_15perCent" ) { - WCSimDetector->SuperK_12inchBandL_15perCent(); - } else if ( newValue == "SuperK_20inchBandL_14perCent" ) { - WCSimDetector->SuperK_20inchBandL_14perCent(); - } else if ( newValue == "Cylinder_60x74_20inchBandL_14perCent" ) { - WCSimDetector->Cylinder_60x74_20inchBandL_14perCent(); - } else if ( newValue == "Cylinder_60x74_20inchBandL_40perCent" ) { - WCSimDetector->Cylinder_60x74_20inchBandL_40perCent(); - } else if ( newValue == "Cylinder_12inchHPD_15perCent" ){ - WCSimDetector->Cylinder_12inchHPD_15perCent(); - } else if ( newValue == "HyperK") { - WCSimDetector->SetHyperKGeometry(); - } else if ( newValue == "HyperK_3inch") { - WCSimDetector->SetHyperK_3inchGeometry(); - } else if ( newValue == "HyperK_8inch") { - WCSimDetector->SetHyperK_8inchGeometry(); - } else if ( newValue == "HyperK_10inch") { - WCSimDetector->SetHyperK_10inchGeometry(); - } else if ( newValue == "HyperK_SKPMT") { - WCSimDetector->SetHyperK_SKPMTGeometry(); - } else if ( newValue == "HyperK_mPMT") { - WCSimDetector->SetHyperK_mPMTGeometry(); - } else if ( newValue == "HyperK_HybridmPMT") { - WCSimDetector->SetHyperK_HybridmPMTGeometry(); - } else if ( newValue == "HyperK_HybridmPMT10PC") { - WCSimDetector->SetHyperK_HybridmPMT10PCGeometry(); - } else if ( newValue == "HyperK_HybridFake") { - WCSimDetector->SetHyperK_HybridFakeGeometry(); - } else if ( newValue == "EggShapedHyperK") { - WCSimDetector->SetIsEggShapedHyperK(true); - WCSimDetector->SetEggShapedHyperKGeometry(); - } else if ( newValue == "EggShapedHyperK_withHPD") { - WCSimDetector->SetIsEggShapedHyperK(true); - WCSimDetector->SetEggShapedHyperKGeometry_withHPD(); - } else if(newValue == "TestSinglemPMT") { - WCSimDetector->SetTestSinglemPMTGeometry(); - } else if(newValue == "Cylinder_60x74_3inchmPMT_14perCent" ) { - WCSimDetector->Cylinder_60x74_3inchmPMT_14perCent(); // MUST be Called after the mPMT settings, otherwise unknown - } else if(newValue == "Cylinder_60x74_3inchmPMT_40perCent" ) { - WCSimDetector->Cylinder_60x74_3inchmPMT_40perCent(); // MUST be Called after the mPMT settings, otherwise unknown - } else if(newValue == "Cylinder_60x74_3inch_14perCent" ) { - WCSimDetector->Cylinder_60x74_3inch_14perCent(); // MUST be Called after the mPMT settings, otherwise unknown - } else if(newValue == "Cylinder_60x74_3inch_40perCent" ) { - WCSimDetector->Cylinder_60x74_3inch_40perCent(); // MUST be Called after the mPMT settings, otherwise unknown - } else if ( newValue == "nuPRISM") { - WCSimDetector->SetIsNuPrism(true); - WCSimDetector->SetDefaultNuPrismGeometry(); - } else if ( newValue == "nuPRISM_mPMT") { - WCSimDetector->SetIsNuPrism(true); - WCSimDetector->SetNuPrism_mPMTGeometry(); - } else if ( newValue == "nuPRISMShort_mPMT") { - WCSimDetector->SetIsNuPrism(true); - WCSimDetector->SetNuPrismShort_mPMTGeometry(); - } else - G4cout << "That geometry choice is not defined!" << G4endl; - } - - if (command == SavePi0){ - G4cout << "Set the flag for saving pi0 info " << newValue << G4endl; - if (newValue=="true"){ - WCSimDetector->SavePi0Info(true); - }else if (newValue == "false"){ - WCSimDetector->SavePi0Info(false); - }else{ - - } - } - - if (command == SaveCapture){ - G4cout << "Set the flag for saving neutron capture info " << newValue << G4endl; - if (newValue=="true"){ - WCSimDetector->SaveCaptureInfo(true); - }else if (newValue == "false"){ - WCSimDetector->SaveCaptureInfo(false); - }else{ - - } - } - - if (command == PMTQEMethod){ - G4cout << "Set PMT QE Method " << newValue << " "; - if (newValue == "Stacking_Only"){ - WCSimDetector->SetPMT_QE_Method(1); - G4cout << "1"; - }else if (newValue == "Stacking_And_SensitiveDetector"){ - WCSimDetector->SetPMT_QE_Method(2); - G4cout << "2"; - }else if (newValue == "SensitiveDetector_Only"){ - WCSimDetector->SetPMT_QE_Method(3); - G4cout << "3"; - }else if (newValue == "DoNotApplyQE"){ - WCSimDetector->SetPMT_QE_Method(4); - G4cout << "4"; - }else{ - - } - G4cout << G4endl; - } - - if (command == PMTCollEff){ - G4cout << "Set PMT Collection Efficiency " << newValue << " "; - if (newValue == "on"){ - WCSimDetector->SetPMT_Coll_Eff(1); - G4cout << "1"; - }else if (newValue == "off"){ - WCSimDetector->SetPMT_Coll_Eff(0); - G4cout << "0"; - } - G4cout << G4endl; - } - - if (command == WCVisChoice){ - G4cout << "Set Vis Choice " << newValue << " "; - if (newValue == "OGLSX"){ - WCSimDetector->SetVis_Choice("OGLSX"); - }else if (newValue == "RayTracer"){ - WCSimDetector->SetVis_Choice("RayTracer"); - } - G4cout << G4endl; - } - - if (command == waterTank_Length){ - bool isEggShapedHyperK = WCSimDetector->GetIsEggShapedHyperK(); - if(isEggShapedHyperK == true){ - G4cout << "Set Partition Length in a cylinder " << newValue << " " << G4endl; - WCSimDetector->SetwaterTank_Length(waterTank_Length->GetNewDoubleValue(newValue)); - WCSimDetector->SetWaterTubeLength(waterTank_Length->GetNewDoubleValue(newValue)); - // WCSimDetector->SetIsEggShapedHyperK(true); - } - else { - G4cout << "Not egg-shaped HyperK Geometry. Detector length unchanged." << G4endl; - } - } - - if(command == DopedWater) { - G4cout << "Setting Gadolinium doping of water: " << newValue << G4endl; - WCSimDetector->SetDopedWater(DopedWater->GetNewBoolValue(newValue)); - } - - - if(command == DopingConcentration) { - G4cout << "Setting Gadolinium doping concentration: " << newValue << "percent" << G4endl; - WCSimDetector->AddDopedWater(DopingConcentration->GetNewDoubleValue(newValue)); - } - - if(command == PMTSize) { - G4cout << "SET PMT SIZE" << G4endl; - if ( newValue == "20inch"){ -// WCSimDetector->Set20inchPMTs(); - }else if (newValue == "10inch"){ -// WCSimDetector->Set10inchPMTs(); - }else - G4cout << "That PMT size is not defined!" << G4endl; - } - - if(command == PMTSize2) { - G4cout << "SET PMT of TYPE 2 SIZE" << G4endl; - if ( newValue == "20inch"){ - // WCSimDetector->Set20inchPMTs(); - }else if (newValue == "10inch"){ - // WCSimDetector->Set10inchPMTs(); - }else - G4cout << "That PMT size is not defined!" << G4endl; - } - - if (command == mPMT_CylHeight){ - G4cout << "Set Vessel Cylinder Height of MultiPMT to " << newValue << " " << G4endl; //doesn't work - //std::cout << "Set Cylinder Height of MultiPMT to " << newValue << " " << std::endl; - WCSimDetector->SetmPMT_VesselCylHeight(mPMT_CylHeight->GetNewDoubleValue(newValue)); - } - - - if (command == mPMT_CylRadiusCurv){ - G4cout << "Set Vessel Radius of Curvature of MultiPMT to " << newValue << " " << G4endl; //doesn't work - //std::cout << "Set Cylinder Radius of MultiPMT to " << newValue << " " << std::endl; - WCSimDetector->SetmPMT_VesselRadiusCurv(mPMT_CylRadiusCurv->GetNewDoubleValue(newValue)); - } - - if (command == mPMT_CylRadius){ - G4cout << "Set Vessel Radius of MultiPMT to " << newValue << " " << G4endl; //doesn't work - //std::cout << "Set Cylinder Radius of MultiPMT to " << newValue << " " << std::endl; - WCSimDetector->SetmPMT_VesselRadius(mPMT_CylRadius->GetNewDoubleValue(newValue)); - } - - - if (command == mPMT_orientation){ - if(newValue == "Horizontal") - WCSimDetector->SetmPMT_Orientation(HORIZONTAL); - else if(newValue == "Vertical") - WCSimDetector->SetmPMT_Orientation(VERTICAL); - else if(newValue == "Perpendicular") - WCSimDetector->SetmPMT_Orientation(PERPENDICULAR); - else - G4cout << "Undefined orientation wrt wall: reverting to default Perpendicular" << G4endl; - - } - if (command == mPMT_ID_reflector_height){ - G4cout << "Set Height of reflector cone to " << newValue << " " << G4endl; - WCSimDetector->SetmPMT_ReflectorHeight(mPMT_ID_reflector_height->GetNewDoubleValue(newValue)); - } - if (command == mPMT_ID_reflector_z_offset){ - G4cout << "Set Z position offset of reflector cone to " << newValue << " " << G4endl; - WCSimDetector->SetmPMT_ReflectorZoffset(mPMT_ID_reflector_z_offset->GetNewDoubleValue(newValue)); - } - if (command == mPMT_ID_reflector_angle){ - G4cout << "Set Angle of reflector cone to " << newValue << " " << G4endl; - WCSimDetector->SetmPMT_ReflectorAngle(mPMT_ID_reflector_angle->GetNewDoubleValue(newValue)); - } - if (command == mPMT_material_outer){ - WCSimDetector->SetmPMT_MaterialOuter(newValue); - } - if (command == mPMT_material_inner){ - WCSimDetector->SetmPMT_MaterialInner(newValue); - } - if(command == mPMT_material_outer_thickness){ - WCSimDetector->SetmPMT_MaterialOuterThickness(mPMT_material_outer_thickness->GetNewDoubleValue(newValue)); - } - if(command == mPMT_nID_PMTs){ - WCSimDetector->SetmPMT_nID(mPMT_nID_PMTs->GetNewIntValue(newValue)); - } - if(command == mPMT_config){ - WCSimDetector->SetmPMT_Config(newValue); - } - - if (command == mPMT_PMTtype_inner) - WCSimDetector->SetmPMT_PMTtype_inner(newValue); - if (command == mPMT_PMTtype_outer) - WCSimDetector->SetmPMT_PMTtype_outer(newValue); - - - if (command == mPMT_DistPMTVessel){ - G4cout << "Set Distance of PMT(s) to pressure vessel to " << newValue << " " << G4endl; //doesn't work - //std::cout << "Set Cylinder Radius of MultiPMT to " << newValue << " " << std::endl; - WCSimDetector->SetmPMT_DistPMTVessel(mPMT_DistPMTVessel->GetNewDoubleValue(newValue)); - } - - if (command == mPMT_numModulesFixed){ - G4cout << "Do you want fix the number of mPMT modules: " << newValue << G4endl; - WCSimDetector->SetmPMT_FixModules(mPMT_numModulesFixed->GetNewBoolValue(newValue)); - } - - if (command == mPMT_pmtOpeningAngle){ - WCSimDetector->SetmPMT_OpeningAngle(mPMT_pmtOpeningAngle->GetNewDoubleValue(newValue)); - } - - if (command == mPMT_material_pmtAssembly){ - WCSimDetector->SetmPMT_MaterialPMTassembly(newValue); - } - if (command == SetHybridPMT){ - G4cout << "Set hybrid config = " << newValue << G4endl; - WCSimDetector->SetHybridPMT(SetHybridPMT->GetNewBoolValue(newValue)); - } - - // Customize nuPRISM tank setup - if( WCSimDetector->GetIsNuPrism()){ - if (command == SetPMTType) WCSimDetector->SetPMTType(newValue); - else if (command == SetPMTCoverage) WCSimDetector->SetPMTCoverage(atof(newValue)); - else if (command == SetDetectorHeight) WCSimDetector->SetDetectorHeight(SetDetectorHeight->GetNewDoubleValue(newValue)); - else if (command == SetDetectorVerticalPosition) WCSimDetector->SetDetectorVerticalPosition(SetDetectorVerticalPosition->GetNewDoubleValue(newValue)); - else if (command == SetDetectorDiameter) WCSimDetector->SetDetectorDiameter(SetDetectorDiameter->GetNewDoubleValue(newValue)); - /* TF: deprecated. Don't remake geometry, just update variables in the setters. - else if (command == UpdateNuPrism){ - WCSimDetector->SetNuPrismGeometry(WCSimDetector->GetPMTType(), - WCSimDetector->GetPMTCoverage(), - WCSimDetector->GetWCIDHeight(), - WCSimDetector->GetWCIDDiameter(), - WCSimDetector->GetWCIDVerticalPosition()); - } */ - } - - if(command == WCConstruct) { - WCSimDetector->UpdateGeometry(); - } -} diff --git a/libWCSIM/src/WCSimEnumerations.cc b/libWCSIM/src/WCSimEnumerations.cc deleted file mode 100644 index 96d9cad..0000000 --- a/libWCSIM/src/WCSimEnumerations.cc +++ /dev/null @@ -1,66 +0,0 @@ -#include -#include - -#include "WCSimEnumerations.hh" - -std::string WCSimEnumerations::EnumAsString(DigitizerType_t d) -{ - switch(d) { - case (kDigitizerSKI) : - return "SKI"; - break; - default: - return ""; - break; - } - return ""; -} - -std::string WCSimEnumerations::EnumAsString(TriggerType_t t) -{ - switch(t) { - case (kTriggerNDigits) : - return "NDigits"; - break; - case (kTriggerNDigitsTest) : - return "NDigits_TEST"; - break; - case (kTriggerFailure) : - return "No_trigger_passed"; - break; - default: - return ""; - break; - } - return ""; -} - -std::string WCSimEnumerations::EnumAsString(WCSimRandomGenerator_t r) -{ - switch(r) { - case (RANDOM_E_RANLUX) : - return "RANLUX"; - break; - case (RANDOM_E_RANECU) : - return "RANECU"; - break; - case (RANDOM_E_HEPJAMES) : - return "HEPJAMES"; - break; - default: - return ""; - break; - } - return ""; -} - -TriggerType_t WCSimEnumerations::TriggerTypeFromString(std::string s) -{ - for(int i = int(kTriggerUndefined)+1; i <= kTriggerFailure; i++) { - if(s.compare(WCSimEnumerations::EnumAsString((TriggerType_t)i)) == 0) { - return (TriggerType_t)i; - } - } - std::cerr << "WCSimEnumerations::TriggerTypeFromString() Unknown string value " << s << std::endl; - return kTriggerUndefined; -} diff --git a/libWCSIM/src/WCSimEventAction.cc b/libWCSIM/src/WCSimEventAction.cc deleted file mode 100644 index cc05304..0000000 --- a/libWCSIM/src/WCSimEventAction.cc +++ /dev/null @@ -1,2122 +0,0 @@ -#include "WCSimEventAction.hh" -#include "WCSimTrajectory.hh" -#include "WCSimRunAction.hh" -#include "WCSimPrimaryGeneratorAction.hh" -#include "WCSimWCHit.hh" -#include "WCSimWCDigi.hh" -#include "WCSimWCDigitizer.hh" -#include "WCSimWCTrigger.hh" -#include "WCSimWCAddDarkNoise.hh" -#include "WCSimWCPMT.hh" -#include "WCSimDetectorConstruction.hh" - -#include "G4Event.hh" -#include "G4RunManager.hh" -#include "G4EventManager.hh" -#include "G4UImanager.hh" -#include "G4TrajectoryContainer.hh" -#include "G4VVisManager.hh" -#include "G4ios.hh" -#include "globals.hh" -#include "G4ThreeVector.hh" -#include "G4TransportationManager.hh" -#include "G4Navigator.hh" -#include "G4SDManager.hh" -#include "G4DigiManager.hh" -#include "G4UnitsTable.hh" -#include "G4UIcmdWith3VectorAndUnit.hh" - -#include "G4PhysicalConstants.hh" -#include "G4SystemOfUnits.hh" - -#include -#include -#include -#include - -#include "jhfNtuple.h" -#include "TTree.h" -#include "TFile.h" -#include "WCSimRootEvent.hh" -#include "TStopwatch.h" - -#ifndef _SAVE_RAW_HITS -#define _SAVE_RAW_HITS -#ifndef _SAVE_RAW_HITS_VERBOSE -//#define _SAVE_RAW_HITS_VERBOSE -#endif -#endif -#ifndef SAVE_DIGITS_VERBOSE -//#define SAVE_DIGITS_VERBOSE -#endif -#ifndef TIME_DAQ_STEPS -//#define TIME_DAQ_STEPS -#endif - -#ifndef NPMTS_VERBOSE -#define NPMTS_VERBOSE 10 -#endif -//#define DEBUG - -WCSimEventAction::WCSimEventAction(WCSimRunAction* myRun, - WCSimDetectorConstruction* myDetector, - WCSimPrimaryGeneratorAction* myGenerator) - :runAction(myRun), generatorAction(myGenerator), - detectorConstructor(myDetector), - ConstructedDAQClasses(false), - SavedOptions(false) -{ - DAQMessenger = new WCSimWCDAQMessenger(this); - - G4DigiManager* DMman = G4DigiManager::GetDMpointer(); - - //create PMT response module - WCSimWCPMT* WCDMPMT = new WCSimWCPMT( "WCReadoutPMT", myDetector,"tank"); - DMman->AddNewModule(WCDMPMT); - - randGen = new TRandom3(); - - //create dark noise module - WCSimWCAddDarkNoise* WCDNM = new WCSimWCAddDarkNoise("WCDarkNoise", detectorConstructor,"tank"); - DMman->AddNewModule(WCDNM); - - WCSimWCPMT* WCDMPMT2; - WCSimWCAddDarkNoise* WCDNM2; - //if(myDetector->GetHybridPMT()){ - WCDMPMT2 = new WCSimWCPMT( "WCReadoutPMT2", myDetector,"tankPMT2"); - DMman->AddNewModule(WCDMPMT2); - - WCDNM2 = new WCSimWCAddDarkNoise("WCDarkNoise2", detectorConstructor,"tankPMT2"); - DMman->AddNewModule(WCDNM2); - //} -} - -WCSimEventAction::~WCSimEventAction() -{ - delete DAQMessenger; -} - -void WCSimEventAction::CreateDAQInstances() -{ - if(ConstructedDAQClasses) { - G4cerr << "WCSimEventAction::CreateDAQInstances() has already been called. Exiting..." << G4endl; - exit(-1); - } - - G4cout << "Creating digitizer and trigger class instances in WCSimEventAction::CreateDAQInstances()" << G4endl; - - G4DigiManager* DMman = G4DigiManager::GetDMpointer(); - - //create your choice of digitizer module - if(DigitizerChoice == "SKI") { - WCSimWCDigitizerSKI* WCDM = new WCSimWCDigitizerSKI("WCReadoutDigits", detectorConstructor, DAQMessenger,"tank"); - DMman->AddNewModule(WCDM); - } - else { - G4cerr << "Unknown DigitizerChoice " << DigitizerChoice << G4endl; - exit(-1); - } - - //create your choice of trigger module - if(TriggerChoice == "NDigits") { - WCSimWCTriggerNDigits* WCTM = new WCSimWCTriggerNDigits("WCReadout", detectorConstructor, DAQMessenger,"tank"); - DMman->AddNewModule(WCTM); - } - else if(TriggerChoice == "NDigits2") { - WCSimWCTriggerNDigits2* WCTM = new WCSimWCTriggerNDigits2("WCReadout", detectorConstructor, DAQMessenger,"tank"); - DMman->AddNewModule(WCTM); - } - else { - G4cerr << "Unknown TriggerChoice " << TriggerChoice << G4endl; - exit(-1); - } - - //B.Q: For the second type of PMT in the hybrid configuration - //create your choice of digitizer module - if(DigitizerChoice == "SKI") { - WCSimWCDigitizerSKI* WCDM2; - if(detectorConstructor->GetHybridPMT()){ - WCDM2 = new WCSimWCDigitizerSKI("WCReadoutDigits2", detectorConstructor, DAQMessenger,"tankPMT2"); - DMman->AddNewModule(WCDM2); - } - } - else { - G4cerr << "Unknown DigitizerChoice " << DigitizerChoice << G4endl; - exit(-1); - } - - //create your choice of trigger module - if(TriggerChoice == "NDigits") { - WCSimWCTriggerNDigits* WCTM2; - if(detectorConstructor->GetHybridPMT()){ - WCTM2 = new WCSimWCTriggerNDigits("WCReadout2", detectorConstructor, DAQMessenger,"tankPMT2"); - DMman->AddNewModule(WCTM2); - } - } - else if(TriggerChoice == "NDigits2") { - WCSimWCTriggerNDigits2* WCTM2; - if(detectorConstructor->GetHybridPMT()){ - WCTM2 = new WCSimWCTriggerNDigits2("WCReadout2", detectorConstructor, DAQMessenger,"tankPMT2"); - DMman->AddNewModule(WCTM2); - } - } - else { - G4cerr << "Unknown TriggerChoice " << TriggerChoice << G4endl; - exit(-1); - } - - ConstructedDAQClasses = true; -} - - -void WCSimEventAction::BeginOfEventAction(const G4Event*) -{ - if(!ConstructedDAQClasses) { - CreateDAQInstances(); - - //and save options in output file - G4DigiManager* DMman = G4DigiManager::GetDMpointer(); - - } -} - -void WCSimEventAction::EndOfEventAction(const G4Event* evt) -{ - - // ---------------------------------------------------------------------- - // Get Particle Table - // ---------------------------------------------------------------------- - - G4ParticleTable* particleTable = G4ParticleTable::GetParticleTable(); - - // ---------------------------------------------------------------------- - // Get Trajectory Container - // ---------------------------------------------------------------------- - - G4TrajectoryContainer* trajectoryContainer = evt->GetTrajectoryContainer(); - - G4int n_trajectories = 0; - if (trajectoryContainer) n_trajectories = trajectoryContainer->entries(); - - if(generatorAction->GetIsRooTrackerFileFinished()){ - const G4Run* run; - GetRunAction()->EndOfRunAction(run); - exit(0); - } - - // ---------------------------------------------------------------------- - // Get Event Information - // ---------------------------------------------------------------------- - - G4int event_id = evt->GetEventID(); - //G4int mode = generatorAction->GetMode(); - InteractionType_t mode = generatorAction->GetMode(); - G4ThreeVector vtx = generatorAction->GetVtx(); - G4int vtxvol = WCSimEventFindStartingVolume(vtx); - G4int vecRecNumber = generatorAction->GetVecRecNumber(); - - // ---------------------------------------------------------------------- - // Get WC Hit Collection - // ---------------------------------------------------------------------- - - G4SDManager* SDman = G4SDManager::GetSDMpointer(); - - // Get Hit collection of this event - G4HCofThisEvent* HCE = evt->GetHCofThisEvent(); - WCSimWCHitsCollection* WCHC = 0; - G4String WCIDCollectionName = detectorConstructor->GetIDCollectionName(); - if (HCE) - { - G4String name = WCIDCollectionName; - G4int collectionID = SDman->GetCollectionID(name); - WCHC = (WCSimWCHitsCollection*)HCE->GetHC(collectionID); - } - //B.Q for the hybrid version - WCSimWCHitsCollection* WCHC2 = 0; - G4String WCIDCollectionName2; - if(detectorConstructor->GetHybridPMT()) WCIDCollectionName2 = detectorConstructor->GetIDCollectionName2(); - if (HCE) - { - G4String name; - G4int collectionID; - if(detectorConstructor->GetHybridPMT()){ - name = WCIDCollectionName2; - collectionID = SDman->GetCollectionID(name); - WCHC2 = (WCSimWCHitsCollection*)HCE->GetHC(collectionID); - } - } - - // If this option is chosen - // pe's will be generated on the pmts. - if( generatorAction->IsUsingPoissonPMT() ){ - - // Loop through PMTs in detector - for ( - std::vector::iterator pmtIt = detectorConstructor->Get_Pmts()->begin(); - pmtIt != detectorConstructor->Get_Pmts()->end(); - pmtIt++ ){ - - // Generate number of PEs acording to Poisson with macro defined mean - int nPoisson = randGen->Poisson(generatorAction->GetPoissonPMTMean()); - - // Check if PMTs have existing hits - bool hitExists = false; - int hitIndex = -1; - for (int existingHit = 0; existingHit < WCHC->GetSize(); existingHit++){ - - if( (*WCHC)[existingHit]->GetTubeID() == (*pmtIt)->Get_tubeid() ){ - - hitExists = true; - hitIndex = existingHit; - - G4cout << "/mygen/pmtPoisson Error: Hits found when placing dummy PEs. Use lower energy primary particles or set '/mygen/pmtPoisson' to false for a real run." << G4endl; - - if ((*WCHC)[existingHit]->GetTotalPe() > nPoisson){ - G4cout << "/mygen/pmtPoisson Error: More PEs in original hit than in the Poisson throw. Poisson distribution will be distorted. Use lower energy primary particles or set '/mygen/pmtPoisson' to false for a real run." << G4endl; - - } else { - // Take into account already exising PEs... this is not ideal as they might not be in time. - nPoisson -= (*WCHC)[existingHit]->GetTotalPe(); - } - } - if (hitExists) break; - } - - if (! hitExists){ - WCHC->insert((WCSimWCHit*) new WCSimWCHit() ); - hitIndex = WCHC->GetSize()-1; - - } - - (*WCHC)[hitIndex]->SetTubeID((*pmtIt)->Get_tubeid()); - (*WCHC)[hitIndex]->SetTrackID(0); - (*WCHC)[hitIndex]->SetEdep(0.); - (*WCHC)[hitIndex]->SetPos(detectorConstructor->GetTubeTransform((*pmtIt)->Get_tubeid()).getTranslation()); - (*WCHC)[hitIndex]->SetRot(detectorConstructor->GetTubeTransform((*pmtIt)->Get_tubeid()).getRotation()); - - // Ignore logical volume for now... - for (int pe = 0; pe < nPoisson; pe++) { - (*WCHC)[hitIndex]->AddPe(G4RandGauss::shoot(0.0,10.)); - (*WCHC)[hitIndex]->AddParentID(0); // Make parent a geantino (whatever that is) - } - - G4cout << "The option using pmtPoisson is not implemented for the hybrid version yet." << G4endl; - } - } - - - - // To use Do like This: - // -------------------- - // if (WCHC) - // for (G4int i=0; i< WCHC->entries() ;i++) - // G4cout << (*WCHC)[i]->GetTotalPe() << G4endl; - - - // ---------------------------------------------------------------------- - // Get Digitized Hit Collection - // ---------------------------------------------------------------------- - G4cout<< "WCSimEventAction::EndOfEventAction() (WCSimWCHitsCollection*)" << WCIDCollectionName - << " has " << WCHC->entries() << " entries" << G4endl; - G4cout << G4endl; - - // Get a pointer to the Digitizing Module Manager - G4DigiManager* DMman = G4DigiManager::GetDMpointer(); - - // Get a pointer to the WC PMT module - WCSimWCPMT* WCDMPMT = - (WCSimWCPMT*)DMman->FindDigitizerModule("WCReadoutPMT"); - - // new MFechner, aug 2006 - // need to clear up the old info inside PMT - WCDMPMT->ReInitialize(); - - -#ifdef TIME_DAQ_STEPS - TStopwatch* ms = new TStopwatch(); - ms->Start(); -#endif - - //Convert the hits to PMT pulse - WCDMPMT->Digitize(); - - // - // Do the Dark Noise, then Digitization, then Trigger - // - - // - // First, add Dark noise hits before digitizing - - //Get a pointer to the WC Dark Noise Module - WCSimWCAddDarkNoise* WCDNM = - (WCSimWCAddDarkNoise*)DMman->FindDigitizerModule("WCDarkNoise"); - - //WCDNM->SetDarkMode(1); - - //Add the dark noise - WCDNM->AddDarkNoise(); - - // - // Next, do the digitization - - //Get a pointer to the WC Digitizer Module - WCSimWCDigitizerBase* WCDM = - (WCSimWCDigitizerBase*)DMman->FindDigitizerModule("WCReadoutDigits"); - - //Digitize the hits - WCDM->Digitize(); - - // - // Finally, apply the trigger - - //Get a pointer to the WC Trigger Module - WCSimWCTriggerBase* WCTM = - (WCSimWCTriggerBase*)DMman->FindDigitizerModule("WCReadout"); - - //tell it the dark noise rate (for calculating the average dark occupancy -> can adjust the NDigits threshold) - WCTM->SetDarkRate(WCDNM->GetDarkRate()); - - //Apply the trigger - // This takes the digits, and places them into trigger gates - // Also throws away digits not contained in an trigger gate - WCTM->Digitize(); - -#ifdef TIME_DAQ_STEPS - ms->Stop(); - G4cout << " Digtization : Real = " << ms->RealTime() - << " ; CPU = " << ms->CpuTime() << "\n"; -#endif - - // Get the post-noise hit collection for the WC - G4int WCDChitsID = DMman->GetDigiCollectionID("WCRawPMTSignalCollection"); - WCSimWCDigitsCollection * WCDC_hits = (WCSimWCDigitsCollection*) DMman->GetDigiCollection(WCDChitsID); - - // Get the digitized collection for the WC - G4int WCDCID = DMman->GetDigiCollectionID("WCDigitizedCollection"); - WCSimWCTriggeredDigitsCollection * WCDC = (WCSimWCTriggeredDigitsCollection*) DMman->GetDigiCollection(WCDCID); - /* - // To use Do like This: - // -------------------- - if(WCDC) - for (G4int i=0; i < WCDC->entries(); i++) - { - G4int tubeID = (*WCDC)[i]->GetTubeID(); - G4float photoElectrons = (*WCDC)[i]->GetPe(i); - G4float time = (*WCDC)[i]->GetTime(i); - // G4cout << "time " << i << " " <