mkfit::DataFile Struct Reference

#include <Event.h>

Public Types
enum	ExtraSection {   ES_SimTrackStates = 0x1, ES_Seeds = 0x2, ES_CmsswTracks = 0x4, ES_HitIterMasks = 0x8,   ES_BeamSpot = 0x10 }

Public Member Functions
int	advancePosToNextEvent (FILE *fp)
void	close ()
void	CloseWrite (int n_written)
bool	hasBeamSpot () const
bool	hasCmsswTracks () const
bool	hasHitIterMasks () const
bool	hasSeeds () const
bool	hasSimTrackStates () const
int	openRead (const std::string &fname, int expected_n_layers)
void	openWrite (const std::string &fname, int n_layers, int n_ev, int extra_sections=0)
void	rewind ()
void	skipNEvents (int n_to_skip)

Public Attributes
FILE *	f_fp = 0
DataFileHeader	f_header
std::mutex	f_next_ev_mutex
long	f_pos = sizeof(DataFileHeader)

Detailed Description

Definition at line 100 of file Event.h.

Member Enumeration Documentation

ExtraSection

enum mkfit::DataFile::ExtraSection

Enumerator
ES_SimTrackStates
ES_Seeds
ES_CmsswTracks
ES_HitIterMasks
ES_BeamSpot

Definition at line 101 of file Event.h.

                      {
      ES_SimTrackStates = 0x1,
      ES_Seeds = 0x2,
      ES_CmsswTracks = 0x4,
      ES_HitIterMasks = 0x8,
      ES_BeamSpot = 0x10
    };

Member Function Documentation

advancePosToNextEvent()

int mkfit::DataFile::advancePosToNextEvent

(

FILE *

fp

)

Definition at line 1013 of file Event.cc.

References f_next_ev_mutex, f_pos, personalPlayback::fp, and mkfit::Config::loopOverFile.

Referenced by mkfit::Event::read_in().

                                              {
    int evsize;

    std::lock_guard<std::mutex> readlock(f_next_ev_mutex);

    fseek(fp, f_pos, SEEK_SET);
    fread(&evsize, sizeof(int), 1, fp);
    if (Config::loopOverFile) {
      // File ended, rewind back to beginning
      if (feof(fp) != 0) {
        f_pos = sizeof(DataFileHeader);
        fseek(fp, f_pos, SEEK_SET);
        fread(&evsize, sizeof(int), 1, fp);
      }
    }

    f_pos += evsize;

    return evsize;
  }

close()

void mkfit::DataFile::close

(

void

)

Definition at line 1046 of file Event.cc.

References f_fp, and f_header.

Referenced by CloseWrite(), esMonitoring.AsyncLineReaderMixin::handle_close(), esMonitoring.FDJsonServer::handle_close(), and test_standard().

                       {
    fclose(f_fp);
    f_fp = 0;
    f_header = DataFileHeader();
  }

CloseWrite()

void mkfit::DataFile::CloseWrite

(

int

n_written

)

Definition at line 1052 of file Event.cc.

References close(), f_fp, f_header, and mkfit::DataFileHeader::f_n_events.

Referenced by main().

                                         {
    if (f_header.f_n_events != n_written) {
      fseek(f_fp, 0, SEEK_SET);
      f_header.f_n_events = n_written;
      fwrite(&f_header, sizeof(DataFileHeader), 1, f_fp);
    }
    close();
  }

hasBeamSpot()

bool mkfit::DataFile::hasBeamSpot ( ) const

inline

Definition at line 122 of file Event.h.

References ES_BeamSpot, mkfit::DataFileHeader::f_extra_sections, and f_header.

Referenced by mkfit::Event::read_in(), and mkfit::Event::write_out().

{ return f_header.f_extra_sections & ES_BeamSpot; }

hasCmsswTracks()

bool mkfit::DataFile::hasCmsswTracks ( ) const

inline

Definition at line 120 of file Event.h.

References ES_CmsswTracks, mkfit::DataFileHeader::f_extra_sections, and f_header.

Referenced by openRead(), mkfit::Event::read_in(), and mkfit::Event::write_out().

{ return f_header.f_extra_sections & ES_CmsswTracks; }

hasHitIterMasks()

bool mkfit::DataFile::hasHitIterMasks ( ) const

inline

Definition at line 121 of file Event.h.

References ES_HitIterMasks, mkfit::DataFileHeader::f_extra_sections, and f_header.

Referenced by mkfit::Event::read_in(), and mkfit::Event::write_out().

{ return f_header.f_extra_sections & ES_HitIterMasks; }

hasSeeds()

bool mkfit::DataFile::hasSeeds ( ) const

inline

Definition at line 119 of file Event.h.

References ES_Seeds, mkfit::DataFileHeader::f_extra_sections, and f_header.

Referenced by openRead(), mkfit::Event::read_in(), and mkfit::Event::write_out().

{ return f_header.f_extra_sections & ES_Seeds; }

hasSimTrackStates()

bool mkfit::DataFile::hasSimTrackStates ( ) const

inline

Definition at line 118 of file Event.h.

References ES_SimTrackStates, mkfit::DataFileHeader::f_extra_sections, and f_header.

Referenced by mkfit::Event::read_in(), and mkfit::Event::write_out().

{ return f_header.f_extra_sections & ES_SimTrackStates; }

openRead()

int mkfit::DataFile::openRead	(	const std::string &	fname,
		int	expected_n_layers
	)

Definition at line 919 of file Event.cc.

References cms::cuda::assert(), mkfit::cmsswSeeds, ES_CmsswTracks, ES_Seeds, ES_SimTrackStates, beamvalidation::exit(), mkfit::DataFileHeader::f_extra_sections, mkfit::DataFileHeader::f_format_version, f_fp, f_header, mkfit::DataFileHeader::f_magic, mkfit::DataFileHeader::f_n_events, mkfit::DataFileHeader::f_n_layers, mkfit::DataFileHeader::f_sizeof_hit, mkfit::DataFileHeader::f_sizeof_hot, mkfit::DataFileHeader::f_sizeof_track, alignmentValidation::fname, hasCmsswTracks(), hasSeeds(), mkfit::Config::readCmsswTracks, mkfit::Config::seedInput, and submitPVResolutionJobs::stderr.

Referenced by mkfit::Shell::Shell(), and test_standard().

                                                                      {
    constexpr int min_ver = 7;
    constexpr int max_ver = 7;

    f_fp = fopen(fname.c_str(), "r");
    assert(f_fp != 0 && "Opening of input file failed.");

    fread(&f_header, sizeof(DataFileHeader), 1, f_fp);

    if (f_header.f_magic != 0xBEEF) {
      fprintf(stderr, "Incompatible input file (wrong magick).\n");
      exit(1);
    }
    if (f_header.f_format_version < min_ver || f_header.f_format_version > max_ver) {
      fprintf(stderr,
              "Unsupported file version %d. Supported versions are from %d to %d.\n",
              f_header.f_format_version,
              min_ver,
              max_ver);
      exit(1);
    }
    if (f_header.f_sizeof_track != sizeof(Track)) {
      fprintf(stderr,
              "sizeof(Track) on file (%d) different from current value (%d).\n",
              f_header.f_sizeof_track,
              (int)sizeof(Track));
      exit(1);
    }
    if (f_header.f_sizeof_hit != sizeof(Hit)) {
      fprintf(stderr,
              "sizeof(Hit) on file (%d) different from current value (%d).\n",
              f_header.f_sizeof_hit,
              (int)sizeof(Hit));
      exit(1);
    }
    if (f_header.f_sizeof_hot != sizeof(HitOnTrack)) {
      fprintf(stderr,
              "sizeof(HitOnTrack) on file (%d) different from current value (%d).\n",
              f_header.f_sizeof_hot,
              (int)sizeof(HitOnTrack));
      exit(1);
    }
    if (f_header.f_n_layers != expected_n_layers) {
      fprintf(stderr,
              "Number of layers on file (%d) is different from current TrackerInfo (%d).\n",
              f_header.f_n_layers,
              expected_n_layers);
      exit(1);
    }

    printf("Opened file '%s', format version %d, n_layers %d, n_events %d\n",
           fname.c_str(),
           f_header.f_format_version,
           f_header.f_n_layers,
           f_header.f_n_events);
    if (f_header.f_extra_sections) {
      printf("  Extra sections:");
      if (f_header.f_extra_sections & ES_SimTrackStates)
        printf(" SimTrackStates");
      if (f_header.f_extra_sections & ES_Seeds)
        printf(" Seeds");
      if (f_header.f_extra_sections & ES_CmsswTracks)
        printf(" CmsswTracks");
      printf("\n");
    }

    if (Config::seedInput == cmsswSeeds && !hasSeeds()) {
      fprintf(stderr, "Reading of CmsswSeeds requested but data not available on file.\n");
      exit(1);
    }

    if (Config::readCmsswTracks && !hasCmsswTracks()) {
      fprintf(stderr, "Reading of CmsswTracks requested but data not available on file.\n");
      exit(1);
    }

    return f_header.f_n_events;
  }

openWrite()

void mkfit::DataFile::openWrite	(	const std::string &	fname,
		int	n_layers,
		int	n_ev,
		int	extra_sections = 0
	)

Definition at line 998 of file Event.cc.

References mkfit::DataFileHeader::f_extra_sections, f_fp, f_header, mkfit::DataFileHeader::f_n_events, mkfit::DataFileHeader::f_n_layers, and alignmentValidation::fname.

Referenced by main().

                                                                                             {
    f_fp = fopen(fname.c_str(), "w");
    f_header.f_n_layers = n_layers;
    f_header.f_n_events = n_ev;
    f_header.f_extra_sections = extra_sections;

    fwrite(&f_header, sizeof(DataFileHeader), 1, f_fp);
  }

rewind()

void mkfit::DataFile::rewind

(

)

Definition at line 1007 of file Event.cc.

References f_fp, f_next_ev_mutex, and f_pos.

Referenced by mkfit::Shell::GoToEvent().

                        {
    std::lock_guard<std::mutex> readlock(f_next_ev_mutex);
    f_pos = sizeof(DataFileHeader);
    fseek(f_fp, f_pos, SEEK_SET);
  }

skipNEvents()

void mkfit::DataFile::skipNEvents

(

int

n_to_skip

)

Definition at line 1034 of file Event.cc.

References f_fp, f_next_ev_mutex, and f_pos.

Referenced by mkfit::Shell::GoToEvent(), and test_standard().

                                          {
    int evsize;

    std::lock_guard<std::mutex> readlock(f_next_ev_mutex);

    while (n_to_skip-- > 0) {
      fseek(f_fp, f_pos, SEEK_SET);
      fread(&evsize, sizeof(int), 1, f_fp);
      f_pos += evsize;
    }
  }

Member Data Documentation

f_fp

FILE* mkfit::DataFile::f_fp = 0

Definition at line 109 of file Event.h.

Referenced by close(), CloseWrite(), openRead(), openWrite(), mkfit::Event::read_in(), rewind(), skipNEvents(), and mkfit::Event::write_out().

f_header

DataFileHeader mkfit::DataFile::f_header

Definition at line 112 of file Event.h.

Referenced by close(), CloseWrite(), hasBeamSpot(), hasCmsswTracks(), hasHitIterMasks(), hasSeeds(), hasSimTrackStates(), openRead(), and openWrite().

f_next_ev_mutex

std::mutex mkfit::DataFile::f_next_ev_mutex

Definition at line 114 of file Event.h.

Referenced by advancePosToNextEvent(), rewind(), and skipNEvents().

f_pos

long mkfit::DataFile::f_pos = sizeof(DataFileHeader)

Definition at line 110 of file Event.h.

Referenced by advancePosToNextEvent(), rewind(), and skipNEvents().