beamtime/cosy-jan12/go4/STS/TSTSTopology.cxx (r4864/r3122)

Go to the documentation of this file.

#include "TSTSTopology.h"
#include "stdlib.h"



TSTSTopology::~TSTSTopology()
{
    for( UInt_t iDet = 0; iDet < fMappings.size(); ++iDet ) {
        for( UInt_t iSide = 0; iSide < fMappings.at( iDet ).size(); ++iSide ) {
            for( UInt_t iConn = 0; iConn < fMappings.at( iDet).at( iSide ).size(); ++iConn ) {
                TF1 * & f = fMappings.at( iDet).at( iSide ).at( iConn );
                if( f ) {
                    delete f;
                    f = 0;
                }
            }
        }
    }
}



void TSTSTopology::SetConnection( UInt_t roc, UInt_t nx, UInt_t det, ESTSSide side, UInt_t conn )
{
    if( roc >= fRocNxConnections.size() ) {
        fRocNxConnections.resize( roc + 1 );
    }
    if( nx >= fRocNxConnections.at( roc ).size() ) {
        fRocNxConnections.at( roc ).resize( nx + 1 );
    }
    fRocNxConnections.at( roc ).at( nx ).fDet = det;
    fRocNxConnections.at( roc ).at( nx ).fSide = side;
    fRocNxConnections.at( roc ).at( nx ).fConn = conn;
    fRocNxConnections.at( roc ).at( nx ).fSet = kTRUE;
}



void TSTSTopology::SetMapping( UInt_t det, UInt_t side, UInt_t conn, const Char_t * mapping )
{
    if( side >= kNSides ) {
        printf( "Error in TSTSTopology::SetMapping: side is expected to be < 2, received %d\n", side );
        return;
    }
    if( det >= fMappings.size() ) {
        fMappings.resize( det + 1 );
    }
    if( side >= fMappings.at( det ).size() ) {
        fMappings.at( det ).resize( side + 1 );
    }
    if( conn >= fMappings.at( det ).at( side ).size() ) {
        fMappings.at( det ).at( side ).resize( conn + 1 );
    }
    fMappings.at( det ).at( side ).at( conn ) = new TF1( Form("mapping_det%d_side%d_conn%d", det, side, conn), mapping, 0, GetNNxCh() - 1 );
}



UInt_t TSTSTopology::GetRocArrSize() const
{
    for( UInt_t iRoc = fRocNxConnections.size() - 1; iRoc >=0 ; --iRoc ) {
        for( UInt_t iNx = 0; iNx < fRocNxConnections.at(iRoc).size(); ++iNx ) {
            if( fRocNxConnections.at( iRoc ).at( iNx ).fSet ) {
                return iRoc + 1;
            }
        }
    }
    return 0;
}



UInt_t TSTSTopology::GetNxArrSize() const
{
    UInt_t nxArrSize = 0;
    for( UInt_t iRoc = 0; iRoc < fRocNxConnections.size(); ++iRoc ) {
        nxArrSize= TMath::Max( nxArrSize, GetNxArrSize( iRoc ) );
    }
    return nxArrSize;
}



UInt_t TSTSTopology::GetNxArrSize( UInt_t roc ) const
{
    for( UInt_t iNx = fRocNxConnections.at( roc ).size() - 1; iNx >= 0; --iNx ) {
        if( fRocNxConnections.at( roc ).at( iNx ).fSet ) return iNx + 1;
    }
    return 0;
}



UInt_t TSTSTopology::GetDetArrSize() const
{
    UInt_t detArrSize = 0;
    for( UInt_t iRoc = 0; iRoc < fRocNxConnections.size(); ++iRoc ) {
        for( UInt_t iNx = 0; iNx < fRocNxConnections.at( iRoc ).size(); ++iNx ) {
            detArrSize = TMath::Max( fRocNxConnections.at( iRoc ).at( iNx ).fDet + 1, detArrSize );
        }
    }
    return detArrSize;
}



UInt_t TSTSTopology::GetDetector( UInt_t roc, UInt_t nx ) const
{
    return fRocNxConnections.at( roc ).at( nx ).fDet;
}



TSTSTopology::ESTSSide TSTSTopology::GetSide( UInt_t roc, UInt_t nx ) const
{
    return fRocNxConnections.at( roc ).at( nx ).fSide;
}



UInt_t TSTSTopology::GetConnector( UInt_t roc, UInt_t nx ) const
{
    return fRocNxConnections.at( roc ).at( nx ).fConn;
}



Double_t TSTSTopology::GetStripNoCheck( UInt_t det, UInt_t side, UInt_t conn, UInt_t ch ) const
{
    return fMappings.at( det ).at( side ).at( conn )->Eval( ch );
}



UInt_t TSTSTopology::GetStripCheckNegative( UInt_t det, UInt_t side, UInt_t conn, UInt_t ch ) const
{
    Double_t strip = GetStripNoCheck( det, side, conn, ch );
    if( strip < 0 ) {
        printf( "Error in TSTSTopology::GetStripCheckNegative: det=%d side=%c conn=%d ch=%d is mapped to negative strip %f\n", det, GetSideName( side ), conn, ch, strip );
        return 0;
    }
    return static_cast< UInt_t >( strip );
}



UInt_t TSTSTopology::GetStrip( UInt_t det, UInt_t side, UInt_t conn, UInt_t ch ) const
{
    return static_cast< UInt_t >( GetStripNoCheck( det, side, conn, ch ) );
}



UInt_t TSTSTopology::GetStrip( UInt_t roc, UInt_t nx, UInt_t ch ) const
{
    UInt_t det = GetDetector( roc, nx );
    ESTSSide side = GetSide( roc, nx );
    UInt_t conn = GetConnector( roc, nx );
    return GetStrip( det, side, conn, ch );
}



UInt_t TSTSTopology::GetStripArrSize( UInt_t det, UInt_t side ) const
{
    UInt_t stripArrSize = 0;
    for( UInt_t iConn = 0; iConn < fMappings.at( det ).at( side ).size(); ++iConn ) {
        if( ! IsMappingSet( det, side, iConn ) ) {
            continue;
        }
        for( UInt_t iCh = 0; iCh < GetNNxCh(); ++iCh ) {
            stripArrSize = TMath::Max( stripArrSize, GetStrip( det, side, iConn, iCh ) + 1 );
        }
    }
    return stripArrSize;
}


/*
 * True if at connection from this roc to at least one nX is specified.
 */
Bool_t TSTSTopology::IsValidRoc( UInt_t roc ) const
{
    if( roc > fRocNxConnections.size() || roc < 0 ) {
        return kFALSE;
    }
    for( UInt_t iNx = 0; iNx < fRocNxConnections.at( roc ).size(); ++iNx ) {
        if( fRocNxConnections.at( roc ).at( iNx ).fSet ) {
            return kTRUE;
        }
    }
    return kFALSE;
}



Bool_t TSTSTopology::IsValidNx( UInt_t roc, UInt_t nx ) const
{
    if( roc > fRocNxConnections.size() || roc < 0 ) {
        return kFALSE;
    }
    if( nx > fRocNxConnections.at( roc ).size() || nx < 0 ) {
        return kFALSE;
    }
    return fRocNxConnections.at( roc ).at( nx ).fSet;
}



Bool_t TSTSTopology::IsValidCh( UInt_t ch ) const
{
    return ch < GetNNxCh() ;
}



Bool_t TSTSTopology::IsMappingSet( UInt_t det, UInt_t side, UInt_t conn ) const
{
    if( det >= fMappings.size() ) {
        return kFALSE;
    }
    if( side >= fMappings.at( det ).size() ) {
        return kFALSE;
    }
    if( conn >= fMappings.at( det ).at( side ).size() ) {
        return kFALSE;
    }
    return fMappings.at( det ).at( side ).at( conn ) != 0;
}



void TSTSTopology::TestMapping( Bool_t doPrint ) const
{
    std::vector< std::vector < std::vector< TRocNxChConn > > > mapping;
    const UInt_t detArrSize = GetDetArrSize();
    mapping.resize( detArrSize );
    for( UInt_t iDet = 0; iDet < detArrSize; ++iDet ) {
        mapping.at( iDet ).resize( kNSides );
        for( UInt_t iSide = 0; iSide < kNSides; iSide++ ) {
            mapping.at( iDet ).at( iSide ).resize( GetStripArrSize( iDet, iSide ) );
        }
    }

    UInt_t rocArrSize = GetRocArrSize();
    for( UInt_t iRoc = 0; iRoc < rocArrSize; ++iRoc ) {
        if( IsValidRoc( iRoc ) ) {
            UInt_t nxArrSize = GetNxArrSize( iRoc );
            for( UInt_t iNx = 0; iNx < nxArrSize; ++iNx ) {
                if( IsValidNx( iRoc, iNx ) ) {
                    for( UInt_t iCh = 0; iCh < GetNNxCh(); ++iCh ) {
                        UInt_t det = GetDetector( iRoc, iNx );
                        UInt_t side = GetSide( iRoc, iNx );
                        UInt_t conn = GetConnector( iRoc, iNx );
                        UInt_t strip = GetStripCheckNegative( det, side, conn, iCh );
                        if( ! mapping.at( det ).at( side ).at( strip ).fSet ) {
                            mapping.at( det ).at( side ).at( strip ).fRoc = iRoc;
                            mapping.at( det ).at( side ).at( strip ).fNx  = iNx;
                            mapping.at( det ).at( side ).at( strip ).fCh  = iCh;
                            mapping.at( det ).at( side ).at( strip ).fConn= conn;
                            mapping.at( det ).at( side ).at( strip ).fSet = kTRUE;
                        }
                        else {
                            UInt_t oldRoc = mapping.at( det ).at( side ).at( strip ).fRoc;
                            UInt_t oldNx  = mapping.at( det ).at( side ).at( strip ).fNx;
                            UInt_t oldCh  = mapping.at( det ).at( side ).at( strip ).fCh;
                            UInt_t oldConn= mapping.at( det ).at( side ).at( strip ).fConn;
                            printf( "Error in TSTSTopology::TestMapping: ");
                            printf( "Two readout channels map to strip %d on %c-side of detector %d: ", strip, GetSideName( side ), det );
                            printf( " roc %d, nx %d, ch%d (connector %d) and ", oldRoc, oldNx, oldCh, oldConn );
                            printf( " roc %d, nx %d, ch%d (connector %d)\n", iRoc, iNx, iCh, conn );
                        }
                    }
                }
            }
        }
    }

    if( doPrint ) {
        UInt_t nNotConnectedStrips = 0;
        printf( "======================= Printing STS topology =================================\n");
        for( UInt_t iDet = 0; iDet < mapping.size(); ++iDet ) {
            for( UInt_t iSide = 0; iSide < mapping.at( iDet ).size(); ++iSide ) {
                for( UInt_t iStrip = 0; iStrip < mapping.at( iDet ).at( iSide ).size(); ++iStrip ) {
                    printf( "Det=%2d  side=%c  strip=%4d   <--   ", iDet, GetSideName( iSide ), iStrip );
                    const TRocNxChConn & rocNxChConn = mapping.at( iDet ).at( iSide ).at( iStrip );
                    if( rocNxChConn.fSet ) {
                        printf( "Roc=%2d  Nx=%2d  Ch=%3d  (Conn=%2d)\n", rocNxChConn.fRoc, rocNxChConn.fNx, rocNxChConn.fCh, rocNxChConn.fConn );
                    }
                    else {
                        printf( "not connected\n");
                        ++nNotConnectedStrips;
                    }
                }
            }
        }
        for( UInt_t iDet = 0; iDet < mapping.size(); ++iDet ) {
            for( UInt_t iSide = 0; iSide < mapping.at( iDet ).size(); ++iSide ) {
                UInt_t maxStrip = GetStripArrSize( iDet, iSide ) - 1;
                printf( "Max strip on %c-side of det %d is %d", GetSideName( iSide ), iDet, maxStrip );
                if( maxStrip != 255 && maxStrip != 1023 ) {
                    printf( " <-- WARNING!!!" );
                }
                printf("\n");
            }
        }
        if( nNotConnectedStrips ) {
            printf( "WARNING: %d strips are not connected!!!\n", nNotConnectedStrips );
        }
        printf("\n");
    }

}