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