ClusterFitter3D.h

Go to the documentation of this file.

#pragma once
 
#include <TF3.h>
#include <TGraph2D.h>
#include <TH3D.h>
#include <TCanvas.h>
#include <TStyle.h>
#include <TMinuit.h>
#include <TFitResult.h>
 
#include "AMSTrdMCTrack.h"
#include "TRDConstants.h"
#include "Line.h"
 
 
namespace ClusterFitter3D
{    
    Line::FitLine3D fitClustersToLine3D(const AMSTrdMCTrack::Track &track, const int& eventID, TFile * file)
    {
        Line::FitLine3D result;
        result.trkID = track.trkID;
        result.g3PID = track.g3PID;
        
        TH3D * hist3D = setClustersInTH3D(track);
 
        TF3 * lin_fit_3D = new TF3("lin_fit_3D", Line::distanceFromLine3D, -100, 100, -100, 100, 90, 140, 6); 
        
        lin_fit_3D->SetParameters(0, 0, 100, 1, 1, 1);  
 
        TFitResultPtr fitResult = hist3D->Fit(lin_fit_3D, "SV"); 
        
        if (!fitResult)  
            printf("\n\nWARNING: Fit failed for Track with trkID: %d and g3PID: %d\n", result.trkID, result.g3PID);
 
        double chi2Ndf = fitResult->Chi2() / fitResult->Ndf();
        printf("For track.trkID = %d, chi2/NDF = %f\n\n\n", track.trkID, chi2Ndf);
        
        // if (file && result.fitResult)
        {
            file->cd();
            hist3D->Write(Form("graph2D_%d_%d_%d_%d", eventID, track.trkID, track.g3PID, track.clusters.size()));
            lin_fit_3D->Write(Form("fit3D_%d_%d_%d_%d", eventID, track.trkID, track.g3PID, track.clusters.size()));
        }
        
        delete lin_fit_3D;
        delete hist3D;
        
        result.fit = fitResult;
        return result;
    }
}