DST_MC/old_code/test_z.C

0001 #include "INTTReadTree.h"
0002 #include "INTTZvtx.h"
0003
0004 void test_z()
0005 {
0006     string input_directory = "/sphenix/user/ChengWei/sPH_dNdeta/dNdEta_INTT_MC";
0007     string file_name = "MC_ZF_1_30400";
0008     string out_folder_directory = input_directory + "/codetest3_" + file_name;
0009     string MC_list_name = "dst_INTTdNdEta.list";
0010     string tree_name = "EventTree";
0011
0012     pair<double, double> beam_origin = {-0.015, 0.012};
0013     pair<double, double> DCA_cut ={-0.5, 0.5};
0014     pair<double, double> zvtx_QA_width = {42.32, 68.12};
0015     double zvtx_QA_ratio = 0.00005;
0016     double phi_diff_cut = 0.11;
0017     int clu_sum_adc_cut = -1;
0018     int clu_size_cut = 4;
0019     int N_clu_cut = 10000;
0020     int N_clu_cutl = 20;
0021
0022     bool draw_event_display = true;
0023     double peek = 3.324;
0024     int data_type = 0; // note : MC
0025     int geo_mode_id = 0; // note : 0 -> perfect geo
0026     int zvtx_cal_require = 15;
0027
0028     INTTReadTree * INTTClu = new INTTReadTree(data_type, input_directory, MC_list_name, tree_name, clu_size_cut, clu_sum_adc_cut);
0029     INTTZvtx     * MCz     = new INTTZvtx(INTTClu -> GetRunType(), out_folder_directory, beam_origin, geo_mode_id, phi_diff_cut, DCA_cut, N_clu_cutl, N_clu_cut, zvtx_cal_require, zvtx_QA_width, zvtx_QA_ratio, draw_event_display, peek);
0030
0031     cout<<"Total event : "<<INTTClu -> GetNEvt()<<endl;
0032
0033     for (int event_i = 0; event_i < 1000; event_i ++)
0034     {
0035         INTTClu -> EvtInit(event_i);
0036         INTTClu -> EvtSetCluGroup();
0037
0038         MCz     -> ProcessEvt(
0039             event_i,
0040             INTTClu -> temp_sPH_inner_nocolumn_vec, INTTClu -> temp_sPH_outer_nocolumn_vec,
0041             INTTClu -> temp_sPH_nocolumn_vec, INTTClu -> temp_sPH_nocolumn_rz_vec,
0042             INTTClu -> GetNvtxMC(), INTTClu -> GetTrigZvtxMC(), INTTClu -> GetPhiCheckTag(), -1  // note : no bco_full for MC
0043         );
0044
0045         MCz -> ClearEvt();
0046         INTTClu -> EvtClear();
0047     }
0048
0049     MCz -> PrintPlots();
0050     MCz -> EndRun();
0051 }
0052
0053
0054         // cout<< INTTClu -> temp_sPH_inner_nocolumn_vec.size()<<" "
0055         //     << INTTClu -> temp_sPH_outer_nocolumn_vec.size()<<" "
0056         //     << INTTClu -> temp_sPH_nocolumn_vec.size()<<" "
0057         //     << INTTClu -> temp_sPH_nocolumn_rz_vec.size()<<" "
0058         //     << INTTClu -> GetNvtxMC()<<" "
0059         //     << INTTClu -> GetTrigZvtxMC()<<" "
0060         //     << INTTClu -> GetPhiCheckTag()<<" "<<endl;