コード例 #1
0
  def write(self, data):
    """
    Write database to file.

    @param data Dictionary of the following form:
      data = {'points': 2-D array (numLocs, spaceDim),
              'x': Array of x coordinates,
              'y': Array of y coordinates,
              'z': Array of z coordinates,
              'coordsys': Coordinate system associated with locations,
              'data_dim': Dimension of spatial distribution,
              'values': [{'name': Name of value,
                          'units': Units of value,
                          'data': Data for value (numLocs)}]}
    """
    import numpy

    (numLocs, spaceDim) = data['points'].shape
    numValues = len(data['values'])    
    names = []
    units = []
    values = numpy.zeros( (numLocs, numValues), dtype=numpy.float64)
    i = 0
    for value in data['values']:
      names.append(value['name'])
      units.append(value['units'])
      values[:,i] = value['data'][:]
      i += 1

    numX = data['x'].shape[0]
    numY = data['y'].shape[0]
    if data['coordsys'].spaceDim() == 2:
      numZ = 0
      if (numLocs != numX*numY):
        raise ValueError("Number of locations (%d) does not match coordinate dimensions (%d, %d)." % (numLocs, numX, numY))
    else:
      numZ = data['z'].shape[0]
      if (numLocs != numX*numY*numZ):
        raise ValueError("Number of locations (%d) does not match coordinate dimensions (%d, %d, %d)." % (numLocs, numX, numY, numZ))

    from SimpleGridDB import SimpleGridDB
    db = SimpleGridDB()
    db.inventory.label = "Temporary database for I/O"
    db.inventory.filename = self.filename
    db._configure()
    db.coordsys(data['coordsys'])
    db.allocate(numX, numY, numZ, numValues, spaceDim, data['data_dim'])
    db.x(data['x'])
    db.y(data['y'])
    if data['coordsys'].spaceDim() == 3:
      db.z(data['z'])
    db.data(data['points'], values)
    db.names(names)
    db.units(units)

    ModuleSimpleGridAscii.write(db)
    return
コード例 #2
0
    def write(self, data):
        """
    Write database to file.

    @param data Dictionary of the following form:
      data = {'points': 2-D array (numLocs, spaceDim),
              'x': Array of x coordinates,
              'y': Array of y coordinates,
              'z': Array of z coordinates,
              'coordsys': Coordinate system associated with locations,
              'data_dim': Dimension of spatial distribution,
              'values': [{'name': Name of value,
                          'units': Units of value,
                          'data': Data for value (numLocs)}]}
    """
        import numpy

        (numLocs, spaceDim) = data['points'].shape
        numValues = len(data['values'])
        names = []
        units = []
        values = numpy.zeros((numLocs, numValues), dtype=numpy.float64)
        i = 0
        for value in data['values']:
            names.append(value['name'])
            units.append(value['units'])
            values[:, i] = value['data'][:]
            i += 1

        numX = data['x'].shape[0]
        numY = data['y'].shape[0]
        if data['coordsys'].spaceDim() == 2:
            numZ = 0
            if (numLocs != numX * numY):
                raise ValueError(
                    "Number of locations (%d) does not match coordinate dimensions (%d, %d)."
                    % (numLocs, numX, numY))
        else:
            numZ = data['z'].shape[0]
            if (numLocs != numX * numY * numZ):
                raise ValueError(
                    "Number of locations (%d) does not match coordinate dimensions (%d, %d, %d)."
                    % (numLocs, numX, numY, numZ))

        from SimpleGridDB import SimpleGridDB
        db = SimpleGridDB()
        db.inventory.label = "Temporary database for I/O"
        db.inventory.filename = self.filename
        db._configure()
        db.coordsys(data['coordsys'])
        db.allocate(numX, numY, numZ, numValues, spaceDim, data['data_dim'])
        db.x(data['x'])
        db.y(data['y'])
        if data['coordsys'].spaceDim() == 3:
            db.z(data['z'])
        db.data(data['points'], values)
        db.names(names)
        db.units(units)

        ModuleSimpleGridAscii.write(db)
        return
コード例 #3
0
 def _createModuleObj(self):
     """
 Create Python module object.
 """
     ModuleSimpleGridAscii.__init__(self)
     return
コード例 #4
0
 def _createModuleObj(self):
   """
   Create Python module object.
   """
   ModuleSimpleGridAscii.__init__(self)
   return