Exemplo n.º 1
0
def main():
    scpm.init_logger()

    sampler = Sampler()
    try:
        sampler.open()
        sampler.start()
        try:
            calibration = scpm.load_calibration_file()
        except:
            calibration = None
        adc_data = sampler.adc_data_queue.get()
    except KeyboardInterrupt:
        sampler.terminate()
    else:
        sampler.terminate()
        adc_data = scpm.join_frames_and_widen(adc_data)
        split_adc_data = scpm.split_channels(adc_data)
        adc_rms = scpm.calculate_adc_rms(split_adc_data)
        if calibration is not None:
            calcd_data = scpm.calculate_calibrated_power(
                split_adc_data, adc_rms, calibration)
            print("")
            scpm.print_power(calcd_data)
        voltage, current = scpm.convert_adc_to_numpy_float(split_adc_data)
        voltage, current = scpm.shift_phase(voltage, current, calibration)
        scpm.plot(voltage, current, calibration)
        logging.shutdown()
Exemplo n.º 2
0
def main():
    scpm.init_logger()
    
    sampler = Sampler()
    try:
        sampler.open()
        sampler.start()
        try:
            calibration = scpm.load_calibration_file()
        except:
            calibration = None
        adc_data = sampler.adc_data_queue.get()
    except KeyboardInterrupt:
        sampler.terminate()
    else:
        sampler.terminate()
        adc_data = scpm.join_frames_and_widen(adc_data)
        split_adc_data = scpm.split_channels(adc_data)
        adc_rms = scpm.calculate_adc_rms(split_adc_data)
        if calibration is not None:
            calcd_data = scpm.calculate_calibrated_power(split_adc_data, 
                                                         adc_rms, calibration)
            print("")
            scpm.print_power(calcd_data)
        voltage, current = scpm.convert_adc_to_numpy_float(split_adc_data)
        voltage, current = scpm.shift_phase(voltage, current, calibration)
        scpm.plot(voltage, current, calibration)
        logging.shutdown()
Exemplo n.º 3
0
 def __init__(self):
     self.wavfile = None
     self.sox_process = None # a subprocess.Popen object
     self.wavfile_name = None
     self.sampler = Sampler()
     
     # Initialise zmq context and socket
     self.context = zmq.Context()
     self.socket = self.context.socket(zmq.PUB)
Exemplo n.º 4
0
def main():
    scpm.init_logger()
    sampler = Sampler()
    wu = WattsUp()

    try:
        sampler.open()
        sampler.start()
        wu.open()
        calibration = scpm.load_calibration_file()
        log.info("Comparing SCPM data with WattsUp data..."
                 " press CTRL+C to stop.")
        while True:
            wu_data = wu.get()  # blocking
            if wu_data is None:
                log.warn("wu_data is none!")
                continue

            time.sleep(config.RECORD_SECONDS)
            # Now get ADC data recorded at wu_data.time
            adc_data = scpm.find_time(sampler.adc_data_queue, wu_data.time)
            if adc_data:
                print("")
                adc_data = scpm.join_frames_and_widen(adc_data)
                split_adc_data = scpm.split_channels(adc_data)
                adc_rms = scpm.calculate_adc_rms(split_adc_data)
                calcd_data = scpm.calculate_calibrated_power(
                    split_adc_data, adc_rms, calibration)

                scpm.print_power(calcd_data, wu_data)
            else:
                log.warn("Could not find ADC data for time {}".format(
                    wu_data.time))

    except KeyboardInterrupt:
        pass
    except Exception:
        log.exception("")
        wu.terminate()
        sampler.terminate()
        raise

    wu.terminate()
    sampler.terminate()
Exemplo n.º 5
0
def main():
    scpm.init_logger()
    sampler = Sampler()
    wu = WattsUp()

    try:
        sampler.open()
        sampler.start()
        wu.open()
        scpm.calibrate(sampler.adc_data_queue, wu)
    except KeyboardInterrupt:
        pass
    except Exception:
        log.exception("")
        wu.terminate()
        sampler.terminate()
        raise

    wu.terminate()
    sampler.terminate()
Exemplo n.º 6
0
def main():
    scpm.init_logger()
    sampler = Sampler()
    wu = WattsUp()
    
    try:
        sampler.open()
        sampler.start()
        wu.open()        
        calibration = scpm.load_calibration_file()
        log.info("Comparing SCPM data with WattsUp data..."
                 " press CTRL+C to stop.")
        while True:
            wu_data = wu.get() # blocking
            if wu_data is None:
                log.warn("wu_data is none!")
                continue
            
            time.sleep(config.RECORD_SECONDS)
            # Now get ADC data recorded at wu_data.time
            adc_data = scpm.find_time(sampler.adc_data_queue, wu_data.time)
            if adc_data:
                print("")
                adc_data = scpm.join_frames_and_widen(adc_data)
                split_adc_data = scpm.split_channels(adc_data)
                adc_rms = scpm.calculate_adc_rms(split_adc_data)
                calcd_data = scpm.calculate_calibrated_power(split_adc_data, 
                                                        adc_rms, calibration)
                
                scpm.print_power(calcd_data, wu_data)
            else:
                log.warn("Could not find ADC data for time {}"
                         .format(wu_data.time))

    except KeyboardInterrupt:
        pass
    except Exception:
        log.exception("")
        wu.terminate()
        sampler.terminate()
        raise
    
    wu.terminate()
    sampler.terminate()
Exemplo n.º 7
0
def main():
    scpm.init_logger()
    sampler = Sampler()
    wu = WattsUp()
    
    try:
        sampler.open()
        sampler.start()
        wu.open()        
        scpm.calibrate(sampler.adc_data_queue, wu)
    except KeyboardInterrupt:
        pass
    except Exception:
        log.exception("")
        wu.terminate()
        sampler.terminate()
        raise
    
    wu.terminate()
    sampler.terminate()
Exemplo n.º 8
0
class Recorder(object):
    
    def __init__(self):
        self.wavfile = None
        self.sox_process = None # a subprocess.Popen object
        self.wavfile_name = None
        self.sampler = Sampler()
        
        # Initialise zmq context and socket
        self.context = zmq.Context()
        self.socket = self.context.socket(zmq.PUB)

    def start(self):
        calibration = scpm.load_calibration_file()
        self.sampler.open()
        self.sampler.start()
        log.info("Recording power data. Press CTRL+C to stop.")

        # Bind to port
        self.socket.bind("tcp://*:%s" % config.BROADCAST_PORT)

        prev_conv_time = None # the last time sox was run
        while True:
            raw_adc_data = self.sampler.adc_data_queue.get()
            if raw_adc_data is None:
                log.warn("adc_data was None!")
                continue
            
            # Calculate real power, apparent power, v_rms
            adc_data = scpm.join_frames_and_widen(raw_adc_data) 
            split_adc_data = scpm.split_channels(adc_data)
            adc_rms = scpm.calculate_adc_rms(split_adc_data)
            power = scpm.calculate_calibrated_power(split_adc_data, adc_rms, 
                                                    calibration)

            # Dump power data to disk
            with open(config.DAT_FILENAME, 'a') as data_file:
                data_file.write('{:.1f} {:.2f} {:.2f} {:.2f}\n'
                                .format(adc_data.time, power.real_power,
                                        power.apparent_power, power.volts_rms))

            # Broadcast latest values
            self.socket.send_string('{:.1f} {:.2f} {:.2f} {:.2f}'
                                    .format(adc_data.time, power.real_power,
                                            power.apparent_power, power.volts_rms))
            
            # Check if it's time to create a new FLAC file
            t = datetime.datetime.utcfromtimestamp(adc_data.time)
            if self.wavfile is None or t.hour == (prev_conv_time.hour+1)%24:
                self._close_and_compress_wavfile()
                self.wavfile_name = (config.FLAC_FILENAME_PREFIX + 
                                     '{:.6f}'.format(adc_data.time)
                                     .replace('.', '_'))
                self.wavfile = scpm.get_wavfile(self.wavfile_name)
            else:
                # Poll the sox process to see if it has finished.
                # We don't really need to do this but it should
                # prevent the sox process shell from appearing as a 
                # zombie process.
                self._check_sox_return_code_no_wait()
                
            prev_conv_time = t
            
            # Write uncompressed wav data to disk
            self.wavfile.writeframes(b''.join(raw_adc_data.data))
            
    def _close_and_compress_wavfile(self):
        if self.wavfile is None:
            return

        self.wavfile.close()
        
        # Check if the previous conversion process has completed
        self._blocking_check_sox_process_has_completed()
        
        # Run new sox process to compress wav file                
        # sox is a high quality audio conversion program
        # the "rate -v -L" option forces very high quality
        # rate conversion with linear phase.
        cmd = ("sox --no-dither {filename}.wav"
               " --compression 8 {filename}.flac"
               " rate -v -L {downsampled_rate}"
               " && rm {filename}.wav"
               .format(filename=self.wavfile_name,
                       downsampled_rate=config.DOWNSAMPLED_RATE))

        log.info("Running: " + cmd)                                
        self.sox_process = subprocess.Popen(cmd, shell=True,
                                            stdout=subprocess.PIPE,
                                            stderr=subprocess.PIPE)

    def _check_sox_return_code_no_wait(self):
        if (self.sox_process is not None 
            and self.sox_process.returncode is None):
               
            self.sox_process.poll()
            if self.sox_process.returncode is not None:     
                if self.sox_process.returncode == 0:
                    log.info("Previous sox conversion successfully completed!")
                else:
                    log.warn("Previous sox conversion FAILED!")
                    
                # Log sox output
                stdout = self.sox_process.stdout.read()
                stderr = self.sox_process.stderr.read()
                if stdout:
                    log.debug("sox stdout: {}".format(stdout))
                if stderr:
                    log.debug("sox stderr: {}".format(stderr))
    

    def _blocking_check_sox_process_has_completed(self):
        if self.sox_process is None:
            return
        
        self._check_sox_return_code_no_wait()
        if self.sox_process.returncode is None:
            log.warn("Previous sox conversion has not terminated yet!"
                     " Waiting...")
            self.sox_process.wait()
            self._check_sox_return_code_no_wait()
    

    def terminate(self):
        log.info("Recorder shutdown.\n")
        
        self._close_and_compress_wavfile()
            
        if self.sampler is not None:
            self.sampler.terminate()
        
        self._blocking_check_sox_process_has_completed()

        self.socket.close()
        self.context.term()

        logging.shutdown()
Exemplo n.º 9
0
 def __init__(self):
     self.wavfile = None
     self.sox_process = None  # a subprocess.Popen object
     self.wavfile_name = None
     self.sampler = Sampler()
Exemplo n.º 10
0
class Recorder(object):
    def __init__(self):
        self.wavfile = None
        self.sox_process = None  # a subprocess.Popen object
        self.wavfile_name = None
        self.sampler = Sampler()

    def start(self):
        calibration = scpm.load_calibration_file()
        self.sampler.open()
        self.sampler.start()
        log.info("Recording power data. Press CTRL+C to stop.")

        prev_conv_time = None  # the last time sox was run
        while True:
            raw_adc_data = self.sampler.adc_data_queue.get()
            if raw_adc_data is None:
                log.warn("adc_data was None!")
                continue

            # Calculate real power, apparent power, v_rms
            adc_data = scpm.join_frames_and_widen(raw_adc_data)
            split_adc_data = scpm.split_channels(adc_data)
            adc_rms = scpm.calculate_adc_rms(split_adc_data)
            power = scpm.calculate_calibrated_power(split_adc_data, adc_rms,
                                                    calibration)

            # Dump power data to disk
            with open(config.DAT_FILENAME, 'a') as data_file:
                data_file.write('{:.1f} {:.2f} {:.2f} {:.2f}\n'.format(
                    adc_data.time, power.real_power, power.apparent_power,
                    power.volts_rms))

            # Check if it's time to create a new FLAC file
            t = datetime.datetime.utcfromtimestamp(adc_data.time)
            if self.wavfile is None or t.hour == (prev_conv_time.hour +
                                                  1) % 24:
                self._close_and_compress_wavfile()
                self.wavfile_name = (
                    config.FLAC_FILENAME_PREFIX +
                    '{:.6f}'.format(adc_data.time).replace('.', '_'))
                self.wavfile = scpm.get_wavfile(self.wavfile_name)
            else:
                # Poll the sox process to see if it has finished.
                # We don't really need to do this but it should
                # prevent the sox process shell from appearing as a
                # zombie process.
                self._check_sox_return_code_no_wait()

            prev_conv_time = t

            # Write uncompressed wav data to disk
            self.wavfile.writeframes(b''.join(raw_adc_data.data))

    def _close_and_compress_wavfile(self):
        if self.wavfile is None:
            return

        self.wavfile.close()

        # Check if the previous conversion process has completed
        self._blocking_check_sox_process_has_completed()

        # Run new sox process to compress wav file
        # sox is a high quality audio conversion program
        # the "rate -v -L" option forces very high quality
        # rate conversion with linear phase.
        cmd = ("sox --no-dither {filename}.wav"
               " --compression 8 {filename}.flac"
               " rate -v -L {downsampled_rate}"
               " && rm {filename}.wav".format(
                   filename=self.wavfile_name,
                   downsampled_rate=config.DOWNSAMPLED_RATE))

        log.info("Running: " + cmd)
        self.sox_process = subprocess.Popen(cmd,
                                            shell=True,
                                            stdout=subprocess.PIPE,
                                            stderr=subprocess.PIPE)

    def _check_sox_return_code_no_wait(self):
        if (self.sox_process is not None
                and self.sox_process.returncode is None):

            self.sox_process.poll()
            if self.sox_process.returncode is not None:
                if self.sox_process.returncode == 0:
                    log.info("Previous sox conversion successfully completed!")
                else:
                    log.warn("Previous sox conversion FAILED!")

                # Log sox output
                stdout = self.sox_process.stdout.read()
                stderr = self.sox_process.stderr.read()
                if stdout:
                    log.debug("sox stdout: {}".format(stdout))
                if stderr:
                    log.debug("sox stderr: {}".format(stderr))

    def _blocking_check_sox_process_has_completed(self):
        if self.sox_process is None:
            return

        self._check_sox_return_code_no_wait()
        if self.sox_process.returncode is None:
            log.warn("Previous sox conversion has not terminated yet!"
                     " Waiting...")
            self.sox_process.wait()
            self._check_sox_return_code_no_wait()

    def terminate(self):
        log.info("Recorder shutdown.\n")

        self._close_and_compress_wavfile()

        if self.sampler is not None:
            self.sampler.terminate()

        self._blocking_check_sox_process_has_completed()

        logging.shutdown()
Exemplo n.º 11
0
 def __init__(self):
     self.wavfile = None
     self.sox_process = None # a subprocess.Popen object
     self.wavfile_name = None
     self.sampler = Sampler()