| @@ -372,6 +372,7 class HDFWriter(Operation): | |||
|
|
372 | 372 | #----------------------- |
|
|
373 | 373 | Typename = None |
|
|
374 | 374 | mask = False |
|
|
375 | setChannel = None | |
|
|
375 | 376 | |
|
|
376 | 377 | def __init__(self): |
|
|
377 | 378 | |
| @@ -388,11 +389,12 class HDFWriter(Operation): | |||
|
|
388 | 389 | for key, value in kwargs.items(): |
|
|
389 | 390 | setattr(obj, key, value) |
|
|
390 | 391 | |
|
|
391 | def setup(self, path=None, blocksPerFile=10, metadataList=None, dataList=None, setType=None, description=None,type_data=None, localtime=True, **kwargs): | |
|
|
392 | def setup(self, path=None, blocksPerFile=10, metadataList=None, dataList=None, setType=None, description=None,type_data=None, localtime=True,setChannel=None, **kwargs): | |
|
|
392 | 393 | self.path = path |
|
|
393 | 394 | self.blocksPerFile = blocksPerFile |
|
|
394 | 395 | self.metadataList = metadataList |
|
|
395 | 396 | self.dataList = [s.strip() for s in dataList] |
|
|
397 | self.setChannel = setChannel | |
|
|
396 | 398 | self.setType = setType |
|
|
397 | 399 | if self.setType == "weather": |
|
|
398 | 400 | self.set_kwargs(**kwargs) |
| @@ -426,7 +428,11 class HDFWriter(Operation): | |||
|
|
426 | 428 | if hasattr(self.dataOut, self.dataList[i]): |
|
|
427 | 429 | dataAux = getattr(self.dataOut, self.dataList[i]) |
|
|
428 | 430 | if self.setType == 'weather' and self.dataList[i] == 'data_param': |
|
|
431 | if self.setChannel is None: | |
|
|
429 | 432 | dataAux = dataAux[:,self.weather_vars[self.weather_var],:] |
|
|
433 | else: | |
|
|
434 | dataAux = dataAux[self.setChannel,self.weather_vars[self.weather_var],:] | |
|
|
435 | dataAux = numpy.reshape(dataAux,(1,dataAux.shape[0],dataAux.shape[1])) | |
|
|
430 | 436 | dsDict['variable'] = self.dataList[i] |
|
|
431 | 437 | else: |
|
|
432 | 438 | log.warning('Attribute {} not found in dataOut'.format(self.dataList[i]), self.name) |
| @@ -679,6 +685,9 class HDFWriter(Operation): | |||
|
|
679 | 685 | else: |
|
|
680 | 686 | shape = (self.blocksPerFile, ) + dsInfo['shape'][1:] |
|
|
681 | 687 | for i in range(dsInfo['dsNumber']): |
|
|
688 | if dsInfo['dsNumber']==1: | |
|
|
689 | if self.setChannel==1: | |
|
|
690 | i=1 | |
|
|
682 | 691 | ds = sgrp.create_dataset( |
|
|
683 | 692 | self.getLabel(dsInfo['variable'], i), |
|
|
684 | 693 | shape, |
General Comments 0
You need to be logged in to leave comments.
Login now