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