bytefield.cpp

强大的并行科学计算软件包

// -*- C++ -*-
// ACL:license
// ----------------------------------------------------------------------
// This software and ancillary information (herein called "SOFTWARE")
// called POOMA (Parallel Object-Oriented Methods and Applications) is
// made available under the terms described here.  The SOFTWARE has been
// approved for release with associated LA-CC Number LA-CC-98-65.
// 
// Unless otherwise indicated, this SOFTWARE has been authored by an
// employee or employees of the University of California, operator of the
// Los Alamos National Laboratory under Contract No.  W-7405-ENG-36 with
// the U.S. Department of Energy.  The U.S. Government has rights to use,
// reproduce, and distribute this SOFTWARE, and to allow others to do so.
// The public may copy and use this SOFTWARE, FOR NONCOMMERCIAL USE ONLY,
// without charge, provided that this Notice and any statement of
// authorship are reproduced on all copies.  Neither the Government nor
// the University makes any warranty, express or implied, or assumes any
// liability or responsibility for the use of this SOFTWARE.
// 
// If SOFTWARE is modified to produce derivative works, such modified
// SOFTWARE should be clearly marked, so as not to confuse it with the
// version available from LANL.
// 
// For more information about POOMA, send e-mail to [e-mail],
// or visit the POOMA web page at http://www.acl.lanl.gov/pooma/.
// ----------------------------------------------------------------------
// ACL:license
//-----------------------------------------------------------------------------
// Lux demo: Display a series of byte-fields
//-----------------------------------------------------------------------------

#include "Pooma/Pooma.h"
#include "Pooma/Lux.h"
#include "Pooma/Arrays.h"
#include "Pooma/Domains.h"
#include <string>
#include <stdio.h>
#include <stdlib.h>


int main(int argc, char *argv[])
{
  // Initialize POOMA and output stream

  Pooma::initialize(argc, argv);
  Inform msg(argv[0]);

  int i, sizeX = 0, sizeY = 0, sizeZ = 0;
  int numfiles = 0;

  // Get array size from command-line args

  for (i=1; i < argc; ++i)
    {
      std::string item(argv[i]);
      if (item == "-x" && i < (argc - 1))
	sizeX = atoi(argv[++i]);
      else if (item == "-y" && i < (argc - 1))
	sizeY = atoi(argv[++i]);
      else if (item == "-z" && i < (argc - 1))
	sizeZ = atoi(argv[++i]);
      else if (i < (argc - 1))
	numfiles++;
    }

  // Do some error-checking

  if (sizeX < 1 || sizeY < 1 || sizeZ < 1)
    {
      msg << "Bad size values, all must be > 0." << std::endl;
      exit(1);
    }

  if (numfiles < 1)
    {
      msg << "You must specify some files to display." << std::endl;
      exit(1);
    }

  // Create an array to display the data

  msg << "Initializing array ..." << std::endl;
  Interval<3> domain(sizeX, sizeY, sizeZ);
  Array<3, unsigned char, Brick> data(domain);
  data = 0;
  Pooma::blockAndEvaluate();

  // Create a Lux connection, and connect up the storage array

  msg << "Creating LuxConnection object ..." << std::endl;
  Connection<Lux> lux(argv[0]);

  msg << "Connecting data storage array ..." << std::endl;
  lux.connect("data", data);

  // Do, in a loop, updates of the arrays, and redisplay/interact.

  int count = 0;
  for (i=1; i < argc; ++i)
    {
      std::string item(argv[i]);
      if (item == "-x" || item == "-y" || item == "-z")
	{
	  ++i;
	}
      else
	{
	  msg << "Reading data from file '" << item << "' ..." << std::endl;

	  // Open the file

	  FILE *f = fopen(argv[i], "r");
	  if (f == 0)
	    {
	      msg << "Error opening file '" << item << "'." << std::endl;
	      break;
	    }

	  // Read in the data

	  if (fread(&(data(data.firsts())), domain.size(), 1, f) != 1)
	    {
	      msg << "Error reading from file '" << item << "'." << std::endl;
	      fclose(f);
	      break;
	    }

	  // Sanity check

	  int s = sum(data);
	  msg << "Sum of data read = " << s << std::endl;
	  msg << "Middle 1D slice of data:" << std::endl;
	  msg << data(Interval<1>(sizeX), sizeY/2, sizeZ/2) << std::endl;

	  // Update the display

	  msg << "Updating the display, for file " << ++count << " out of ";
	  msg << numfiles << " ..." << std::endl;
	  lux.ready();
	}
    }

  // Delete LUX connection, closing the window.

  msg << "Closing LUX connection ..." << std::endl;
  lux.close();

  // Finish up

  Pooma::finalize();
  return 0;
}

// ACL:rcsinfo
// ----------------------------------------------------------------------
// $RCSfile: bytefield.cpp,v $   $Author: bfh $
// $Revision: 1.1 $   $Date: 1999/10/01 03:26:37 $
// ----------------------------------------------------------------------
// ACL:rcsinfo