Changeset 19917 for trunk/Mars/msimreflector/MSimRays.cc

Timestamp:

12/17/19 17:26:49 (5 years ago)

Author:

tbretz

Message:

Added a possibility to produce a pre-defined wavelength distirbution.

File:

• trunk/Mars/msimreflector/MSimRays.cc (modified) (4 diffs)

trunk/Mars/msimreflector/MSimRays.cc

@@ -52 +52 @@
 #include "MParList.h"
 
+#include "MSpline3.h"
 #include "MQuaternion.h"
 
@@ -71 +72 @@
     : fEvt(0), fReflector(0), fPointPos(0), fSource(0),
     fNumPhotons(1000), fHeight(-1), fWavelengthMin(-1), fWavelengthMax(-1),
-    fNameReflector("MReflector"), fNamePointPos("MPointingPos"),
+    fRandomDist(0), fNameReflector("MReflector"), fNamePointPos("MPointingPos"),
     fNameSource("Source")
 {
     fName  = name  ? name  : "MSimRays";
     fTitle = title ? title : "Task to calculate reflection os a mirror";
+}
+
+MSimRays::~MSimRays()
+{
+    delete fRandomDist;
 }
 
@@ -234 +240 @@
         dat.SetDirection(w);
 
-        if (fWavelengthMin>0 && fWavelengthMax>0)
-            dat.SimWavelength(fWavelengthMin, fWavelengthMax);
+        if (fRandomDist)
+        {
+            dat.SetWavelength(fRandomDist->FindX(gRandom->Uniform()));
+        }
+        else
+        {
+            if (fWavelengthMin>0 && fWavelengthMax>0)
+                dat.SimWavelength(fWavelengthMin, fWavelengthMax);
+        }
 
         idx++;
@@ -247 +260 @@
 }
 
+void MSimRays::SetDistribution(const MSpline3 &s)
+{
+    delete fRandomDist;
+    fRandomDist = new MSpline3(s.GetIntegralSpline());
+}
+
+bool MSimRays::ReadDistribution(const char *filename, const char *fmt)
+{
+    TGraph g(filename, fmt);
+    if (g.GetN()<2)
+        return false;
+
+    SetDistribution(MSpline3(g));
+    return true;
+}
+
 // --------------------------------------------------------------------------
 //