664 def get_det_data_all(self,frag):
665 frh = frag.get_header()
666 trigger_number = frh.trigger_number
667
668 get_ana_data = (self.ana_data_prescale is not None and (trigger_number % self.ana_data_prescale)==0)
669 get_wvfm_data = (self.wvfm_data_prescale is not None and (trigger_number % self.wvfm_data_prescale)==0)
670
671
672
673 if not (get_ana_data or get_wvfm_data):
674 return None,None
675
676 ana_data = None
677 wvfm_data = None
678
679 adcs = self.unpacker.np_array_adc(frag)
680 det, crate, slot, stream = self.get_det_crate_slot_stream(frag)
681 channels = [ self.channel_map.get_offline_channel_from_det_crate_slot_stream_chan(det, crate, slot, stream, c) for c in range(self.N_CHANNELS_PER_FRAME) ]
682 planes = [ self.channel_map.get_plane_from_offline_channel(uc) for uc in channels ]
683 elements = [ self.channel_map.get_element_id_from_offline_channel(uc) for uc in channels ]
684 tde_chans = range(self.N_CHANNELS_PER_FRAME)
685
686 if get_ana_data:
687 adc_mean = np.mean(adcs,axis=0)
688 adc_rms = np.std(adcs,axis=0)
689 adc_max = np.max(adcs,axis=0)
690 adc_min = np.min(adcs,axis=0)
691 adc_median = np.median(adcs,axis=0)
692 ana_data = [ TDEEthAnalysisData(run=frh.run_number,
693 trigger=frh.trigger_number,
694 sequence=frh.sequence_number,
695 src_id=frh.element_id.id,
696 channel=channels[i_ch],
697 plane=planes[i_ch],
698 element=elements[i_ch],
699 tde_chan=tde_chans[i_ch],
700 adc_mean=adc_mean[i_ch],
701 adc_rms=adc_rms[i_ch],
702 adc_max=adc_max[i_ch],
703 adc_min=adc_min[i_ch],
704 adc_median=adc_median[i_ch]) for i_ch in range(self.N_CHANNELS_PER_FRAME) ]
705 if get_wvfm_data:
706 timestamps = self.unpacker.np_array_timestamp(frag)
707 ffts = np.abs(np.fft.rfft(adcs,axis=0))
708 wvfm_data = [ TDEEthWaveformData(run=frh.run_number,
709 trigger=frh.trigger_number,
710 sequence=frh.sequence_number,
711 src_id=frh.element_id.id,
712 channel=channels[i_ch],
713 plane=planes[i_ch],
714 element=elements[i_ch],
715 tde_chan=tde_chans[i_ch],
716 timestamps=timestamps,
717 adcs=adcs[:,i_ch],
718 fft_mag=ffts[:,i_ch]) for i_ch in range(self.N_CHANNELS_PER_FRAME) ]
719
720 return ana_data, wvfm_data
721