Subversion Repositories f9daq

#include "include/guide.h"

#include "src/raySimulator.cpp"

#include "TFile.h"

#include "TPolyMarker3D.h"

#include "TCanvas.h"

//extern int show_3d;

//extern int show_data;

// Set the simulation parameters

void Show3D(int b) {show_3d = b;}

void ShowData(int b) {show_data = b;}

// Set the initial detector parameters (a, b, d, activeDetector, n1, n2, n3, gap)

DetectorParameters parameters(3.0, 5.0, 3.0, 3.0, 1, 1.53, 1.52, TVector3(0.3, 0, 0));

//void SetLGType(int in = 1, int side = 1, int out = 0)

        //{detector->SetLGType(in, side, out);}

void SetLG(double SiPM0 = 3.0, double b0 = 5.0, double d0 = 3.0, double n10 = 1.0, double n20 = 1.53, double n30 = 1.52)

        { parameters.setGuide(SiPM0, b0, d0, n10, n20, n30); }

void SetGap(double x_gap0 = 0.3, double y_gap0 = 0.0, double z_gap0 = 0.0)

  { parameters.setGap(x_gap0, y_gap0, z_gap0); }

void SetGlass(double glassOn, double glassD)

  { parameters.setGlass(glassOn, glassD); };

void SetPlate(int plateOn = 1, double plateWidth = 1)

  { parameters.setPlate(plateOn, plateWidth); }

void SetFresnel(int fresnel = 1) { parameters.setFresnel(fresnel); }

int save_ary = 0;

//------------------------------------------------------------------------------------------

void Help()

{

        printf("void SetCenter(x, y, z)\n");

        printf("void SetLGType(in, side, out)\n");

        printf("SURF_DUMMY 0, ");

        printf("SURF_REFRA 1, ");

        printf("SURF_REFLE 2, ");

        printf("SURF_TOTAL 3, ");

        printf("SURF_IMPER 4\n");

        printf("void SetLG(SiPM0, M0, d0, n10, n20, n30, R0)\n");

        printf("void SetR(R0)\n");

        printf("void SetGlass(glass_on0, glass_d0)\n");

        printf("void SetGap(x_gap0, y_gap0, z_gap0)\n");

        printf("LGG(NN, theta)\n");

        printf("LGR(NN, theta, phi)\n");

        printf("LGI(NN, theta)\n");

        printf("LGI_gap(NN, min, max, steps, theta)\n");

        printf("LGR_gap(NN, min, max, steps, theta, phi)\n");

        printf("LGR_th(NN, min, max, steps, phi)\n");

        printf("LGG_th(NN, min, max, steps)\n");

        printf("LGI_ad(NN, mina, maxa, mind, maxd, steps(memory dependent), theta)");  

}

//------------------------------------------------------------------------------------------

int show_in_steps = 0;

void SetShowInSteps(int in = 1)

{                                                                                

        show_in_steps = in;

}

//------------------------------------------------------------------------------------------

int gs_set = 0;

void SetGS()

{                                                                                

        const UInt_t Number = 2;

        Double_t Red[Number]   = { 1.00, 0.00};

        Double_t Green[Number] = { 1.00, 0.00};

        Double_t Blue[Number]  = { 1.00, 0.00};

        Double_t Stops[Number] = { 0.00, 1.00};

        Int_t nb=50;

        if(!gs_set) {

                TColor::CreateGradientColorTable(Number,Stops,Red,Green,Blue,nb);

                gs_set = 1;

        }

}

//------------------------------------------------------------------------------------------

void Draw1(CDetector *detector, int gs = 0, double range = 0.0)

{

        if(gs) SetGS();

        TCanvas *cdata = new TCanvas("cdata", "2dscan", 350, 0, 500, 500);

        gStyle->SetOptTitle(0);

        gStyle->SetOptStat(0);

        cdata->cd(0);

                if(range > 0.1) {

                        ((detector->GetHLaser())->GetXaxis())->SetRangeUser(-range,range);

                        ((detector->GetHLaser())->GetYaxis())->SetRangeUser(-range,range);

                }

                (detector->GetHLaser())->SetTitle("Hits at LG entrance; x [mm]; y [mm]");

                (detector->GetHLaser())->Draw("COLZ");

        cdata->SaveAs("2dscan.gif");

        //gStyle->SetOptTitle(1);

        //gStyle->SetOptStat(1);

}

//------------------------------------------------------------------------------------------

TVector3 p_pol0(0.0, 0.0, 1.0);

void SetPol(double x, double y, double z)

{p_pol0.SetXYZ(x,y,z);}

void PolTest(double theta = 0.0)

{

  int p_type = 1;

  double p_n1 = 1.5;

  double p_n2 = 1.0;

        theta = 3.141593*theta/180.0; if(theta < 1e-6) theta = 1e-6;

        TVector3 p_pol;

        Init();

        double cx = 0.0;

        TVector3 vodnik_edge[4];

        double t = 3.0;

        vodnik_edge[0].SetXYZ(cx, t,-t); vodnik_edge[1].SetXYZ(cx, t, t);

        vodnik_edge[2].SetXYZ(cx,-t, t); vodnik_edge[3].SetXYZ(cx,-t,-t);

        /*

#define SURF_DUMMY 0

#define SURF_REFRA 1

#define SURF_REFLE 2

#define SURF_TOTAL 3

#define SURF_IMPER 4

        */

        CSurface *surf = new CSurface(p_type, vodnik_edge, p_n1, p_n2, 0.96); //surf->FlipN();

        surf->SetFresnel(0);

        surf->Draw();

        CRay *ray = new CRay(cx, 0.0, 0.0, TMath::Cos(theta), 0.0, TMath::Sin(theta));

        ray->SetColor(1);

        //p_pol = rotatey(p_pol0, -theta);

        p_pol = p_pol0; p_pol.RotateY(-theta);

        //printf("p_pol = "); printv(p_pol);

        ray->SetPolarization(p_pol);

        ray->DrawS(cx, -5.0);

        CRay *out = new CRay; out->SetColor(2);

        TVector3 *inters = new TVector3;

        int fate = surf->PropagateRay(*ray, out, inters);

        if(fate == 1) out->DrawS(cx, 5.0);

        CRay *pp = new CRay; pp->SetColor(3);

        pp->Set(ray->GetR(), p_pol);

        pp->DrawS(cx, 2.0);

        CRay *pn = new CRay; pn->SetColor(4);

        pn->Set(ray->GetR(), surf->GetN());

        pn->DrawS(cx, 3.0);

}

void ptt()

{

        for(double th = 0.0; th < 91.0; th += 5.0) {

                printf("%lf ", th);

                PolTest(th);

        }

}

//------------------------------------------------------------------------------------------

// Propagate single ray and show the statistics

void LGS(double xh = 0.0, double yh = 0.0, double zh = 0.0, double theta = 0.0, double phi = 0.0)

{

  CDetector *detector = new CDetector(CENTER, parameters);

        Init();

        double izkoristek = Single(detector, parameters, TVector3(xh,yh,zh),theta, phi);

        PrintGuideHead();

        PrintGuideStat(izkoristek);

        DrawData(detector, parameters, theta, izkoristek);

}

//------------------------------------------------------------------------------------------

// Propagate NN rays generated as grid and show the statistics

void LGG(int NN = 10, double theta = 0.0)

{

  CDetector *detector = new CDetector(TVector3(-2,0,0), parameters);

        Init();

        double izkoristek = Grid(detector, parameters, NN, theta);

        //printf("izkoristek = %.3lf\n", izkoristek);

        PrintGuideHead();

        PrintGuideStat(izkoristek);

        DrawData(detector, parameters, theta, izkoristek);

}

//------------------------------------------------------------------------------------------

// Propagate NN rays genarated under the same angle theta, phi with random spacing and statistics

void LGR(int NN = 1e4, double theta = 0.0, double phi = 0.0)

{

  CDetector *detector = new CDetector(TVector3(-2,0,0), parameters);

        Init();

        double izkoristek = RandYZ(detector, parameters, NN, theta, phi);

        //printf("izkoristek = %.3lf\n", izkoristek);

        PrintGuideHead(); PrintGuideStat(izkoristek);

        DrawData(detector, parameters, theta, izkoristek);

}

//------------------------------------------------------------------------------------------

// Propagate NN rays isotropically generated in solid angle theta and show the statistics

void LGI(int NN = 1e4, double theta = 30.0, int nnrays = 30, int showr = 0)

{

  Init();

  CDetector *detector = new CDetector(CENTER, parameters);

  //CDetector detector = new CDetector();

        double izkoristek = RandIso(detector, parameters, NN, theta, nnrays, showr);

        //printf("izkoristek = %.3lf\n", izkoristek);

        PrintGuideHead(); PrintGuideStat(izkoristek);

        DrawData(detector, parameters, theta, izkoristek);

        //TCanvas *canvasDetector = new TCanvas("canvasDetector","canvasDetector",500,500);

        //canvasDetector->cd();

        //detector->Draw();

}

//------------------------------------------------------------------------------------------

void LGI_gap(int NN = 1e4, double min = 0.0, double max = 2.0, const int steps = 10, double theta = 30.0)

{

int tmp3d = show_3d, tmpdata = show_data;

double step[steps], acc[steps];

        show_3d = 0; show_data = 0;

        //printf(" x_gap | acceptance \n");

        PrintGuideHead();

        for(int i=0; i<=steps; i++) {

                step[i] = min + i*(max - min)/(double)steps;

                parameters.setGap(step[i], 0.0, 0.0);

                CDetector *detector = new CDetector(CENTER, parameters);

                Init();

                acc[i] = RandIso(detector, parameters, NN, theta);

                //printf("%6.2lf |%6.1lf\n", step[i], acc[i]*100.0);

                PrintGuideStat(acc[i]);

        }

        //char sbuff[256];

        //sprintf(sbuff, "SiPM = %.1lf, M = %.1lf, d = %.1lf | gap(y,z) = (%.1lf, %.1lf) | #theta = %.1lf",

                //      detector->GetSiPM(), detector->GetM(), detector->GetD(), 

                //      (detector->GetVGap()).y(), (detector->GetVGap()).z(), theta);

        DrawAcc(steps+1, step, acc, (char*)"; x_gap;Acceptance", min, max);

        show_3d = tmp3d; show_data = tmpdata;

}

//------------------------------------------------------------------------------------------

//------------------------------------------------------------------------------------------

void LGR_th(int NN = 1e4, double min = 0.0, double max = 30.0, const int steps = 15, double phi = 5.0)

{

int tmp3d = show_3d, tmpdata = show_data;

double step[steps], acc[steps];

        show_3d = 0; show_data = 0;

        //printf(" theta | acceptance \n");

        PrintGuideHead();

        for(int i=0; i<=steps; i++) {

                Init();

                step[i] = min + i*(max - min)/(double)steps;

                CDetector *detector = new CDetector(TVector3(-2,0,0), parameters);

                acc[i] = RandYZ(detector, parameters, NN, step[i], phi);

                //printf("%6.1lf |%6.1lf\n", step[i], acc[i]*100.0);

                PrintGuideStat(acc[i]);

                //delete detector;

        }

        //char sbuff[256];

        //sprintf(sbuff, "SiPM = %.1lf, M = %.1lf, d = %.1lf | gap = (%.1lf, %.1lf, %.1lf)",

                //      detector->GetSiPM(), detector->GetM(), detector->GetD(), 

                //      (detector->GetVGap()).x(), (detector->GetVGap()).y(), (detector->GetVGap()).z());

        DrawAcc(steps+1, step, acc, (char*)"; #theta [deg];Acceptance", min, max);

        show_3d = tmp3d; show_data = tmpdata;

}

//------------------------------------------------------------------------------------------

// a vs. d

void LGI_ad(int NN = 1e4, double min = 2.5, double max = 3.5, double minD = 1, double maxD = 6, const int steps = 10, double theta = 30.0)

{

  int tmp3d = show_3d;

        int tmpdata = show_data;

        //char sbuff[256];

        show_3d = 0; // don't show simulations 

        show_data = 1;

        //double d  = detector->GetD();

        const double b = parameters.getB(); // upper side of LG

        //const double SiPM = 3.0; // the length of the detector itself

        //double reflectivity = detector->guide->getR();

        TH2F *hAcceptance = new TH2F("hAcceptance","Acceptance",steps-1,minD,maxD,steps-1,min,max);

        // Use the Fresnel eq. instead of fixed reflectivity 96%

        //detector->SetFresnel(1);

        printf("   d |   a  | Acceptance\n");

        for(int i=0; i<steps; i++) {

                const double d = hAcceptance->GetXaxis()->GetBinCenter(i);

                for(int j=0; j<steps; j++) {

                  const double a = hAcceptance->GetYaxis()->GetBinCenter(j);

                  const double M = b/a;

      parameters.setGuide(a, M, d);

                  CDetector *detector = new CDetector(CENTER, parameters);

                  //detector->guide->setLG(y, M, x);

                  //Init(); exclude simulation

                  double acceptance = RandIso(detector, parameters, NN, theta, 0, 0);

                  //double acceptance = Grid(NN, theta);

                  //double acceptance = -1.0;

                  hAcceptance->Fill(d, a, acceptance);

                  //printf("%.2lf | %.2lf | ", x,y);

                  //PrintGuideStat(izkoristek);

                  delete detector; //works fine, 50x50 grid takes ~4MB of RAM

                  }

        }

        TCanvas *cp = new TCanvas("Acceptance", "Acceptance", 0, 0, 640, 480);

        TVirtualPad *pacc = cp->cd(0);

        pacc->SetRightMargin(0.10);

        pacc->SetLeftMargin(0.10);

        pacc->SetTopMargin(0.10);

        TFile *file = new TFile("acceptance.root","RECREATE");

        hAcceptance->Write();

        file->Close();

        //delete file;

        //hAcceptance->SetContour(100);

        //gStyle->SetPalette(1,0);

        hAcceptance->SetTitle(";d [mm];a [mm]");

        hAcceptance->GetXaxis()->SetRangeUser(minD,maxD);

        hAcceptance->Draw("COLZ");

        show_3d = tmp3d; show_data = tmpdata;

}

//------------------------------------------------------------------------------------------

// collection efficiency vs length

// a changes, b rests the same, d changes

// still can't use this function, it has hardcoded tan 10 deg

// !very stupid!

// new 3D graph NEEDED

// d vs a vs acceptance

void LGI_d2(int NN = 1e4, double min = 0.5, double max = 3.0, const int steps = 25, double theta = 30.0)

{

  TVector3 gapGuideSiPM(0.3, 0, 0);

  CDetector *detector = new CDetector(TVector3(-2,0,0), parameters);

        int tmp3d = show_3d, tmpdata = show_data;

        double step[steps], acc[steps];

        double sipm_d, M0;

        char sbuff[256];

        show_3d = 1; show_data = 1;

        M0 = parameters.getM();

        PrintGuideHead();

        for(int i=0; i<=steps; i++) {

                step[i] = min + i*(max - min)/(double)steps;           

                sipm_d = M0 - 2.0*step[i]*0.199819;//tan 11.3 deg

                parameters.setGuide(sipm_d, M0/sipm_d, step[i]);

                //printf("sipm = %lf, M = %lf, d = %lf\n", detector->GetSiPM(), detector->GetM(), step[i]);

                printf("%.1lf | %.2lf | ", step[i], sipm_d);

                Init();

                acc[i] = RandIso(detector, parameters, NN, theta);

                PrintGuideStat(acc[i]);

                sprintf(sbuff, "d%2d.gif", i);

                //c3dview->SaveAs(sbuff);

        }

        //sprintf(sbuff, "SiPM = %.1lf, M = %.1lf, d = %.1lf | gap(y,z) = (%.1lf, %.1lf) | #theta = %.1lf",

                //      detector->GetSiPM(), detector->GetM(), detector->GetD(), 

                        //                                (detector->GetVGap()).y(), (detector->GetVGap()).z(), theta);

        //DrawAcc(steps+1, step, acc, sbuff, (char*)"; d;Acceptance", min, max);

        TCanvas *cp = new TCanvas("Acceptance", "Acceptance", 0, 0, 640, 480);

        TVirtualPad *pacc = cp->cd(0);

        pacc->SetRightMargin(0.05);

        pacc->SetLeftMargin(0.13);

        pacc->SetTopMargin(0.08);

        TGraph *gacceptance = new TGraph(steps+1, step, acc);

        gacceptance->SetTitle("; d [mm];Collection efficiency");

        gacceptance->SetMarkerStyle(8);

        gacceptance->SetLineColor(12);

        gacceptance->SetLineWidth(1);

        (gacceptance->GetXaxis())->SetRangeUser(min, max);

        //(gacceptance->GetYaxis())->SetRangeUser(ymin, ymax);

        gacceptance->Draw("ACP");

        show_3d = tmp3d; show_data = tmpdata;

}

//------------------------------------------------------------------------------------------

// Acceptance vs. the length

// The magnification ratio M rests the same all the time

void LGI_d(int NN = 1e4, double min = 1, double max = 6.0, const int steps = 25, double theta = 30.0)

{

        int tmp3d = show_3d, tmpdata = show_data;

        double step[steps], acc[steps];

        const double a = parameters.getA();

        const double magnif = parameters.getM();

        //show_3d = show_in_steps; show_data = 1;

        // Use Fresnel equations

        //detector->SetFresnel(1);

        // Set glass (n=1.5) at the exit

        //detector->SetGlass(1,0);

        PrintGuideHead();

        for(int i=0; i<=steps; i++) {

                step[i] = min + i*(max - min)/(double)steps;

                parameters.setGuide(a, magnif, step[i]);

                CDetector *detector = new CDetector(CENTER, parameters);

                //printf("sipm = %lf, M = %lf, d = %lf\n", detector->GetSiPM(), detector->GetM(), step[i]);

                printf("%.1lf | ", step[i]);

                Init();

                acc[i] = RandIso(detector, parameters, NN, theta);

                PrintGuideStat(acc[i]);

                delete detector;

        }

        //char sbuff[256];

        //sprintf(sbuff, "SiPM = %.1lf, M = %.1lf, d = %.1lf | gap(y,z) = (%.1lf, %.1lf) | #theta = %.1lf",

                //      detector->GetSiPM(), detector->GetM(), detector->GetD(), 

                        //                                (detector->GetVGap()).y(), (detector->GetVGap()).z(), theta);

        DrawAcc(steps+1, step, acc, (char*)"; d [mm];Acceptance", min, max);

        show_3d = tmp3d; show_data = tmpdata;

}

//------------------------------------------------------------------------------------------

// Magnification optimization. a and d are fixed, M and b change in steps

void LGI_M(int NN = 1e4, double min = 1.0, double max = 3.0, const int steps = 30, double theta = 30.0)

{

        int tmp3d = show_3d, tmpdata = show_data;

        double step[steps], acc[steps];

        const double a = parameters.getA();

        const double d = parameters.getD();

        show_3d = 0; show_data = 0;

        PrintGuideHead();

        for(int i=0; i<=steps; i++) {

                step[i] = min + i*(max - min)/(double)steps;

                parameters.setGuide(a, step[i], d);

                CDetector *detector = new CDetector(CENTER, parameters);

                Init();

                acc[i] = RandIso(detector, parameters, NN, theta);

                PrintGuideStat(acc[i]);

        }

        //char sbuff[256];

        //sprintf(sbuff, "SiPM = %.1lf, M = %.1lf, d = %.1lf | gap(y,z) = (%.1lf, %.1lf) | #theta = %.1lf",

                //      detector->GetSiPM(), detector->GetM(), detector->GetD(), 

                        //                                (detector->GetVGap()).y(), (detector->GetVGap()).z(), theta);

        DrawAcc(steps+1, step, acc, (char*)"; M;Acceptance", min, max);

        show_3d = tmp3d; show_data = tmpdata;

}

//------------------------------------------------------------------------------------------