tests/src/RealignCosmics.cpp

Go to the documentation of this file.

#include "MapsTrackManager.hh"
#include "MapsTrack.hh"
#include "MapsSensor.hh"
#include "MapsException.hh"
#include <cmath>
#include <vector>
#include "TRandom2.h"
#include "TFile.h"
#include "TH1F.h"
#include "TH2F.h"
int main(int argc, const char **argv) {

        for (int d(0); d < argc; d++) {
                std::cout << argv[d] << "\n";
        }
        if (argc != 3) {
                std::cout << "Usage: Rewriter <input file> <outputfile>\n";
                return 0;
        }

        MapsTrackManager mtm2;
        std::cout << "Recreating from root file...\n";
        char input[100];
        strcpy(input, argv[1]);

        mtm2.recreateFromRootFile(input);
        std::cout << "Rewriting to new root file...\n";
        std::vector<MapsSensor*> s(mtm2.getSensors());
        MapsSensor* left = 0;
        MapsSensor* right = 0;
        MapsSensor* m1 = 0;
        MapsSensor* m2 = 0;
        for (std::vector<MapsSensor*>::iterator it = s.begin(); it != s.end(); ++it) {
                MapsSensor* sensor = *it;
                if (sensor->id() == 2) {
                        left = sensor;

                }
                if (sensor->id() == 7) {
                        right = sensor;
                        //sensor->setPhi(1.5 * 3.14159265358979323846);
                }
                if (sensor->id() == 6) {
                        m1 = sensor;
                        //sensor->setPhi(0.5 * 3.14159265358979323846);
                }
                if (sensor->id() == 8) {
                        m2 = sensor;
                }

        }

        mtm2.setRequiredLeftSensor(left);
        mtm2.setRequiredRightSensor(right);
        MapsTrackManager mtm3;
        mtm3.setSensors(s);

        std::vector<MapsTrack*>& tracks = mtm2.getTracks();
        for (std::vector<MapsTrack*>::iterator it = tracks.begin(); it
                        != tracks.end(); it++) {
                MapsTrack* t = *it;
                t->eraseGlobalHits();
                std::pair<double, double> error(0.25, 0.25);
                t->setTrackError(error);
                MapsTrack threeHit;
                t->make3HitTrack(threeHit);
                if (threeHit.chiSqProb(0) > 0.05 && threeHit.chiSqProb(1) > 0.05) {
                        diagnose(std::cout, *t);

                        if (t->fourthSensor()->isDeadArea(t->findXYPrediction(t->fourthSensor()->zPosition())) && t->getHits().size() == 3) {
                                std::cout << "Three hit prediction falls in dead area.\n";
                        } else {
                                mtm3.addTrack(t);
                        }
                }
                t->tellSensorsOfResiduals(left, right);
        }

        TH2F s1, s2;
        TFile* f = new TFile("CosmicAlignment.root", "recreate");
        m1->getResidualXYPlot(s1);
        m2->getResidualXYPlot(s2);
        s1.Write();
        s2.Write();
        f->Write();
        f->Close();
        char output[100];
        strcpy(output, argv[2]);
        //mtm2.exportToRootFile(output);
        mtm3.exportToRootFile("GoodCosmicTracks.root");
        std::cout << "Done.\n";
        return 0;
}

---