tests/src/Realignment.cpp

Go to the documentation of this file.

#include "MapsTrackManager.hh"
#include "MapsTrack.hh"
#include "MapsSensor.hh"
#include "MapsException.hh"
#include <cmath>
#include <algorithm>
#include <iostream>
#include "TRandom2.h"
#include "TFile.h"
#include "TH1F.h"


/** Realignment.cpp
 * 
 * Extremely useful tool for reprocessing stored tracks, without having to recreate them
 * from a DAQ binary file. One reads in a file of stored tracks, realigns the sensors,
 * and writes the new data in a new ROOT file.
 * 
 * At the moment, all parameters are hardcoded, except for the filenames; copy and 
 * edit this file as requried.
 * 
 * Jamie Ballin, Imperial College London
 *              February 2008
 * 
 * j.ballin06@imperial.ac.uk
 */
int main(int argc, const char **argv) {

        if (argc != 3) {
                std::cout << "Usage: Realignment <input file> <outputfile>\n";
                return 0;
        }


        MapsTrackManager mtm2;
        std::cout << "Recreating from root file...\n";
        std::cout << "NOTE: that this tool uses hardcoded parameters! Adjust and recompile for your own purposes...\n";
        char input[100];
        strcpy(input, argv[1]);
        mtm2.recreateFromRootFile(input);

        std::vector<MapsSensor*>& sensors = mtm2.getSensors();

        for (std::vector<MapsSensor*>::iterator it = sensors.begin(); it
                        != sensors.end(); it++) {
                MapsSensor* s = *it;
                MapsSensor::physXY align;
                //Adjust these switches as required.
                switch (s->id()) {
                //Sensors 8 and 6 define the world coordinate system for the DESY beam test runs
                case 8: {
                        mtm2.setRequiredLeftSensor(s);
                        break;
                }
                case 6: {
                        s->setPhi(0.0);
                        mtm2.setRequiredRightSensor(s);
                        break;
                }
                case 2: {
                        s->setPhi(3.14159265358979323846);
                        align.first = 0.09233;
                        align.second = 0.1428;
                        break;
                }
                case 7: {
                        align.first = 0.1513;
                        align.second = -0.1024;
                        break;
                }
                }
                s->setAlignment(align);

        }

        std::vector<MapsTrack*>& tracks = mtm2.getTracks();
        for (std::vector<MapsTrack*>::iterator it = tracks.begin(); it
                        != tracks.end(); it++) {
                MapsTrack* t = *it;
                std::pair<double, double> error(0.018, 0.018);
                t->setTrackError(error);
                //diagnose(std::cout, *t);
        }

        std::cout << "Rewriting to new root file...\n";
        char output[100];
        strcpy(output, argv[2]);
        mtm2.exportToRootFile(output);

        std::cout << "Done.\n";
        return 0;
}

---