CalFileFromILL.cxx

Go to the documentation of this file.

#include <iostream>
#include <iomanip>
#include <fstream>
#include <sstream>
#include <string>
#include <vector>
 
#include "TChannel.h"
#include "TFipps.h"
 
int main(int argc, char** argv) {
   if(argc < 4) {
      std::cerr<<"Usage: "<<argv[0]<<" <LUT file> <cal file(s)> <cross talk file>"<<std::endl;
      return 1;
   }
 
   std::stringstream str;
   std::string line;
 
   TChannel* channel = nullptr;
 
   int adc;
   int detType;
   int detNumber;
   int cryNumber;
   int globId;
   int timeOffset;
 
   //////////////////////////////////////// look up table ////////////////////////////////////////
   std::ifstream lutFile(argv[1]);
 
   while(std::getline(lutFile, line)) {
      if(line.empty() || std::all_of(line.begin(), line.end(), [](char c){ return std::isspace(c); }) || line[0] == '#') continue;
 
      str.clear();
      str.str(line);
 
      str>>adc>>detType>>detNumber>>cryNumber>>globId>>timeOffset;// ignoring rangemin, rangemax at the end of the line
      //std::cout<<"line \""<<line<<"\" adc "<<adc<<", detType "<<detType<<", cryNumber "<<cryNumber<<", globId "<<globId<<", timeOffset "<<timeOffset<<std::endl;
 
      // Detector Types:
      //  -- 0 -> NotUsed
      //  -- 1 -> FIPPS Ge
      //  -- 2 -> IFIN Ge
      //  -- 3 -> FIPPS AC
      //  -- 4 -> IFIN AC
      //  -- 5 -> Fission Tag
      //  -- 6 -> LaBr3
      //  -- 7 -> TAC
      //
      // In case of AC, CloverId -> Associated clover
      //
      // DetId:  0->3 for Ge
      //         0    for IFIN AC
      //         0->2 for FIPPS AC (Back, Front, Side)
      //         id   for TAG, LaBr3, TAC
 
      str.clear();
      str.str(std::string());
      //std::cout<<"cleared - "<<str.str()<<std::endl;
      switch(detType) {
         case 1: // FIPPS Ge
            str<<"FIG";
            break;
         case 2: // IFIN Ge
            str<<"IFG";
            break;
         case 3: // FIPPS BGO
            str<<"FIS";
            break;
         case 4: // IFIN BGO
            str<<"IFS";
            break;
         default:
            std::cerr<<"unknown detector type "<<detType<<std::endl;
            continue;
      }
      //std::cout<<"system - "<<str.str()<<std::endl;
      str<<std::setfill('0')<<std::setw(2)<<detNumber+1; // LUT starts counting at 0, cal file starts at 1
      //std::cout<<"detector # - "<<str.str()<<std::endl;
      if(detType == 1 || detType == 2) { // Ge detectors - set crystals as colors
         str<<TFipps::GetColorFromNumber(cryNumber);
      } else {
         str<<std::setw(1)<<cryNumber;
      }
      //std::cout<<"crystal - "<<str.str()<<std::endl;
      str<<"N00X";
      //std::cout<<"final - "<<str.str()<<std::endl;
      channel = new TChannel(str.str().c_str());
      channel->SetAddress((adc/16)*0x40+adc%16);
      channel->SetNumber(TPriorityValue<int>(globId+1, EPriority::kForce));
      channel->SetTimeOffset(TPriorityValue<Long64_t>(-timeOffset, EPriority::kForce));
      channel->SetDigitizerType(TPriorityValue<std::string>("V1725", EPriority::kForce));
      TChannel::AddChannel(channel);
      //channel->Print();
   }
 
   //////////////////////////////////////// energy calibration ////////////////////////////////////////
   Float_t tmpFloat;
   std::vector<Float_t> coefficients;
   int minRange = 0;
   int maxRange = 30000;
   for(int i = 2; i < argc-1; ++i) {
      std::ifstream calFile(argv[i]);
      //std::cout<<"Opened energy calibration file '"<<argv[i]<<"' for range "<<i-2<<std::endl;
      while(std::getline(calFile, line)) {
         // erase all trailing whitespace
         line.erase(std::find_if(line.rbegin(), line.rend(), [](int ch) { return !std::isspace(ch); }).base(), line.end());
 
         if(line.empty() || std::all_of(line.begin(), line.end(), [](char c){ return std::isspace(c); }) || line[0] == '#') continue;
 
         str.clear();
         str.str(line);
         str>>globId;
         channel = TChannel::GetChannelByNumber(globId+1);
         if(channel != nullptr) {
            if(i == 2) {
               channel->DestroyENGCal();
               channel->ResizeENG(argc-3); // -3 = program itself, LUT file, and cross talk file
               //std::cout<<"resized ENG to "<<argc-3<<" - "<<channel->GetMnemonic()->GetName()<<std::endl;
            }
            //std::cout<<"Setting "<<i-2<<" energy coefficients for channel "<<globId+1<<" - "<<channel->GetMnemonic()->GetName()<<std::endl;
            //std::cout<<"From line \""<<line<<"\": "<<std::endl;
            while(str.good()) {
               str>>tmpFloat;
               coefficients.push_back(tmpFloat);
            }
            //for(auto val : coefficients) std::cout<<val<<"\t";
            //std::cout<<std::endl;
            // we only expect the last two columns to be the ranges if we have more than one cal-file
            if(argc > 4) {
               maxRange = coefficients.back();
               coefficients.pop_back();
               minRange = coefficients.back();
               coefficients.pop_back();
            }
            channel->SetENGCoefficients(coefficients, i-2);
            coefficients.clear();
channel->Print();
            //std::cout<<minRange<<"-"<<maxRange<<std::endl;
            channel->SetENGRange(std::make_pair(minRange, maxRange), i-2);
channel->Print();
         } else {
            std::cerr<<"Failed to find detector ID "<<globId+1<<" in TChannel"<<std::endl;
         }
      }
   }
 
   //////////////////////////////////////// cross talk ////////////////////////////////////////
   std::ifstream xTalkFile(argv[argc-1]);
   detNumber = 1;
   int col;
   int row;
   double val;
   while(std::getline(xTalkFile, line)) {
      if(line.empty() || std::all_of(line.begin(), line.end(), [](char c){ return std::isspace(c); }) || line[0] == '#') continue;
 
      str.clear();
      str.str(line);
      str>>col>>row>>val;
      //std::cout<<"from line \""<<line<<"\" got col "<<col<<", row "<<row<<", val "<<val<<std::endl;
 
      // col = affected crystal, row = affecting crystal
      if(row == 0) {
         channel = TChannel::FindChannelByName(Form("FIG%02d%sN00X", detNumber, TFipps::GetColorFromNumber(col)));
         if(channel == nullptr) {
            channel = TChannel::FindChannelByName(Form("IFG%02d%sN00X", detNumber, TFipps::GetColorFromNumber(col)));
         }
         if(channel == nullptr) {
            std::cerr<<"Failed to find channel \""<<Form("FI/IFG%02d%sN00X", detNumber, TFipps::GetColorFromNumber(col))<<"\""<<std::endl;
            std::cerr<<"Got color "<<TFipps::GetColorFromNumber(col)<<" from number "<<col<<std::endl;
            break;
         }
         channel->DestroyCTCal();
         
         //std::cout<<"destroyed cross talk coefficients for channel "<<channel->GetMnemonic()->GetName()<<std::endl;
      }
      channel->AddCTCoefficient(val);
      if(col == 3 && row == 3) ++detNumber;
   }
 
   //std::cout<<std::endl<<"done"<<std::endl<<std::endl;
 
   TChannel::WriteCalFile();
}