def load_graph(self): path = self.path # path = os.path.join(self.root, self.filename) from pychron.lasers.power.power_map_processor import PowerMapProcessor pmp = PowerMapProcessor() if path.endswith('.h5') or path.endswith('.hdf5'): reader = H5DataManager() # reader = self._data_manager_factory() reader.open_data(path) else: with open(path, 'r') as f: reader = csv.reader(f) # trim off header reader.next() # self.graph = pmp.load_graph(reader) self.graph.width = 625 self.graph.height = 500 reader.open_data(path) z, _ = pmp._extract_h5(reader) if self.surf: self.graph3D.plot_data(z, func='surf', representation=self.representation, warp_scale=self.vertical_ex , outline=self.surf_outline ) if self.contour: self.graph3D.plot_data(z, func='contour_surf', contours=self.levels, warp_scale=self.vertical_ex, outline=self.contour_outline )
def load(self, path): pmp = PowerMapProcessor() reader = H5DataManager() reader.open_data(path) cg = pmp.load_graph(reader) self.beam_diameter, self.power = pmp.extract_attrs(['beam_diameter', 'power']) self.component = cg.plotcontainer self.was_executed = True self.processor = pmp