This addresses the bugs from the upstream changes in pprofile identified in #11.

With these changes, the tests pass:

Basic test to check pyheat class works end to end. ... ok

----------------------------------------------------------------------
Ran 1 test in 0.272s

Longer (hundreds LOC) scripts will get squished by this. The solution would be breaking heatmap into manageable chunks and introducing GUI that would let you scroll through. Preferably with a minimap (a la Sublime Text) that shows where you are and overall heatmap of the script.

Hello, thanks for making this neat tool!

It looks like this commit in pprofile changed the Profile.file_dict value to be a list instead of a single dictionary. This breaks the way pyheat parses the Profile object.

Running the existing tests gives this failure:

======================================================================
ERROR: test_pyheat_show (tests.pyheat_test.PyHeatTest)
Basic test to check pyheat class works end to end.
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/hhinnefeld/repos/pyheat-original/tests/pyheat_test.py", line 26, in test_pyheat_show
    self.pyheat.create_heatmap()
  File "/Users/hhinnefeld/repos/pyheat-original/pyheat/pyheat.py", line 48, in create_heatmap
    self.__fetch_heatmap_data_from_profile()
  File "/Users/hhinnefeld/repos/pyheat-original/pyheat/pyheat.py", line 100, in __fetch_heatmap_data_from_profile
    line_profiles = self.__get_line_profile_data()
  File "/Users/hhinnefeld/repos/pyheat-original/pyheat/pyheat.py", line 85, in __get_line_profile_data
    return self.line_profiler.file_dict[self.pyfile.path].line_dict
AttributeError: 'list' object has no attribute 'line_dict'

----------------------------------------------------------------------

I'll take a stab at fixing this myself, if that's alright.

pyheat --pyfile <path_to_python_file>

should simply be:

pyheat <path_to_python_file>

Additionally, it could be great to support output to a file:

pyheat --out outpath.png <path_to_python_file>

When the 'heat' gets too hot, and the highlight goes a dark maroon, black text is hardly visible. There should be some threshold that inverts text colour when the 'heat' goes to high (and thus the highlight gets too dark).

Steps to reproduce:

git clone https://github.com/csurfer/pyheat.git
cd pyheat
python3 setup.py install --user
pyheat tests/test_program.py

Results in the following warning on the line which calls set_yticklabels:

/home/aaron/.local/lib/python3.6/site-packages/py_heat-0.0.6-py3.6.egg/pyheat/pyheat.py:158: UserWarning: FixedFormatter should only be used together with FixedLocator

Strangely enough the warning doesn't appear when running the unit tests with:

python3 setup.py test

$ python3.6 -m pip install pyheat
Defaulting to user installation because normal site-packages is not writeable
Collecting pyheat
  Using cached PyHeat-0.2.tar.gz (66 kB)
  Preparing metadata (setup.py) ... done
Building wheels for collected packages: pyheat
  Building wheel for pyheat (setup.py) ... done
  Created wheel for pyheat: filename=PyHeat-0.2-py3-none-any.whl size=6085 sha256=57f1673c93432b413832766cb67b8c56bc84870642eebd77343dd7d7dce54fe4
  Stored in directory: /home/james/.cache/pip/wheels/8f/9b/31/73da16b55b6b6e2f78c2d6a821be9c0ad41ce70fd2dd1e81ab
Successfully built pyheat
Installing collected packages: pyheat
Successfully installed pyheat-0.2
james:aoc$ python3.6 -m pyheat
Traceback (most recent call last):
  File "/usr/lib64/python3.6/runpy.py", line 183, in _run_module_as_main
    mod_name, mod_spec, code = _get_module_details(mod_name, _Error)
  File "/usr/lib64/python3.6/runpy.py", line 142, in _get_module_details
    return _get_module_details(pkg_main_name, error)
  File "/usr/lib64/python3.6/runpy.py", line 109, in _get_module_details
    __import__(pkg_name)
  File "/home/james/.local/lib/python3.6/site-packages/pyheat/__init__.py", line 1, in <module>
    from pyheat.heatmap import HeatMap
  File "/home/james/.local/lib/python3.6/site-packages/pyheat/heatmap.py", line 88
    glEnableClientState(GL_VERTEX_ARRAY)
                                       ^
TabError: inconsistent use of tabs and spaces in indentation

Also there are some old style print var (no brackets) statements in there.

Hi,

I have a customized module xyz.pyd, on windows 10, which my program loads/executes perfectly as it's in the same directory. But pyheat fails to load it. Does it load the modules only from the site-packages?

Regards

Prabhat

I believe a simple web based interface can help visualize larger files better, so instead of showing the matplotlib window it opens the browser and displays a similar heatmap there.