LArOFCSubsetCnv_p1.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "LArRawConditions/LArConditionsSubset.h"
#include "LArCondTPCnv/LArOFCSubsetCnv_p1.h"
 
void
LArOFCSubsetCnv_p1::persToTrans(const LArOFCPersType* persObj, 
                                  LArOFCTransType* transObj, 
                                  MsgStream & log) const
{
    // Copy basic metadata
    transObj->setChannel       (persObj->m_subset.m_channel);
    transObj->setGroupingType  (persObj->m_subset.m_groupingType);
 
    transObj->initialize (persObj->m_subset.m_febIds, persObj->m_subset.m_gain);
 
    // Copy conditions
    unsigned int nfebids    = persObj->m_subset.m_febIds.size();
    const unsigned int nChannelsPerFeb  = persObj->m_subset.subsetSize();
    unsigned int nPhases    = persObj->m_nPhases;
    unsigned int nSamples   = persObj->m_nSamples;
    unsigned int dataIndex  = 0;
    unsigned int timeIndex  = 0;
 
    // Loop over febs
    unsigned int ifebWithData = 0; // counter for febs with data
 
    auto subsetIt = transObj->subsetBegin();
    for (unsigned int i = 0; i < nfebids; ++i, ++subsetIt){
        // Set febid
        unsigned int febid = (*subsetIt).first;
 
        bool hasSparseData       = false;
        unsigned int chansSet    = 0;
        unsigned int chansOffset = 0;
        if (ifebWithData < persObj->m_subset.m_febsWithSparseData.size() &&
            febid == persObj->m_subset.m_febsWithSparseData[ifebWithData]) {
            // Found feb with sparse data
            hasSparseData = true;
            ifebWithData++;
            chansSet    = persObj->m_subset.m_febsWithSparseData[ifebWithData];
            chansOffset = 0;
            ifebWithData++;
        }
            
        // Loop over channels in feb - only some channels are filled
        for (unsigned int j = 0; j < nChannelsPerFeb; ++j){
 
            bool copyChannel = true;
            if (hasSparseData) {
                if (!(chansSet & (1 << (j - chansOffset)))) {
                    // Channel is missing data - skip
                    copyChannel = false;
                            }
                if (j%32 == 31 && j < nChannelsPerFeb-2) {
                    chansSet     = persObj->m_subset.m_febsWithSparseData[ifebWithData];
                    chansOffset += 32;
                    ifebWithData++;
                }
            }
            if (copyChannel) {
                        
                // Channel has ofcs - loop over ofcs per channel and
                // copy to the persistent object
 
                // check indexes
                if (dataIndex  >= persObj->m_vOFC_a.size()    ||
                    dataIndex  >= persObj->m_vOFC_b.size()    ||
                    timeIndex >= persObj->m_timeOffset.size() ||
                    timeIndex >= persObj->m_timeBinWidth.size()) {
                    log << MSG::ERROR 
                        << "LArOFCSubsetCnv_p1::persToTrans - ofc index too large: dataIndex size ofc_a, size ofc_b, timeIndex timeOffset size, timeBinWidth size " 
                        << dataIndex  << " " << persObj->m_vOFC_a.size() << " " 
                        << persObj->m_vOFC_b.size() << " " << timeIndex << " "
                        << persObj->m_timeOffset.size() << " " 
                        << persObj->m_timeBinWidth.size()
                        << endmsg;
                    return;
                }
 
                LArOFCTransType::Reference ofcs =
                  (*subsetIt).second[j];
                LArOFCP1 tmp (persObj->m_timeOffset[timeIndex],
                              persObj->m_timeBinWidth[timeIndex],
                              nPhases,
                              nSamples,
                              persObj->m_vOFC_a,
                              persObj->m_vOFC_b,
                              dataIndex);
                ofcs.assign (tmp);
                ++timeIndex;
                dataIndex += nPhases * nSamples;
            }
        }
    }
 
    // Copy corrections
    unsigned int ncorrs     = persObj->m_subset.m_corrChannels.size();
    LArOFCTransType::CorrectionVec corrs;
 
    if (ncorrs) {
        // corrs exist - resize vector
        std::vector<float>               vSamples(nSamples, 0.0);
        std::vector<std::vector<float> > vOFC(nPhases, vSamples);
        LArOFCP1  larOFCP1(0.0, 0.0, vOFC, vOFC);
 
        corrs.resize(ncorrs, LArOFCTransType::CorrectionPair(0, larOFCP1));
    }
 
    // Loop over corrections
    for (unsigned int i = 0; i < ncorrs; ++i){
        // check indexes
        if (dataIndex  >= persObj->m_vOFC_a.size()    ||
            dataIndex  >= persObj->m_vOFC_b.size()    ||
            timeIndex >= persObj->m_timeOffset.size() ||
            timeIndex >= persObj->m_timeBinWidth.size()) {
            log << MSG::ERROR 
                << "LArOFCSubsetCnv_p1::persToTrans - ofc index too large: dataIndex size ofc_a, size ofc_b, timeIndex timeOffset size, timeBinWidth size " 
                << dataIndex  << " " << persObj->m_vOFC_a.size() << " " 
                << persObj->m_vOFC_b.size() << " " << timeIndex << " "
                << persObj->m_timeOffset.size() << " " 
                << persObj->m_timeBinWidth.size()
            << endmsg;
            return;
        }
 
        // copy channel id
        corrs[i].first = persObj->m_subset.m_corrChannels[i];
 
        LArOFCP1& ofcs = corrs[i].second;
        LArOFCP1 tmp (persObj->m_timeOffset[timeIndex],
                      persObj->m_timeBinWidth[timeIndex],
                      nPhases,
                      nSamples,
                      persObj->m_vOFC_a,
                      persObj->m_vOFC_b,
                      dataIndex);
        ofcs.setFrom (tmp);
        ++timeIndex;
        dataIndex += nPhases * nSamples;
    }
    transObj->insertCorrections (std::move (corrs));
 
    transObj->shrink_to_fit();
}
 
 
void
LArOFCSubsetCnv_p1::transToPers(const LArOFCTransType* transObj, 
                                LArOFCPersType* persObj, 
                                MsgStream &log) const
{
    // Copy conditions
 
    // We copy all ofcs into a few simple vectors. 
    // For the conditions, there are two situations to treat: 
    //   1) dense data: normal conditions where each feb has 128
    //      channels and all channels have data,
    //   2) sparse data: conditions data where some channels are
    //      missing data. This is true for MC conditions (only some
    //      channels have data, and symmetry is used to obtain
    //      conditions for the rest of the channels), as well for
    //      'normal' conditions it may happen that some channels may
    //      be missing data.
    //
    // Treating 1) is straight-forward. For 2) we need to keep track
    // of which channels are present. We do so with
    // m_subset.m_febsWithSparseData where we store the febid followed by
    // four unsigned ints which contain the full bit pattern of the
    // channels set (i.e. bits 0-127).
    //
    // Note that one may also have a subset with all channels missing
    // data. In this case, we do not write out the empty subset.
    //
    // Finally, for corrections, we save the channel ids in
    // m_subset.m_corrChannels and the ofcs in the same vectors as the
    // rest of the conditions data.
    //
    // For each channel with data, the number of ofcs is assumed
    // constant. This is calculated at the beginning, along with
    // whether a feb is sparse or not.
    //
 
    // Get the number of channels, corrections and the size of ofc vectors
    unsigned int nsubsetsNotEmpty = 0;
    unsigned int ncorrs           = transObj->correctionVecSize();
    const unsigned int nChannelsPerFeb  = transObj->channelVectorSize();
    unsigned int nchans           = 0;
    unsigned int nPhases          = 0;
    unsigned int nSamples         = 0;
    bool foundOFCs                = false;
    std::vector<unsigned int> febsWithSparseData;
 
    // Find the number of ofcs and check for sparse conditions,
    // e.g. MC conditions
    const auto subsetEnd = transObj->subsetEnd();
    for (auto subsetIt = transObj->subsetBegin();
         subsetIt != subsetEnd;
         ++subsetIt)
    {
        unsigned int nfebChans = (*subsetIt).second.size();
 
        if (nfebChans != 0 && nfebChans != nChannelsPerFeb) {
            log << MSG::ERROR 
                << "LArOFCSubsetCnv_p1::transToPers - found incorrect number of channels per feb: " << nfebChans
                << endmsg;
            return;
        }
        if (nfebChans) ++nsubsetsNotEmpty; // count number of non-empty subsets
 
        // Loop over channels and check if this subset has sparse data
        bool subsetIsSparse = false;
        for (unsigned int j = 0; j < nfebChans; ++j) {
            LArOFCTransType::ConstReference ofc = 
              (*subsetIt).second[j];
            if (ofc.OFC_aSize() == 0) {
                if (!subsetIsSparse) {
                    // save febids for sparse subsets
                    subsetIsSparse = true;
                    febsWithSparseData.push_back((*subsetIt).first);
                }
            }
            else {
                nchans++; // count number of channels
                if (!foundOFCs) {
                    // Save the number of phases and samples for each
                    // ofc from first channels present
                    nPhases       = ofc.OFC_aSize();
                    nSamples      = ofc.OFC_a(0).size();
                    foundOFCs     = true;
                }
            }
        }
    }
    if (!foundOFCs && ncorrs>0) {
        // Save the number of phases and samples for each ofc from
        // first correction - couldn't find it from channels
        const LArOFCP1& ofc = transObj->correctionVecBegin()->second;
        nPhases             = ofc.OFC_aSize();
        nSamples            = ofc.OFC_a(0).size();
    }
 
    // Save sizes
    persObj->m_nPhases    = nPhases;
    persObj->m_nSamples   = nSamples;
 
    // Reserve space in vectors
    persObj->m_subset.m_febIds.reserve(nsubsetsNotEmpty);
    persObj->m_subset.m_corrChannels.reserve(ncorrs);
    unsigned int ndataTot  = (nchans + ncorrs)*nPhases*nSamples;
    unsigned int nTime     = (nchans + ncorrs);
    persObj->m_vOFC_a.reserve(ndataTot);
    persObj->m_vOFC_b.reserve(ndataTot);
    persObj->m_timeOffset.reserve(nTime);
    persObj->m_timeBinWidth.reserve(nTime);
 
    // For subsets with sparse data, reserve space for identifying
    // channels written out:
    //   1 - febid
    //   4 - for 128 bits (4*32)
    if (febsWithSparseData.size())
        persObj->m_subset.m_febsWithSparseData.reserve(febsWithSparseData.size()*5);
 
    // Copy conditions in subset
    unsigned int isparse = 0;
    for (auto subsetIt = transObj->subsetBegin();
         subsetIt != subsetEnd;
         ++subsetIt)
    {
        unsigned int nfebChans = (*subsetIt).second.size();
 
        // skip subsets without any channels
        if (nfebChans == 0) continue;
        
        unsigned int febid = (*subsetIt).first;
        persObj->m_subset.m_febIds.push_back(febid);
 
 
        bool isSparse = false;
        if (isparse < febsWithSparseData.size() && 
            febsWithSparseData[isparse] == febid) {
            // sparse subset, save channels with data
            isparse++;
            isSparse = true;
            // save febid
            persObj->m_subset.m_febsWithSparseData.push_back(febid);            
        }
            
        // Now loop over channels and save ofc and times
        unsigned int chansSet    = 0;
        unsigned int chansOffset = 0;
        for (unsigned int j = 0; j < nfebChans; ++j){
 
            bool saveOFCs = true;
            if (isSparse) {
                // subset with sparse data
          if ((*subsetIt).second[j].OFC_aSize() > 0) {
                    // store the channel number in bit map
                    assert (j >= chansOffset && (j - chansOffset) <= 31);
                    chansSet |= (1 << (j - chansOffset));
                }
                else {
          saveOFCs = false;
                }
                // Save chansSet
                if  (j == (chansOffset + 31) || j == nfebChans-1) {
                    persObj->m_subset.m_febsWithSparseData.push_back(chansSet);
                    chansSet    =   0;
                    chansOffset += 32;
                }
            }
            if (saveOFCs) {
            bool tooSmall=false;
        // Loop over phases and samples per channel
                for (unsigned int k = 0; k < nPhases; ++k) {
                    for (unsigned int l = 0; l < nSamples; ++l) {
              //check if data object is big enough
              if (k >= (*subsetIt).second[j].OFC_aSize() ||
                          l >= (*subsetIt).second[j].OFC_a(k).size())
                      {
            tooSmall=true;
            persObj->m_vOFC_a.push_back(0.);
            persObj->m_vOFC_b.push_back(0.);
              }
              else {
                        persObj->m_vOFC_a.push_back((*subsetIt).second[j].OFC_a(k)[l]);
                        persObj->m_vOFC_b.push_back((*subsetIt).second[j].OFC_b(k)[l]);
//                      std::cout << "WL Data: FEB=" << std::hex << febid << std::dec << " [" << i << "] Channel=" 
//                << j << " Phase="<< k<< " Sample " << l << " OFC=" 
//                << (*subsetIt).second[j].m_vOFC_a[k][l] << std::endl;
              }
                    }
                }
                // set time offset and binwidth
                persObj->m_timeOffset.push_back((*subsetIt).second[j].timeOffset());
                persObj->m_timeBinWidth.push_back((*subsetIt).second[j].timeBinWidth());
        if (tooSmall)
          log << MSG::ERROR << "Feb 0x" << std::hex << febid << std::dec << " channel " << j <<": OFC object too small. Expected " 
              << nPhases << " phases and " << nSamples << " samples. Padded with 0.0" << endmsg;
 
            }// end if saveOFC
 
//             static unsigned int nch = 0;
//             ++nch;
//             std::cout << "transToPers - i, j, save " << i << " " << j << " " 
//                       << saveOFCs << " " << nch << " febid " << febid
//                       << " chansSet " << std::hex << chansSet << std::dec
//                       << " chansOffset " << chansOffset
//                       << std::endl;
 
        }
    }
 
    // Copy corrections
    const auto corrEnd = transObj->correctionVecEnd();
    for (auto corrIt = transObj->correctionVecBegin();
         corrIt != corrEnd;
         ++corrIt)
    {
        persObj->m_subset.m_corrChannels.push_back(corrIt->first);
        // OFCs
    bool tooSmall=false;
    // Loop over phases and samples per channel
        for (unsigned int k = 0; k < nPhases; ++k) {
            for (unsigned int l = 0; l < nSamples; ++l) {
           //check if data object is big enough
          if (k >= corrIt->second.OFC_aSize() ||
                  l >= corrIt->second.OFC_a(k).size())
              {
        tooSmall=true;
        persObj->m_vOFC_a.push_back(0.);
        persObj->m_vOFC_b.push_back(0.);
          }
          else {
                persObj->m_vOFC_a.push_back(corrIt->second.OFC_a(k)[l]);
                persObj->m_vOFC_b.push_back(corrIt->second.OFC_b(k)[l]);
          }
        }
        }
        // set time offset and binwidth
        persObj->m_timeOffset.push_back(corrIt->second.timeOffset());
        persObj->m_timeBinWidth.push_back(corrIt->second.timeBinWidth());
    if (tooSmall)
      log << MSG::ERROR << "Correction (channel 0x" << std::hex << corrIt->first << std::dec <<  
        "): OFC object too small. Expected " << nPhases << " phases and " << nSamples << " samples. Padded with 0.0" << endmsg;
    }
 
    // Copy the rest
    persObj->m_subset.m_gain          = transObj->gain(); 
    persObj->m_subset.m_channel       = transObj->channel();
    persObj->m_subset.m_groupingType  = transObj->groupingType();
    
}