Resolve "mc_info_extractor for neutrino reco chain"

Closes #15 (closed)

orcasong/mc_info_extr.py

@@ -130,7 +130,7 @@ def get_std_reco(blob):
 		'std_pos_x':std_pos_x,'std_pos_y':std_pos_y,'std_pos_z':std_pos_z,
 		'std_energy':std_energy,'std_lik_energy':lik_energy,'std_length':std_length,
 		}
-
+		
 	return std_reco_info
 	
 	
@@ -175,7 +175,7 @@ def get_real_data_info_extr(input_file):
 
 		""" 
 	
-		event_info = blob['EventInfo']
+		event_info = blob['EventInfo'][0]
 	
 		#add n_hits info for the cut
 		n_hits = len(blob['Hits'])
@@ -242,7 +242,7 @@ def get_random_noise_mc_info_extr(input_file):
 
 		track = {
 			'event_id': event_info.event_id[0],
-			'run_id': event_info.run_id,
+			'run_id': event_info.run_id[0],
 			'particle_type': 0,
 		}
 		
@@ -307,12 +307,12 @@ def get_neutrino_mc_info_extr(input_file):
 		""" 
 		
 		#get general info about the event
-		event_id = blob['EventInfo'].event_id
-		run_id = blob["EventInfo"].run_id
+		event_id = blob['EventInfo'].event_id[0]
+		run_id = blob["EventInfo"].run_id[0]
 		#weights for neutrino analysis  
-		weight_w1 = blob["EventInfo"].weight_w1
-		weight_w2 = blob["EventInfo"].weight_w2
-		weight_w3 = blob["EventInfo"].weight_w3
+		weight_w1 = blob["EventInfo"].weight_w1[0]
+		weight_w2 = blob["EventInfo"].weight_w2[0]
+		weight_w3 = blob["EventInfo"].weight_w3[0]
 		
 		# first, look for the particular neutrino index of the production
 		p = 0 #for ORCA4 (and probably subsequent productions)
@@ -339,7 +339,7 @@ def get_neutrino_mc_info_extr(input_file):
 				 'weight_w1': weight_w1,'weight_w2': weight_w2,'weight_w3': weight_w3,
 				 'n_gen':n_gen,
 				}				 
-
+		print(is_cc)
  		#get all the std reco info		
 		if has_std_reco:
 	
@@ -373,7 +373,7 @@ def get_muon_mc_info_extr(input_file):
 	#check if std reco is present
 	f = File(input_file,"r")
 	has_std_reco = "reco" in f.keys()
-	
+
 	#no n_gen here, but needed for concatenation
 	n_gen = 1
 	
@@ -400,9 +400,8 @@ def get_muon_mc_info_extr(input_file):
 
 		""" 
 		
-		# take for ORCA7 for example index 1, index 0 is the empty neutrino mc_track
-		event_id = blob['EventInfo'].event_id
-		run_id = blob["EventInfo"].run_id
+		event_id = blob['EventInfo'].event_id[0]
+		run_id = blob["EventInfo"].run_id[0]
 
 		p = 0 #for ORCA4 (and probably subsequent productions)