Changeset 12343 for trunk/Mars/mcore

Timestamp:

10/31/11 14:52:20 (13 years ago)

Author:

tbretz

Message:

Moved some common code between MARS and FACT++ to DrsCalib.h as new class DrsCalibration; renamed CalibData to DrsCalibrate.

File:

• trunk/Mars/mcore/DrsCalib.h (modified) (6 diffs)

trunk/Mars/mcore/DrsCalib.h

@@ -1 @@
-#ifndef MARS_CalibData
-#define MARS_CalibData
-
-class CalibData
+#ifndef MARS_DrsCalib
+#define MARS_DrsCalib
+
+#include "fits.h"
+
+class DrsCalibrate
 {
 protected:
@@ -10 +12 @@
     size_t fNumChannels;
 
-    vector<int64_t> fSum;
-    vector<int64_t> fSum2;
+    std::vector<int64_t> fSum;
+    std::vector<int64_t> fSum2;
 
     /*
@@ -18 +20 @@
      */
 public:
-    CalibData() : fNumEntries(0), fNumSamples(0), fNumChannels(0) { }
+    DrsCalibrate() : fNumEntries(0), fNumSamples(0), fNumChannels(0) { }
     void Reset()
     {
@@ -240 +242 @@
     }
 
-    pair<vector<double>,vector<double> > GetSampleStats() const
+    std::pair<std::vector<double>,std::vector<double> > GetSampleStats() const
     {
         if (fNumEntries==0)
-            return make_pair(vector<double>(),vector<double>());
-
-        vector<double> mean(fSum.size());
-        vector<double> error(fSum.size());
-
-        vector<int64_t>::const_iterator it = fSum.begin();
-        vector<int64_t>::const_iterator i2 = fSum2.begin();
-        vector<double>::iterator        im = mean.begin();
-        vector<double>::iterator        ie = error.begin();
+            return make_pair(std::vector<double>(),std::vector<double>());
+
+        std::vector<double> mean(fSum.size());
+        std::vector<double> error(fSum.size());
+
+        std::vector<int64_t>::const_iterator it = fSum.begin();
+        std::vector<int64_t>::const_iterator i2 = fSum2.begin();
+        std::vector<double>::iterator        im = mean.begin();
+        std::vector<double>::iterator        ie = error.begin();
 
         while (it!=fSum.end())
@@ -283 +285 @@
         }
 
-        vector<int64_t>::const_iterator it = fSum.begin();
-        vector<int64_t>::const_iterator i2 = fSum2.begin();
+        std::vector<int64_t>::const_iterator it = fSum.begin();
+        std::vector<int64_t>::const_iterator i2 = fSum2.begin();
 
         while (it!=fSum.end())
@@ -348 +350 @@
     }
 
-    const vector<int64_t> &GetSum() const { return fSum; }
+    const std::vector<int64_t> &GetSum() const { return fSum; }
 
     uint64_t GetNumEntries() const { return fNumEntries; }
 };
 
+struct DrsCalibration
+{
+    std::vector<int32_t> fOffset;
+    std::vector<int64_t> fGain;
+    std::vector<int64_t> fTrgOff;
+
+    uint64_t fNumOffset;
+    uint64_t fNumGain;
+    uint64_t fNumTrgOff;
+
+    uint32_t fStep;
+
+    DrsCalibration() :
+        fOffset(1440*1024, 0),
+        fGain(1440*1024, 4096),
+        fTrgOff(1440*1024, 0),
+        fNumOffset(1),
+        fNumGain(2000),
+        fNumTrgOff(1),
+        fStep(0)
+    {
+    }
+
+    void Clear()
+    {
+        // Default gain:
+        // 0.575*[45590]*2.5V / 2^16 = 0.99999 V
+        fOffset.assign(1440*1024, 0);
+        fGain.assign  (1440*1024, 4096);
+        fTrgOff.assign(1440*1024, 0);
+
+        fNumOffset = 1;
+        fNumGain   = 2000;
+        fNumTrgOff = 1;
+
+        fStep = 0;
+    }
+
+    string ReadFitsImp(const string &str, std::vector<float> &vec)
+    {
+        std::fits file(str);
+        if (!file)
+        {
+            std::ostringstream msg;
+            msg << "Could not open file " << str << ": " << strerror(errno);
+            return msg.str();
+        }
+
+        if (file.GetStr("TELESCOP")!="FACT")
+        {
+            std::ostringstream msg;
+            msg << "Reading " << str << " failed: Not a valid FACT file (TELESCOP not FACT in header)";
+            return msg.str();
+        }
+
+        if (!file.HasKey("STEP"))
+        {
+            std::ostringstream msg;
+            msg << "Reading " << str << " failed: Is not a DRS calib file (STEP not found in header)";
+            return msg.str();
+        }
+
+        if (file.GetNumRows()!=1)
+        {
+            std::ostringstream msg;
+            msg << "Reading " << str << " failed: Number of rows in table is not 1.";
+            return msg.str();
+        }
+
+        vec.resize(1440*1024*6+3);
+
+        float *base = vec.data();
+
+        file.SetPtrAddress("RunNumberBaseline",      base,   1);
+        file.SetPtrAddress("RunNumberGain",          base+1, 1);
+        file.SetPtrAddress("RunNumberTriggerOffset", base+2, 1);
+        file.SetPtrAddress("BaselineMean",           base+0*1024*1440+3, 1024*1440);
+        file.SetPtrAddress("BaselineRms",            base+1*1024*1440+3, 1024*1440);
+        file.SetPtrAddress("GainMean",               base+2*1024*1440+3, 1024*1440);
+        file.SetPtrAddress("GainRms",                base+3*1024*1440+3, 1024*1440);
+        file.SetPtrAddress("TriggerOffsetMean",      base+4*1024*1440+3, 1024*1440);
+        file.SetPtrAddress("TriggerOffsetRms",       base+5*1024*1440+3, 1024*1440);
+
+        if (!file.GetNextRow())
+        {
+            std::ostringstream msg;
+            msg << "Reading data from " << str << " failed.";
+            return msg.str();
+        }
+
+        fStep      = file.GetUInt("STEP");
+        fNumOffset = file.GetUInt("NBOFFSET");
+        fNumGain   = file.GetUInt("NBGAIN");
+        fNumTrgOff = file.GetUInt("NBTRGOFF");
+
+        fOffset.resize(1024*1440);
+        fGain.resize(1024*1440);
+        fTrgOff.resize(1024*1440);
+
+        // Convert back to ADC counts: 256/125 = 4096/2000
+        // Convert back to sum (mean * num_entries)
+        for (int i=0; i<1024*1440; i++)
+        {
+            fOffset[i] = fNumOffset           *256*base[i+1024*1440*0+3]/125;
+            fGain[i]   = fNumOffset*fNumGain  *256*base[i+1024*1440*2+3]/125;
+            fTrgOff[i] = fNumOffset*fNumTrgOff*256*base[i+1024*1440*4+3]/125;
+        }
+
+        // DAC:  0..2.5V == 0..65535
+        // V-mV: 1000
+        //fNumGain *= 2500*50000;
+        //for (int i=0; i<1024*1440; i++)
+        //    fGain[i] *= 65536;
+        if (fStep==0)
+        {
+            for (int i=0; i<1024*1440; i++)
+                fGain[i] = fNumOffset*4096;
+        }
+        else
+        {
+            fNumGain *= 1953125;
+            for (int i=0; i<1024*1440; i++)
+                fGain[i] *= 1024;
+        }
+
+        // Now mark the stored DRS data as "officially valid"
+        // However, this is not thread safe. It only ensures that
+        // this data is not used before it is completely and correctly
+        // read.
+        fStep++;
+
+        return string();
+    }
+    string ReadFitsImp(const string &str)
+    {
+        std::vector<float> vec;
+        return ReadFitsImp(str, vec);
+    }
+
+    bool IsValid() { return fStep>2; }
+
+    void Apply(float *vec, int16_t *val, const int16_t *start, uint32_t roi)
+    {
+        DrsCalibrate::Apply(vec, val, start, roi,
+                            fOffset.data(), fNumOffset,
+                            fGain.data(),   fNumGain,
+                            fTrgOff.data(), fNumTrgOff);
+
+    }
+};
+
 #endif