Newer
Older
Janik Zikovsky
committed
/*WIKI*
The LoadEventPreNeXus algorithm stores data from the pre-nexus neutron event data file in an [[EventWorkspace]]. The default histogram bin boundaries consist of a single bin able to hold all events (in all pixels), and will have their [[units]] set to time-of-flight. Since it is an [[EventWorkspace]], it can be rebinned to finer bins with no loss of data.
=== Optional properties ===
Specific pulse ID and mapping files can be specified if needed; these are guessed at automatically from the neutron filename, if not specified.
*WIKI*/
#include "MantidDataHandling/LoadEventPreNexus.h"
Peterson, Peter
committed
#include "MantidAPI/FileFinder.h"
Gigg, Martyn Anthony
committed
#include "MantidAPI/LoadAlgorithmFactory.h"
#include "MantidDataObjects/EventWorkspace.h"
#include "MantidDataObjects/EventList.h"
#include "MantidKernel/ArrayProperty.h"
Peterson, Peter
committed
#include "MantidKernel/FileValidator.h"
Janik Zikovsky
committed
#include "MantidKernel/DateAndTime.h"
Peterson, Peter
committed
#include "MantidKernel/Glob.h"
Gigg, Martyn Anthony
committed
#include "MantidAPI/FileProperty.h"
Peterson, Peter
committed
#include "MantidKernel/ConfigService.h"
#include "MantidKernel/BinaryFile.h"
#include "MantidKernel/System.h"
#include "MantidKernel/TimeSeriesProperty.h"
#include "MantidKernel/UnitFactory.h"
#include "MantidKernel/DateAndTime.h"
#include "MantidGeometry/IDetector.h"
Janik Zikovsky
committed
#include "MantidKernel/CPUTimer.h"
#include "MantidKernel/VisibleWhenProperty.h"
#include "MantidKernel/BoundedValidator.h"
#include "MantidKernel/ListValidator.h"
#include "MantidAPI/MemoryManager.h"
Peterson, Peter
committed
#include <algorithm>
#include <sstream>
#include <stdexcept>
#include <functional>
#include <iostream>
#include <set>
#include <vector>
#include <Poco/File.h>
#include <Poco/Path.h>
#include <boost/timer.hpp>
Peterson, Peter
committed
// Register the algorithm into the AlgorithmFactory
DECLARE_ALGORITHM(LoadEventPreNexus)
DECLARE_LOADALGORITHM(LoadEventPreNexus)
Peterson, Peter
committed
Janik Zikovsky
committed
/// Sets documentation strings for this algorithm
void LoadEventPreNexus::initDocs()
Janik Zikovsky
committed
{
this->setWikiSummary("Loads SNS raw neutron event data format and stores it in a [[workspace]] ([[EventWorkspace]] class). ");
this->setOptionalMessage("Loads SNS raw neutron event data format and stores it in a workspace (EventWorkspace class).");
}
using namespace Kernel;
using namespace API;
using namespace Geometry;
using boost::posix_time::ptime;
using boost::posix_time::time_duration;
using DataObjects::EventList;
using DataObjects::EventWorkspace;
using DataObjects::EventWorkspace_sptr;
using DataObjects::TofEvent;
using std::cout;
using std::endl;
using std::ifstream;
using std::runtime_error;
using std::stringstream;
using std::string;
using std::vector;
// constants for locating the parameters to use in execution
static const string EVENT_PARAM("EventFilename");
static const string PULSEID_PARAM("PulseidFilename");
static const string MAP_PARAM("MappingFilename");
static const string PID_PARAM("SpectrumList");
static const string PARALLEL_PARAM("UseParallelProcessing");
static const string BLOCK_SIZE_PARAM("LoadingBlockSize");
static const string OUT_PARAM("OutputWorkspace");
Peterson, Peter
committed
static const string PULSE_EXT("pulseid.dat");
static const string EVENT_EXT("event.dat");
Janik Zikovsky
committed
static const size_t DEFAULT_BLOCK_SIZE = 1000000; // 100,000
/// All pixel ids with matching this mask are errors.
/// The maximum possible tof as native type
static const uint32_t MAX_TOF_UINT32 = std::numeric_limits<uint32_t>::max();
/// Conversion factor between 100 nanoseconds and 1 microsecond.
static const double TOF_CONVERSION = .1;
/// Conversion factor between picoColumbs and microAmp*hours
static const double CURRENT_CONVERSION = 1.e-6 / 3600.;
LoadEventPreNexus::LoadEventPreNexus() : Mantid::API::IDataFileChecker(), eventfile(NULL), max_events(0)
Peterson, Peter
committed
LoadEventPreNexus::~LoadEventPreNexus()
Peterson, Peter
committed
{
delete this->eventfile;
Peterson, Peter
committed
Gigg, Martyn Anthony
committed
/**
* Returns the name of the property to be considered as the Filename for Load
* @returns A character string containing the file property's name
*/
const char * LoadEventPreNexus::filePropertyName() const
Gigg, Martyn Anthony
committed
{
return EVENT_PARAM.c_str();
}
/**
* Do a quick file type check by looking at the first 100 bytes of the file
* @param filePath :: path of the file including name.
* @param nread :: no.of bytes read
* @param header :: The first 100 bytes of the file as a union
* @return true if the given file is of type which can be loaded by this algorithm
*/
bool LoadEventPreNexus::quickFileCheck(const std::string& filePath,size_t,const file_header&)
Gigg, Martyn Anthony
committed
{
std::string ext = extension(filePath);
return (ext.rfind("dat") != std::string::npos);
}
/**
* Checks the file by opening it and reading few lines
* @param filePath :: name of the file inluding its path
* @return an integer value how much this algorithm can load the file
*/
int LoadEventPreNexus::fileCheck(const std::string& filePath)
Gigg, Martyn Anthony
committed
{
int confidence(0);
try
{
// If this looks like a binary file where the exact file length is a multiple
Gigg, Martyn Anthony
committed
// of the DasEvent struct then we're probably okay.
// NOTE: Putting this on the stack gives a segfault on Windows when for some reason
// the BinaryFile destructor is called twice! I'm sure there is something I don't understand there
// but heap allocation seems to work so go for that.
BinaryFile<DasEvent> *event_file = new BinaryFile<DasEvent>(filePath);
Gigg, Martyn Anthony
committed
confidence = 80;
Gigg, Martyn Anthony
committed
delete event_file;
Gigg, Martyn Anthony
committed
}
catch(std::runtime_error &)
{
Gigg, Martyn Anthony
committed
// This BinaryFile constructor throws if the file does not contain an
// exact multiple of the sizeof(DasEvent) objects.
Gigg, Martyn Anthony
committed
}
return confidence;
}
Janik Zikovsky
committed
//-----------------------------------------------------------------------------
/** Initialize the algorithm */
void LoadEventPreNexus::init()
Janik Zikovsky
committed
declareProperty(new FileProperty(EVENT_PARAM, "", FileProperty::Load, EVENT_EXT),
"The name of the neutron event file to read, including its full or relative path. The file typically ends in neutron_event.dat (N.B. case sensitive if running on Linux).");
declareProperty(new FileProperty(PULSEID_PARAM, "", FileProperty::OptionalLoad, PULSE_EXT),
"File containing the accelerator pulse information; the filename will be found automatically if not specified.");
declareProperty(new FileProperty(MAP_PARAM, "", FileProperty::OptionalLoad, ".dat"),
"File containing the pixel mapping (DAS pixels to pixel IDs) file (typically INSTRUMENT_TS_YYYY_MM_DD.dat). The filename will be found automatically if not specified.");
Janik Zikovsky
committed
declareProperty(new ArrayProperty<int64_t>(PID_PARAM),
"A list of individual spectra (pixel IDs) to read, specified as e.g. 10:20. Only used if set.");
Peterson, Peter
committed
auto mustBePositive = boost::make_shared<BoundedValidator<int> >();
mustBePositive->setLower(1);
declareProperty("ChunkNumber", EMPTY_INT(), mustBePositive,
"If loading the file by sections ('chunks'), this is the section number of this execution of the algorithm.");
declareProperty("TotalChunks", EMPTY_INT(), mustBePositive,
"If loading the file by sections ('chunks'), this is the total number of sections.");
// TotalChunks is only meaningful if ChunkNumber is set
// Would be nice to be able to restrict ChunkNumber to be <= TotalChunks at validation
setPropertySettings("TotalChunks", new VisibleWhenProperty("ChunkNumber", IS_NOT_DEFAULT));
Janik Zikovsky
committed
std::vector<std::string> propOptions;
propOptions.push_back("Auto");
propOptions.push_back("Serial");
propOptions.push_back("Parallel");
declareProperty("UseParallelProcessing", "Auto", boost::make_shared<StringListValidator>(propOptions),
Janik Zikovsky
committed
"Use multiple cores for loading the data?\n"
" Auto: Use serial loading for small data sets, parallel for large data sets.\n"
" Serial: Use a single core.\n"
" Parallel: Use all available cores.");
// the output workspace name
Janik Zikovsky
committed
declareProperty(new WorkspaceProperty<IEventWorkspace>(OUT_PARAM,"",Direction::Output),
"The name of the workspace that will be created, filled with the read-in data and stored in the [[Analysis Data Service]].");
Peterson, Peter
committed
}
Janik Zikovsky
committed
//-----------------------------------------------------------------------------
Janik Zikovsky
committed
static string generatePulseidName(string eventfile)
Peterson, Peter
committed
{
size_t start;
string ending;
// normal ending
ending = "neutron_event.dat";
start = eventfile.find(ending);
if (start != string::npos)
return eventfile.replace(start, ending.size(), "pulseid.dat");
// split up event files - yes this is copy and pasted code
ending = "neutron0_event.dat";
start = eventfile.find(ending);
if (start != string::npos)
return eventfile.replace(start, ending.size(), "pulseid0.dat");
ending = "neutron1_event.dat";
start = eventfile.find(ending);
if (start != string::npos)
return eventfile.replace(start, ending.size(), "pulseid1.dat");
return "";
}
Janik Zikovsky
committed
//-----------------------------------------------------------------------------
Peterson, Peter
committed
static string generateMappingfileName(EventWorkspace_sptr &wksp)
{//
Peterson, Peter
committed
// get the name of the mapping file as set in the parameter files
Gigg, Martyn Anthony
committed
std::vector<string> temp = wksp->getInstrument()->getStringParameter("TS_mapping_file");
Peterson, Peter
committed
if (temp.empty())
return "";
string mapping = temp[0];
Lynch, Vickie
committed
// Try to get it from the working directory
Poco::File localmap(mapping);
if (localmap.exists())
return mapping;
Peterson, Peter
committed
// Try to get it from the data directories
string dataversion = Mantid::API::FileFinder::Instance().getFullPath(mapping);
if (!dataversion.empty())
return dataversion;
// get a list of all proposal directories
Peterson, Peter
committed
string instrument = wksp->getInstrument()->getName();
Peterson, Peter
committed
Poco::File base("/SNS/" + instrument + "/");
Lynch, Vickie
committed
// try short instrument name
Peterson, Peter
committed
if (!base.exists())
Lynch, Vickie
committed
{
instrument = Kernel::ConfigService::Instance().getInstrument(instrument).shortName();
Lynch, Vickie
committed
if (!base.exists())
return "";
}
Peterson, Peter
committed
vector<string> dirs; // poco won't let me reuse temp
base.list(dirs);
// check all of the proposals for the mapping file in the canonical place
const string CAL("_CAL");
Peterson, Peter
committed
const size_t CAL_LEN = CAL.length(); // cache to make life easier
Peterson, Peter
committed
vector<string> files;
for (size_t i = 0; i < dirs.size(); ++i) {
Peterson, Peter
committed
if ( (dirs[i].length() > CAL_LEN)
&& (dirs[i].compare(dirs[i].length() - CAL.length(), CAL.length(), CAL) == 0) ) {
Peterson, Peter
committed
if (Poco::File(base.path() + "/" + dirs[i] + "/calibrations/" + mapping).exists())
files.push_back(base.path() + "/" + dirs[i] + "/calibrations/" + mapping);
}
}
Peterson, Peter
committed
if (files.empty())
return "";
Peterson, Peter
committed
else if (files.size() == 1)
return files[0];
else // just assume that the last one is the right one, this should never be fired
return *(files.rbegin());
Peterson, Peter
committed
}
namespace { // anonymous namespace
string getRunnumber(const string &filename) {
// start by trimming the filename
string runnumber(Poco::Path(filename).getBaseName());
Janik Zikovsky
committed
if (runnumber.find("neutron") >= string::npos)
return "0";
std::size_t left = runnumber.find("_");
std::size_t right = runnumber.find("_", left+1);
return runnumber.substr(left+1, right-left-1);
}
}
Janik Zikovsky
committed
Janik Zikovsky
committed
//-----------------------------------------------------------------------------
/** Execute the algorithm */
void LoadEventPreNexus::exec()
Peterson, Peter
committed
{
// Check 'chunk' properties are valid, if set
const int chunks = getProperty("TotalChunks");
if ( !isEmpty(chunks) && int(getProperty("ChunkNumber")) > chunks )
{
throw std::out_of_range("ChunkNumber cannot be larger than TotalChunks");
}
Janik Zikovsky
committed
prog = new Progress(this,0.0,1.0,100);
// what spectra (pixel ID's) to load
this->spectra_list = this->getProperty(PID_PARAM);
Peterson, Peter
committed
// the event file is needed in case the pulseid fileanme is empty
string event_filename = this->getPropertyValue(EVENT_PARAM);
Peterson, Peter
committed
bool throwError = true;
Peterson, Peter
committed
if (pulseid_filename.empty())
{
Janik Zikovsky
committed
pulseid_filename = generatePulseidName(event_filename);
Peterson, Peter
committed
if (!pulseid_filename.empty())
{
if (Poco::File(pulseid_filename).exists())
Peterson, Peter
committed
{
Peterson, Peter
committed
this->g_log.information() << "Found pulseid file " << pulseid_filename << std::endl;
Peterson, Peter
committed
throwError = false;
}
Peterson, Peter
committed
{
pulseid_filename = "";
Peterson, Peter
committed
}
Peterson, Peter
committed
}
Janik Zikovsky
committed
prog->report("Loading Pulse ID file");
Peterson, Peter
committed
this->readPulseidFile(pulseid_filename, throwError);
Peterson, Peter
committed
this->openEventFile(event_filename);
Janik Zikovsky
committed
prog->report("Creating output workspace");
// prep the output workspace
EventWorkspace_sptr localWorkspace = EventWorkspace_sptr(new EventWorkspace());
Janik Zikovsky
committed
//Make sure to initialize.
// We can use dummy numbers for arguments, for event workspace it doesn't matter
localWorkspace->initialize(1,1,1);
Janik Zikovsky
committed
// Set the units
localWorkspace->getAxis(0)->unit() = UnitFactory::Instance().create("TOF");
localWorkspace->setYUnit("Counts");
// TODO localWorkspace->setTitle(title);
Janik Zikovsky
committed
// Add the run_start property
// Use the first pulse as the run_start time.
Janik Zikovsky
committed
if (this->num_pulses > 0)
Janik Zikovsky
committed
{
// add the start of the run as a ISO8601 date/time string. The start = the first pulse.
// (this is used in LoadInstrument to find the right instrument file to use).
localWorkspace->mutableRun().addProperty("run_start", pulsetimes[0].toISO8601String(), true );
Janik Zikovsky
committed
}
// determine the run number and add it to the run object
localWorkspace->mutableRun().addProperty("run_number", getRunnumber(event_filename));
//Get the instrument!
Janik Zikovsky
committed
prog->report("Loading Instrument");
Peterson, Peter
committed
this->runLoadInstrument(event_filename, localWorkspace);
Janik Zikovsky
committed
Peterson, Peter
committed
// load the mapping file
Janik Zikovsky
committed
prog->report("Loading Mapping File");
Peterson, Peter
committed
string mapping_filename = this->getPropertyValue(MAP_PARAM);
Peterson, Peter
committed
if (mapping_filename.empty()) {
mapping_filename = generateMappingfileName(localWorkspace);
if (!mapping_filename.empty())
this->g_log.information() << "Found mapping file \"" << mapping_filename << "\"" << std::endl;
}
Peterson, Peter
committed
this->loadPixelMap(mapping_filename);
Janik Zikovsky
committed
//Process the events into pixels
this->procEvents(localWorkspace);
Janik Zikovsky
committed
//Save output
Gigg, Martyn Anthony
committed
this->setProperty<IEventWorkspace_sptr>(OUT_PARAM, localWorkspace);
Janik Zikovsky
committed
// Cleanup
delete prog;
Peterson, Peter
committed
}
Janik Zikovsky
committed
//-----------------------------------------------------------------------------
/** Load the instrument geometry File
Janik Zikovsky
committed
* @param eventfilename :: Used to pick the instrument.
* @param localWorkspace :: MatrixWorkspace in which to put the instrument geometry
Janik Zikovsky
committed
*/
void LoadEventPreNexus::runLoadInstrument(const std::string &eventfilename, MatrixWorkspace_sptr localWorkspace)
Janik Zikovsky
committed
{
Peterson, Peter
committed
// determine the instrument parameter file
string instrument = Poco::Path(eventfilename).getFileName();
size_t pos = instrument.rfind("_"); // get rid of 'event.dat'
pos = instrument.rfind("_", pos-1); // get rid of 'neutron'
pos = instrument.rfind("_", pos-1); // get rid of the run number
instrument = instrument.substr(0, pos);
// do the actual work
Janik Zikovsky
committed
IAlgorithm_sptr loadInst= createSubAlgorithm("LoadInstrument");
// Now execute the sub-algorithm. Catch and log any error, but don't stop.
Peterson, Peter
committed
loadInst->setPropertyValue("InstrumentName", instrument);
loadInst->setProperty<MatrixWorkspace_sptr> ("Workspace", localWorkspace);
Lynch, Vickie
committed
loadInst->setProperty("RewriteSpectraMap", false);
Peterson, Peter
committed
loadInst->executeAsSubAlg();
Peterson, Peter
committed
Peterson, Peter
committed
// Populate the instrument parameters in this workspace - this works around a bug
localWorkspace->populateInstrumentParameters();
Janik Zikovsky
committed
}
Janik Zikovsky
committed
Janik Zikovsky
committed
//-----------------------------------------------------------------------------
/** Turn a pixel id into a "corrected" pixelid and period.
*
*/
Janik Zikovsky
committed
inline void LoadEventPreNexus::fixPixelId(PixelType &pixel, uint32_t &period) const
Janik Zikovsky
committed
if (!this->using_mapping_file) { // nothing to do here
period = 0;
return;
}
PixelType unmapped_pid = pixel % this->numpixel;
period = (pixel - unmapped_pid) / this->numpixel;
pixel = this->pixelmap[unmapped_pid];
}
Janik Zikovsky
committed
//-----------------------------------------------------------------------------
/** Process the event file properly.
Janik Zikovsky
committed
* @param workspace :: EventWorkspace to write to.
*/
void LoadEventPreNexus::procEvents(DataObjects::EventWorkspace_sptr & workspace)
this->num_ignored_events = 0;
//Default values in the case of no parallel
size_t loadBlockSize = Mantid::Kernel::DEFAULT_BLOCK_SIZE * 2;
shortest_tof = static_cast<double>(MAX_TOF_UINT32) * TOF_CONVERSION;
longest_tof = 0.;
Janik Zikovsky
committed
//Initialize progress reporting.
size_t numBlocks = (max_events + loadBlockSize - 1) / loadBlockSize;
Janik Zikovsky
committed
Peterson, Peter
committed
// We want to pad out empty pixels.
Peterson, Peter
committed
detid2det_map detector_map;
workspace->getInstrument()->getDetectors(detector_map);
Janik Zikovsky
committed
// -------------- Determine processing mode
std::string procMode = getProperty("UseParallelProcessing");
if (procMode == "Serial")
parallelProcessing = false;
else if (procMode == "Parallel")
parallelProcessing = true;
else
{
// Automatic determination. Loading serially (for me) is about 3 million events per second,
// (which is sped up by ~ x 3 with parallel processing, say 10 million per second, e.g. 7 million events more per seconds).
// compared to a setup time/merging time of about 10 seconds per million detectors.
double setUpTime = double(detector_map.size()) * 10e-6;
parallelProcessing = ((double(max_events) / 7e6) > setUpTime);
Janik Zikovsky
committed
g_log.debug() << (parallelProcessing ? "Using" : "Not using") << " parallel processing." << std::endl;
}
Peterson, Peter
committed
// determine maximum pixel id
detid2det_map::iterator it;
detid_max = 0; // seems like a safe lower bound
Peterson, Peter
committed
for (it = detector_map.begin(); it != detector_map.end(); it++)
if (it->first > detid_max)
detid_max = it->first;
Janik Zikovsky
committed
// Pad all the pixels
Janik Zikovsky
committed
prog->report("Padding Pixels");
this->pixel_to_wkspindex.reserve(detid_max+1); //starting at zero up to and including detid_max
Janik Zikovsky
committed
// Set to zero
this->pixel_to_wkspindex.assign(detid_max+1, 0);
Peterson, Peter
committed
size_t workspaceIndex = 0;
for (it = detector_map.begin(); it != detector_map.end(); it++)
Peterson, Peter
committed
{
Peterson, Peter
committed
if (!it->second->isMonitor())
Peterson, Peter
committed
{
Peterson, Peter
committed
this->pixel_to_wkspindex[it->first] = workspaceIndex;
Janik Zikovsky
committed
EventList & spec = workspace->getOrAddEventList(workspaceIndex);
spec.addDetectorID(it->first);
Lynch, Vickie
committed
// Start the spectrum number at 1
spec.setSpectrumNo(specid_t(workspaceIndex+1));
Peterson, Peter
committed
workspaceIndex += 1;
Peterson, Peter
committed
}
}
Janik Zikovsky
committed
workspace->doneAddingEventLists();
Peterson, Peter
committed
//For slight speed up
loadOnlySomeSpectra = (this->spectra_list.size() > 0);
Janik Zikovsky
committed
//Turn the spectra list into a map, for speed of access
for (std::vector<int64_t>::iterator it = spectra_list.begin(); it != spectra_list.end(); it++)
spectraLoadMap[*it] = true;
Janik Zikovsky
committed
CPUTimer tim;
// --------------- Create the partial workspaces ------------------------------------------
// Vector of partial workspaces, for parallel processing.
std::vector<EventWorkspace_sptr> partWorkspaces;
std::vector<DasEvent *> buffers;
Janik Zikovsky
committed
/// Pointer to the vector of events
typedef std::vector<TofEvent> * EventVector_pt;
/// Bare array of arrays of pointers to the EventVectors
EventVector_pt ** eventVectors;
/// How many threads will we use?
size_t numThreads = 1;
Janik Zikovsky
committed
if (parallelProcessing)
Janik Zikovsky
committed
numThreads = size_t(PARALLEL_GET_MAX_THREADS);
partWorkspaces.resize(numThreads);
buffers.resize(numThreads);
eventVectors = new EventVector_pt *[numThreads];
// cppcheck-suppress syntaxError
Janik Zikovsky
committed
PRAGMA_OMP( parallel for if (parallelProcessing) )
for (int i=0; i < int(numThreads); i++)
Janik Zikovsky
committed
{
Janik Zikovsky
committed
// This is the partial workspace we are about to create (if in parallel)
EventWorkspace_sptr partWS;
if (parallelProcessing)
Janik Zikovsky
committed
{
prog->report("Creating Partial Workspace");
// Create a partial workspace
Janik Zikovsky
committed
partWS = EventWorkspace_sptr(new EventWorkspace());
Janik Zikovsky
committed
//Make sure to initialize.
partWS->initialize(1,1,1);
// Copy all the spectra numbers and stuff (no actual events to copy though).
partWS->copyDataFrom(*workspace);
// Push it in the array
partWorkspaces[i] = partWS;
}
Janik Zikovsky
committed
else
partWS = workspace;
//Allocate the buffers
buffers[i] = new DasEvent[loadBlockSize];
// For each partial workspace, make an array where index = detector ID and value = pointer to the events vector
eventVectors[i] = new EventVector_pt[detid_max+1];
Janik Zikovsky
committed
EventVector_pt * theseEventVectors = eventVectors[i];
Janik Zikovsky
committed
{
size_t wi = pixel_to_wkspindex[j];
// Save a POINTER to the vector<tofEvent>
theseEventVectors[j] = &partWS->getEventList(wi).getEvents();
}
Janik Zikovsky
committed
}
Janik Zikovsky
committed
g_log.debug() << tim << " to create " << partWorkspaces.size() << " workspaces for parallel loading." << std::endl;
Janik Zikovsky
committed
prog->resetNumSteps( numBlocks, 0.1, 0.8);
// ---------------------------------- LOAD THE DATA --------------------------
PRAGMA_OMP( parallel for schedule(dynamic, 1) if (parallelProcessing) )
for (int blockNum=0; blockNum<int(numBlocks); blockNum++)
{
Janik Zikovsky
committed
PARALLEL_START_INTERUPT_REGION
Janik Zikovsky
committed
// Find the workspace for this particular thread
EventWorkspace_sptr ws;
size_t threadNum = 0;
if (parallelProcessing)
{
threadNum = PARALLEL_THREAD_NUMBER;
ws = partWorkspaces[threadNum];
}
else
ws = workspace;
Janik Zikovsky
committed
Janik Zikovsky
committed
// Get the buffer (for this thread)
DasEvent * event_buffer = buffers[threadNum];
Janik Zikovsky
committed
// Get the speeding-up array of vector<tofEvent> where index = detid.
EventVector_pt * theseEventVectors = eventVectors[threadNum];
Janik Zikovsky
committed
// Where to start in the file?
size_t fileOffset = first_event + (loadBlockSize * blockNum);
// May need to reduce size of last (or only) block
size_t current_event_buffer_size =
( blockNum == int(numBlocks-1) ) ? ( max_events - (numBlocks-1)*loadBlockSize ) : loadBlockSize;
Janik Zikovsky
committed
// Load this chunk of event data (critical block)
PARALLEL_CRITICAL( LoadEventPreNexus_fileAccess )
{
current_event_buffer_size = eventfile->loadBlockAt(event_buffer, fileOffset, current_event_buffer_size);
Janik Zikovsky
committed
}
// This processes the events. Can be done in parallel!
Janik Zikovsky
committed
procEventsLinear(ws, theseEventVectors, event_buffer, current_event_buffer_size, fileOffset);
Janik Zikovsky
committed
// Report progress
prog->report("Load Event PreNeXus");
Janik Zikovsky
committed
PARALLEL_END_INTERUPT_REGION
Janik Zikovsky
committed
}
Janik Zikovsky
committed
PARALLEL_CHECK_INTERUPT_REGION
Janik Zikovsky
committed
g_log.debug() << tim << " to load the data." << std::endl;
Janik Zikovsky
committed
// ---------------------------------- MERGE WORKSPACES BACK TOGETHER --------------------------
if (parallelProcessing)
{
Janik Zikovsky
committed
PARALLEL_START_INTERUPT_REGION
Janik Zikovsky
committed
prog->resetNumSteps( workspace->getNumberHistograms(), 0.8, 0.95);
size_t memoryCleared = 0;
MemoryManager::Instance().releaseFreeMemory();
Janik Zikovsky
committed
Janik Zikovsky
committed
// Merge all workspaces, index by index.
PARALLEL_FOR_NO_WSP_CHECK()
for (int iwi=0; iwi<int(workspace->getNumberHistograms()); iwi++)
{
size_t wi = size_t(iwi);
// The output event list.
EventList & el = workspace->getEventList(wi);
el.clear(false);
// How many events will it have?
size_t numEvents = 0;
for (size_t i=0; i<numThreads; i++)
numEvents += partWorkspaces[i]->getEventList(wi).getNumberEvents();
// This will avoid too much copying.
el.reserve(numEvents);
Janik Zikovsky
committed
// Now merge the event lists
for (size_t i=0; i<numThreads; i++)
{
EventList & partEl = partWorkspaces[i]->getEventList(wi);
el += partEl.getEvents();
// Free up memory as you go along.
partEl.clear(false);
}
Janik Zikovsky
committed
Janik Zikovsky
committed
// With TCMalloc, release memory when you accumulate enough to make sense
PARALLEL_CRITICAL( LoadEventPreNexus_trackMemory )
{
memoryCleared += numEvents;
if (memoryCleared > 10000000) // ten million events = about 160 MB
{
MemoryManager::Instance().releaseFreeMemory();
memoryCleared = 0;
}
}
prog->report("Merging Workspaces");
}
// Final memory release
MemoryManager::Instance().releaseFreeMemory();
g_log.debug() << tim << " to merge workspaces together." << std::endl;
Janik Zikovsky
committed
PARALLEL_END_INTERUPT_REGION
Janik Zikovsky
committed
}
Janik Zikovsky
committed
PARALLEL_CHECK_INTERUPT_REGION
Janik Zikovsky
committed
Janik Zikovsky
committed
// Delete the buffers for each thread.
Janik Zikovsky
committed
for (size_t i=0; i<numThreads; i++)
{
delete [] buffers[i];
delete [] eventVectors[i];
}
delete [] eventVectors;
//delete [] pulsetimes;
Janik Zikovsky
committed
prog->resetNumSteps( 3, 0.94, 1.00);
Janik Zikovsky
committed
Peterson, Peter
committed
//finalize loading
Janik Zikovsky
committed
prog->report("Deleting Empty Lists");
Peterson, Peter
committed
if(loadOnlySomeSpectra)
workspace->deleteEmptyLists();
Janik Zikovsky
committed
prog->report("Setting proton charge");
Janik Zikovsky
committed
this->setProtonCharge(workspace);
Janik Zikovsky
committed
g_log.debug() << tim << " to set the proton charge log." << std::endl;
Janik Zikovsky
committed
//Make sure the MRU is cleared
workspace->clearMRU();
Janik Zikovsky
committed
//Now, create a default X-vector for histogramming, with just 2 bins.
Kernel::cow_ptr<MantidVec> axis;
MantidVec& xRef = axis.access();
xRef.resize(2);
xRef[0] = shortest_tof - 1; //Just to make sure the bins hold it all
xRef[1] = longest_tof + 1;
workspace->setAllX(axis);
Peterson, Peter
committed
this->pixel_to_wkspindex.clear();
Janik Zikovsky
committed
Janik Zikovsky
committed
g_log.information() << "Read " << this->num_good_events << " events + "
<< this->num_error_events << " errors"
Peterson, Peter
committed
<< ". Shortest TOF: " << shortest_tof << " microsec; longest TOF: "
<< longest_tof << " microsec." << std::endl;
Janik Zikovsky
committed
}
//-----------------------------------------------------------------------------
/** Linear-version of the procedure to process the event file properly.
Janik Zikovsky
committed
* @param workspace :: EventWorkspace to write to.
* @param arrayOfVectors :: For speed up: this is an array, of size detid_max+1, where the
Janik Zikovsky
committed
* index is a pixel ID, and the value is a pointer to the vector<tofEvent> in the given EventList.
Janik Zikovsky
committed
* @param event_buffer :: The buffer containing the DAS events
* @param current_event_buffer_size :: The length of the given DAS buffer
* @param fileOffset :: Value for an offset into the binary file
*/
Janik Zikovsky
committed
void LoadEventPreNexus::procEventsLinear(DataObjects::EventWorkspace_sptr & /*workspace*/,
std::vector<TofEvent> ** arrayOfVectors, DasEvent * event_buffer,
size_t current_event_buffer_size, size_t fileOffset)
{
Janik Zikovsky
committed
//Starting pulse time
Gigg, Martyn Anthony
committed
DateAndTime pulsetime;
Janik Zikovsky
committed
int64_t numPulses = static_cast<int64_t>(num_pulses);
{
g_log.warning() << "Event_indices vector is smaller than the pulsetimes array.\n";
Janik Zikovsky
committed
size_t local_num_error_events = 0;
size_t local_num_ignored_events = 0;
size_t local_num_good_events = 0;
double local_shortest_tof = static_cast<double>(MAX_TOF_UINT32) * TOF_CONVERSION;
double local_longest_tof = 0.;
// process the individual events
for (size_t i = 0; i < current_event_buffer_size; i++)
{
Janik Zikovsky
committed
DasEvent & temp = *(event_buffer + i);
PixelType pid = temp.pid;
if ((pid & ERROR_PID) == ERROR_PID) // marked as bad
{
Janik Zikovsky
committed
local_num_error_events++;
continue;
}
//Covert the pixel ID from DAS pixel to our pixel ID
Janik Zikovsky
committed
if (this->using_mapping_file)
{
PixelType unmapped_pid = pid % this->numpixel;
pid = this->pixelmap[unmapped_pid];
}
// Avoid segfaults for wrong pixel IDs
if (pid > static_cast<PixelType>(detid_max))
{
local_num_error_events++;
continue;
}
//Now check if this pid we want to load.
if (loadOnlySomeSpectra)
{
Janik Zikovsky
committed
it = spectraLoadMap.find(pid);
if (it == spectraLoadMap.end())
{
//Pixel ID was not found, so the event is being ignored.
Janik Zikovsky
committed
local_num_ignored_events++;
continue;
}
}
// work with the good guys
Janik Zikovsky
committed
Janik Zikovsky
committed
//Find the pulse time for this event index
if (pulse_i < numPulses-1)
{
//This is the total offset into the file
size_t total_i = i + fileOffset;
//Go through event_index until you find where the index increases to encompass the current index. Your pulse = the one before.
while (!((total_i >= event_indices[pulse_i]) && (total_i < event_indices[pulse_i+1])) )
{
Janik Zikovsky
committed
pulse_i++;
if (pulse_i >= (numPulses-1))
break;
}
//if (pulsetimes[pulse_i] != pulsetime) std::cout << pulse_i << " at " << pulsetimes[pulse_i] << "\n";
Janik Zikovsky
committed
//Save the pulse time at this index for creating those events
pulsetime = pulsetimes[pulse_i];
}
Janik Zikovsky
committed
double tof = static_cast<double>(temp.tof) * TOF_CONVERSION;
Janik Zikovsky
committed
TofEvent event(tof, pulsetime);
//Find the overall max/min tof
Janik Zikovsky
committed
if (tof < local_shortest_tof)
local_shortest_tof = tof;
if (tof > local_longest_tof)
local_longest_tof = tof;
//The addEventQuickly method does not clear the cache, making things slightly faster.
Janik Zikovsky
committed
//workspace->getEventList(this->pixel_to_wkspindex[pid]).addEventQuickly(event);
// This is equivalent to workspace->getEventList(this->pixel_to_wkspindex[pid]).addEventQuickly(event);
// But should be faster as a bunch of these calls were cached.
arrayOfVectors[pid]->push_back(event);
// TODO work with period
Janik Zikovsky
committed
local_num_good_events++;
}//for each event
Janik Zikovsky
committed
PARALLEL_CRITICAL( LoadEventPreNexus_global_statistics )
{
this->num_good_events += local_num_good_events;
this->num_ignored_events += local_num_ignored_events;
this->num_error_events += local_num_error_events;
if (local_shortest_tof < shortest_tof)
shortest_tof = local_shortest_tof;
if (local_longest_tof > longest_tof)
longest_tof = local_longest_tof;
}
}
//-----------------------------------------------------------------------------
/// Comparator for sorting dasevent lists
bool intermediatePixelIDComp(IntermediateEvent x, IntermediateEvent y)
{
return (x.pid < y.pid);
}
Janik Zikovsky
committed
//-----------------------------------------------------------------------------
/**
Janik Zikovsky
committed
* Add a sample environment log for the proton chage (charge of the pulse in picoCoulombs)
Janik Zikovsky
committed
* and set the scalar value (total proton charge, microAmps*hours, on the sample)
*
Janik Zikovsky
committed
* @param workspace :: Event workspace to set the proton charge on
*/
void LoadEventPreNexus::setProtonCharge(DataObjects::EventWorkspace_sptr & workspace)
{
if (this->proton_charge.empty()) // nothing to do
Peterson, Peter
committed
return;
Janik Zikovsky
committed
Gigg, Martyn Anthony
committed
Run& run = workspace->mutableRun();
Janik Zikovsky
committed
//Add the proton charge entries.
Janik Zikovsky
committed
TimeSeriesProperty<double>* log = new TimeSeriesProperty<double>("proton_charge");
Janik Zikovsky
committed
log->setUnits("picoCoulombs");
Janik Zikovsky
committed
//Add the time and associated charge to the log
log->addValues(this->pulsetimes, this->proton_charge);
Janik Zikovsky
committed
Janik Zikovsky
committed
/// TODO set the units for the log
Gigg, Martyn Anthony
committed
run.addLogData(log);
Janik Zikovsky
committed
double integ = run.integrateProtonCharge();
//run.setProtonCharge(this->proton_charge_tot); //This is now redundant
this->g_log.information() << "Total proton charge of " << integ << " microAmp*hours found by integrating.\n";
}
Janik Zikovsky
committed
//-----------------------------------------------------------------------------
/** Load a pixel mapping file
Janik Zikovsky
committed
* @param filename :: Path to file.
*/
void LoadEventPreNexus::loadPixelMap(const std::string &filename)
Janik Zikovsky
committed
this->using_mapping_file = false;
this->pixelmap.clear();
// check that there is a mapping file
if (filename.empty()) {
this->g_log.information("NOT using a mapping file");
return;
}
// actually deal with the file
this->g_log.debug("Using mapping file \"" + filename + "\"");
Janik Zikovsky
committed
//Open the file; will throw if there is any problem
BinaryFile<PixelType> pixelmapFile(filename);
PixelType max_pid = static_cast<PixelType>(pixelmapFile.getNumElements());
//Load all the data
pixelmapFile.loadAllInto( this->pixelmap );
Janik Zikovsky
committed
Janik Zikovsky
committed
//Check for funky file
if (std::find_if(pixelmap.begin(), pixelmap.end(), std::bind2nd(std::greater<PixelType>(), max_pid))
!= pixelmap.end())
{
this->g_log.warning("Pixel id in mapping file was out of bounds. Loading without mapping file");
this->numpixel = 0;
this->pixelmap.clear();
this->using_mapping_file = false;
Janik Zikovsky
committed
return;
Janik Zikovsky
committed
//If we got here, the mapping file was loaded correctly and we'll use it
this->using_mapping_file = true;
Janik Zikovsky
committed
//Let's assume that the # of pixels in the instrument matches the mapping file length.
this->numpixel = static_cast<uint32_t>(pixelmapFile.getNumElements());
Janik Zikovsky
committed
Janik Zikovsky
committed
//-----------------------------------------------------------------------------
/** Open an event file
Janik Zikovsky
committed
* @param filename :: file to open.
*/
void LoadEventPreNexus::openEventFile(const std::string &filename)
Janik Zikovsky
committed
//Open the file
eventfile = new BinaryFile<DasEvent>(filename);
num_events = eventfile->getNumElements();
g_log.debug() << "File contains " << num_events << " event records.\n";
Peterson, Peter
committed
// Check if we are only loading part of the event file
const int chunk = getProperty("ChunkNumber");
if ( isEmpty(chunk) ) // We are loading the whole file
{
first_event = 0;
max_events = num_events;
}
else // We are loading part - work out the event number range
{
const int totalChunks = getProperty("TotalChunks");
max_events = num_events/totalChunks;
first_event = (chunk - 1) * max_events;
// Need to add any remainder to the final chunk
if ( chunk == totalChunks ) max_events += num_events%totalChunks;
}
g_log.information()<< "Reading " << max_events << " event records\n";
//-----------------------------------------------------------------------------
/** Read a pulse ID file
Janik Zikovsky
committed
* @param filename :: file to load.
* @param throwError :: Flag to trigger error throwing instead of just logging
*/
void LoadEventPreNexus::readPulseidFile(const std::string &filename, const bool throwError)
Peterson, Peter
committed
this->proton_charge_tot = 0.;
Janik Zikovsky
committed
this->num_pulses = 0;
Peterson, Peter
committed
// jump out early if there isn't a filename
if (filename.empty()) {
this->g_log.information("NOT using a pulseid file");
return;
}
Peterson, Peter
committed
std::vector<Pulse> * pulses;
Janik Zikovsky
committed
//Open the file; will throw if there is any problem
Peterson, Peter
committed
try {
BinaryFile<Pulse> pulseFile(filename);
//Get the # of pulse
this->num_pulses = pulseFile.getNumElements();
this->g_log.information() << "Using pulseid file \"" << filename << "\", with " << num_pulses
<< " pulses.\n";
//Load all the data
pulses = pulseFile.loadAll();
} catch (runtime_error &e) {
if (throwError)
{
throw;
}
else
{
this->g_log.information() << "Encountered error in pulseidfile (ignoring file): " << e.what() << "\n";
return;
}
}
Janik Zikovsky
committed
double temp;
Janik Zikovsky
committed
if (num_pulses > 0)
Janik Zikovsky
committed
{
Janik Zikovsky
committed
this->pulsetimes.reserve(num_pulses);
for (size_t i=0; i < num_pulses; i++)
{
Pulse & it = (*pulses)[i];
this->pulsetimes.push_back( DateAndTime( (int64_t) it.seconds, (int64_t) it.nanoseconds) );
Janik Zikovsky
committed
this->event_indices.push_back(it.event_index);
temp = it.pCurrent;