PndSttStrawMap.cxx

Go to the documentation of this file.

/*
 * PndSTTStrawMap.cxx
 *
 *  Created on: Apr 16, 2013
 *      Author: mertens
 */


#include "PndSttStrawMap.h"
#include "TClonesArray.h"
#include "PndSttTube.h"
#include "TMath.h"

#include <iostream>
#include <vector>

using std::cout;
using std::endl;
using std::vector;

PndSttStrawMap::PndSttStrawMap()
: fStrawMapInitialized(false), fTubeArray(0)
{
        GenerateAngles();
}

PndSttStrawMap::PndSttStrawMap(TClonesArray* const stt_tube_array)
: fStrawMapInitialized(false), fTubeArray(0)
{
        GenerateAngles();
        GenerateStrawMap(stt_tube_array);
}

void PndSttStrawMap::GenerateAngles()
{
        fSectorStart.push_back(1.57);
        fSectorEnd.push_back(2.62);
        fSectorStart.push_back(2.62);
        fSectorEnd.push_back(-2.62+2*TMath::Pi());
        fSectorStart.push_back(-2.62+2*TMath::Pi());
        fSectorEnd.push_back(-1.57+2*TMath::Pi());
        fSectorStart.push_back(-1.57+2*TMath::Pi());
        fSectorEnd.push_back(-0.52+2*TMath::Pi());
        fSectorStart.push_back(-0.52+2*TMath::Pi());
        fSectorEnd.push_back(0.52);
        fSectorStart.push_back(0.52);
        fSectorEnd.push_back(1.57);
}

void PndSttStrawMap::GenerateStrawMap(TClonesArray* const stt_tube_array)
{
        fTubeArray = stt_tube_array;

        vector<int> currentRow;
        int sector = 0;
        int lastsector = 0;
        int row = 0;
        fSectorOfStraw.push_back(-1);
        fRowOfStraw.push_back(-1);
        fAxialStraw.push_back(false);
        // cout << "Generating straw map for " << fTubeArray->GetEntriesFast() << " straws." << endl;
        for (int i = 1; i < fTubeArray->GetEntriesFast(); i++) {
                // cout << "Finding tube." << endl;
                PndSttTube* tube = GetTube(i);
                // cout << "Tube address: " << tube << endl;
                bool isaxial = (tube->GetWireDirection().Theta() < 0.001);
                // cout << "Axial Straw: " << isaxial << endl;
                // cout << "Wire Direction: ";
//              tube->GetWireDirection().Print();
                double phi = tube->GetPosition().Phi();
                // cout << "Phi: " << phi << endl;
                if (phi < 0) phi += 2*TMath::Pi();
                // cout << "Checking sector." << endl;
                while ( !( (phi > fSectorStart.at(sector)) && (phi < fSectorEnd.at(sector)) )
                                && !( (sector == 4) && ( (phi > fSectorStart.at(4)) || (phi < fSectorEnd.at(4)) ) ) ) {
                        sector++;
                        sector %= 6;
                }
                if (sector != lastsector) {
                        fStrawIndex[lastsector].push_back(currentRow);
                        currentRow.clear();
                        // cout << "SECTOR COMPLETE: Row " << row << " added to sector " << lastsector << endl;
                }
                if (sector < lastsector) row++;
                lastsector = sector;
                currentRow.push_back(i);
                fSectorOfStraw.push_back(sector);
                fRowOfStraw.push_back(row);
                fAxialStraw.push_back(isaxial);
                // cout << "Straw " << i << " added to " << sector << ", " << row << endl;
        }
        fStrawIndex[lastsector].push_back(currentRow);
        // cout << "STT COMPLETE. SECTOR COMPLETE: Row " << row << " added to sector " << lastsector << endl;

        fStrawMapInitialized = true;
}

int PndSttStrawMap::FindPhiSector(double phi) const
{
        int sector = 0;
        if (phi < 0) phi += 2*TMath::Pi();
        while ( !( (phi > fSectorStart.at(sector)) && (phi < fSectorEnd.at(sector)) )
                        && !( (sector == 4) && ( (phi > fSectorStart.at(4)) || (phi < fSectorEnd.at(4)) ) ) ) {
                sector++;
                sector %= 6;
        }
        return sector;
}

bool PndSttStrawMap::IsEdgeStraw(int strawindex) const
{
        if (GetRow(strawindex) == 0) return true;
        int endstraw = 0; //there is no straw index 0
        switch (GetSector(strawindex)) {
        case 0: endstraw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).front();
        break;
        case 2: endstraw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).back();
        break;
        case 3: endstraw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).front();
        break;
        case 5: endstraw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).back();
        break;
        default: return false;
        };
        return (strawindex == endstraw);
}

int PndSttStrawMap::IsSectorBorderStraw(int strawindex) const
{
        int endstrawcw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).front();
        int endstrawccw = GetStrawRow(GetSector(strawindex), GetRow(strawindex)).back();
        if (strawindex == endstrawcw) return -1;
        if (strawindex == endstrawccw) return 1;
        return 0;
}

bool PndSttStrawMap::IsAxialRow(int rowindex) const
{
        //only works if all straws in one row are of the same type
        //which is the case for the current geometry
        return IsAxialStraw(GetStrawRow(0, rowindex).at(0));
}

bool PndSttStrawMap::IsSkewedRow(int rowindex) const
{
        //only works if all straws in one row are of the same type
        //which is the case for the current geometry
        return IsSkewedStraw(GetStrawRow(0, rowindex).at(0));
}


ClassImp(PndSttStrawMap)